mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
cb02a6897d
* proto+ssz * refactor GetBlindedBlockSSZ (cherry picked from commit 97483c339f99b0d96bd81846a979383ffd2b0cda) # Conflicts: # beacon-chain/rpc/eth/beacon/blocks.go (cherry picked from commit 9e4e82d2c55e09be7568b28eaa33cdd1141445f4) # Conflicts: # beacon-chain/rpc/eth/beacon/blocks.go * add Capella version (cherry picked from commit 5d6fd0bbe663e5dd16df5b2e773f68982bbcd24e) (cherry picked from commit 82f6ddb693ac9e8b4336b30fae724e478b9e8ec0) * support SSZ lol (cherry picked from commit 52bc2c8d617ac3e1254c493fa053cdce4a1ebd63) (cherry picked from commit d7d70bc25b3ee8acbea10aaf77d26cd1f8c5f26f) * update withdrawals proto * refactor and test GetBlockV2 (cherry picked from commit c1d4eaa79d4df04bd284ec65cf261b6f5f260a97) # Conflicts: # beacon-chain/rpc/eth/beacon/blocks.go * refactor and test GetSSZBlockV2 (cherry picked from commit fbc4e73d31c2f68d55d1e2bb8e7f2d8c7458c0a0) # Conflicts: # beacon-chain/rpc/eth/beacon/blocks.go * test other functions (cherry picked from commit 31d4a4cd1165b882d823696e5983ac6635262ec2) * move stuff to blinded_blocks.go (cherry picked from commit 0a9e1658ddb28f45ae5c1cb9fc2703cbb8c6708d) # Conflicts: # beacon-chain/rpc/eth/beacon/blocks.go * fix migration code * add Capella to SubmitBlock * custom hooks * missing structs * fix tests * fix tests * review * fix build issues * replace ioutil with io Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
116 lines
4.2 KiB
Go
116 lines
4.2 KiB
Go
package state_native
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/v3/config/params"
|
|
types "github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v3/encoding/bytesutil"
|
|
enginev1 "github.com/prysmaticlabs/prysm/v3/proto/engine/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v3/runtime/version"
|
|
"github.com/prysmaticlabs/prysm/v3/time/slots"
|
|
)
|
|
|
|
const ETH1AddressOffset = 12
|
|
|
|
// NextWithdrawalIndex returns the index that will be assigned to the next withdrawal.
|
|
func (b *BeaconState) NextWithdrawalIndex() (uint64, error) {
|
|
if b.version < version.Capella {
|
|
return 0, errNotSupported("NextWithdrawalIndex", b.version)
|
|
}
|
|
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
return b.nextWithdrawalIndex, nil
|
|
}
|
|
|
|
// NextWithdrawalValidatorIndex returns the index of the validator which is
|
|
// next in line for a withdrawal.
|
|
func (b *BeaconState) NextWithdrawalValidatorIndex() (types.ValidatorIndex, error) {
|
|
if b.version < version.Capella {
|
|
return 0, errNotSupported("NextWithdrawalValidatorIndex", b.version)
|
|
}
|
|
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
return b.nextWithdrawalValidatorIndex, nil
|
|
}
|
|
|
|
// ExpectedWithdrawals returns the withdrawals that a proposer will need to pack in the next block
|
|
// applied to the current state. It is also used by validators to check that the execution payload carried
|
|
// the right number of withdrawals
|
|
func (b *BeaconState) ExpectedWithdrawals() ([]*enginev1.Withdrawal, error) {
|
|
if b.version < version.Capella {
|
|
return nil, errNotSupported("ExpectedWithdrawals", b.version)
|
|
}
|
|
|
|
b.lock.RLock()
|
|
defer b.lock.RUnlock()
|
|
|
|
withdrawals := make([]*enginev1.Withdrawal, 0, params.BeaconConfig().MaxWithdrawalsPerPayload)
|
|
validatorIndex := b.nextWithdrawalValidatorIndex
|
|
withdrawalIndex := b.nextWithdrawalIndex
|
|
epoch := slots.ToEpoch(b.slot)
|
|
for range b.validators {
|
|
val := b.validators[validatorIndex]
|
|
balance := b.balances[validatorIndex]
|
|
if balance > 0 && isFullyWithdrawableValidator(val, epoch) {
|
|
withdrawals = append(withdrawals, &enginev1.Withdrawal{
|
|
Index: withdrawalIndex,
|
|
ValidatorIndex: validatorIndex,
|
|
Address: bytesutil.SafeCopyBytes(val.WithdrawalCredentials[ETH1AddressOffset:]),
|
|
Amount: balance,
|
|
})
|
|
withdrawalIndex++
|
|
} else if isPartiallyWithdrawableValidator(val, balance) {
|
|
withdrawals = append(withdrawals, &enginev1.Withdrawal{
|
|
Index: withdrawalIndex,
|
|
ValidatorIndex: validatorIndex,
|
|
Address: bytesutil.SafeCopyBytes(val.WithdrawalCredentials[ETH1AddressOffset:]),
|
|
Amount: balance - params.BeaconConfig().MaxEffectiveBalance,
|
|
})
|
|
withdrawalIndex++
|
|
}
|
|
if uint64(len(withdrawals)) == params.BeaconConfig().MaxWithdrawalsPerPayload {
|
|
break
|
|
}
|
|
validatorIndex += 1
|
|
if uint64(validatorIndex) == uint64(len(b.validators)) {
|
|
validatorIndex = 0
|
|
}
|
|
}
|
|
return withdrawals, nil
|
|
}
|
|
|
|
// hasETH1WithdrawalCredential returns whether the validator has an ETH1
|
|
// Withdrawal prefix. It assumes that the caller has a lock on the state
|
|
func hasETH1WithdrawalCredential(val *ethpb.Validator) bool {
|
|
if val == nil {
|
|
return false
|
|
}
|
|
cred := val.WithdrawalCredentials
|
|
return len(cred) > 0 && cred[0] == params.BeaconConfig().ETH1AddressWithdrawalPrefixByte
|
|
}
|
|
|
|
// isFullyWithdrawableValidator returns whether the validator is able to perform a full
|
|
// withdrawal. This differ from the spec helper in that the balance > 0 is not
|
|
// checked. This function assumes that the caller holds a lock on the state
|
|
func isFullyWithdrawableValidator(val *ethpb.Validator, epoch types.Epoch) bool {
|
|
if val == nil {
|
|
return false
|
|
}
|
|
return hasETH1WithdrawalCredential(val) && val.WithdrawableEpoch <= epoch
|
|
}
|
|
|
|
// isPartiallyWithdrawable returns whether the validator is able to perform a
|
|
// partial withdrawal. This function assumes that the caller has a lock on the state
|
|
func isPartiallyWithdrawableValidator(val *ethpb.Validator, balance uint64) bool {
|
|
if val == nil {
|
|
return false
|
|
}
|
|
hasMaxBalance := val.EffectiveBalance == params.BeaconConfig().MaxEffectiveBalance
|
|
hasExcessBalance := balance > params.BeaconConfig().MaxEffectiveBalance
|
|
return hasETH1WithdrawalCredential(val) && hasExcessBalance && hasMaxBalance
|
|
}
|