done

Author: domiwei

cl/beacon/synced_data/interface.go

@@ -6,6 +6,7 @@ import "github.com/ledgerwatch/erigon/cl/phase1/core/state"
 type SyncedData interface {
 	OnHeadState(newState *state.CachingBeaconState) (err error)
 	HeadState() *state.CachingBeaconState
+	HeadStateReader() state.BeaconStateReader
 	Syncing() bool
 	HeadSlot() uint64
 }

cl/beacon/synced_data/mock_services/synced_data_mock.go

@@ -43,6 +43,10 @@ func (s *SyncedDataManager) HeadState() *state.CachingBeaconState {
 	return nil
 }
 
+func (s *SyncedDataManager) HeadStateReader() state.BeaconStateReader {
+	return s.HeadState()
+}
+
 func (s *SyncedDataManager) Syncing() bool {
 	if !s.enabled {
 		return false

cl/beacon/synced_data/synced_data.go

@@ -0,0 +1,12 @@
+package state
+
+import libcommon "github.com/ledgerwatch/erigon-lib/common"
+
+// BeaconStateReader is an interface for reading the beacon state.
+//
+//go:generate mockgen -destination=./mock_services/beacon_state_reader.go -package=mock_services . BeaconStateReader
+type BeaconStateReader interface {
+	ValidatorPublicKey(index int) (libcommon.Bytes48, error)
+	GetDomain(domainType [4]byte, epoch uint64) ([]byte, error)
+	CommitteeCount(epoch uint64) uint64
+}

cl/phase1/core/state/interface.go

@@ -21,6 +21,8 @@ import (
 var (
 	computeSubnetForAttestation  = subnets.ComputeSubnetForAttestation
 	computeCommitteeCountPerSlot = subnets.ComputeCommitteeCountPerSlot
+	computeSigningRoot           = fork.ComputeSigningRoot
+	blsVerify                    = bls.Verify
 )
 
 type attestationService struct {
@@ -61,7 +63,7 @@ func (s *attestationService) ProcessMessage(ctx context.Context, subnet *uint64,
 		committeeIndex = att.AttestantionData().CommitteeIndex()
 		targetEpoch    = att.AttestantionData().Target().Epoch()
 	)
-	headState := s.syncedDataManager.HeadState()
+	headState := s.syncedDataManager.HeadStateReader()
 	if headState == nil {
 		return ErrIgnore
 	}
@@ -144,11 +146,11 @@ func (s *attestationService) ProcessMessage(ctx context.Context, subnet *uint64,
 	if err != nil {
 		return fmt.Errorf("unable to get the domain: %v", err)
 	}
-	signingRoot, err := fork.ComputeSigningRoot(att.AttestantionData(), domain)
+	signingRoot, err := computeSigningRoot(att.AttestantionData(), domain)
 	if err != nil {
 		return fmt.Errorf("unable to get signing root: %v", err)
 	}
-	if valid, err := bls.Verify(signature[:], signingRoot[:], pubKey[:]); err != nil {
+	if valid, err := blsVerify(signature[:], signingRoot[:], pubKey[:]); err != nil {
 		return err
 	} else if !valid {
 		return fmt.Errorf("invalid signature")