1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
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()
|
