FreeDATA/tnc/data_handler.py

3617 lines
140 KiB
Python
Raw Normal View History

# -*- coding: UTF-8 -*-
2021-02-24 13:22:28 +00:00
"""
Created on Sun Dec 27 20:43:40 2020
@author: DJ2LS
"""
# pylint: disable=invalid-name, line-too-long, c-extension-no-member
# pylint: disable=import-outside-toplevel, attribute-defined-outside-init
import os
import base64
import sys
2021-02-24 13:22:28 +00:00
import threading
import time
import uuid
2022-12-12 14:02:57 +00:00
import lzma
2023-01-25 11:47:18 +00:00
from random import randrange
import hmac
import hashlib
import codec2
import helpers
import modem
import numpy as np
import sock
2023-04-27 19:43:56 +00:00
from static import ARQ, AudioParam, Beacon, Channel, Daemon, HamlibParam, ModemParam, Station, Statistics, TCIParam, TNC
import structlog
2023-02-02 17:03:22 +00:00
import stats
import ujson as json
2023-04-23 13:59:30 +00:00
from codec2 import FREEDV_MODE, FREEDV_MODE_USED_SLOTS
2023-06-13 12:03:36 +00:00
from queues import DATA_QUEUE_RECEIVED, DATA_QUEUE_TRANSMIT, RX_BUFFER, MESH_RECEIVED_QUEUE
from static import FRAME_TYPE as FR_TYPE
2023-05-15 06:56:01 +00:00
import broadcast
2023-04-26 16:23:49 +00:00
TESTMODE = False
2022-05-09 00:41:49 +00:00
2022-05-09 01:27:24 +00:00
2022-05-19 20:15:24 +00:00
class DATA:
"""Terminal Node Controller for FreeDATA"""
log = structlog.get_logger("DATA")
2023-02-06 15:56:50 +00:00
def __init__(self) -> None:
2023-02-06 15:56:50 +00:00
self.stats = stats.stats()
# Initial call sign. Will be overwritten later
2023-04-27 19:43:56 +00:00
self.mycallsign = Station.mycallsign
self.dxcallsign = Station.dxcallsign
2022-01-04 15:34:20 +00:00
self.data_queue_transmit = DATA_QUEUE_TRANSMIT
self.data_queue_received = DATA_QUEUE_RECEIVED
2022-01-04 15:34:20 +00:00
2022-09-16 15:07:58 +00:00
# length of signalling frame
self.length_sig0_frame = 14
self.length_sig1_frame = 14
2022-09-16 15:07:58 +00:00
2023-04-22 13:28:50 +00:00
# duration of signalling frame
2023-05-04 09:22:44 +00:00
self.duration_sig0_frame = 2.3
self.duration_sig1_frame = 2.3
2023-04-22 20:50:32 +00:00
self.longest_duration = 5.8 # datac5
2023-04-22 13:28:50 +00:00
2022-10-05 17:24:50 +00:00
# hold session id
self.session_id = bytes(1)
# ------- ARQ SESSION
self.arq_file_transfer = False
self.IS_ARQ_SESSION_MASTER = False
self.arq_session_last_received = 0
self.arq_session_timeout = 30
self.session_connect_max_retries = 15
self.irs_buffer_position = 0
2022-10-05 18:27:38 +00:00
# actual n retries of burst
self.tx_n_retry_of_burst = 0
self.transmission_uuid = ""
2022-05-09 00:41:49 +00:00
2023-04-22 20:50:32 +00:00
self.burst_last_received = 0.0 # time of last "live sign" of a burst
2022-05-19 20:15:24 +00:00
self.data_channel_last_received = 0.0 # time of last "live sign" of a frame
self.burst_ack_snr = 0 # SNR from received burst ack frames
2023-04-25 16:51:51 +00:00
# Flag to indicate if we received an ACK frame for a burst
self.burst_ack = False
2023-04-25 16:51:51 +00:00
# Flag to indicate if we received an ACK frame for a data frame
self.data_frame_ack_received = False
2023-04-25 16:51:51 +00:00
# Flag to indicate if we received a request for repeater frames
self.rpt_request_received = False
2022-05-19 20:15:24 +00:00
self.rpt_request_buffer = [] # requested frames, saved in a list
self.burst_rpt_counter = 0
2022-05-19 20:15:24 +00:00
self.rx_start_of_transmission = 0 # time of transmission start
# 3 bytes for the BOF Beginning of File indicator in a data frame
self.data_frame_bof = b"BOF"
# 3 bytes for the EOF End of File indicator in a data frame
self.data_frame_eof = b"EOF"
2021-09-26 15:51:11 +00:00
self.tx_n_max_retries_per_burst = 40
self.rx_n_max_retries_per_burst = 40
self.n_retries_per_burst = 0
self.rx_n_frame_of_burst = 0
self.rx_n_frames_per_burst = 0
2023-04-25 14:32:36 +00:00
self.max_n_frames_per_burst = 1
2022-05-09 00:41:49 +00:00
2023-05-15 06:56:01 +00:00
self.broadcast = broadcast.broadcastHandler()
2023-05-14 17:44:35 +00:00
2023-04-25 19:08:48 +00:00
# Flag to indicate if we received a low bandwidth mode channel opener
self.received_LOW_BANDWIDTH_MODE = False
self.data_channel_max_retries = 15
self.datachannel_timeout = False
2022-05-09 00:41:49 +00:00
# -------------- AVAILABLE MODES START-----------
# IMPORTANT: LISTS MUST BE OF EQUAL LENGTH
# --------------------- LOW BANDWIDTH
# List of codec2 modes to use in "low bandwidth" mode.
self.mode_list_low_bw = [
FREEDV_MODE.datac4.value,
]
# List for minimum SNR operating level for the corresponding mode in self.mode_list
2023-04-25 16:51:51 +00:00
self.snr_list_low_bw = [-100]
# List for time to wait for corresponding mode in seconds
2023-05-04 09:22:44 +00:00
self.time_list_low_bw = [6 + self.duration_sig0_frame + 1]
# --------------------- HIGH BANDWIDTH
# List of codec2 modes to use in "high bandwidth" mode.
self.mode_list_high_bw = [
FREEDV_MODE.datac4.value,
FREEDV_MODE.datac3.value,
FREEDV_MODE.datac1.value,
]
# List for minimum SNR operating level for the corresponding mode in self.mode_list
2023-04-25 16:51:51 +00:00
self.snr_list_high_bw = [-100, 0, 3]
# List for time to wait for corresponding mode in seconds
# test with 6,7 --> caused sometimes a frame timeout if ack frame takes longer
# TODO: Need to check why ACK frames needs more time
# TODO: Adjust these times
2023-05-04 09:22:44 +00:00
self.time_list_high_bw = [6 + self.duration_sig0_frame + 1, 6 + self.duration_sig0_frame + 1, 6 + self.duration_sig0_frame + 1]
# -------------- AVAILABLE MODES END-----------
# Mode list for selecting between low bandwidth ( 500Hz ) and modes with higher bandwidth
# but ability to fall back to low bandwidth modes if needed.
2023-04-27 19:43:56 +00:00
if TNC.low_bandwidth_mode:
# List of codec2 modes to use in "low bandwidth" mode.
self.mode_list = self.mode_list_low_bw
# list of times to wait for corresponding mode in seconds
self.time_list = self.time_list_low_bw
else:
# List of codec2 modes to use in "high bandwidth" mode.
self.mode_list = self.mode_list_high_bw
# list of times to wait for corresponding mode in seconds
self.time_list = self.time_list_high_bw
2022-05-09 00:41:49 +00:00
2022-05-19 20:15:24 +00:00
self.speed_level = len(self.mode_list) - 1 # speed level for selecting mode
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
2022-05-09 00:41:49 +00:00
2023-02-23 09:54:52 +00:00
# minimum payload for arq burst
# import for avoiding byteorder bug and buffer search area
self.arq_burst_header_size = 3
2023-04-22 13:28:50 +00:00
self.arq_burst_minimum_payload = 56 - self.arq_burst_header_size
2023-02-23 09:54:52 +00:00
self.arq_burst_maximum_payload = 510 - self.arq_burst_header_size
2023-04-24 09:52:03 +00:00
# save last used payload for optimising buffer search area
self.arq_burst_last_payload = self.arq_burst_maximum_payload
2023-02-23 09:54:52 +00:00
self.is_IRS = False
self.burst_nack = False
self.burst_nack_counter = 0
self.frame_nack_counter = 0
self.frame_received_counter = 0
2022-05-09 00:41:49 +00:00
2022-01-07 10:44:35 +00:00
self.rx_frame_bof_received = False
self.rx_frame_eof_received = False
2022-10-05 18:27:38 +00:00
# TIMEOUTS
2023-04-22 20:50:32 +00:00
self.burst_ack_timeout_seconds = 4.5 # timeout for burst acknowledges
self.data_frame_ack_timeout_seconds = 4.5 # timeout for data frame acknowledges
self.rpt_ack_timeout_seconds = 4.5 # timeout for rpt frame acknowledges
self.transmission_timeout = 180 # transmission timeout in seconds
2021-02-24 13:22:28 +00:00
# Dictionary of functions and log messages used in process_data
# instead of a long series of if-elif-else statements.
self.rx_dispatcher = {
FR_TYPE.ARQ_DC_OPEN_ACK_N.value: (
self.arq_received_channel_is_open,
"ARQ OPEN ACK (Narrow)",
),
FR_TYPE.ARQ_DC_OPEN_ACK_W.value: (
self.arq_received_channel_is_open,
"ARQ OPEN ACK (Wide)",
),
FR_TYPE.ARQ_DC_OPEN_N.value: (
self.arq_received_data_channel_opener,
"ARQ Data Channel Open (Narrow)",
),
FR_TYPE.ARQ_DC_OPEN_W.value: (
self.arq_received_data_channel_opener,
"ARQ Data Channel Open (Wide)",
),
FR_TYPE.ARQ_SESSION_CLOSE.value: (
self.received_session_close,
"ARQ CLOSE SESSION",
),
FR_TYPE.ARQ_SESSION_HB.value: (
self.received_session_heartbeat,
"ARQ HEARTBEAT",
),
FR_TYPE.ARQ_SESSION_OPEN.value: (
self.received_session_opener,
"ARQ OPEN SESSION",
),
FR_TYPE.ARQ_STOP.value: (self.received_stop_transmission, "ARQ STOP TX"),
FR_TYPE.BEACON.value: (self.received_beacon, "BEACON"),
FR_TYPE.BURST_ACK.value: (self.burst_ack_nack_received, "BURST ACK"),
FR_TYPE.BURST_NACK.value: (self.burst_ack_nack_received, "BURST NACK"),
FR_TYPE.CQ.value: (self.received_cq, "CQ"),
FR_TYPE.FR_ACK.value: (self.frame_ack_received, "FRAME ACK"),
FR_TYPE.FR_NACK.value: (self.frame_nack_received, "FRAME NACK"),
FR_TYPE.FR_REPEAT.value: (self.burst_rpt_received, "REPEAT REQUEST"),
FR_TYPE.PING_ACK.value: (self.received_ping_ack, "PING ACK"),
FR_TYPE.PING.value: (self.received_ping, "PING"),
FR_TYPE.QRV.value: (self.received_qrv, "QRV"),
2023-02-12 17:34:05 +00:00
FR_TYPE.IS_WRITING.value: (self.received_is_writing, "IS_WRITING"),
2023-05-15 06:56:01 +00:00
FR_TYPE.FEC.value: (self.broadcast.received_fec, "FEC"),
FR_TYPE.FEC_WAKEUP.value: (self.broadcast.received_fec_wakeup, "FEC WAKEUP"),
2023-02-12 17:34:05 +00:00
}
self.command_dispatcher = {
# "CONNECT": (self.arq_session_handler, "CONNECT"),
"CQ": (self.transmit_cq, "CQ"),
"DISCONNECT": (self.close_session, "DISCONNECT"),
"SEND_TEST_FRAME": (self.send_test_frame, "TEST"),
"STOP": (self.stop_transmission, "STOP"),
}
# Start worker and watchdog threads
worker_thread_transmit = threading.Thread(
target=self.worker_transmit, name="worker thread transmit", daemon=True
)
worker_thread_transmit.start()
2022-05-09 00:41:49 +00:00
worker_thread_receive = threading.Thread(
target=self.worker_receive, name="worker thread receive", daemon=True
)
worker_thread_receive.start()
2022-05-09 00:41:49 +00:00
# START THE THREAD FOR THE TIMEOUT WATCHDOG
watchdog_thread = threading.Thread(
target=self.watchdog, name="watchdog", daemon=True
)
watchdog_thread.start()
2022-05-09 00:41:49 +00:00
arq_session_thread = threading.Thread(
target=self.heartbeat, name="watchdog", daemon=True
)
2022-05-09 00:41:49 +00:00
arq_session_thread.start()
self.beacon_interval = 0
self.beacon_interval_timer = 0
self.beacon_thread = threading.Thread(
target=self.run_beacon, name="watchdog", daemon=True
)
self.beacon_thread.start()
2022-05-09 00:41:49 +00:00
def worker_transmit(self) -> None:
"""Dispatch incoming UI instructions for transmitting operations"""
while True:
data = self.data_queue_transmit.get()
# if we are already in ARQ_STATE, or we're receiving codec2 traffic
# let's wait with processing data
# this should avoid weird toggle states where both stations
# stuck in IRS
#
# send transmission queued information once
2023-04-27 19:43:56 +00:00
if ARQ.arq_state or ModemParam.is_codec2_traffic:
2023-02-09 12:26:25 +00:00
self.log.debug(
"[TNC] TX DISPATCHER - waiting with processing command ",
2023-04-27 19:43:56 +00:00
arq_state=ARQ.arq_state,
2023-02-09 12:26:25 +00:00
)
2023-01-25 12:23:09 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
2023-01-25 13:17:09 +00:00
command=data[0],
2023-01-25 13:17:41 +00:00
status="queued",
)
# now stay in while loop until state released
2023-04-27 19:43:56 +00:00
while ARQ.arq_state or ModemParam.is_codec2_traffic:
threading.Event().wait(0.01)
# and finally sleep some time
threading.Event().wait(1.0)
# Dispatch commands known to command_dispatcher
if data[0] in self.command_dispatcher:
self.log.debug(f"[TNC] TX {self.command_dispatcher[data[0]][1]}...")
self.command_dispatcher[data[0]][0]()
# Dispatch commands that need more arguments.
elif data[0] == "CONNECT":
2022-12-05 15:57:45 +00:00
# [1] mycallsign
# [2] dxcallsign
# [3] attempts
self.arq_session_handler(data[1], data[2], data[3])
elif data[0] == "PING":
2022-12-05 15:57:45 +00:00
# [1] mycallsign
# [2] dxcallsign
self.transmit_ping(data[1], data[2])
2022-05-09 00:41:49 +00:00
elif data[0] == "BEACON":
# [1] INTERVAL int
# [2] STATE bool
if data[2]:
self.beacon_interval = data[1]
2023-04-27 19:43:56 +00:00
Beacon.beacon_state = True
else:
2023-04-27 19:43:56 +00:00
Beacon.beacon_state = False
2022-05-09 00:41:49 +00:00
elif data[0] == "ARQ_RAW":
# [1] DATA_OUT bytes
2023-05-08 10:58:25 +00:00
# [2] self.transmission_uuid str
# [3] mycallsign with ssid
# [4] dxcallsign with ssid
# [5] attempts
# [6] hmac salt hash
self.open_dc_and_transmit(data[1], data[2], data[3], data[4], data[5], data[6])
2022-03-19 11:42:10 +00:00
2023-02-12 17:24:42 +00:00
elif data[0] == "FEC_IS_WRITING":
# [1] DATA_OUT bytes
# [2] MODE str datac0/1/3...
2023-02-12 17:38:00 +00:00
self.send_fec_is_writing(data[1])
elif data[0] == "FEC":
# [1] WAKEUP bool
# [2] MODE str datac0/1/3...
# [3] PAYLOAD
2023-05-14 16:12:06 +00:00
# [4] MYCALLSIGN
self.send_fec(data[1], data[2], data[3], data[4])
else:
self.log.error(
"[TNC] worker_transmit: received invalid command:", data=data
)
def worker_receive(self) -> None:
"""Queue received data for processing"""
while True:
data = self.data_queue_received.get()
# [0] bytes
# [1] freedv instance
# [2] bytes_per_frame
self.process_data(
bytes_out=data[0], freedv=data[1], bytes_per_frame=data[2]
)
def process_data(self, bytes_out, freedv, bytes_per_frame: int) -> None:
"""
Process incoming data and decide what to do with the frame.
Args:
2022-05-09 00:41:49 +00:00
bytes_out:
freedv:
bytes_per_frame:
Returns:
"""
self.log.debug(
"[TNC] process_data:", n_retries_per_burst=self.n_retries_per_burst
)
# Process data only if broadcast or we are the receiver
2022-05-09 00:41:49 +00:00
# bytes_out[1:4] == callsign check for signalling frames,
# bytes_out[2:5] == transmission
# we could also create an own function, which returns True.
2022-04-11 09:03:54 +00:00
frametype = int.from_bytes(bytes(bytes_out[:1]), "big")
2022-10-05 17:24:50 +00:00
# check for callsign CRC
_valid1, _ = helpers.check_callsign(self.mycallsign, bytes(bytes_out[1:4]))
_valid2, _ = helpers.check_callsign(self.mycallsign, bytes(bytes_out[2:5]))
2022-10-05 17:24:50 +00:00
# check for session ID
2022-10-06 09:21:36 +00:00
# signalling frames
2022-10-05 17:24:50 +00:00
_valid3 = helpers.check_session_id(self.session_id, bytes(bytes_out[1:2]))
2022-10-06 09:21:36 +00:00
# arq data frames
_valid4 = helpers.check_session_id(self.session_id, bytes(bytes_out[2:3]))
if (
2022-09-08 13:47:23 +00:00
_valid1
or _valid2
2022-10-05 17:24:50 +00:00
or _valid3
2022-10-06 09:21:36 +00:00
or _valid4
2022-09-08 13:47:23 +00:00
or frametype
in [
FR_TYPE.CQ.value,
FR_TYPE.QRV.value,
FR_TYPE.PING.value,
FR_TYPE.BEACON.value,
FR_TYPE.IS_WRITING.value,
2023-05-14 19:55:17 +00:00
FR_TYPE.FEC.value,
FR_TYPE.FEC_WAKEUP.value,
]
):
2022-05-09 00:41:49 +00:00
# CHECK IF FRAMETYPE IS BETWEEN 10 and 50 ------------------------
2022-10-05 18:27:38 +00:00
# frame = frametype - 10
# n_frames_per_burst = int.from_bytes(bytes(bytes_out[1:2]), "big")
# Dispatch activity based on received frametype
if frametype in self.rx_dispatcher:
# Process frames "known" by rx_dispatcher
# self.log.debug(f"[TNC] {self.rx_dispatcher[frametype][1]} RECEIVED....")
self.rx_dispatcher[frametype][0](bytes_out[:-2])
# Process frametypes requiring a different set of arguments.
elif FR_TYPE.BURST_51.value >= frametype >= FR_TYPE.BURST_01.value:
# get snr of received data
# FIXME: find a fix for this - after moving to classes, this no longer works
2022-05-19 20:15:24 +00:00
# snr = self.calculate_snr(freedv)
2023-04-27 19:43:56 +00:00
snr = ModemParam.snr
self.log.debug("[TNC] RX SNR", snr=snr)
# send payload data to arq checker without CRC16
self.arq_data_received(
bytes(bytes_out[:-2]), bytes_per_frame, snr, freedv
)
# if we received the last frame of a burst or the last remaining rpt frame, do a modem unsync
2023-04-27 19:43:56 +00:00
# if ARQ.rx_burst_buffer.count(None) <= 1 or (frame+1) == n_frames_per_burst:
# self.log.debug(f"[TNC] LAST FRAME OF BURST --> UNSYNC {frame+1}/{n_frames_per_burst}")
2022-01-07 10:44:35 +00:00
# self.c_lib.freedv_set_sync(freedv, 0)
# TESTFRAMES
elif frametype == FR_TYPE.TEST_FRAME.value:
self.log.debug("[TNC] TESTFRAME RECEIVED", frame=bytes_out[:])
2022-05-09 00:41:49 +00:00
# Unknown frame type
else:
2022-07-02 20:20:51 +00:00
self.log.warning(
"[TNC] ARQ - other frame type", frametype=FR_TYPE(frametype).name
)
2021-03-12 13:14:36 +00:00
else:
# for debugging purposes to receive all data
self.log.debug(
"[TNC] Foreign frame received",
frame=bytes_out[:-2].hex(),
frame_type=FR_TYPE(int.from_bytes(bytes_out[:1], byteorder="big")).name,
)
def enqueue_frame_for_tx(
2022-09-08 13:47:23 +00:00
self,
frame_to_tx, # : list[bytearray], # this causes a crash on python 3.7
c2_mode=FREEDV_MODE.sig0.value,
2022-09-08 13:47:23 +00:00
copies=1,
repeat_delay=0,
) -> None:
"""
Send (transmit) supplied frame to TNC
:param frame_to_tx: Frame data to send
:type frame_to_tx: list of bytearrays
:param c2_mode: Codec2 mode to use, defaults to datac13
:type c2_mode: int, optional
:param copies: Number of frame copies to send, defaults to 1
:type copies: int, optional
:param repeat_delay: Delay time before sending repeat frame, defaults to 0
:type repeat_delay: int, optional
"""
2023-05-06 14:10:31 +00:00
#print(frame_to_tx[0])
#print(frame_to_tx)
2023-01-06 14:08:58 +00:00
frame_type = FR_TYPE(int.from_bytes(frame_to_tx[0][:1], byteorder="big")).name
self.log.debug("[TNC] enqueue_frame_for_tx", c2_mode=FREEDV_MODE(c2_mode).name, data=frame_to_tx,
type=frame_type)
# Set the TRANSMITTING flag before adding an object to the transmit queue
# TODO: This is not that nice, we could improve this somehow
2023-04-27 19:43:56 +00:00
TNC.transmitting = True
modem.MODEM_TRANSMIT_QUEUE.put([c2_mode, copies, repeat_delay, frame_to_tx])
# Wait while transmitting
2023-04-27 19:43:56 +00:00
while TNC.transmitting:
threading.Event().wait(0.01)
2022-07-04 21:09:08 +00:00
def send_data_to_socket_queue(self, **jsondata):
"""
Send information to the UI via JSON and the sock.SOCKET_QUEUE.
Args:
Dictionary containing the data to be sent, in the format:
key=value, for each item. E.g.:
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
arq="received",
2022-06-05 17:11:09 +00:00
status="success",
2022-06-06 19:07:41 +00:00
uuid=self.transmission_uuid,
timestamp=timestamp,
2022-06-06 19:07:41 +00:00
mycallsign=str(self.mycallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
dxcallsign=str(Station.dxcallsign, "UTF-8"),
dxgrid=str(Station.dxgrid, "UTF-8"),
data=base64_data,
)
"""
# add mycallsign and dxcallsign to network message if they not exist
# and make sure we are not overwrite them if they exist
try:
2022-12-16 16:09:48 +00:00
if "mycallsign" not in jsondata:
jsondata["mycallsign"] = str(self.mycallsign, "UTF-8")
2022-12-16 16:09:48 +00:00
if "dxcallsign" not in jsondata:
2023-04-27 19:43:56 +00:00
jsondata["dxcallsign"] = str(Station.dxcallsign, "UTF-8")
except Exception as e:
self.log.debug("[TNC] error adding callsigns to network message", e=e)
2022-12-01 08:06:28 +00:00
# run json dumps
json_data_out = json.dumps(jsondata)
2022-12-01 08:06:28 +00:00
self.log.debug("[TNC] send_data_to_socket_queue:", jsondata=json_data_out)
# finally push data to our network queue
sock.SOCKET_QUEUE.put(json_data_out)
def send_ident_frame(self, transmit) -> None:
2022-10-21 19:59:26 +00:00
"""Build and send IDENT frame """
ident_frame = bytearray(self.length_sig1_frame)
ident_frame[:1] = bytes([FR_TYPE.IDENT.value])
ident_frame[1:self.length_sig1_frame] = self.mycallsign
# Transmit frame
if transmit:
self.enqueue_frame_for_tx([ident_frame], c2_mode=FREEDV_MODE.sig0.value)
else:
return ident_frame
2022-10-21 19:59:26 +00:00
def send_burst_ack_frame(self, snr) -> None:
"""Build and send ACK frame for burst DATA frame"""
2022-11-09 19:47:46 +00:00
ack_frame = bytearray(self.length_sig1_frame)
2022-06-24 13:48:50 +00:00
ack_frame[:1] = bytes([FR_TYPE.BURST_ACK.value])
2022-10-05 17:24:50 +00:00
ack_frame[1:2] = self.session_id
2022-11-09 19:47:46 +00:00
ack_frame[2:3] = helpers.snr_to_bytes(snr)
2022-10-10 07:00:45 +00:00
ack_frame[3:4] = bytes([int(self.speed_level)])
2023-04-27 19:43:56 +00:00
ack_frame[4:8] = len(ARQ.rx_frame_buffer).to_bytes(4, byteorder="big")
2022-12-29 16:57:55 +00:00
# wait while timeout not reached and our busy state is busy
channel_busy_timeout = time.time() + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
2022-12-29 16:57:55 +00:00
threading.Event().wait(0.01)
# Transmit frame
2022-11-02 21:48:50 +00:00
self.enqueue_frame_for_tx([ack_frame], c2_mode=FREEDV_MODE.sig1.value)
def send_data_ack_frame(self, snr) -> None:
"""Build and send ACK frame for received DATA frame"""
2022-11-09 19:47:46 +00:00
ack_frame = bytearray(self.length_sig1_frame)
ack_frame[:1] = bytes([FR_TYPE.FR_ACK.value])
2022-10-05 17:24:50 +00:00
ack_frame[1:2] = self.session_id
2022-11-09 19:47:46 +00:00
ack_frame[2:3] = helpers.snr_to_bytes(snr)
2022-12-29 16:57:55 +00:00
# wait while timeout not reached and our busy state is busy
channel_busy_timeout = time.time() + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
2022-12-29 16:57:55 +00:00
threading.Event().wait(0.01)
# reset burst timeout in case we had to wait too long
self.burst_last_received = time.time() + channel_busy_timeout + 8
# Transmit frame
2022-10-28 09:02:23 +00:00
# TODO: Do we have to send , self.send_ident_frame(False) ?
# self.enqueue_frame_for_tx([ack_frame, self.send_ident_frame(False)], c2_mode=FREEDV_MODE.sig1.value, copies=3, repeat_delay=0)
self.enqueue_frame_for_tx([ack_frame], c2_mode=FREEDV_MODE.sig1.value, copies=3, repeat_delay=0)
2023-01-06 14:50:25 +00:00
def send_retransmit_request_frame(self) -> None:
2022-10-28 09:02:23 +00:00
# check where a None is in our burst buffer and do frame+1, because lists start at 0
# FIXME: Check to see if there's a `frame - 1` in the receive portion. Remove both if there is.
2023-05-06 14:10:31 +00:00
#print(ARQ.rx_burst_buffer)
missing_frames = [
frame + 1
2023-04-27 19:43:56 +00:00
for frame, element in enumerate(ARQ.rx_burst_buffer)
if element is None
]
rpt_frame = bytearray(self.length_sig1_frame)
rpt_frame[:1] = bytes([FR_TYPE.FR_REPEAT.value])
2022-10-05 17:24:50 +00:00
rpt_frame[1:2] = self.session_id
rpt_frame[2:2 + len(missing_frames)] = missing_frames
self.log.info("[TNC] ARQ | RX | Requesting", frames=missing_frames)
# Transmit frame
2022-11-02 21:48:50 +00:00
self.enqueue_frame_for_tx([rpt_frame], c2_mode=FREEDV_MODE.sig1.value, copies=1, repeat_delay=0)
def send_burst_nack_frame(self, snr: bytes) -> None:
"""Build and send NACK frame for received DATA frame"""
2022-11-09 19:47:46 +00:00
nack_frame = bytearray(self.length_sig1_frame)
nack_frame[:1] = bytes([FR_TYPE.FR_NACK.value])
2022-10-05 17:24:50 +00:00
nack_frame[1:2] = self.session_id
2022-11-12 11:40:32 +00:00
nack_frame[2:3] = helpers.snr_to_bytes(snr)
2022-10-10 07:00:45 +00:00
nack_frame[3:4] = bytes([int(self.speed_level)])
2023-04-27 19:43:56 +00:00
nack_frame[4:8] = len(ARQ.rx_frame_buffer).to_bytes(4, byteorder="big")
# TRANSMIT NACK FRAME FOR BURST
2022-10-28 09:03:15 +00:00
# TODO: Do we have to send ident frame?
# self.enqueue_frame_for_tx([ack_frame, self.send_ident_frame(False)], c2_mode=FREEDV_MODE.sig1.value, copies=3, repeat_delay=0)
2022-12-29 16:57:55 +00:00
# wait while timeout not reached and our busy state is busy
channel_busy_timeout = time.time() + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
2022-12-29 16:57:55 +00:00
threading.Event().wait(0.01)
2023-04-21 16:14:44 +00:00
self.enqueue_frame_for_tx([nack_frame], c2_mode=FREEDV_MODE.sig1.value, copies=3, repeat_delay=0)
2023-01-06 14:50:25 +00:00
# reset burst timeout in case we had to wait too long
self.burst_last_received = time.time()
def send_burst_nack_frame_watchdog(self, snr: bytes, tx_n_frames_per_burst) -> None:
"""Build and send NACK frame for watchdog timeout"""
# increment nack counter for transmission stats
self.frame_nack_counter += 1
2023-04-25 19:08:48 +00:00
# we need to clear our rx burst buffer
2023-04-27 19:43:56 +00:00
ARQ.rx_burst_buffer = []
2023-04-25 19:08:48 +00:00
# Create and send ACK frame
self.log.info("[TNC] ARQ | RX | SENDING NACK")
nack_frame = bytearray(self.length_sig1_frame)
nack_frame[:1] = bytes([FR_TYPE.BURST_NACK.value])
2022-10-05 17:24:50 +00:00
nack_frame[1:2] = self.session_id
2022-11-09 19:47:46 +00:00
nack_frame[2:3] = helpers.snr_to_bytes(snr)
2022-10-10 07:00:45 +00:00
nack_frame[3:4] = bytes([int(self.speed_level)])
2023-04-25 19:08:48 +00:00
nack_frame[4:5] = bytes([int(tx_n_frames_per_burst)])
nack_frame[5:9] = len(ARQ.rx_frame_buffer).to_bytes(4, byteorder="big")
2022-12-29 16:57:55 +00:00
# wait while timeout not reached and our busy state is busy
channel_busy_timeout = time.time() + 5 + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
2022-12-29 16:57:55 +00:00
threading.Event().wait(0.01)
# TRANSMIT NACK FRAME FOR BURST
2022-11-02 21:48:50 +00:00
self.enqueue_frame_for_tx([nack_frame], c2_mode=FREEDV_MODE.sig1.value, copies=1, repeat_delay=0)
2023-01-11 15:18:29 +00:00
2023-02-23 10:10:54 +00:00
# reset frame counter for not increasing speed level
self.frame_received_counter = 0
def send_disconnect_frame(self) -> None:
"""Build and send a disconnect frame"""
disconnection_frame = bytearray(self.length_sig1_frame)
disconnection_frame[:1] = bytes([FR_TYPE.ARQ_SESSION_CLOSE.value])
2022-10-05 17:24:50 +00:00
disconnection_frame[1:2] = self.session_id
2023-04-27 19:43:56 +00:00
disconnection_frame[2:5] = Station.dxcallsign_crc
2022-10-10 07:00:45 +00:00
# TODO: Needed? disconnection_frame[7:13] = helpers.callsign_to_bytes(self.mycallsign)
# self.enqueue_frame_for_tx([disconnection_frame, self.send_ident_frame(False)], c2_mode=FREEDV_MODE.sig0.value, copies=5, repeat_delay=0)
2022-11-02 14:29:22 +00:00
# TODO: We need to add the ident frame feature with a seperate PR after publishing latest protocol
# TODO: We need to wait some time between last arq related signalling frame and ident frame
# TODO: Maybe about 500ms - 1500ms to avoid confusion and too much PTT toggles
2022-12-29 16:57:55 +00:00
# wait while timeout not reached and our busy state is busy
channel_busy_timeout = time.time() + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
2022-12-29 16:57:55 +00:00
threading.Event().wait(0.01)
2023-04-21 16:14:44 +00:00
self.enqueue_frame_for_tx([disconnection_frame], c2_mode=FREEDV_MODE.sig0.value, copies=3, repeat_delay=0)
def arq_data_received(
self, data_in: bytes, bytes_per_frame: int, snr: float, freedv
) -> None:
"""
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
bytes_per_frame:int:
snr:float:
freedv:
Returns:
"""
# We've arrived here from process_data which already checked that the frame
# is intended for this station.
2022-05-09 00:41:49 +00:00
data_in = bytes(data_in)
# only process data if we are in ARQ and BUSY state else return to quit
2023-04-27 19:43:56 +00:00
if not ARQ.arq_state and TNC.tnc_state not in ["BUSY"]:
self.log.warning("[TNC] wrong tnc state - dropping data", arq_state=ARQ.arq_state,
tnc_state=TNC.tnc_state)
return
self.arq_file_transfer = True
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "BUSY"
ARQ.arq_state = True
# Update data_channel timestamp
self.data_channel_last_received = int(time.time())
2023-01-04 10:04:10 +00:00
self.burst_last_received = int(time.time())
2022-05-09 00:41:49 +00:00
# Extract some important data from the frame
# Get sequence number of burst frame
self.rx_n_frame_of_burst = int.from_bytes(bytes(data_in[:1]), "big") - 10
# Get number of bursts from received frame
self.rx_n_frames_per_burst = int.from_bytes(bytes(data_in[1:2]), "big")
# The RX burst buffer needs to have a fixed length filled with "None".
# We need this later for counting the "Nones" to detect missing data.
# Check if burst buffer has expected length else create it
2023-04-27 19:43:56 +00:00
if len(ARQ.rx_burst_buffer) != self.rx_n_frames_per_burst:
ARQ.rx_burst_buffer = [None] * self.rx_n_frames_per_burst
# Append data to rx burst buffer
2023-04-27 19:43:56 +00:00
ARQ.rx_burst_buffer[self.rx_n_frame_of_burst] = data_in[self.arq_burst_header_size:] # type: ignore
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
snr,
2023-04-27 19:43:56 +00:00
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
2022-05-09 00:41:49 +00:00
# Check if we received all frames in the burst by checking if burst buffer has no more "Nones"
# This is the ideal case because we received all data
2023-04-27 19:43:56 +00:00
if None not in ARQ.rx_burst_buffer:
# then iterate through burst buffer and stick the burst together
2022-11-02 14:29:22 +00:00
# the temp burst buffer is needed for checking, if we already received data
temp_burst_buffer = b""
2023-04-27 19:43:56 +00:00
for value in ARQ.rx_burst_buffer:
# ARQ.rx_frame_buffer += ARQ.rx_burst_buffer[i]
temp_burst_buffer += bytes(value) # type: ignore
# free up burst buffer
2023-04-27 19:43:56 +00:00
ARQ.rx_burst_buffer = []
# TODO: Needs to be removed as soon as mode error is fixed
# catch possible modem error which leads into false byteorder
# modem possibly decodes too late - data then is pushed to buffer
# which leads into wrong byteorder
2023-02-09 17:06:00 +00:00
# Lets put this in try/except so we are not crashing tnc as its highly experimental
# This might only work for datac1 and datac3
try:
2023-02-09 17:06:00 +00:00
# area_of_interest = (modem.get_bytes_per_frame(self.mode_list[speed_level] - 1) -3) * 2
2023-04-27 19:43:56 +00:00
if ARQ.rx_frame_buffer.endswith(temp_burst_buffer[:246]) and len(temp_burst_buffer) >= 246:
self.log.warning(
"[TNC] ARQ | RX | wrong byteorder received - dropping data"
)
# we need to run a return here, so we are not sending an ACK
# return
except Exception as e:
self.log.warning(
"[TNC] ARQ | RX | wrong byteorder check failed", e=e
)
self.log.debug("[TNC] temp_burst_buffer", buffer=temp_burst_buffer)
2023-04-27 19:43:56 +00:00
self.log.debug("[TNC] ARQ.rx_frame_buffer", buffer=ARQ.rx_frame_buffer)
# if frame buffer ends not with the current frame, we are going to append new data
# if data already exists, we received the frame correctly,
# but the ACK frame didn't receive its destination (ISS)
2023-04-27 19:43:56 +00:00
if ARQ.rx_frame_buffer.endswith(temp_burst_buffer):
self.log.info(
"[TNC] ARQ | RX | Frame already received - sending ACK again"
)
2022-05-09 00:41:49 +00:00
else:
# Here we are going to search for our data in the last received bytes.
# This reduces the chance we will lose the entire frame in the case of signalling frame loss
2023-04-27 19:43:56 +00:00
# ARQ.rx_frame_buffer --> existing data
# temp_burst_buffer --> new data
# search_area --> area where we want to search
search_area = self.arq_burst_last_payload * self.rx_n_frames_per_burst
2023-04-27 19:43:56 +00:00
search_position = len(ARQ.rx_frame_buffer) - search_area
# if search position < 0, then search position = 0
search_position = max(0, search_position)
# find position of data. returns -1 if nothing found in area else >= 0
# we are beginning from the end, so if data exists twice or more,
# only the last one should be replaced
2023-02-23 09:54:52 +00:00
# we are going to only check position against minimum data frame payload
# use case: receive data, which already contains received data
# while the payload of data received before is shorter than actual payload
2023-04-27 19:43:56 +00:00
get_position = ARQ.rx_frame_buffer[search_position:].rfind(
2023-02-23 09:54:52 +00:00
temp_burst_buffer[:self.arq_burst_minimum_payload]
)
# if we find data, replace it at this position with the new data and strip it
if get_position >= 0:
2023-04-27 19:43:56 +00:00
ARQ.rx_frame_buffer = ARQ.rx_frame_buffer[
2022-09-08 13:47:23 +00:00
: search_position + get_position
]
self.log.warning(
"[TNC] ARQ | RX | replacing existing buffer data",
area=search_area,
pos=get_position,
)
else:
self.log.debug("[TNC] ARQ | RX | appending data to buffer")
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
ARQ.rx_frame_buffer += temp_burst_buffer
2023-04-24 09:52:03 +00:00
self.arq_burst_last_payload = len(temp_burst_buffer)
# Check if we didn't receive a BOF and EOF yet to avoid sending
# ack frames if we already received all data
if (
2022-09-08 13:47:23 +00:00
not self.rx_frame_bof_received
and not self.rx_frame_eof_received
and data_in.find(self.data_frame_eof) < 0
):
self.arq_calculate_speed_level(snr)
self.data_channel_last_received = int(time.time()) + 6 + 6
2023-04-25 14:32:36 +00:00
self.burst_last_received = int(time.time()) + 6 + 6
# Create and send ACK frame
2023-04-27 19:43:56 +00:00
self.log.info("[TNC] ARQ | RX | SENDING ACK", finished=ARQ.arq_seconds_until_finish,
bytesperminute=ARQ.bytes_per_minute)
self.send_burst_ack_frame(snr)
# Reset n retries per burst counter
2022-04-05 09:29:20 +00:00
self.n_retries_per_burst = 0
2022-05-09 00:41:49 +00:00
2022-04-05 09:29:20 +00:00
# calculate statistics
self.calculate_transfer_rate_rx(
2023-04-27 19:43:56 +00:00
self.rx_start_of_transmission, len(ARQ.rx_frame_buffer)
)
2022-05-09 00:41:49 +00:00
2022-12-05 07:23:18 +00:00
# send a network message with information
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="receiving",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-04-27 19:43:56 +00:00
finished=ARQ.arq_seconds_until_finish,
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-12-05 07:23:18 +00:00
)
2023-02-23 09:54:52 +00:00
# elif self.rx_n_frame_of_burst == self.rx_n_frames_per_burst:
# # We have "Nones" in our rx buffer,
# # Check if we received last frame of burst - this is an indicator for missed frames.
# # With this way of doing this, we always MUST receive the last
# # frame of a burst otherwise the entire burst is lost
# # TODO: See if a timeout on the send side with re-transmit last burst would help.
# self.log.debug(
# "[TNC] last frames of burst received:",
# frame=self.rx_n_frame_of_burst,
# frames=self.rx_n_frames_per_burst,
#
# )
# self.calculate_transfer_rate_rx(
2023-04-27 19:43:56 +00:00
# self.rx_start_of_transmission, len(ARQ.rx_frame_buffer)
# )
# elif self.rx_n_frame_of_burst not in [self.rx_n_frames_per_burst - 1]:
# self.log.info(
# "[TNC] data_handler: received burst",
# frame=self.rx_n_frame_of_burst + 1,
# frames=self.rx_n_frames_per_burst,
# )
# else:
# self.log.error(
# "[TNC] data_handler: Should not reach this point...",
# frame=self.rx_n_frame_of_burst + 1,
# frames=self.rx_n_frames_per_burst,
# )
else:
self.log.warning(
"[TNC] data_handler: missing data in burst buffer...",
frame=self.rx_n_frame_of_burst + 1,
frames=self.rx_n_frames_per_burst
)
# We have a BOF and EOF flag in our data. If we received both we received our frame.
2022-05-19 20:37:27 +00:00
# In case of loosing data, but we received already a BOF and EOF we need to make sure, we
2022-05-09 00:41:49 +00:00
# received the complete last burst by checking it for Nones
2023-04-27 19:43:56 +00:00
bof_position = ARQ.rx_frame_buffer.find(self.data_frame_bof)
eof_position = ARQ.rx_frame_buffer.find(self.data_frame_eof)
2022-11-02 14:29:22 +00:00
# get total bytes per transmission information as soon we received a frame with a BOF
2022-05-09 00:41:49 +00:00
2022-05-19 20:15:24 +00:00
if bof_position >= 0:
2023-02-09 17:15:00 +00:00
self.arq_extract_statistics_from_data_frame(bof_position, eof_position)
if (
2022-09-08 13:47:23 +00:00
bof_position >= 0
and eof_position > 0
2023-04-27 19:43:56 +00:00
and None not in ARQ.rx_burst_buffer
):
self.log.debug(
"[TNC] arq_data_received:",
bof_position=bof_position,
eof_position=eof_position,
)
2022-01-07 10:44:35 +00:00
self.rx_frame_bof_received = True
self.rx_frame_eof_received = True
2022-05-09 00:41:49 +00:00
# Extract raw data from buffer
2023-04-27 19:43:56 +00:00
payload = ARQ.rx_frame_buffer[
2022-09-08 13:47:23 +00:00
bof_position + len(self.data_frame_bof): eof_position
]
# Get the data frame crc
data_frame_crc = payload[:4] # 0:4 = 4 bytes
# Get the data frame length
frame_length = int.from_bytes(payload[4:8], "big") # 4:8 = 4 bytes
2023-04-27 19:43:56 +00:00
ARQ.total_bytes = frame_length
# 8:9 = compression factor
data_frame = payload[9:]
data_frame_crc_received = helpers.get_crc_32(data_frame)
2022-05-09 00:41:49 +00:00
# check if hmac signing enabled
if TNC.enable_hmac:
2023-08-16 07:58:49 +00:00
# now check if we have valid hmac signature - returns salt or bool
salt_found = helpers.search_hmac_salt(self.dxcallsign, self.mycallsign, data_frame_crc, data_frame, token_iters=100)
if salt_found:
# hmac digest received
self.arq_process_received_data_frame(data_frame, snr, signed=True)
else:
2023-08-16 07:58:49 +00:00
# hmac signature wrong
self.arq_process_received_data_frame(data_frame, snr, signed=False)
elif data_frame_crc == data_frame_crc_received:
2023-08-16 07:58:49 +00:00
self.log.warning(
"[TNC] [HMAC] Disabled, using CRC",
)
self.arq_process_received_data_frame(data_frame, snr, signed=False)
else:
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
status="failed",
2022-06-06 19:07:41 +00:00
uuid=self.transmission_uuid,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
duration = time.time() - self.rx_start_of_transmission
self.log.warning(
2022-06-06 18:31:15 +00:00
"[TNC] ARQ | RX | DATA FRAME NOT SUCCESSFULLY RECEIVED!",
e="wrong crc",
expected=data_frame_crc.hex(),
received=data_frame_crc_received.hex(),
2023-04-27 19:43:56 +00:00
overflows=AudioParam.buffer_overflow_counter,
nacks=self.frame_nack_counter,
duration=duration,
2023-04-27 19:43:56 +00:00
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
2022-12-31 10:55:41 +00:00
data=data_frame,
)
2023-04-27 19:43:56 +00:00
if TNC.enable_stats:
2023-02-06 15:56:50 +00:00
self.stats.push(frame_nack_counter=self.frame_nack_counter, status="wrong_crc", duration=duration)
2023-04-27 19:43:56 +00:00
self.log.info("[TNC] ARQ | RX | Sending NACK", finished=ARQ.arq_seconds_until_finish,
bytesperminute=ARQ.bytes_per_minute)
self.send_burst_nack_frame(snr)
2022-05-09 00:41:49 +00:00
# Update arq_session timestamp
self.arq_session_last_received = int(time.time())
2022-05-09 00:41:49 +00:00
# Finally cleanup our buffers and states,
self.arq_cleanup()
2022-05-09 00:41:49 +00:00
2023-02-09 17:15:00 +00:00
def arq_extract_statistics_from_data_frame(self, bof_position, eof_position):
2023-04-27 19:43:56 +00:00
payload = ARQ.rx_frame_buffer[
2023-02-09 17:15:00 +00:00
bof_position + len(self.data_frame_bof): eof_position
]
frame_length = int.from_bytes(payload[4:8], "big") # 4:8 4bytes
2023-04-27 19:43:56 +00:00
ARQ.total_bytes = frame_length
2023-02-09 17:15:00 +00:00
compression_factor = int.from_bytes(payload[8:9], "big") # 4:8 4bytes
# limit to max value of 255
compression_factor = np.clip(compression_factor, 0, 255)
2023-04-27 19:43:56 +00:00
ARQ.arq_compression_factor = compression_factor / 10
2023-02-09 17:15:00 +00:00
self.calculate_transfer_rate_rx(
2023-04-27 19:43:56 +00:00
self.rx_start_of_transmission, len(ARQ.rx_frame_buffer)
2023-02-09 17:15:00 +00:00
)
def check_if_mode_fits_to_busy_slot(self):
"""
Check if actual mode is fitting into given busy state
Returns:
"""
mode_name = FREEDV_MODE(self.mode_list[self.speed_level]).name
mode_slots = FREEDV_MODE_USED_SLOTS[mode_name].value
2023-04-27 19:43:56 +00:00
if mode_slots in [ModemParam.channel_busy_slot]:
self.log.warning(
"[TNC] busy slot detection",
2023-04-27 19:43:56 +00:00
slots=ModemParam.channel_busy_slot,
mode_slots=mode_slots,
)
return False
else:
return True
def arq_calculate_speed_level(self, snr):
current_speed_level = self.speed_level
2023-02-09 17:15:00 +00:00
self.frame_received_counter += 1
# try increasing speed level only if we had two successful decodes
if self.frame_received_counter >= 2:
self.frame_received_counter = 0
# make sure new speed level isn't higher than available modes
new_speed_level = min(self.speed_level + 1, len(self.mode_list) - 1)
# check if actual snr is higher than minimum snr for next mode
2023-04-27 19:43:56 +00:00
if ModemParam.snr >= self.snr_list[new_speed_level]:
2023-02-09 17:15:00 +00:00
self.speed_level = new_speed_level
2023-04-23 13:59:30 +00:00
2023-02-09 17:15:00 +00:00
else:
self.log.info("[TNC] ARQ | increasing speed level not possible because of SNR limit",
2023-04-27 19:43:56 +00:00
given_snr=ModemParam.snr,
2023-02-09 17:15:00 +00:00
needed_snr=self.snr_list[new_speed_level]
)
2023-04-23 13:59:30 +00:00
# calculate if speed level fits to busy condition
if not self.check_if_mode_fits_to_busy_slot():
self.speed_level = current_speed_level
2023-04-23 13:59:30 +00:00
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
2023-02-09 17:15:00 +00:00
# Update modes we are listening to
self.set_listening_modes(False, True, self.mode_list[self.speed_level])
def arq_process_received_data_frame(self, data_frame, snr, signed):
"""
2023-02-09 17:06:00 +00:00
"""
# transmittion duration
duration = time.time() - self.rx_start_of_transmission
self.calculate_transfer_rate_rx(
2023-04-27 19:43:56 +00:00
self.rx_start_of_transmission, len(ARQ.rx_frame_buffer)
)
self.log.info("[TNC] ARQ | RX | DATA FRAME SUCCESSFULLY RECEIVED", nacks=self.frame_nack_counter,
bytesperminute=ARQ.bytes_per_minute, total_bytes=ARQ.total_bytes, duration=duration, signed=signed)
2023-02-09 17:06:00 +00:00
# Decompress the data frame
data_frame_decompressed = lzma.decompress(data_frame)
2023-04-27 19:43:56 +00:00
ARQ.arq_compression_factor = len(data_frame_decompressed) / len(
data_frame
)
data_frame = data_frame_decompressed
2023-02-09 17:06:00 +00:00
self.transmission_uuid = str(uuid.uuid4())
timestamp = int(time.time())
2023-02-09 17:06:00 +00:00
# Re-code data_frame in base64, UTF-8 for JSON UI communication.
base64_data = base64.b64encode(data_frame).decode("UTF-8")
2023-02-09 17:06:00 +00:00
# check if RX_BUFFER isn't full
if not RX_BUFFER.full():
# make sure we have always the correct buffer size
2023-04-27 19:43:56 +00:00
RX_BUFFER.maxsize = int(ARQ.rx_buffer_size)
else:
# if full, free space by getting an item
2023-02-09 17:06:00 +00:00
self.log.info(
"[TNC] ARQ | RX | RX_BUFFER FULL - dropping old data",
buffer_size=RX_BUFFER.qsize(),
2023-04-27 19:43:56 +00:00
maxsize=int(ARQ.rx_buffer_size)
2023-02-09 17:06:00 +00:00
)
RX_BUFFER.get()
# add item to RX_BUFFER
self.log.info(
"[TNC] ARQ | RX | saving data to rx buffer",
buffer_size=RX_BUFFER.qsize() + 1,
maxsize=RX_BUFFER.maxsize
)
try:
RX_BUFFER.put(
[
self.transmission_uuid,
timestamp,
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
base64_data,
]
)
except Exception as e:
# File "/usr/lib/python3.7/queue.py", line 133, in put
# if self.maxsize > 0
# TypeError: '>' not supported between instances of 'str' and 'int'
#
# Occurs on Raspberry Pi and Python 3.7
self.log.error(
"[TNC] ARQ | RX | error occurred when saving data!",
e=e,
uuid=self.transmission_uuid,
timestamp=timestamp,
2023-04-27 19:43:56 +00:00
dxcall=Station.dxcallsign,
dxgrid=Station.dxgrid,
data=base64_data
)
2023-04-27 19:43:56 +00:00
if ARQ.arq_save_to_folder:
2023-02-09 17:06:00 +00:00
try:
self.save_data_to_folder(
self.transmission_uuid,
timestamp,
self.mycallsign,
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
data_frame
2023-02-09 17:06:00 +00:00
)
except Exception as e:
self.log.error(
"[TNC] ARQ | RX | can't save file to folder",
2023-02-09 17:06:00 +00:00
e=e,
uuid=self.transmission_uuid,
timestamp=timestamp,
2023-04-27 19:43:56 +00:00
dxcall=Station.dxcallsign,
dxgrid=Station.dxgrid,
2023-02-09 17:06:00 +00:00
data=base64_data
)
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="received",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
timestamp=timestamp,
finished=0,
mycallsign=str(self.mycallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
dxcallsign=str(Station.dxcallsign, "UTF-8"),
dxgrid=str(Station.dxgrid, "UTF-8"),
data=base64_data,
irs=helpers.bool_to_string(self.is_IRS),
signed=signed
)
2023-02-09 17:06:00 +00:00
2023-04-27 19:43:56 +00:00
if TNC.enable_stats:
duration = time.time() - self.rx_start_of_transmission
self.stats.push(frame_nack_counter=self.frame_nack_counter, status="received", duration=duration)
2023-02-09 17:06:00 +00:00
self.log.info(
"[TNC] ARQ | RX | SENDING DATA FRAME ACK")
2023-02-09 17:06:00 +00:00
self.send_data_ack_frame(snr)
# Update statistics AFTER the frame ACK is sent
self.calculate_transfer_rate_rx(
2023-04-27 19:43:56 +00:00
self.rx_start_of_transmission, len(ARQ.rx_frame_buffer)
)
2023-02-09 17:06:00 +00:00
self.log.info(
"[TNC] | RX | DATACHANNEL ["
+ str(self.mycallsign, "UTF-8")
+ "]<< >>["
2023-04-27 19:43:56 +00:00
+ str(Station.dxcallsign, "UTF-8")
+ "]",
snr=snr,
)
2023-02-09 19:32:21 +00:00
2023-08-16 07:58:49 +00:00
def arq_transmit(self, data_out: bytes, hmac_salt: bytes):
"""
Transmit ARQ frame
Args:
2022-05-09 00:41:49 +00:00
data_out:bytes:
2023-05-08 13:30:54 +00:00
"""
2023-04-25 14:32:36 +00:00
2022-10-10 07:00:45 +00:00
# set signalling modes we want to listen to
# we are in an ongoing arq transmission, so we don't need sig0 actually
modem.RECEIVE_SIG0 = False
modem.RECEIVE_SIG1 = True
2022-05-19 20:15:24 +00:00
self.tx_n_retry_of_burst = 0 # retries we already sent data
# Maximum number of retries to send before declaring a frame is lost
2023-02-09 17:45:09 +00:00
# save len of data_out to TOTAL_BYTES for our statistics
2023-04-27 19:43:56 +00:00
ARQ.total_bytes = len(data_out)
2023-02-09 17:27:06 +00:00
self.arq_file_transfer = True
frame_total_size = len(data_out).to_bytes(4, byteorder="big")
2022-05-09 00:41:49 +00:00
# Compress data frame
data_frame_compressed = lzma.compress(data_out)
compression_factor = len(data_out) / len(data_frame_compressed)
2023-04-27 19:43:56 +00:00
ARQ.arq_compression_factor = np.clip(compression_factor, 0, 255)
compression_factor = bytes([int(ARQ.arq_compression_factor * 10)])
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
arq="transmission",
status="transmitting",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
finished=ARQ.arq_seconds_until_finish,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
)
2022-05-09 00:41:49 +00:00
self.log.info(
"[TNC] | TX | DATACHANNEL",
2023-04-27 19:43:56 +00:00
Bytes=ARQ.total_bytes,
)
2022-05-09 00:41:49 +00:00
data_out = data_frame_compressed
# Reset data transfer statistics
tx_start_of_transmission = time.time()
self.calculate_transfer_rate_tx(tx_start_of_transmission, 0, len(data_out))
# check if hmac signature is available
if hmac_salt not in ['', False]:
2023-08-16 07:58:49 +00:00
print(data_out)
# create hmac digest
hmac_digest = hmac.new(hmac_salt, data_out, hashlib.sha256).digest()
# truncate to 32bit
frame_payload_crc = hmac_digest[:4]
self.log.debug("[TNC] frame payload HMAC:", crc=frame_payload_crc.hex())
else:
# Append a crc at the beginning and end of file indicators
frame_payload_crc = helpers.get_crc_32(data_out)
self.log.debug("[TNC] frame payload CRC:", crc=frame_payload_crc.hex())
# Assemble the data frame
data_out = (
2022-09-08 13:47:23 +00:00
self.data_frame_bof
+ frame_payload_crc
+ frame_total_size
+ compression_factor
+ data_out
+ self.data_frame_eof
)
2022-12-31 15:05:08 +00:00
self.log.debug("[TNC] frame raw data:", data=data_out)
# Initial bufferposition is 0
2023-04-25 14:32:36 +00:00
bufferposition = 0
bufferposition_end = 0
bufferposition_burst_start = 0
# Iterate through data_out buffer
2023-04-27 19:43:56 +00:00
while not self.data_frame_ack_received and ARQ.arq_state:
2022-10-05 18:27:38 +00:00
# we have self.tx_n_max_retries_per_burst attempts for sending a burst
for self.tx_n_retry_of_burst in range(self.tx_n_max_retries_per_burst):
2022-06-23 22:16:32 +00:00
# Bound speed level to:
# - minimum of either the speed or the length of mode list - 1
# - maximum of either the speed or zero
self.speed_level = min(self.speed_level, len(self.mode_list) - 1)
self.speed_level = max(self.speed_level, 0)
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
2022-06-23 22:16:32 +00:00
data_mode = self.mode_list[self.speed_level]
2022-05-09 00:41:49 +00:00
2022-06-23 22:16:32 +00:00
self.log.debug(
"[TNC] Speed-level:",
level=self.speed_level,
retry=self.tx_n_retry_of_burst,
mode=FREEDV_MODE(data_mode).name,
2022-06-23 22:16:32 +00:00
)
2022-05-09 00:41:49 +00:00
# Payload information
2022-05-19 20:15:24 +00:00
payload_per_frame = modem.get_bytes_per_frame(data_mode) - 2
2023-04-25 19:32:39 +00:00
self.log.info("[TNC] early buffer info",
bufferposition=bufferposition,
bufferposition_end=bufferposition_end,
bufferposition_burst_start=bufferposition_burst_start
2023-04-25 19:32:39 +00:00
)
# check for maximum frames per burst for remaining data
n_frames_per_burst = 1
2023-04-25 14:32:36 +00:00
if self.max_n_frames_per_burst > 1:
while (payload_per_frame * n_frames_per_burst) % len(data_out[bufferposition_burst_start:]) == (
payload_per_frame * n_frames_per_burst):
2023-04-25 19:32:39 +00:00
threading.Event().wait(0.01)
2023-05-06 14:10:31 +00:00
#print((payload_per_frame * n_frames_per_burst) % len(data_out))
n_frames_per_burst += 1
2023-04-25 14:32:36 +00:00
if n_frames_per_burst == self.max_n_frames_per_burst:
break
else:
n_frames_per_burst = 1
self.log.info("[TNC] calculated frames_per_burst:", n=n_frames_per_burst)
tempbuffer = []
self.rpt_request_buffer = []
2023-04-25 14:32:36 +00:00
# Append data frames with n_frames_per_burst to tempbuffer
for n_frame in range(n_frames_per_burst):
arqheader = bytearray()
arqheader[:1] = bytes([FR_TYPE.BURST_01.value + n_frame])
#####arqheader[:1] = bytes([FR_TYPE.BURST_01.value])
arqheader[1:2] = bytes([n_frames_per_burst])
arqheader[2:3] = self.session_id
# only check for buffer position if at least one NACK received
self.log.info("[TNC] ----- data buffer position:", iss_buffer_pos=bufferposition,
irs_bufferposition=self.irs_buffer_position)
if self.frame_nack_counter > 0 and self.irs_buffer_position != bufferposition:
self.log.error("[TNC] ----- data buffer offset:", iss_buffer_pos=bufferposition,
irs_bufferposition=self.irs_buffer_position)
# only adjust buffer position for experimental versions
2023-04-27 19:43:56 +00:00
if 'exp' in TNC.version:
self.log.warning("[TNC] ----- data adjustment disabled!")
# bufferposition = self.irs_buffer_position
bufferposition_end = bufferposition + payload_per_frame - len(arqheader)
# Normal condition
if bufferposition_end <= len(data_out):
frame = data_out[bufferposition:bufferposition_end]
frame = arqheader + frame
# Pad the last bytes of a frame
else:
extended_data_out = data_out[bufferposition:]
extended_data_out += bytes([0]) * (
payload_per_frame - len(extended_data_out) - len(arqheader)
)
frame = arqheader + extended_data_out
######tempbuffer = frame # [frame]
tempbuffer.append(frame)
# add data to our repeat request buffer for easy access if we received a request
self.rpt_request_buffer.append(frame)
# set new buffer position
bufferposition = bufferposition_end
self.log.debug("[TNC] tempbuffer:", tempbuffer=tempbuffer)
self.log.info(
"[TNC] ARQ | TX | FRAMES",
mode=FREEDV_MODE(data_mode).name,
fpb=n_frames_per_burst,
retry=self.tx_n_retry_of_burst,
)
2022-05-09 00:41:49 +00:00
self.enqueue_frame_for_tx(tempbuffer, c2_mode=data_mode)
2022-05-09 00:41:49 +00:00
# After transmission finished, wait for an ACK or RPT frame
2023-02-09 17:27:06 +00:00
while (
2023-04-27 19:43:56 +00:00
ARQ.arq_state
and not self.burst_ack
and not self.burst_nack
and not self.rpt_request_received
and not self.data_frame_ack_received
):
threading.Event().wait(0.01)
2022-05-09 00:41:49 +00:00
# Once we receive a burst ack, reset its state and break the RETRIES loop
2022-01-07 11:55:03 +00:00
if self.burst_ack:
2022-05-19 20:15:24 +00:00
self.burst_ack = False # reset ack state
self.tx_n_retry_of_burst = 0 # reset retries
self.log.debug(
"[TNC] arq_transmit: Received BURST ACK. Sending next chunk."
, irs_snr=self.burst_ack_snr)
2023-04-25 14:32:36 +00:00
# update temp bufferposition for n frames per burst early calculation
bufferposition_burst_start = bufferposition_end
2022-05-19 20:15:24 +00:00
break # break retry loop
if self.data_frame_ack_received:
self.log.debug(
2023-02-09 19:32:21 +00:00
"[TNC] arq_transmit: Received FRAME ACK. Braking retry loop."
)
2022-05-19 20:15:24 +00:00
break # break retry loop
2022-05-09 00:41:49 +00:00
if self.burst_nack:
self.tx_n_retry_of_burst += 1
self.log.warning(
"[TNC] arq_transmit: Received BURST NACK. Resending data",
bufferposition_burst_start=bufferposition_burst_start,
bufferposition=bufferposition
)
bufferposition = bufferposition_burst_start
self.burst_nack = False # reset nack state
# We need this part for leaving the repeat loop
2023-04-27 19:43:56 +00:00
# ARQ.arq_state == "DATA" --> when stopping transmission manually
if not ARQ.arq_state:
self.log.debug(
"[TNC] arq_transmit: ARQ State changed to FALSE. Breaking retry loop."
)
break
2022-05-09 00:41:49 +00:00
self.calculate_transfer_rate_tx(
tx_start_of_transmission, bufferposition_end, len(data_out)
)
# NEXT ATTEMPT
self.log.debug(
"[TNC] ATTEMPT:",
retry=self.tx_n_retry_of_burst,
2022-10-05 18:27:38 +00:00
maxretries=self.tx_n_max_retries_per_burst,
2023-04-27 19:43:56 +00:00
overflows=AudioParam.buffer_overflow_counter,
)
2022-05-09 00:41:49 +00:00
# update buffer position
bufferposition = bufferposition_end
2021-03-12 13:14:36 +00:00
# update stats
self.calculate_transfer_rate_tx(
tx_start_of_transmission, bufferposition_end, len(data_out)
)
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
arq="transmission",
status="transmitting",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
finished=ARQ.arq_seconds_until_finish,
2022-11-11 11:57:14 +00:00
irs_snr=self.burst_ack_snr,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
)
2022-05-09 00:41:49 +00:00
2022-11-02 14:29:22 +00:00
# Stay in the while loop until we receive a data_frame_ack. Otherwise,
# the loop exits after sending the last frame only once and doesn't
# wait for an acknowledgement.
if self.data_frame_ack_received and bufferposition > len(data_out):
self.log.debug("[TNC] arq_tx: Last fragment sent and acknowledged.")
break
2023-02-09 17:27:06 +00:00
# GOING TO NEXT ITERATION
2022-05-09 00:41:49 +00:00
if self.data_frame_ack_received:
2023-02-09 19:32:21 +00:00
self.arq_transmit_success()
else:
self.arq_transmit_failed()
2022-05-09 00:41:49 +00:00
2023-02-09 19:32:21 +00:00
if TESTMODE:
# Quit after transmission
self.log.debug("[TNC] TESTMODE: arq_transmit exiting.")
sys.exit(0)
2023-02-09 19:32:21 +00:00
def arq_transmit_success(self):
"""
will be called if we successfully transmitted all of queued data
2023-02-03 15:48:37 +00:00
2023-02-09 19:32:21 +00:00
"""
# we need to wait until sending "transmitted" state
# gui database is too slow for handling this within 0.001 seconds
# so let's sleep a little
threading.Event().wait(0.2)
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="transmitted",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
finished=ARQ.arq_seconds_until_finish,
2023-02-09 19:32:21 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
irs=helpers.bool_to_string(self.is_IRS)
)
2021-03-12 13:14:36 +00:00
2023-02-09 19:32:21 +00:00
self.log.info(
"[TNC] ARQ | TX | DATA TRANSMITTED!",
2023-04-27 19:43:56 +00:00
BytesPerMinute=ARQ.bytes_per_minute,
total_bytes=ARQ.total_bytes,
BitsPerSecond=ARQ.arq_bits_per_second,
overflows=AudioParam.buffer_overflow_counter,
2022-05-09 00:41:49 +00:00
2023-02-09 19:32:21 +00:00
)
2023-02-02 17:03:22 +00:00
2023-02-09 19:32:21 +00:00
# finally do an arq cleanup
self.arq_cleanup()
2021-03-17 10:22:06 +00:00
2023-02-09 19:32:21 +00:00
def arq_transmit_failed(self):
"""
will be called if we not successfully transmitted all of queued data
"""
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="failed",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
2023-02-09 19:32:21 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
irs=helpers.bool_to_string(self.is_IRS)
)
self.log.info(
"[TNC] ARQ | TX | TRANSMISSION FAILED OR TIME OUT!",
2023-04-27 19:43:56 +00:00
overflows=AudioParam.buffer_overflow_counter,
2023-02-09 19:32:21 +00:00
)
self.stop_transmission()
def burst_ack_nack_received(self, data_in: bytes) -> None:
"""
2022-11-02 14:29:22 +00:00
Received an ACK/NACK for a transmitted frame, keep track and
make adjustments to speed level if needed.
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
Returns:
"""
# Process data only if we are in ARQ and BUSY state
2023-04-27 19:43:56 +00:00
if ARQ.arq_state:
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
self.dxcallsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
frametype = int.from_bytes(bytes(data_in[:1]), "big")
if frametype == FR_TYPE.BURST_ACK.value:
# Increase speed level if we received a burst ack
# self.speed_level = min(self.speed_level + 1, len(self.mode_list) - 1)
2022-07-02 20:20:51 +00:00
# Force data retry loops of TX TNC to stop and continue with next frame
self.burst_ack = True
# Reset burst nack counter
self.burst_nack_counter = 0
# Reset n retries per burst counter
self.n_retries_per_burst = 0
self.irs_buffer_position = int.from_bytes(data_in[4:8], "big")
2023-02-10 09:41:26 +00:00
self.burst_ack_snr = helpers.snr_from_bytes(data_in[2:3])
else:
2023-02-21 19:47:22 +00:00
2022-07-02 20:20:51 +00:00
# Decrease speed level if we received a burst nack
# self.speed_level = max(self.speed_level - 1, 0)
# Set flag to retry frame again.
2022-07-02 20:20:51 +00:00
self.burst_nack = True
# Increment burst nack counter
self.burst_nack_counter += 1
2023-02-10 09:41:26 +00:00
self.burst_ack_snr = 'NaN'
self.irs_buffer_position = int.from_bytes(data_in[5:9], "big")
2023-02-21 19:47:22 +00:00
self.log.warning(
"[TNC] ARQ | TX | Burst NACK received",
burst_nack_counter=self.burst_nack_counter,
irs_buffer_position=self.irs_buffer_position,
)
# Update data_channel timestamp
self.data_channel_last_received = int(time.time())
2022-11-09 19:47:46 +00:00
# self.burst_ack_snr = int.from_bytes(bytes(data_in[2:3]), "big")
self.burst_ack_snr = helpers.snr_from_bytes(data_in[2:3])
# self.log.info("SNR ON IRS", snr=self.burst_ack_snr)
2022-11-09 19:47:46 +00:00
2022-10-10 07:00:45 +00:00
self.speed_level = int.from_bytes(bytes(data_in[3:4]), "big")
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
def frame_ack_received(
2022-09-08 13:47:23 +00:00
self, data_in: bytes # pylint: disable=unused-argument
) -> None:
"""Received an ACK for a transmitted frame"""
# Process data only if we are in ARQ and BUSY state
2023-04-27 19:43:56 +00:00
if ARQ.arq_state:
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
# Force data loops of TNC to stop and continue with next frame
self.data_frame_ack_received = True
# Update arq_session and data_channel timestamp
self.data_channel_last_received = int(time.time())
self.arq_session_last_received = int(time.time())
def frame_nack_received(
2022-09-08 13:47:23 +00:00
self, data_in: bytes # pylint: disable=unused-argument
) -> None:
"""
Received a NACK for a transmitted frame
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
2022-12-11 12:07:56 +00:00
self.log.warning("[TNC] ARQ FRAME NACK RECEIVED - cleanup!",
arq="transmission",
status="failed",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
2022-12-11 12:07:56 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-12-11 12:07:56 +00:00
)
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
arq="transmission",
status="failed",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
)
# Update data_channel timestamp
self.arq_session_last_received = int(time.time())
2022-05-09 00:41:49 +00:00
self.arq_cleanup()
2022-05-19 20:15:24 +00:00
def burst_rpt_received(self, data_in: bytes):
"""
Repeat request frame received for transmitted frame
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
# Only process data if we are in ARQ and BUSY state
2023-04-27 19:43:56 +00:00
if not ARQ.arq_state or TNC.tnc_state != "BUSY":
2023-02-09 17:24:37 +00:00
return
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
2023-02-09 17:24:37 +00:00
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
2023-02-09 17:24:37 +00:00
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
2023-02-09 17:24:37 +00:00
)
2022-05-09 00:41:49 +00:00
self.log.info("[TNC] ARQ REPEAT RECEIVED")
# self.rpt_request_received = True
2023-02-09 17:24:37 +00:00
# Update data_channel timestamp
self.data_channel_last_received = int(time.time())
# self.rpt_request_buffer = []
missing_area = bytes(data_in[2:12]) # 1:9
missing_area = missing_area.strip(b"\x00")
print(missing_area)
print(self.rpt_request_buffer)
2023-04-25 19:08:48 +00:00
tempbuffer_rptframes = []
for i in range(len(missing_area)):
print(missing_area[i])
missing_frames_buffer_position = missing_area[i] - 1
2023-04-25 19:08:48 +00:00
tempbuffer_rptframes.append(self.rpt_request_buffer[missing_frames_buffer_position])
2021-03-12 13:14:36 +00:00
self.log.info("[TNC] SENDING REPEAT....")
data_mode = self.mode_list[self.speed_level]
2023-04-25 19:08:48 +00:00
self.enqueue_frame_for_tx(tempbuffer_rptframes, c2_mode=data_mode)
2021-02-28 14:24:14 +00:00
# for i in range(0, 6, 2):
# if not missing_area[i: i + 2].endswith(b"\x00\x00"):
# self.rpt_request_buffer.insert(0, missing_area[i: i + 2])
2021-03-12 13:14:36 +00:00
############################################################################################################
# ARQ SESSION HANDLER
############################################################################################################
def arq_session_handler(self, mycallsign, dxcallsign, attempts) -> bool:
"""
2023-04-27 19:43:56 +00:00
Create a session with `Station.dxcallsign` and wait until the session is open.
Returns:
True if the session was opened successfully
False if the session open request failed
"""
# override connection attempts
self.session_connect_max_retries = attempts
self.mycallsign = mycallsign
self.dxcallsign = dxcallsign
2023-04-27 19:43:56 +00:00
Station.dxcallsign = self.dxcallsign
Station.dxcallsign_crc = helpers.get_crc_24(self.dxcallsign)
2022-05-23 07:37:24 +00:00
# TODO: we need to check this, maybe placing it to class init
self.datachannel_timeout = False
self.log.info(
"[TNC] SESSION ["
+ str(self.mycallsign, "UTF-8")
+ "]>> <<["
+ str(self.dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
)
2022-05-09 00:41:49 +00:00
2022-11-17 23:03:18 +00:00
# Let's check if we have a busy channel
2023-04-27 19:43:56 +00:00
if ModemParam.channel_busy:
2022-11-17 23:03:18 +00:00
self.log.warning("[TNC] Channel busy, waiting until free...")
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="waiting",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-11-17 23:03:18 +00:00
)
# wait while timeout not reached and our busy state is busy
2022-12-30 20:29:22 +00:00
channel_busy_timeout = time.time() + 15
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
threading.Event().wait(0.01)
2022-11-17 23:03:18 +00:00
# if channel busy timeout reached stop connecting
if time.time() > channel_busy_timeout:
self.log.warning("[TNC] Channel busy, try again later...")
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "failed"
2022-11-17 23:03:18 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="failed",
reason="busy",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-11-17 23:03:18 +00:00
)
return False
self.open_session()
# wait until data channel is open
2023-04-27 19:43:56 +00:00
while not ARQ.arq_session and not self.arq_session_timeout:
threading.Event().wait(0.01)
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "connecting"
2022-06-15 08:40:54 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connecting",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-15 08:40:54 +00:00
)
2023-04-27 19:43:56 +00:00
if ARQ.arq_session and ARQ.arq_session_state == "connected":
# ARQ.arq_session_state = "connected"
2022-06-15 08:40:54 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connected",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-15 08:40:54 +00:00
)
return True
2022-11-20 17:34:45 +00:00
self.log.warning(
"[TNC] SESSION FAILED ["
+ str(self.mycallsign, "UTF-8")
+ "]>>X<<["
+ str(self.dxcallsign, "UTF-8")
2022-11-20 17:34:45 +00:00
+ "]",
2022-11-20 17:37:44 +00:00
attempts=self.session_connect_max_retries, # Adjust for 0-based for user display
2022-11-20 17:34:45 +00:00
reason="maximum connection attempts reached",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
2022-11-20 17:34:45 +00:00
)
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "failed"
2022-06-15 08:40:54 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="failed",
reason="timeout",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-15 08:40:54 +00:00
)
return False
def open_session(self) -> bool:
"""
Create and send the frame to request a connection.
Returns:
True if the session was opened successfully
False if the session open request failed
"""
self.IS_ARQ_SESSION_MASTER = True
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "connecting"
2022-10-05 17:24:50 +00:00
# create a random session id
self.session_id = np.random.bytes(1)
2022-10-05 17:24:50 +00:00
connection_frame = bytearray(self.length_sig0_frame)
connection_frame[:1] = bytes([FR_TYPE.ARQ_SESSION_OPEN.value])
2022-10-05 17:24:50 +00:00
connection_frame[1:2] = self.session_id
2023-04-27 19:43:56 +00:00
connection_frame[2:5] = Station.dxcallsign_crc
connection_frame[5:8] = Station.mycallsign_crc
2022-10-12 05:40:39 +00:00
connection_frame[8:14] = helpers.callsign_to_bytes(self.mycallsign)
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
while not ARQ.arq_session:
threading.Event().wait(0.01)
for attempt in range(self.session_connect_max_retries):
self.log.info(
"[TNC] SESSION ["
+ str(self.mycallsign, "UTF-8")
+ "]>>?<<["
+ str(self.dxcallsign, "UTF-8")
+ "]",
a=f"{str(attempt + 1)}/{str(self.session_connect_max_retries)}",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
)
2022-05-09 00:41:49 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connecting",
2022-12-01 09:09:44 +00:00
attempt=attempt + 1,
maxattempts=self.session_connect_max_retries,
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
)
self.enqueue_frame_for_tx([connection_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
# Wait for a time, looking to see if `ARQ.arq_session`
# indicates we've received a positive response from the far station.
2022-05-09 00:41:49 +00:00
timeout = time.time() + 3
while time.time() < timeout:
threading.Event().wait(0.01)
# Stop waiting if data channel is opened
2023-04-27 19:43:56 +00:00
if ARQ.arq_session:
2022-04-24 15:05:11 +00:00
return True
2022-11-17 13:55:20 +00:00
# Stop waiting and interrupt if data channel is getting closed while opening
2023-04-27 19:43:56 +00:00
if ARQ.arq_session_state == "disconnecting":
# disabled this session close as its called twice
# self.close_session()
2022-11-17 13:55:20 +00:00
return False
2022-04-24 15:05:11 +00:00
# Session connect timeout, send close_session frame to
2022-05-28 12:31:58 +00:00
# attempt to clean up the far-side, if it received the
2022-04-24 15:05:11 +00:00
# open_session frame and can still hear us.
2023-04-27 19:43:56 +00:00
if not ARQ.arq_session:
2022-04-24 15:05:11 +00:00
self.close_session()
return False
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
# Given the while condition, it will only exit when `ARQ.arq_session` is True
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connected",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
)
return True
def received_session_opener(self, data_in: bytes) -> None:
"""
Received a session open request packet.
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
2022-12-04 15:56:12 +00:00
# if we don't want to respond to calls, return False
2023-04-27 19:43:56 +00:00
if not TNC.respond_to_call:
2022-12-04 15:56:12 +00:00
return False
# ignore channel opener if already in ARQ STATE
# use case: Station A is connecting to Station B while
# Station B already tries connecting to Station A.
# For avoiding ignoring repeated connect request in case of packet loss
# we are only ignoring packets in case we are ISS
2023-04-27 19:43:56 +00:00
if ARQ.arq_session and self.IS_ARQ_SESSION_MASTER:
return False
self.IS_ARQ_SESSION_MASTER = False
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "connecting"
# Update arq_session timestamp
self.arq_session_last_received = int(time.time())
2022-10-12 05:40:39 +00:00
self.session_id = bytes(data_in[1:2])
2023-04-27 19:43:56 +00:00
Station.dxcallsign_crc = bytes(data_in[5:8])
2022-12-03 12:59:05 +00:00
self.dxcallsign = helpers.bytes_to_callsign(bytes(data_in[8:14]))
2023-04-27 19:43:56 +00:00
Station.dxcallsign = self.dxcallsign
2022-05-09 00:41:49 +00:00
2022-12-07 15:30:59 +00:00
# check if callsign ssid override
valid, mycallsign = helpers.check_callsign(self.mycallsign, data_in[2:5])
self.mycallsign = mycallsign
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
self.log.info(
"[TNC] SESSION ["
+ str(self.mycallsign, "UTF-8")
+ "]>>|<<["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
)
2023-04-27 19:43:56 +00:00
ARQ.arq_session = True
TNC.tnc_state = "BUSY"
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connected",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
)
self.transmit_session_heartbeat()
def close_session(self) -> None:
"""Close the ARQ session"""
2023-04-27 19:43:56 +00:00
ARQ.arq_session_state = "disconnecting"
self.log.info(
"[TNC] SESSION ["
+ str(self.mycallsign, "UTF-8")
+ "]<<X>>["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
)
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="session",
status="close",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-05 17:11:09 +00:00
)
self.IS_ARQ_SESSION_MASTER = False
2023-04-27 19:43:56 +00:00
ARQ.arq_session = False
2022-05-09 00:41:49 +00:00
2022-10-06 09:47:19 +00:00
# we need to send disconnect frame before doing arq cleanup
# we would lose our session id then
self.send_disconnect_frame()
2022-10-06 09:47:19 +00:00
self.arq_cleanup()
2022-05-19 20:15:24 +00:00
def received_session_close(self, data_in: bytes):
"""
Closes the session when a close session frame is received and
the DXCALLSIGN_CRC matches the remote station participating in the session.
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
# We've arrived here from process_data which already checked that the frame
# is intended for this station.
# Close the session if the CRC matches the remote station in static.
2022-12-07 15:30:59 +00:00
_valid_crc, mycallsign = helpers.check_callsign(self.mycallsign, bytes(data_in[2:5]))
2022-10-05 18:27:38 +00:00
_valid_session = helpers.check_session_id(self.session_id, bytes(data_in[1:2]))
2023-04-27 19:43:56 +00:00
if (_valid_crc or _valid_session) and ARQ.arq_session_state not in ["disconnected"]:
ARQ.arq_session_state = "disconnected"
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
self.log.info(
"[TNC] SESSION ["
2022-12-07 15:30:59 +00:00
+ str(mycallsign, "UTF-8")
+ "]<<X>>["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
state=ARQ.arq_session_state,
)
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="session",
status="close",
2022-12-07 15:30:59 +00:00
mycallsign=str(mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-05 17:11:09 +00:00
)
self.IS_ARQ_SESSION_MASTER = False
2023-04-27 19:43:56 +00:00
ARQ.arq_session = False
self.arq_cleanup()
def transmit_session_heartbeat(self) -> None:
"""Send ARQ sesion heartbeat while connected"""
2023-04-27 19:43:56 +00:00
# ARQ.arq_session = True
# TNC.tnc_state = "BUSY"
# ARQ.arq_session_state = "connected"
connection_frame = bytearray(self.length_sig0_frame)
connection_frame[:1] = bytes([FR_TYPE.ARQ_SESSION_HB.value])
2022-10-05 17:24:50 +00:00
connection_frame[1:2] = self.session_id
2022-03-19 11:42:10 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connected",
heartbeat="transmitting",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
)
self.enqueue_frame_for_tx([connection_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
def received_session_heartbeat(self, data_in: bytes) -> None:
"""
Received an ARQ session heartbeat, record and update state accordingly.
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
2022-10-05 17:24:50 +00:00
# Accept session data if the DXCALLSIGN_CRC matches the station in static or session id.
2022-12-03 12:59:05 +00:00
_valid_crc, _ = helpers.check_callsign(self.dxcallsign, bytes(data_in[4:7]))
2022-10-05 17:24:50 +00:00
_valid_session = helpers.check_session_id(self.session_id, bytes(data_in[1:2]))
2023-04-27 19:43:56 +00:00
if _valid_crc or _valid_session and ARQ.arq_session_state in ["connected", "connecting"]:
self.log.debug("[TNC] Received session heartbeat")
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2022-12-03 12:59:05 +00:00
self.dxcallsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"SESSION-HB",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
2022-05-09 00:41:49 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="session",
status="connected",
heartbeat="received",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
)
2023-04-27 19:43:56 +00:00
ARQ.arq_session = True
ARQ.arq_session_state = "connected"
TNC.tnc_state = "BUSY"
# Update the timeout timestamps
self.arq_session_last_received = int(time.time())
self.data_channel_last_received = int(time.time())
# transmit session heartbeat only
# -> if not session master
# --> this will be triggered by heartbeat watchdog
# -> if not during a file transfer
# -> if ARQ_SESSION_STATE != disconnecting, disconnected, failed
# --> to avoid heartbeat toggle loops while disconnecting
if (
not self.IS_ARQ_SESSION_MASTER
and not self.arq_file_transfer
2023-04-27 19:43:56 +00:00
and ARQ.arq_session_state != 'disconnecting'
and ARQ.arq_session_state != 'disconnected'
and ARQ.arq_session_state != 'failed'
):
self.transmit_session_heartbeat()
##########################################################################################################
# ARQ DATA CHANNEL HANDLER
##########################################################################################################
def open_dc_and_transmit(
2022-09-08 13:47:23 +00:00
self,
data_out: bytes,
transmission_uuid: str,
mycallsign,
2022-12-04 11:22:35 +00:00
dxcallsign,
2022-11-20 11:02:53 +00:00
attempts: int,
hmac_salt: str
) -> bool:
"""
Open data channel and transmit data
Args:
2022-05-09 00:41:49 +00:00
data_out:bytes:
2022-10-05 18:27:38 +00:00
transmission_uuid:str:
mycallsign:bytes:
attempts:int: Overriding number of attempts initiating a connection
Returns:
True if the data session was opened and the data was sent
False if the data session was not opened
"""
# overwrite mycallsign in case of different SSID
self.mycallsign = mycallsign
2022-12-04 11:22:35 +00:00
self.dxcallsign = dxcallsign
2022-05-09 00:41:49 +00:00
Station.dxcallsign = dxcallsign
Station.dxcallsign_crc = helpers.get_crc_24(Station.dxcallsign)
# override session connection attempts
self.data_channel_max_retries = attempts
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "BUSY"
self.arq_file_transfer = True
2022-05-09 00:41:49 +00:00
self.transmission_uuid = transmission_uuid
2022-05-09 00:41:49 +00:00
2022-05-28 12:31:58 +00:00
# wait a moment for the case, a heartbeat is already on the way back to us
# this makes channel establishment more clean
2023-04-27 19:43:56 +00:00
if ARQ.arq_session:
2023-04-22 11:37:51 +00:00
threading.Event().wait(2.5)
2022-05-09 00:41:49 +00:00
self.datachannel_timeout = False
2022-05-09 00:41:49 +00:00
# we need to compress data for getting a compression factor.
2022-05-28 12:31:58 +00:00
# so we are compressing twice. This is not that nice and maybe there is another way
# for calculating transmission statistics
2023-04-27 19:43:56 +00:00
# ARQ.arq_compression_factor = len(data_out) / len(lzma.compress(data_out))
2022-05-09 00:41:49 +00:00
2023-05-08 13:29:18 +00:00
self.arq_open_data_channel(mycallsign)
2022-05-09 00:41:49 +00:00
# wait until data channel is open
2023-05-08 14:13:37 +00:00
while not ARQ.arq_state and not self.datachannel_timeout and TNC.tnc_state in ["BUSY"]:
threading.Event().wait(0.01)
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
if ARQ.arq_state:
self.arq_transmit(data_out, hmac_salt)
return True
2022-05-09 00:41:49 +00:00
return False
def arq_open_data_channel(
2023-05-08 11:02:48 +00:00
self, mycallsign
) -> bool:
"""
Open an ARQ data channel.
Args:
2022-10-05 18:27:38 +00:00
mycallsign:bytes:
Returns:
True if the data channel was opened successfully
False if the data channel failed to open
"""
2022-05-09 00:41:49 +00:00
self.is_IRS = False
2022-10-05 18:27:38 +00:00
# init a new random session id if we are not in an arq session
2023-04-27 19:43:56 +00:00
if not ARQ.arq_session:
self.session_id = np.random.bytes(1)
2022-10-05 17:24:50 +00:00
# Update data_channel timestamp
self.data_channel_last_received = int(time.time())
2023-04-27 19:43:56 +00:00
if TNC.low_bandwidth_mode:
frametype = bytes([FR_TYPE.ARQ_DC_OPEN_N.value])
self.log.debug("[TNC] Requesting low bandwidth mode")
else:
frametype = bytes([FR_TYPE.ARQ_DC_OPEN_W.value])
self.log.debug("[TNC] Requesting high bandwidth mode")
connection_frame = bytearray(self.length_sig0_frame)
2022-05-19 20:15:24 +00:00
connection_frame[:1] = frametype
2023-04-27 19:43:56 +00:00
connection_frame[1:4] = Station.dxcallsign_crc
connection_frame[4:7] = Station.mycallsign_crc
2022-05-19 20:15:24 +00:00
connection_frame[7:13] = helpers.callsign_to_bytes(mycallsign)
2022-10-05 17:24:50 +00:00
connection_frame[13:14] = self.session_id
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
while not ARQ.arq_state:
threading.Event().wait(0.01)
for attempt in range(self.data_channel_max_retries):
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
status="opening",
2022-12-04 11:22:35 +00:00
mycallsign=str(mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
self.log.info(
"[TNC] ARQ | DATA | TX | ["
+ str(mycallsign, "UTF-8")
+ "]>> <<["
2022-12-04 11:22:35 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2022-09-08 13:47:23 +00:00
attempt=f"{str(attempt + 1)}/{str(self.data_channel_max_retries)}",
)
# Let's check if we have a busy channel and if we are not in a running arq session.
2023-04-27 19:43:56 +00:00
if ModemParam.channel_busy and not ARQ.arq_state:
self.log.warning("[TNC] Channel busy, waiting until free...")
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="waiting",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
)
# wait while timeout not reached and our busy state is busy
2023-04-22 13:28:50 +00:00
channel_busy_timeout = time.time() + 5
2023-04-27 19:43:56 +00:00
while ModemParam.channel_busy and time.time() < channel_busy_timeout and not self.check_if_mode_fits_to_busy_slot():
threading.Event().wait(0.01)
self.enqueue_frame_for_tx([connection_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
timeout = time.time() + self.duration_sig1_frame * 3 + (ModemParam.tx_delay / 1000 * 2)
2022-05-09 00:41:49 +00:00
while time.time() < timeout:
threading.Event().wait(0.01)
# Stop waiting if data channel is opened
2023-04-27 19:43:56 +00:00
if ARQ.arq_state:
return True
2023-04-27 19:43:56 +00:00
if TNC.tnc_state in ["IDLE"]:
return False
# `data_channel_max_retries` attempts have been sent. Aborting attempt & cleaning up
2022-06-05 17:11:09 +00:00
self.log.debug(
"[TNC] arq_open_data_channel:", transmission_uuid=self.transmission_uuid
)
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
arq="transmission",
status="failed",
2022-06-06 18:44:35 +00:00
reason="unknown",
uuid=self.transmission_uuid,
2023-04-27 19:43:56 +00:00
percent=ARQ.arq_transmission_percent,
bytesperminute=ARQ.bytes_per_minute,
compression=ARQ.arq_compression_factor,
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
)
self.log.warning(
"[TNC] ARQ | TX | DATA ["
+ str(mycallsign, "UTF-8")
+ "]>>X<<["
2022-12-04 11:22:35 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]"
)
self.datachannel_timeout = True
2022-10-06 09:47:19 +00:00
2022-11-02 14:29:22 +00:00
# Attempt to clean up the far-side, if it received the
# open_session frame and can still hear us.
self.close_session()
2022-10-06 09:47:19 +00:00
return False
2022-11-02 14:29:22 +00:00
# Shouldn't get here...
return True
2022-05-19 20:15:24 +00:00
def arq_received_data_channel_opener(self, data_in: bytes):
"""
2022-11-02 14:29:22 +00:00
Received request to open data channel frame
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
# We've arrived here from process_data which already checked that the frame
# is intended for this station.
2022-12-04 15:56:12 +00:00
# stop processing if we don't want to respond to a call when not in a arq session
2023-04-27 19:43:56 +00:00
if not TNC.respond_to_call and not ARQ.arq_session:
2022-12-04 15:56:12 +00:00
return False
# stop processing if not in arq session, but tnc state is busy and we have a different session id
# use-case we get a connection request while connecting to another station
2023-04-27 19:43:56 +00:00
if not ARQ.arq_session and TNC.tnc_state in ["BUSY"] and data_in[13:14] != self.session_id:
return False
self.arq_file_transfer = True
2022-12-19 14:42:48 +00:00
# check if callsign ssid override
_, self.mycallsign = helpers.check_callsign(self.mycallsign, data_in[1:4])
# ignore channel opener if already in ARQ STATE
# use case: Station A is connecting to Station B while
# Station B already tries connecting to Station A.
# For avoiding ignoring repeated connect request in case of packet loss
# we are only ignoring packets in case we are ISS
2023-04-27 19:43:56 +00:00
if ARQ.arq_state and not self.is_IRS:
return False
2023-01-29 11:20:30 +00:00
self.is_IRS = True
2023-04-27 19:43:56 +00:00
Station.dxcallsign_crc = bytes(data_in[4:7])
self.dxcallsign = helpers.bytes_to_callsign(bytes(data_in[7:13]))
2023-04-27 19:43:56 +00:00
Station.dxcallsign = self.dxcallsign
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
status="opening",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
# n_frames_per_burst is currently unused
# n_frames_per_burst = int.from_bytes(bytes(data_in[13:14]), "big")
frametype = int.from_bytes(bytes(data_in[:1]), "big")
2022-05-23 01:11:40 +00:00
# check if we received low bandwidth mode
# possible channel constellations
# ISS(w) <-> IRS(w)
# ISS(w) <-> IRS(n)
# ISS(n) <-> IRS(w)
# ISS(n) <-> IRS(n)
2023-04-27 19:43:56 +00:00
if frametype == FR_TYPE.ARQ_DC_OPEN_W.value and not TNC.low_bandwidth_mode:
# ISS(w) <-> IRS(w)
constellation = "ISS(w) <-> IRS(w)"
self.received_LOW_BANDWIDTH_MODE = False
2022-05-09 00:41:49 +00:00
self.mode_list = self.mode_list_high_bw
self.time_list = self.time_list_high_bw
self.snr_list = self.snr_list_high_bw
2023-02-09 17:24:37 +00:00
elif frametype == FR_TYPE.ARQ_DC_OPEN_W.value:
# ISS(w) <-> IRS(n)
constellation = "ISS(w) <-> IRS(n)"
self.received_LOW_BANDWIDTH_MODE = False
self.mode_list = self.mode_list_low_bw
self.time_list = self.time_list_low_bw
self.snr_list = self.snr_list_low_bw
2023-04-27 19:43:56 +00:00
elif frametype == FR_TYPE.ARQ_DC_OPEN_N.value and not TNC.low_bandwidth_mode:
# ISS(n) <-> IRS(w)
constellation = "ISS(n) <-> IRS(w)"
self.received_LOW_BANDWIDTH_MODE = True
self.mode_list = self.mode_list_low_bw
self.time_list = self.time_list_low_bw
self.snr_list = self.snr_list_low_bw
2023-02-09 17:24:37 +00:00
elif frametype == FR_TYPE.ARQ_DC_OPEN_N.value:
# ISS(n) <-> IRS(n)
constellation = "ISS(n) <-> IRS(n)"
self.received_LOW_BANDWIDTH_MODE = True
self.mode_list = self.mode_list_low_bw
self.time_list = self.time_list_low_bw
self.snr_list = self.snr_list_low_bw
else:
constellation = "not matched"
self.received_LOW_BANDWIDTH_MODE = True
2022-05-09 00:41:49 +00:00
self.mode_list = self.mode_list_low_bw
self.time_list = self.time_list_low_bw
self.snr_list = self.snr_list_low_bw
2022-09-08 13:36:53 +00:00
# get mode which fits to given SNR
2023-02-09 17:45:09 +00:00
# initially set speed_level 0 in case of bad SNR and no matching mode
2022-09-08 13:47:23 +00:00
self.speed_level = 0
2023-04-23 13:59:30 +00:00
# TODO: MOVE THIS TO arq_calculate_speed_level()
# calculate speed level in correlation to latest known SNR
for i in range(len(self.mode_list)):
2023-04-27 19:43:56 +00:00
if ModemParam.snr >= self.snr_list[i]:
self.speed_level = i
2022-09-08 13:47:23 +00:00
2023-04-23 13:59:30 +00:00
# calculate if speed level fits to busy condition
mode_name = codec2.FREEDV_MODE(self.mode_list[self.speed_level]).name
mode_slots = codec2.FREEDV_MODE_USED_SLOTS[mode_name].value
2023-04-27 19:43:56 +00:00
if mode_slots in [ModemParam.channel_busy_slot]:
2023-04-23 13:59:30 +00:00
self.speed_level = 0
self.log.warning(
"[TNC] busy slot detection",
2023-04-27 19:43:56 +00:00
slots=ModemParam.channel_busy_slot,
2023-04-23 13:59:30 +00:00
mode_slots=mode_slots,
)
2022-09-08 13:47:23 +00:00
self.log.debug(
"[TNC] calculated speed level",
speed_level=self.speed_level,
2023-04-27 19:43:56 +00:00
given_snr=ModemParam.snr,
2022-09-08 13:47:23 +00:00
min_snr=self.snr_list[self.speed_level],
)
2022-05-09 00:41:49 +00:00
# Update modes we are listening to
2022-10-10 07:00:45 +00:00
self.set_listening_modes(True, True, self.mode_list[self.speed_level])
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
2022-05-09 00:41:49 +00:00
2022-10-05 17:24:50 +00:00
self.session_id = data_in[13:14]
2022-12-19 14:42:48 +00:00
# check again if callsign ssid override
_, self.mycallsign = helpers.check_callsign(self.mycallsign, data_in[1:4])
2022-05-09 00:41:49 +00:00
self.log.info(
"[TNC] ARQ | DATA | RX | ["
2022-12-19 14:42:48 +00:00
+ str(self.mycallsign, "UTF-8")
+ "]>> <<["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
channel_constellation=constellation,
)
2022-05-09 00:41:49 +00:00
2023-01-06 13:41:15 +00:00
# Reset data_channel/burst timestamps
self.data_channel_last_received = int(time.time())
self.burst_last_received = int(time.time() + 10) # we might need some more time so lets increase this
2023-01-06 13:41:15 +00:00
# Set ARQ State AFTER resetting timeouts
# this avoids timeouts starting too early
2023-04-27 19:43:56 +00:00
ARQ.arq_state = True
TNC.tnc_state = "BUSY"
self.reset_statistics()
2021-03-16 14:21:58 +00:00
# Select the frame type based on the current TNC mode
2023-04-27 19:43:56 +00:00
if TNC.low_bandwidth_mode or self.received_LOW_BANDWIDTH_MODE:
frametype = bytes([FR_TYPE.ARQ_DC_OPEN_ACK_N.value])
self.log.debug("[TNC] Responding with low bandwidth mode")
else:
frametype = bytes([FR_TYPE.ARQ_DC_OPEN_ACK_W.value])
self.log.debug("[TNC] Responding with high bandwidth mode")
2022-05-09 00:41:49 +00:00
connection_frame = bytearray(self.length_sig0_frame)
2022-05-19 20:15:24 +00:00
connection_frame[:1] = frametype
2022-10-05 17:24:50 +00:00
connection_frame[1:2] = self.session_id
connection_frame[8:9] = bytes([self.speed_level])
2023-04-27 19:43:56 +00:00
connection_frame[13:14] = bytes([ARQ.arq_protocol_version])
2022-01-07 11:55:03 +00:00
self.enqueue_frame_for_tx([connection_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
2022-12-05 07:23:18 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
arq="transmission",
status="opened",
2022-12-19 14:42:48 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
2022-12-05 07:23:18 +00:00
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-12-05 07:23:18 +00:00
)
self.log.info(
"[TNC] ARQ | DATA | RX | ["
2022-12-19 14:42:48 +00:00
+ str(self.mycallsign, "UTF-8")
+ "]>>|<<["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2022-05-23 01:11:40 +00:00
bandwidth="wide",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
)
2022-05-09 00:41:49 +00:00
# set start of transmission for our statistics
self.rx_start_of_transmission = time.time()
2021-02-28 14:24:14 +00:00
2023-01-06 13:41:15 +00:00
# Reset data_channel/burst timestamps once again for avoiding running into timeout
2022-05-19 20:15:24 +00:00
self.data_channel_last_received = int(time.time())
self.burst_last_received = int(time.time() + 10) # we might need some more time so lets increase this
2022-05-19 20:15:24 +00:00
def arq_received_channel_is_open(self, data_in: bytes) -> None:
"""
2022-03-06 16:23:04 +00:00
Called if we received a data channel opener
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
protocol_version = int.from_bytes(bytes(data_in[13:14]), "big")
2023-04-27 19:43:56 +00:00
if protocol_version == ARQ.arq_protocol_version:
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
2022-06-06 18:31:15 +00:00
status="opened",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
2022-05-09 00:41:49 +00:00
frametype = int.from_bytes(bytes(data_in[:1]), "big")
if frametype == FR_TYPE.ARQ_DC_OPEN_ACK_N.value:
self.received_LOW_BANDWIDTH_MODE = True
2022-05-09 00:41:49 +00:00
self.mode_list = self.mode_list_low_bw
self.time_list = self.time_list_low_bw
self.log.debug("[TNC] low bandwidth mode", modes=self.mode_list)
else:
self.received_LOW_BANDWIDTH_MODE = False
2022-05-09 00:41:49 +00:00
self.mode_list = self.mode_list_high_bw
self.time_list = self.time_list_high_bw
self.log.debug("[TNC] high bandwidth mode", modes=self.mode_list)
# set speed level from session opener frame which is selected by SNR measurement
self.speed_level = int.from_bytes(bytes(data_in[8:9]), "big")
self.log.debug("[TNC] speed level selected for given SNR", speed_level=self.speed_level)
2022-09-08 13:47:23 +00:00
# self.speed_level = len(self.mode_list) - 1
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
"DATA-CHANNEL",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
self.log.info(
"[TNC] ARQ | DATA | TX | ["
+ str(self.mycallsign, "UTF-8")
+ "]>>|<<["
2022-12-03 12:59:05 +00:00
+ str(self.dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
)
2022-05-09 00:41:49 +00:00
# as soon as we set ARQ_STATE to DATA, transmission starts
2023-04-27 19:43:56 +00:00
ARQ.arq_state = True
# Update data_channel timestamp
self.data_channel_last_received = int(time.time())
else:
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "IDLE"
ARQ.arq_state = False
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
status="failed",
reason="protocol version missmatch",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
# TODO: We should display a message to this effect on the UI.
self.log.warning(
"[TNC] protocol version mismatch:",
received=protocol_version,
2023-04-27 19:43:56 +00:00
own=ARQ.arq_protocol_version,
)
self.stop_transmission()
2021-02-28 14:24:14 +00:00
# ---------- PING
2022-12-05 15:57:45 +00:00
def transmit_ping(self, mycallsign: bytes, dxcallsign: bytes) -> None:
"""
2022-03-06 16:23:04 +00:00
Funktion for controlling pings
Args:
2022-12-05 15:57:45 +00:00
mycallsign:bytes:
2022-05-09 00:41:49 +00:00
dxcallsign:bytes:
"""
2022-06-17 23:48:47 +00:00
if not str(dxcallsign).strip():
# TODO: We should display a message to this effect on the UI.
2022-06-17 23:48:47 +00:00
self.log.warning("[TNC] Missing required callsign", dxcallsign=dxcallsign)
return
2023-04-27 19:43:56 +00:00
Station.dxcallsign = dxcallsign
Station.dxcallsign_crc = helpers.get_crc_24(Station.dxcallsign)
2022-12-19 14:42:48 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
ping="transmitting",
dxcallsign=str(dxcallsign, "UTF-8"),
mycallsign=str(mycallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
snr=str(ModemParam.snr),
2022-12-19 14:42:48 +00:00
)
self.log.info(
"[TNC] PING REQ ["
2022-12-05 15:57:45 +00:00
+ str(mycallsign, "UTF-8")
+ "] >>> ["
2022-12-03 12:59:05 +00:00
+ str(dxcallsign, "UTF-8")
+ "]"
)
ping_frame = bytearray(self.length_sig0_frame)
ping_frame[:1] = bytes([FR_TYPE.PING.value])
2023-04-27 19:43:56 +00:00
ping_frame[1:4] = Station.dxcallsign_crc
2022-12-19 15:00:47 +00:00
ping_frame[4:7] = helpers.get_crc_24(mycallsign)
2022-12-05 15:57:45 +00:00
ping_frame[7:13] = helpers.callsign_to_bytes(mycallsign)
2023-04-27 19:43:56 +00:00
if TNC.enable_fsk:
self.log.info("[TNC] ENABLE FSK", state=TNC.enable_fsk)
self.enqueue_frame_for_tx([ping_frame], c2_mode=FREEDV_MODE.fsk_ldpc_0.value)
2022-03-31 10:45:44 +00:00
else:
self.enqueue_frame_for_tx([ping_frame], c2_mode=FREEDV_MODE.sig0.value)
2022-05-09 00:41:49 +00:00
def received_ping(self, data_in: bytes) -> None:
"""
2022-03-06 16:23:04 +00:00
Called if we received a ping
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
dxcallsign_crc = bytes(data_in[4:7])
dxcallsign = helpers.bytes_to_callsign(bytes(data_in[7:13]))
2022-05-09 00:41:49 +00:00
2022-03-19 11:42:10 +00:00
# check if callsign ssid override
valid, mycallsign = helpers.check_callsign(self.mycallsign, data_in[1:4])
2022-04-10 20:45:05 +00:00
if not valid:
# PING packet not for me.
self.log.debug("[TNC] received_ping: ping not for this station.")
2022-04-10 20:45:05 +00:00
return
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
Station.dxcallsign_crc = dxcallsign_crc
Station.dxcallsign = dxcallsign
self.log.info(
"[TNC] PING REQ ["
+ str(mycallsign, "UTF-8")
+ "] <<< ["
2022-12-03 12:59:05 +00:00
+ str(dxcallsign, "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
)
2021-02-24 15:47:52 +00:00
2023-04-27 19:43:56 +00:00
Station.dxgrid = b'------'
helpers.add_to_heard_stations(
2022-12-03 13:04:09 +00:00
dxcallsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"PING",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
self.send_data_to_socket_queue(
freedata="tnc-message",
ping="received",
uuid=str(uuid.uuid4()),
timestamp=int(time.time()),
2023-04-27 19:43:56 +00:00
dxgrid=str(Station.dxgrid, "UTF-8"),
dxcallsign=str(dxcallsign, "UTF-8"),
2022-12-07 15:30:59 +00:00
mycallsign=str(mycallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
snr=str(ModemParam.snr),
)
2023-04-27 19:43:56 +00:00
if TNC.respond_to_call:
2023-02-09 17:24:37 +00:00
self.transmit_ping_ack()
def transmit_ping_ack(self):
"""
transmit a ping ack frame
called by def received_ping
"""
ping_frame = bytearray(self.length_sig0_frame)
ping_frame[:1] = bytes([FR_TYPE.PING_ACK.value])
2023-04-27 19:43:56 +00:00
ping_frame[1:4] = Station.dxcallsign_crc
ping_frame[4:7] = Station.mycallsign_crc
ping_frame[7:11] = helpers.encode_grid(Station.mygrid.decode("UTF-8"))
ping_frame[13:14] = helpers.snr_to_bytes(ModemParam.snr)
2023-02-09 17:24:37 +00:00
2023-04-27 19:43:56 +00:00
if TNC.enable_fsk:
2023-02-09 17:24:37 +00:00
self.enqueue_frame_for_tx([ping_frame], c2_mode=FREEDV_MODE.fsk_ldpc_0.value)
else:
self.enqueue_frame_for_tx([ping_frame], c2_mode=FREEDV_MODE.sig0.value)
2022-05-09 00:41:49 +00:00
def received_ping_ack(self, data_in: bytes) -> None:
"""
2022-03-06 16:23:04 +00:00
Called if a PING ack has been received
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
2022-12-03 13:04:09 +00:00
# check if we received correct ping
2022-12-19 15:00:47 +00:00
# check if callsign ssid override
_valid, mycallsign = helpers.check_callsign(self.mycallsign, data_in[1:4])
if _valid:
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
Station.dxgrid = bytes(helpers.decode_grid(data_in[7:11]), "UTF-8")
2022-12-10 19:46:17 +00:00
dxsnr = helpers.snr_from_bytes(data_in[13:14])
2022-12-03 13:04:09 +00:00
self.send_data_to_socket_queue(
freedata="tnc-message",
ping="acknowledge",
uuid=str(uuid.uuid4()),
timestamp=int(time.time()),
2023-04-27 19:43:56 +00:00
dxgrid=str(Station.dxgrid, "UTF-8"),
dxcallsign=str(Station.dxcallsign, "UTF-8"),
2022-12-19 15:00:47 +00:00
mycallsign=str(mycallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
snr=str(ModemParam.snr),
2022-12-10 19:46:17 +00:00
dxsnr=str(dxsnr)
2022-12-03 13:04:09 +00:00
)
2022-12-10 19:46:17 +00:00
# combined_snr = own rx snr / snr on dx side
2023-04-27 19:43:56 +00:00
combined_snr = f"{ModemParam.snr}/{dxsnr}"
2022-12-03 13:04:09 +00:00
helpers.add_to_heard_stations(
2023-04-27 19:43:56 +00:00
Station.dxcallsign,
Station.dxgrid,
2022-12-03 13:04:09 +00:00
"PING-ACK",
2022-12-10 19:46:17 +00:00
combined_snr,
2023-04-27 19:43:56 +00:00
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
2022-12-03 13:04:09 +00:00
)
2022-05-09 00:41:49 +00:00
2022-12-03 13:04:09 +00:00
self.log.info(
"[TNC] PING ACK ["
2022-12-19 15:00:47 +00:00
+ str(mycallsign, "UTF-8")
2022-12-03 13:04:09 +00:00
+ "] >|< ["
2023-04-27 19:43:56 +00:00
+ str(Station.dxcallsign, "UTF-8")
2022-12-03 13:04:09 +00:00
+ "]",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
2022-12-10 19:46:17 +00:00
dxsnr=dxsnr,
2022-12-03 13:04:09 +00:00
)
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "IDLE"
2022-12-03 13:04:09 +00:00
else:
self.log.info(
"[TNC] FOREIGN PING ACK ["
+ str(self.mycallsign, "UTF-8")
+ "] ??? ["
+ str(bytes(data_in[4:7]), "UTF-8")
+ "]",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
2022-12-03 13:04:09 +00:00
)
2022-12-03 13:05:39 +00:00
def stop_transmission(self) -> None:
2022-03-06 16:23:04 +00:00
"""
Force a stop of the running transmission
"""
self.log.warning("[TNC] Stopping transmission!")
2022-12-24 22:13:21 +00:00
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "IDLE"
ARQ.arq_state = False
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
2022-12-24 22:13:21 +00:00
status="stopped",
mycallsign=str(self.mycallsign, 'UTF-8'),
2023-02-05 13:25:53 +00:00
dxcallsign=str(self.dxcallsign, 'UTF-8')
2022-06-06 18:31:15 +00:00
)
2022-12-24 22:13:21 +00:00
stop_frame = bytearray(self.length_sig0_frame)
stop_frame[:1] = bytes([FR_TYPE.ARQ_STOP.value])
2023-04-27 19:43:56 +00:00
stop_frame[1:4] = Station.dxcallsign_crc
stop_frame[4:7] = Station.mycallsign_crc
2022-12-24 22:13:21 +00:00
# TODO: Not sure if we really need the session id when disconnecting
# stop_frame[1:2] = self.session_id
stop_frame[7:13] = helpers.callsign_to_bytes(self.mycallsign)
2023-04-21 16:14:44 +00:00
self.enqueue_frame_for_tx([stop_frame], c2_mode=FREEDV_MODE.sig1.value, copies=3, repeat_delay=0)
2022-12-24 22:13:21 +00:00
self.arq_cleanup()
def received_stop_transmission(
2022-09-08 13:47:23 +00:00
self, data_in: bytes
) -> None: # pylint: disable=unused-argument
2022-03-06 16:23:04 +00:00
"""
Received a transmission stop
"""
self.log.warning("[TNC] Stopping transmission!")
2023-04-27 19:43:56 +00:00
TNC.tnc_state = "IDLE"
ARQ.arq_state = False
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
2022-12-24 22:13:21 +00:00
status="stopped",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:25:53 +00:00
uuid=self.transmission_uuid
2022-06-05 17:11:09 +00:00
)
self.arq_cleanup()
2022-05-09 00:41:49 +00:00
# ----------- BROADCASTS
def run_beacon(self) -> None:
"""
2022-05-19 20:15:24 +00:00
Controlling function for running a beacon
Args:
2022-05-19 20:15:24 +00:00
self: arq class
Returns:
"""
try:
while True:
threading.Event().wait(0.5)
2023-04-27 19:43:56 +00:00
while Beacon.beacon_state:
if (
2023-04-27 19:43:56 +00:00
not ARQ.arq_session
2022-09-08 13:47:23 +00:00
and not self.arq_file_transfer
2023-04-27 19:43:56 +00:00
and not Beacon.beacon_pause
and not ModemParam.channel_busy
2023-05-08 14:17:11 +00:00
and TNC.tnc_state not in ["BUSY"]
2023-04-27 19:43:56 +00:00
and not ARQ.arq_state
):
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
beacon="transmitting",
dxcallsign="None",
2022-06-06 18:44:35 +00:00
interval=self.beacon_interval,
2022-06-05 17:11:09 +00:00
)
self.log.info(
"[TNC] Sending beacon!", interval=self.beacon_interval
)
2022-05-09 00:41:49 +00:00
beacon_frame = bytearray(self.length_sig0_frame)
beacon_frame[:1] = bytes([FR_TYPE.BEACON.value])
2022-05-19 20:15:24 +00:00
beacon_frame[1:7] = helpers.callsign_to_bytes(self.mycallsign)
2023-04-27 19:43:56 +00:00
beacon_frame[7:11] = helpers.encode_grid(Station.mygrid.decode("UTF-8"))
2023-04-27 19:43:56 +00:00
if TNC.enable_fsk:
self.log.info("[TNC] ENABLE FSK", state=TNC.enable_fsk)
self.enqueue_frame_for_tx(
[beacon_frame],
c2_mode=FREEDV_MODE.fsk_ldpc_0.value,
)
2022-03-31 10:45:44 +00:00
else:
self.enqueue_frame_for_tx([beacon_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1,
repeat_delay=0)
2022-05-09 00:41:49 +00:00
self.beacon_interval_timer = time.time() + self.beacon_interval
while (
time.time() < self.beacon_interval_timer
2023-04-27 19:43:56 +00:00
and Beacon.beacon_state
and not Beacon.beacon_pause
):
threading.Event().wait(0.01)
except Exception as err:
self.log.debug("[TNC] run_beacon: ", exception=err)
def received_beacon(self, data_in: bytes) -> None:
"""
2022-03-06 16:23:04 +00:00
Called if we received a beacon
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
"""
# here we add the received station to the heard stations buffer
beacon_callsign = helpers.bytes_to_callsign(bytes(data_in[1:7]))
2023-04-27 19:43:56 +00:00
Station.dxgrid = bytes(helpers.decode_grid(data_in[7:11]), "UTF-8")
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-07 08:08:15 +00:00
beacon="received",
uuid=str(uuid.uuid4()),
timestamp=int(time.time()),
dxcallsign=str(beacon_callsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
dxgrid=str(Station.dxgrid, "UTF-8"),
snr=str(ModemParam.snr),
)
2022-06-07 08:08:15 +00:00
self.log.info(
"[TNC] BEACON RCVD ["
+ str(beacon_callsign, "UTF-8")
+ "]["
2023-04-27 19:43:56 +00:00
+ str(Station.dxgrid, "UTF-8")
+ "] ",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
)
helpers.add_to_heard_stations(
beacon_callsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"BEACON",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
def transmit_cq(self) -> None:
"""
Transmit a CQ
Args:
2022-10-05 18:27:38 +00:00
self
Returns:
Nothing
"""
self.log.info("[TNC] CQ CQ CQ")
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
cq="transmitting",
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, "UTF-8"),
dxcallsign="None",
2022-06-05 17:11:09 +00:00
)
cq_frame = bytearray(self.length_sig0_frame)
cq_frame[:1] = bytes([FR_TYPE.CQ.value])
2022-05-19 20:15:24 +00:00
cq_frame[1:7] = helpers.callsign_to_bytes(self.mycallsign)
2023-04-27 19:43:56 +00:00
cq_frame[7:11] = helpers.encode_grid(Station.mygrid.decode("UTF-8"))
2022-05-09 00:41:49 +00:00
self.log.debug("[TNC] CQ Frame:", data=[cq_frame])
2023-04-27 19:43:56 +00:00
if TNC.enable_fsk:
self.log.info("[TNC] ENABLE FSK", state=TNC.enable_fsk)
self.enqueue_frame_for_tx([cq_frame], c2_mode=FREEDV_MODE.fsk_ldpc_0.value)
else:
self.enqueue_frame_for_tx([cq_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
# FIXME: Remove or change this in later versions for full CW support
# TNC.transmitting = True
# modem.MODEM_TRANSMIT_QUEUE.put(["morse", 1, 0, "123"])
def received_cq(self, data_in: bytes) -> None:
"""
Called when we receive a CQ frame
Args:
2022-05-09 00:41:49 +00:00
data_in:bytes:
Returns:
Nothing
"""
# here we add the received station to the heard stations buffer
dxcallsign = helpers.bytes_to_callsign(bytes(data_in[1:7]))
self.log.debug("[TNC] received_cq:", dxcallsign=dxcallsign)
2023-04-27 19:43:56 +00:00
Station.dxgrid = bytes(helpers.decode_grid(data_in[7:11]), "UTF-8")
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-07 08:08:15 +00:00
cq="received",
2022-06-06 19:07:41 +00:00
mycallsign=str(self.mycallsign, "UTF-8"),
dxcallsign=str(dxcallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
dxgrid=str(Station.dxgrid, "UTF-8"),
2022-06-05 17:11:09 +00:00
)
self.log.info(
"[TNC] CQ RCVD ["
+ str(dxcallsign, "UTF-8")
+ "]["
2023-04-27 19:43:56 +00:00
+ str(Station.dxgrid, "UTF-8")
+ "] ",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
)
helpers.add_to_heard_stations(
dxcallsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"CQ CQ CQ",
2023-04-27 19:43:56 +00:00
ModemParam.snr,
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
2023-04-27 19:43:56 +00:00
if TNC.respond_to_cq and TNC.respond_to_call:
self.transmit_qrv(dxcallsign)
def transmit_qrv(self, dxcallsign: bytes) -> None:
2022-03-06 16:23:04 +00:00
"""
Called when we send a QRV frame
Args:
self,
dxcallsign
2022-03-06 16:23:04 +00:00
"""
# Sleep a random amount of time before responding to make it more likely to be
# heard when many stations respond. Each DATAC0 frame is 0.44 sec (440ms) in
2023-04-22 13:28:50 +00:00
# duration, plus overhead. Set the wait interval to be random between 0 and
# self.duration_sig1_frame * 4 == 4 slots
# in self.duration_sig1_frame increments.
2023-05-08 09:24:52 +00:00
# FIXME: This causes problems when running ctests - we need to figure out why
if not TESTMODE:
self.log.info("[TNC] Waiting for QRV slot...")
helpers.wait(randrange(0, int(self.duration_sig1_frame * 4), self.duration_sig1_frame * 10 // 10.0))
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
qrv="transmitting",
dxcallsign=str(dxcallsign, "UTF-8"),
2022-06-05 17:11:09 +00:00
)
self.log.info("[TNC] Sending QRV!")
qrv_frame = bytearray(self.length_sig0_frame)
qrv_frame[:1] = bytes([FR_TYPE.QRV.value])
2022-05-19 20:15:24 +00:00
qrv_frame[1:7] = helpers.callsign_to_bytes(self.mycallsign)
2023-04-27 19:43:56 +00:00
qrv_frame[7:11] = helpers.encode_grid(Station.mygrid.decode("UTF-8"))
qrv_frame[11:12] = helpers.snr_to_bytes(ModemParam.snr)
2023-04-27 19:43:56 +00:00
if TNC.enable_fsk:
self.log.info("[TNC] ENABLE FSK", state=TNC.enable_fsk)
self.enqueue_frame_for_tx([qrv_frame], c2_mode=FREEDV_MODE.fsk_ldpc_0.value)
2022-03-31 10:45:44 +00:00
else:
2023-05-08 09:24:52 +00:00
self.enqueue_frame_for_tx([qrv_frame], c2_mode=FREEDV_MODE.sig0.value, copies=1, repeat_delay=0)
2021-09-26 15:51:11 +00:00
def received_qrv(self, data_in: bytes) -> None:
"""
Called when we receive a QRV frame
Args:
data_in:bytes:
"""
# here we add the received station to the heard stations buffer
dxcallsign = helpers.bytes_to_callsign(bytes(data_in[1:7]))
2023-04-27 19:43:56 +00:00
Station.dxgrid = bytes(helpers.decode_grid(data_in[7:11]), "UTF-8")
2022-12-19 15:46:18 +00:00
dxsnr = helpers.snr_from_bytes(data_in[11:12])
2023-04-27 19:43:56 +00:00
combined_snr = f"{ModemParam.snr}/{dxsnr}"
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-06 18:44:35 +00:00
qrv="received",
2022-12-01 09:09:44 +00:00
dxcallsign=str(dxcallsign, "UTF-8"),
2023-04-27 19:43:56 +00:00
dxgrid=str(Station.dxgrid, "UTF-8"),
snr=str(ModemParam.snr),
2022-12-19 15:46:18 +00:00
dxsnr=str(dxsnr)
)
self.log.info(
"[TNC] QRV RCVD ["
+ str(dxcallsign, "UTF-8")
+ "]["
2023-04-27 19:43:56 +00:00
+ str(Station.dxgrid, "UTF-8")
+ "] ",
2023-04-27 19:43:56 +00:00
snr=ModemParam.snr,
2022-12-19 15:46:18 +00:00
dxsnr=dxsnr
)
helpers.add_to_heard_stations(
dxcallsign,
2023-04-27 19:43:56 +00:00
Station.dxgrid,
"QRV",
2022-12-19 15:46:18 +00:00
combined_snr,
2023-04-27 19:43:56 +00:00
ModemParam.frequency_offset,
HamlibParam.hamlib_frequency,
)
2023-02-13 21:41:09 +00:00
2023-05-14 17:44:35 +00:00
2023-02-12 17:34:05 +00:00
def received_is_writing(self, data_in: bytes) -> None:
"""
Called when we receive a IS WRITING frame
Args:
data_in:bytes:
"""
# here we add the received station to the heard stations buffer
dxcallsign = helpers.bytes_to_callsign(bytes(data_in[1:7]))
self.send_data_to_socket_queue(
freedata="tnc-message",
fec="is_writing",
dxcallsign=str(dxcallsign, "UTF-8")
)
self.log.info(
"[TNC] IS_WRITING RCVD ["
+ str(dxcallsign, "UTF-8")
+ "] ",
)
2022-11-02 14:29:22 +00:00
# ------------ CALCULATE TRANSFER RATES
def calculate_transfer_rate_rx(
2022-09-08 13:47:23 +00:00
self, rx_start_of_transmission: float, receivedbytes: int
) -> list:
"""
Calculate transfer rate for received data
Args:
2022-05-09 00:41:49 +00:00
rx_start_of_transmission:float:
receivedbytes:int:
Returns: List of:
bits_per_second: float,
bytes_per_minute: float,
transmission_percent: float
"""
2022-05-09 00:41:49 +00:00
try:
2023-04-27 19:43:56 +00:00
if ARQ.total_bytes == 0:
ARQ.total_bytes = 1
ARQ.arq_transmission_percent = min(
int(
(
2022-09-08 13:47:23 +00:00
receivedbytes
2023-04-27 19:43:56 +00:00
* ARQ.arq_compression_factor
/ ARQ.total_bytes
)
* 100
),
100,
)
2022-05-09 00:41:49 +00:00
transmissiontime = time.time() - self.rx_start_of_transmission
2022-05-09 00:41:49 +00:00
if receivedbytes > 0:
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second = int((receivedbytes * 8) / transmissiontime)
ARQ.bytes_per_minute = int(
2022-10-05 18:27:38 +00:00
receivedbytes / (transmissiontime / 60)
)
2023-04-27 19:43:56 +00:00
ARQ.arq_seconds_until_finish = int(((ARQ.total_bytes - receivedbytes) / (
ARQ.bytes_per_minute * ARQ.arq_compression_factor)) * 60) - 20 # offset because of frame ack/nack
2023-04-27 19:43:56 +00:00
speed_chart = {"snr": ModemParam.snr, "bpm": ARQ.bytes_per_minute, "timestamp": int(time.time())}
# check if data already in list
2023-04-27 19:43:56 +00:00
if speed_chart not in ARQ.speed_list:
ARQ.speed_list.append(speed_chart)
else:
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second = 0
ARQ.bytes_per_minute = 0
ARQ.arq_seconds_until_finish = 0
except Exception as err:
self.log.error(f"[TNC] calculate_transfer_rate_rx: Exception: {err}")
2023-04-27 19:43:56 +00:00
ARQ.arq_transmission_percent = 0.0
ARQ.arq_bits_per_second = 0
ARQ.bytes_per_minute = 0
return [
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second,
ARQ.bytes_per_minute,
ARQ.arq_transmission_percent,
]
def reset_statistics(self) -> None:
2022-03-06 16:23:04 +00:00
"""
Reset statistics
"""
# reset ARQ statistics
2023-05-06 15:32:15 +00:00
ARQ.bytes_per_minute_burst = 0
2023-04-27 19:43:56 +00:00
ARQ.bytes_per_minute = 0
ARQ.arq_bits_per_second_burst = 0
ARQ.arq_bits_per_second = 0
ARQ.arq_transmission_percent = 0
ARQ.total_bytes = 0
ARQ.arq_seconds_until_finish = 0
2022-05-09 00:41:49 +00:00
def calculate_transfer_rate_tx(
2022-09-08 13:47:23 +00:00
self, tx_start_of_transmission: float, sentbytes: int, tx_buffer_length: int
) -> list:
"""
2022-05-23 07:37:24 +00:00
Calculate transfer rate for transmission
Args:
2022-05-09 00:41:49 +00:00
tx_start_of_transmission:float:
sentbytes:int:
tx_buffer_length:int:
Returns: List of:
bits_per_second: float,
bytes_per_minute: float,
transmission_percent: float
"""
try:
2023-04-27 19:43:56 +00:00
ARQ.arq_transmission_percent = min(
int((sentbytes / tx_buffer_length) * 100), 100
)
2022-05-09 00:41:49 +00:00
transmissiontime = time.time() - tx_start_of_transmission
if sentbytes > 0:
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second = int((sentbytes * 8) / transmissiontime)
ARQ.bytes_per_minute = int(sentbytes / (transmissiontime / 60))
ARQ.arq_seconds_until_finish = int(((tx_buffer_length - sentbytes) / (
ARQ.bytes_per_minute * ARQ.arq_compression_factor)) * 60)
2023-04-27 19:43:56 +00:00
speed_chart = {"snr": self.burst_ack_snr, "bpm": ARQ.bytes_per_minute,
"timestamp": int(time.time())}
# check if data already in list
2023-04-27 19:43:56 +00:00
if speed_chart not in ARQ.speed_list:
ARQ.speed_list.append(speed_chart)
else:
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second = 0
ARQ.bytes_per_minute = 0
ARQ.arq_seconds_until_finish = 0
2022-05-09 00:41:49 +00:00
except Exception as err:
self.log.error(f"[TNC] calculate_transfer_rate_tx: Exception: {err}")
2023-04-27 19:43:56 +00:00
ARQ.arq_transmission_percent = 0.0
ARQ.arq_bits_per_second = 0
ARQ.bytes_per_minute = 0
2021-09-26 15:51:11 +00:00
return [
2023-04-27 19:43:56 +00:00
ARQ.arq_bits_per_second,
ARQ.bytes_per_minute,
ARQ.arq_transmission_percent,
]
# ----------------------CLEANUP AND RESET FUNCTIONS
def arq_cleanup(self) -> None:
2022-03-06 16:23:04 +00:00
"""
Cleanup function which clears all ARQ states
2022-03-06 16:23:04 +00:00
"""
if TESTMODE:
self.log.debug("[TNC] TESTMODE: arq_cleanup: Not performing cleanup.")
return
self.log.debug("[TNC] arq_cleanup")
2023-01-29 17:50:09 +00:00
# wait a second for smoother arq behaviour
helpers.wait(1.0)
2023-01-11 15:18:29 +00:00
2022-01-07 10:44:35 +00:00
self.rx_frame_bof_received = False
self.rx_frame_eof_received = False
2022-01-07 11:55:03 +00:00
self.burst_ack = False
self.rpt_request_received = False
self.burst_rpt_counter = 0
self.data_frame_ack_received = False
2023-04-27 19:43:56 +00:00
ARQ.rx_burst_buffer = []
ARQ.rx_frame_buffer = b""
2023-01-12 23:14:42 +00:00
self.burst_ack_snr = 0
2023-04-24 09:52:03 +00:00
self.arq_burst_last_payload = 0
self.rx_n_frame_of_burst = 0
self.rx_n_frames_per_burst = 0
# reset modem receiving state to reduce cpu load
2022-10-10 07:00:45 +00:00
modem.RECEIVE_SIG0 = True
modem.RECEIVE_SIG1 = False
modem.RECEIVE_DATAC1 = False
modem.RECEIVE_DATAC3 = False
2023-04-21 16:14:44 +00:00
modem.RECEIVE_DATAC4 = False
2022-05-19 20:15:24 +00:00
# modem.RECEIVE_FSK_LDPC_0 = False
modem.RECEIVE_FSK_LDPC_1 = False
2022-05-09 00:41:49 +00:00
# reset buffer overflow counter
2023-04-27 19:43:56 +00:00
AudioParam.buffer_overflow_counter = [0, 0, 0, 0, 0]
self.is_IRS = False
self.burst_nack = False
self.burst_nack_counter = 0
self.frame_nack_counter = 0
self.frame_received_counter = 0
self.speed_level = len(self.mode_list) - 1
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
2022-05-09 00:41:49 +00:00
2022-05-23 01:11:40 +00:00
# low bandwidth mode indicator
self.received_LOW_BANDWIDTH_MODE = False
2022-05-09 00:41:49 +00:00
# reset retry counter for rx channel / burst
self.n_retries_per_burst = 0
2022-05-09 00:41:49 +00:00
# reset max retries possibly overriden by api
2022-12-25 13:24:39 +00:00
self.session_connect_max_retries = 10
self.data_channel_max_retries = 10
2023-01-11 15:18:29 +00:00
# we need to keep these values if in ARQ_SESSION
2023-04-27 19:43:56 +00:00
if not ARQ.arq_session:
TNC.tnc_state = "IDLE"
self.dxcallsign = b"AA0AA-0"
2023-04-27 19:43:56 +00:00
self.mycallsign = Station.mycallsign
2023-01-11 15:18:29 +00:00
self.session_id = bytes(1)
2022-05-09 00:41:49 +00:00
2023-05-08 14:13:37 +00:00
ARQ.arq_session_state = "disconnected"
2023-04-27 19:43:56 +00:00
ARQ.speed_list = []
ARQ.arq_state = False
self.arq_file_transfer = False
2022-05-09 00:41:49 +00:00
2023-04-27 19:43:56 +00:00
Beacon.beacon_pause = False
# reset beacon interval timer for not directly starting beacon after ARQ
self.beacon_interval_timer = time.time() + self.beacon_interval
2022-05-09 00:41:49 +00:00
def arq_reset_ack(self, state: bool) -> None:
"""
2022-03-06 16:23:04 +00:00
Funktion for resetting acknowledge states
Args:
2022-05-09 00:41:49 +00:00
state:bool:
"""
2022-01-07 11:55:03 +00:00
self.burst_ack = state
self.rpt_request_received = state
self.data_frame_ack_received = state
2021-09-26 15:51:11 +00:00
2022-10-10 07:00:45 +00:00
def set_listening_modes(self, enable_sig0: bool, enable_sig1: bool, mode: int) -> None:
2023-02-09 17:24:37 +00:00
# sourcery skip: extract-duplicate-method
"""
2022-03-06 16:23:04 +00:00
Function for setting the data modes we are listening to for saving cpu power
Args:
2022-10-10 07:00:45 +00:00
enable_sig0:int: Enable/Disable signalling mode 0
enable_sig1:int: Enable/Disable signalling mode 1
mode:int: Codec2 mode to listen for
"""
# set modes we want to listen to
2022-10-10 07:00:45 +00:00
modem.RECEIVE_SIG0 = enable_sig0
modem.RECEIVE_SIG1 = enable_sig1
2023-04-21 16:14:44 +00:00
if mode == codec2.FREEDV_MODE.datac1.value:
modem.RECEIVE_DATAC1 = True
modem.RECEIVE_DATAC3 = False
2023-04-21 16:14:44 +00:00
modem.RECEIVE_DATAC4 = False
modem.RECEIVE_FSK_LDPC_1 = False
self.log.debug("[TNC] Changing listening data mode", mode="datac1")
2023-04-21 16:14:44 +00:00
elif mode == codec2.FREEDV_MODE.datac3.value:
modem.RECEIVE_DATAC1 = False
modem.RECEIVE_DATAC3 = True
2023-04-21 16:14:44 +00:00
modem.RECEIVE_DATAC4 = False
modem.RECEIVE_FSK_LDPC_1 = False
self.log.debug("[TNC] Changing listening data mode", mode="datac3")
2023-04-21 16:14:44 +00:00
elif mode == codec2.FREEDV_MODE.datac4.value:
modem.RECEIVE_DATAC1 = False
modem.RECEIVE_DATAC3 = False
2023-04-21 16:14:44 +00:00
modem.RECEIVE_DATAC4 = True
modem.RECEIVE_FSK_LDPC_1 = False
self.log.debug("[TNC] Changing listening data mode", mode="datac4")
elif mode == codec2.FREEDV_MODE.fsk_ldpc_1.value:
modem.RECEIVE_DATAC1 = False
modem.RECEIVE_DATAC3 = False
modem.RECEIVE_DATAC4 = False
modem.RECEIVE_FSK_LDPC_1 = True
self.log.debug("[TNC] Changing listening data mode", mode="fsk_ldpc_1")
else:
modem.RECEIVE_DATAC1 = True
modem.RECEIVE_DATAC3 = True
2023-04-21 16:14:44 +00:00
modem.RECEIVE_DATAC4 = True
modem.RECEIVE_FSK_LDPC_1 = True
self.log.debug(
"[TNC] Changing listening data mode", mode="datac1/datac3/fsk_ldpc"
)
2022-05-09 00:41:49 +00:00
# ------------------------- WATCHDOG FUNCTIONS FOR TIMER
def watchdog(self) -> None:
"""Author: DJ2LS
2022-05-09 00:41:49 +00:00
Watchdog master function. From here, "pet" the watchdogs
"""
while True:
threading.Event().wait(0.1)
self.data_channel_keep_alive_watchdog()
self.burst_watchdog()
self.arq_session_keep_alive_watchdog()
2021-09-26 15:51:11 +00:00
def burst_watchdog(self) -> None:
2022-03-06 16:23:04 +00:00
"""
Watchdog which checks if we are running into a connection timeout
2022-03-06 16:23:04 +00:00
DATA BURST
"""
2022-05-09 00:41:49 +00:00
# IRS SIDE
2022-05-19 20:15:24 +00:00
# TODO: We need to redesign this part for cleaner state handling
# Return if not ARQ STATE and not ARQ SESSION STATE as they are different use cases
if (
2023-04-27 19:43:56 +00:00
not ARQ.arq_state
and ARQ.arq_session_state != "connected"
2022-09-08 13:47:23 +00:00
or not self.is_IRS
):
2022-05-20 12:38:43 +00:00
return
2022-12-05 14:49:10 +00:00
# get modem error state
modem_error_state = modem.get_modem_error_state()
# We want to reach this state only if connected ( == return above not called )
if self.rx_n_frames_per_burst > 1:
2023-04-25 14:32:36 +00:00
# uses case for IRS: reduce time for waiting by counting "None" in burst buffer
2023-04-27 19:43:56 +00:00
frames_left = ARQ.rx_burst_buffer.count(None)
2023-04-25 16:51:51 +00:00
elif self.rx_n_frame_of_burst == 0 and self.rx_n_frames_per_burst == 0:
2023-04-25 14:32:36 +00:00
# use case for IRS: We didn't receive a burst yet, because the first one got lost
# in this case we don't have any information about the expected burst length
# we must assume, we are getting a burst with max_n_frames_per_burst
frames_left = self.max_n_frames_per_burst
else:
frames_left = 1
2023-04-25 14:32:36 +00:00
2023-04-25 19:08:48 +00:00
# make sure we don't have a 0 here for avoiding too short timeouts
if frames_left == 0:
frames_left = 1
2023-04-25 16:51:51 +00:00
timeout = self.burst_last_received + (self.time_list[self.speed_level] * frames_left)
2023-04-25 19:32:39 +00:00
# TODO: Enable this for development
# print(f"timeout expected in:{round(timeout - time.time())} | frames left: {frames_left} of {self.rx_n_frames_per_burst} | speed level: {self.speed_level}")
2023-01-04 10:04:10 +00:00
if timeout <= time.time() or modem_error_state:
2023-02-09 19:32:21 +00:00
self.log.warning(
"[TNC] Burst decoding error or timeout",
attempt=self.n_retries_per_burst,
max_attempts=self.rx_n_max_retries_per_burst,
speed_level=self.speed_level,
modem_error_state=modem_error_state
)
2023-01-04 10:04:10 +00:00
print(
2023-04-27 19:43:56 +00:00
f"frames_per_burst {self.rx_n_frame_of_burst} / {self.rx_n_frames_per_burst}, Repeats: {self.burst_rpt_counter} Nones: {ARQ.rx_burst_buffer.count(None)}")
2023-04-25 19:08:48 +00:00
2023-04-27 19:43:56 +00:00
if self.rx_n_frames_per_burst > 1 and self.burst_rpt_counter < 3 and ARQ.rx_burst_buffer.count(None) > 0:
# reset self.burst_last_received
2023-04-25 19:08:48 +00:00
self.burst_last_received = time.time() + self.time_list[self.speed_level] * frames_left
self.burst_rpt_counter += 1
self.send_retransmit_request_frame()
2023-04-25 16:51:51 +00:00
else:
2023-01-04 10:04:10 +00:00
# reset self.burst_last_received
self.burst_last_received = time.time() + self.time_list[self.speed_level]
2022-05-09 00:41:49 +00:00
# reduce speed level if nack counter increased
self.frame_received_counter = 0
self.burst_nack_counter += 1
if self.burst_nack_counter >= 2:
self.burst_nack_counter = 0
self.speed_level = max(self.speed_level - 1, 0)
2023-04-27 19:43:56 +00:00
ARQ.arq_speed_level = self.speed_level
2022-05-09 00:41:49 +00:00
2023-04-25 19:08:48 +00:00
# TODO: Create better mechanisms for handling n frames per burst for bad channels
# reduce frames per burst
if self.burst_rpt_counter >= 2:
tx_n_frames_per_burst = max(self.rx_n_frames_per_burst - 1, 1)
else:
tx_n_frames_per_burst = self.rx_n_frames_per_burst
# Update modes we are listening to
self.set_listening_modes(True, True, self.mode_list[self.speed_level])
2023-04-25 19:08:48 +00:00
# TODO: Does SNR make sense for NACK if we dont have an actual SNR information?
self.send_burst_nack_frame_watchdog(0, tx_n_frames_per_burst)
# Update data_channel timestamp
# TODO: Disabled this one for testing.
# self.data_channel_last_received = time.time()
self.n_retries_per_burst += 1
else:
# print((self.data_channel_last_received + self.time_list[self.speed_level])-time.time())
pass
if self.n_retries_per_burst >= self.rx_n_max_retries_per_burst:
self.stop_transmission()
2022-05-09 00:41:49 +00:00
def data_channel_keep_alive_watchdog(self) -> None:
2022-03-06 16:23:04 +00:00
"""
watchdog which checks if we are running into a connection timeout
2022-05-09 00:41:49 +00:00
DATA CHANNEL
2022-03-06 16:23:04 +00:00
"""
# and not static.ARQ_SEND_KEEP_ALIVE:
2023-04-27 19:43:56 +00:00
if ARQ.arq_state and TNC.tnc_state == "BUSY":
threading.Event().wait(0.01)
if (
2022-09-08 13:47:23 +00:00
self.data_channel_last_received + self.transmission_timeout
> time.time()
):
2022-12-11 12:07:56 +00:00
2023-01-04 10:04:10 +00:00
timeleft = int((self.data_channel_last_received + self.transmission_timeout) - time.time())
if timeleft % 10 == 0:
2023-04-25 19:08:48 +00:00
self.log.debug("Time left until channel timeout", seconds=timeleft)
2023-01-04 10:04:10 +00:00
# threading.Event().wait(5)
2022-05-19 20:15:24 +00:00
# print(self.data_channel_last_received + self.transmission_timeout - time.time())
# pass
else:
# Clear the timeout timestamp
self.data_channel_last_received = 0
self.log.info(
"[TNC] DATA ["
+ str(self.mycallsign, "UTF-8")
+ "]<<T>>["
2023-04-27 19:43:56 +00:00
+ str(Station.dxcallsign, "UTF-8")
+ "]"
)
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="transmission",
status="failed",
2022-06-06 19:07:41 +00:00
uuid=self.transmission_uuid,
2022-12-05 07:23:18 +00:00
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2023-02-05 13:35:10 +00:00
irs=helpers.bool_to_string(self.is_IRS)
2022-06-05 17:11:09 +00:00
)
self.arq_cleanup()
2022-05-09 00:41:49 +00:00
def arq_session_keep_alive_watchdog(self) -> None:
2022-03-06 16:23:04 +00:00
"""
watchdog which checks if we are running into a connection timeout
ARQ SESSION
"""
if (
2023-04-27 19:43:56 +00:00
ARQ.arq_session
and TNC.tnc_state == "BUSY"
2022-09-08 13:47:23 +00:00
and not self.arq_file_transfer
):
if self.arq_session_last_received + self.arq_session_timeout > time.time():
threading.Event().wait(0.01)
else:
self.log.info(
"[TNC] SESSION ["
+ str(self.mycallsign, "UTF-8")
+ "]<<T>>["
+ str(self.dxcallsign, "UTF-8")
+ "]"
)
2022-06-05 17:11:09 +00:00
self.send_data_to_socket_queue(
2022-06-06 18:50:36 +00:00
freedata="tnc-message",
2022-06-05 17:11:09 +00:00
arq="session",
status="failed",
reason="timeout",
mycallsign=str(self.mycallsign, 'UTF-8'),
dxcallsign=str(self.dxcallsign, 'UTF-8'),
2022-06-05 17:11:09 +00:00
)
self.close_session()
2022-05-09 00:41:49 +00:00
def heartbeat(self) -> None:
2022-03-06 16:23:04 +00:00
"""
Heartbeat thread which auto pauses and resumes the heartbeat signal when in an arq session
2022-03-06 16:23:04 +00:00
"""
while True:
threading.Event().wait(0.01)
# additional check for smoother stopping if heartbeat transmission
while not self.arq_file_transfer:
threading.Event().wait(0.01)
if (
2023-04-27 19:43:56 +00:00
ARQ.arq_session
and self.IS_ARQ_SESSION_MASTER
2023-04-27 19:43:56 +00:00
and ARQ.arq_session_state == "connected"
2022-11-02 14:29:22 +00:00
# and not self.arq_file_transfer
):
threading.Event().wait(1)
self.transmit_session_heartbeat()
threading.Event().wait(2)
2022-05-09 00:41:49 +00:00
def send_test_frame(self) -> None:
"""Send an empty test frame"""
2023-01-06 16:02:21 +00:00
test_frame = bytearray(126)
test_frame[:1] = bytes([FR_TYPE.TEST_FRAME.value])
self.enqueue_frame_for_tx(
2023-04-22 10:23:59 +00:00
frame_to_tx=[test_frame], c2_mode=FREEDV_MODE.datac13.value
)
2023-05-14 16:12:06 +00:00
def send_fec(self, mode, wakeup, payload, mycallsign):
2023-02-12 16:39:13 +00:00
"""Send an empty test frame"""
2023-05-14 19:55:17 +00:00
print(wakeup)
print(payload)
print(mycallsign)
mode_int = codec2.freedv_get_mode_value_by_name(mode)
payload_per_frame = modem.get_bytes_per_frame(mode_int) - 2
fec_payload_length = payload_per_frame - 1
2023-05-14 17:44:35 +00:00
if wakeup:
2023-05-14 19:55:17 +00:00
mode_int_wakeup = codec2.freedv_get_mode_value_by_name("sig0")
payload_per_wakeup_frame = modem.get_bytes_per_frame(mode_int_wakeup) - 2
fec_wakeup_frame = bytearray(payload_per_wakeup_frame)
fec_wakeup_frame[:1] = bytes([FR_TYPE.FEC_WAKEUP.value])
2023-05-14 16:12:06 +00:00
fec_wakeup_frame[1:7] = helpers.callsign_to_bytes(mycallsign)
2023-05-14 17:44:35 +00:00
fec_wakeup_frame[7:8] = bytes([mode_int])
fec_wakeup_frame[8:9] = bytes([1]) # n payload bursts
2023-05-14 19:55:17 +00:00
print(mode_int_wakeup)
2023-05-14 17:44:35 +00:00
self.enqueue_frame_for_tx(
2023-05-14 19:55:17 +00:00
frame_to_tx=[fec_wakeup_frame], c2_mode=codec2.FREEDV_MODE["sig1"].value
)
time.sleep(1)
2023-02-12 16:39:13 +00:00
fec_frame = bytearray(payload_per_frame)
fec_frame[:1] = bytes([FR_TYPE.FEC.value])
fec_frame[1:payload_per_frame] = bytes(payload[:fec_payload_length])
self.enqueue_frame_for_tx(
frame_to_tx=[fec_frame], c2_mode=codec2.FREEDV_MODE[mode].value
)
2023-02-12 17:24:42 +00:00
def send_fec_is_writing(self, mycallsign) -> None:
"""Send an fec is writing frame"""
2023-02-12 17:24:42 +00:00
fec_frame = bytearray(14)
fec_frame[:1] = bytes([FR_TYPE.IS_WRITING.value])
fec_frame[1:7] = helpers.callsign_to_bytes(mycallsign)
2023-02-12 18:07:59 +00:00
# send burst only if channel not busy - but without waiting
# otherwise burst will be dropped
2023-04-27 19:43:56 +00:00
if not ModemParam.channel_busy and not TNC.transmitting:
2023-02-12 18:07:59 +00:00
self.enqueue_frame_for_tx(
frame_to_tx=[fec_frame], c2_mode=codec2.FREEDV_MODE["sig0"].value
2023-02-12 18:07:59 +00:00
)
else:
return False
def save_data_to_folder(self,
transmission_uuid,
timestamp,
mycallsign,
dxcallsign,
dxgrid,
data_frame
):
2022-12-26 09:49:37 +00:00
"""
Save received data to folder
Also supports chat messages
"""
2022-12-26 09:49:37 +00:00
try:
2022-12-26 09:49:37 +00:00
self.log.info("[TNC] ARQ | RX | saving data to folder")
2022-12-26 09:35:58 +00:00
2022-12-26 09:49:37 +00:00
mycallsign = str(mycallsign, "UTF-8")
dxcallsign = str(dxcallsign, "UTF-8")
folder_path = "received"
if not os.path.exists(folder_path):
os.makedirs(folder_path)
callsign_path = f"{mycallsign}_{dxcallsign}"
if not os.path.exists(f"{folder_path}/{callsign_path}"):
os.makedirs(f"{folder_path}/{callsign_path}")
split_char = b"\0;\1;"
2022-12-28 17:03:05 +00:00
n_objects = 9
2022-12-26 09:49:37 +00:00
decoded_data = data_frame.split(split_char)
2022-12-28 16:37:50 +00:00
# if we have a false positive in case our split_char is available in data
# lets stick the data together, so we are not loosing it
if len(decoded_data) > n_objects:
2022-12-28 22:47:13 +00:00
file_data = b''.join(decoded_data[n_objects:])
2022-12-28 22:21:39 +00:00
# slice is crashing nuitka
# decoded_data = [*decoded_data[:n_objects], file_data]
decoded_data = decoded_data[:n_objects] + [file_data]
2022-12-26 09:49:37 +00:00
if decoded_data[0] in [b'm']:
2022-12-27 22:30:15 +00:00
checksum_delivered = str(decoded_data[2], "utf-8").lower()
2022-12-27 21:57:54 +00:00
# transmission_uuid = decoded_data[3]
message = decoded_data[5]
filename = decoded_data[6]
# filetype = decoded_data[7]
# timestamp = decoded_data[4]
data = decoded_data[8]
2022-12-26 09:49:37 +00:00
else:
message = b''
filename = b''
2022-12-26 09:49:37 +00:00
# save file to folder
2022-12-26 09:35:58 +00:00
if filename not in [b'', b'undefined']:
2022-12-27 21:58:18 +00:00
# doing crc check
2022-12-27 22:02:52 +00:00
crc = helpers.get_crc_32(data).hex().lower()
2022-12-27 21:57:54 +00:00
validity = checksum_delivered == crc
2022-12-27 22:04:57 +00:00
self.log.info(
2022-12-27 21:57:54 +00:00
"[TNC] ARQ | RX | checking data crc",
crc_delivered=checksum_delivered,
crc_calculated=crc,
valid=validity,
)
2022-12-26 09:35:58 +00:00
filename = str(filename, "UTF-8")
filename_complex = f"{timestamp}_{transmission_uuid}_{filename}"
with open(f"{folder_path}/{callsign_path}/{filename_complex}", "wb") as file:
file.write(data)
if message not in [b'', b'undefined']:
# save message to folder
message_name = f"{timestamp}_{transmission_uuid}_msg.txt"
with open(f"{folder_path}/{callsign_path}/{message_name}", "wb") as file:
file.write(message)
except Exception as e:
self.log.error("[TNC] error saving data to folder", e=e)