mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
83179376d4
* Clean testutil, change tool names to Deterministic * Cleanup errors * Merge branch 'master' of https://github.com/prysmaticlabs/prysm into clean-testutil * Fix bug with generating deposits * Fix a few tests * Fix most tests * Clean up some tests * Remove err pt. 1 * Remove err pt. 2 * Change tests to use genesis state util * Remove err from deposits * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Remove circular dependency * Remove uncompressed signature test * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Merge branch 'master' into clean-testutil * Goimports * gazelle * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Add back error handling * New attestation pool (#4185) * New pool * Better namings * Fmt * Gazelle * Merge branch 'master' of https://github.com/prysmaticlabs/prysm into define-pool * Raul's feedback * Raul's feedback * Log peer conected log for incoming connections (#4173) * Log peer conected log for incoming connections * Merge branch 'master' into peerconnected * Merge branch 'master' into peerconnected * Update handshake.go * Update handshake.go * Merge branch 'master' into peerconnected * Merge branch 'master' into peerconnected * Attestation pool to use go-cache (#4187) * Update EthereumAPIs (#4186) * include new patch targeting latest ethapis master * ensure project builds * Merge branch 'master' into update-all-api * fix up committees * Merge branch 'update-all-api' of github.com:prysmaticlabs/prysm into update-all-api * include latest eth apis * Merge branch 'master' into update-all-api * update block tests * Merge branch 'update-all-api' of github.com:prysmaticlabs/prysm into update-all-api * Merge branch 'master' into update-all-api * add todos * Implement GetValidator RPC Endpoint (#4188) * include new patch targeting latest ethapis master * ensure project builds * Merge branch 'master' into update-all-api * fix up committees * Merge branch 'update-all-api' of github.com:prysmaticlabs/prysm into update-all-api * include latest eth apis * Merge branch 'master' into update-all-api * update block tests * Merge branch 'update-all-api' of github.com:prysmaticlabs/prysm into update-all-api * Merge branch 'master' into update-all-api * add todos * implement get validator rpc * add test for get validator * table driven test * fix up test * fix confs * tests for more cases * fix up tests and add out of range * Slasher optimization (#4172) * size * batching and concurrency improvements * gaz * merge fixes * fix comment * fix test * fix test * fix build * ethpb * ethpb * fix test * fix comment * add benchmark * fix benchmark * Handle error for all testutil uses * Fix errors * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Revert error handling Revert "Fix errors" This reverts commit db081f5486f62c854e7a34e792f5e380cfa922e7. Revert "Handle error for all testutil uses" This reverts commit bdabef230632dd184491c2dad63c7d3e74a3861b. Revert "Add back error handling" This reverts commit da7e3d2020cd906f45f452d4e441b566f2d5c8aa. * Change genesis state func to use testing.T * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Fix conflict * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Merge branch 'master' into clean-testutil * Merge branch 'master' into clean-testutil * Captialize other logs * Merge branch 'clean-testutil' of https://github.com/0xKiwi/Prysm into clean-testutil * Merge branch 'master' of https://github.com/prysmaticlabs/Prysm into clean-testutil * Merge branch 'master' into clean-testutil
194 lines
6.5 KiB
Go
194 lines
6.5 KiB
Go
package testutil
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/pkg/errors"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/state"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/hashutil"
|
|
"github.com/prysmaticlabs/prysm/shared/interop"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/trieutil"
|
|
)
|
|
|
|
var lock sync.Mutex
|
|
|
|
// Caches
|
|
var cachedDeposits []*ethpb.Deposit
|
|
var privKeys []*bls.SecretKey
|
|
var trie *trieutil.MerkleTrie
|
|
|
|
// DeterministicDepositsAndKeys returns the entered amount of deposits and secret keys.
|
|
// The deposits are configured such that for deposit n the validator
|
|
// account is key n and the withdrawal account is key n+1. As such,
|
|
// if all secret keys for n validators are required then numDeposits
|
|
// should be n+1.
|
|
func DeterministicDepositsAndKeys(numDeposits uint64) ([]*ethpb.Deposit, []*bls.SecretKey, error) {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
var err error
|
|
|
|
// Populate trie cache, if not initialized yet.
|
|
if trie == nil {
|
|
trie, err = trieutil.NewTrie(int(params.BeaconConfig().DepositContractTreeDepth))
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "failed to create new trie")
|
|
}
|
|
}
|
|
|
|
// If more deposits requested than cached, generate more.
|
|
if numDeposits > uint64(len(cachedDeposits)) {
|
|
numExisting := uint64(len(cachedDeposits))
|
|
numRequired := numDeposits - uint64(len(cachedDeposits))
|
|
// Fetch the required number of keys.
|
|
secretKeys, publicKeys, err := interop.DeterministicallyGenerateKeys(numExisting, numRequired+1)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "could not create deterministic keys: ")
|
|
}
|
|
privKeys = append(privKeys, secretKeys[:len(secretKeys)-1]...)
|
|
|
|
// Create the new deposits and add them to the trie.
|
|
for i := uint64(0); i < numRequired; i++ {
|
|
withdrawalCreds := hashutil.Hash(publicKeys[i+1].Marshal())
|
|
withdrawalCreds[0] = params.BeaconConfig().BLSWithdrawalPrefixByte
|
|
|
|
depositData := ðpb.Deposit_Data{
|
|
PublicKey: publicKeys[i].Marshal()[:],
|
|
Amount: params.BeaconConfig().MaxEffectiveBalance,
|
|
WithdrawalCredentials: withdrawalCreds[:],
|
|
}
|
|
|
|
domain := bls.ComputeDomain(params.BeaconConfig().DomainDeposit)
|
|
root, err := ssz.SigningRoot(depositData)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "could not get signing root of deposit data")
|
|
}
|
|
depositData.Signature = secretKeys[i].Sign(root[:], domain).Marshal()
|
|
|
|
deposit := ðpb.Deposit{
|
|
Data: depositData,
|
|
}
|
|
cachedDeposits = append(cachedDeposits, deposit)
|
|
|
|
hashedDeposit, err := ssz.HashTreeRoot(deposit.Data)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "could not tree hash deposit data")
|
|
}
|
|
|
|
if err := trie.InsertIntoTrie(hashedDeposit[:], int(numExisting+i)); err != nil {
|
|
return nil, nil, errors.Wrap(err, "could not tree hash deposit data")
|
|
}
|
|
}
|
|
}
|
|
|
|
depositTrie, _, err := DeterministicDepositTrie(int(numDeposits))
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "failed to create deposit trie")
|
|
}
|
|
requestedDeposits := cachedDeposits[:numDeposits]
|
|
for i := range requestedDeposits {
|
|
proof, err := depositTrie.MerkleProof(int(i))
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "could not create merkle proof")
|
|
}
|
|
requestedDeposits[i].Proof = proof
|
|
}
|
|
|
|
return requestedDeposits, privKeys[0:numDeposits], nil
|
|
}
|
|
|
|
// DeterministicDepositTrie returns a merkle trie of the requested size from the
|
|
// deterministic deposits.
|
|
func DeterministicDepositTrie(size int) (*trieutil.MerkleTrie, [][32]byte, error) {
|
|
items := trie.Items()
|
|
if size > len(items) {
|
|
return nil, [][32]byte{}, errors.New("requested a larger tree than amount of deposits")
|
|
}
|
|
|
|
if trie == nil {
|
|
return nil, [][32]byte{}, errors.New("trie cache is empty, generate deposits at an earlier point")
|
|
}
|
|
|
|
items = items[:size]
|
|
depositTrie, err := trieutil.GenerateTrieFromItems(items, int(params.BeaconConfig().DepositContractTreeDepth))
|
|
if err != nil {
|
|
return nil, [][32]byte{}, errors.Wrapf(err, "could not generate trie of %d length", size)
|
|
}
|
|
|
|
roots := make([][32]byte, len(items))
|
|
for i, dep := range items {
|
|
roots[i] = bytesutil.ToBytes32(dep)
|
|
}
|
|
|
|
return depositTrie, roots, nil
|
|
}
|
|
|
|
// DeterministicEth1Data takes an array of deposits and returns the eth1Data made from the deposit trie.
|
|
func DeterministicEth1Data(size int) (*ethpb.Eth1Data, error) {
|
|
depositTrie, _, err := DeterministicDepositTrie(size)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to create trie")
|
|
}
|
|
root := depositTrie.Root()
|
|
eth1Data := ðpb.Eth1Data{
|
|
BlockHash: root[:],
|
|
DepositRoot: root[:],
|
|
DepositCount: uint64(size),
|
|
}
|
|
return eth1Data, nil
|
|
}
|
|
|
|
// DeterministicGenesisState returns a genesis state made using the deterministic deposits.
|
|
func DeterministicGenesisState(t testing.TB, numValidators uint64) (*pb.BeaconState, []*bls.SecretKey) {
|
|
deposits, privKeys, err := DeterministicDepositsAndKeys(numValidators)
|
|
if err != nil {
|
|
t.Fatal(errors.Wrapf(err, "failed to get %d deposits", numValidators))
|
|
}
|
|
eth1Data, err := DeterministicEth1Data(len(deposits))
|
|
if err != nil {
|
|
t.Fatal(errors.Wrapf(err, "failed to get eth1data for %d deposits", numValidators))
|
|
}
|
|
beaconState, err := state.GenesisBeaconState(deposits, uint64(0), eth1Data)
|
|
if err != nil {
|
|
t.Fatal(errors.Wrapf(err, "failed to get genesis beacon state of %d validators", numValidators))
|
|
}
|
|
return beaconState, privKeys
|
|
}
|
|
|
|
// DepositTrieFromDeposits takes an array of deposits and returns the deposit trie.
|
|
func DepositTrieFromDeposits(deposits []*ethpb.Deposit) (*trieutil.MerkleTrie, [][32]byte, error) {
|
|
encodedDeposits := make([][]byte, len(deposits))
|
|
for i := 0; i < len(encodedDeposits); i++ {
|
|
hashedDeposit, err := ssz.HashTreeRoot(deposits[i].Data)
|
|
if err != nil {
|
|
return nil, [][32]byte{}, errors.Wrap(err, "could not tree hash deposit data")
|
|
}
|
|
encodedDeposits[i] = hashedDeposit[:]
|
|
}
|
|
|
|
depositTrie, err := trieutil.GenerateTrieFromItems(encodedDeposits, int(params.BeaconConfig().DepositContractTreeDepth))
|
|
if err != nil {
|
|
return nil, [][32]byte{}, errors.Wrap(err, "Could not generate deposit trie")
|
|
}
|
|
|
|
roots := make([][32]byte, len(deposits))
|
|
for i, dep := range encodedDeposits {
|
|
roots[i] = bytesutil.ToBytes32(dep)
|
|
}
|
|
|
|
return depositTrie, roots, nil
|
|
}
|
|
|
|
// ResetCache clears out the old trie, private keys and deposits.
|
|
func ResetCache() {
|
|
trie = nil
|
|
privKeys = []*bls.SecretKey{}
|
|
cachedDeposits = []*ethpb.Deposit{}
|
|
}
|