mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-28 14:17:17 +00:00
83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
package powchain
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/trieutil"
|
|
)
|
|
|
|
// processDeposit is a copy of the core function of the same name which includes some optimizations
|
|
// and removes the requirement to pass in beacon state. This is for determining genesis validators.
|
|
func (s *Service) processDeposit(
|
|
eth1Data *ethpb.Eth1Data,
|
|
deposit *ethpb.Deposit,
|
|
) error {
|
|
if err := verifyDeposit(eth1Data, deposit); err != nil {
|
|
return errors.Wrapf(err, "could not verify deposit from #%x", bytesutil.Trunc(deposit.Data.PublicKey))
|
|
}
|
|
pubKey := bytesutil.ToBytes48(deposit.Data.PublicKey)
|
|
amount := deposit.Data.Amount
|
|
currBal, ok := s.depositedPubkeys[pubKey]
|
|
if !ok {
|
|
pub, err := bls.PublicKeyFromBytes(pubKey[:])
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not deserialize validator public key")
|
|
}
|
|
domain := bls.ComputeDomain(params.BeaconConfig().DomainDeposit)
|
|
sig, err := bls.SignatureFromBytes(deposit.Data.Signature)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not convert bytes to signature")
|
|
}
|
|
root, err := ssz.SigningRoot(deposit.Data)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not sign root for deposit data")
|
|
}
|
|
if !sig.Verify(root[:], pub, domain) {
|
|
return fmt.Errorf("deposit signature did not verify")
|
|
}
|
|
s.depositedPubkeys[pubKey] = amount
|
|
|
|
if amount >= params.BeaconConfig().MaxEffectiveBalance {
|
|
s.activeValidatorCount++
|
|
}
|
|
} else {
|
|
newBal := currBal + amount
|
|
s.depositedPubkeys[pubKey] = newBal
|
|
// exit if the validator is already an active validator previously
|
|
if currBal >= params.BeaconConfig().MaxEffectiveBalance {
|
|
return nil
|
|
}
|
|
if newBal >= params.BeaconConfig().MaxEffectiveBalance {
|
|
s.activeValidatorCount++
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func verifyDeposit(eth1Data *ethpb.Eth1Data, deposit *ethpb.Deposit) error {
|
|
// Verify Merkle proof of deposit and deposit trie root.
|
|
receiptRoot := eth1Data.DepositRoot
|
|
leaf, err := ssz.HashTreeRoot(deposit.Data)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not tree hash deposit data")
|
|
}
|
|
if ok := trieutil.VerifyMerkleProof(
|
|
receiptRoot,
|
|
leaf[:],
|
|
int(eth1Data.DepositCount-1),
|
|
deposit.Proof,
|
|
); !ok {
|
|
return fmt.Errorf(
|
|
"deposit merkle branch of deposit root did not verify for root: %#x",
|
|
receiptRoot,
|
|
)
|
|
}
|
|
return nil
|
|
}
|