From 725b8d3a3d322b0242f685f37047edf8bf3ea757 Mon Sep 17 00:00:00 2001
From: Mehdi Abaakouk <sileht@sileht.net>
Date: Fri, 27 Jul 2012 23:12:11 +0200
Subject: [PATCH] basic caracter escape for pydot html label

---
 lg.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/lg.py b/lg.py
index 0913317..2c69953 100755
--- a/lg.py
+++ b/lg.py
@@ -361,9 +361,15 @@ def show_bgpmap():
     nodes = {}
     edges = {}
 
+    def escape(label):
+        label = label.replace("&", "&amp;")
+        label = label.replace(">", "&gt;")
+        label = label.replace("<", "&lt;")
+        return label
+
     def add_node(_as, **kwargs):
         if _as not in nodes:
-            kwargs["label"] = '<<TABLE CELLBORDER="0" BORDER="0" CELLPADDING="0" CELLSPACING="0"><TR><TD ALIGN="CENTER">' + kwargs.get("label", get_as_name(_as)).replace("\r","<BR/>") + "</TD></TR></TABLE>>"
+            kwargs["label"] = '<<TABLE CELLBORDER="0" BORDER="0" CELLPADDING="0" CELLSPACING="0"><TR><TD ALIGN="CENTER">' + escape(kwargs.get("label", get_as_name(_as))).replace("\r","<BR/>") + "</TD></TR></TABLE>>"
             nodes[_as] = pydot.Node(_as, style="filled", fontsize="10", **kwargs)
             graph.add_node(nodes[_as])
         return nodes[_as]
@@ -385,8 +391,9 @@ def show_bgpmap():
             if "%s*" % label_without_star not in labels:
                 labels = [ kwargs["label"] ]  + [ l for l in labels if not l.startswith(label_without_star) ] 
                 labels = sorted(labels, cmp=lambda x,y: x.endswith("*") and -1 or 1)
-
-                e.set_label("\r".join(labels))
+                
+                label = escape("\r".join(labels))
+                e.set_label(label)
         return edges[edge_tuple]
 
     for host, asmaps in data.iteritems():