1 files changed, 68 insertions, 0 deletions
diff --git a/haircontrol/discovery.py b/haircontrol/discovery.py
new file mode 100644
index 0000000..d454283
--- /dev/null
+++ b/haircontrol/discovery.py
@@ -0,0 +1,68 @@
+
+import re
+import xml.etree.ElementTree
+from haircontrol.data import Equipment
+
+class Discovery:
+    def __init__(self, inspector):
+        self.inspector = inspector
+        self.equipments = {}
+        self.ip2mac = {}
+        self.mac2ip = {}
+
+    def add_equipment(self,e):
+        old = self.equipments.get(e.mgmtip)
+        if old:
+            print("Warn : %s replaced by %s"%(old,e))
+        self.equipments[e.mgmtip] = e
+
+    def index_mac_ip(self,mac, ip):
+        oldmac = self.ip2mac.get(ip)
+        if oldmac:
+            print("Warn : %s replaced by %s for %s"%(oldmac, mac, ip))
+        oldip = self.mac2ip.get(mac)
+        if oldip:
+            print("Warn : %s replaced by %s for %s"%(oldip, ip, mac))
+        self.ip2mac[ip] = mac
+        self.mac2ip[mac] = ip
+
+    def discover_hinting_from_lldp(self, lldpnode):
+        self.inspector.connect(lldpnode)
+        fd = self.inspector.command('lldp')
+        root = xml.etree.ElementTree.parse(fd).getroot()
+        for iface in root.iter('interface'):
+            ifname = iface.get('name')
+            chassis = iface.find('chassis')
+            e = Equipment()
+            e.name = chassis.find('name').text
+            e.mgmtip = chassis.find('mgmt-ip').text
+            self.add_equipment(e)
+            for port in iface.findall('port'):
+                #remote_ifname = port.find('id').text
+                e.add_seen_mac(ifname, 'lldp') #XXX
+        fd.close()
+        self.inspector.disconnect()
+
+    def discover_from_root(self, rootnode):
+        self.inspector.connect(rootnode)
+        fd = self.inspector.command('ip-neigh')
+        IPNEIGH = re.compile("^(?P<ip>[a-f0-9:.]+) dev (?P<ifname>.*) lladdr (?P<mac>[a-f0-9:]*)")
+        for line in fd:
+            matches = IPNEIGH.search(line)
+            if matches:
+                ip, ifname, mac = [ matches.group(k) for k in ['ip','ifname','mac'] ]
+                self.index_mac_ip(mac, ip)
+                rootnode.add_seen_mac(ifname, mac)
+        fd.close()
+        self.inspector.disconnect()
+
+# fe80::8300 dev eth1 lladdr 10:fe:ed:f1:e1:f3 router STALE
+# 172.16.11.46 dev eth1 lladdr 24:a4:3c:ee:89:ca STALE
+# 172.16.20.3 dev eth1 lladdr 00:27:22:0e:74:15 STALE
+# 172.16.20.210 dev eth1 lladdr c0:4a:00:fe:1f:87 REACHABLE
+# 172.16.21.69 dev eth1 lladdr e8:de:27:b5:f2:b1 DELAY
+# 172.16.20.216 dev eth1 lladdr c0:4a:00:fe:09:bd PERMANENT
+# 172.16.11.41 dev eth1 lladdr 04:18:d6:0e:37:d4 STALE
+# 172.16.11.104 dev eth1 lladdr 00:15:6d:8e:22:46 STALE
+# 172.16.10.8 dev eth1 lladdr 00:27:22:0e:67:f9 STALE
+#