mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57: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>
109 lines
3.6 KiB
Go
109 lines
3.6 KiB
Go
package state_native
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/state/fieldtrie"
|
|
nativetypes "github.com/prysmaticlabs/prysm/v3/beacon-chain/state/state-native/types"
|
|
"github.com/prysmaticlabs/prysm/v3/encoding/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/v3/runtime/version"
|
|
)
|
|
|
|
const (
|
|
finalizedRootIndex = uint64(105) // Precomputed value.
|
|
)
|
|
|
|
// FinalizedRootGeneralizedIndex for the beacon state.
|
|
func FinalizedRootGeneralizedIndex() uint64 {
|
|
return finalizedRootIndex
|
|
}
|
|
|
|
// CurrentSyncCommitteeGeneralizedIndex for the beacon state.
|
|
func (b *BeaconState) CurrentSyncCommitteeGeneralizedIndex() (uint64, error) {
|
|
if b.version == version.Phase0 {
|
|
return 0, errNotSupported("CurrentSyncCommitteeGeneralizedIndex", b.version)
|
|
}
|
|
|
|
return uint64(nativetypes.CurrentSyncCommittee.RealPosition()), nil
|
|
}
|
|
|
|
// NextSyncCommitteeGeneralizedIndex for the beacon state.
|
|
func (b *BeaconState) NextSyncCommitteeGeneralizedIndex() (uint64, error) {
|
|
if b.version == version.Phase0 {
|
|
return 0, errNotSupported("NextSyncCommitteeGeneralizedIndex", b.version)
|
|
}
|
|
|
|
return uint64(nativetypes.NextSyncCommittee.RealPosition()), nil
|
|
}
|
|
|
|
// CurrentSyncCommitteeProof from the state's Merkle trie representation.
|
|
func (b *BeaconState) CurrentSyncCommitteeProof(ctx context.Context) ([][]byte, error) {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version == version.Phase0 {
|
|
return nil, errNotSupported("CurrentSyncCommitteeProof", b.version)
|
|
}
|
|
|
|
// In case the Merkle layers of the trie are not populated, we need
|
|
// to perform some initialization.
|
|
if err := b.initializeMerkleLayers(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
// Our beacon state uses a "dirty" fields pattern which requires us to
|
|
// recompute branches of the Merkle layers that are marked as dirty.
|
|
if err := b.recomputeDirtyFields(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
return fieldtrie.ProofFromMerkleLayers(b.merkleLayers, nativetypes.CurrentSyncCommittee.RealPosition()), nil
|
|
}
|
|
|
|
// NextSyncCommitteeProof from the state's Merkle trie representation.
|
|
func (b *BeaconState) NextSyncCommitteeProof(ctx context.Context) ([][]byte, error) {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version == version.Phase0 {
|
|
return nil, errNotSupported("NextSyncCommitteeProof", b.version)
|
|
}
|
|
|
|
if err := b.initializeMerkleLayers(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := b.recomputeDirtyFields(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
return fieldtrie.ProofFromMerkleLayers(b.merkleLayers, nativetypes.NextSyncCommittee.RealPosition()), nil
|
|
}
|
|
|
|
// FinalizedRootProof crafts a Merkle proof for the finalized root
|
|
// contained within the finalized checkpoint of a beacon state.
|
|
func (b *BeaconState) FinalizedRootProof(ctx context.Context) ([][]byte, error) {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version == version.Phase0 {
|
|
return nil, errNotSupported("FinalizedRootProof", b.version)
|
|
}
|
|
|
|
if err := b.initializeMerkleLayers(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := b.recomputeDirtyFields(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
cpt := b.finalizedCheckpointVal()
|
|
// The epoch field of a finalized checkpoint is the neighbor
|
|
// index of the finalized root field in its Merkle tree representation
|
|
// of the checkpoint. This neighbor is the first element added to the proof.
|
|
epochBuf := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(epochBuf, uint64(cpt.Epoch))
|
|
epochRoot := bytesutil.ToBytes32(epochBuf)
|
|
proof := make([][]byte, 0)
|
|
proof = append(proof, epochRoot[:])
|
|
branch := fieldtrie.ProofFromMerkleLayers(b.merkleLayers, nativetypes.FinalizedCheckpoint.RealPosition())
|
|
proof = append(proof, branch...)
|
|
return proof, nil
|
|
}
|