diff --git a/apps/contract-migration/scripts/import_balance.py b/apps/contract-migration/scripts/import_balance.py new file mode 100644 index 00000000..7c6ce3ef --- /dev/null +++ b/apps/contract-migration/scripts/import_balance.py @@ -0,0 +1,275 @@ +# standard imports +import os +import sys +import logging +import time +import argparse +import sys +import re +import hashlib +import csv +import json + +# third-party impotts +import eth_abi +import confini +from hexathon import ( + strip_0x, + add_0x, + ) +from cic_registry.chain import ChainSpec +from chainsyncer.backend import MemBackend +from chainsyncer.driver import HeadSyncer +from chainlib.eth.connection import HTTPConnection +from chainlib.eth.block import ( + block_latest, + block_by_number, + Block, + ) +from chainlib.eth.hash import keccak256_string_to_hex +from chainlib.eth.address import to_checksum +from chainlib.eth.erc20 import ERC20TxFactory +from chainlib.eth.gas import DefaultGasOracle +from chainlib.eth.nonce import DefaultNonceOracle +from chainlib.eth.tx import TxFactory +from chainlib.eth.rpc import jsonrpc_template +from chainlib.eth.error import EthException +from crypto_dev_signer.eth.signer import ReferenceSigner as EIP155Signer +from crypto_dev_signer.keystore import DictKeystore +from cic_types.models.person import Person + + +logging.basicConfig(level=logging.WARNING) +logg = logging.getLogger() + +config_dir = '/usr/local/etc/cic-syncer' + +argparser = argparse.ArgumentParser(description='daemon that monitors transactions in new blocks') +argparser.add_argument('-p', '--provider', dest='p', type=str, help='chain rpc provider address') +argparser.add_argument('-y', '--key-file', dest='y', type=str, help='Ethereum keystore file to use for signing') +argparser.add_argument('-c', type=str, default=config_dir, help='config root to use') +argparser.add_argument('-i', '--chain-spec', type=str, dest='i', help='chain spec') +argparser.add_argument('-a', '--index-address', type=str, dest='a', help='account index contract address') +argparser.add_argument('-r', '--registry-address', type=str, dest='r', help='CIC Registry address') +argparser.add_argument('--head', action='store_true', help='start at current block height (overrides --offset)') +argparser.add_argument('--env-prefix', default=os.environ.get('CONFINI_ENV_PREFIX'), dest='env_prefix', type=str, help='environment prefix for variables to overwrite configuration') +argparser.add_argument('-q', type=str, default='cic-eth', help='celery queue to submit transaction tasks to') +argparser.add_argument('--offset', type=int, default=0, help='block offset to start syncer from') +argparser.add_argument('-v', help='be verbose', action='store_true') +argparser.add_argument('-vv', help='be more verbose', action='store_true') +argparser.add_argument('user_dir', type=str, help='user export directory') +args = argparser.parse_args(sys.argv[1:]) + +if args.v == True: + logging.getLogger().setLevel(logging.INFO) +elif args.vv == True: + logging.getLogger().setLevel(logging.DEBUG) + +config_dir = os.path.join(args.c) +os.makedirs(config_dir, 0o777, True) +config = confini.Config(config_dir, args.env_prefix) +config.process() +# override args +args_override = { + 'CIC_CHAIN_SPEC': getattr(args, 'i'), + 'ETH_PROVIDER': getattr(args, 'p'), + 'CIC_REGISTRY_ADDRESS': getattr(args, 'r'), + } +config.dict_override(args_override, 'cli flag') +config.censor('PASSWORD', 'DATABASE') +config.censor('PASSWORD', 'SSL') +logg.debug('config loaded from {}:\n{}'.format(config_dir, config)) + +#app = celery.Celery(backend=config.get('CELERY_RESULT_URL'), broker=config.get('CELERY_BROKER_URL')) + +signer_address = None +keystore = DictKeystore() +if args.y != None: + logg.debug('loading keystore file {}'.format(args.y)) + signer_address = keystore.import_keystore_file(args.y) + logg.debug('now have key for signer address {}'.format(signer_address)) +signer = EIP155Signer(keystore) + +queue = args.q +chain_str = config.get('CIC_CHAIN_SPEC') +block_offset = 0 +if args.head: + block_offset = -1 +else: + block_offset = args.offset +account_index_address = args.a + +chain_spec = ChainSpec.from_chain_str(chain_str) + +user_dir = args.user_dir # user_out_dir from import_users.py + + +class Handler: + + account_index_add_signature = keccak256_string_to_hex('add(address)')[:8] + + def __init__(self, conn, user_dir, addresses, balances, token_address): + self.conn = conn + self.token_address = token_address + self.user_dir = user_dir + self.addresses = addresses + self.balances = balances + + + def name(self): + return 'balance_handler' + + + def handle(self, getter, block, tx): + try: + if tx.payload[:8] == self.account_index_add_signature: + recipient = eth_abi.decode_single('address', bytes.fromhex(tx.payload[-64:])) + original_address = to_checksum(self.addresses[to_checksum(recipient)]) + user_file = '{}/{}/{}.json'.format( + recipient[2:4].upper(), + recipient[4:6].upper(), + recipient[2:].upper(), + ) + filepath = os.path.join(self.user_dir, user_file) + f = open(filepath, 'r') + o = json.load(f) + f.close() + u = Person(o) + balance = self.balances[original_address] + logg.info('registered {} originally {} ({}) tx hash {} balance {}'.format(recipient, original_address, u, tx.hash, balance)) + + (tx_hash_hex, o) = getter.tx_factory.erc20_transfer(self.token_address, signer_address, recipient, balance) + logg.info('submitting erc20 transfer tx {} for recipient {}'.format(tx_hash_hex, recipient)) + r = self.conn.do(o) + except TypeError: + pass + except IndexError: + pass + except EthException as e: + logg.error('send error {}'.format(e).ljust(200)) + #except KeyError as e: + # logg.error('key error {}'.format(e).ljust(200)) + + +class BlockGetter: + + def __init__(self, conn, gas_oracle, nonce_oracle, chain_id): + self.conn = conn + self.tx_factory = ERC20TxFactory(signer=signer, gas_oracle=gas_oracle, nonce_oracle=nonce_oracle, chain_id=chain_id) + + + def get(self, n): + o = block_by_number(n) + r = self.conn.do(o) + b = None + try: + b = Block(r) + except TypeError as e: + if r == None: + logg.debug('block not found {}'.format(n)) + else: + logg.error('block retrieve error {}'.format(e)) + return b + + +def progress_callback(s, block_number, tx_index): + sys.stdout.write(s.ljust(200) + "\r") + + + +def main(): + global chain_str, block_offset, user_dir + + conn = HTTPConnection(config.get('ETH_PROVIDER')) + gas_oracle = DefaultGasOracle(conn) + nonce_oracle = DefaultNonceOracle(signer_address, conn) + + # Get Token registry address + txf = TxFactory(signer=signer, gas_oracle=gas_oracle, nonce_oracle=None, chain_id=chain_spec.chain_id()) + tx = txf.template(signer_address, config.get('CIC_REGISTRY_ADDRESS')) + + registry_addressof_method = keccak256_string_to_hex('addressOf(bytes32)')[:8] + data = add_0x(registry_addressof_method) + data += eth_abi.encode_single('bytes32', b'TokenRegistry').hex() + txf.set_code(tx, data) + + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(txf.normalize(tx)) + o['params'].append('latest') + r = conn.do(o) + token_index_address = to_checksum(eth_abi.decode_single('address', bytes.fromhex(strip_0x(r)))) + logg.info('found token index address {}'.format(token_index_address)) + + + # Get Sarafu token address + tx = txf.template(signer_address, token_index_address) + data = add_0x(registry_addressof_method) + h = hashlib.new('sha256') + h.update(b'SRF') + z = h.digest() + data += eth_abi.encode_single('bytes32', z).hex() + txf.set_code(tx, data) + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(txf.normalize(tx)) + o['params'].append('latest') + r = conn.do(o) + print('r {}'.format(r)) + sarafu_token_address = to_checksum(eth_abi.decode_single('address', bytes.fromhex(strip_0x(r)))) + logg.info('found token address {}'.format(sarafu_token_address)) + + + getter = BlockGetter(conn, gas_oracle, nonce_oracle, chain_spec.chain_id()) + syncer_backend = MemBackend(chain_str, 0) + + if block_offset == -1: + o = block_latest() + r = conn.do(o) + block_offset = int(strip_0x(r), 16) + 1 + + addresses = {} + f = open('{}/addresses.csv'.format(user_dir, 'r')) + while True: + l = f.readline() + if l == None: + break + r = l.split(',') + try: + k = r[0] + v = r[1].rstrip() + addresses[k] = v + sys.stdout.write('loading address mapping {} -> {}'.format(k, v).ljust(200) + "\r") + except IndexError as e: + break + f.close() + + balances = {} + f = open('{}/balances.csv'.format(user_dir, 'r')) + remove_zeros = 10**12 + i = 0 + while True: + l = f.readline() + if l == None: + break + r = l.split(',') + try: + address = to_checksum(r[0]) + sys.stdout.write('loading balance {} {}'.format(i, address).ljust(200) + "\r") + except ValueError: + break + balance = int(int(r[1].rstrip()) / remove_zeros) + balances[address] = balance + i += 1 + + f.close() + + syncer_backend.set(block_offset, 0) + syncer = HeadSyncer(syncer_backend, progress_callback=progress_callback) + handler = Handler(conn, user_dir, addresses, balances, sarafu_token_address) + syncer.add_filter(handler) + syncer.loop(1, getter) + + +if __name__ == '__main__': + main() diff --git a/apps/contract-migration/scripts/import_users.py b/apps/contract-migration/scripts/import_users.py new file mode 100644 index 00000000..1118e74b --- /dev/null +++ b/apps/contract-migration/scripts/import_users.py @@ -0,0 +1,160 @@ +# standard imports +import os +import sys +import json +import logging +import argparse +import uuid +import datetime +import shutil +from glob import glob + +# third-party imports +import redis +import confini +import celery +from hexathon import ( + add_0x, + strip_0x, + ) +from chainlib.eth.address import to_checksum +from cic_types.models.person import Person +from cic_eth.api.api_task import Api +from cic_registry.chain import ChainSpec + +logging.basicConfig(level=logging.WARNING) +logg = logging.getLogger() + +default_config_dir = '/usr/local/etc/cic' + +argparser = argparse.ArgumentParser() +argparser.add_argument('-c', type=str, default=default_config_dir, help='config file') +argparser.add_argument('-i', '--chain-spec', dest='i', type=str, help='Chain specification string') +argparser.add_argument('--redis-host', dest='redis_host', type=str, help='redis host to use for task submission') +argparser.add_argument('--redis-port', dest='redis_port', type=int, help='redis host to use for task submission') +argparser.add_argument('--redis-db', dest='redis_db', type=int, help='redis db to use for task submission and callback') +argparser.add_argument('--redis-host-callback', dest='redis_host_callback', default='localhost', type=str, help='redis host to use for callback') +argparser.add_argument('--redis-port-callback', dest='redis_port_callback', default=6379, type=int, help='redis port to use for callback') +argparser.add_argument('--timeout', default=20.0, type=float, help='Callback timeout') +argparser.add_argument('-q', type=str, default='cic-eth', help='Task queue') +argparser.add_argument('-v', action='store_true', help='Be verbose') +argparser.add_argument('-vv', action='store_true', help='Be more verbose') +argparser.add_argument('user_dir', type=str, help='path to users export dir tree') +args = argparser.parse_args() + +if args.v: + logg.setLevel(logging.INFO) +elif args.vv: + logg.setLevel(logging.DEBUG) + +config_dir = args.c +config = confini.Config(config_dir, os.environ.get('CONFINI_ENV_PREFIX')) +config.process() +args_override = { + 'CIC_CHAIN_SPEC': getattr(args, 'i'), + 'REDIS_HOST': getattr(args, 'redis_host'), + 'REDIS_PORT': getattr(args, 'redis_port'), + 'REDIS_DB': getattr(args, 'redis_db'), + } +config.dict_override(args_override, 'cli') +celery_app = celery.Celery(broker=config.get('CELERY_BROKER_URL'), backend=config.get('CELERY_RESULT_URL')) + +redis_host = config.get('REDIS_HOST') +redis_port = config.get('REDIS_PORT') +redis_db = config.get('REDIS_DB') +r = redis.Redis(redis_host, redis_port, redis_db) + +ps = r.pubsub() + +user_dir = args.user_dir +user_out_dir = '{}_cic_eth'.format(user_dir) +os.makedirs(user_out_dir) +shutil.copy( + os.path.join(user_dir, 'balances.csv'), + os.path.join(user_out_dir, 'balances.csv'), + ) + +chain_spec = ChainSpec.from_chain_str(config.get('CIC_CHAIN_SPEC')) +chain_str = str(chain_spec) + + +def register_eth(u): + redis_channel = str(uuid.uuid4()) + ps.subscribe(redis_channel) + ps.get_message() + api = Api( + config.get('CIC_CHAIN_SPEC'), + queue=args.q, + callback_param='{}:{}:{}:{}'.format(args.redis_host_callback, args.redis_port_callback, redis_db, redis_channel), + callback_task='cic_eth.callbacks.redis.redis', + callback_queue=args.q, + ) + t = api.create_account(register=True) + + ps.get_message() + m = ps.get_message(timeout=args.timeout) + address = json.loads(m['data']) + logg.debug('register eth {} {}'.format(u, address)) + + return address + + +def register_ussd(u): + #logg.warning('missing ussd register') + pass + + +if __name__ == '__main__': + + fi = open(os.path.join(user_out_dir, 'addresses.csv'), 'a') + + i = 0 + for x in os.walk(user_dir): + for y in x[2]: + if y[len(y)-5:] != '.json': + continue + filepath = os.path.join(x[0], y) + f = open(filepath, 'r') + try: + o = json.load(f) + except json.decoder.JSONDecodeError as e: + f.close() + logg.error('load error for {}: {}'.format(y, e)) + continue + f.close() + u = Person(o) + + new_address = register_eth(u) + u.identities['evm'][chain_str] = [new_address] + + register_ussd(u) + +# part = [] +# for j in range(3): +# (head, tail) = os.path.split(filepath) +# part.append(tail) +# filepath = head +# part.reverse() +# filepath = os.path.join(user_out_dir, '/'.join(part)) + new_address_clean = strip_0x(new_address) + filepath = os.path.join( + user_out_dir, + new_address_clean[:2].upper(), + new_address_clean[2:4].upper(), + new_address_clean.upper() + '.json', + ) + logg.debug('outpath {}'.format(filepath)) + os.makedirs(os.path.dirname(filepath), exist_ok=True) + + o = u.serialize() + f = open(filepath, 'w') + f.write(json.dumps(o)) + f.close() + + old_address = to_checksum(add_0x(y[:len(y)-5])) + fi.write('{},{}\n'.format(new_address, old_address)) + + i += 1 + sys.stdout.write('imported {} {}'.format(i, u).ljust(200) + "\r") + + fi.close() diff --git a/apps/contract-migration/scripts/requirements.txt b/apps/contract-migration/scripts/requirements.txt new file mode 100644 index 00000000..daee02f5 --- /dev/null +++ b/apps/contract-migration/scripts/requirements.txt @@ -0,0 +1,9 @@ +psycopg2==2.8.6 +cic-types==0.1.0a2 +chainlib==0.0.1a7 +chainsyncer==0.0.1a5 +cic-eth==0.10.0a26 +confini==0.3.6b1 +celery==4.4.7 +redis==3.5.3 +hexathon==0.0.1a2 diff --git a/apps/contract-migration/scripts/verify.py b/apps/contract-migration/scripts/verify.py new file mode 100644 index 00000000..d1543c26 --- /dev/null +++ b/apps/contract-migration/scripts/verify.py @@ -0,0 +1,240 @@ +# standard imports +import os +import sys +import logging +import time +import argparse +import sys +import re +import hashlib +import csv +import json + +# third-party impotts +import celery +import eth_abi +import confini +from hexathon import ( + strip_0x, + add_0x, + ) +from cic_registry.chain import ChainSpec +from chainsyncer.backend import MemBackend +from chainsyncer.driver import HeadSyncer +from chainlib.eth.connection import HTTPConnection +from chainlib.eth.constant import ZERO_ADDRESS +from chainlib.eth.block import ( + block_latest, + block_by_number, + Block, + ) +from chainlib.eth.hash import keccak256_string_to_hex +from chainlib.eth.address import to_checksum +from chainlib.eth.erc20 import ERC20TxFactory +from chainlib.eth.gas import DefaultGasOracle +from chainlib.eth.nonce import DefaultNonceOracle +from chainlib.eth.tx import TxFactory +from chainlib.eth.rpc import jsonrpc_template +from chainlib.eth.error import EthException +from crypto_dev_signer.eth.signer import ReferenceSigner as EIP155Signer +from crypto_dev_signer.keystore import DictKeystore +from cic_eth.api.api_admin import AdminApi + +logging.basicConfig(level=logging.WARNING) +logg = logging.getLogger() + +config_dir = '/usr/local/etc/cic-syncer' + +argparser = argparse.ArgumentParser(description='daemon that monitors transactions in new blocks') +argparser.add_argument('-p', '--provider', dest='p', type=str, help='chain rpc provider address') +argparser.add_argument('-c', type=str, default=config_dir, help='config root to use') +argparser.add_argument('-i', '--chain-spec', type=str, dest='i', help='chain spec') +argparser.add_argument('-r', '--registry-address', type=str, dest='r', help='CIC Registry address') +argparser.add_argument('--env-prefix', default=os.environ.get('CONFINI_ENV_PREFIX'), dest='env_prefix', type=str, help='environment prefix for variables to overwrite configuration') +argparser.add_argument('-v', help='be verbose', action='store_true') +argparser.add_argument('-vv', help='be more verbose', action='store_true') +argparser.add_argument('user_dir', type=str, help='user export directory') +args = argparser.parse_args(sys.argv[1:]) + +if args.v == True: + logging.getLogger().setLevel(logging.INFO) +elif args.vv == True: + logging.getLogger().setLevel(logging.DEBUG) + +config_dir = os.path.join(args.c) +os.makedirs(config_dir, 0o777, True) +config = confini.Config(config_dir, args.env_prefix) +config.process() +# override args +args_override = { + 'CIC_CHAIN_SPEC': getattr(args, 'i'), + 'ETH_PROVIDER': getattr(args, 'p'), + 'CIC_REGISTRY_ADDRESS': getattr(args, 'r'), + } +config.dict_override(args_override, 'cli flag') +config.censor('PASSWORD', 'DATABASE') +config.censor('PASSWORD', 'SSL') +logg.debug('config loaded from {}:\n{}'.format(config_dir, config)) + +celery_app = celery.Celery(backend=config.get('CELERY_RESULT_URL'), broker=config.get('CELERY_BROKER_URL')) + +chain_str = config.get('CIC_CHAIN_SPEC') +chain_spec = ChainSpec.from_chain_str(chain_str) +user_dir = args.user_dir # user_out_dir from import_users.py + + +class Verifier: + + def __init__(self, conn, cic_eth_api, gas_oracle, chain_spec, index_address, token_address): + self.conn = conn + self.gas_oracle = gas_oracle + self.chain_spec = chain_spec + self.index_address = index_address + self.token_address = token_address + self.erc20_tx_factory = ERC20TxFactory(chain_id=chain_spec.chain_id(), gas_oracle=gas_oracle) + self.tx_factory = TxFactory(chain_id=chain_spec.chain_id(), gas_oracle=gas_oracle) + self.api = cic_eth_api + + + def verify_accounts_index(self, address): + tx = self.tx_factory.template(ZERO_ADDRESS, self.index_address) + data = keccak256_string_to_hex('have(address)')[:8] + data += eth_abi.encode_single('address', address).hex() + tx = self.tx_factory.set_code(tx, data) + tx = self.tx_factory.normalize(tx) + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(tx) + r = self.conn.do(o) + logg.debug('index check for {}: {}'.format(address, r)) + n = eth_abi.decode_single('uint256', bytes.fromhex(strip_0x(r))) + if n != 1: + raise ValueError(n) + + + def verify_balance(self, address, balance): + o = self.erc20_tx_factory.erc20_balance(self.token_address, address) + r = self.conn.do(o) + actual_balance = int(strip_0x(r), 16) + logg.debug('balance for {}: {}'.format(address, balance)) + if balance != actual_balance: + raise ValueError((actual_balance, balance)) + + + def verify_local_key(self, address): + r = self.api.have_account(address, str(self.chain_spec)) + logg.debug('verify local key result {}'.format(r)) + if r != address: + raise ValueError(address, r) + + + def verify(self, address, balance): + logg.debug('verify {} {}'.format(address, balance)) + + self.verify_local_key(address) + self.verify_accounts_index(address) + self.verify_balance(address, balance) + + +class MockClient: + + w3 = None + +def main(): + global chain_str, block_offset, user_dir + + conn = HTTPConnection(config.get('ETH_PROVIDER')) + gas_oracle = DefaultGasOracle(conn) + + # Get Token registry address + txf = TxFactory(signer=None, gas_oracle=gas_oracle, nonce_oracle=None, chain_id=chain_spec.chain_id()) + tx = txf.template(ZERO_ADDRESS, config.get('CIC_REGISTRY_ADDRESS')) + + registry_addressof_method = keccak256_string_to_hex('addressOf(bytes32)')[:8] + data = add_0x(registry_addressof_method) + data += eth_abi.encode_single('bytes32', b'TokenRegistry').hex() + txf.set_code(tx, data) + + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(txf.normalize(tx)) + o['params'].append('latest') + r = conn.do(o) + token_index_address = to_checksum(eth_abi.decode_single('address', bytes.fromhex(strip_0x(r)))) + logg.info('found token index address {}'.format(token_index_address)) + + data = add_0x(registry_addressof_method) + data += eth_abi.encode_single('bytes32', b'AccountRegistry').hex() + txf.set_code(tx, data) + + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(txf.normalize(tx)) + o['params'].append('latest') + r = conn.do(o) + account_index_address = to_checksum(eth_abi.decode_single('address', bytes.fromhex(strip_0x(r)))) + logg.info('found account index address {}'.format(account_index_address)) + + + # Get Sarafu token address + tx = txf.template(ZERO_ADDRESS, token_index_address) + data = add_0x(registry_addressof_method) + h = hashlib.new('sha256') + h.update(b'SRF') + z = h.digest() + data += eth_abi.encode_single('bytes32', z).hex() + txf.set_code(tx, data) + o = jsonrpc_template() + o['method'] = 'eth_call' + o['params'].append(txf.normalize(tx)) + o['params'].append('latest') + r = conn.do(o) + print('r {}'.format(r)) + sarafu_token_address = to_checksum(eth_abi.decode_single('address', bytes.fromhex(strip_0x(r)))) + logg.info('found token address {}'.format(sarafu_token_address)) + + addresses = {} + f = open('{}/addresses.csv'.format(user_dir, 'r')) + while True: + l = f.readline() + if l == None: + break + r = l.split(',') + try: + k = r[0] + v = r[1].rstrip() + addresses[k] = v + sys.stdout.write('loading address mapping {} -> {}'.format(k, v).ljust(200) + "\r") + except IndexError as e: + break + f.close() + + balances = {} + f = open('{}/balances.csv'.format(user_dir, 'r')) + remove_zeros = 10**12 + i = 0 + while True: + l = f.readline() + if l == None: + break + r = l.split(',') + try: + address = to_checksum(r[0]) + sys.stdout.write('loading balance {} {}'.format(i, address).ljust(200) + "\r") + except ValueError: + break + balance = int(int(r[1].rstrip()) / remove_zeros) + balances[address] = balance + i += 1 + + f.close() + + api = AdminApi(MockClient()) + + verifier = Verifier(conn, api, gas_oracle, chain_spec, account_index_address, sarafu_token_address) + for k in addresses.keys(): + verifier.verify(k, balances[addresses[k]]) + + +if __name__ == '__main__': + main()