Merge branch 'master' of github.com:krebscode/painload

1 files changed, 23 insertions, 4 deletions

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()