diff options
Diffstat (limited to 'example/asm/shellcode.py')
| -rw-r--r-- | example/asm/shellcode.py | 9 |
1 files changed, 7 insertions, 2 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)) |