From f1bd6c0caad44acd41286b1c8876302a90bc6e2f Mon Sep 17 00:00:00 2001
From: Fabrice Desclaux <fabrice.desclaux@cea.fr>
Date: Thu, 22 Oct 2015 09:51:06 +0200
Subject: Loader/pe: fix iat rebuild

---
 miasm2/jitter/loader/pe.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'miasm2/jitter/loader/pe.py')

diff --git a/miasm2/jitter/loader/pe.py b/miasm2/jitter/loader/pe.py
index 3233cd4b..8cda2c99 100644
--- a/miasm2/jitter/loader/pe.py
+++ b/miasm2/jitter/loader/pe.py
@@ -391,7 +391,8 @@ class libimp_pe(libimp):
                 # Find libname's Import Address Table
                 othunk = all_ads[0]
                 i = 0
-                while i + 1 < len(all_ads) and all_ads[i] + 4 == all_ads[i + 1]:
+                while (i + 1 < len(all_ads) and
+                       all_ads[i] + target_pe._wsize / 8 == all_ads[i + 1]):
                     i += 1
                 # 'i + 1' is IAT's length
 
-- 
cgit 1.4.1