152 lines
4.0 KiB
Python
152 lines
4.0 KiB
Python
# standard imports
|
||
import logging
|
||
|
||
# third-party imports
|
||
import sha3
|
||
from hexathon import strip_0x
|
||
from eth_keys import KeyAPI
|
||
from eth_keys.backends import NativeECCBackend
|
||
from rlp import decode as rlp_decode
|
||
from rlp import encode as rlp_encode
|
||
from crypto_dev_signer.eth.transaction import EIP155Transaction
|
||
|
||
# local imports
|
||
from .address import to_checksum
|
||
from .constant import (
|
||
MINIMUM_FEE_UNITS,
|
||
MINIMUM_FEE_PRICE,
|
||
)
|
||
|
||
logg = logging.getLogger(__name__)
|
||
|
||
|
||
field_debugs = [
|
||
'nonce',
|
||
'gasPrice',
|
||
'gas',
|
||
'to',
|
||
'value',
|
||
'data',
|
||
'v',
|
||
'r',
|
||
's',
|
||
]
|
||
|
||
def unpack_signed(tx_raw_bytes, chain_id=1):
|
||
d = rlp_decode(tx_raw_bytes)
|
||
|
||
logg.debug('decoding using chain id {}'.format(chain_id))
|
||
j = 0
|
||
for i in d:
|
||
logg.debug('decoded {}: {}'.format(field_debugs[j], i.hex()))
|
||
j += 1
|
||
vb = chain_id
|
||
if chain_id != 0:
|
||
v = int.from_bytes(d[6], 'big')
|
||
vb = v - (chain_id * 2) - 35
|
||
s = b''.join([d[7], d[8], bytes([vb])])
|
||
so = KeyAPI.Signature(signature_bytes=s)
|
||
|
||
h = sha3.keccak_256()
|
||
h.update(rlp_encode(d))
|
||
signed_hash = h.digest()
|
||
|
||
d[6] = chain_id
|
||
d[7] = b''
|
||
d[8] = b''
|
||
|
||
h = sha3.keccak_256()
|
||
h.update(rlp_encode(d))
|
||
unsigned_hash = h.digest()
|
||
|
||
p = so.recover_public_key_from_msg_hash(unsigned_hash)
|
||
a = p.to_checksum_address()
|
||
logg.debug('decoded recovery byte {}'.format(vb))
|
||
logg.debug('decoded address {}'.format(a))
|
||
logg.debug('decoded signed hash {}'.format(signed_hash.hex()))
|
||
logg.debug('decoded unsigned hash {}'.format(unsigned_hash.hex()))
|
||
|
||
to = d[3].hex() or None
|
||
if to != None:
|
||
to = to_checksum(to)
|
||
|
||
return {
|
||
'from': a,
|
||
'nonce': int.from_bytes(d[0], 'big'),
|
||
'gasPrice': int.from_bytes(d[1], 'big'),
|
||
'gas': int.from_bytes(d[2], 'big'),
|
||
'to': to,
|
||
'value': int.from_bytes(d[4], 'big'),
|
||
'data': '0x' + d[5].hex(),
|
||
'v': chain_id,
|
||
'r': '0x' + s[:32].hex(),
|
||
's': '0x' + s[32:64].hex(),
|
||
'chainId': chain_id,
|
||
'hash': '0x' + signed_hash.hex(),
|
||
'hash_unsigned': '0x' + unsigned_hash.hex(),
|
||
}
|
||
|
||
|
||
class TxFactory:
|
||
|
||
def __init__(self, signer=None, gas_oracle=None, nonce_oracle=None, chain_id=1):
|
||
self.gas_oracle = gas_oracle
|
||
self.nonce_oracle = nonce_oracle
|
||
self.chain_id = chain_id
|
||
self.signer = signer
|
||
|
||
|
||
def template(self, sender, recipient):
|
||
gas_price = MINIMUM_FEE_PRICE
|
||
if self.gas_oracle != None:
|
||
gas_price = self.gas_oracle.get()
|
||
logg.debug('using gas price {}'.format(gas_price))
|
||
nonce = 0
|
||
if self.nonce_oracle != None:
|
||
nonce = self.nonce_oracle.next()
|
||
logg.debug('using nonce {} for address {}'.format(nonce, sender))
|
||
return {
|
||
'from': sender,
|
||
'to': recipient,
|
||
'value': 0,
|
||
'data': '0x',
|
||
'nonce': nonce,
|
||
'gasPrice': gas_price,
|
||
'gas': MINIMUM_FEE_UNITS,
|
||
'chainId': self.chain_id,
|
||
}
|
||
|
||
|
||
def normalize(self, tx):
|
||
txe = EIP155Transaction(tx, tx['nonce'], tx['chainId'])
|
||
txes = txe.serialize()
|
||
print(txes)
|
||
return {
|
||
'from': tx['from'],
|
||
'to': txes['to'],
|
||
'gasPrice': txes['gasPrice'],
|
||
'gas': txes['gas'],
|
||
'data': txes['data'],
|
||
}
|
||
|
||
|
||
def set_code(self, tx, data, update_fee=True):
|
||
tx['data'] = data
|
||
if update_fee:
|
||
logg.debug('using hardcoded gas limit of 8000000 until we have reliable vm executor')
|
||
tx['gas'] = 8000000
|
||
return tx
|
||
|
||
|
||
class Tx:
|
||
|
||
def __init__(self, src, block):
|
||
self.index = int(strip_0x(src['transactionIndex']), 16)
|
||
self.nonce = src['nonce']
|
||
self.hash = src['hash']
|
||
self.block = block
|
||
|
||
|
||
def __str__(self):
|
||
return 'block {} tx {} {}'.format(self.block.number, self.index, self.hash)
|