Add xrefs command line option

mythril/ether/contractstorage.py

@@ -1,5 +1,5 @@
 from mythril.rpc.client import EthJsonRpc
-from mythril.ether.ethcontract import ETHContract, InstanceList
+from .mythril.ether.ethcontract import ETHContract, InstanceList
 import hashlib
 import os
 import persistent
@@ -37,6 +37,11 @@ class ContractStorage(persistent.Persistent):
         self.instance_lists= BTree()
         self.last_block = 0
 
+
+    def get_contract_by_hash(self, contract_hash):
+        return self.contracts[contract_hash]
+
+
     def initialize(self, rpchost, rpcport, sync_all):
 
         eth = EthJsonRpc(rpchost, rpcport)

mythril/mythril

@@ -35,6 +35,7 @@ parser.add_argument('-a', '--address', default='0x0123456789ABCDEF0123456789ABCD
 parser.add_argument('-o', '--outfile')
 parser.add_argument('--data', help='message call input data for tracing')
 parser.add_argument('--search', help='search the contract database')
+parser.add_argument('--xrefs', help='get xrefs from contract in database', metavar='CONTRACT_HASH')
 parser.add_argument('--hash', help='calculate function signature hash', metavar='SIGNATURE')
 parser.add_argument('--init-db', action='store_true', help='Initialize the contract database')
 parser.add_argument('--sync-all', action='store_true', help='Also sync contracts with zero balance')
@@ -110,6 +111,12 @@ elif (args.search):
 
     contract_storage.search(args.search, searchCallback)
 
+elif (args.xrefs):
+
+    contract_hash = bytes.fromhex(args.xrefs)
+    contract = contract_storage.get_contract_by_hash(contract_hash)
+    print("\n".join(contract.get_xrefs()))
+
 elif (args.init_db):
     contract_storage.initialize(args.rpchost, args.rpcport, args.sync_all)
 

setup.py

@@ -9,7 +9,7 @@ long_description = '''
 setup(
     name='mythril',
 
-    version='0.2.7',
+    version='0.2.8',
 
     description='A bug hunting tool framework for the Ethereum blockchain',
     long_description=long_description,

~mythril/contractstorage.fs.index

@@ -0,0 +1 @@
+<EFBFBD>K.<EFBFBD>N.

~mythril/contractstorage.fs.lock

@@ -0,0 +1 @@
+ 63031