From 23b4198eac3f70e708ffcc3ccecb19dfa5cd497e Mon Sep 17 00:00:00 2001
From: coco <coco@hexgolems.com>
Date: Wed, 2 Aug 2017 12:19:19 +0200
Subject: fix atomic_mode leak

---
 miasm2/core/cpu.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'miasm2/core')

diff --git a/miasm2/core/cpu.py b/miasm2/core/cpu.py
index 3502397d..aa94773f 100644
--- a/miasm2/core/cpu.py
+++ b/miasm2/core/cpu.py
@@ -1169,7 +1169,11 @@ class cls_mn(object):
                     if bs_l * 8 - offset_b < l:
                         getok = False
                         break
-                    bv = cls.getbits(bs, mode, offset_b, l)
+                    try:
+                        bv = cls.getbits(bs, mode, offset_b, l)
+                    except:
+                        bs_o.leave_atomic_mode()
+                        raise
                     offset_b += l
                     if not f.fname in fname_values:
                         fname_values[f.fname] = bv
-- 
cgit 1.4.1