wifi-probe.py

#!/usr/bin/env python

from scapy.all import *
from datetime import datetime
import os
import humanize
import signal
import sys

# config
DISPLAY_PROBES_MAX = 16

# colors, control sequences
TERM_RED           = '\033[91m'
TERM_GREEN         = '\033[92m'
TERM_YELLOW        = '\033[93m'
TERM_BLUE          = '\033[94m'
TERM_MAGENTA       = '\033[95m'
TERM_BOLD          = '\033[1m'
TERM_RESET         = '\033[0m'
TERM_POS_ZERO      = '\033[0;0H'

# probe database
probes = [ ]


def signal_handler(signal, frame):
    print TERM_RESET
    sys.exit(0)


def get_termsize():
    y, x = os.popen('stty size', 'r').read().split()
    return int(x), int(y)


def update_probes(probe):
    probe['last_seen'] = datetime.now()

    # if ssid in database, update
    updated = False
    for i in range(len(probes)):
        if probes[i]['ssid'] == probe['ssid']:
            probes[i] = probe
            updated = True

    # if ssid not in database, insert
    if not updated:
        if len(probes) < DISPLAY_PROBES_MAX:
            probes.append(probe)
        else:
            while len(probes) > DISPLAY_PROBES_MAX:
                oldest = 0
                for i in range(len(probes)):
                    if probes[i]['last_seen'] < probes[oldest]['last_seen']:
                        oldest = i
                del probes[oldest]
            oldest = 0
            for i in range(len(probes)):
                if probes[i]['last_seen'] < probes[oldest]['last_seen']:
                    oldest = i
            probes[oldest] = probe


def print_probes():
    termx, termy = get_termsize()
    col1_width = termx // 2
    col2_width = termx - col1_width - 1
    DISPLAY_PROBES_MAX = termy - 2
    print TERM_RESET + TERM_POS_ZERO + str(datetime.now()).center(termx)
    for i in range(len(probes)):
        probe = probes[i]
        # ssid
        out = ''
        out += TERM_BOLD
        age = (datetime.now() - probe['last_seen']).total_seconds()
        if age < 60:
            out += TERM_RED
        elif age < 180:
            out += TERM_YELLOW
        else:
            out += TERM_GREEN
        out += probe['ssid'].rjust(col1_width)
        # time
        out += ' '
        out += TERM_RESET + TERM_BLUE
        out += humanize.naturaltime(probe['last_seen']).ljust(col2_width)
        print out


def packet_handler(pkt):
    if pkt.haslayer(Dot11):
        if pkt.type == 0 and pkt.subtype == 4:
            ssid = pkt.getlayer(Dot11ProbeReq).info
            if len(ssid):
                try:
                    extra = pkt.notdecoded
                    signal_strength = -(256 - ord(extra[-4:-3]))
                except:
                    signal_strength = 0
                probe = { }
                probe['ssid'] = ssid
                probe['signal'] = signal_strength
                probe['source'] = pkt.addr2
                probe['target'] = pkt.addr3
                update_probes(probe)
                print_probes()


if __name__=="__main__":
    signal.signal(signal.SIGINT, signal_handler)
    os.system('clear')
    while(True):
        print_probes()
        sniff(iface=sys.argv[1], prn=packet_handler, count=75)