mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
49a0d3caf0
* Fix a few deps to work with go.mod, check in generated files * Update Gossipsub to 1.1 (#5998) * update libs * add new validators * add new deps * new set of deps * tls * further fix gossip update * get everything to build * clean up * gaz * fix build * fix all tests * add deps to images * imports Co-authored-by: rauljordan <raul@prysmaticlabs.com> * Beacon chain builds with go build * fix bazel * fix dep * lint * Add github action for testing go * on PR for any branch * fix libp2p test failure * Fix TestProcessBlock_PassesProcessingConditions by updating the proposer index in test * Revert "Fix TestProcessBlock_PassesProcessingConditions by updating the proposer index in test" This reverts commit 43676894ab01f03fe90a9b8ee3ecfbc2ec1ec4e4. * Compute and set proposer index instead of hard code * Add back go mod/sum, fix deps * go build ./... * Temporarily skip two tests * Fix kafka confluent patch * Fix kafka confluent patch * fix kafka build * fix kafka * Add info in DEPENDENCIES. Added a stub link for Why Bazel? until https://github.com/prysmaticlabs/documentation/issues/138 * Update fuzz ssz files as well * Update fuzz ssz files as well * getting closer * rollback rules_go and gazelle * fix gogo protobuf * install librdkafka-dev as part of github actions * Update kafka to a recent version where librkafkfa is not required for go modules * clarify comment * fix kafka build * disable go tests * comment * Fix geth dependencies for end to end * rename word * lint * fix docker Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: rauljordan <raul@prysmaticlabs.com> Co-authored-by: terence tsao <terence@prysmaticlabs.com>
220 lines
5.9 KiB
Go
220 lines
5.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"math/big"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/accounts/keystore"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/ethclient"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
"github.com/pkg/errors"
|
|
contracts "github.com/prysmaticlabs/prysm/contracts/deposit-contract"
|
|
"github.com/prysmaticlabs/prysm/shared/version"
|
|
"github.com/sirupsen/logrus"
|
|
prefixed "github.com/x-cray/logrus-prefixed-formatter"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func main() {
|
|
var keystoreUTCPath string
|
|
var passwordFile string
|
|
var httpPath string
|
|
var privKeyString string
|
|
|
|
customFormatter := new(prefixed.TextFormatter)
|
|
customFormatter.TimestampFormat = "2006-01-02 15:04:05"
|
|
customFormatter.FullTimestamp = true
|
|
logrus.SetFormatter(customFormatter)
|
|
|
|
app := cli.App{}
|
|
app.Name = "drainContracts"
|
|
app.Usage = "this is a util to drain all (testing) deposit contracts of their ETH."
|
|
app.Version = version.GetVersion()
|
|
app.Flags = []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "keystoreUTCPath",
|
|
Usage: "Location of keystore",
|
|
Destination: &keystoreUTCPath,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "httpPath",
|
|
Value: "https://goerli.infura.io/v3/be3fb7ed377c418087602876a40affa1",
|
|
Usage: "HTTP-RPC server listening interface",
|
|
Destination: &httpPath,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "passwordFile",
|
|
Value: "./password.txt",
|
|
Usage: "Password file for unlock account",
|
|
Destination: &passwordFile,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "privKey",
|
|
Usage: "Private key to send ETH transaction",
|
|
Destination: &privKeyString,
|
|
},
|
|
}
|
|
|
|
app.Action = func(c *cli.Context) error {
|
|
// Set up RPC client
|
|
var rpcClient *rpc.Client
|
|
var err error
|
|
var txOps *bind.TransactOpts
|
|
|
|
// Uses HTTP-RPC if IPC is not set
|
|
rpcClient, err = rpc.Dial(httpPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := ethclient.NewClient(rpcClient)
|
|
|
|
// User inputs private key, sign tx with private key
|
|
if privKeyString != "" {
|
|
privKey, err := crypto.HexToECDSA(privKeyString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
txOps = bind.NewKeyedTransactor(privKey)
|
|
txOps.Value = big.NewInt(0)
|
|
txOps.GasLimit = 4000000
|
|
nonce, err := client.NonceAt(context.Background(), crypto.PubkeyToAddress(privKey.PublicKey), nil)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not get account nonce")
|
|
}
|
|
txOps.Nonce = big.NewInt(int64(nonce))
|
|
fmt.Printf("current address is %s\n", crypto.PubkeyToAddress(privKey.PublicKey).String())
|
|
fmt.Printf("nonce is %d\n", nonce)
|
|
// User inputs keystore json file, sign tx with keystore json
|
|
} else {
|
|
password := loadTextFromFile(passwordFile)
|
|
|
|
// #nosec - Inclusion of file via variable is OK for this tool.
|
|
keyJSON, err := ioutil.ReadFile(keystoreUTCPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
privKey, err := keystore.DecryptKey(keyJSON, password)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
txOps = bind.NewKeyedTransactor(privKey.PrivateKey)
|
|
txOps.Value = big.NewInt(0)
|
|
txOps.GasLimit = 4000000
|
|
nonce, err := client.NonceAt(context.Background(), privKey.Address, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
txOps.Nonce = big.NewInt(int64(nonce))
|
|
fmt.Printf("current address is %s\n", privKey.Address.String())
|
|
fmt.Printf("nonce is %d\n", nonce)
|
|
}
|
|
|
|
addresses, err := allDepositContractAddresses(client)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Could not get all deposit contract address")
|
|
}
|
|
|
|
fmt.Printf("%d contracts ready to drain found\n", len(addresses))
|
|
|
|
for _, address := range addresses {
|
|
bal, err := client.BalanceAt(context.Background(), address, nil /*blockNum*/)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if bal.Cmp(big.NewInt(0)) < 1 {
|
|
continue
|
|
}
|
|
depositContract, err := contracts.NewDepositContract(address, client)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tx, err := depositContract.Drain(txOps)
|
|
if err != nil {
|
|
log.Fatalf("unable to send transaction to contract: %v", err)
|
|
}
|
|
|
|
txOps.Nonce = txOps.Nonce.Add(txOps.Nonce, big.NewInt(1))
|
|
|
|
fmt.Printf("Contract address %s drained in TX hash: %s\n", address.String(), tx.Hash().String())
|
|
time.Sleep(time.Duration(1) * time.Second)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func loadTextFromFile(filepath string) string {
|
|
// #nosec - Inclusion of file via variable is OK for this tool.
|
|
file, err := os.Open(filepath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
scanner.Split(bufio.ScanWords)
|
|
scanner.Scan()
|
|
return scanner.Text()
|
|
}
|
|
|
|
func allDepositContractAddresses(client *ethclient.Client) ([]common.Address, error) {
|
|
log.Print("Looking up contracts")
|
|
addresses := make(map[common.Address]bool)
|
|
|
|
// Hash of deposit log signature
|
|
// DepositEvent: event({
|
|
// pubkey: bytes[48],
|
|
// withdrawal_credentials: bytes[32],
|
|
// amount: bytes[8],
|
|
// signature: bytes[96],
|
|
// index: bytes[8],
|
|
// })
|
|
depositTopicHash := common.HexToHash("0x649bbc62d0e31342afea4e5cd82d4049e7e1ee912fc0889aa790803be39038c5")
|
|
fmt.Println(depositTopicHash.Hex())
|
|
|
|
query := ethereum.FilterQuery{
|
|
Addresses: []common.Address{},
|
|
Topics: [][]common.Hash{
|
|
{depositTopicHash},
|
|
},
|
|
FromBlock: big.NewInt(800000), // Contracts before this may not have drain().
|
|
}
|
|
|
|
logs, err := client.FilterLogs(context.Background(), query)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not get all deposit logs")
|
|
}
|
|
|
|
fmt.Printf("%d deposit logs found\n", len(logs))
|
|
for i := len(logs)/2 - 1; i >= 0; i-- {
|
|
opp := len(logs) - 1 - i
|
|
logs[i], logs[opp] = logs[opp], logs[i]
|
|
}
|
|
|
|
for _, ll := range logs {
|
|
addresses[ll.Address] = true
|
|
}
|
|
|
|
keys := make([]common.Address, 0, len(addresses))
|
|
for key := range addresses {
|
|
keys = append(keys, key)
|
|
}
|
|
|
|
return keys, nil
|
|
}
|