wiliot-certificate 4.4.3__py3-none-any.whl → 4.5.0a1__py3-none-any.whl

gw_certificate/interface/ble_sniffer.py

@@ -1,189 +0,0 @@
-import ast
-import logging
-import pandas as pd
-import time
-import datetime
-import threading
-import binascii
-
-from gw_certificate.interface.uart_if import UARTInterface
-from gw_certificate.common.debug import debug_print
-from gw_certificate.interface.if_defines import *
-
-class SnifferPkt():
-    def __init__(self, raw_output, time_received, rx_channel):
-        self.adva = raw_output[:12]
-        self.packet = raw_output[12:74]
-        try:
-            self.rssi = int.from_bytes(binascii.unhexlify(raw_output[74:]), 'big') * -1
-        except ValueError:
-            self.rssi = 99
-        self.time_received = time_received
-        self.rx_channel = rx_channel
-
-    def __repr__(self):
-        return f'CH{self.rx_channel}|{self.adva}{self.packet} RSSI:{self.rssi} {self.time_received}'
-
-    def to_dict(self):
-        return {'adva': self.adva, 'packet': self.packet, 'rssi': self.rssi, 'time_received': self.time_received, 'rx_channel': self.rx_channel}
-
-class SnifferPkts():
-    def __init__(self, pkts=[]):
-        self.pkts = pkts
-
-    def __add__(self, other):
-        return SnifferPkts(self.pkts + other.pkts)
-
-    def __len__(self):
-        return len(self.pkts)
-
-    def __repr__(self):
-        return self.pkts
-
-    def process_pkt(self, raw_output, time_received, rx_channel, print_pkt=False, adva_filter=None):
-        pkt = SnifferPkt(raw_output, time_received, rx_channel)
-        if adva_filter:
-            if pkt.adva != adva_filter:
-                return None
-        self.pkts.append(pkt)
-        if print_pkt:
-            print(pkt)
-        return pkt
-
-    def filter_pkts(self, raw_packet=None, adva=None, time_range:tuple=None):
-        result = []
-        for pkt in self.pkts:
-            if (raw_packet is not None) and (pkt.packet == raw_packet) or \
-                (adva is not None) and (pkt.adva == adva) or \
-                (time_range is not None) and (time_range[0] < pkt.time_received < time_range[1]):
-                result.append(pkt)
-        return SnifferPkts(result)
-
-    def flush_pkts(self):
-        self.pkts = []
-
-    def to_list(self):
-        return [p.to_dict() for p in self.pkts]
-
-    def to_pandas(self):
-        return pd.DataFrame().from_dict(self.to_list())
-
-def cur_time():
-    return datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-
-
-class BLESniffer():
-    def __init__(self, uart:UARTInterface, print_pkt=False, logger_filepath=None, adva_filter=None):
-        self.uart = uart
-        self.listener_thread = None
-        self.sniffer_pkts = SnifferPkts()
-        self.listening = False
-        self.listener_lock = threading.Lock()
-        self.print = print_pkt
-        self.rx_channel = 0
-        self.adva_filter = None
-        if adva_filter is not None:
-            adva_filter = adva_filter.upper()
-            assert isinstance(adva_filter, str)\
-                and len(adva_filter) == 12\
-                and all(c in '0123456789ABCDEF' for c in adva_filter), "Input must be a 12-character hexadecimal string"
-            self.adva_filter = adva_filter
-        # Configure Logger
-        logger = logging.getLogger('ble_sniffer')
-        logger.setLevel(logging.DEBUG)
-        if logger_filepath is not None:
-            # create file handler
-            fh = logging.FileHandler(logger_filepath)
-            fh.setLevel(logging.DEBUG)
-            formatter = logging.Formatter('%(message)s')
-            fh.setFormatter(formatter)
-            logger.addHandler(fh)
-            logger.propagate = False # Do not send logs to 'root' logger
-            debug_print(f'BLE Sniffer Logger initialized at {logger_filepath}')
-        self.logger = logger
-
-    def packet_listener(self):
-        while self.listening:
-            line = self.uart.read_line()
-            if line is not None and len(line) == 76:
-                with self.listener_lock:
-                    pkt = self.sniffer_pkts.process_pkt(line, datetime.datetime.now(), self.rx_channel, self.print, self.adva_filter)
-                    if pkt:
-                        self.logger.info(pkt)
-
-    # Change sniffing modes
-    def start_sniffer(self, rx_channel):
-        self.logger.debug(f'{cur_time()} | Starting Sniffer on CH{rx_channel}')
-        self.uart.set_sniffer(rx_channel)
-        self.rx_channel = rx_channel
-        self.listener_thread = threading.Thread(target=self.packet_listener)
-        self.listening = True
-        self.listener_thread.start()
-
-    def stop_sniffer(self):
-        self.logger.debug(f'{cur_time()} | Stopping Sniffer')
-        self.flush_pkts()
-        self.listening = False
-        if self.listener_thread is not None:
-            self.listener_thread.join()
-            self.listener_thread = None
-            self.uart.cancel_sniffer()
-        self.rx_channel = 0
-
-    def reset_sniffer(self, rx_channel):
-        self.logger.debug(f'{cur_time()} | Reseting Sniffer')
-        self.stop_sniffer()
-        self.start_sniffer(rx_channel)
-        self.flush_pkts()
-
-    # Data Handling
-    def get_all_pkts(self):
-        return self.sniffer_pkts
-
-    def get_filtered_packets(self, raw_packet=None, adva=None, time_range:tuple=None):
-        return self.sniffer_pkts.filter_pkts(raw_packet, adva, time_range)
-
-    def flush_pkts(self):
-        self.logger.debug(f'{cur_time()} | Flushing packets')
-        self.sniffer_pkts.flush_pkts()
-
-    def to_pandas(self):
-        return self.sniffer_pkts.to_pandas()
-
-    # Packet Counters
-    def get_pkts_cntrs(self, channel, lookout_time=DEFAULT_LOOKOUT_TIME):
-        self.logger.debug(f'{cur_time()} | Getting pkt counters for CH{channel}')
-        if self.rx_channel != channel:
-            self.uart.set_rx(channel)
-            self.rx_channel = channel
-        
-        lookout_time = lookout_time
-        pkt_cntrs = None
-        self.uart.flush()
-        self.uart.write_ble_command(GET_LOGGER_COUNTERS)
-        start_time = time.time()
-        current_time = start_time
-        while current_time - start_time < lookout_time:
-            line = self.uart.read_line()
-            if line is not None and "'bad_crc'" in line:
-                start_of_cntr_index = line.find('{')
-                pkt_cntrs = line[start_of_cntr_index:]
-                debug_print(f"pkt_cntrs: {pkt_cntrs}")
-                return ast.literal_eval(pkt_cntrs)
-            current_time = time.time()
-        debug_print(f"No counter received within the time limit of {lookout_time} seconds")
-        return pkt_cntrs
-
-class BLESnifferContext():
-    def __init__(self, ble_sniffer:BLESniffer, rx_channel):
-        self.ble_sniffer = ble_sniffer
-        self.rx_channel = rx_channel
-
-    def __enter__(self):
-        self.ble_sniffer.flush_pkts()
-        self.ble_sniffer.start_sniffer(self.rx_channel)
-        return self.ble_sniffer
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        if self.ble_sniffer:
-            self.ble_sniffer.stop_sniffer()

