#!/usr/bin/python3 ''' # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - # - # Example of how to filter messages from the :8081 websocket - # - # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - You will need websocket-client: apt install python3-websocket ''' import sys if sys.version_info[0] != 3: print("This script requires Python 3") exit() import argparse import json import logging import pprint import traceback import time from time import sleep import websocket import re try: import thread except ImportError: import _thread as thread from LANforge import LFUtils cre={ "phy": re.compile(r'^(1\.\d+):\s+(\S+)\s+\(phy', re.I), "ifname": re.compile(r'(1\.\d+):\s+IFNAME=(\S+)\s+', re.I), "port": re.compile(r'Port (\S+)', re.I), "connected": re.compile(r'.*?CTRL-EVENT-CONNECTED - Connection to ([a-f0-9:]+) complete', re.I), "associated": re.compile(r'^.*?Associated with ([a-f0-9:]+)$', re.I), "auth": re.compile(r'.*: auth ([a-f0-9:]+) -> ([a-f0-9:]+) status: 0: Successful', re.I), "authenticated": re.compile(r'.*?Authenticated with ([a-f0-9:]+)', re.I), "associating": re.compile(r'.*?Trying to associate with ([a-f0-9:]+)', re.I), "authenticating": re.compile(r'.*?[>]SME: Trying to authenticate with ([a-f0-9:]+)', re.I), } ignore=[ ": scan finished", ": scan started", ": scan aborted: ", "CTRL-EVENT-SCAN-STARTED", "SCAN-STARTED", "SSID-TEMP-DISABLED", "CTRL-EVENT-DISCONNECTED", "CTRL-EVENT-REGDOM-CHANGE", "CTRL-EVENT-SUBNET-STATUS-UPDATE", "Reject scan trigger since one is already pending", "Failed to initiate AP scan", "new station", "del station", "ping", ": Key negotiation completed with ", "deleted-alert", ": deauth ", ": disconnected ", "regulatory domain change", ] rebank = { "ifname" : re.compile("IFNAME=(\S+)") } websock = None host = "localhost" base_url = None port = 8081 # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - def usage(): print("""Example: __file__ --host 192.168.1.101 --port 8081\n""") def main(): global websock # global host # global base_url # resource_id = 1 # typically you're using resource 1 in stand alone realm parser = argparse.ArgumentParser(description="test creating a station") parser.add_argument("-m", "--host", type=str, help="websocket host to connect to") parser.add_argument("-p", "--port", type=str, help="websoket port") host = "unset" base_url = "unset" try: args = parser.parse_args() if (args.host is None): host = "localhost" elif (type(args) is tuple) or (type(args) is list): host = args.host[0] else: host = args.host base_url = "ws://%s:%s" % (host, port) except Exception as e: print("Exception: "+e) logging.exception(e) usage() exit(2) # open websocket # print("Main: base_url: %s, host:%s, port:%s" % (base_url, host, port)) websock = start_websocket(base_url, websock) # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- def sock_filter(wsock, text): global ignore global interesting global rebank global cre debug = 0 station_name = None resource = None for test in ignore: if (test in text): if (debug): print (" ignoring ",text) return; try: message = json.loads(text) except Exception as ex: print ("Json Exception: ", repr(ex)) traceback.print_exc() try: # big generic filter for wifi-message or details keys try: if ("details" in message.keys()): for test in ignore: if (test in message["details"]): return; except KeyError: print("Message lacks key 'details'") try: if ("wifi-event" in message.keys()): for test in ignore: # print (" is ",test, " in ", message["wifi-event"]) if (test in message["wifi-event"]): return; except KeyError: print("Message lacks key 'wifi-event'") if (("time" in message.keys()) and ("timestamp" in message.keys())): return if ("name" in message.keys()): station_name = message["name"] if ("resource" in message.keys()): resource = "1.", message["resource"] if ("event_type" in message.keys()): match_result = cre["port"].match(message["details"]) if (match_result is not None): station_name = match_result.group(1) if (message["is_alert"]): print("alert: ", message["details"]) # LFUtils.debug_printer.pprint(message) return else: # LFUtils.debug_printer.pprint(message) if (" IP change from " in message["details"]): if (" to 0.0.0.0" in messsage["details"]): print("e: %s.%s lost IP address", [resource, station_name]) else: print("e: %s.%s gained IP address", [resource, station_name]) if ("Link DOWN" in message["details"]): return # duplicates alert print("event: ", message["details"]) return if ("wifi-event" in message.keys()): if ("CTRL-EVENT-CONNECTED" in message["wifi-event"]): # redunant return if (("CTRL-EVENT-CONNECTED - Connection to " in message["wifi-event"]) and ( " complete" in message["wifi-event"])): return; if ((": assoc " in message["wifi-event"]) and ("status: 0: Successful" in message["wifi-event"])): return if ((station_name is None) or (resource is None)): try: match_result = cre["phy"].match(message["wifi-event"]) if (match_result is not None): # LFUtils.debug_printer.pprint(match_result) # LFUtils.debug_printer.pprint(match_result.groups()) resource = match_result.group(1) station_name = match_result.group(2) else: match_result = cre["ifname"].match(message["wifi-event"]) # LFUtils.debug_printer.pprint(match_result) # LFUtils.debug_printer.pprint(match_result.groups()) if (match_result is not None): resource = match_result.group(1) station_name = match_result.group(2) else: print("Is there some other combination??? :", message["wifi-event"]) station_name = 'no-sta' resource_name = 'no-resource' print("bleh!") except Exception as ex2: print("No regex match:") print(repr(ex2)) traceback.print_exc() sleep(1) # print ("Determined station name: as %s.%s"%(resource, station_name)) if ((": auth ") and ("status: 0: Successful" in message["wifi-event"])): match_result = cre["auth"].match(message["wifi-event"]) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s auth with %s" % (resource, station_name, bssid)) return else: print("station %s.%s auth with ??" % (resource, station_name)) LFUtils.debug_printer.pprint(match_result) if ("Associated with " in message["wifi-event"]): match_result = cre["associated"].match(message["wifi-event"]) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s assocated with %s" % (resource, station_name, bssid)) return else: print("station %s.%s assocated with ??" % (resource, station_name)) LFUtils.debug_printer.pprint(match_result) if (" - Connection to " in message["wifi-event"]): match_result = cre["connected"].match(message["wifi-event"]) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s connected to %s" % (resource, station_name, bssid)) return else: print("station %s.%s connected to ??" % (resource, station_name)) LFUtils.debug_printer.pprint(match_result) if ("disconnected" in message["wifi-event"]): print("Station %s.%s down" % (resource, station_name)) return if ("Trying to associate with " in message["wifi-event"]): match_result = cre["associating"].match(message["wifi-event"]) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s associating with %s" % (resource, station_name, bssid)) return else: print("station %s.%s associating with ??" % (resource, station_name)) LFUtils.debug_printer.pprint(match_result) if ("Trying to authenticate" in message["wifi-event"]): match_result = cre["authenticating"].match(message["wifi-event"]) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s authenticating with %s" % (resource, station_name, bssid)) return else: print("station %s.%s authenticating with ??" % (resource, station_name)) LFUtils.debug_printer.pprint(match_result) if ("Authenticated" in message["wifi-event"]): match_result = cre["authenticed"].match(message["wifi-event"]) LFUtils.debug_printer.pprint(match_result) if (match_result and match_result.groups()): bssid = match_result.group(1) print("station %s.%s authenticated with %s" % (resource, station_name, bssid)) else: print("station %s.%s authenticated with ??" % (resource, station_name)) print("w: ", message["wifi-event"]) else: print("\nUnhandled: ") LFUtils.debug_printer.pprint(message) except KeyError as kerr: print("# ----- Bad Key: ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") print("input: ", text) print(repr(kerr)) traceback.print_exc() print("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") sleep(1) return except json.JSONDecodeError as derr: print("# ----- Decode err: ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") print("input: ", text) print(repr(derr)) traceback.print_exc() print("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") sleep(1) return except Exception as ex: print("# ----- Exception: ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") print(repr(ex)) print("input: ", text) LFUtils.debug_printer.pprint(message) traceback.print_exc() print("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----") sleep(1) return # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- def m_error(wsock, err): print("# ----- Error: ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----\n") LFUtils.debug_printer.pprint(err) print("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----\n") # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- def m_open(wsock): def run(*args): time.sleep(0.1) # ping = json.loads(); wsock.send('{"text":"ping"}') thread.start_new_thread(run, ()) print("Connected...") # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- def m_close(wsock): LFUtils.debug_printer.pprint(wsock) # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- def start_websocket(uri, websock): websock = websocket.WebSocketApp(uri, on_message=sock_filter, on_error=m_error, on_close=m_close) websock.on_open = m_open websock.run_forever() return websock # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- if __name__ == '__main__': main() #### #### ####