Whitelist permitted types when adding constraint in integer_underflow.py

mythril/analysis/modules/integer_underflow.py

@@ -52,25 +52,28 @@ def execute(statespace):
 
                 logging.debug("[INTEGER_UNDERFLOW] Checking SUB " + str(op0) + ", " + str(op1) + " at address " + str(instruction['address']))
 
-                constraints.append(UGT(op1,op0))
+                allowed_types = [int, BitVecRef, BitVecNumRef]
 
-                try:
+                if type(op0) in allowed_types and type(op1) in allowed_types:
+                    constraints.append(UGT(op1,op0))
 
-                    model = solver.get_model(constraints)
+                    try:
                         
-                    issue = Issue(node.module_name, node.function_name, instruction['address'], "Integer Underflow", "Warning")
+                        model = solver.get_model(constraints)
 
-                    issue.description = "A possible integer underflow exists in the function " + node.function_name + ".\n" \
+                        issue = Issue(node.module_name, node.function_name, instruction['address'], "Integer Underflow", "Warning")
-                        "The SUB instruction at address " + str(instruction['address']) + " may result in a value < 0." 
 
-                    issue.debug = "(" + str(op0) + ") - (" + str(op1) + ").]"
+                        issue.description = "A possible integer underflow exists in the function " + node.function_name + ".\n" \
+                            "The SUB instruction at address " + str(instruction['address']) + " may result in a value < 0." 
 
-                    issues.append(issue)
+                        issue.debug = "(" + str(op0) + ") - (" + str(op1) + ").]"
 
-                    for d in model.decls():
+                        issues.append(issue)
-                        logging.debug("[INTEGER_UNDERFLOW] model: %s = 0x%x" % (d.name(), model[d].as_long()))
 
-                except UnsatError:
+                        for d in model.decls():
-                    logging.debug("[INTEGER_UNDERFLOW] no model found")         
+                            logging.debug("[INTEGER_UNDERFLOW] model: %s = 0x%x" % (d.name(), model[d].as_long()))
+
+                    except UnsatError:
+                        logging.debug("[INTEGER_UNDERFLOW] no model found")         
 
     return issues