diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/arch/aarch64/unit/asm_test.py | 2 | ||||
| -rw-r--r-- | test/arch/mips32/unit/asm_test.py | 2 | ||||
| -rw-r--r-- | test/arch/x86/sem.py | 4 | ||||
| -rw-r--r-- | test/arch/x86/unit/asm_test.py | 2 |
4 files changed, 5 insertions, 5 deletions
diff --git a/test/arch/aarch64/unit/asm_test.py b/test/arch/aarch64/unit/asm_test.py index 60ed418e..9e0d5ea8 100644 --- a/test/arch/aarch64/unit/asm_test.py +++ b/test/arch/aarch64/unit/asm_test.py @@ -41,7 +41,7 @@ class Asm_Test(object): # 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) + patches = asmbloc.asm_resolve_final(mn_aarch64, blocs, symbol_pool) for offset, raw in patches.items(): s[offset] = raw diff --git a/test/arch/mips32/unit/asm_test.py b/test/arch/mips32/unit/asm_test.py index b6cb7b2d..a00d0842 100644 --- a/test/arch/mips32/unit/asm_test.py +++ b/test/arch/mips32/unit/asm_test.py @@ -41,7 +41,7 @@ class Asm_Test(object): # fix shellcode addr symbol_pool.set_offset(symbol_pool.getby_name("main"), 0x0) s = StrPatchwork() - patches = asmbloc.asm_resolve_final(mn_mips32, blocs[0], symbol_pool) + patches = asmbloc.asm_resolve_final(mn_mips32, blocs, symbol_pool) for offset, raw in patches.items(): s[offset] = raw diff --git a/test/arch/x86/sem.py b/test/arch/x86/sem.py index b80ab33d..617b929b 100644 --- a/test/arch/x86/sem.py +++ b/test/arch/x86/sem.py @@ -47,9 +47,9 @@ def compute(ir, mode, asm, inputstate={}, debug=False): def compute_txt(ir, mode, txt, inputstate={}, debug=False): blocs, symbol_pool = parse_asm.parse_txt(mn, mode, txt) symbol_pool.set_offset(symbol_pool.getby_name("main"), 0x0) - patches = asmbloc.asm_resolve_final(mn, blocs[0], symbol_pool) + patches = asmbloc.asm_resolve_final(mn, blocs, symbol_pool) interm = ir(symbol_pool) - for bbl in blocs[0]: + for bbl in blocs: interm.add_bloc(bbl) return symb_exec(interm, inputstate, debug) diff --git a/test/arch/x86/unit/asm_test.py b/test/arch/x86/unit/asm_test.py index c6381d9e..bf609aa5 100644 --- a/test/arch/x86/unit/asm_test.py +++ b/test/arch/x86/unit/asm_test.py @@ -41,7 +41,7 @@ class Asm_Test(object): # fix shellcode addr symbol_pool.set_offset(symbol_pool.getby_name("main"), 0x0) s = StrPatchwork() - patches = asmbloc.asm_resolve_final(mn_x86, blocs[0], symbol_pool) + patches = asmbloc.asm_resolve_final(mn_x86, blocs, symbol_pool) for offset, raw in patches.items(): s[offset] = raw |