@ -35,6 +35,7 @@ from mythril.ethereum.interface.leveldb.client import EthLevelDB
# logging.basicConfig(level=logging.DEBUG)
class Mythril ( object ) :
"""
Mythril main interface class .
@ -75,9 +76,10 @@ class Mythril(object):
mythril . get_state_variable_from_storage ( args )
"""
def __init__ ( self , solv = None ,
solc_args = None , dynld = False ,
enable_online_lookup = False ) :
def __init__ (
self , solv = None , solc_args = None , dynld = False , enable_online_lookup = False
) :
self . solv = solv
self . solc_args = solc_args
@ -86,18 +88,28 @@ class Mythril(object):
self . mythril_dir = self . _init_mythril_dir ( )
self . sigs = signatures . SignatureDb ( enable_online_lookup = self . enable_online_lookup )
self . sigs = signatures . SignatureDb (
enable_online_lookup = self . enable_online_lookup
)
try :
self . sigs . open ( ) # tries mythril_dir/signatures.json by default (provide path= arg to make this configurable)
except FileNotFoundError :
logging . info (
" No signature database found. Creating database if sigs are loaded in: " + self . sigs . signatures_file + " \n " +
" Consider replacing it with the pre-initialized database at https://raw.githubusercontent.com/ConsenSys/mythril/master/signatures.json " )
" No signature database found. Creating database if sigs are loaded in: "
+ self . sigs . signatures_file
+ " \n "
+ " Consider replacing it with the pre-initialized database at https://raw.githubusercontent.com/ConsenSys/mythril/master/signatures.json "
)
except json . JSONDecodeError as jde :
raise CriticalError ( " Invalid JSON in signatures file " + self . sigs . signatures_file + " \n " + str ( jde ) )
raise CriticalError (
" Invalid JSON in signatures file "
+ self . sigs . signatures_file
+ " \n "
+ str ( jde )
)
self . solc_binary = self . _init_solc_binary ( solv )
self . config_path = os . path . join ( self . mythril_dir , ' config.ini ' )
self . config_path = os . path . join ( self . mythril_dir , " config.ini " )
self . leveldb_dir = self . _init_config ( )
self . eth = None # ethereum API client
@ -108,9 +120,9 @@ class Mythril(object):
@staticmethod
def _init_mythril_dir ( ) :
try :
mythril_dir = os . environ [ ' MYTHRIL_DIR ' ]
mythril_dir = os . environ [ " MYTHRIL_DIR " ]
except KeyError :
mythril_dir = os . path . join ( os . path . expanduser ( ' ~ ' ) , " .mythril " )
mythril_dir = os . path . join ( os . path . expanduser ( " ~ " ) , " .mythril " )
# Initialize data directory and signature database
@ -128,59 +140,75 @@ class Mythril(object):
"""
system = platform . system ( ) . lower ( )
leveldb_fallback_dir = os . path . expanduser ( ' ~ ' )
leveldb_fallback_dir = os . path . expanduser ( " ~ " )
if system . startswith ( " darwin " ) :
leveldb_fallback_dir = os . path . join ( leveldb_fallback_dir , " Library " , " Ethereum " )
leveldb_fallback_dir = os . path . join (
leveldb_fallback_dir , " Library " , " Ethereum "
)
elif system . startswith ( " windows " ) :
leveldb_fallback_dir = os . path . join ( leveldb_fallback_dir , " AppData " , " Roaming " , " Ethereum " )
leveldb_fallback_dir = os . path . join (
leveldb_fallback_dir , " AppData " , " Roaming " , " Ethereum "
)
else :
leveldb_fallback_dir = os . path . join ( leveldb_fallback_dir , " .ethereum " )
leveldb_fallback_dir = os . path . join ( leveldb_fallback_dir , " geth " , " chaindata " )
if not os . path . exists ( self . config_path ) :
logging . info ( " No config file found. Creating default: " + self . config_path )
open ( self . config_path , ' a ' ) . close ( )
open ( self . config_path , " a " ) . close ( )
config = ConfigParser ( allow_no_value = True )
config . optionxform = str
config . read ( self . config_path , ' utf-8 ' )
if ' defaults ' not in config . sections ( ) :
config . read ( self . config_path , " utf-8 " )
if " defaults " not in config . sections ( ) :
self . _add_default_options ( config )
if not config . has_option ( ' defaults ' , ' leveldb_dir ' ) :
if not config . has_option ( " defaults " , " leveldb_dir " ) :
self . _add_leveldb_option ( config , leveldb_fallback_dir )
if not config . has_option ( ' defaults ' , ' dynamic_loading ' ) :
if not config . has_option ( " defaults " , " dynamic_loading " ) :
self . _add_dynamic_loading_option ( config )
with codecs . open ( self . config_path , ' w ' , ' utf-8 ' ) as fp :
with codecs . open ( self . config_path , " w " , " utf-8 " ) as fp :
config . write ( fp )
leveldb_dir = config . get ( ' defaults ' , ' leveldb_dir ' , fallback = leveldb_fallback_dir )
leveldb_dir = config . get (
" defaults " , " leveldb_dir " , fallback = leveldb_fallback_dir
)
return os . path . expanduser ( leveldb_dir )
@staticmethod
def _add_default_options ( config ) :
config . add_section ( ' defaults ' )
config . add_section ( " defaults " )
@staticmethod
def _add_leveldb_option ( config , leveldb_fallback_dir ) :
config . set ( ' defaults ' , " #Default chaindata locations: " )
config . set ( ' defaults ' , " #– Mac: ~/Library/Ethereum/geth/chaindata " )
config . set ( ' defaults ' , " #– Linux: ~/.ethereum/geth/chaindata " )
config . set ( ' defaults ' , " #– Windows: % USERPROFILE % \\ AppData \\ Roaming \\ Ethereum \\ geth \\ chaindata " )
config . set ( ' defaults ' , ' leveldb_dir ' , leveldb_fallback_dir )
config . set ( " defaults " , " #Default chaindata locations: " )
config . set ( " defaults " , " #– Mac: ~/Library/Ethereum/geth/chaindata " )
config . set ( " defaults " , " #– Linux: ~/.ethereum/geth/chaindata " )
config . set (
" defaults " ,
" #– Windows: % USERPROFILE % \\ AppData \\ Roaming \\ Ethereum \\ geth \\ chaindata " ,
)
config . set ( " defaults " , " leveldb_dir " , leveldb_fallback_dir )
@staticmethod
def _add_dynamic_loading_option ( config ) :
config . set ( ' defaults ' , ' #– To connect to Infura use dynamic_loading: infura ' )
config . set ( ' defaults ' , ' #– To connect to Rpc use '
' dynamic_loading: HOST:PORT / ganache / infura-[network_name] ' )
config . set ( ' defaults ' , ' #– To connect to local host use dynamic_loading: localhost ' )
config . set ( ' defaults ' , ' dynamic_loading ' , ' infura ' )
config . set ( " defaults " , " #– To connect to Infura use dynamic_loading: infura " )
config . set (
" defaults " ,
" #– To connect to Rpc use "
" dynamic_loading: HOST:PORT / ganache / infura-[network_name] " ,
)
config . set (
" defaults " , " #– To connect to local host use dynamic_loading: localhost "
)
config . set ( " defaults " , " dynamic_loading " , " infura " )
def analyze_truffle_project ( self , * args , * * kwargs ) :
return analyze_truffle_project ( self . sigs , * args , * * kwargs ) # just passthru by passing signatures for now
return analyze_truffle_project (
self . sigs , * args , * * kwargs
) # just passthru by passing signatures for now
@staticmethod
def _init_solc_binary ( version ) :
@ -190,27 +218,31 @@ class Mythril(object):
if version :
# tried converting input to semver, seemed not necessary so just slicing for now
if version == str ( solc . main . get_solc_version ( ) ) [ : 6 ] :
logging . info ( ' Given version matches installed version ' )
logging . info ( " Given version matches installed version " )
try :
solc_binary = os . environ [ ' SOLC ' ]
solc_binary = os . environ [ " SOLC " ]
except KeyError :
solc_binary = ' solc '
solc_binary = " solc "
else :
if util . solc_exists ( version ) :
logging . info ( ' Given version is already installed ' )
logging . info ( " Given version is already installed " )
else :
try :
solc . install_solc ( ' v ' + version )
solc . install_solc ( " v " + version )
except SolcError :
raise CriticalError ( " There was an error when trying to install the specified solc version " )
raise CriticalError (
" There was an error when trying to install the specified solc version "
)
solc_binary = os . path . join ( os . environ [ ' HOME ' ] , " .py-solc/solc-v " + version , " bin/solc " )
solc_binary = os . path . join (
os . environ [ " HOME " ] , " .py-solc/solc-v " + version , " bin/solc "
)
logging . info ( " Setting the compiler to " + str ( solc_binary ) )
else :
try :
solc_binary = os . environ [ ' SOLC ' ]
solc_binary = os . environ [ " SOLC " ]
except KeyError :
solc_binary = ' solc '
solc_binary = " solc "
return solc_binary
def set_api_leveldb ( self , leveldb ) :
@ -219,22 +251,24 @@ class Mythril(object):
return self . eth
def set_api_rpc_infura ( self ) :
self . eth = EthJsonRpc ( ' mainnet.infura.io ' , 443 , True )
self . eth = EthJsonRpc ( " mainnet.infura.io " , 443 , True )
logging . info ( " Using INFURA for RPC queries " )
def set_api_rpc ( self , rpc = None , rpctls = False ) :
if rpc == ' ganache ' :
rpcconfig = ( ' localhost ' , 8545 , False )
if rpc == " ganache " :
rpcconfig = ( " localhost " , 8545 , False )
else :
m = re . match ( r ' infura-(.*) ' , rpc )
if m and m . group ( 1 ) in [ ' mainnet ' , ' rinkeby ' , ' kovan ' , ' ropsten ' ] :
rpcconfig = ( m . group ( 1 ) + ' .infura.io ' , 443 , True )
m = re . match ( r " infura-(.*) " , rpc )
if m and m . group ( 1 ) in [ " mainnet " , " rinkeby " , " kovan " , " ropsten " ] :
rpcconfig = ( m . group ( 1 ) + " .infura.io " , 443 , True )
else :
try :
host , port = rpc . split ( " : " )
rpcconfig = ( host , int ( port ) , rpctls )
except ValueError :
raise CriticalError ( " Invalid RPC argument, use ' ganache ' , ' infura-[network] ' or ' HOST:PORT ' " )
raise CriticalError (
" Invalid RPC argument, use ' ganache ' , ' infura-[network] ' or ' HOST:PORT ' "
)
if rpcconfig :
self . eth = EthJsonRpc ( rpcconfig [ 0 ] , int ( rpcconfig [ 1 ] ) , rpcconfig [ 2 ] )
@ -243,26 +277,25 @@ class Mythril(object):
raise CriticalError ( " Invalid RPC settings, check help for details. " )
def set_api_rpc_localhost ( self ) :
self . eth = EthJsonRpc ( ' localhost ' , 8545 )
self . eth = EthJsonRpc ( " localhost " , 8545 )
logging . info ( " Using default RPC settings: http://localhost:8545 " )
def set_api_from_config_path ( self ) :
config = ConfigParser ( allow_no_value = False )
config . optionxform = str
config . read ( self . config_path , ' utf-8 ' )
if config . has_option ( ' defaults ' , ' dynamic_loading ' ) :
dynamic_loading = config . get ( ' defaults ' , ' dynamic_loading ' )
config . read ( self . config_path , " utf-8 " )
if config . has_option ( " defaults " , " dynamic_loading " ) :
dynamic_loading = config . get ( " defaults " , " dynamic_loading " )
else :
dynamic_loading = ' infura '
if dynamic_loading == ' infura ' :
dynamic_loading = " infura "
if dynamic_loading == " infura " :
self . set_api_rpc_infura ( )
elif dynamic_loading == ' localhost ' :
elif dynamic_loading == " localhost " :
self . set_api_rpc_localhost ( )
else :
self . set_api_rpc ( dynamic_loading )
def search_db ( self , search ) :
def search_callback ( _ , address , balance ) :
print ( " Address: " + address + " , balance: " + str ( balance ) )
@ -273,18 +306,22 @@ class Mythril(object):
raise CriticalError ( " Syntax error in search expression. " )
def contract_hash_to_address ( self , hash ) :
if not re . match ( r ' 0x[a-fA-F0-9] {64} ' , hash ) :
if not re . match ( r " 0x[a-fA-F0-9] {64} " , hash ) :
raise CriticalError ( " Invalid address hash. Expected format is ' 0x... ' . " )
print ( self . eth_db . contract_hash_to_address ( hash ) )
def load_from_bytecode ( self , code ) :
address = util . get_indexed_address ( 0 )
self . contracts . append ( ETHContract ( code , name = " MAIN " , enable_online_lookup = self . enable_online_lookup ) )
self . contracts . append (
ETHContract (
code , name = " MAIN " , enable_online_lookup = self . enable_online_lookup
)
)
return address , self . contracts [ - 1 ] # return address and contract object
def load_from_address ( self , address ) :
if not re . match ( r ' 0x[a-fA-F0-9] {40} ' , address ) :
if not re . match ( r " 0x[a-fA-F0-9] {40} " , address ) :
raise CriticalError ( " Invalid contract address. Expected format is ' 0x... ' . " )
try :
@ -292,14 +329,24 @@ class Mythril(object):
except FileNotFoundError as e :
raise CriticalError ( " IPC error: " + str ( e ) )
except ConnectionError :
raise CriticalError ( " Could not connect to RPC server. Make sure that your node is running and that RPC parameters are set correctly. " )
raise CriticalError (
" Could not connect to RPC server. Make sure that your node is running and that RPC parameters are set correctly. "
)
except Exception as e :
raise CriticalError ( " IPC / RPC error: " + str ( e ) )
else :
if code == " 0x " or code == " 0x0 " :
raise CriticalError ( " Received an empty response from eth_getCode. Check the contract address and verify that you are on the correct chain. " )
raise CriticalError (
" Received an empty response from eth_getCode. Check the contract address and verify that you are on the correct chain. "
)
else :
self . contracts . append ( ETHContract ( code , name = address , enable_online_lookup = self . enable_online_lookup ) )
self . contracts . append (
ETHContract (
code ,
name = address ,
enable_online_lookup = self . enable_online_lookup ,
)
)
return address , self . contracts [ - 1 ] # return address and contract object
def load_from_solidity ( self , solidity_files ) :
@ -320,57 +367,105 @@ class Mythril(object):
try :
# import signatures from solidity source
self . sigs . import_from_solidity_source ( file , solc_binary = self . solc_binary , solc_args = self . solc_args )
self . sigs . import_from_solidity_source (
file , solc_binary = self . solc_binary , solc_args = self . solc_args
)
# Save updated function signatures
self . sigs . write ( ) # dump signatures to disk (previously opened file or default location)
if contract_name is not None :
contract = SolidityContract ( file , contract_name , solc_args = self . solc_args )
contract = SolidityContract (
file , contract_name , solc_args = self . solc_args
)
self . contracts . append ( contract )
contracts . append ( contract )
else :
for contract in get_contracts_from_file ( file , solc_args = self . solc_args ) :
for contract in get_contracts_from_file (
file , solc_args = self . solc_args
) :
self . contracts . append ( contract )
contracts . append ( contract )
except FileNotFoundError :
raise CriticalError ( " Input file not found: " + file )
except CompilerError as e :
raise CriticalError ( e )
except NoContractFoundError :
logging . info ( " The file " + file + " does not contain a compilable contract. " )
logging . info (
" The file " + file + " does not contain a compilable contract. "
)
return address , contracts
def dump_statespace ( self , strategy , contract , address = None , max_depth = None ,
execution_timeout = None , create_timeout = None ) :
sym = SymExecWrapper ( contract , address , strategy ,
def dump_statespace (
self ,
strategy ,
contract ,
address = None ,
max_depth = None ,
execution_timeout = None ,
create_timeout = None ,
) :
sym = SymExecWrapper (
contract ,
address ,
strategy ,
dynloader = DynLoader ( self . eth ) if self . dynld else None ,
max_depth = max_depth , execution_timeout = execution_timeout , create_timeout = create_timeout )
max_depth = max_depth ,
execution_timeout = execution_timeout ,
create_timeout = create_timeout ,
)
return get_serializable_statespace ( sym )
def graph_html ( self , strategy , contract , address , max_depth = None , enable_physics = False ,
phrackify = False , execution_timeout = None , create_timeout = None ) :
sym = SymExecWrapper ( contract , address , strategy ,
def graph_html (
self ,
strategy ,
contract ,
address ,
max_depth = None ,
enable_physics = False ,
phrackify = False ,
execution_timeout = None ,
create_timeout = None ,
) :
sym = SymExecWrapper (
contract ,
address ,
strategy ,
dynloader = DynLoader ( self . eth ) if self . dynld else None ,
max_depth = max_depth , execution_timeout = execution_timeout , create_timeout = create_timeout )
max_depth = max_depth ,
execution_timeout = execution_timeout ,
create_timeout = create_timeout ,
)
return generate_graph ( sym , physics = enable_physics , phrackify = phrackify )
def fire_lasers ( self , strategy , contracts = None , address = None ,
modules = None , verbose_report = False , max_depth = None , execution_timeout = None , create_timeout = None ,
max_transaction_count = None ) :
def fire_lasers (
self ,
strategy ,
contracts = None ,
address = None ,
modules = None ,
verbose_report = False ,
max_depth = None ,
execution_timeout = None ,
create_timeout = None ,
max_transaction_count = None ,
) :
all_issues = [ ]
for contract in ( contracts or self . contracts ) :
sym = SymExecWrapper ( contract , address , strategy ,
for contract in contracts or self . contracts :
sym = SymExecWrapper (
contract ,
address ,
strategy ,
dynloader = DynLoader ( self . eth ) if self . dynld else None ,
max_depth = max_depth , execution_timeout = execution_timeout ,
max_depth = max_depth ,
execution_timeout = execution_timeout ,
create_timeout = create_timeout ,
max_transaction_count = max_transaction_count )
max_transaction_count = max_transaction_count ,
)
issues = fire_lasers ( sym , modules )
@ -398,9 +493,14 @@ class Mythril(object):
position = int ( params [ 1 ] )
position_formatted = utils . zpad ( utils . int_to_big_endian ( position ) , 32 )
for i in range ( 2 , len ( params ) ) :
key = bytes ( params [ i ] , ' utf8 ' )
key = bytes ( params [ i ] , " utf8 " )
key_formatted = utils . rzpad ( key , 32 )
mappings . append ( int . from_bytes ( utils . sha3 ( key_formatted + position_formatted ) , byteorder = ' big ' ) )
mappings . append (
int . from_bytes (
utils . sha3 ( key_formatted + position_formatted ) ,
byteorder = " big " ,
)
)
length = len ( mappings )
if length == 1 :
@ -415,30 +515,51 @@ class Mythril(object):
if len ( params ) > = 2 :
length = int ( params [ 1 ] )
if len ( params ) == 3 and params [ 2 ] == " array " :
position_formatted = utils . zpad ( utils . int_to_big_endian ( position ) , 32 )
position = int . from_bytes ( utils . sha3 ( position_formatted ) , byteorder = ' big ' )
position_formatted = utils . zpad (
utils . int_to_big_endian ( position ) , 32
)
position = int . from_bytes (
utils . sha3 ( position_formatted ) , byteorder = " big "
)
except ValueError :
raise CriticalError ( " Invalid storage index. Please provide a numeric value. " )
raise CriticalError (
" Invalid storage index. Please provide a numeric value. "
)
outtxt = [ ]
try :
if length == 1 :
outtxt . append ( " {} : {} " . format ( position , self . eth . eth_getStorageAt ( address , position ) ) )
outtxt . append (
" {} : {} " . format (
position , self . eth . eth_getStorageAt ( address , position )
)
)
else :
if len ( mappings ) > 0 :
for i in range ( 0 , len ( mappings ) ) :
position = mappings [ i ]
outtxt . append ( " {} : {} " . format ( hex ( position ) , self . eth . eth_getStorageAt ( address , position ) ) )
outtxt . append (
" {} : {} " . format (
hex ( position ) ,
self . eth . eth_getStorageAt ( address , position ) ,
)
)
else :
for i in range ( position , position + length ) :
outtxt . append ( " {} : {} " . format ( hex ( i ) , self . eth . eth_getStorageAt ( address , i ) ) )
outtxt . append (
" {} : {} " . format (
hex ( i ) , self . eth . eth_getStorageAt ( address , i )
)
)
except FileNotFoundError as e :
raise CriticalError ( " IPC error: " + str ( e ) )
except ConnectionError :
raise CriticalError ( " Could not connect to RPC server. Make sure that your node is running and that RPC parameters are set correctly. " )
return ' \n ' . join ( outtxt )
raise CriticalError (
" Could not connect to RPC server. Make sure that your node is running and that RPC parameters are set correctly. "
)
return " \n " . join ( outtxt )
@staticmethod
def disassemble ( contract ) :
JoranHonig
6 years ago
committed by
GitHub