From bb334f0ab0667181d6e069a8f98ce191fb1fa3dc Mon Sep 17 00:00:00 2001
From: Brian Martin <bmartin@qualbe.com>
Date: Fri, 31 Jul 2015 15:34:30 -0400
Subject: [PATCH] - update route fetcher code

---
 traceroute.py | 31 ++++++++++++++++++++++++-------
 1 file changed, 24 insertions(+), 7 deletions(-)

diff --git a/traceroute.py b/traceroute.py
index 38724dc..1f13e08 100755
--- a/traceroute.py
+++ b/traceroute.py
@@ -303,16 +303,33 @@ class Traceroute(object):
 
     def __get_network_routes(self):
         """
-        Gather network routes on localhost. Only grabs default gateway. Need to play around on different hosts to see what output
-        should be
+        Gather routes from netifaces module
         """
         routes = []
 
-        gw = netifaces.gateways()
-        if 'default' in gw.keys():
-            routes.append( {
-                'default' : gw['default'][netifaces.AF_INET]
-            })
+        gws = netifaces.gateways()
+        for k in gws.keys():
+            if k == 'default':
+                continue
+
+            if len(gws[k]) == 1:
+                (ip,interface,is_gateway) = gws[k][0]
+
+                if is_gateway:
+                    gw_name = 'default'
+                else: # just use the index value from netifaces
+                    gw_name = "{0}".format(k)
+
+                routes.append({
+                       gw_name : {
+                            'ip_address' : ip,
+                            'interface'  : interface
+                       }
+                
+                    }
+                )
+            else:
+                print "Error interpretting network routes. check netifaces output"
 
         return routes