Codegen: simplify high level op

1 files changed, 8 insertions, 0 deletions

diff --git a/miasm2/jitter/codegen.py b/miasm2/jitter/codegen.py
index abba9843..fec10f41 100644
--- a/miasm2/jitter/codegen.py
+++ b/miasm2/jitter/codegen.py
@@ -8,6 +8,7 @@ from miasm2.ir.ir import IRBlock, AssignBlock
 
 from miasm2.ir.translators.C import TranslatorC, int_size_to_bn
 from miasm2.core.asmblock import AsmBlockBad
+from miasm2.expression.simplifications_high_level import expr_simp
 
 TRANSLATOR_NO_SYMBOL = TranslatorC(loc_db=None)
 
@@ -166,6 +167,13 @@ class CGen(object):
             irblock_head = self.assignblk_to_irbloc(instr, assignblk_head)
             irblocks = [irblock_head] + assignblks_extra
 
+            # Simplify high level operators
+            out = []
+            for irblock in irblocks:
+                new_irblock = irblock.simplify(expr_simp)[1]
+                out.append(new_irblock)
+            irblocks = out
+
             for irblock in irblocks:
                 assert irblock.dst is not None
             irblocks_list.append(irblocks)