P25Clients/P25Gateway/P25Network.cpp

/*
 *   Copyright (C) 2009-2014,2016,2020,2024,2025 by Jonathan Naylor G4KLX
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include "P25Network.h"
#include "Utils.h"
#include "Log.h"

#include <cstdio>
#include <cassert>
#include <cstring>

CP25Network::CP25Network(unsigned short port, const std::string& callsign, bool debug) :
m_callsign(callsign),
m_socket4(nullptr),
m_socket6(nullptr),
m_debug(debug)
{
	assert(port > 0U);

	m_socket4 = new CUDPSocket(port);
	m_socket6 = new CUDPSocket(port);

	m_callsign.resize(10U, ' ');
}

CP25Network::~CP25Network()
{
	delete m_socket4;
	delete m_socket6;
}

bool CP25Network::open()
{
	LogInfo("Opening P25 network connection");

	sockaddr_storage addr4;
	addr4.ss_family = AF_INET;

	bool ret4 = m_socket4->open(addr4);
	if (!ret4) {
		delete m_socket4;
		m_socket4 = nullptr;
	}

	sockaddr_storage addr6;
	addr6.ss_family = AF_INET6;

	bool ret6 = m_socket6->open(addr6);
	if (!ret6) {
		delete m_socket6;
		m_socket6 = nullptr;
	}

	if (!ret4 && !ret6) {
		LogError("Unable to open an IPv4 or an IPv6 socket");
		return false;
	}

	return true;
}

bool CP25Network::write(const unsigned char* data, unsigned int length, const CP25Reflector& address)
{
	assert(data != nullptr);
	assert(length > 0U);

	if (m_debug)
		CUtils::dump(1U, "P25 Network Data Sent", data, length);

	if (address.hasIPv6() && hasIPv6()) {
		return m_socket6->write(data, length , address.IPv6.m_addr, address.IPv6.m_addrLen);
	} else if (address.hasIPv4() && hasIPv4()) {
		return m_socket4->write(data, length, address.IPv4.m_addr, address.IPv4.m_addrLen);
	} else {
		LogError("No suitable IP address to write data to TG%u", address.m_id);
		return false;
	}
}

bool CP25Network::poll(const CP25Reflector& address)
{
	unsigned char data[15U];

	data[0U] = 0xF0U;

	for (unsigned int i = 0U; i < 10U; i++)
		data[i + 1U] = m_callsign.at(i);

	if (m_debug)
		CUtils::dump(1U, "P25 Network Poll Sent", data, 11U);

	if (address.hasIPv6() && hasIPv6()) {
		return m_socket6->write(data, 11U, address.IPv6.m_addr, address.IPv6.m_addrLen);
	} else if (address.hasIPv4() && hasIPv4()) {
		return m_socket4->write(data, 11U, address.IPv4.m_addr, address.IPv4.m_addrLen);
	} else {
		LogError("No suitable IP address to poll TG%u", address.m_id);
		return false;
	}
}

bool CP25Network::unlink(const CP25Reflector& address)
{
	unsigned char data[15U];

	data[0U] = 0xF1U;

	for (unsigned int i = 0U; i < 10U; i++)
		data[i + 1U] = m_callsign.at(i);

	if (m_debug)
		CUtils::dump(1U, "P25 Network Unlink Sent", data, 11U);

	if (address.hasIPv6() && hasIPv6()) {
		return m_socket6->write(data, 11U, address.IPv6.m_addr, address.IPv6.m_addrLen);
	} else if (address.hasIPv4() && hasIPv4()) {
		return m_socket4->write(data, 11U, address.IPv4.m_addr, address.IPv4.m_addrLen);
	} else {
		LogError("No suitable IP address to unlink from TG%u", address.m_id);
		return false;
	}
}

unsigned int CP25Network::read(unsigned char* data, unsigned int length, sockaddr_storage& addr, unsigned int& addrLen)
{
	assert(data != nullptr);
	assert(length > 0U);

	if (hasIPv4()) {
		int len = m_socket4->read(data, length, addr, addrLen);
		if (len > 0) {
			if (m_debug)
				CUtils::dump(1U, "P25 Network Data Received", data, len);
			return len;
		}
	}

	if (hasIPv6()) {
		int len = m_socket6->read(data, length, addr, addrLen);
		if (len > 0) {
			if (m_debug)
				CUtils::dump(1U, "P25 Network Data Received", data, len);
			return len;
		}
	}

	return 0U;
}

void CP25Network::close()
{
	if (hasIPv4()) {
		m_socket4->close();
		delete m_socket4;
		m_socket4 = nullptr;
	}

	if (hasIPv6()) {
		m_socket6->close();
		delete m_socket6;
		m_socket6 = nullptr;
	}

	LogInfo("Closing P25 network connection");
}

bool CP25Network::hasIPv4() const
{
	return m_socket4 != nullptr;
}

bool CP25Network::hasIPv6() const
{
	return m_socket6 != nullptr;
}

bool CP25Network::match(const sockaddr_storage& address, const CP25Reflector& reflector)
{
	switch (address.ss_family) {
		case AF_INET:
			if (!reflector.hasIPv4())
				return false;
			return CUDPSocket::match(address, reflector.IPv4.m_addr);

		case AF_INET6:
			if (!reflector.hasIPv6())
				return false;
			return CUDPSocket::match(address, reflector.IPv6.m_addr);

		default:
			LogError("Trying to match an unknown protocol family - %d", address.ss_family);
			return false;
	}
}