mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-11 12:10:05 +00:00
7aee67af90
* add proto for withdrawal containers * process BLSToExecutionChange * unit tests * go fmt * gaz * unused error * Fix import error * Radek's review * failed test Co-authored-by: terence tsao <terence@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
79 lines
2.8 KiB
Go
79 lines
2.8 KiB
Go
package blocks
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/signing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
"github.com/prysmaticlabs/prysm/crypto/hash/htr"
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/time/slots"
|
|
)
|
|
|
|
const executionToBLSPadding = 12
|
|
|
|
// ProcessBLSToExecutionChange validates a SignedBLSToExecution message and
|
|
// changes the validator's withdrawal address accordingly.
|
|
//
|
|
// Spec pseudocode definition:
|
|
//
|
|
//def process_bls_to_execution_change(state: BeaconState, signed_address_change: SignedBLSToExecutionChange) -> None:
|
|
// validator = state.validators[address_change.validator_index]
|
|
//
|
|
// assert validator.withdrawal_credentials[:1] == BLS_WITHDRAWAL_PREFIX
|
|
// assert validator.withdrawal_credentials[1:] == hash(address_change.from_bls_pubkey)[1:]
|
|
//
|
|
// domain = get_domain(state, DOMAIN_BLS_TO_EXECUTION_CHANGE)
|
|
// signing_root = compute_signing_root(address_change, domain)
|
|
// assert bls.Verify(address_change.from_bls_pubkey, signing_root, signed_address_change.signature)
|
|
//
|
|
// validator.withdrawal_credentials = (
|
|
// ETH1_ADDRESS_WITHDRAWAL_PREFIX
|
|
// + b'\x00' * 11
|
|
// + address_change.to_execution_address
|
|
// )
|
|
//
|
|
func ProcessBLSToExecutionChange(st state.BeaconState, signed *ethpb.SignedBLSToExecutionChange) (state.BeaconState, error) {
|
|
if signed == nil {
|
|
return st, errNilSignedWithdrawalMessage
|
|
}
|
|
message := signed.Message
|
|
if message == nil {
|
|
return st, errNilWithdrawalMessage
|
|
}
|
|
|
|
val, err := st.ValidatorAtIndex(message.ValidatorIndex)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cred := val.WithdrawalCredentials
|
|
if cred[0] != params.BeaconConfig().BLSWithdrawalPrefixByte {
|
|
return nil, errInvalidBLSPrefix
|
|
}
|
|
|
|
// hash the public key and verify it matches the withdrawal credentials
|
|
fromPubkey := message.FromBlsPubkey
|
|
pubkeyChunks := [][32]byte{bytesutil.ToBytes32(fromPubkey[:32]), bytesutil.ToBytes32(fromPubkey[32:])}
|
|
digest := make([][32]byte, 1)
|
|
htr.VectorizedSha256(pubkeyChunks, digest)
|
|
if !bytes.Equal(digest[0][1:], cred[1:]) {
|
|
return nil, errInvalidWithdrawalCredentials
|
|
}
|
|
|
|
epoch := slots.ToEpoch(st.Slot())
|
|
domain, err := signing.Domain(st.Fork(), epoch, params.BeaconConfig().DomainBLSToExecutionChange, st.GenesisValidatorsRoot())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := signing.VerifySigningRoot(message, fromPubkey, signed.Signature, domain); err != nil {
|
|
return nil, signing.ErrSigFailedToVerify
|
|
}
|
|
newCredentials := make([]byte, executionToBLSPadding)
|
|
newCredentials[0] = params.BeaconConfig().ETH1AddressWithdrawalPrefixByte
|
|
val.WithdrawalCredentials = append(newCredentials, message.ToExecutionAddress...)
|
|
err = st.UpdateValidatorAtIndex(message.ValidatorIndex, val)
|
|
return st, err
|
|
}
|