2 files changed, 94 insertions, 0 deletions
diff --git a/test/arch/aarch64/unit/asm_test.py b/test/arch/aarch64/unit/asm_test.py
new file mode 100644
index 00000000..60ed418e
--- /dev/null
+++ b/test/arch/aarch64/unit/asm_test.py
@@ -0,0 +1,64 @@
+#! /usr/bin/env python
+import sys
+import os
+
+from miasm2.core.cpu import parse_ast
+from miasm2.arch.aarch64.arch import mn_aarch64, base_expr, variable
+from miasm2.core import parse_asm
+from miasm2.expression.expression import *
+from miasm2.core import asmbloc
+from elfesteem.strpatchwork import StrPatchwork
+from miasm2.analysis.machine import Machine
+from miasm2.jitter.csts import *
+from pdb import pm
+
+
+filename = os.environ.get('PYTHONSTARTUP')
+if filename and os.path.isfile(filename):
+    execfile(filename)
+
+
+reg_and_id = dict(mn_aarch64.regs.all_regs_ids_byname)
+
+class Asm_Test(object):
+    def __init__(self):
+        self.myjit = Machine("aarch64l").jitter()
+        self.myjit.init_stack()
+
+        self.myjit.jit.log_regs = False
+        self.myjit.jit.log_mn = False
+
+
+    def __call__(self):
+        self.asm()
+        self.run()
+        self.check()
+
+
+    def asm(self):
+        blocs, symbol_pool = parse_asm.parse_txt(mn_aarch64, 'l', self.TXT,
+                                                 symbol_pool = self.myjit.ir_arch.symbol_pool)
+        # fix shellcode addr
+        symbol_pool.set_offset(symbol_pool.getby_name("main"), 0x0)
+        s = StrPatchwork()
+        patches = asmbloc.asm_resolve_final(mn_aarch64, blocs[0], symbol_pool)
+        for offset, raw in patches.items():
+            s[offset] = raw
+
+        self.assembly = str(s)
+
+    def run(self):
+        run_addr = 0
+        self.myjit.vm.add_memory_page(run_addr, PAGE_READ | PAGE_WRITE, self.assembly)
+
+        self.myjit.cpu.LR = 0x1337beef
+
+        self.myjit.add_breakpoint(0x1337beef, lambda x:False)
+
+        self.myjit.init_run(run_addr)
+        self.myjit.continue_run()
+
+        assert(self.myjit.pc == 0x1337beef)
+
+    def check(self):
+        raise NotImplementedError('abstract method')
diff --git a/test/arch/aarch64/unit/mn_ubfm.py b/test/arch/aarch64/unit/mn_ubfm.py
new file mode 100644
index 00000000..938f13cf
--- /dev/null
+++ b/test/arch/aarch64/unit/mn_ubfm.py
@@ -0,0 +1,30 @@
+#! /usr/bin/env python
+from asm_test import Asm_Test
+from pdb import pm
+
+
+class Test_UBFM1(Asm_Test):
+    TXT = '''
+main:
+       MOVZ    X0, 0x5600
+       UBFM    X0, X0, 8, 15
+       RET     LR
+    '''
+    def check(self):
+        assert(self.myjit.cpu.X0 == 0x56)
+        pass
+
+class Test_UBFM2(Asm_Test):
+    TXT = '''
+main:
+       MOVZ    X0, 0x56
+       UBFM    X0, X0, 4, 55
+       RET     LR
+    '''
+    def check(self):
+        assert(self.myjit.cpu.X0 == 0x5)
+        pass
+
+
+if __name__ == "__main__":
+    [test()() for test in [Test_UBFM1, Test_UBFM2 ]]