diff options
| -rw-r--r-- | miasm2/expression/simplifications_common.py | 33 | ||||
| -rw-r--r-- | test/expression/simplifications.py | 14 |
2 files changed, 47 insertions, 0 deletions
diff --git a/miasm2/expression/simplifications_common.py b/miasm2/expression/simplifications_common.py index fe69d0b9..a52debe6 100644 --- a/miasm2/expression/simplifications_common.py +++ b/miasm2/expression/simplifications_common.py @@ -327,6 +327,39 @@ def simp_cst_propagation(e_s, e): new_args[i] = ExprOp(op, *arg), bound[i][0], bound[i][1] return ExprCompose(new_args) + # <<<c_rez, >>>c_rez + if op in [">>>c_rez", "<<<c_rez"]: + assert len(args) == 3 + dest, rounds, cf = args + # Skipped if rounds is 0 + if (isinstance(rounds, ExprInt) and + int(rounds.arg) == 0): + return dest + elif all(map(lambda x: isinstance(x, ExprInt), args)): + # The expression can be resolved + tmp = int(dest.arg) + cf = int(cf.arg) + size = dest.size + tmp_count = (int(rounds.arg) & + (0x3f if size == 64 else 0x1f)) % (size + 1) + if op == ">>>c_rez": + while (tmp_count != 0): + tmp_cf = tmp & 1; + tmp = (tmp >> 1) + (cf << (size - 1)) + cf = tmp_cf + tmp_count -= 1 + tmp &= int(dest.mask.arg) + elif op == "<<<c_rez": + while (tmp_count != 0): + tmp_cf = (tmp >> (size - 1)) & 1 + tmp = (tmp << 1) + cf + cf = tmp_cf + tmp_count -= 1 + tmp &= int(dest.mask.arg) + else: + raise RuntimeError("Unknown operation: %s" % op) + return ExprInt(tmp, size=dest.size) + return ExprOp(op, *args) diff --git a/test/expression/simplifications.py b/test/expression/simplifications.py index 60609df4..7b68c1bf 100644 --- a/test/expression/simplifications.py +++ b/test/expression/simplifications.py @@ -18,8 +18,10 @@ f = ExprId('f', size=64) m = ExprMem(a) s = a[:8] +i0 = ExprInt(uint32(0x0)) i1 = ExprInt(uint32(0x1)) i2 = ExprInt(uint32(0x2)) +icustom = ExprInt(uint32(0x12345678)) cc = ExprCond(a, b, c) o = ExprCompose([(a[:8], 8, 16), @@ -301,6 +303,18 @@ to_test = [(ExprInt32(1) - ExprInt32(1), ExprInt32(0)), (expr_cmps(ExprInt32(-10), ExprInt32(-5)), ExprInt1(0)), + (ExprOp("<<<c_rez", i1, i0, i0), + i1), + (ExprOp("<<<c_rez", i1, i1, i0), + ExprInt32(2)), + (ExprOp("<<<c_rez", i1, i1, i1), + ExprInt32(3)), + (ExprOp(">>>c_rez", icustom, i0, i0), + icustom), + (ExprOp(">>>c_rez", icustom, i1, i0), + ExprInt32(0x91A2B3C)), + (ExprOp(">>>c_rez", icustom, i1, i1), + ExprInt32(0x891A2B3C)), ] for e, e_check in to_test[:]: |