fix expression division and add test

cea-sec · Aug 19, 2024 · 17e8f3e · 17e8f3e
1 parent 1ba37c3
commit 17e8f3e
Show file tree

Hide file tree

Showing 2 changed files with 10 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,24 @@
 
 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 F in F
 
 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