Reg test: add arm jit little/big endian

3 files changed, 49 insertions, 6 deletions

diff --git a/example/asm_arm.py b/example/asm_arm.py
index 60e85259..e9ffa330 100644
--- a/example/asm_arm.py
+++ b/example/asm_arm.py
@@ -31,8 +31,7 @@ main:
   MOV    R1, mystrend & 0xffff
   ORR    R1, R1, mystrend & 0xffff0000
 xxx:
-  LDR    R2, [PC, key-$]
-  LDR    R6, [PC, test-$]
+  LDRB    R2, [PC, key-$]
 loop:
   LDRB   R3, [R0]
   EOR    R3, R3, R2
@@ -43,13 +42,11 @@ end:
   MOV    R0, R4
   LDMFD  SP!, {R4, R5, PC}
 key:
-.long 0x11223344
+.byte 0x11
 mystr:
 .string "test string"
 mystrend:
 .long 0
-test:
-.long mystrend - key + 0x1122
 '''
 
 blocs_b, symbol_pool_b = parse_asm.parse_txt(my_mn, "b", txt)
diff --git a/example/test_jit_arm.py b/example/test_jit_arm.py
index 997fa4ff..da4a0e6d 100644
--- a/example/test_jit_arm.py
+++ b/example/test_jit_arm.py
@@ -32,4 +32,3 @@ if options.address is None:
 
 sb.run()
 
-
diff --git a/example/test_jit_arm_sc.py b/example/test_jit_arm_sc.py
new file mode 100644
index 00000000..1ecafe69
--- /dev/null
+++ b/example/test_jit_arm_sc.py
@@ -0,0 +1,47 @@
+#!/usr/bin/env python
+#-*- coding:utf-8 -*-
+from miasm2.analysis import debugging, gdbserver
+
+from miasm2.analysis.sandbox import Sandbox_Linux_armb_str
+from miasm2.analysis.sandbox import Sandbox_Linux_arml_str
+from miasm2.jitter.jitload import vm_load_elf, libimp, preload_elf
+from miasm2.analysis.machine import Machine
+from elfesteem.strpatchwork import StrPatchwork
+import logging
+
+from pdb import pm
+
+parser = Sandbox_Linux_arml_str.parser(description="""Sandbox an elf binary with arm engine
+(ex: test_jit_arm_sc.py example/demo_arm_l.bin)""")
+parser.add_argument("filename", help="string Filename")
+parser.add_argument("endianess", help="endianness [b/l]")
+parser.add_argument('-v', "--verbose",
+                    help="verbose mode", action="store_true")
+
+options = parser.parse_args()
+
+if options.endianess == 'b':
+    sandbox = Sandbox_Linux_armb_str
+elif options.endianess == 'l':
+    sandbox = Sandbox_Linux_arml_str
+else:
+    raise ValueError("Bad endianess!")
+
+sb = sandbox(options.filename, options, globals())
+
+if options.address is None:
+    raise ValueError('invalid address')
+
+sb.run()
+
+# test correct de xor
+start = sb.jitter.cpu.R0
+stop = sb.jitter.cpu.R1
+s = sb.jitter.vm.get_mem(start, stop-start)
+s = StrPatchwork(s)
+for i, c in enumerate(s):
+    s[i] = chr(ord(c)^0x11)
+s = str(s)
+assert(s == "test string\x00")
+
+