You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
dvmhost/src/common/lookups/RadioIdLookup.cpp

312 lines
8.2 KiB

// SPDX-License-Identifier: GPL-2.0-only
/*
* Digital Voice Modem - Common Library
* GPLv2 Open Source. Use is subject to license terms.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* Copyright (C) 2016 Jonathan Naylor, G4KLX
* Copyright (C) 2017-2022,2025 Bryan Biedenkapp, N2PLL
* Copyright (c) 2024 Patrick McDonnell, W3AXL
*
*/
#include "lookups/RadioIdLookup.h"
#include "p25/P25Defines.h"
#include "Log.h"
using namespace lookups;
#include <cstdlib>
#include <string>
#include <vector>
#include <fstream>
// ---------------------------------------------------------------------------
// Static Class Members
// ---------------------------------------------------------------------------
std::mutex RadioIdLookup::s_mutex;
bool RadioIdLookup::s_locked = false;
// ---------------------------------------------------------------------------
// Macros
// ---------------------------------------------------------------------------
// Lock the table.
#define __LOCK_TABLE() \
std::lock_guard<std::mutex> lock(s_mutex); \
s_locked = true;
// Unlock the table.
#define __UNLOCK_TABLE() s_locked = false;
// Spinlock wait for table to be read unlocked.
#define __SPINLOCK() \
if (s_locked) { \
while (s_locked) \
Thread::sleep(2U); \
}
// ---------------------------------------------------------------------------
// Public Class Members
// ---------------------------------------------------------------------------
/* Initializes a new instance of the RadioIdLookup class. */
RadioIdLookup::RadioIdLookup(const std::string& filename, uint32_t reloadTime, bool ridAcl, bool verbose) : LookupTable(filename, reloadTime),
m_acl(ridAcl),
m_verbose(verbose)
{
/* stub */
}
/* Clears all entries from the lookup table. */
void RadioIdLookup::clear()
{
__LOCK_TABLE();
m_table.clear();
__UNLOCK_TABLE();
}
/* Toggles the specified radio ID enabled or disabled. */
void RadioIdLookup::toggleEntry(uint32_t id, bool enabled)
{
RadioId rid = find(id);
addEntry(id, enabled, rid.radioAlias());
}
/* Adds a new entry to the lookup table by the specified unique ID. */
void RadioIdLookup::addEntry(uint32_t id, bool enabled, const std::string& alias, const std::string& ipAddress)
{
if ((id == p25::defines::WUID_ALL) || (id == p25::defines::WUID_FNE)) {
return;
}
RadioId entry = RadioId(enabled, false, alias, ipAddress);
__LOCK_TABLE();
try {
RadioId _entry = m_table.at(id);
// if either the alias or the enabled flag doesn't match, update the entry
if (_entry.radioEnabled() != enabled || _entry.radioAlias() != alias) {
//LogDebug(LOG_HOST, "Updating existing RID %d (%s) in ACL", id, alias.c_str());
_entry = RadioId(enabled, false, alias, ipAddress);
m_table[id] = _entry;
} else {
//LogDebug(LOG_HOST, "No changes made to RID %d (%s) in ACL", id, alias.c_str());
}
} catch (...) {
//LogDebug(LOG_HOST, "Adding new RID %d (%s) to ACL", id, alias.c_str());
m_table[id] = entry;
}
__UNLOCK_TABLE();
}
/* Erases an existing entry from the lookup table by the specified unique ID. */
void RadioIdLookup::eraseEntry(uint32_t id)
{
__LOCK_TABLE();
try {
RadioId entry = m_table.at(id); // this value will get discarded
(void)entry; // but some variants of C++ mark the unordered_map<>::at as nodiscard
m_table.erase(id);
}
catch (...) {
/* stub */
}
__UNLOCK_TABLE();
}
/* Finds a table entry in this lookup table. */
RadioId RadioIdLookup::find(uint32_t id)
{
RadioId entry;
if ((id == p25::defines::WUID_ALL) || (id == p25::defines::WUID_FNE)) {
return RadioId(true, false);
}
__SPINLOCK();
try {
entry = m_table.at(id);
} catch (...) {
entry = RadioId(false, true);
}
return entry;
}
/* Saves loaded talkgroup rules. */
void RadioIdLookup::commit(bool quiet)
{
save(quiet);
}
/* Flag indicating whether radio ID access control is enabled or not. */
bool RadioIdLookup::getACL()
{
return m_acl;
}
// ---------------------------------------------------------------------------
// Private Class Members
// ---------------------------------------------------------------------------
/* Loads the table from the passed lookup table file. */
bool RadioIdLookup::load()
{
if (m_filename.empty()) {
return false;
}
std::ifstream file (m_filename, std::ifstream::in);
if (file.fail()) {
LogError(LOG_HOST, "Cannot open the radio ID lookup file - %s", m_filename.c_str());
return false;
}
// clear table
clear();
__LOCK_TABLE();
// read lines from file
std::string line;
while (std::getline(file, line)) {
if (line.length() > 0) {
// Skip comments with #
if (line.at(0) == '#')
continue;
// tokenize line
std::vector<std::string> parsed;
std::stringstream ss(line);
std::string field;
char delim = ',';
while (std::getline(ss, field, delim))
parsed.push_back(field);
// ensure we have at least 2 fields
if (parsed.size() < 2) {
LogError(LOG_HOST, "Invalid entry in radio ID lookup table - %s", line.c_str());
continue;
}
// parse tokenized line
uint32_t id = ::atoi(parsed[0].c_str());
bool radioEnabled = ::atoi(parsed[1].c_str()) == 1;
std::string alias = "";
std::string ipAddress = "";
// check for an optional alias field
if (parsed.size() >= 3) {
alias = parsed[2];
}
// check for an optional IP address field
if (parsed.size() >= 4) {
ipAddress = parsed[3];
}
m_table[id] = RadioId(radioEnabled, false, alias, ipAddress);
if (m_verbose) {
LogInfoEx(LOG_HOST, "Radio NAME: %s RID: %u ENABLED: %u IPADDR: %s", alias.c_str(), id, radioEnabled, ipAddress.c_str());
}
}
}
file.close();
__UNLOCK_TABLE();
size_t size = m_table.size();
if (size == 0U)
return false;
uint64_t now = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count();
m_lastLoadTime = now;
LogInfoEx(LOG_HOST, "Loaded %lu entries into radio ID lookup table", size);
return true;
}
/* Saves the table to the passed lookup table file. */
bool RadioIdLookup::save(bool quiet)
{
if (m_filename.empty()) {
return false;
}
std::ofstream file (m_filename, std::ofstream::out);
if (file.fail()) {
LogError(LOG_HOST, "Cannot open the radio ID lookup file - %s", m_filename.c_str());
return false;
}
if (!quiet)
LogInfoEx(LOG_HOST, "Saving RID lookup file to %s", m_filename.c_str());
// Counter for lines written
unsigned int lines = 0;
std::lock_guard<std::mutex> lock(s_mutex);
// String for writing
std::string line;
// iterate over each entry in the RID lookup and write it to the open file
for (auto& entry: m_table) {
// get the parameters
uint32_t rid = entry.first;
bool enabled = entry.second.radioEnabled();
std::string alias = entry.second.radioAlias();
std::string ipAddress = entry.second.radioIPAddress();
// format into a string
line = std::to_string(rid) + "," + std::to_string(enabled) + ",";
// add the alias if we have one
if (alias.length() > 0) {
line += alias;
line += ",";
}
// add the IP address if we have one
if (ipAddress.length() > 0) {
line += ipAddress;
line += ",";
}
line += "\n";
file << line;
lines++;
}
file.close();
if (lines != m_table.size())
return false;
if (!quiet)
LogInfoEx(LOG_HOST, "Saved %u entries to lookup table file %s", lines, m_filename.c_str());
return true;
}

Powered by TurnKey Linux.