mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
376d248c22
* Add in progress handler for committee cache * Remove debug print * Update validators.go * Fix all the tests * More tests * Update committee_disabled.go * Update committee_disabled.go * Update testing util * Update main.go Co-authored-by: Nishant Das <nishdas93@gmail.com>
96 lines
3.2 KiB
Go
96 lines
3.2 KiB
Go
package debug
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db/filters"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
pbrpc "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/attestation"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// GetBlock in an ssz-encoded format by block root.
|
|
func (ds *Server) GetBlock(
|
|
ctx context.Context,
|
|
req *pbrpc.BlockRequestByRoot,
|
|
) (*pbrpc.SSZResponse, error) {
|
|
root := bytesutil.ToBytes32(req.BlockRoot)
|
|
signedBlock, err := ds.BeaconDB.Block(ctx, root)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not retrieve block by root: %v", err)
|
|
}
|
|
if signedBlock == nil || signedBlock.IsNil() {
|
|
return &pbrpc.SSZResponse{Encoded: make([]byte, 0)}, nil
|
|
}
|
|
encoded, err := signedBlock.MarshalSSZ()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not marshal block: %v", err)
|
|
}
|
|
return &pbrpc.SSZResponse{
|
|
Encoded: encoded,
|
|
}, nil
|
|
}
|
|
|
|
// GetInclusionSlot of an attestation in block.
|
|
func (ds *Server) GetInclusionSlot(ctx context.Context, req *pbrpc.InclusionSlotRequest) (*pbrpc.InclusionSlotResponse, error) {
|
|
ds.GenesisTimeFetcher.CurrentSlot()
|
|
|
|
// Attestation has one epoch to get included in the chain. This blocks users from requesting too soon.
|
|
epochBack := types.Slot(0)
|
|
if ds.GenesisTimeFetcher.CurrentSlot() > params.BeaconConfig().SlotsPerEpoch {
|
|
epochBack = ds.GenesisTimeFetcher.CurrentSlot() - params.BeaconConfig().SlotsPerEpoch
|
|
}
|
|
if epochBack < req.Slot {
|
|
return nil, fmt.Errorf("attestation has one epoch window, please request slot older than %d", epochBack)
|
|
}
|
|
|
|
// Attestation could be in blocks between slot + 1 to slot + epoch_duration.
|
|
startSlot := req.Slot + params.BeaconConfig().MinAttestationInclusionDelay
|
|
endSlot := req.Slot + params.BeaconConfig().SlotsPerEpoch
|
|
|
|
filter := filters.NewFilter().SetStartSlot(startSlot).SetEndSlot(endSlot)
|
|
blks, _, err := ds.BeaconDB.Blocks(ctx, filter)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not retrieve blocks: %v", err)
|
|
}
|
|
|
|
inclusionSlot := types.Slot(1<<64 - 1)
|
|
targetStates := make(map[[32]byte]state.ReadOnlyBeaconState)
|
|
for _, blk := range blks {
|
|
for _, a := range blk.Block().Body().Attestations() {
|
|
tr := bytesutil.ToBytes32(a.Data.Target.Root)
|
|
s, ok := targetStates[tr]
|
|
if !ok {
|
|
s, err = ds.StateGen.StateByRoot(ctx, tr)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not retrieve state: %v", err)
|
|
}
|
|
targetStates[tr] = s
|
|
}
|
|
c, err := helpers.BeaconCommitteeFromState(ctx, s, a.Data.Slot, a.Data.CommitteeIndex)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not get committee: %v", err)
|
|
}
|
|
indices, err := attestation.AttestingIndices(a.AggregationBits, c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, i := range indices {
|
|
if req.Id == i && req.Slot == a.Data.Slot {
|
|
inclusionSlot = blk.Block().Slot()
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return &pbrpc.InclusionSlotResponse{Slot: inclusionSlot}, nil
|
|
}
|