mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-03 08:37:37 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
package stateutil
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/ssz"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
)
|
|
|
|
// PendingAttRootWithHasher describes a method from which the hash tree root
|
|
// of a pending attestation is returned.
|
|
func PendingAttRootWithHasher(att *ethpb.PendingAttestation) ([32]byte, error) {
|
|
var fieldRoots [][32]byte
|
|
|
|
// Bitfield.
|
|
aggregationRoot, err := ssz.BitlistRoot(att.AggregationBits, params.BeaconConfig().MaxValidatorsPerCommittee)
|
|
if err != nil {
|
|
return [32]byte{}, err
|
|
}
|
|
// Attestation data.
|
|
attDataRoot, err := attDataRootWithHasher(att.Data)
|
|
if err != nil {
|
|
return [32]byte{}, err
|
|
}
|
|
inclusionBuf := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(inclusionBuf, uint64(att.InclusionDelay))
|
|
// Inclusion delay.
|
|
inclusionRoot := bytesutil.ToBytes32(inclusionBuf)
|
|
|
|
proposerBuf := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(proposerBuf, uint64(att.ProposerIndex))
|
|
// Proposer index.
|
|
proposerRoot := bytesutil.ToBytes32(proposerBuf)
|
|
|
|
fieldRoots = [][32]byte{aggregationRoot, attDataRoot, inclusionRoot, proposerRoot}
|
|
|
|
return ssz.BitwiseMerkleize(fieldRoots, uint64(len(fieldRoots)), uint64(len(fieldRoots)))
|
|
}
|
|
|
|
func attDataRootWithHasher(data *ethpb.AttestationData) ([32]byte, error) {
|
|
fieldRoots := make([][32]byte, 5)
|
|
|
|
if data != nil {
|
|
// Slot.
|
|
slotBuf := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(slotBuf, uint64(data.Slot))
|
|
fieldRoots[0] = bytesutil.ToBytes32(slotBuf)
|
|
|
|
// CommitteeIndex.
|
|
indexBuf := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(indexBuf, uint64(data.CommitteeIndex))
|
|
fieldRoots[1] = bytesutil.ToBytes32(indexBuf)
|
|
|
|
// Beacon block root.
|
|
fieldRoots[2] = bytesutil.ToBytes32(data.BeaconBlockRoot)
|
|
|
|
// Source
|
|
sourceRoot, err := ssz.CheckpointRoot(data.Source)
|
|
if err != nil {
|
|
return [32]byte{}, errors.Wrap(err, "could not compute source checkpoint merkleization")
|
|
}
|
|
fieldRoots[3] = sourceRoot
|
|
|
|
// Target
|
|
fieldRoots[4], err = ssz.CheckpointRoot(data.Target)
|
|
if err != nil {
|
|
return [32]byte{}, errors.Wrap(err, "could not compute target checkpoint merkleization")
|
|
}
|
|
}
|
|
|
|
return ssz.BitwiseMerkleize(fieldRoots, uint64(len(fieldRoots)), uint64(len(fieldRoots)))
|
|
}
|