about summary refs log tree commit diff stats
diff options
context:
space:
mode:
authorFabrice Desclaux <fabrice.desclaux@cea.fr>2016-01-21 16:05:32 +0100
committerserpilliere <fabrice.desclaux@cea.fr>2016-01-21 18:17:35 +0100
commit2df43b1d5a10ab0c883aa440bd1f68141fdfb651 (patch)
treea5c8d72b9b1cff194f5aa6867d22619a79a68b70
parentd49e05f1aef8da814fa7bfc1b99e7051e51db0b0 (diff)
downloadmiasm-2df43b1d5a10ab0c883aa440bd1f68141fdfb651.tar.gz
miasm-2df43b1d5a10ab0c883aa440bd1f68141fdfb651.zip
Asmbloc: Fix asmbloc label (creds @commial)
-rw-r--r--miasm2/core/asmbloc.py3
-rw-r--r--miasm2/ir/ir.py2
2 files changed, 3 insertions, 2 deletions
diff --git a/miasm2/core/asmbloc.py b/miasm2/core/asmbloc.py
index a6d652be..7089e888 100644
--- a/miasm2/core/asmbloc.py
+++ b/miasm2/core/asmbloc.py
@@ -103,7 +103,8 @@ class asm_constraint_to(asm_constraint):
 
 class asm_bloc(object):
 
-    def __init__(self, label=None, alignment=1):
+    def __init__(self, label, alignment=1):
+        assert isinstance(label, asm_label)
         self.bto = set()
         self.lines = []
         self.label = label
diff --git a/miasm2/ir/ir.py b/miasm2/ir/ir.py
index f957fcab..9d90f79a 100644
--- a/miasm2/ir/ir.py
+++ b/miasm2/ir/ir.py
@@ -202,7 +202,7 @@ class ir(object):
         return self.blocs.get(label, None)
 
     def add_instr(self, l, ad=0, gen_pc_updt = False):
-        b = asm_bloc(l)
+        b = asm_bloc(self.gen_label())
         b.lines = [l]
         self.add_bloc(b, gen_pc_updt)