FreeDATA/sock.py

127 lines
4.4 KiB
Python
Raw Normal View History

2021-02-16 13:23:57 +00:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Dec 25 21:25:14 2020
@author: DJ2LS
"""
import socketserver
import threading
import logging
2021-02-16 18:39:08 +00:00
import time
2021-02-16 13:23:57 +00:00
import static
import arq
2021-02-16 19:49:02 +00:00
import helpers
2021-02-16 13:23:57 +00:00
class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
2021-02-16 13:36:01 +00:00
2021-02-16 18:39:08 +00:00
def handle(self):
2021-02-16 19:49:02 +00:00
2021-02-18 14:11:37 +00:00
encoding = 'utf-8'
2021-02-16 18:39:08 +00:00
data = str(self.request.recv(1024), 'utf-8')
2021-02-18 14:11:37 +00:00
# SOCKETTEST
2021-02-16 18:39:08 +00:00
if data == 'SOCKETTEST':
2021-02-16 21:41:06 +00:00
cur_thread = threading.current_thread()
2021-02-18 14:11:37 +00:00
response = bytes("WELL DONE! YOU ARE ABLE TO COMMUNICATE WITH THE TNC", encoding)
2021-02-16 18:39:08 +00:00
self.request.sendall(response)
# TRANSMIT ARQ MESSAGE
2021-02-19 08:50:04 +00:00
if data.startswith('ARQ:') and static.TNC_STATE == b'IDLE':
2021-02-16 18:39:08 +00:00
logging.info("CMD | NEW ARQ DATA")
2021-02-19 08:50:04 +00:00
static.TNC_STATE = b'BUSY'
2021-02-16 21:41:06 +00:00
arqdata = data.split('ARQ:')
data_out = bytes(arqdata[1], 'utf-8')
2021-02-16 13:36:01 +00:00
TRANSMIT_ARQ = threading.Thread(target=arq.transmit, args=[data_out], name="TRANSMIT_ARQ")
TRANSMIT_ARQ.start()
2021-02-16 19:49:02 +00:00
# SETTINGS AND STATUS
if data.startswith('SET:MYCALLSIGN:'):
data = data.split('SET:MYCALLSIGN:')
2021-02-18 14:11:37 +00:00
if bytes(data[1], encoding) == b'':
self.request.sendall(b'INVALID CALLSIGN')
else:
static.MYCALLSIGN = bytes(data[1], encoding)
static.MYCALLSIGN_CRC8 = helpers.get_crc_8(static.MYCALLSIGN)
self.request.sendall(static.MYCALLSIGN)
logging.info("CMD | MYCALLSIGN: " + str(static.MYCALLSIGN))
2021-02-16 19:49:02 +00:00
if data == 'GET:MYCALLSIGN':
self.request.sendall(bytes(static.MYCALLSIGN, encoding))
if data == 'GET:MYCALLSIGN_CRC8':
self.request.sendall(bytes(static.MYCALLSIGN_CRC8, encoding))
if data == 'GET:DXCALLSIGN':
self.request.sendall(bytes(static.DXCALLSIGN, encoding))
# ARQ
if data == 'GET:ARQ_STATE':
self.request.sendall(bytes(static.ARQ_STATE, encoding))
if data == 'GET:TX_N_MAX_RETRIES':
self.request.sendall(bytes([static.TX_N_MAX_RETRIES], encoding))
if data == 'GET:TX_N_RETRIES':
self.request.sendall(bytes([static.TX_N_RETRIES], encoding))
if data == 'GET:ARQ_TX_N_FRAMES_PER_BURST':
self.request.sendall(bytes([static.ARQ_TX_N_FRAMES_PER_BURST], encoding))
if data == 'GET:ARQ_TX_N_BURSTS':
self.request.sendall(bytes([static.ARQ_TX_N_BURSTS], encoding))
if data == 'GET:ARQ_TX_N_CURRENT_ARQ_FRAME':
self.request.sendall(bytes([static.ARQ_TX_N_CURRENT_ARQ_FRAME], encoding))
if data == 'GET:ARQ_TX_N_TOTAL_ARQ_FRAMES':
self.request.sendall(bytes([static.ARQ_TX_N_TOTAL_ARQ_FRAMES], encoding))
if data == 'GET:ARQ_RX_FRAME_N_BURSTS':
self.request.sendall(bytes([static.ARQ_RX_FRAME_N_BURSTS], encoding))
if data == 'GET:ARQ_RX_N_CURRENT_ARQ_FRAME':
self.request.sendall(bytes([static.ARQ_RX_N_CURRENT_ARQ_FRAME], encoding))
if data == 'GET:ARQ_N_ARQ_FRAMES_PER_DATA_FRAME':
self.request.sendall(bytes([static.ARQ_N_ARQ_FRAMES_PER_DATA_FRAME], encoding))
if data == 'GET:RX_BUFFER_LENGTH':
self.request.sendall(bytes(str(len(static.RX_BUFFER)),encoding))
2021-02-16 13:23:57 +00:00
2021-02-16 19:49:02 +00:00
if data.startswith('GET:RX_BUFFER:'):
data = data.split('GET:RX_BUFFER:')
bufferposition = int(data[1])-1
2021-02-16 21:41:06 +00:00
if bufferposition == -1:
2021-02-16 19:49:02 +00:00
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]))
2021-02-18 14:11:37 +00:00
if data == 'DEL:RX_BUFFER':
static.RX_BUFFER = []
2021-02-16 13:36:01 +00:00
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()