about summary refs log tree commit diff stats
diff options
context:
space:
mode:
-rw-r--r--miasm2/core/graph.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py
index d97ca8be..ec9eac36 100644
--- a/miasm2/core/graph.py
+++ b/miasm2/core/graph.py
@@ -100,6 +100,11 @@ class DiGraph(object):
         self._nodes_succ[src].remove(dst)
         self._nodes_pred[dst].remove(src)
 
+    def discard_edge(self, src, dst):
+        """Remove edge between @src and @dst if it exits"""
+        if (src, dst) in self._edges:
+            self.del_edge(src, dst)
+
     def predecessors_iter(self, node):
         if not node in self._nodes_pred:
             raise StopIteration