tests: jump_hook: Add is_pic() test

Author: shaymargolis

tests/test_jump_hook.py

@@ -9,6 +9,12 @@
 SECTOR_SIZE = 0x2000
 
 
+@pytest.mark.parametrize('shellcode_run_addr', [
+    (0x82000010),
+    (0xbc100010),
+    (0xbcd00010),
+    (0x91100118),
+])
 @pytest.mark.parametrize('jump_hook_location', [
     0x81000010,
     0xbc000010,
@@ -21,11 +27,12 @@
     0xbcf00070,
     0x910f0218,
 ])
-def test_jump_hook_sanity(temp_dir_path, jump_hook_location, jump_hook_goto):
+def test_jump_hook_sanity(temp_dir_path, shellcode_run_addr, jump_hook_location, jump_hook_goto):
     # Generate shellcode
     # ------------------
     shellcode_address = 0xbfc00000
     jump_hook_sector = int(jump_hook_location/SECTOR_SIZE) * SECTOR_SIZE
+    shellcode_run_sector = int(shellcode_run_addr/SECTOR_SIZE) * SECTOR_SIZE
 
     step = ShellcodeStep(
         "first_step",
@@ -57,15 +64,15 @@ def test_jump_hook_sanity(temp_dir_path, jump_hook_location, jump_hook_goto):
     # --------------------
 
     mu = Uc(UC_ARCH_MIPS, UC_MODE_32 | UC_MODE_BIG_ENDIAN)
-    mu.mem_map(shellcode_address, 0x2000)
+    mu.mem_map(shellcode_run_sector, 0x2000)
     mu.mem_map(jump_hook_sector, 0x2000)
 
     # write machine code to be emulated to memory
-    mu.mem_write(shellcode_address, shellcode)
+    mu.mem_write(shellcode_run_addr, shellcode)
     mu.mem_write(jump_hook_sector, b"\x00" * 0x1000)
 
     # emulate code in infinite time & unlimited instructions
-    mu.emu_start(shellcode_address, shellcode_address + len(shellcode))
+    mu.emu_start(shellcode_run_addr, shellcode_run_addr + len(shellcode))
 
     assert mu.mem_read(jump_hook_location, len(EXPECTED_HOOK)) == EXPECTED_HOOK
     assert mu.mem_read(jump_hook_location+len(EXPECTED_HOOK), 1) == (b"\x00")