FreeDATA/sock.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Dec 25 21:25:14 2020

@author: DJ2LS
"""

import socketserver
import threading
import logging
import json
import asyncio

import static
import data_handler
import helpers


class CMDTCPRequestHandler(socketserver.BaseRequestHandler):

    def handle(self):

        encoding = 'utf-8'
        #data = str(self.request.recv(1024), 'utf-8')

        data = bytes()
        while True:
            chunk = self.request.recv(8192)  # .strip()
            data += chunk
            if chunk.endswith(b'\n'):
                break
        data = data[:-1]  # remove b'\n'
        data = str(data, 'utf-8')
        
        # convert data to json object
        received_json = json.loads(data)

        # GET COMMANDS
        # "command" : "..."
        
        # SET COMMANDS
        # "command" : "..."
        # "parameter" : " ..."
        
        # DATA COMMANDS
        # "command" : "..."
        # "type" : "..."
        # "dxcallsign" : "..."
        # "data" : "..."
        
        

        # SOCKETTEST ---------------------------------------------------
        #if data == 'SOCKETTEST':
        if received_json["command"] == "SOCKETTEST":
            #cur_thread = threading.current_thread()
            response = bytes("WELL DONE! YOU ARE ABLE TO COMMUNICATE WITH THE TNC", encoding)
            self.request.sendall(response)

        # CQ CQ CQ -----------------------------------------------------
        #if data == 'CQCQCQ':
        if received_json["command"] == "CQCQCQ":
            asyncio.run(data_handler.transmit_cq())


        # PING ----------------------------------------------------------
        #if data.startswith('PING:'):
        if received_json["command"] == "PING":
            # send ping frame and wait for ACK
            dxcallsign = received_json["dxcallsign"]
            asyncio.run(data_handler.transmit_ping(dxcallsign))

        # ARQ CONNECT TO CALLSIGN ----------------------------------------
        #if data.startswith('ARQ:CONNECT:'):
        if received_json["command"] == "ARQ:CONNECT":
        
            dxcallsign = received_json["dxcallsign"]
            static.DXCALLSIGN = bytes(dxcallsign, 'utf-8')
            static.DXCALLSIGN_CRC8 = helpers.get_crc_8(static.DXCALLSIGN)

            if static.ARQ_STATE == 'CONNECTED':
                # here we could disconnect
                pass

            if static.TNC_STATE == 'IDLE':

                asyncio.run(data_handler.arq_connect())

        # ARQ DISCONNECT FROM CALLSIGN ----------------------------------------
        if received_json["command"] == "ARQ:DISCONNECT":
            asyncio.run(data_handler.arq_disconnect())


        if received_json["command"] == "ARQ:OPEN_DATA_CHANNEL": # and static.ARQ_STATE == 'CONNECTED':
            static.ARQ_READY_FOR_DATA = False
            static.TNC_STATE = 'BUSY'
            
            dxcallsign = received_json["dxcallsign"]
            static.DXCALLSIGN = bytes(dxcallsign, 'utf-8')
            static.DXCALLSIGN_CRC8 = helpers.get_crc_8(static.DXCALLSIGN)
            
            asyncio.run(data_handler.arq_open_data_channel())
            

        if received_json["command"] == "ARQ:DATA" and static.ARQ_READY_FOR_DATA == True: # and static.ARQ_STATE == 'CONNECTED' :
            static.TNC_STATE = 'BUSY'

            data_out = bytes(received_json["data"], 'utf-8')

            ARQ_DATA_THREAD = threading.Thread(target=data_handler.arq_transmit, args=[data_out], name="ARQ_DATA")
            ARQ_DATA_THREAD.start()
            # asyncio.run(data_handler.arq_transmit(data_out))

        # SETTINGS AND STATUS ---------------------------------------------
        if received_json["command"] == 'SET:MYCALLSIGN':
            callsign = received_json["parameter"]

            if bytes(callsign, encoding) == b'':
                self.request.sendall(b'INVALID CALLSIGN')
            else:
                static.MYCALLSIGN = bytes(callsign, encoding)
                static.MYCALLSIGN_CRC8 = helpers.get_crc_8(static.MYCALLSIGN)
                logging.info("CMD | MYCALLSIGN: " + str(static.MYCALLSIGN))
  
            
        if received_json["command"] == 'GET:STATION_INFO':
            output = {
                "MY_CALLSIGN": str(static.MYCALLSIGN, encoding),
                "DX_CALLSIGN": str(static.DXCALLSIGN, encoding)  
            }
            
            jsondata = json.dumps(output)
            self.request.sendall(bytes(jsondata, encoding))

        if received_json["command"] == 'GET:TNC_STATE':
            output = {
                "PTT_STATE": str(static.PTT_STATE),
                "CHANNEL_STATE": str(static.CHANNEL_STATE),
                "TNC_STATE": str(static.TNC_STATE),
                "ARQ_STATE": str(static.ARQ_STATE),
                "AUDIO_RMS": str(static.AUDIO_RMS),
                "BER": str(static.BER),
                "SNR": str(static.SNR)
            }
            
            jsondata = json.dumps(output)
            self.request.sendall(bytes(jsondata, encoding))

        if received_json["command"] == 'GET:DATA_STATE':
            output = {
                "RX_BUFFER_LENGTH": str(len(static.RX_BUFFER)),
                "TX_N_MAX_RETRIES": str(static.TX_N_MAX_RETRIES),
                "ARQ_TX_N_FRAMES_PER_BURST": str(static.ARQ_TX_N_FRAMES_PER_BURST),
                "ARQ_TX_N_BURSTS": str(static.ARQ_TX_N_BURSTS),
                "ARQ_TX_N_CURRENT_ARQ_FRAME": str(int.from_bytes(bytes(static.ARQ_TX_N_CURRENT_ARQ_FRAME), "big")),
                "ARQ_TX_N_TOTAL_ARQ_FRAMES": str(int.from_bytes(bytes(static.ARQ_TX_N_TOTAL_ARQ_FRAMES), "big")),
                "ARQ_RX_FRAME_N_BURSTS": str(static.ARQ_RX_FRAME_N_BURSTS),
                "ARQ_RX_N_CURRENT_ARQ_FRAME": str(static.ARQ_RX_N_CURRENT_ARQ_FRAME),
                "ARQ_N_ARQ_FRAMES_PER_DATA_FRAME": str(static.ARQ_N_ARQ_FRAMES_PER_DATA_FRAME)
            }
            
            jsondata = json.dumps(output)
            self.request.sendall(bytes(jsondata, encoding))

        if received_json["command"] == 'GET:HEARD_STATIONS':
            output = []
            for i in range(0, len(static.HEARD_STATIONS)):
                output.append({"CALLSIGN": str(static.HEARD_STATIONS[i][0], 'utf-8'), "TIMESTAMP": static.HEARD_STATIONS[i][1], "DATATYPE": static.HEARD_STATIONS[i][2]})

            jsondata = json.dumps(output)
            self.request.sendall(bytes(jsondata, encoding))


        if received_json["command"] == 'GET:RX_BUFFER':
            data = data.split('GET:RX_BUFFER:')
            bufferposition = int(data[1]) - 1
            if bufferposition == -1:
                if len(static.RX_BUFFER) > 0:
                    self.request.sendall(static.RX_BUFFER[-1])

            if bufferposition <= len(static.RX_BUFFER) > 0:
                self.request.sendall(bytes(static.RX_BUFFER[bufferposition]))

        if received_json["command"] == 'DEL:RX_BUFFER':
            static.RX_BUFFER = []

def start_cmd_socket():

    try:
        logging.info("SRV | STARTING TCP/IP SOCKET FOR CMD ON PORT: " + str(static.PORT))
        socketserver.TCPServer.allow_reuse_address = True  # https://stackoverflow.com/a/16641793
        cmdserver = socketserver.TCPServer((static.HOST, static.PORT), CMDTCPRequestHandler)
        cmdserver.serve_forever()

    finally:
        cmdserver.server_close()