diff options
Diffstat (limited to 'example')
| -rw-r--r-- | example/asm/shellcode.py | 9 | ||||
| -rw-r--r-- | example/jitter/unpack_upx.py | 2 |
2 files changed, 8 insertions, 3 deletions
diff --git a/example/asm/shellcode.py b/example/asm/shellcode.py index 253386b5..ce84238a 100644 --- a/example/asm/shellcode.py +++ b/example/asm/shellcode.py @@ -96,8 +96,13 @@ if args.encrypt: patches = new_patches print patches -for offset, raw in patches.items(): - virt[offset] = raw +if isinstance(virt, StrPatchwork): + for offset, raw in patches.items(): + virt[offset] = raw +else: + for offset, raw in patches.items(): + virt.set(offset, raw) + # Produce output open(args.output, 'wb').write(str(output)) diff --git a/example/jitter/unpack_upx.py b/example/jitter/unpack_upx.py index 58507506..c27ecdd8 100644 --- a/example/jitter/unpack_upx.py +++ b/example/jitter/unpack_upx.py @@ -86,7 +86,7 @@ def update_binary(jitter): logging.info('updating binary') for s in sb.pe.SHList: sdata = sb.jitter.vm.get_mem(sb.pe.rva2virt(s.addr), s.rawsize) - sb.pe.virt[sb.pe.rva2virt(s.addr)] = sdata + sb.pe.rva.set(s.addr, sdata) # Stop execution jitter.run = False |