VVarCategory: Add a TMP category. (#248)

angr · Oct 17, 2024 · 0767b5e · 0767b5e
1 parent 0989ca3
commit 0767b5e
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/ailment/expression.py b/ailment/expression.py
@@ -213,7 +213,8 @@ class VirtualVariableCategory(IntEnum):
     STACK = 1
     MEMORY = 2
     PARAMETER = 3
-    UNKNOWN = 4
+    TMP = 4
+    UNKNOWN = 5
 
 
 class VirtualVariable(Atom):
@@ -257,6 +258,10 @@ def was_stack(self) -> bool:
     def was_parameter(self) -> bool:
         return self.category == VirtualVariableCategory.PARAMETER
 
+    @property
+    def was_tmp(self) -> bool:
+        return self.category == VirtualVariableCategory.TMP
+
     @property
     def reg_offset(self) -> int | None:
         if self.was_reg:
@@ -269,6 +274,10 @@ def stack_offset(self) -> int | None:
             return self.oident
         return None
 
+    @property
+    def tmp_idx(self) -> int | None:
+        return self.oident if self.was_tmp else None
+
     def likes(self, atom):
         return (
             isinstance(atom, VirtualVariable)