1 files changed, 87 insertions, 0 deletions
diff --git a/snmp_users.py b/snmp_users.py
new file mode 100755
index 00000000..df78cc8d
--- /dev/null
+++ b/snmp_users.py
@@ -0,0 +1,87 @@
+#!/usr/bin/python2
+import subprocess,re
+import logging
+from arping import arpingy,pingy
+log = logging.getLogger('snmp_users')
+
+class snmp_users:
+  mac_list = {}
+
+  def __init__(self,server='10.42.0.1',community='shammunity'):
+    self.server = server
+    self.community = community
+
+  def call_external(self):
+    """ creates a list of lines produced by snmpwal
+    """
+    out = subprocess.Popen(
+        ['snmpwalk',
+          '-v2c',
+          '-c',self.community,
+          self.server,
+          '1.3.6.1.2.1.3.1.1.2'],
+        stdout=subprocess.PIPE).communicate()[0]
+    return out.split('\n')
+
+  def parse_output(self,output):
+    """ parses output lines produced by snmpwalk"""
+    for i in output:
+      if i == '':
+        continue
+
+      ip,mac = re.sub(r'.*\.(\d+\.\d+\.\d+\.\d+) = Hex-STRING: ([ 0-9A-F]*) ',
+          r'\1 : \2',i).split(' : ')
+
+      verified = False
+      if self.verify_data(mac,ip,arpingy(ip)):
+        log.info("verfied by arping %s => %s" % (mac,ip))
+        verified = True
+      
+      #elif self.verify_data(mac,ip,pingy(ip) ): builds up onto arp'ing...
+      #  log.info("verfied by fallback ping %s => %s" % (mac,ip))
+      #  verfied = True
+
+      if verified:
+        if mac in self.mac_list.keys():
+          log.info( "%s already assigned for ip %s (new IP %s) " %
+              ( mac ,self.mac_list[mac], ip))
+          self.mac_list[mac]['counter'] +=1
+          continue
+        else:
+          log.debug("%s => %s" % (mac,ip))
+          self.mac_list[mac] = { 'ip' : ip, 'counter' : 1}
+      else:
+        log.warning("Verification failed %s => %s" % (mac,ip))
+
+  def verify_data(self,mac,ip,ret):
+    """ Verifies ip and mac via ARP Scan """ 
+    mac = ':'.join(mac.split())
+    log.debug(ret)
+
+    if type(ret) == type(True):
+      return True
+    elif ret != []:
+      ret = ret[0]
+      ret_mac = ret[1].upper()
+      if ret_mac == mac:
+        return True
+      else:
+        log.info('return mac not equal to expected, %s != %s' % (ret_mac,mac))
+    else:
+      return False
+
+  def collect(self):
+    output = self.call_external()
+    self.parse_output(output)
+    self.print_results()
+  def print_results(self):
+    print '\n'.join([ mac + " => " + ip['ip'] + " ( %d active leases )" %
+      ip['counter'] for mac,ip in self.mac_list.items() ])
+    print '%d *unique* nodes in network' % len(self.mac_list)
+
+
+if __name__ == "__main__":
+  logging.basicConfig(level=logging.INFO)
+  a = snmp_users()
+  a.collect()
+