mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
251 lines
9.0 KiB
Go
251 lines
9.0 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
slashpb "github.com/prysmaticlabs/prysm/proto/slashing"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/featureconfig"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
mockSlasher "github.com/prysmaticlabs/prysm/validator/testing"
|
|
)
|
|
|
|
func TestPreSignatureValidation(t *testing.T) {
|
|
config := &featureconfig.Flags{
|
|
SlasherProtection: true,
|
|
}
|
|
reset := featureconfig.InitWithReset(config)
|
|
defer reset()
|
|
validator, _, validatorKey, finish := setup(t)
|
|
defer finish()
|
|
pubKey := [48]byte{}
|
|
copy(pubKey[:], validatorKey.PublicKey().Marshal())
|
|
att := ðpb.IndexedAttestation{
|
|
AttestingIndices: []uint64{1, 2},
|
|
Data: ðpb.AttestationData{
|
|
Slot: 5,
|
|
CommitteeIndex: 2,
|
|
BeaconBlockRoot: []byte("great block"),
|
|
Source: ðpb.Checkpoint{
|
|
Epoch: 4,
|
|
Root: []byte("good source"),
|
|
},
|
|
Target: ðpb.Checkpoint{
|
|
Epoch: 10,
|
|
Root: []byte("good target"),
|
|
},
|
|
},
|
|
}
|
|
mockProtector := &mockSlasher.MockProtector{AllowAttestation: false}
|
|
validator.protector = mockProtector
|
|
err := validator.preAttSignValidations(context.Background(), att, pubKey)
|
|
require.ErrorContains(t, failedPreAttSignExternalErr, err)
|
|
mockProtector.AllowAttestation = true
|
|
err = validator.preAttSignValidations(context.Background(), att, pubKey)
|
|
require.NoError(t, err, "Expected allowed attestation not to throw error")
|
|
}
|
|
|
|
func TestPreSignatureValidation_NilLocal(t *testing.T) {
|
|
config := &featureconfig.Flags{
|
|
SlasherProtection: false,
|
|
}
|
|
reset := featureconfig.InitWithReset(config)
|
|
defer reset()
|
|
validator, _, _, finish := setup(t)
|
|
defer finish()
|
|
att := ðpb.IndexedAttestation{
|
|
AttestingIndices: []uint64{1, 2},
|
|
Data: ðpb.AttestationData{
|
|
Slot: 5,
|
|
CommitteeIndex: 2,
|
|
BeaconBlockRoot: []byte("great block"),
|
|
Source: ðpb.Checkpoint{
|
|
Epoch: 4,
|
|
Root: []byte("good source"),
|
|
},
|
|
Target: ðpb.Checkpoint{
|
|
Epoch: 10,
|
|
Root: []byte("good target"),
|
|
},
|
|
},
|
|
}
|
|
fakePubkey := bytesutil.ToBytes48([]byte("test"))
|
|
err := validator.preAttSignValidations(context.Background(), att, fakePubkey)
|
|
require.NoError(t, err, "Expected allowed attestation not to throw error")
|
|
}
|
|
|
|
func TestPostSignatureUpdate(t *testing.T) {
|
|
config := &featureconfig.Flags{
|
|
SlasherProtection: true,
|
|
}
|
|
reset := featureconfig.InitWithReset(config)
|
|
defer reset()
|
|
validator, _, validatorKey, finish := setup(t)
|
|
defer finish()
|
|
pubKey := [48]byte{}
|
|
copy(pubKey[:], validatorKey.PublicKey().Marshal())
|
|
att := ðpb.IndexedAttestation{
|
|
AttestingIndices: []uint64{1, 2},
|
|
Data: ðpb.AttestationData{
|
|
Slot: 5,
|
|
CommitteeIndex: 2,
|
|
BeaconBlockRoot: []byte("great block"),
|
|
Source: ðpb.Checkpoint{
|
|
Epoch: 4,
|
|
Root: []byte("good source"),
|
|
},
|
|
Target: ðpb.Checkpoint{
|
|
Epoch: 10,
|
|
Root: []byte("good target"),
|
|
},
|
|
},
|
|
}
|
|
mockProtector := &mockSlasher.MockProtector{AllowAttestation: false}
|
|
validator.protector = mockProtector
|
|
err := validator.postAttSignUpdate(context.Background(), att, pubKey)
|
|
require.ErrorContains(t, failedPostAttSignExternalErr, err, "Expected error on post signature update is detected as slashable")
|
|
mockProtector.AllowAttestation = true
|
|
err = validator.postAttSignUpdate(context.Background(), att, pubKey)
|
|
require.NoError(t, err, "Expected allowed attestation not to throw error")
|
|
}
|
|
|
|
func TestPostSignatureUpdate_NilLocal(t *testing.T) {
|
|
config := &featureconfig.Flags{
|
|
SlasherProtection: false,
|
|
}
|
|
reset := featureconfig.InitWithReset(config)
|
|
defer reset()
|
|
validator, _, _, finish := setup(t)
|
|
defer finish()
|
|
att := ðpb.IndexedAttestation{
|
|
AttestingIndices: []uint64{1, 2},
|
|
Data: ðpb.AttestationData{
|
|
Slot: 5,
|
|
CommitteeIndex: 2,
|
|
BeaconBlockRoot: []byte("great block"),
|
|
Source: ðpb.Checkpoint{
|
|
Epoch: 4,
|
|
Root: []byte("good source"),
|
|
},
|
|
Target: ðpb.Checkpoint{
|
|
Epoch: 10,
|
|
Root: []byte("good target"),
|
|
},
|
|
},
|
|
}
|
|
fakePubkey := bytesutil.ToBytes48([]byte("test"))
|
|
err := validator.postAttSignUpdate(context.Background(), att, fakePubkey)
|
|
require.NoError(t, err, "Expected allowed attestation not to throw error")
|
|
}
|
|
|
|
func TestAttestationHistory_BlocksDoubleAttestation(t *testing.T) {
|
|
newMap := make(map[uint64]uint64)
|
|
newMap[0] = params.BeaconConfig().FarFutureEpoch
|
|
attestations := &slashpb.AttestationHistory{
|
|
TargetToSource: newMap,
|
|
LatestEpochWritten: 0,
|
|
}
|
|
|
|
// Mark an attestation spanning epochs 0 to 3.
|
|
newAttSource := uint64(0)
|
|
newAttTarget := uint64(3)
|
|
attestations = markAttestationForTargetEpoch(attestations, newAttSource, newAttTarget)
|
|
require.Equal(t, newAttTarget, attestations.LatestEpochWritten, "Unexpected latest epoch written")
|
|
|
|
// Try an attestation that should be slashable (double att) spanning epochs 1 to 3.
|
|
newAttSource = uint64(1)
|
|
newAttTarget = uint64(3)
|
|
if !isNewAttSlashable(attestations, newAttSource, newAttTarget) {
|
|
t.Fatalf("Expected attestation of source %d and target %d to be considered slashable", newAttSource, newAttTarget)
|
|
}
|
|
}
|
|
|
|
func TestAttestationHistory_Prunes(t *testing.T) {
|
|
wsPeriod := params.BeaconConfig().WeakSubjectivityPeriod
|
|
newMap := make(map[uint64]uint64)
|
|
newMap[0] = params.BeaconConfig().FarFutureEpoch
|
|
attestations := &slashpb.AttestationHistory{
|
|
TargetToSource: newMap,
|
|
LatestEpochWritten: 0,
|
|
}
|
|
|
|
// Try an attestation on totally unmarked history, should not be slashable.
|
|
require.Equal(t, false, isNewAttSlashable(attestations, 0, wsPeriod+5), "Should not be slashable")
|
|
|
|
// Mark attestations spanning epochs 0 to 3 and 6 to 9.
|
|
prunedNewAttSource := uint64(0)
|
|
prunedNewAttTarget := uint64(3)
|
|
attestations = markAttestationForTargetEpoch(attestations, prunedNewAttSource, prunedNewAttTarget)
|
|
newAttSource := prunedNewAttSource + 6
|
|
newAttTarget := prunedNewAttTarget + 6
|
|
attestations = markAttestationForTargetEpoch(attestations, newAttSource, newAttTarget)
|
|
require.Equal(t, newAttTarget, attestations.LatestEpochWritten, "Unexpected latest epoch")
|
|
|
|
// Mark an attestation spanning epochs 54000 to 54003.
|
|
farNewAttSource := newAttSource + wsPeriod
|
|
farNewAttTarget := newAttTarget + wsPeriod
|
|
attestations = markAttestationForTargetEpoch(attestations, farNewAttSource, farNewAttTarget)
|
|
require.Equal(t, farNewAttTarget, attestations.LatestEpochWritten, "Unexpected latest epoch")
|
|
|
|
target := safeTargetToSource(attestations, prunedNewAttTarget)
|
|
require.Equal(t, params.BeaconConfig().FarFutureEpoch, target, "Unexpectedly marked attestation")
|
|
require.Equal(t, farNewAttSource, safeTargetToSource(attestations, farNewAttTarget), "Unexpectedly marked attestation")
|
|
|
|
// Try an attestation from existing source to outside prune, should slash.
|
|
if !isNewAttSlashable(attestations, newAttSource, farNewAttTarget) {
|
|
t.Fatalf("Expected attestation of source %d, target %d to be considered slashable", newAttSource, farNewAttTarget)
|
|
}
|
|
// Try an attestation from before existing target to outside prune, should slash.
|
|
if !isNewAttSlashable(attestations, newAttTarget-1, farNewAttTarget) {
|
|
t.Fatalf("Expected attestation of source %d, target %d to be considered slashable", newAttTarget-1, farNewAttTarget)
|
|
}
|
|
// Try an attestation larger than pruning amount, should slash.
|
|
if !isNewAttSlashable(attestations, 0, farNewAttTarget+5) {
|
|
t.Fatalf("Expected attestation of source 0, target %d to be considered slashable", farNewAttTarget+5)
|
|
}
|
|
}
|
|
|
|
func TestAttestationHistory_BlocksSurroundedAttestation(t *testing.T) {
|
|
newMap := make(map[uint64]uint64)
|
|
newMap[0] = params.BeaconConfig().FarFutureEpoch
|
|
attestations := &slashpb.AttestationHistory{
|
|
TargetToSource: newMap,
|
|
LatestEpochWritten: 0,
|
|
}
|
|
|
|
// Mark an attestation spanning epochs 0 to 3.
|
|
newAttSource := uint64(0)
|
|
newAttTarget := uint64(3)
|
|
attestations = markAttestationForTargetEpoch(attestations, newAttSource, newAttTarget)
|
|
require.Equal(t, newAttTarget, attestations.LatestEpochWritten)
|
|
|
|
// Try an attestation that should be slashable (being surrounded) spanning epochs 1 to 2.
|
|
newAttSource = uint64(1)
|
|
newAttTarget = uint64(2)
|
|
require.Equal(t, true, isNewAttSlashable(attestations, newAttSource, newAttTarget), "Expected slashable attestation")
|
|
}
|
|
|
|
func TestAttestationHistory_BlocksSurroundingAttestation(t *testing.T) {
|
|
newMap := make(map[uint64]uint64)
|
|
newMap[0] = params.BeaconConfig().FarFutureEpoch
|
|
attestations := &slashpb.AttestationHistory{
|
|
TargetToSource: newMap,
|
|
LatestEpochWritten: 0,
|
|
}
|
|
|
|
// Mark an attestation spanning epochs 1 to 2.
|
|
newAttSource := uint64(1)
|
|
newAttTarget := uint64(2)
|
|
attestations = markAttestationForTargetEpoch(attestations, newAttSource, newAttTarget)
|
|
require.Equal(t, newAttTarget, attestations.LatestEpochWritten)
|
|
require.Equal(t, newAttSource, attestations.TargetToSource[newAttTarget])
|
|
|
|
// Try an attestation that should be slashable (surrounding) spanning epochs 0 to 3.
|
|
newAttSource = uint64(0)
|
|
newAttTarget = uint64(3)
|
|
require.Equal(t, true, isNewAttSlashable(attestations, newAttSource, newAttTarget))
|
|
}
|