Minor improvements to user_assertions.py

5 years ago · bd8c3f8cd4
parent 2a34a5d712
commit bd8c3f8cd4
1 changed files with 2 additions and 7 deletions
--- a/mythril/analysis/modules/user_assertions.py
+++ b/mythril/analysis/modules/user_assertions.py
@ -55,14 +55,9 @@ class UserAssertions(DetectionModule):
        :param state:
        :return:
        """
-        mem_start = state.mstate.stack[-1]
-        size = state.mstate.stack[-2]
-        topic = state.mstate.stack[-3]
+        topic, size, mem_start = state.mstate.stack[-3:]

-        if topic.symbolic:
-            return []
-
-        if topic.value != assertion_failed_hash:
+        if topic.symbolic or topic.value != assertion_failed_hash:
            return []

        message = None