From 207a4ec405d3229a9d2e200b7803c3068d513265 Mon Sep 17 00:00:00 2001
From: krebs <krebs@fuerkrebs>
Date: Fri, 4 Nov 2011 12:44:25 -0400
Subject: //census: fix arping scripts, print data in json

---
 census/arping.py       |  3 +--
 census/arping_users.py | 27 +++++++++++++++++++++++----
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/census/arping.py b/census/arping.py
index 3cffa6be..ed257441 100755
--- a/census/arping.py
+++ b/census/arping.py
@@ -11,11 +11,10 @@ try:
   from scapy.all import * #might throws "no such module"
 
   def arpingy(iprange="10.42.1.0/24",iface='eth0'):
-    log.debug("pinging "+ str(iprange))
     """Arping function takes IP Address or Network, returns nested mac/ip list"""
     try:
       conf.verb=0
-      ans,unans=arping(iprange,iface=iface,timeout=0.4,retry=1)
+      ans,unans=arping(iprange,iface=iface,timeout=3,retry=1)
 
       collection = []
       for snd, rcv in ans:
diff --git a/census/arping_users.py b/census/arping_users.py
index f4f356bf..eb47f308 100755
--- a/census/arping_users.py
+++ b/census/arping_users.py
@@ -1,11 +1,11 @@
 #!/usr/bin/python
 import subprocess,re,logging,sys
-
+import json
 from arping import arpingy
 from multiprocessing import Pool
-logging.basicConfig(level=logging.DEBUG)
+logging.basicConfig(level=logging.WARNING)
 log = logging.getLogger("main")
-DEV='eth0'
+DEV='eth1'
 MAC_NAMES='mac_names.lst'
 data = []
 my_addr = False
@@ -29,12 +29,14 @@ def load_names(mac_file):
     names[mac] = name.replace('\n','')
   f.close()
   return names
+
 def print_config():
   log.info("My Addr : %s" %str(my_addr))
   log.info("MAC Names file: %s " %MAC_NAMES)
   log.debug("Loaded names : ")
   for mac,name in my_names.iteritems():
     log.debug("%s => %s" %(mac,name))
+
 def init():
   my_addr = get_own_addr()
   my_names = load_names(MAC_NAMES)
@@ -52,22 +54,39 @@ def main():
       data.append({'iprange':'10.42.'+str(first)+'.'+str(second),'iface':DEV})
   try:
     log.info("creating new Pool")
-    p = Pool(30)
+    p = Pool(35)
     ret = filter(lambda x:x , p.map(arping_helper, data))
     log.info("killing it")
     p.terminate()
 
   except Exception as e:
     print 'you fail '+str(e)
+    sys.exit(1)
   myip,mymac = get_own_addr()
   ret.append([mymac,myip]) 
 
+  print_json(ret)
+  #print_names(ret)
+
+def print_names(ret):
   for p in ret:
     if not quiet:
       print p[0] + " => " + p[1]
     if p[1] in names:
       print names[p[1]]+ " is online"
 
+def print_json(ret):
+  from time import time
+  output = {}
+  output["timestamp"] = time()
+  for i in ret:
+    mac = i[0]
+    ip  = i[1]
+    if i[0] not in output:
+      output[mac] = []
+    output[mac].append(ip)
+  print json.dumps(output)
+
 if __name__ == "__main__":
   log.debug("starting arping_users")
   main()
-- 
cgit v1.2.3