issue crytic#2085 fix: return UnaryType instead of UnaryOperationType…

… for Unary IR operation
dokzai · Sep 4, 2023 · 4486a65 · 4486a65
1 parent 0de7a25
commit 4486a65
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/slither/slithir/operations/unary.py b/slither/slithir/operations/unary.py
@@ -35,7 +35,7 @@ def __init__(
         self,
         result: Union[TemporaryVariableSSA, TemporaryVariable],
         variable: Union[Constant, LocalIRVariable, LocalVariable],
-        operation_type: UnaryOperationType,
+        operation_type: UnaryType,
     ) -> None:
         assert is_valid_rvalue(variable)
         assert is_valid_lvalue(result)
@@ -53,7 +53,7 @@ def rvalue(self) -> Union[Constant, LocalVariable]:
         return self._variable
 
     @property
-    def type(self) -> UnaryOperationType:
+    def type(self) -> UnaryType:
         return self._type
 
     @property

diff --git a/slither/visitors/slithir/expression_to_slithir.py b/slither/visitors/slithir/expression_to_slithir.py
@@ -50,6 +50,7 @@
     Member,
     TypeConversion,
     Unary,
+    UnaryType,
     Unpack,
     Return,
     SolidityCall,
@@ -109,6 +110,13 @@ def set_val(expression: Expression, val: Any) -> None:
     BinaryOperationType.OROR: BinaryType.OROR,
 }
 
+
+_unary_to_unary = {
+    UnaryOperationType.BANG: UnaryType.BANG,
+    UnaryOperationType.TILD: UnaryType.TILD,
+}
+
+
 _signed_to_unsigned = {
     BinaryOperationType.DIVISION_SIGNED: BinaryType.DIVISION,
     BinaryOperationType.MODULO_SIGNED: BinaryType.MODULO,
@@ -585,7 +593,7 @@ def _post_unary_operation(self, expression: UnaryOperation) -> None:
         operation: Operation
         if expression.type in [UnaryOperationType.BANG, UnaryOperationType.TILD]:
             lvalue = TemporaryVariable(self._node)
-            operation = Unary(lvalue, value, expression.type)
+            operation = Unary(lvalue, value, _unary_to_unary[expression.type])
             operation.set_expression(expression)
             self._result.append(operation)
             set_val(expression, lvalue)