mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-28 22:27:17 +00:00
ce65b11801
* beacon attestation subscriber * register beacon attestation handler * fix tests
40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/karlseguin/ccache"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
)
|
|
|
|
// recentlySeenBlockRoots cache with max size of ~3Mib
|
|
var recentlySeenRoots = ccache.New(ccache.Configure().MaxSize(100000))
|
|
|
|
// validateBeaconBlockPubSub checks that the incoming block has a valid BLS signature.
|
|
// Blocks that have already been seen are ignored. If the BLS signature is any valid signature,
|
|
// this method rebroadcasts the message.
|
|
func (r *RegularSync) validateBeaconBlockPubSub(ctx context.Context, msg proto.Message, p p2p.Broadcaster) bool {
|
|
m := msg.(*ethpb.BeaconBlock)
|
|
|
|
blockRoot, err := ssz.SigningRoot(m)
|
|
if err != nil {
|
|
log.WithField("validate", "beacon block").WithError(err).Error("Failed to get signing root of block")
|
|
return false
|
|
}
|
|
if recentlySeenRoots.Get(string(blockRoot[:])) != nil || r.db.HasBlock(ctx, blockRoot) {
|
|
return false
|
|
}
|
|
recentlySeenRoots.Set(string(blockRoot[:]), true /*value*/, 365*24*time.Hour /*TTL*/)
|
|
|
|
_, err = bls.SignatureFromBytes(m.Signature)
|
|
if err == nil {
|
|
p.Broadcast(ctx, m)
|
|
}
|
|
return err == nil
|
|
}
|