#!/usr/bin/env python

import subprocess
import signal
import getopt
import re
import logging
logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
from scapy.all import *

global interface

def usage():
	print " "
	print "Clients usage:"
	print " "
	print "-i <interface>    : The interface you want to sniff"
	print " "
	sys.exit()

	return

def ListClients(pkt):
	try:
		source_mac = pkt.getlayer(Ether).src
	except:
		return

	try:
		source_ip = pkt.getlayer(IP).src
	except:
		return

	if source_ip == "0.0.0.0":
		return

	try:
                routegw = subprocess.check_output("route -n | grep -i 'UG' | awk '{{print $2}}'", shell=True).strip()
                if source_ip == routegw:
                        return
        except:
                pass

	f1 = open('fclients2', 'r')
	f2 = open('fclients2', 'a')
	state = 0

	for line in f1:
		space = string.find(line, " ")
		mac = line[0:space]
		space = space + 1
		ip = line[space:len(line) - 1]

		if mac == source_mac and ip == source_ip:
			state = 1
			break
		elif mac == source_mac and ip != source_ip:
			state = 2
			break

	if state == 0:
		print "[  clients2][00000000] Added", source_mac, source_ip
		f2.write(source_mac + ' ' + source_ip + '\n')
	elif state == 2:
		f1.close()
		f2.close()

		f1 = open('fclients2', 'r')

		try:
			os.remove('fclients2.tmp')
		except:
			pass

		f2 = open('fclients2.tmp', 'w+')

		for line in f1:
			space = string.find(line, " ")
			mac = line[0:space]

			if mac == source_mac:
				f2.write(source_mac + ' ' + source_ip + '\n')
			else:
				f2.write(line)
	
	f1.close()
	f2.close()

	if state == 2:
		os.rename('fclients2.tmp', 'fclients2')
	
	return

def SIGNAL_handler(signum, frame):
	global interface

	print "[  clients2][00000000] Signal caught."
	print "[  clients2][00000000] Sniffing on", interface, "off"
	print "[  clients2][00000000]", interface, "promisc mode off"
	subprocess.call(['ifconfig', interface, '-promisc'])

	sys.exit()

	return

def main():
	global interface

	try:
		opts, args = getopt.getopt(sys.argv[1:],"i:")
	except:
		usage()
		sys.exit(1)

	interface = None

	for o, a in opts:
		if o == '-i':
			interface = a

	if args or not interface:
		usage()
		sys.exit(1)

	try:
		os.makedirs('/opt/td-config/run/besside/')
	except:
		pass
	
	os.chdir('/opt/td-config/run/besside/')

	f = open('fclients2', 'w')
	f.close()

	signal.signal(signal.SIGINT, SIGNAL_handler)
	signal.signal(signal.SIGTERM, SIGNAL_handler)

	print "[  clients2][00000000]", interface, "promisc mode on"
	subprocess.call(["ifconfig", interface, 'promisc'])

	print "[  clients2][00000000] Sniffing on", interface, "on"

	try:
	    sniff(iface=interface, filter="ip", store=0, prn=ListClients)
	except:
	    return

	return

if __name__ == "__main__":
	main()