/backend/hosttracking/getmacs.py

https://github.com/BillTheBest/FooNMS · Python · 75 lines · 55 code · 7 blank · 13 comment · 8 complexity · c2dacd73359d60fc5e608bacee933c20 MD5 · raw file 

    

  1. #!/usr/bin/python
    2. 

  2. #http://pysnmp.sourceforge.net/docs/4.x/index.html#SYNCH-ONELINER-APPS
    3. 

  3. 

  4. #includes
    5. 

  5. from pysnmp.entity.rfc3413.oneliner import cmdgen
    6. 

  6. 

  7. # functions
    8. 

  8. def hexify( octets ):
    9. 

  9. 	return ":".join( [ '%x'%(ord(c)) for c in octets ]) #http://www.velocityreviews.com/forums/t320222-octet-string-conversion.html
    10. 

  10. 

  11. def grabmacs (host,community):	#gets a list of mac addresses, returns a dict of macid and mac address
    12. 

  12. 	errorIndication, errorStatus, errorIndex, varBinds = cmdgen.CommandGenerator().nextCmd(
    13. 

  13. 	cmdgen.CommunityData('my-agent', community, 0),
    14. 

  14. 	cmdgen.UdpTransportTarget((host, 161)),
    15. 

  15. 	(1,3,6,1,2,1,17,4,3,1,1))
    16. 

  16. 	table = {}
    17. 

  17. 	#http://www.cisco.com/en/US/tech/tk648/tk362/technologies_tech_note09186a00801c9199.shtml
    18. 

  18. 	for varBind in varBinds:
    19. 

  19. 		macid, mac = varBind[0]
    20. 

  20. 		table[str(macid.prettyPrint().split(".")[-6::])] = hexify(mac)
    21. 

  21. 	return(table)
    22. 

  22. 

  23. def grabbridgeports (host, community): #grabs a list of macid and port numbers. returns a dict of macid and portid
    24. 

  24.         errorIndication, errorStatus, errorIndex, varBinds = cmdgen.CommandGenerator().nextCmd(
    25. 

  25.         cmdgen.CommunityData('my-agent', community, 0),
    26. 

  26.         cmdgen.UdpTransportTarget((host, 161)),
    27. 

  27.         (1,3,6,1,2,1,17,4,3,1,2))
    28. 

  28. 	table = {}
    29. 

  29.         #http://www.cisco.com/en/US/tech/tk648/tk362/technologies_tech_note09186a00801c9199.shtml
    30. 

  30.         for varBind in varBinds:
    31. 

  31.                 macid, portid = varBind[0]
    32. 

  32.                 table[str(macid.prettyPrint().split(".")[-6::])] = portid
    33. 

  33. 	return(table)
    34. 

  34. 

  35. def grabportindex (host, community): #grabs a list of macid and port numbers. returns a dict of macid and portid
    36. 

  36.         errorIndication, errorStatus, errorIndex, varBinds = cmdgen.CommandGenerator().nextCmd(
    37. 

  37.         cmdgen.CommunityData('my-agent', community, 0),
    38. 

  38.         cmdgen.UdpTransportTarget((host, 161)),
    39. 

  39.         (1,3,6,1,2,1,17,1,4,1,2))
    40. 

  40. 	table = {}
    41. 

  41.         #http://www.cisco.com/en/US/tech/tk648/tk362/technologies_tech_note09186a00801c9199.shtml
    42. 

  42.         for varBind in varBinds:
    43. 

  43.                 index, portid = varBind[0]
    44. 

  44.                 table[index.prettyPrint().split(".")[-1]] = portid
    45. 

  45. 	return(table)
    46. 

  46. 

  47. def grabportnames (host,community): #grabs a list of portnames. returns a portid and a human name of the port as a dict
    48. 

  48. 	errorIndication, errorStatus, errorIndex, varBinds = cmdgen.CommandGenerator().nextCmd(
    49. 

  49.         cmdgen.CommunityData('my-agent', community, 0),
    50. 

  50.         cmdgen.UdpTransportTarget((host, 161)),
    51. 

  51.         (1,3,6,1,2,1,31,1,1,1,1))
    52. 

  52. 	table = {}
    53. 

  53.         #http://www.cisco.com/en/US/tech/tk648/tk362/technologies_tech_note09186a00801c9199.shtml
    54. 

  54.         for varBind in varBinds:
    55. 

  55.                 portid, portname = varBind[0]
    56. 

  56. 		table[portid.prettyPrint().split(".")[-1]] = portname
    57. 

  57. 	return(table)
    58. 

  58. 

  59. def getmacs(host,community):
    60. 

  60. 	macs = grabmacs(host,community)
    61. 

  61. 	bridgeports = grabbridgeports(host,community)
    62. 

  62. 	index = grabportindex(host,community)
    63. 

  63. 	portnames = grabportnames(host,community)
    64. 

  64. 	table = {}
    65. 

  65. 	for mac in macs:
    66. 

  66. 		if index == {}:
    67. 

  67. 			table[macs[mac]] = portnames[str(bridgeports[mac])]
    68. 

  68. 		else:		
    69. 

  69. 			table[macs[mac]] = portnames[str(index[str(bridgeports[mac])])]
    70. 

  70. 	return(table)		
    71. 

  71. #testing
    72. 

  72. #print grabmacs("172.27.2.1","public")
    73. 

  73. #print grabbridgeports("172.27.2.1","public")
    74. 

  74. #print grabportnames("172.27.2.1","public")
    75. 

  75. # print getmacs("172.27.2.43","public")
    76.