diff options
Diffstat (limited to 'test/arch/x86')
| -rw-r--r-- | test/arch/x86/sem.py | 4 | ||||
| -rw-r--r-- | test/arch/x86/unit/asm_test.py | 2 |
2 files changed, 3 insertions, 3 deletions
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 |