Add dynamic timeout presets

mythril/exceptions.py

@@ -39,3 +39,10 @@ class AddressNotFoundError(MythrilBaseException):
     found."""
 
     pass
+
+
+class UnsupportedModeError(MythrilBaseException):
+    """A Mythril exception denoting an unsupported dynamic timeout mode
+    constraints."""
+
+    pass

mythril/interfaces/cli.py

@@ -15,6 +15,7 @@ import coloredlogs
 import traceback
 
 import mythril.support.signatures as sigs
+from mythril.support import dynamic_timeout
 from mythril.exceptions import AddressNotFoundError, CriticalError
 from mythril.mythril import (
     MythrilAnalyzer,
@@ -217,6 +218,11 @@ def create_parser(parser: argparse.ArgumentParser) -> None:
         default=86400,
         help="The amount of seconds to spend on symbolic execution",
     )
+    options.add_argument(
+        "--dynamic-timeout",
+        choices=["fast", "deep"],
+        help="Choose timeout dynamically for fast or extensive analysis",
+    )
     options.add_argument(
         "--create-timeout",
         type=int,
@@ -401,12 +407,20 @@ def execute_command(
         print(storage)
         return
 
+    if args.dynamic_timeout:
+        timeout = dynamic_timeout.get_timeout(
+            len(disassembler.contracts[0].creation_code), args.dynamic_timeout
+        )
+    else:
+        timeout = args.execution_timeout
+        log.debug("Set dynamic execution timeout: {}".format(timeout))
+
     analyzer = MythrilAnalyzer(
         strategy=args.strategy,
         disassembler=disassembler,
         address=address,
         max_depth=args.max_depth,
-        execution_timeout=args.execution_timeout,
+        execution_timeout=timeout,
         create_timeout=args.create_timeout,
         enable_iprof=args.enable_iprof,
         onchain_storage_access=not args.no_onchain_storage_access,

mythril/support/dynamic_timeout.py

@@ -0,0 +1,17 @@
+from math import exp
+from mythril.exceptions import UnsupportedModeError
+
+
+def get_timeout(bytecode_length: int, mode: str) -> int:
+    """
+    Returns a dynamically calculated timeout based on the length of the bytecode string.
+
+    :param bytecode_length: Length of the bytecode: len("0x....")
+    :param mode: Preset mode (fast or deep)
+    """
+    if mode == "fast":
+        return int(min(120, 15 * exp(bytecode_length / 6000)))
+    elif mode == "deep":
+        return int(min(3600, 30 * exp(bytecode_length / 3000)))
+
+    raise UnsupportedModeError