dot11sniffer.py

#!/bin/python2

import time

from threading import *

import signal
import sys
import netifaces

from sniffer_module import *
from dblib.dbSender import *

# globals
shouldQuit = None
otherThread = None
sender = Sender()
hostname = ""

# signal handler
def sigintHandler(signum, frame):
    print("Quitting uncleanly")
    sys.exit()

# search for interfaces until we found a monitor interface
def findIface():
    chosenIface = None
    while chosenIface is None:
        ifaces = netifaces.interfaces()
        for iface in ifaces:
            if iface[:3] == 'mon':
                chosenIface = iface

    return chosenIface

# gets hostname for computer
def loadHostname():
    global hostname

    hostnameFile = open("/etc/hostname", "r")
    hostname = hostnameFile.readline()
    hostnameFile.close()

    hostname = hostname[:-1]

# send the contents of the list
def sendData(otherThread):
    sender.clear()

    # lock for clearing
    otherThread.listLock.acquire()

    # add each of the packets to the list
    for mac in otherThread.devices:
        sender.add(mac, hostname, otherThread.devices[mac][0], otherThread.devices[mac][1])

    # clear out the device buffer
    otherThread.devices = {}

    otherThread.listLock.release()

    # and send off the data
    sender.send()

def main():
    # register signal handler and then sleep until we get a response
    signal.signal(signal.SIGINT, sigintHandler)

    # init the processing thread
    shouldQuit = False
    listLock = Lock()
    otherThread = ProcessingThread(listLock, shouldQuit)

    # find monitor inferaces
    print("Looking for monitor interfaces...")
    chosenIface = findIface()
    print("Found Interface: " + chosenIface)

    # get the hostname
    loadHostname()
    print("Hostname: " + hostname)

    # start the other thread to start sniffing
    otherThread.start()

    # sit and spin while the other thread does things
    while True:
        print("sending off data")
        sendData(otherThread)
        time.sleep(15)

# run main
if __name__ == "__main__":
    main()