From 271cc69e7f8f4411b2021830efbddca305cd690f Mon Sep 17 00:00:00 2001
From: serpilliere <devnull@localhost>
Date: Mon, 5 Sep 2011 13:16:58 +0200
Subject: fix x86_asm ep

---
 example/asm_x86.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'example/asm_x86.py')

diff --git a/example/asm_x86.py b/example/asm_x86.py
index 8e911676..79b61886 100755
--- a/example/asm_x86.py
+++ b/example/asm_x86.py
@@ -16,7 +16,7 @@ my_mn = x86_mn
 my_mn = x86_mn
 
 e = pe_init.PE()
-e.SHList.add_section(name = "text", addr = 0x1000, rawsize = 0x4000)
+s_text = e.SHList.add_section(name = "text", addr = 0x1000, rawsize = 0x4000)
 
 ####filelogger sc####
 all_bloc, symbol_pool = parse_asm.parse_txt(my_mn,r'''
@@ -56,6 +56,7 @@ eend:
 #fix shellcode addr
 symbol_pool.add(asmbloc.asm_label('base_address', 0x400000))
 symbol_pool.getby_name("main").offset = 0x401000
+e.Opthdr.AddressOfEntryPoint = s_text.addr
 
 for b in all_bloc[0]:
     print b
-- 
cgit 1.4.1