Add transform-dis-cases-arm64.S and fix a serious bug thus discovered.

Commit transform-dis-cases-x86_64.S, which I forgot to earlier.

2 files changed, 48 insertions, 0 deletions

diff --git a/test/transform-dis-cases-arm64.S b/test/transform-dis-cases-arm64.S
new file mode 100644
index 0000000..cc3bd4c
--- /dev/null
+++ b/test/transform-dis-cases-arm64.S
@@ -0,0 +1,35 @@
+#define GIVEN .ascii "GIVEN";
+#define EXPECT .ascii "EXPECT";
+#define EXPECT_ERR .ascii "EXPECT_ERR";
+
+/* yay clang, no semicolons allowed */
+
+GIVEN
+    blr x5
+EXPECT
+    blr x5
+
+GIVEN
+    blr x5
+    nop
+EXPECT_ERR
+
+GIVEN
+    cbnz x8, .+0x100
+EXPECT
+    cbz x8, 1f
+    mov  x18, #0x0100
+    movk x18, #0xdead, lsl #16
+    br x18
+    1:
+
+GIVEN
+    0: mov x18, #0x123
+    cbnz x18, 0b+0x100
+EXPECT
+    mov x18, #0x123
+    cbz x18, 1f
+    mov  x17, #0x0100
+    movk x17, #0xdead, lsl #16
+    br x17
+    1:
diff --git a/test/transform-dis-cases-x86_64.S b/test/transform-dis-cases-x86_64.S
new file mode 100644
index 0000000..7dccf24
--- /dev/null
+++ b/test/transform-dis-cases-x86_64.S
@@ -0,0 +1,13 @@
+#define GIVEN .ascii "GIVEN";
+#define EXPECT .ascii "EXPECT";
+#define EXPECT_ERR .ascii "EXPECT_ERR";
+
+GIVEN nopl (%rip)
+EXPECT push %rcx; mov $0xdead0007, %rcx; nopl (%rcx); pop %rcx
+GIVEN mov (%rip), %rax
+EXPECT push %rcx; mov $0xdead0007, %rcx; mov (%rcx), %rax; pop %rcx
+GIVEN mov (%rip), %rcx
+EXPECT push %rax; mov $0xdead0007, %rax; mov (%rax), %rcx; pop %rax
+GIVEN jmpq *(%rip)
+EXPECT push %rax; mov $0xdead0006, %rax; mov %rax, -8(%rsp); pop %rax;
+       jmp *-0x10(%rsp)