diff options
Diffstat (limited to 'example/asm')
| -rwxr-xr-x | example/asm/shellcode.py | 3 | ||||
| -rw-r--r-- | example/asm/simple.py | 12 |
2 files changed, 9 insertions, 6 deletions
diff --git a/example/asm/shellcode.py b/example/asm/shellcode.py index 67c882e9..70d844a9 100755 --- a/example/asm/shellcode.py +++ b/example/asm/shellcode.py @@ -83,7 +83,7 @@ with open(args.source) as fstream: loc_db = LocationDB() -asmcfg, loc_db = parse_asm.parse_txt(machine.mn, attrib, source, loc_db) +asmcfg = parse_asm.parse_txt(machine.mn, attrib, source, loc_db) # Fix shellcode addrs loc_db.set_location_offset(loc_db.get_name_location("main"), addr_main) @@ -106,7 +106,6 @@ open("graph.dot", "w").write(asmcfg.dot()) patches = asmblock.asm_resolve_final( machine.mn, asmcfg, - loc_db, dst_interval ) if args.encrypt: diff --git a/example/asm/simple.py b/example/asm/simple.py index 8f6aac92..bfa3ace6 100644 --- a/example/asm/simple.py +++ b/example/asm/simple.py @@ -4,10 +4,12 @@ from pprint import pprint from miasm.arch.x86.arch import mn_x86 from miasm.core import parse_asm, asmblock - +from miasm.core.locationdb import LocationDB # Assemble code -asmcfg, loc_db = parse_asm.parse_txt(mn_x86, 32, ''' +loc_db = LocationDB() +asmcfg = parse_asm.parse_txt( + mn_x86, 32, ''' main: MOV EAX, 1 MOV EBX, 2 @@ -20,13 +22,15 @@ loop: ADD EAX, ECX JZ loop RET -''') +''', + loc_db +) # Set 'main' loc_key's offset loc_db.set_location_offset(loc_db.get_name_location("main"), 0x0) # Spread information and resolve instructions offset -patches = asmblock.asm_resolve_final(mn_x86, asmcfg, loc_db) +patches = asmblock.asm_resolve_final(mn_x86, asmcfg) # Show resolved asmcfg for block in asmcfg.blocks: |