Merge pull request #1502 from W0ni/fix_expression_division

fix expression division
cea-sec · Aug 20, 2024 · 7ffe286 · 7ffe286
2 parents c8aafe1 + 0dfc550
commit 7ffe286
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/miasm/expression/expression.py b/miasm/expression/expression.py
@@ -568,11 +568,11 @@ def __add__(self, other):
     def __sub__(self, other):
         return ExprOp('+', self, ExprOp('-', other))
 
-    def __div__(self, other):
+    def __truediv__(self, other):
         return ExprOp('/', self, other)
 
     def __floordiv__(self, other):
-        return self.__div__(other)
+        return self.__truediv__(other)
 
     def __mod__(self, other):
         return ExprOp('%', self, other)

diff --git a/test/expression/expression.py b/test/expression/expression.py
@@ -80,16 +80,22 @@
 
 C = A+B
 D = C + A
+E = A / B
+F = A // B
+assert E is F
 
 assert A in A
 assert A in C
 assert B in C
 assert C in C
+assert E in E
 
 assert A in D
 assert B in D
 assert C in D
 assert D in D
+assert A in E
+assert B in E
 
 assert C not in A
 assert C not in B