1 files changed, 92 insertions, 0 deletions
diff --git a/test/arch/mep/asm/ut_helpers_asm.py b/test/arch/mep/asm/ut_helpers_asm.py
new file mode 100644
index 00000000..af010afc
--- /dev/null
+++ b/test/arch/mep/asm/ut_helpers_asm.py
@@ -0,0 +1,92 @@
+# Toshiba MeP-c4 - unit tests helpers
+# Guillaume Valadon <guillaume@valadon.net>
+
+from miasm2.arch.mep.arch import mn_mep
+from miasm2.core.cpu import Disasm_Exception
+from miasm2.core.locationdb import LocationDB
+from miasm2.expression.expression import ExprId, ExprInt, ExprLoc
+
+import re
+
+
+def dis(mn_hex):
+    """Disassembly helper"""
+    mn_bin = mn_hex.decode("hex")
+    try:
+        return mn_mep.dis(mn_bin, "b")
+    except Disasm_Exception:
+        assert(False)  # miasm don't know what to do
+
+
+def check_instruction(mn_str, mn_hex, multi=None, offset=0):
+    """Try to disassemble and assemble this instruction"""
+
+    # Rename objdump registers names
+    mn_str = re.sub("\$([0-9]+)", lambda m: "R"+m.group(1), mn_str)
+    mn_str = mn_str.replace("$", "")
+
+    # Disassemble
+    mn = dis(mn_hex)
+    mn.offset = offset
+    if mn.dstflow():
+
+        # Remember ExprInt arguments sizes
+        args_size = list()
+        for i in range(len(mn.args)):
+            if isinstance(mn.args[i], ExprInt):
+                args_size.append(mn.args[i].size)
+            else:
+                args_size.append(None)
+
+        # Adjust arguments values using the instruction offset
+        loc_db = LocationDB()
+        mn.dstflow2label(loc_db)
+
+        # Convert ExprLoc to ExprInt
+        for i in range(len(mn.args)):
+            if args_size[i] is None:
+                continue
+            if isinstance(mn.args[i], ExprLoc):
+                addr = loc_db.get_location_offset(mn.args[i].loc_key)
+                mn.args[i] = ExprInt(addr, args_size[i])
+
+    print "dis: %s -> %s" % (mn_hex.rjust(20), str(mn).rjust(20))
+    assert(str(mn) == mn_str)  # disassemble assertion
+
+    # Assemble and return all possible candidates
+    instr = mn_mep.fromstring(mn_str, "b")
+    instr.offset = offset
+    instr.mode = "b"
+    if instr.offset:
+        instr.fixDstOffset()
+    asm_list = [i.encode("hex") for i in mn_mep.asm(instr)]
+
+    # Check instructions variants
+    if multi:
+        print "Instructions count:", len(asm_list)
+        assert(len(asm_list) == multi)
+
+        # Ensure that variants correspond to the same disassembled instruction
+        for mn_hex in asm_list:
+            mn = dis(mn_hex)
+            print "dis: %s -> %s" % (mn_hex.rjust(20), str(mn).rjust(20))
+
+    # Check the assembly result
+    print "asm: %s -> %s" % (mn_str.rjust(20),
+                             ", ".join(asm_list).rjust(20))
+    assert(mn_hex in asm_list)  # assemble assertion
+
+
+def launch_tests(obj):
+    """Call test methods by name"""
+
+    test_methods = [name for name in dir(obj) if name.startswith("test")]
+
+    for method in test_methods:
+        print method
+        try:
+            getattr(obj, method)()
+        except AttributeError as e:
+            print "Method not found: %s" % method
+            assert(False)
+        print '-' * 42