mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 11:32:09 +00:00
1112e01c06
* `TestStore_GenesisValidatorsRoot_ReadAndWrite`: Make all test cases independents. In a test with multiple test cases, each test case should be independents. (aka: Removing test case `A` should not impact test case `B`) * `SaveGenesisValidatorsRoot`: Allow to overwrite the genesis validator root if the root is the same. * `ProposalHistoryForSlot`: Add `signingRootExists` Currently, it is not possible with `ProposalHistoryForSlot` to know if a proposal is stored with and `0x00000....` signing root or with an empty signing root. Both cases result to `proposalExists == true` and `signingRoot == 0x00000`. This commit adds a new return boolean: `signingRootExists`. If a proposal has been saved with a `0x00000...` signing root, then: - `proposalExists` is set to `true`, and - `signingRootExists` is set to `true`, and - `signingRoot` is set to `0x00000...` If a proposal has been saved with an empty signing root, then: - `proposalExists` is set to `true`, and - `signingRootExists` is set to `false`, and - (`signingRoot` is set to `0x00000...`) * `ImportStandardProtectionJSON`: When importing EIP-3076 Slashing Protection Interchange Format, do not filter any more slashable keys. Note: Those keys are still saved into the black-listed public keys list. There is two reason not to do so: - The EIP-3076 test cases do not know about Prysm's internal black-listed public keys list. Tests will expect, without looking into this internal black-listed public keys list, to deny a further signature. If we filter these keys from the DB (even if we keep them into the black-listed keys list), then some tests will fail. - If we import a interchange file containing slashable keys and we filter them, then, if we re-export the DB, those slashing offences won't appear in the exported interchange file. * `transformSignedBlocks`: Store an 0-len byte slice When importing an EIP-3076 interchange format, and when no signing root is specified into the file, we currently store a `0x00000.....` signing root. In such a case, instead storing `0x00000...`, this commit stores a 0-len byte array, so we can differentiate real `0x000.....` signing root and no signing-root at all. * `slashableProposalCheck`: Manage lack of sign root Currently, `slashableProposalCheck` does not really make a difference between a `0x0000.....` signing root and a missing signing root. (Signing roots can be missing when importing an EIP-3076 interchange file.) This commit differentiate, for `slashableProposalCheck`, `0x0000....` signing root and a missing signing root. * `AttestationRecord.SigningRoot`: ==> `[]byte` When importing attestations from EIP-3076 interchange format, the signing root of an attestation may be missing. Currently, Prysm consider any missing attestation signing root as `0x000...`. However, it may conflict with signing root which really are equal to `0x000...`. This commit transforms `AttestationRecord.SigningRoot` from `[32]byte` to `[]byte`, and change the minimal set of functions (sic) to support this new type. * `CheckSlashableAttestation`: Empty signing root Regarding slashing roots, 2 attestations are slashable, if: - both signing roots are defined and differs, or - one attestation exists, but without a signing root * `filterSlashablePubKeysFromAttestations`: Err sort Rergarding `CheckSlashableAttestation`, we consider that: - If slashable == NotSlashable and err != nil, then CheckSlashableAttestation failed. - If slashable != NotSlashable, then err contains the reason why the attestation is slashable. * `setupEIP3076SpecTests`: Update to `v5.3.0` This commit: - Updates the version of EIP-3076 tests to `v.5.2.1`. - Setups on anti-slashing DB per test case, instead per step. * `ImportStandardProtectionJSON`: Reduce cycl cmplxt * `AttestationHistoryForPubKey`: copy signing root BoltDB documentation specifies: | Byte slices returned from Bolt are only valid during a transaction. | Once the transaction has been committed or rolled back then the memory | they point to can be reused by a new page or can be unmapped | from virtual memory and you'll see an unexpected fault address panic | when accessing it.
380 lines
14 KiB
Go
380 lines
14 KiB
Go
package history_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/db/kv"
|
|
dbtest "github.com/prysmaticlabs/prysm/v4/validator/db/testing"
|
|
history "github.com/prysmaticlabs/prysm/v4/validator/slashing-protection-history"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/slashing-protection-history/format"
|
|
slashtest "github.com/prysmaticlabs/prysm/v4/validator/testing"
|
|
)
|
|
|
|
func TestImportExport_RoundTrip(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 10
|
|
publicKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, publicKeys)
|
|
|
|
// First we setup some mock attesting and proposal histories and create a mock
|
|
// standard slashing protection format JSON struct.
|
|
attestingHistory, proposalHistory := slashtest.MockAttestingAndProposalHistories(publicKeys)
|
|
require.NoError(t, err)
|
|
wanted, err := slashtest.MockSlashingProtectionJSON(publicKeys, attestingHistory, proposalHistory)
|
|
require.NoError(t, err)
|
|
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(wanted)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
require.NoError(t, err)
|
|
|
|
rawPublicKeys := make([][]byte, numValidators)
|
|
for i := 0; i < numValidators; i++ {
|
|
rawPublicKeys[i] = publicKeys[i][:]
|
|
}
|
|
|
|
// Next up, we export our slashing protection database into the EIP standard file.
|
|
// Next, we attempt to import it into our validator database.
|
|
eipStandard, err := history.ExportStandardProtectionJSON(ctx, validatorDB)
|
|
require.NoError(t, err)
|
|
|
|
// We compare the metadata fields from import to export.
|
|
require.Equal(t, wanted.Metadata, eipStandard.Metadata)
|
|
|
|
// The values in the data field of the EIP struct are not guaranteed to be sorted,
|
|
// so we create a map to verify we have the data we expected.
|
|
require.Equal(t, len(wanted.Data), len(eipStandard.Data))
|
|
|
|
dataByPubKey := make(map[string]*format.ProtectionData)
|
|
for _, item := range wanted.Data {
|
|
dataByPubKey[item.Pubkey] = item
|
|
}
|
|
for _, item := range eipStandard.Data {
|
|
want, ok := dataByPubKey[item.Pubkey]
|
|
require.Equal(t, true, ok)
|
|
require.Equal(t, len(want.SignedAttestations), len(item.SignedAttestations))
|
|
require.Equal(t, len(want.SignedBlocks), len(item.SignedBlocks))
|
|
wantedAttsByRoot := make(map[string]*format.SignedAttestation)
|
|
for _, att := range want.SignedAttestations {
|
|
wantedAttsByRoot[att.SigningRoot] = att
|
|
}
|
|
for _, att := range item.SignedAttestations {
|
|
wantedAtt, ok := wantedAttsByRoot[att.SigningRoot]
|
|
require.Equal(t, true, ok)
|
|
require.DeepEqual(t, wantedAtt, att)
|
|
}
|
|
require.DeepEqual(t, want.SignedBlocks, item.SignedBlocks)
|
|
}
|
|
}
|
|
|
|
func TestImportExport_RoundTrip_SkippedAttestationEpochs(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 1
|
|
pubKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, pubKeys)
|
|
wanted := &format.EIPSlashingProtectionFormat{
|
|
Metadata: struct {
|
|
InterchangeFormatVersion string `json:"interchange_format_version"`
|
|
GenesisValidatorsRoot string `json:"genesis_validators_root"`
|
|
}{
|
|
InterchangeFormatVersion: format.InterchangeFormatVersion,
|
|
GenesisValidatorsRoot: fmt.Sprintf("%#x", [32]byte{1}),
|
|
},
|
|
Data: []*format.ProtectionData{
|
|
{
|
|
Pubkey: fmt.Sprintf("%#x", pubKeys[0]),
|
|
SignedAttestations: []*format.SignedAttestation{
|
|
{
|
|
SourceEpoch: "1",
|
|
TargetEpoch: "2",
|
|
},
|
|
{
|
|
SourceEpoch: "8",
|
|
TargetEpoch: "9",
|
|
},
|
|
},
|
|
SignedBlocks: make([]*format.SignedBlock, 0),
|
|
},
|
|
},
|
|
}
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(wanted)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
require.NoError(t, err)
|
|
|
|
rawPublicKeys := make([][]byte, numValidators)
|
|
for i := 0; i < numValidators; i++ {
|
|
rawPublicKeys[i] = pubKeys[i][:]
|
|
}
|
|
|
|
// Next up, we export our slashing protection database into the EIP standard file.
|
|
// Next, we attempt to import it into our validator database.
|
|
eipStandard, err := history.ExportStandardProtectionJSON(ctx, validatorDB)
|
|
require.NoError(t, err)
|
|
|
|
// We compare the metadata fields from import to export.
|
|
require.Equal(t, wanted.Metadata, eipStandard.Metadata)
|
|
|
|
// The values in the data field of the EIP struct are not guaranteed to be sorted,
|
|
// so we create a map to verify we have the data we expected.
|
|
require.Equal(t, len(wanted.Data), len(eipStandard.Data))
|
|
require.DeepEqual(t, wanted.Data, eipStandard.Data)
|
|
}
|
|
|
|
func TestImportExport_FilterKeys(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 10
|
|
publicKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, publicKeys)
|
|
|
|
// First we setup some mock attesting and proposal histories and create a mock
|
|
// standard slashing protection format JSON struct.
|
|
attestingHistory, proposalHistory := slashtest.MockAttestingAndProposalHistories(publicKeys)
|
|
require.NoError(t, err)
|
|
wanted, err := slashtest.MockSlashingProtectionJSON(publicKeys, attestingHistory, proposalHistory)
|
|
require.NoError(t, err)
|
|
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(wanted)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
require.NoError(t, err)
|
|
|
|
// Next up, we export our slashing protection database into the EIP standard file.
|
|
// Next, we attempt to import it into our validator database.
|
|
rawKeys := make([][]byte, 5)
|
|
for i := 0; i < len(rawKeys); i++ {
|
|
rawKeys[i] = publicKeys[i][:]
|
|
}
|
|
|
|
eipStandard, err := history.ExportStandardProtectionJSON(ctx, validatorDB, rawKeys...)
|
|
require.NoError(t, err)
|
|
|
|
// We compare the metadata fields from import to export.
|
|
require.Equal(t, wanted.Metadata, eipStandard.Metadata)
|
|
require.Equal(t, len(rawKeys), len(eipStandard.Data))
|
|
}
|
|
|
|
func TestImportInterchangeData_OK(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 10
|
|
publicKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, publicKeys)
|
|
|
|
// First we setup some mock attesting and proposal histories and create a mock
|
|
// standard slashing protection format JSON struct.
|
|
attestingHistory, proposalHistory := slashtest.MockAttestingAndProposalHistories(publicKeys)
|
|
require.NoError(t, err)
|
|
standardProtectionFormat, err := slashtest.MockSlashingProtectionJSON(publicKeys, attestingHistory, proposalHistory)
|
|
require.NoError(t, err)
|
|
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(standardProtectionFormat)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
require.NoError(t, err)
|
|
|
|
// Next, we attempt to retrieve the attesting and proposals histories from our database and
|
|
// verify those indeed match the originally generated mock histories.
|
|
for i := 0; i < len(publicKeys); i++ {
|
|
receivedAttestingHistory, err := validatorDB.AttestationHistoryForPubKey(ctx, publicKeys[i])
|
|
require.NoError(t, err)
|
|
|
|
wantedAttsByRoot := make(map[[32]byte]*kv.AttestationRecord)
|
|
for _, att := range attestingHistory[i] {
|
|
var signingRoot [32]byte
|
|
copy(signingRoot[:], att.SigningRoot)
|
|
wantedAttsByRoot[signingRoot] = att
|
|
}
|
|
for _, att := range receivedAttestingHistory {
|
|
var signingRoot [32]byte
|
|
copy(signingRoot[:], att.SigningRoot)
|
|
wantedAtt, ok := wantedAttsByRoot[signingRoot]
|
|
require.Equal(t, true, ok)
|
|
require.DeepEqual(t, wantedAtt, att)
|
|
}
|
|
|
|
proposals := proposalHistory[i].Proposals
|
|
receivedProposalHistory, err := validatorDB.ProposalHistoryForPubKey(ctx, publicKeys[i])
|
|
require.NoError(t, err)
|
|
rootsBySlot := make(map[primitives.Slot][]byte)
|
|
for _, proposal := range receivedProposalHistory {
|
|
rootsBySlot[proposal.Slot] = proposal.SigningRoot
|
|
}
|
|
for _, proposal := range proposals {
|
|
receivedRoot, ok := rootsBySlot[proposal.Slot]
|
|
require.DeepEqual(t, true, ok)
|
|
require.DeepEqual(
|
|
t,
|
|
receivedRoot,
|
|
proposal.SigningRoot,
|
|
"Imported proposals are different then the generated ones",
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestImportInterchangeData_OK_SavesBlacklistedPublicKeys(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 3
|
|
publicKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, publicKeys)
|
|
|
|
// First we setup some mock attesting and proposal histories and create a mock
|
|
// standard slashing protection format JSON struct.
|
|
attestingHistory, proposalHistory := slashtest.MockAttestingAndProposalHistories(publicKeys)
|
|
require.NoError(t, err)
|
|
|
|
standardProtectionFormat, err := slashtest.MockSlashingProtectionJSON(publicKeys, attestingHistory, proposalHistory)
|
|
require.NoError(t, err)
|
|
|
|
// We add a slashable block for public key at index 1.
|
|
pubKey0 := standardProtectionFormat.Data[0].Pubkey
|
|
standardProtectionFormat.Data[0].SignedBlocks = append(
|
|
standardProtectionFormat.Data[0].SignedBlocks,
|
|
&format.SignedBlock{
|
|
Slot: "700",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{1}),
|
|
},
|
|
&format.SignedBlock{
|
|
Slot: "700",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{2}),
|
|
},
|
|
)
|
|
|
|
// We add a slashable attestation for public key at index 1
|
|
// representing a double vote event.
|
|
pubKey1 := standardProtectionFormat.Data[1].Pubkey
|
|
standardProtectionFormat.Data[1].SignedAttestations = append(
|
|
standardProtectionFormat.Data[1].SignedAttestations,
|
|
&format.SignedAttestation{
|
|
TargetEpoch: "700",
|
|
SourceEpoch: "699",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{1}),
|
|
},
|
|
&format.SignedAttestation{
|
|
TargetEpoch: "700",
|
|
SourceEpoch: "699",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{2}),
|
|
},
|
|
)
|
|
|
|
// We add a slashable attestation for public key at index 2
|
|
// representing a surround vote event.
|
|
pubKey2 := standardProtectionFormat.Data[2].Pubkey
|
|
standardProtectionFormat.Data[2].SignedAttestations = append(
|
|
standardProtectionFormat.Data[2].SignedAttestations,
|
|
&format.SignedAttestation{
|
|
TargetEpoch: "800",
|
|
SourceEpoch: "805",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{4}),
|
|
},
|
|
&format.SignedAttestation{
|
|
TargetEpoch: "801",
|
|
SourceEpoch: "804",
|
|
SigningRoot: fmt.Sprintf("%#x", [32]byte{5}),
|
|
},
|
|
)
|
|
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(standardProtectionFormat)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
require.NoError(t, err)
|
|
|
|
// Assert the three slashable keys in the imported JSON were saved to the database.
|
|
sKeys, err := validatorDB.EIPImportBlacklistedPublicKeys(ctx)
|
|
require.NoError(t, err)
|
|
slashableKeys := make(map[string]bool)
|
|
for _, pubKey := range sKeys {
|
|
pkString := fmt.Sprintf("%#x", pubKey)
|
|
slashableKeys[pkString] = true
|
|
}
|
|
ok := slashableKeys[pubKey0]
|
|
assert.Equal(t, true, ok)
|
|
ok = slashableKeys[pubKey1]
|
|
assert.Equal(t, true, ok)
|
|
ok = slashableKeys[pubKey2]
|
|
assert.Equal(t, true, ok)
|
|
}
|
|
|
|
func TestStore_ImportInterchangeData_BadFormat_PreventsDBWrites(t *testing.T) {
|
|
ctx := context.Background()
|
|
numValidators := 5
|
|
publicKeys, err := slashtest.CreateRandomPubKeys(numValidators)
|
|
require.NoError(t, err)
|
|
validatorDB := dbtest.SetupDB(t, publicKeys)
|
|
|
|
// First we setup some mock attesting and proposal histories and create a mock
|
|
// standard slashing protection format JSON struct.
|
|
attestingHistory, proposalHistory := slashtest.MockAttestingAndProposalHistories(publicKeys)
|
|
require.NoError(t, err)
|
|
standardProtectionFormat, err := slashtest.MockSlashingProtectionJSON(publicKeys, attestingHistory, proposalHistory)
|
|
require.NoError(t, err)
|
|
|
|
// We replace a slot of one of the blocks with junk data.
|
|
standardProtectionFormat.Data[0].SignedBlocks[0].Slot = "BadSlot"
|
|
|
|
// We encode the standard slashing protection struct into a JSON format.
|
|
blob, err := json.Marshal(standardProtectionFormat)
|
|
require.NoError(t, err)
|
|
buf := bytes.NewBuffer(blob)
|
|
|
|
// Next, we attempt to import it into our validator database and check that
|
|
// we obtain an error during the import process.
|
|
err = history.ImportStandardProtectionJSON(ctx, validatorDB, buf)
|
|
assert.NotNil(t, err)
|
|
|
|
// Next, we attempt to retrieve the attesting and proposals histories from our database and
|
|
// verify nothing was saved to the DB. If there is an error in the import process, we need to make
|
|
// sure writing is an atomic operation: either the import succeeds and saves the slashing protection
|
|
// data to our DB, or it does not.
|
|
for i := 0; i < len(publicKeys); i++ {
|
|
receivedAttestingHistory, err := validatorDB.AttestationHistoryForPubKey(ctx, publicKeys[i])
|
|
require.NoError(t, err)
|
|
require.Equal(
|
|
t,
|
|
0,
|
|
len(receivedAttestingHistory),
|
|
"Imported attestation protection history is different than the empty default",
|
|
)
|
|
receivedHistory, err := validatorDB.ProposalHistoryForPubKey(ctx, publicKeys[i])
|
|
require.NoError(t, err)
|
|
require.DeepEqual(
|
|
t,
|
|
make([]*kv.Proposal, 0),
|
|
receivedHistory,
|
|
"Imported proposal signing root is different than the empty default",
|
|
)
|
|
}
|
|
}
|