summaryrefslogtreecommitdiffstats
path: root/god/census/arping_users.py
blob: eb47f308d50a8db402ff44ee4d645e1c58813b3b (plain)
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
88
89
90
91
92
#!/usr/bin/python
import subprocess,re,logging,sys
import json
from arping import arpingy
from multiprocessing import Pool
logging.basicConfig(level=logging.WARNING)
log = logging.getLogger("main")
DEV='eth1'
MAC_NAMES='mac_names.lst'
data = []
my_addr = False
my_names = {}
ret = {}
quiet=False

names = {}
if len(sys.argv) > 1 and sys.argv[1] == 'q':
  quiet=True
def get_own_addr():
  data = subprocess.Popen(['/sbin/ifconfig',DEV], 
      stdout=subprocess.PIPE).communicate()[0].replace('\n','')
  return re.sub(r'.*HWaddr ([0-9A-Fa-f:]*).*inet addr:([0-9.]*).*' ,
      r'\1 \2',data).split()

def load_names(mac_file):
  f = open(mac_file)
  for l in f:
    mac,name = l.split(' ',1)
    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)

def arping_helper(dic):
  log.debug("trying arpingy(%s)" %dic)
  return arpingy(**dic)

def main():
  init()
  print_config()

  for first in range(1,4):
    for second in range(256):
      data.append({'iprange':'10.42.'+str(first)+'.'+str(second),'iface':DEV})
  try:
    log.info("creating new Pool")
    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()