gw_certificate/interface/flash_fw.py

@@ -1,90 +0,0 @@
-import sys
-import serial
-import time
-import argparse
-import subprocess
-import serial.tools.list_ports
-import datetime
-import tkinter as tk
-from tkinter import filedialog
-SEP = "#"*100
-
-NRFUTIL_MAP = {"linux": "nrfutil-linux", "linux2": "nrfutil-linux", "darwin": "nrfutil-mac", "win32": ".\\nrfutil.exe"}
-
-def read_from_ble(ble_ser):
-    ble_ser_bytes = ble_ser.readline()
-    input = ble_ser_bytes.decode("utf-8", "ignore").strip()
-    if input:
-        print(input)
-    return input
-
-class Command:
-    def __init__(self, cmd, expected, delay=0):
-        self.cmd = cmd
-        self.cmd_exec = b'\r\n!'+bytes(cmd, encoding='utf-8')+b'\r\n'
-        self.expected = expected
-        self.delay = delay
-    def exec_cmd(self, ble_ser):
-        print("==>> !{}".format(self.cmd))
-        ble_ser.write(self.cmd_exec)
-        start_time = datetime.datetime.now()
-        while (datetime.datetime.now() - start_time).seconds < 2:
-            if self.expected in read_from_ble(ble_ser):
-                print("success!\n{}".format(SEP))
-                return 0
-        print("failure!\n{}".format(SEP))
-        return 1
-
-GW_COMMANDS = [
-    Command(cmd="version", expected="WILIOT_GW_BLE_CHIP_SW_VER"),
-    Command(cmd="move_to_bootloader", expected='')
-]
-
-def run_cmd(cmd):
-    print("Running: " + cmd)
-    p = subprocess.Popen(cmd, shell=True)
-    p.wait()
-    if p.returncode:
-        print(f"\nFailed running : {cmd}\n")
-        sys.exit(-1)
-    else:
-        print(f"\nSuccess running : {cmd}\n")
-
-def main():
-
-    ports = serial.tools.list_ports.comports()
-    print(SEP + "\nAvailable ports:")
-    for port, desc, hwid in sorted(ports):
-            print("{}: {} [{}]".format(port, desc, hwid))
-    print(SEP + "\n")
-
-    parser = argparse.ArgumentParser(description='Used to load gw image')
-    parser.add_argument('--p', type=str, help='COM for the ble - meaning loading gw zip file from UART')
-    args = parser.parse_args()
-
-
-    if args.p:
-        ble_ser = serial.Serial(port=args.p, baudrate=921600, timeout=0.1)
-        ble_ser.flushInput()
-
-        failures = 0
-        
-        for cmd in GW_COMMANDS:
-            failures += cmd.exec_cmd(ble_ser)
-            time.sleep(cmd.delay)
-        if failures == 0:
-            ble_ser.close()
-            time.sleep(2)
-            print("Please pick a file! (<X.Y.Z>_app.zip)")
-            root = tk.Tk()
-            root.withdraw()
-            file_path = filedialog.askopenfilename()
-            print(file_path)
-            cmd = f'{NRFUTIL_MAP[sys.platform]} dfu serial --package "{file_path}" -p {args.p} -fc 0 -b 115200 -t 10'
-            run_cmd(cmd)
-        else:
-            print("ERROR: failed to run UART commands!")
-
-
-if __name__ == "__main__":
-    main()

