mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
2d10bcf179
* move state protos * regen ssz * edit v1 code * fix imports * building * beacon chain builds * validator and shared builds * fuzz builds * changes * spectest builds * tools build * remove import cycle * generate ssz * pcli * gaz * kafka * gaz
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package fuzz
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
stateutil "github.com/prysmaticlabs/prysm/beacon-chain/core/state"
|
|
v1 "github.com/prysmaticlabs/prysm/beacon-chain/state/v1"
|
|
statepb "github.com/prysmaticlabs/prysm/proto/prysm/v2/state"
|
|
"github.com/prysmaticlabs/prysm/shared/featureconfig"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
func init() {
|
|
featureconfig.Init(&featureconfig.Flags{
|
|
EnableSSZCache: false,
|
|
})
|
|
}
|
|
|
|
// BeaconStateFuzz --
|
|
func BeaconStateFuzz(input []byte) {
|
|
params.UseMainnetConfig()
|
|
st := &statepb.BeaconState{}
|
|
if err := st.UnmarshalSSZ(input); err != nil {
|
|
return
|
|
}
|
|
s, err := v1.InitializeFromProtoUnsafe(st)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
validateStateHTR(s)
|
|
nextEpoch := helpers.SlotToEpoch(s.Slot()) + 1
|
|
slot, err := helpers.StartSlot(nextEpoch)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if _, err := stateutil.ProcessSlots(context.Background(), s, slot); err != nil {
|
|
_ = err
|
|
return
|
|
}
|
|
validateStateHTR(s)
|
|
}
|
|
|
|
func validateStateHTR(s *v1.BeaconState) {
|
|
rawState, ok := s.InnerStateUnsafe().(*statepb.BeaconState)
|
|
if !ok {
|
|
panic("non valid type assertion")
|
|
}
|
|
rt, err := s.HashTreeRoot(context.Background())
|
|
nxtRt, err2 := rawState.HashTreeRoot()
|
|
|
|
if err == nil && err2 != nil {
|
|
panic("HTR from state had only and error from cached state HTR method")
|
|
}
|
|
if err != nil && err2 == nil {
|
|
panic("HTR from state had only and error from fast-ssz HTR method")
|
|
}
|
|
if err != nil && err2 != nil {
|
|
return
|
|
}
|
|
if rt != nxtRt {
|
|
panic(fmt.Sprintf("cached HTR gave a root of %#x while fast-ssz gave a root of %#x", rt, nxtRt))
|
|
}
|
|
}
|