mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 18:58:01 +00:00
361 lines
13 KiB
Python
Executable File
361 lines
13 KiB
Python
Executable File
#!/usr/bin/python3
|
|
"""
|
|
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
|
# -
|
|
# Example of how to filter messages from the :8081 websocket -
|
|
# -
|
|
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
|
|
You will need websocket-client:
|
|
apt install python3-websocket
|
|
"""
|
|
import sys
|
|
import os
|
|
import importlib
|
|
import argparse
|
|
import json
|
|
import logging
|
|
import traceback
|
|
from time import sleep
|
|
import websocket
|
|
import re
|
|
|
|
try:
|
|
import thread
|
|
except ImportError:
|
|
import _thread as thread
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit()
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
LFUtils = importlib.import_module("py-json.LANforge.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)
|
|
# 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 message["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 Exception as ex:
|
|
traceback.print_exc()
|
|
raise ("Json Exception: ", repr(ex))
|
|
|
|
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):
|
|
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()
|
|
|
|
####
|
|
####
|
|
####
|