hamnet70/impl/test/layer1/test_loopback.c

/*
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 * Copyright (C) 2024  Thomas Kolb
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <time.h>

#include <liquid/liquid.h>

#include "layer1/tx.h"
#include "layer1/rx.h"

#include "logger.h"

#include "utils.h"
#include "config.h"

#define RESULT_CHECK(stmt) { \
	result_t res = stmt; \
	if(res != OK) { \
		LOG(LVL_ERR, "Error %d in %s:%d!", res, __FILE__, __LINE__); \
		exit(1); \
	} \
}


void print_complex_array(const char *varname, float complex const *array, size_t len)
{
	fprintf(stderr, "%s=np.array([%f%+fj", varname, crealf(array[0]), cimagf(array[0]));
	for(size_t k = 1; k < len; k++) {
		fprintf(stderr, ", %f%+fj", crealf(array[k]), cimagf(array[k]));
	}
	fprintf(stderr, "])\n");
}


void cb_rx(rx_evt_t evt, const layer1_rx_t *rx, uint8_t *packet_data, size_t packet_len)
{
	(void)rx;

	unsigned int data_size;

	switch(evt)
	{
		case RX_EVT_DECODE_FAILED:
			LOG(LVL_ERR, "Received a message of %zu bytes, but decoding failed.", packet_len);
			LOG(LVL_INFO, "=== FAILED PAYLOAD ===");
			hexdump(packet_data, packet_len);
			LOG(LVL_INFO, "=======================");
			break;

		case RX_EVT_PACKET_RECEIVED:
			data_size = packet_len - crc_sizeof_key(PAYLOAD_CRC_SCHEME);

			LOG(LVL_INFO, "=== DECODED PAYLOAD (%4zu bytes) ===", packet_len);
			hexdump(packet_data, packet_len);
			LOG(LVL_INFO, "====================================");

			// FIXME: move to layer 2
			if(!crc_check_key(PAYLOAD_CRC_SCHEME, packet_data, data_size)) {
				LOG(LVL_WARN, "payload CRC check failed!");
				break;
			}

			LOG(LVL_INFO, "A message of %zu bytes was decoded successfully.", packet_len);
			break;

		case RX_EVT_HEADER_ERROR:
			LOG(LVL_ERR, "Header decoding failed!");
			break;

		case RX_EVT_PREAMBLE_FOUND:
			LOG(LVL_INFO, "Found preamble!");
			break;

		case RX_EVT_PACKET_DEBUG_INFO_COMPLETE:
			// FIXME: print debug info
			break;
	}
}


int main(void)
{
	layer1_tx_t tx;
	layer1_rx_t rx;

	// ** Initialize **

	logger_init();

	srand(time(NULL));

	RESULT_CHECK(layer1_tx_init(&tx));
	RESULT_CHECK(layer1_rx_init(&rx, cb_rx));


	// ** Generate packets **

	RESULT_CHECK(layer1_tx_reset(&tx));

	uint8_t packetbuf1[64] = "Hello World! 1234567890";
	size_t crc_size = crc_sizeof_key(PAYLOAD_CRC_SCHEME);
	size_t l = strlen((char*)packetbuf1);

	crc_append_key(PAYLOAD_CRC_SCHEME, packetbuf1, l);

	LOG(LVL_INFO, "Transmitting packet with %zd bytes.", l + crc_size);

	RESULT_CHECK(layer1_tx_add_packet_to_burst(&tx, packetbuf1, l + crc_size));

	uint8_t packetbuf2[128] = "This is just a test message. Just writing some stuff here to create a longer packet.";
	l = strlen((char*)packetbuf2);

	crc_append_key(PAYLOAD_CRC_SCHEME, packetbuf2, l);

	LOG(LVL_INFO, "Transmitting packet with %zd bytes.", l + crc_size);

	RESULT_CHECK(layer1_tx_add_packet_to_burst(&tx, packetbuf2, l + crc_size));

	RESULT_CHECK(layer1_tx_finalize_burst(&tx));

	size_t burst_len = layer1_tx_get_sample_count(&tx);
	const float complex *whole_burst = layer1_tx_get_sample_data(&tx);

	dump_array_cf(whole_burst, burst_len, 10e-6f, "/tmp/tx.cpx");

	// ** Simulate channel effects **
	channel_cccf ch = channel_cccf_create();
	channel_cccf_add_awgn(ch, -30, 30);
	//channel_cccf_add_carrier_offset(ch, 0.05f, 1.23f);

	float complex whole_burst_ch[burst_len];
	channel_cccf_execute_block(ch, (float complex*)whole_burst, burst_len, whole_burst_ch);

	dump_array_cf(whole_burst_ch, burst_len, 10e-6f, "/tmp/rx.cpx");


	// ** Receive signal **

	RESULT_CHECK(layer1_rx_process(&rx, whole_burst_ch, burst_len));

	// ** Cleanup **

	layer1_tx_shutdown(&tx);
	layer1_rx_shutdown(&rx);

	LOG(LVL_INFO, "Done.");
}