about summary refs log tree commit diff stats
path: root/miasm2/analysis/depgraph.py
diff options
context:
space:
mode:
authorCamille Mougey <commial@gmail.com>2016-02-26 16:30:54 +0100
committerCamille Mougey <commial@gmail.com>2016-02-26 16:30:54 +0100
commit9c4d5276b8b29d6f900c8184939468b028933cee (patch)
tree687141798f79fb46921289bf96c022d6011fa7c2 /miasm2/analysis/depgraph.py
parente586d46f44e0a57592930434fde48b6f5b1daf52 (diff)
parent04ae248954d5638aa836cf609e2bb4c50246cbdc (diff)
downloadmiasm-9c4d5276b8b29d6f900c8184939468b028933cee.tar.gz
miasm-9c4d5276b8b29d6f900c8184939468b028933cee.zip
Merge pull request #330 from serpilliere/assignblock
Assignblock
Diffstat (limited to 'miasm2/analysis/depgraph.py')
-rw-r--r--miasm2/analysis/depgraph.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/miasm2/analysis/depgraph.py b/miasm2/analysis/depgraph.py
index a7c16a19..897de77b 100644
--- a/miasm2/analysis/depgraph.py
+++ b/miasm2/analysis/depgraph.py
@@ -832,11 +832,11 @@ class DependencyGraph(object):
             read = set()
             modifier = False
 
-            for affect in self._get_affblock(depnode):
-                if affect.dst == depnode.element:
-                    elements = self._follow_apply_cb(affect.src)
-                    read.update(elements)
-                    modifier = True
+            assignblk = self._get_affblock(depnode)
+            if depnode.element in assignblk:
+                elements = self._follow_apply_cb(assignblk[depnode.element])
+                read.update(elements)
+                modifier = True
 
             # If it's not a modifier affblock, reinject current element
             if not modifier: