mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
14f040de48
* GetFinalityCheckpoints and GetGenesis * bzl * remove unused func * use const as base path * bring back headers --------- Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
138 lines
5.0 KiB
Go
138 lines
5.0 KiB
Go
package beacon
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/rpc/eth/helpers"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/rpc/lookup"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/eth/v1"
|
|
eth2 "github.com/prysmaticlabs/prysm/v4/proto/eth/v2"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
"go.opencensus.io/trace"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
type stateRequest struct {
|
|
epoch *primitives.Epoch
|
|
stateId []byte
|
|
}
|
|
|
|
// GetStateRoot calculates HashTreeRoot for state with given 'stateId'. If stateId is root, same value will be returned.
|
|
func (bs *Server) GetStateRoot(ctx context.Context, req *ethpb.StateRequest) (*ethpb.StateRootResponse, error) {
|
|
ctx, span := trace.StartSpan(ctx, "beacon.GetStateRoot")
|
|
defer span.End()
|
|
|
|
stateRoot, err := bs.Stater.StateRoot(ctx, req.StateId)
|
|
if err != nil {
|
|
if rootNotFoundErr, ok := err.(*lookup.StateRootNotFoundError); ok {
|
|
return nil, status.Errorf(codes.NotFound, "State root not found: %v", rootNotFoundErr)
|
|
} else if parseErr, ok := err.(*lookup.StateIdParseError); ok {
|
|
return nil, status.Errorf(codes.InvalidArgument, "Invalid state ID: %v", parseErr)
|
|
}
|
|
return nil, status.Errorf(codes.Internal, "Could not get state root: %v", err)
|
|
}
|
|
st, err := bs.Stater.State(ctx, req.StateId)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not get state: %v", err)
|
|
}
|
|
isOptimistic, err := helpers.IsOptimistic(ctx, req.StateId, bs.OptimisticModeFetcher, bs.Stater, bs.ChainInfoFetcher, bs.BeaconDB)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not check if slot's block is optimistic: %v", err)
|
|
}
|
|
blockRoot, err := st.LatestBlockHeader().HashTreeRoot()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not calculate root of latest block header")
|
|
}
|
|
isFinalized := bs.FinalizationFetcher.IsFinalized(ctx, blockRoot)
|
|
|
|
return ðpb.StateRootResponse{
|
|
Data: ðpb.StateRootResponse_StateRoot{
|
|
Root: stateRoot,
|
|
},
|
|
ExecutionOptimistic: isOptimistic,
|
|
Finalized: isFinalized,
|
|
}, nil
|
|
}
|
|
|
|
// GetRandao fetches the RANDAO mix for the requested epoch from the state identified by state_id.
|
|
// If an epoch is not specified then the RANDAO mix for the state's current epoch will be returned.
|
|
// By adjusting the state_id parameter you can query for any historic value of the RANDAO mix.
|
|
// Ordinarily states from the same epoch will mutate the RANDAO mix for that epoch as blocks are applied.
|
|
func (bs *Server) GetRandao(ctx context.Context, req *eth2.RandaoRequest) (*eth2.RandaoResponse, error) {
|
|
ctx, span := trace.StartSpan(ctx, "beacon.GetRandao")
|
|
defer span.End()
|
|
|
|
st, err := bs.Stater.State(ctx, req.StateId)
|
|
if err != nil {
|
|
return nil, helpers.PrepareStateFetchGRPCError(err)
|
|
}
|
|
|
|
stEpoch := slots.ToEpoch(st.Slot())
|
|
epoch := stEpoch
|
|
if req.Epoch != nil {
|
|
epoch = *req.Epoch
|
|
}
|
|
|
|
// future epochs and epochs too far back are not supported.
|
|
randaoEpochLowerBound := uint64(0)
|
|
// Lower bound should not underflow.
|
|
if uint64(stEpoch) > uint64(st.RandaoMixesLength()) {
|
|
randaoEpochLowerBound = uint64(stEpoch) - uint64(st.RandaoMixesLength())
|
|
}
|
|
if epoch > stEpoch || uint64(epoch) < randaoEpochLowerBound+1 {
|
|
return nil, status.Errorf(codes.InvalidArgument, "Epoch is out of range for the randao mixes of the state")
|
|
}
|
|
idx := epoch % params.BeaconConfig().EpochsPerHistoricalVector
|
|
randao, err := st.RandaoMixAtIndex(uint64(idx))
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not get randao mix at index %d", idx)
|
|
}
|
|
|
|
isOptimistic, err := helpers.IsOptimistic(ctx, req.StateId, bs.OptimisticModeFetcher, bs.Stater, bs.ChainInfoFetcher, bs.BeaconDB)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not check if slot's block is optimistic: %v", err)
|
|
}
|
|
|
|
blockRoot, err := st.LatestBlockHeader().HashTreeRoot()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not calculate root of latest block header")
|
|
}
|
|
isFinalized := bs.FinalizationFetcher.IsFinalized(ctx, blockRoot)
|
|
|
|
return ð2.RandaoResponse{
|
|
Data: ð2.RandaoResponse_Randao{Randao: randao},
|
|
ExecutionOptimistic: isOptimistic,
|
|
Finalized: isFinalized,
|
|
}, nil
|
|
}
|
|
|
|
func (bs *Server) stateFromRequest(ctx context.Context, req *stateRequest) (state.BeaconState, error) {
|
|
if req.epoch != nil {
|
|
slot, err := slots.EpochStart(*req.epoch)
|
|
if err != nil {
|
|
return nil, status.Errorf(
|
|
codes.Internal,
|
|
"Could not calculate start slot for epoch %d: %v",
|
|
*req.epoch,
|
|
err,
|
|
)
|
|
}
|
|
st, err := bs.Stater.State(ctx, []byte(strconv.FormatUint(uint64(slot), 10)))
|
|
if err != nil {
|
|
return nil, helpers.PrepareStateFetchGRPCError(err)
|
|
}
|
|
return st, nil
|
|
}
|
|
var err error
|
|
st, err := bs.Stater.State(ctx, req.stateId)
|
|
if err != nil {
|
|
return nil, helpers.PrepareStateFetchGRPCError(err)
|
|
}
|
|
return st, nil
|
|
}
|