Also pass execution timeout for graph

pull/404/head
Joran Honig 6 years ago
parent b28130cb8e
commit f4da38a2c1
  1. 2
      mythril/interfaces/cli.py
  2. 2
      mythril/laser/ethereum/svm.py
  3. 4
      mythril/mythril.py

@ -194,7 +194,7 @@ def main():
if args.graph: if args.graph:
html = mythril.graph_html(strategy=args.strategy, contract=mythril.contracts[0], address=address, html = mythril.graph_html(strategy=args.strategy, contract=mythril.contracts[0], address=address,
enable_physics=args.enable_physics, phrackify=args.phrack, enable_physics=args.enable_physics, phrackify=args.phrack,
max_depth=args.max_depth) max_depth=args.max_depth, execution_timeout=args.execution_timeout)
try: try:
with open(args.graph, "w") as f: with open(args.graph, "w") as f:

@ -46,7 +46,7 @@ class LaserEVM:
self.work_list = [] self.work_list = []
self.strategy = strategy(self.work_list, max_depth) self.strategy = strategy(self.work_list, max_depth)
self.max_depth = max_depth self.max_depth = max_depth
self.execution_timeout = 20 self.execution_timeout = execution_timeout
self.time = None self.time = None
self.pre_hooks = {} self.pre_hooks = {}

@ -363,10 +363,10 @@ class Mythril(object):
return get_serializable_statespace(sym) return get_serializable_statespace(sym)
def graph_html(self, strategy, contract, address, max_depth=12, enable_physics=False, phrackify=False): def graph_html(self, strategy, contract, address, max_depth=12, enable_physics=False, phrackify=False, execution_timeout=None):
sym = SymExecWrapper(contract, address, strategy, sym = SymExecWrapper(contract, address, strategy,
dynloader=DynLoader(self.eth) if self.dynld else None, dynloader=DynLoader(self.eth) if self.dynld else None,
max_depth=max_depth) max_depth=max_depth, execution_timeout=execution_timeout)
return generate_graph(sym, physics=enable_physics, phrackify=phrackify) return generate_graph(sym, physics=enable_physics, phrackify=phrackify)
def fire_lasers(self, strategy, contracts=None, address=None, def fire_lasers(self, strategy, contracts=None, address=None,

Loading…
Cancel
Save