mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-30 15:06:31 +00:00
1ced4754db
* Enhance logging with sig * Fixed * Merge branch 'master' into add-sig * Merge branch 'master' into add-sig * Merge branch 'master' of https://github.com/prysmaticlabs/prysm into add-sig * signature via debug * Merge branch 'add-sig' of https://github.com/prysmaticlabs/prysm into add-sig * Merge branch 'master' into add-sig * Merge branch 'master' into add-sig * Merge branch 'master' into add-sig
84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package aggregator
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/blockchain"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/state"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/sync"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/sirupsen/logrus"
|
|
"go.opencensus.io/trace"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
var log logrus.FieldLogger
|
|
|
|
func init() {
|
|
log = logrus.WithField("prefix", "rpc/aggregator")
|
|
}
|
|
|
|
// Server defines a server implementation of the gRPC aggregator service.
|
|
type Server struct {
|
|
BeaconDB db.Database
|
|
HeadFetcher blockchain.HeadFetcher
|
|
SyncChecker sync.Checker
|
|
}
|
|
|
|
// SubmitAggregateAndProof is called by a validator when its assigned to be an aggregator.
|
|
// The beacon node will broadcast aggregated attestation and proof on the aggregator's behavior.
|
|
func (as *Server) SubmitAggregateAndProof(ctx context.Context, req *pb.AggregationRequest) (*pb.AggregationResponse, error) {
|
|
ctx, span := trace.StartSpan(ctx, "AggregatorServer.SubmitAggregation")
|
|
defer span.End()
|
|
span.AddAttributes(trace.Int64Attribute("slot", int64(req.Slot)))
|
|
|
|
if as.SyncChecker.Syncing() {
|
|
return nil, status.Errorf(codes.Unavailable, "Syncing to latest head, not ready to respond")
|
|
}
|
|
|
|
headState, err := as.HeadFetcher.HeadState(ctx)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not retrieve head state: %v", err)
|
|
}
|
|
headState, err = state.ProcessSlots(ctx, headState, req.Slot)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not process slots up to %d: %v", req.Slot, err)
|
|
}
|
|
|
|
validatorIndex, exists, err := as.BeaconDB.ValidatorIndex(ctx, bytesutil.ToBytes48(req.PublicKey))
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not get validator index from DB: %v", err)
|
|
}
|
|
if !exists {
|
|
return nil, status.Error(codes.Internal, "Could not locate validator index in DB")
|
|
}
|
|
|
|
// Check if the validator is an aggregator
|
|
sig, err := bls.SignatureFromBytes(req.SlotSignature)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not convert signature to byte: %v", err)
|
|
}
|
|
isAggregator, err := helpers.IsAggregator(headState, req.Slot, req.CommitteeIndex, sig)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "Could not get aggregator status: %v", err)
|
|
}
|
|
if !isAggregator {
|
|
return nil, status.Errorf(codes.InvalidArgument, "Validator is not an aggregator")
|
|
}
|
|
|
|
// TODO(3865): Broadcast aggregated attestation & proof via the aggregation topic
|
|
|
|
log.WithFields(logrus.Fields{
|
|
"slot": req.Slot,
|
|
"validatorIndex": validatorIndex,
|
|
"committeeIndex": req.CommitteeIndex,
|
|
}).Debug("Broadcasting aggregated attestation and proof")
|
|
|
|
return &pb.AggregationResponse{}, nil
|
|
}
|