mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
b4e89fb28b
* add it in * build tags * gaz * remove dependency on fast-ssz for now * copy and comment * add in native state * fix build * no assert on invalid input, just return * Add failing case * fix it up Co-authored-by: prestonvanloon <preston@prysmaticlabs.com>
91 lines
3.0 KiB
Go
91 lines
3.0 KiB
Go
//go:build go1.18
|
|
// +build go1.18
|
|
|
|
package v2_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
coreState "github.com/prysmaticlabs/prysm/beacon-chain/core/transition"
|
|
native "github.com/prysmaticlabs/prysm/beacon-chain/state/state-native"
|
|
v2 "github.com/prysmaticlabs/prysm/beacon-chain/state/v2"
|
|
types "github.com/prysmaticlabs/prysm/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/crypto/rand"
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/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)
|
|
}
|
|
|
|
})
|
|
}
|