mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
d077483577
* v3 import renamings * tidy * fmt * rev * Update beacon-chain/core/epoch/precompute/reward_penalty_test.go * Update beacon-chain/core/helpers/validators_test.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/iface/BUILD.bazel * Update beacon-chain/db/kv/kv.go * Update beacon-chain/db/kv/state.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/sync/initial-sync/service.go * fix deps * fix bad replacements * fix bad replacements * change back * gohashtree version * fix deps Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Potuz <potuz@prysmaticlabs.com>
90 lines
3.0 KiB
Go
90 lines
3.0 KiB
Go
//go:build go1.18
|
|
|
|
package v2_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
coreState "github.com/prysmaticlabs/prysm/v3/beacon-chain/core/transition"
|
|
native "github.com/prysmaticlabs/prysm/v3/beacon-chain/state/state-native"
|
|
v2 "github.com/prysmaticlabs/prysm/v3/beacon-chain/state/v2"
|
|
types "github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v3/crypto/rand"
|
|
"github.com/prysmaticlabs/prysm/v3/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/util"
|
|
)
|
|
|
|
func FuzzV2StateHashTreeRoot(f *testing.F) {
|
|
gState, _ := util.DeterministicGenesisStateAltair(f, 100)
|
|
output, err := gState.MarshalSSZ()
|
|
assert.NoError(f, err)
|
|
randPool := make([]byte, 100)
|
|
_, err = rand.NewDeterministicGenerator().Read(randPool)
|
|
assert.NoError(f, err)
|
|
f.Add(randPool, uint64(10))
|
|
f.Fuzz(func(t *testing.T, diffBuffer []byte, slotsToTransition uint64) {
|
|
stateSSZ := bytesutil.SafeCopyBytes(output)
|
|
for i := 0; i < len(diffBuffer); i += 9 {
|
|
if i+8 >= len(diffBuffer) {
|
|
return
|
|
}
|
|
num := bytesutil.BytesToUint64BigEndian(diffBuffer[i : i+8])
|
|
num %= uint64(len(diffBuffer))
|
|
// Perform a XOR on the byte of the selected index.
|
|
stateSSZ[num] ^= diffBuffer[i+8]
|
|
}
|
|
pbState := ðpb.BeaconStateAltair{}
|
|
err := pbState.UnmarshalSSZ(stateSSZ)
|
|
if err != nil {
|
|
return
|
|
}
|
|
nativeState, err := native.InitializeFromProtoAltair(pbState)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
slotsToTransition %= 100
|
|
stateObj, err := v2.InitializeFromProtoUnsafe(pbState)
|
|
assert.NoError(t, err)
|
|
for stateObj.Slot() < types.Slot(slotsToTransition) {
|
|
stateObj, err = coreState.ProcessSlots(context.Background(), stateObj, stateObj.Slot()+1)
|
|
assert.NoError(t, err)
|
|
stateObj.Copy()
|
|
|
|
nativeState, err = coreState.ProcessSlots(context.Background(), nativeState, nativeState.Slot()+1)
|
|
assert.NoError(t, err)
|
|
nativeState.Copy()
|
|
}
|
|
assert.NoError(t, err)
|
|
// Perform a cold HTR calculation by initializing a new state.
|
|
innerState, ok := stateObj.InnerStateUnsafe().(*ethpb.BeaconStateAltair)
|
|
assert.Equal(t, true, ok, "inner state is a not a beacon state altair proto")
|
|
newState, err := v2.InitializeFromProtoUnsafe(innerState)
|
|
assert.NoError(t, err)
|
|
|
|
newRt, newErr := newState.HashTreeRoot(context.Background())
|
|
rt, err := stateObj.HashTreeRoot(context.Background())
|
|
nativeRt, nativeErr := nativeState.HashTreeRoot(context.Background())
|
|
assert.Equal(t, newErr != nil, err != nil)
|
|
assert.Equal(t, newErr != nil, nativeErr != nil)
|
|
if err == nil {
|
|
assert.Equal(t, rt, newRt)
|
|
assert.Equal(t, rt, nativeRt)
|
|
}
|
|
|
|
newSSZ, newErr := newState.MarshalSSZ()
|
|
stateObjSSZ, err := stateObj.MarshalSSZ()
|
|
nativeSSZ, nativeErr := nativeState.MarshalSSZ()
|
|
assert.Equal(t, newErr != nil, err != nil)
|
|
assert.Equal(t, newErr != nil, nativeErr != nil)
|
|
if err == nil {
|
|
assert.DeepEqual(t, newSSZ, stateObjSSZ)
|
|
assert.DeepEqual(t, newSSZ, nativeSSZ)
|
|
}
|
|
|
|
})
|
|
}
|