129 lines
4.2 KiB
Python
129 lines
4.2 KiB
Python
|
# standard imports
|
|||
|
|
import os
|
|||
|
|
import logging
|
|||
|
|
import uuid
|
|||
|
|
import random
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# external imports
|
|||
|
|
from chainlib.chain import ChainSpec
|
|||
|
|
from chainlib.eth.constant import ZERO_ADDRESS
|
|||
|
|
from chainlib.eth.gas import (
|
|||
|
|
balance,
|
|||
|
|
Gas,
|
|||
|
|
)
|
|||
|
|
from hexathon import (
|
|||
|
|
add_0x,
|
|||
|
|
strip_0x,
|
|||
|
|
)
|
|||
|
|
from chainlib.eth.connection import EthHTTPSignerConnection
|
|||
|
|
from funga.eth.signer import EIP155Signer
|
|||
|
|
from funga.eth.keystore.sql import SQLKeystore
|
|||
|
|
from chainlib.cli.wallet import Wallet
|
|||
|
|
from chainlib.eth.address import AddressChecksum
|
|||
|
|
from chainlib.eth.nonce import RPCNonceOracle
|
|||
|
|
from chainlib.eth.gas import OverrideGasOracle
|
|||
|
|
from chainlib.eth.address import (
|
|||
|
|
is_checksum_address,
|
|||
|
|
to_checksum_address,
|
|||
|
|
)
|
|||
|
|
from chainlib.eth.tx import (
|
|||
|
|
TxFormat,
|
|||
|
|
)
|
|||
|
|
import chainlib.eth.cli
|
|||
|
|
|
|||
|
|
|
|||
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|||
|
|
config_dir = os.path.join(script_dir, '..', 'cic_signer', 'data', 'config')
|
|||
|
|
|
|||
|
|
logging.basicConfig(level=logging.WARNING)
|
|||
|
|
logg = logging.getLogger()
|
|||
|
|
|
|||
|
|
arg_flags = chainlib.eth.cli.argflag_std_write | chainlib.eth.cli.Flag.WALLET
|
|||
|
|
argparser = chainlib.eth.cli.ArgumentParser(arg_flags)
|
|||
|
|
args = argparser.parse_args()
|
|||
|
|
|
|||
|
|
config = chainlib.eth.cli.Config.from_args(args, arg_flags, base_config_dir=config_dir)
|
|||
|
|
|
|||
|
|
# set up rpc
|
|||
|
|
chain_spec = ChainSpec.from_chain_str(config.get('CHAIN_SPEC'))
|
|||
|
|
|
|||
|
|
|
|||
|
|
# connect to database
|
|||
|
|
dsn = 'postgresql://{}:{}@{}:{}/{}'.format(
|
|||
|
|
config.get('DATABASE_USER'),
|
|||
|
|
config.get('DATABASE_PASSWORD'),
|
|||
|
|
config.get('DATABASE_HOST'),
|
|||
|
|
config.get('DATABASE_PORT'),
|
|||
|
|
config.get('DATABASE_NAME'),
|
|||
|
|
)
|
|||
|
|
logg.info('using dsn {}'.format(dsn))
|
|||
|
|
|
|||
|
|
keystore = SQLKeystore(dsn, symmetric_key=bytes.fromhex(config.get('SIGNER_SECRET')))
|
|||
|
|
wallet = Wallet(EIP155Signer, keystore=keystore, checksummer=AddressChecksum)
|
|||
|
|
|
|||
|
|
rpc = chainlib.eth.cli.Rpc(wallet=wallet)
|
|||
|
|
conn = rpc.connect_by_config(config)
|
|||
|
|
|
|||
|
|
wallet.init()
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
if config.get('_RECIPIENT') == None:
|
|||
|
|
sys.stderr.write('Missing sink address\n')
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
sink_address = config.get('_RECIPIENT')
|
|||
|
|
if config.get('_UNSAFE'):
|
|||
|
|
sink_address = to_checksum_address(sink_address)
|
|||
|
|
if not is_checksum_address(sink_address):
|
|||
|
|
sys.stderr.write('Invalid sink address {}\n'.format(sink_address))
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
if (config.get('_RPC_SEND')):
|
|||
|
|
verify_string = random.randbytes(4).hex()
|
|||
|
|
verify_string_check = input("\033[;31m*** WARNING! WARNING! WARNING! ***\033[;39m\nThis action will transfer all remaining gas from all accounts in custodial care to account {}. To confirm, please enter the string: {}\n".format(config.get('_RECIPIENT'), verify_string))
|
|||
|
|
if verify_string != verify_string_check:
|
|||
|
|
sys.stderr.write('Verify string mismatch. Aborting!\n')
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
signer = rpc.get_signer()
|
|||
|
|
|
|||
|
|
gas_oracle = rpc.get_gas_oracle()
|
|||
|
|
gas_pair = gas_oracle.get_fee()
|
|||
|
|
gas_price = gas_pair[0]
|
|||
|
|
gas_limit = 21000
|
|||
|
|
gas_cost = gas_price * gas_limit
|
|||
|
|
gas_oracle = OverrideGasOracle(price=gas_price, limit=gas_limit)
|
|||
|
|
logg.info('using gas price {}'.format(gas_price))
|
|||
|
|
|
|||
|
|
for r in keystore.list():
|
|||
|
|
account = r[0]
|
|||
|
|
|
|||
|
|
o = balance(add_0x(account))
|
|||
|
|
r = conn.do(o)
|
|||
|
|
account_balance = 0
|
|||
|
|
try:
|
|||
|
|
r = strip_0x(r)
|
|||
|
|
account_balance = int(r, 16)
|
|||
|
|
except ValueError:
|
|||
|
|
account_balance = int(r)
|
|||
|
|
|
|||
|
|
transfer_amount = account_balance - gas_cost
|
|||
|
|
if transfer_amount <= 0:
|
|||
|
|
logg.warning('address {} has balance {} which is less than gas cost {}, skipping'.format(account, account_balance, gas_cost))
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
nonce_oracle = RPCNonceOracle(account, conn)
|
|||
|
|
c = Gas(chain_spec, gas_oracle=gas_oracle, nonce_oracle=nonce_oracle, signer=signer)
|
|||
|
|
tx_hash_hex = None
|
|||
|
|
if (config.get('_RPC_SEND')):
|
|||
|
|
(tx_hash_hex, o) = c.create(account, config.get('_RECIPIENT'), transfer_amount)
|
|||
|
|
r = conn.do(o)
|
|||
|
|
else:
|
|||
|
|
(tx_hash_hex, o) = c.create(account, config.get('_RECIPIENT'), transfer_amount, tx_format=TxFormat.RLP_SIGNED)
|
|||
|
|
|
|||
|
|
logg.info('address {} balance {} net transfer {} tx {}'.format(account, account_balance, transfer_amount, tx_hash_hex))
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|