@ -26,23 +26,23 @@ address_mapping_head_key = b'accountMapping' # head (latest) number of indexed
def _format_block_number ( number ) :
'''
"""
formats block number to uint64 big endian
'''
"""
return utils . zpad ( utils . int_to_big_endian ( number ) , 8 )
def _encode_hex ( v ) :
'''
"""
encodes hash as hex
'''
"""
return ' 0x ' + utils . encode_hex ( v )
class LevelDBReader ( object ) :
'''
"""
level db reading interface , can be used with snapshot
'''
"""
def __init__ ( self , db ) :
self . db = db
@ -50,34 +50,34 @@ class LevelDBReader(object):
self . head_state = None
def _get_head_state ( self ) :
'''
"""
gets head state
'''
"""
if not self . head_state :
root = self . _get_head_block ( ) . state_root
self . head_state = State ( self . db , root )
return self . head_state
def _get_account ( self , address ) :
'''
"""
gets account by address
'''
"""
state = self . _get_head_state ( )
account_address = binascii . a2b_hex ( utils . remove_0x_head ( address ) )
return state . get_and_cache_account ( account_address )
def _get_block_hash ( self , number ) :
'''
"""
gets block hash by block number
'''
"""
num = _format_block_number ( number )
hash_key = header_prefix + num + num_suffix
return self . db . get ( hash_key )
def _get_head_block ( self ) :
'''
"""
gets head block header
'''
"""
if not self . head_block_header :
block_hash = self . db . get ( head_header_key )
num = self . _get_block_number ( block_hash )
@ -91,38 +91,29 @@ class LevelDBReader(object):
return self . head_block_header
def _get_block_number ( self , block_hash ) :
'''
gets block number by its hash
'''
""" Get block number by its hash """
number_key = block_hash_prefix + block_hash
return self . db . get ( number_key )
def _get_block_header ( self , block_hash , num ) :
'''
get block header by block header hash & number
'''
""" Get block header by block header hash & number """
header_key = header_prefix + num + block_hash
block_header_data = self . db . get ( header_key )
header = rlp . decode ( block_header_data , sedes = BlockHeader )
return header
def _get_address_by_hash ( self , block_hash ) :
'''
get mapped address by its hash
'''
""" Get mapped address by its hash """
address_key = address_prefix + block_hash
return self . db . get ( address_key )
def _get_last_indexed_number ( self ) :
'''
latest indexed block number
'''
""" Get latest indexed block number """
return self . db . get ( address_mapping_head_key )
def _get_block_receipts ( self , block_hash , num ) :
'''
get block transaction receipts by block header hash & number
'''
""" Get block transaction receipts by block header hash & number """
number = _format_block_number ( num )
receipts_key = block_receipts_prefix + number + block_hash
receipts_data = self . db . get ( receipts_key )
@ -131,44 +122,44 @@ class LevelDBReader(object):
class LevelDBWriter ( object ) :
'''
"""
level db writing interface
'''
"""
def __init__ ( self , db ) :
self . db = db
self . wb = None
def _set_last_indexed_number ( self , number ) :
'''
"""
sets latest indexed block number
'''
"""
return self . db . put ( address_mapping_head_key , _format_block_number ( number ) )
def _start_writing ( self ) :
'''
"""
start writing a batch
'''
"""
self . wb = self . db . write_batch ( )
def _commit_batch ( self ) :
'''
"""
commit batch
'''
"""
self . wb . write ( )
def _store_account_address ( self , address ) :
'''
"""
get block transaction receipts by block header hash & number
'''
"""
address_key = address_prefix + utils . sha3 ( address )
self . wb . put ( address_key , address )
class EthLevelDB ( object ) :
'''
"""
Go - Ethereum LevelDB client class
'''
"""
def __init__ ( self , path ) :
self . path = path
@ -177,9 +168,9 @@ class EthLevelDB(object):
self . writer = LevelDBWriter ( self . db )
def get_contracts ( self ) :
'''
"""
iterate through all contracts
'''
"""
for account in self . reader . _get_head_state ( ) . get_all_accounts ( ) :
if account . code is not None :
code = _encode_hex ( account . code )
@ -188,9 +179,9 @@ class EthLevelDB(object):
yield contract , account . address , account . balance
def search ( self , expression , callback_func ) :
'''
"""
searches through all contract accounts
'''
"""
cnt = 0
indexer = AccountIndexer ( self )
@ -217,9 +208,7 @@ class EthLevelDB(object):
logging . info ( " Searched %d contracts " % cnt )
def contract_hash_to_address ( self , contract_hash ) :
'''
tries to find corresponding account address
'''
""" Tries to find corresponding account address """
address_hash = binascii . a2b_hex ( utils . remove_0x_head ( contract_hash ) )
indexer = AccountIndexer ( self )
@ -227,17 +216,17 @@ class EthLevelDB(object):
return _encode_hex ( indexer . get_contract_by_hash ( address_hash ) )
def eth_getBlockHeaderByNumber ( self , number ) :
'''
"""
gets block header by block number
'''
"""
block_hash = self . reader . _get_block_hash ( number )
block_number = _format_block_number ( number )
return self . reader . _get_block_header ( block_hash , block_number )
def eth_getBlockByNumber ( self , number ) :
'''
"""
gets block body by block number
'''
"""
block_hash = self . reader . _get_block_hash ( number )
block_number = _format_block_number ( number )
body_key = body_prefix + block_number + block_hash
@ -246,22 +235,22 @@ class EthLevelDB(object):
return body
def eth_getCode ( self , address ) :
'''
"""
gets account code
'''
"""
account = self . reader . _get_account ( address )
return _encode_hex ( account . code )
def eth_getBalance ( self , address ) :
'''
"""
gets account balance
'''
"""
account = self . reader . _get_account ( address )
return account . balance
def eth_getStorageAt ( self , address , position ) :
'''
"""
gets account storage data at position
'''
"""
account = self . reader . _get_account ( address )
return _encode_hex ( utils . zpad ( utils . encode_int ( account . get_storage_data ( position ) ) , 32 ) )
Dominik Muhs
6 years ago