From b2f02006df9487905384cd7654e003dbf01419a0 Mon Sep 17 00:00:00 2001
From: serpilliere <devnull@localhost>
Date: Fri, 13 Jun 2014 09:36:58 +0200
Subject: Simplification: detect and handle == condition Add corresponding
 regression tests

---
 miasm2/expression/simplifications.py        |  4 +++-
 miasm2/expression/simplifications_common.py | 15 ---------------
 miasm2/expression/simplifications_cond.py   | 30 ++++++++++++++++++++++++++++-
 3 files changed, 32 insertions(+), 17 deletions(-)

(limited to 'miasm2/expression')

diff --git a/miasm2/expression/simplifications.py b/miasm2/expression/simplifications.py
index d633cf3e..cbffb219 100644
--- a/miasm2/expression/simplifications.py
+++ b/miasm2/expression/simplifications.py
@@ -40,7 +40,9 @@ class ExpressionSimplifier(object):
                                      simplifications_cond.expr_simp_inf_unsigned_inversed],
                  m2_expr.ExprOp: [simplifications_cond.exec_inf_unsigned,
                                   simplifications_cond.exec_inf_signed,
-                                  simplifications_cond.expr_simp_inverse],
+                                  simplifications_cond.expr_simp_inverse,
+                                  simplifications_cond.exec_equal],
+                 m2_expr.ExprCond: [simplifications_cond.expr_simp_equal]
                  }
 
 
diff --git a/miasm2/expression/simplifications_common.py b/miasm2/expression/simplifications_common.py
index c907fe84..792a5a05 100644
--- a/miasm2/expression/simplifications_common.py
+++ b/miasm2/expression/simplifications_common.py
@@ -222,21 +222,6 @@ def simp_cst_propagation(e_s, e):
             2 ** args[1].arg >= args[0].args[1].arg):
             return ExprInt_from(args[0], 0)
 
-    # int == int => 0 or 1
-    if (op == '==' and
-        isinstance(args[0], ExprInt) and
-        isinstance(args[1], ExprInt)):
-        if args[0].arg == args[1].arg:
-            return ExprInt_from(args[0], 1)
-        else:
-            return ExprInt_from(args[0], 0)
-    #(A|int == 0)  => 0  with int != 0
-    if op == '==' and isinstance(args[1], ExprInt) and args[1].arg == 0:
-        if isinstance(args[0], ExprOp) and args[0].op == '|' and\
-                isinstance(args[0].args[1], ExprInt) and \
-                args[0].args[1].arg != 0:
-            return ExprInt_from(args[0], 0)
-
     # parity(int) => int
     if op == 'parity' and isinstance(args[0], ExprInt):
         return ExprInt1(parity(args[0].arg))
diff --git a/miasm2/expression/simplifications_cond.py b/miasm2/expression/simplifications_cond.py
index ee23f6db..6688bc31 100644
--- a/miasm2/expression/simplifications_cond.py
+++ b/miasm2/expression/simplifications_cond.py
@@ -8,7 +8,7 @@
 # - Simplifications to catch known condition forms
 #
 # Conditions currently supported :
-# <u, <s
+# <u, <s, ==
 #
 # Authors : Fabrice DESCLAUX (CEA/DAM), Camille MOUGEY (CEA/DAM)
 #
@@ -54,6 +54,10 @@ def ExprOp_inf_unsigned(arg1, arg2):
     "Return an ExprOp standing for arg1 <s arg2"
     return __ExprOp_cond(TOK_INF_UNSIGNED, arg1, arg2)
 
+def ExprOp_equal(arg1, arg2):
+    "Return an ExprOp standing for arg1 == arg2"
+    return __ExprOp_cond(TOK_EQUAL, arg1, arg2)
+
 
 # Catching conditions forms
 
@@ -174,6 +178,17 @@ def expr_simp_inverse(expr_simp, e):
     else:
         return ExprOp_inf_unsigned(r[jok1], r[jok2])
 
+def expr_simp_equal(expr_simp, e):
+    """(x - y)?(0:1) == (x == y)"""
+
+    to_match = m2_expr.ExprCond(jok1 + jok2, m2_expr.ExprInt1(0), m2_expr.ExprInt1(1))
+    r = __MatchExprWrap(e,
+                        to_match,
+                        [jok1, jok2])
+    if r is False:
+        return e
+
+    return ExprOp_equal(r[jok1], expr_simp(-r[jok2]))
 
 # Compute conditions
 
@@ -206,6 +221,7 @@ def __comp_signed(arg1, arg2):
 
 def exec_inf_signed(expr_simp, e):
     "Compute x <s y"
+
     if e.op != TOK_INF_SIGNED:
         return e
 
@@ -215,3 +231,15 @@ def exec_inf_signed(expr_simp, e):
         return __comp_signed(arg1, arg2)
     else:
         return e
+
+def exec_equal(expr_simp, e):
+    "Compute x == y"
+
+    if e.op != TOK_EQUAL:
+        return e
+
+    arg1, arg2 = e.args
+    if isinstance(arg1, m2_expr.ExprInt) and isinstance(arg2, m2_expr.ExprInt):
+        return m2_expr.ExprInt1(1) if (arg1.arg == arg2.arg) else m2_expr.ExprInt1(0)
+    else:
+        return e
-- 
cgit 1.4.1