about summary refs log tree commit diff stats
path: root/example/asm/shellcode.py
diff options
context:
space:
mode:
authorFabrice Desclaux <fabrice.desclaux@cea.fr>2015-12-13 13:38:54 +0100
committerFabrice Desclaux <fabrice.desclaux@cea.fr>2015-12-16 15:55:55 +0100
commit8ebdbffdc6c86e9b832869046dfc6e11a8fb21be (patch)
treebde0bed0794fd7e52130c00c954212dbc9bed7f3 /example/asm/shellcode.py
parentb998b1a8b2814585692e885e656c48357fbbb9b8 (diff)
downloadmiasm-8ebdbffdc6c86e9b832869046dfc6e11a8fb21be.tar.gz
miasm-8ebdbffdc6c86e9b832869046dfc6e11a8fb21be.zip
Update api according to Elfesteem
Diffstat (limited to 'example/asm/shellcode.py')
-rw-r--r--example/asm/shellcode.py9
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))