mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 03:30:35 +00:00
214 lines
7.0 KiB
Go
214 lines
7.0 KiB
Go
package accounts
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v5/beacon-chain/core/blocks"
|
|
fieldparams "github.com/prysmaticlabs/prysm/v5/config/fieldparams"
|
|
"github.com/prysmaticlabs/prysm/v5/config/params"
|
|
"github.com/prysmaticlabs/prysm/v5/encoding/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/v5/io/file"
|
|
eth "github.com/prysmaticlabs/prysm/v5/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/client"
|
|
beacon_api "github.com/prysmaticlabs/prysm/v5/validator/client/beacon-api"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/client/iface"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/keymanager"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
// PerformExitCfg for account voluntary exits.
|
|
type PerformExitCfg struct {
|
|
ValidatorClient iface.ValidatorClient
|
|
NodeClient iface.NodeClient
|
|
Keymanager keymanager.IKeymanager
|
|
RawPubKeys [][]byte
|
|
FormattedPubKeys []string
|
|
OutputDirectory string
|
|
}
|
|
|
|
// Exit performs a voluntary exit on one or more accounts.
|
|
func (acm *CLIManager) Exit(ctx context.Context) error {
|
|
// User decided to cancel the voluntary exit.
|
|
if acm.rawPubKeys == nil && acm.formattedPubKeys == nil {
|
|
return nil
|
|
}
|
|
|
|
validatorClient, nodeClient, err := acm.prepareBeaconClients(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if nodeClient == nil {
|
|
return errors.New("could not prepare beacon node client")
|
|
}
|
|
syncStatus, err := (*nodeClient).GetSyncStatus(ctx, &emptypb.Empty{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if syncStatus == nil {
|
|
return errors.New("could not get sync status")
|
|
}
|
|
|
|
if syncStatus.Syncing {
|
|
return errors.New("could not perform exit: beacon node is syncing.")
|
|
}
|
|
|
|
cfg := PerformExitCfg{
|
|
*validatorClient,
|
|
*nodeClient,
|
|
acm.keymanager,
|
|
acm.rawPubKeys,
|
|
acm.formattedPubKeys,
|
|
acm.exitJSONOutputPath,
|
|
}
|
|
rawExitedKeys, trimmedExitedKeys, err := PerformVoluntaryExit(ctx, cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
displayExitInfo(rawExitedKeys, trimmedExitedKeys)
|
|
|
|
return nil
|
|
}
|
|
|
|
// PerformVoluntaryExit uses gRPC clients to submit a voluntary exit message to a beacon node.
|
|
func PerformVoluntaryExit(
|
|
ctx context.Context, cfg PerformExitCfg,
|
|
) (rawExitedKeys [][]byte, formattedExitedKeys []string, err error) {
|
|
var rawNotExitedKeys [][]byte
|
|
genesisResponse, err := cfg.NodeClient.GetGenesis(ctx, &emptypb.Empty{})
|
|
if err != nil {
|
|
log.WithError(err).Errorf("voluntary exit failed: %v", err)
|
|
}
|
|
for i, key := range cfg.RawPubKeys {
|
|
// When output directory is present, only create the signed exit, but do not propose it.
|
|
// Otherwise, propose the exit immediately.
|
|
epoch, err := client.CurrentEpoch(genesisResponse.GenesisTime)
|
|
if err != nil {
|
|
log.WithError(err).Errorf("voluntary exit failed: %v", err)
|
|
}
|
|
if len(cfg.OutputDirectory) > 0 {
|
|
sve, err := client.CreateSignedVoluntaryExit(ctx, cfg.ValidatorClient, cfg.Keymanager.Sign, key, epoch)
|
|
if err != nil {
|
|
rawNotExitedKeys = append(rawNotExitedKeys, key)
|
|
msg := err.Error()
|
|
if strings.Contains(msg, blocks.ValidatorAlreadyExitedMsg) ||
|
|
strings.Contains(msg, blocks.ValidatorCannotExitYetMsg) {
|
|
log.Warningf("Could not create voluntary exit for account %s: %s", cfg.FormattedPubKeys[i], msg)
|
|
} else {
|
|
log.WithError(err).Errorf("voluntary exit failed for account %s", cfg.FormattedPubKeys[i])
|
|
}
|
|
} else if err := writeSignedVoluntaryExitJSON(sve, cfg.OutputDirectory); err != nil {
|
|
log.WithError(err).Error("failed to write voluntary exit")
|
|
}
|
|
} else if err := client.ProposeExit(ctx, cfg.ValidatorClient, cfg.Keymanager.Sign, key, epoch); err != nil {
|
|
rawNotExitedKeys = append(rawNotExitedKeys, key)
|
|
|
|
msg := err.Error()
|
|
if strings.Contains(msg, blocks.ValidatorAlreadyExitedMsg) ||
|
|
strings.Contains(msg, blocks.ValidatorCannotExitYetMsg) {
|
|
log.Warningf("Could not perform voluntary exit for account %s: %s", cfg.FormattedPubKeys[i], msg)
|
|
} else {
|
|
log.WithError(err).Errorf("voluntary exit failed for account %s", cfg.FormattedPubKeys[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
rawExitedKeys = make([][]byte, 0)
|
|
formattedExitedKeys = make([]string, 0)
|
|
for i, key := range cfg.RawPubKeys {
|
|
found := false
|
|
for _, notExited := range rawNotExitedKeys {
|
|
if bytes.Equal(notExited, key) {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
rawExitedKeys = append(rawExitedKeys, key)
|
|
formattedExitedKeys = append(formattedExitedKeys, cfg.FormattedPubKeys[i])
|
|
}
|
|
}
|
|
|
|
return rawExitedKeys, formattedExitedKeys, nil
|
|
}
|
|
|
|
func prepareAllKeys(validatingKeys [][fieldparams.BLSPubkeyLength]byte) (raw [][]byte, formatted []string) {
|
|
raw = make([][]byte, len(validatingKeys))
|
|
formatted = make([]string, len(validatingKeys))
|
|
for i, pk := range validatingKeys {
|
|
raw[i] = make([]byte, len(pk))
|
|
copy(raw[i], pk[:])
|
|
formatted[i] = fmt.Sprintf("%#x", bytesutil.Trunc(pk[:]))
|
|
}
|
|
return
|
|
}
|
|
|
|
func displayExitInfo(rawExitedKeys [][]byte, trimmedExitedKeys []string) {
|
|
if len(rawExitedKeys) > 0 {
|
|
urlFormattedPubKeys := make([]string, len(rawExitedKeys))
|
|
for i, key := range rawExitedKeys {
|
|
urlFormattedPubKeys[i] = formatBeaconChaURL(key)
|
|
}
|
|
|
|
ifaceKeys := make([]interface{}, len(urlFormattedPubKeys))
|
|
for i, k := range urlFormattedPubKeys {
|
|
ifaceKeys[i] = k
|
|
}
|
|
|
|
info := fmt.Sprintf("Voluntary exit was successful for the accounts listed. "+
|
|
"URLs where you can track each validator's exit:\n"+strings.Repeat("%s\n", len(ifaceKeys)), ifaceKeys...)
|
|
|
|
log.WithField("pubkeys", strings.Join(trimmedExitedKeys, ", ")).Info(info)
|
|
} else {
|
|
log.Info("No successful voluntary exits")
|
|
}
|
|
}
|
|
|
|
func formatBeaconChaURL(key []byte) string {
|
|
baseURL := "https://%sbeaconcha.in/validator/%s"
|
|
pulseChainURL := "https://%s.pulsechain.com/validator/%s"
|
|
keyWithout0x := hexutil.Encode(key)[2:]
|
|
|
|
switch env := params.BeaconConfig().ConfigName; env {
|
|
case params.PulseChainName:
|
|
return fmt.Sprintf(pulseChainURL, "beacon", keyWithout0x)
|
|
case params.PulseChainTestnetV4Name:
|
|
return fmt.Sprintf(pulseChainURL, "beacon.v4.testnet", keyWithout0x)
|
|
case params.PraterName, params.GoerliName:
|
|
return fmt.Sprintf(baseURL, "prater.", keyWithout0x)
|
|
case params.HoleskyName:
|
|
return fmt.Sprintf(baseURL, "holesky.", keyWithout0x)
|
|
case params.SepoliaName:
|
|
return fmt.Sprintf(baseURL, "sepolia.", keyWithout0x)
|
|
default:
|
|
return fmt.Sprintf(baseURL, "", keyWithout0x)
|
|
}
|
|
}
|
|
|
|
func writeSignedVoluntaryExitJSON(sve *eth.SignedVoluntaryExit, outputDirectory string) error {
|
|
if err := file.MkdirAll(outputDirectory); err != nil {
|
|
return err
|
|
}
|
|
|
|
jsve := beacon_api.JsonifySignedVoluntaryExits([]*eth.SignedVoluntaryExit{sve})[0]
|
|
b, err := json.Marshal(jsve)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to marshal JSON signed voluntary exit")
|
|
}
|
|
|
|
filepath := path.Join(outputDirectory, fmt.Sprintf("validator-exit-%s.json", jsve.Message.ValidatorIndex))
|
|
if err := file.WriteFile(filepath, b); err != nil {
|
|
return errors.Wrap(err, "failed to write validator exist json")
|
|
}
|
|
|
|
log.Infof("Wrote signed validator exit JSON to %s", filepath)
|
|
|
|
return nil
|
|
}
|