package validator import ( "context" "strings" "sync" "testing" "time" "github.com/gogo/protobuf/proto" ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1" "github.com/prysmaticlabs/go-ssz" mock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing" "github.com/prysmaticlabs/prysm/beacon-chain/cache" "github.com/prysmaticlabs/prysm/beacon-chain/core/helpers" dbutil "github.com/prysmaticlabs/prysm/beacon-chain/db/testing" "github.com/prysmaticlabs/prysm/beacon-chain/operations/attestations" mockp2p "github.com/prysmaticlabs/prysm/beacon-chain/p2p/testing" beaconstate "github.com/prysmaticlabs/prysm/beacon-chain/state" "github.com/prysmaticlabs/prysm/beacon-chain/state/stategen" "github.com/prysmaticlabs/prysm/beacon-chain/state/stateutil" mockSync "github.com/prysmaticlabs/prysm/beacon-chain/sync/initial-sync/testing" pbp2p "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1" "github.com/prysmaticlabs/prysm/shared/bls" "github.com/prysmaticlabs/prysm/shared/params" "github.com/prysmaticlabs/prysm/shared/roughtime" "github.com/prysmaticlabs/prysm/shared/testutil" "google.golang.org/grpc/status" ) func TestProposeAttestation_OK(t *testing.T) { db := dbutil.SetupDB(t) ctx := context.Background() attesterServer := &Server{ HeadFetcher: &mock.ChainService{}, P2P: &mockp2p.MockBroadcaster{}, BeaconDB: db, AttestationCache: cache.NewAttestationCache(), AttPool: attestations.NewPool(), OperationNotifier: (&mock.ChainService{}).OperationNotifier(), } head := ðpb.SignedBeaconBlock{ Block: ðpb.BeaconBlock{ Slot: 999, ParentRoot: []byte{'a'}, }, } if err := db.SaveBlock(ctx, head); err != nil { t.Fatal(err) } root, err := stateutil.BlockRoot(head.Block) if err != nil { t.Fatal(err) } validators := make([]*ethpb.Validator, 64) for i := 0; i < len(validators); i++ { validators[i] = ðpb.Validator{ ExitEpoch: params.BeaconConfig().FarFutureEpoch, EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance, } } state := testutil.NewBeaconState() if err := state.SetSlot(params.BeaconConfig().SlotsPerEpoch + 1); err != nil { t.Fatal(err) } if err := state.SetValidators(validators); err != nil { t.Fatal(err) } if err := db.SaveState(ctx, state, root); err != nil { t.Fatal(err) } if err := db.SaveHeadBlockRoot(ctx, root); err != nil { t.Fatal(err) } sk := bls.RandKey() sig := sk.Sign([]byte("dummy_test_data")) req := ðpb.Attestation{ Signature: sig.Marshal(), Data: ðpb.AttestationData{ BeaconBlockRoot: root[:], Source: ðpb.Checkpoint{}, Target: ðpb.Checkpoint{}, }, } if _, err := attesterServer.ProposeAttestation(context.Background(), req); err != nil { t.Errorf("Could not attest head correctly: %v", err) } } func TestProposeAttestation_IncorrectSignature(t *testing.T) { db := dbutil.SetupDB(t) attesterServer := &Server{ HeadFetcher: &mock.ChainService{}, P2P: &mockp2p.MockBroadcaster{}, BeaconDB: db, AttestationCache: cache.NewAttestationCache(), AttPool: attestations.NewPool(), OperationNotifier: (&mock.ChainService{}).OperationNotifier(), } req := ðpb.Attestation{ Data: ðpb.AttestationData{ Source: ðpb.Checkpoint{}, Target: ðpb.Checkpoint{}, }, } wanted := "Incorrect attestation signature" if _, err := attesterServer.ProposeAttestation(context.Background(), req); err == nil || !strings.Contains(err.Error(), wanted) { t.Errorf("Did not get wanted error") } } func TestGetAttestationData_OK(t *testing.T) { ctx := context.Background() db := dbutil.SetupDB(t) block := ðpb.BeaconBlock{ Slot: 3*params.BeaconConfig().SlotsPerEpoch + 1, } targetBlock := ðpb.BeaconBlock{ Slot: 1 * params.BeaconConfig().SlotsPerEpoch, } justifiedBlock := ðpb.BeaconBlock{ Slot: 2 * params.BeaconConfig().SlotsPerEpoch, } blockRoot, err := stateutil.BlockRoot(block) if err != nil { t.Fatalf("Could not hash beacon block: %v", err) } justifiedRoot, err := stateutil.BlockRoot(justifiedBlock) if err != nil { t.Fatalf("Could not get signing root for justified block: %v", err) } targetRoot, err := stateutil.BlockRoot(targetBlock) if err != nil { t.Fatalf("Could not get signing root for target block: %v", err) } slot := 3*params.BeaconConfig().SlotsPerEpoch + 1 beaconState := testutil.NewBeaconState() if err := beaconState.SetSlot(slot); err != nil { t.Fatal(err) } if err := beaconState.SetCurrentJustifiedCheckpoint(ðpb.Checkpoint{ Epoch: 2, Root: justifiedRoot[:], }); err != nil { t.Fatal(err) } blockRoots := beaconState.BlockRoots() blockRoots[1] = blockRoot[:] blockRoots[1*params.BeaconConfig().SlotsPerEpoch] = targetRoot[:] blockRoots[2*params.BeaconConfig().SlotsPerEpoch] = justifiedRoot[:] if err := beaconState.SetBlockRoots(blockRoots); err != nil { t.Fatal(err) } chainService := &mock.ChainService{ Genesis: time.Now(), } attesterServer := &Server{ BeaconDB: db, P2P: &mockp2p.MockBroadcaster{}, SyncChecker: &mockSync.Sync{IsSyncing: false}, AttestationCache: cache.NewAttestationCache(), HeadFetcher: &mock.ChainService{ State: beaconState, Root: blockRoot[:], }, FinalizationFetcher: &mock.ChainService{ CurrentJustifiedCheckPoint: beaconState.CurrentJustifiedCheckpoint(), }, GenesisTimeFetcher: &mock.ChainService{ Genesis: time.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second), }, StateNotifier: chainService.StateNotifier(), } if err := db.SaveState(ctx, beaconState, blockRoot); err != nil { t.Fatal(err) } if err := db.SaveBlock(ctx, ðpb.SignedBeaconBlock{Block: block}); err != nil { t.Fatal(err) } if err := db.SaveHeadBlockRoot(ctx, blockRoot); err != nil { t.Fatal(err) } req := ðpb.AttestationDataRequest{ CommitteeIndex: 0, Slot: 3*params.BeaconConfig().SlotsPerEpoch + 1, } res, err := attesterServer.GetAttestationData(context.Background(), req) if err != nil { t.Fatalf("Could not get attestation info at slot: %v", err) } expectedInfo := ðpb.AttestationData{ Slot: 3*params.BeaconConfig().SlotsPerEpoch + 1, BeaconBlockRoot: blockRoot[:], Source: ðpb.Checkpoint{ Epoch: 2, Root: justifiedRoot[:], }, Target: ðpb.Checkpoint{ Epoch: 3, Root: blockRoot[:], }, } if !proto.Equal(res, expectedInfo) { t.Errorf("Expected attestation info to match, received %v, wanted %v", res, expectedInfo) } } func TestGetAttestationData_SyncNotReady(t *testing.T) { as := &Server{ SyncChecker: &mockSync.Sync{IsSyncing: true}, } _, err := as.GetAttestationData(context.Background(), ðpb.AttestationDataRequest{}) if err == nil || strings.Contains(err.Error(), "syncing to latest head") { t.Error("Did not get wanted error") } } func TestAttestationDataAtSlot_HandlesFarAwayJustifiedEpoch(t *testing.T) { // Scenario: // // State slot = 10000 // Last justified slot = epoch start of 1500 // HistoricalRootsLimit = 8192 // // More background: https://github.com/prysmaticlabs/prysm/issues/2153 // This test breaks if it doesnt use mainnet config db := dbutil.SetupDB(t) ctx := context.Background() // Ensure HistoricalRootsLimit matches scenario params.SetupTestConfigCleanup(t) cfg := params.MainnetConfig() cfg.HistoricalRootsLimit = 8192 params.OverrideBeaconConfig(cfg) block := ðpb.BeaconBlock{ Slot: 10000, } epochBoundaryBlock := ðpb.BeaconBlock{ Slot: helpers.StartSlot(helpers.SlotToEpoch(10000)), } justifiedBlock := ðpb.BeaconBlock{ Slot: helpers.StartSlot(helpers.SlotToEpoch(1500)) - 2, // Imagine two skip block } blockRoot, err := stateutil.BlockRoot(block) if err != nil { t.Fatalf("Could not hash beacon block: %v", err) } justifiedBlockRoot, err := stateutil.BlockRoot(justifiedBlock) if err != nil { t.Fatalf("Could not hash justified block: %v", err) } epochBoundaryRoot, err := stateutil.BlockRoot(epochBoundaryBlock) if err != nil { t.Fatalf("Could not hash justified block: %v", err) } slot := uint64(10000) beaconState := testutil.NewBeaconState() if err := beaconState.SetSlot(slot); err != nil { t.Fatal(err) } if err := beaconState.SetCurrentJustifiedCheckpoint(ðpb.Checkpoint{ Epoch: helpers.SlotToEpoch(1500), Root: justifiedBlockRoot[:], }); err != nil { t.Fatal(err) } blockRoots := beaconState.BlockRoots() blockRoots[1] = blockRoot[:] blockRoots[1*params.BeaconConfig().SlotsPerEpoch] = epochBoundaryRoot[:] blockRoots[2*params.BeaconConfig().SlotsPerEpoch] = justifiedBlockRoot[:] if err := beaconState.SetBlockRoots(blockRoots); err != nil { t.Fatal(err) } chainService := &mock.ChainService{ Genesis: time.Now(), } attesterServer := &Server{ BeaconDB: db, P2P: &mockp2p.MockBroadcaster{}, AttestationCache: cache.NewAttestationCache(), HeadFetcher: &mock.ChainService{State: beaconState, Root: blockRoot[:]}, FinalizationFetcher: &mock.ChainService{ CurrentJustifiedCheckPoint: beaconState.CurrentJustifiedCheckpoint(), }, SyncChecker: &mockSync.Sync{IsSyncing: false}, GenesisTimeFetcher: &mock.ChainService{Genesis: time.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second)}, StateNotifier: chainService.StateNotifier(), } if err := db.SaveState(ctx, beaconState, blockRoot); err != nil { t.Fatal(err) } if err := db.SaveBlock(ctx, ðpb.SignedBeaconBlock{Block: block}); err != nil { t.Fatal(err) } if err := db.SaveHeadBlockRoot(ctx, blockRoot); err != nil { t.Fatal(err) } req := ðpb.AttestationDataRequest{ CommitteeIndex: 0, Slot: 10000, } res, err := attesterServer.GetAttestationData(context.Background(), req) if err != nil { t.Fatalf("Could not get attestation info at slot: %v", err) } expectedInfo := ðpb.AttestationData{ Slot: req.Slot, BeaconBlockRoot: blockRoot[:], Source: ðpb.Checkpoint{ Epoch: helpers.SlotToEpoch(1500), Root: justifiedBlockRoot[:], }, Target: ðpb.Checkpoint{ Epoch: 312, Root: blockRoot[:], }, } if !proto.Equal(res, expectedInfo) { t.Errorf("Expected attestation info to match, received %v, wanted %v", res, expectedInfo) } } func TestAttestationDataSlot_handlesInProgressRequest(t *testing.T) { s := &pbp2p.BeaconState{Slot: 100} state, err := beaconstate.InitializeFromProto(s) if err != nil { t.Fatal(err) } ctx := context.Background() chainService := &mock.ChainService{ Genesis: time.Now(), } slot := uint64(2) server := &Server{ HeadFetcher: &mock.ChainService{State: state}, AttestationCache: cache.NewAttestationCache(), SyncChecker: &mockSync.Sync{IsSyncing: false}, GenesisTimeFetcher: &mock.ChainService{Genesis: time.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second)}, StateNotifier: chainService.StateNotifier(), } req := ðpb.AttestationDataRequest{ CommitteeIndex: 1, Slot: slot, } res := ðpb.AttestationData{ Target: ðpb.Checkpoint{Epoch: 55}, } if err := server.AttestationCache.MarkInProgress(req); err != nil { t.Fatal(err) } var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() response, err := server.GetAttestationData(ctx, req) if err != nil { t.Error(err) } if !proto.Equal(res, response) { t.Error("Expected equal responses from cache") } }() wg.Add(1) go func() { defer wg.Done() if err := server.AttestationCache.Put(ctx, req, res); err != nil { t.Error(err) } if err := server.AttestationCache.MarkNotInProgress(req); err != nil { t.Error(err) } }() wg.Wait() } func TestServer_GetAttestationData_InvalidRequestSlot(t *testing.T) { ctx := context.Background() slot := 3*params.BeaconConfig().SlotsPerEpoch + 1 attesterServer := &Server{ SyncChecker: &mockSync.Sync{IsSyncing: false}, GenesisTimeFetcher: &mock.ChainService{Genesis: time.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second)}, } req := ðpb.AttestationDataRequest{ Slot: 1000000000000, } _, err := attesterServer.GetAttestationData(ctx, req) if s, ok := status.FromError(err); !ok || s.Message() != msgInvalidAttestationRequest { t.Fatalf("Wrong error. Wanted %v, got %v", msgInvalidAttestationRequest, err) } } func TestServer_GetAttestationData_HeadStateSlotGreaterThanRequestSlot(t *testing.T) { // There exists a rare scenario where the validator may request an attestation for a slot less // than the head state's slot. The ETH2 spec constraints require that the block root the // attestation is referencing be less than or equal to the attestation data slot. // See: https://github.com/prysmaticlabs/prysm/issues/5164 ctx := context.Background() db := dbutil.SetupDB(t) slot := 3*params.BeaconConfig().SlotsPerEpoch + 1 block := ðpb.BeaconBlock{ Slot: slot, } block2 := ðpb.BeaconBlock{Slot: slot - 1} targetBlock := ðpb.BeaconBlock{ Slot: 1 * params.BeaconConfig().SlotsPerEpoch, } justifiedBlock := ðpb.BeaconBlock{ Slot: 2 * params.BeaconConfig().SlotsPerEpoch, } blockRoot, err := stateutil.BlockRoot(block) if err != nil { t.Fatalf("Could not hash beacon block: %v", err) } blockRoot2, err := ssz.HashTreeRoot(block2) if err != nil { t.Fatal(err) } if err := db.SaveBlock(ctx, ðpb.SignedBeaconBlock{Block: block2}); err != nil { t.Fatal(err) } justifiedRoot, err := stateutil.BlockRoot(justifiedBlock) if err != nil { t.Fatalf("Could not get signing root for justified block: %v", err) } targetRoot, err := stateutil.BlockRoot(targetBlock) if err != nil { t.Fatalf("Could not get signing root for target block: %v", err) } beaconState := testutil.NewBeaconState() if err := beaconState.SetSlot(slot); err != nil { t.Fatal(err) } if err := beaconState.SetGenesisTime(uint64(time.Now().Unix() - int64(slot*params.BeaconConfig().SecondsPerSlot))); err != nil { t.Fatal(err) } if err := beaconState.SetLatestBlockHeader(ðpb.BeaconBlockHeader{ ParentRoot: blockRoot2[:], StateRoot: make([]byte, 32), BodyRoot: make([]byte, 32), }); err != nil { t.Fatal(err) } if err := beaconState.SetCurrentJustifiedCheckpoint(ðpb.Checkpoint{ Epoch: 2, Root: justifiedRoot[:], }); err != nil { t.Fatal(err) } blockRoots := beaconState.BlockRoots() blockRoots[1] = blockRoot[:] blockRoots[1*params.BeaconConfig().SlotsPerEpoch] = targetRoot[:] blockRoots[2*params.BeaconConfig().SlotsPerEpoch] = justifiedRoot[:] blockRoots[3*params.BeaconConfig().SlotsPerEpoch] = blockRoot2[:] if err := beaconState.SetBlockRoots(blockRoots); err != nil { t.Fatal(err) } beaconState2 := beaconState.Copy() if err := beaconState2.SetSlot(beaconState2.Slot() - 1); err != nil { t.Fatal(err) } if err := db.SaveState(ctx, beaconState2, blockRoot2); err != nil { t.Fatal(err) } chainService := &mock.ChainService{ Genesis: time.Now(), } attesterServer := &Server{ BeaconDB: db, P2P: &mockp2p.MockBroadcaster{}, SyncChecker: &mockSync.Sync{IsSyncing: false}, AttestationCache: cache.NewAttestationCache(), HeadFetcher: &mock.ChainService{State: beaconState, Root: blockRoot[:]}, FinalizationFetcher: &mock.ChainService{CurrentJustifiedCheckPoint: beaconState.CurrentJustifiedCheckpoint()}, GenesisTimeFetcher: &mock.ChainService{Genesis: time.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second)}, StateNotifier: chainService.StateNotifier(), StateGen: stategen.New(db, cache.NewStateSummaryCache()), } if err := db.SaveState(ctx, beaconState, blockRoot); err != nil { t.Fatal(err) } if err := db.SaveBlock(ctx, ðpb.SignedBeaconBlock{Block: block}); err != nil { t.Fatal(err) } if err := db.SaveHeadBlockRoot(ctx, blockRoot); err != nil { t.Fatal(err) } req := ðpb.AttestationDataRequest{ CommitteeIndex: 0, Slot: slot - 1, } res, err := attesterServer.GetAttestationData(ctx, req) if err != nil { t.Fatalf("Could not get attestation info at slot: %v", err) } expectedInfo := ðpb.AttestationData{ Slot: slot - 1, BeaconBlockRoot: blockRoot2[:], Source: ðpb.Checkpoint{ Epoch: 2, Root: justifiedRoot[:], }, Target: ðpb.Checkpoint{ Epoch: 3, Root: blockRoot2[:], }, } if !proto.Equal(res, expectedInfo) { t.Errorf("Expected attestation info to match, received %v, wanted %v", res, expectedInfo) } } func TestGetAttestationData_SucceedsInFirstEpoch(t *testing.T) { ctx := context.Background() db := dbutil.SetupDB(t) slot := uint64(5) block := ðpb.BeaconBlock{ Slot: slot, } targetBlock := ðpb.BeaconBlock{ Slot: 0, } justifiedBlock := ðpb.BeaconBlock{ Slot: 0, } blockRoot, err := stateutil.BlockRoot(block) if err != nil { t.Fatalf("Could not hash beacon block: %v", err) } justifiedRoot, err := stateutil.BlockRoot(justifiedBlock) if err != nil { t.Fatalf("Could not get signing root for justified block: %v", err) } targetRoot, err := stateutil.BlockRoot(targetBlock) if err != nil { t.Fatalf("Could not get signing root for target block: %v", err) } beaconState := testutil.NewBeaconState() if err := beaconState.SetSlot(slot); err != nil { t.Fatal(err) } if err := beaconState.SetCurrentJustifiedCheckpoint(ðpb.Checkpoint{ Epoch: 0, Root: justifiedRoot[:], }); err != nil { t.Fatal(err) } blockRoots := beaconState.BlockRoots() blockRoots[1] = blockRoot[:] blockRoots[1*params.BeaconConfig().SlotsPerEpoch] = targetRoot[:] blockRoots[2*params.BeaconConfig().SlotsPerEpoch] = justifiedRoot[:] if err := beaconState.SetBlockRoots(blockRoots); err != nil { t.Fatal(err) } chainService := &mock.ChainService{ Genesis: time.Now(), } attesterServer := &Server{ BeaconDB: db, P2P: &mockp2p.MockBroadcaster{}, SyncChecker: &mockSync.Sync{IsSyncing: false}, AttestationCache: cache.NewAttestationCache(), HeadFetcher: &mock.ChainService{ State: beaconState, Root: blockRoot[:], }, FinalizationFetcher: &mock.ChainService{ CurrentJustifiedCheckPoint: beaconState.CurrentJustifiedCheckpoint(), }, GenesisTimeFetcher: &mock.ChainService{Genesis: roughtime.Now().Add(time.Duration(-1*int64(slot*params.BeaconConfig().SecondsPerSlot)) * time.Second)}, StateNotifier: chainService.StateNotifier(), } if err := db.SaveState(ctx, beaconState, blockRoot); err != nil { t.Fatal(err) } if err := db.SaveBlock(ctx, ðpb.SignedBeaconBlock{Block: block}); err != nil { t.Fatal(err) } if err := db.SaveHeadBlockRoot(ctx, blockRoot); err != nil { t.Fatal(err) } req := ðpb.AttestationDataRequest{ CommitteeIndex: 0, Slot: 5, } res, err := attesterServer.GetAttestationData(context.Background(), req) if err != nil { t.Fatalf("Could not get attestation info at slot: %v", err) } expectedInfo := ðpb.AttestationData{ Slot: slot, BeaconBlockRoot: blockRoot[:], Source: ðpb.Checkpoint{ Epoch: 0, Root: justifiedRoot[:], }, Target: ðpb.Checkpoint{ Epoch: 0, Root: blockRoot[:], }, } if !proto.Equal(res, expectedInfo) { t.Errorf("Expected attestation info to match, received %v, wanted %v", res, expectedInfo) } }