diff options
| -rw-r--r-- | miasm2/core/graph.py | 17 | ||||
| -rw-r--r-- | miasm2/ir/ir.py | 1 |
2 files changed, 11 insertions, 7 deletions
diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py index ad159771..d29517cc 100644 --- a/miasm2/core/graph.py +++ b/miasm2/core/graph.py @@ -27,13 +27,16 @@ class DiGraph: self._nodes_to[n] = [] self._nodes_from[n] = [] - def del_node(self, n): - if n in self._nodes: - self._nodes.remove(n) - for pred in self.predecessors(n): - self.del_edge(pred, n) - for succ in self.successors(n): - self.del_edge(n, succ) + def del_node(self, node): + """Delete the @node of the graph; Also delete every edge to/from this + @node""" + + if node in self._nodes: + self._nodes.remove(node) + for pred in self.predecessors(node): + self.del_edge(pred, node) + for succ in self.successors(node): + self.del_edge(node, succ) def add_edge(self, a, b): if not a in self._nodes: diff --git a/miasm2/ir/ir.py b/miasm2/ir/ir.py index d51693f1..b955bac7 100644 --- a/miasm2/ir/ir.py +++ b/miasm2/ir/ir.py @@ -53,6 +53,7 @@ class irbloc: return dst def set_dst(self, value): + """Find and replace the IRDst affectation's source by @value""" dst = None for ir in self.irs: for i in ir: |