Disassembler: prevent duplicate edges

Consensys · Oct 4, 2017 · 9006eba · 9006eba
1 parent 0a4b8dc
commit 9006eba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/mythril/disassembler/disassembly.py b/mythril/disassembler/disassembly.py
@@ -114,12 +114,15 @@ def __init__(self, code):
                 except IndexError:
                     continue
 
-                self.xrefs.append((block.id, self.blocks[j].id))
+
+                if not (block.id, self.blocks[j].id) in self.xrefs:
+                    self.xrefs.append((block.id, self.blocks[j].id))
 
             # if the last instruction isn't an unconditional jump or halt, also add a reference to the following block
 
             if (block.id < len(self.blocks)) and (block.instruction_list[block.length - 1]['opcode'] not in ['JUMP', 'STOP', 'THROW', 'REVERT', 'INVALID']):
-                self.xrefs.append((block.id, block.id + 1))
+                if not (block.id, self.blocks[j].id) in self.xrefs:
+                    self.xrefs.append((block.id, block.id + 1))
 
 
     def get_easm(self):

diff --git a/setup.py b/setup.py
@@ -219,7 +219,7 @@
 setup(
     name='mythril',
 
-    version='0.3.8',
+    version='0.3.9',
 
     description='A reversing and bug hunting framework for the Ethereum blockchain',
     long_description=long_description,