gw_certificate/interface/if_defines.py

@@ -1,36 +0,0 @@
-RX_CHANNELS = [37, 38, 39]
-RX_CHANNELS_SNIFFING_KIT = list(range(40))
-SET_SNIFFER = '!set_logger_mode 2'
-CANCEL_SNIFFER = '!set_logger_mode 0'
-GET_LOGGER_COUNTERS = '!get_logger_counters'
-GATEWAY_APP = '!gateway_app'
-STOP_ADVERTISING = '!stop_advertising'
-CANCEL = '!cancel'
-RESET_GW = '!reset'
-VERSION = '!version'
-SEND_ALL_ADV_CHANNELS = 0
-SERIAL_TIMEOUT = 0.1
-SEP = "#"*100
-LENGTH = '1E'
-GAP_TYPE = '16'
-DEFAULT_ADVA = 'FFFFFFFFFFFF'
-DEFAULT_DUPLICATES = 3
-GW_APP_VERSION_HEADER = 'WILIOT_GW_BLE_CHIP_SW_VER'
-GW_MAC_ADDRESS_HEADER = 'WILIOT_GW_BLE_CHIP_MAC_ADDRESS'
-LOCATION = 'location'
-DEFAULT_OUTPUT_POWER = 8
-DEFAULT_DELAY = 20
-BRIDGES = [i for i in range(3)]
-MAX_NFPKT = 65535
-MAX_RSSI = 255
-ADVA_LENGTH = 12
-GAP_LENGTH = 4
-SERVICE_UUID_LENGTH = 4
-GROUP_ID_LENGTH = 6
-UPLINK_DUPLICATIONS = [i for i in range(3, 7)]
-UPLINK_TIME_DELAYS = [20, 130, 255]
-UNIFIED_DUPLICATIONS = [i for i in range(3, 7)]
-UNIFIED_TIME_DELAYS = [20, 130, 255]
-DEFAULT_LOOKOUT_TIME = 2
-
-ADVA_PAYLOAD = 'adva_payload'