From bcfa445359c3ec9136a1b95a476612bd0f4cc98f Mon Sep 17 00:00:00 2001
From: Phil <phil@secdev.org>
Date: Thu, 5 Nov 2009 15:32:55 +0100
Subject: [PATCH] Fixed bug in TracerouteResult.graph() induced by
 [e28139770a3a]

---
 scapy/layers/inet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scapy/layers/inet.py b/scapy/layers/inet.py
index bf670d02..2ff5478e 100644
--- a/scapy/layers/inet.py
+++ b/scapy/layers/inet.py
@@ -1113,8 +1113,8 @@ class TracerouteResult(SndRcvList):
         ports = {}
         ports_done = {}
         for s,r in self.res:
-            r = r[IP] or (conf.ipv6_enabled and r[inet6.IPv6]) or r
-            s = s[IP] or (conf.ipv6_enabled and s[inet6.IPv6]) or s
+            r = r.getlayer(IP) or (conf.ipv6_enabled and r[scapy.layers.inet6.IPv6]) or r
+            s = s.getlayer(IP) or (conf.ipv6_enabled and s[scapy.layers.inet6.IPv6]) or s
             ips[r.src] = None
             if TCP in s:
                 trace_id = (s.src,s.dst,6,s.dport)
-- 
GitLab