add test for #8

purseclab · Feb 13, 2024 · b3cfa1b · b3cfa1b
1 parent 6765fba
commit b3cfa1b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tests/test_binaries/x86_64/issue8 b/tests/test_binaries/x86_64/issue8
diff --git a/tests/test_x86_64.py b/tests/test_x86_64.py
@@ -159,6 +159,19 @@ def test_replace_function_patch_with_function_reference_and_rodata(self):
             expected_returnCode=0,
         )
 
+    def test_insert_instruction_patch_with_force_insert(self):
+        instrs = """
+            mov rdi, rax
+            add rdi, 3
+            call {puts}
+        """
+        self.run_one(
+            "issue8",
+            [InsertInstructionPatch(0x117A, instrs, force_insert=True)],
+            expected_output=b"ched failed\n",
+            expected_returnCode=0,
+        )
+
     def run_one(
         self,
         filename,