mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
21bdbd548a
* v0 * getters/setters * init and copy * hasher * all the nice stuff * make bazel happy * remove tests for smaller PR * remove old states * move files * import fixes * custom MarshalSSZ * fixed deadlock * copy version when copying state * correct issues in state_trie * fix Copy() * better e2e comment * add code to minimal state * spectest test * Revert "Auxiliary commit to revert individual files from 84154423464e8372f7e0a03367403656ac5cd78e" This reverts commit 9602599d183081291dfa0ba4f1036430f63a7822. * native state assert * always error * always log * more native state usage * cleanup * remove empty line * Revert "spectests" This reverts commit 1c49bed5d1cf6224afaf21e18562bf72fae5d2b6. # Conflicts: # beacon-chain/powchain/service.go # beacon-chain/state/v1/state_trie.go # beacon-chain/state/v2/state_trie.go # beacon-chain/state/v3/state_trie.go # testing/spectest/shared/phase0/finality/BUILD.bazel # testing/spectest/shared/phase0/finality/runner.go * dedup field trie * fix test issues * cleanup * use correct field num in FinalizedRootProof * use existing version constant * halfway there * "working" version * some fixes * fix field nums in tests * rename v0types to nativetypes * Revert "Auxiliary commit to revert individual files from dc549b1cf8e724bd08cee1ecc760ff3771d5592d" This reverts commit 7254d3070d8693b283fc686a2e01a822ecbac1b3. * uncomment code * remove map size * Revert "Revert "spectests"" This reverts commit 39c271ae6b57d6a3737e2c202cd8407857475e56. * use reverse map * Revert "Revert "Revert "spectests""" This reverts commit 19ba8cf95cdca689357c8234a262e08cccbafef4. * finally found the bug (cherry picked from commit a5414c4be1bdb61a50b391ea5301895e772cc5e9) * simplify populateFieldIndexes * fix copy (cherry picked from commit 7da4fb8cf51557ef931bb781872ea52fc6731af5) * remove native state from e2e * remove index map * unsupported functions * Use ProtobufBeaconState() from native state * tests * typo * reduce complexity of `SaveStatesEfficient` * remove unused receiver name * update doc.go * fix test assertion * fix test assertion 2 * Phase0 justification bits * bring back state tests * rename fieldIndexRev * versioning of ToProto * remove version check from unexported function * hasher tests * don't return error from JustificationBits * extract fieldConvertersNative * helper error function * use fieldConvertersNative * Introduce RealPosition method on FieldIndex * use RealPosition in hasher * remove unused fields * remove TestAppendBeyondIndicesLimit (cherry picked from commit 3017e700282969c30006b64c95c21ffe6b166f8b) * simplify RealPosition * rename field interface * use helper in proofs.go * Update beacon-chain/core/altair/upgrade.go Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
104 lines
3.9 KiB
Go
104 lines
3.9 KiB
Go
package state_native
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
nativetypes "github.com/prysmaticlabs/prysm/beacon-chain/state/state-native/types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stateutil"
|
|
fieldparams "github.com/prysmaticlabs/prysm/config/fieldparams"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/runtime/version"
|
|
)
|
|
|
|
// RotateAttestations sets the previous epoch attestations to the current epoch attestations and
|
|
// then clears the current epoch attestations.
|
|
func (b *BeaconState) RotateAttestations() error {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version != version.Phase0 {
|
|
return errNotSupported("RotateAttestations", b.version)
|
|
}
|
|
|
|
b.setPreviousEpochAttestations(b.currentEpochAttestationsVal())
|
|
b.setCurrentEpochAttestations([]*ethpb.PendingAttestation{})
|
|
return nil
|
|
}
|
|
|
|
func (b *BeaconState) setPreviousEpochAttestations(val []*ethpb.PendingAttestation) {
|
|
b.sharedFieldReferences[nativetypes.PreviousEpochAttestations].MinusRef()
|
|
b.sharedFieldReferences[nativetypes.PreviousEpochAttestations] = stateutil.NewRef(1)
|
|
|
|
b.previousEpochAttestations = val
|
|
b.markFieldAsDirty(nativetypes.PreviousEpochAttestations)
|
|
b.rebuildTrie[nativetypes.PreviousEpochAttestations] = true
|
|
}
|
|
|
|
func (b *BeaconState) setCurrentEpochAttestations(val []*ethpb.PendingAttestation) {
|
|
b.sharedFieldReferences[nativetypes.CurrentEpochAttestations].MinusRef()
|
|
b.sharedFieldReferences[nativetypes.CurrentEpochAttestations] = stateutil.NewRef(1)
|
|
|
|
b.currentEpochAttestations = val
|
|
b.markFieldAsDirty(nativetypes.CurrentEpochAttestations)
|
|
b.rebuildTrie[nativetypes.CurrentEpochAttestations] = true
|
|
}
|
|
|
|
// AppendCurrentEpochAttestations for the beacon state. Appends the new value
|
|
// to the the end of list.
|
|
func (b *BeaconState) AppendCurrentEpochAttestations(val *ethpb.PendingAttestation) error {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version != version.Phase0 {
|
|
return errNotSupported("AppendCurrentEpochAttestations", b.version)
|
|
}
|
|
|
|
atts := b.currentEpochAttestations
|
|
max := uint64(fieldparams.CurrentEpochAttestationsLength)
|
|
if uint64(len(atts)) >= max {
|
|
return fmt.Errorf("current pending attestation exceeds max length %d", max)
|
|
}
|
|
|
|
if b.sharedFieldReferences[nativetypes.CurrentEpochAttestations].Refs() > 1 {
|
|
// Copy elements in underlying array by reference.
|
|
atts = make([]*ethpb.PendingAttestation, len(b.currentEpochAttestations))
|
|
copy(atts, b.currentEpochAttestations)
|
|
b.sharedFieldReferences[nativetypes.CurrentEpochAttestations].MinusRef()
|
|
b.sharedFieldReferences[nativetypes.CurrentEpochAttestations] = stateutil.NewRef(1)
|
|
}
|
|
|
|
b.currentEpochAttestations = append(atts, val)
|
|
b.markFieldAsDirty(nativetypes.CurrentEpochAttestations)
|
|
b.addDirtyIndices(nativetypes.CurrentEpochAttestations, []uint64{uint64(len(b.currentEpochAttestations) - 1)})
|
|
return nil
|
|
}
|
|
|
|
// AppendPreviousEpochAttestations for the beacon state. Appends the new value
|
|
// to the the end of list.
|
|
func (b *BeaconState) AppendPreviousEpochAttestations(val *ethpb.PendingAttestation) error {
|
|
b.lock.Lock()
|
|
defer b.lock.Unlock()
|
|
|
|
if b.version != version.Phase0 {
|
|
return errNotSupported("AppendPreviousEpochAttestations", b.version)
|
|
}
|
|
|
|
atts := b.previousEpochAttestations
|
|
max := uint64(fieldparams.PreviousEpochAttestationsLength)
|
|
if uint64(len(atts)) >= max {
|
|
return fmt.Errorf("previous pending attestation exceeds max length %d", max)
|
|
}
|
|
|
|
if b.sharedFieldReferences[nativetypes.PreviousEpochAttestations].Refs() > 1 {
|
|
atts = make([]*ethpb.PendingAttestation, len(b.previousEpochAttestations))
|
|
copy(atts, b.previousEpochAttestations)
|
|
b.sharedFieldReferences[nativetypes.PreviousEpochAttestations].MinusRef()
|
|
b.sharedFieldReferences[nativetypes.PreviousEpochAttestations] = stateutil.NewRef(1)
|
|
}
|
|
|
|
b.previousEpochAttestations = append(atts, val)
|
|
b.markFieldAsDirty(nativetypes.PreviousEpochAttestations)
|
|
b.addDirtyIndices(nativetypes.PreviousEpochAttestations, []uint64{uint64(len(b.previousEpochAttestations) - 1)})
|
|
return nil
|
|
}
|