Tom Early
6 years ago
parent
f6643f30eb
commit
3e17f21891
@ -0,0 +1,185 @@
|
||||
/*
|
||||
* Copyright (c) 2020 by Thomas A. Early N7TAE
|
||||
*
|
||||
* 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 "CacheManager.h"
|
||||
|
||||
void CCacheManager::findUserData(const std::string &user, std::string &rptr, std::string &gate, std::string &addr)
|
||||
{
|
||||
mux.lock();
|
||||
rptr.assign(findUserRptr(user));
|
||||
gate.assign(findRptrGate(rptr));
|
||||
addr.assign(findGateAddr(gate));
|
||||
mux.unlock();
|
||||
}
|
||||
|
||||
void CCacheManager::findRptrData(const std::string &rptr, std::string &gate, std::string &addr)
|
||||
{
|
||||
mux.lock();
|
||||
gate.assign(findRptrGate(rptr));
|
||||
addr.assign(findGateAddr(gate));
|
||||
mux.unlock();
|
||||
}
|
||||
|
||||
std::string CCacheManager::findUserAddr(const std::string &user)
|
||||
{
|
||||
mux.lock();
|
||||
std::string addr(findGateAddr(findRptrGate(findUserRptr(user))));
|
||||
mux.unlock();
|
||||
|
||||
return addr;
|
||||
}
|
||||
|
||||
std::string CCacheManager::findUserTime(const std::string &user)
|
||||
{
|
||||
std::string utime;
|
||||
if (user.empty())
|
||||
return utime;
|
||||
mux.lock();
|
||||
auto itt = UserTime.find(user);
|
||||
if (itt != UserTime.end())
|
||||
utime.assign(itt->second);
|
||||
mux.unlock();
|
||||
return utime;
|
||||
}
|
||||
|
||||
std::string CCacheManager::findUserRepeater(const std::string &user)
|
||||
{
|
||||
mux.lock();
|
||||
std::string rptr(findUserRptr(user));
|
||||
mux.unlock();
|
||||
return rptr;
|
||||
}
|
||||
|
||||
std::string CCacheManager::findGateAddress(const std::string &gate)
|
||||
{
|
||||
mux.lock();
|
||||
std::string addr(findGateAddr(gate));
|
||||
mux.unlock();
|
||||
return addr;
|
||||
}
|
||||
|
||||
void CCacheManager::updateUser(const std::string &user, const std::string &rptr, const std::string &gate, const std::string &addr, const std::string &time)
|
||||
{
|
||||
if (user.empty())
|
||||
return;
|
||||
|
||||
mux.lock();
|
||||
if (! time.empty())
|
||||
UserTime[user] = time;
|
||||
|
||||
if (rptr.empty()) {
|
||||
mux.unlock();
|
||||
return;
|
||||
}
|
||||
|
||||
UserRptr[user] = rptr;
|
||||
|
||||
if (gate.empty() || addr.empty()) {
|
||||
mux.unlock();
|
||||
return;
|
||||
}
|
||||
|
||||
if (rptr.compare(0, 7, gate, 0, 7))
|
||||
RptrGate[rptr] = gate; // only do this if they differ
|
||||
|
||||
if (addr.npos == addr.find(':'))
|
||||
GateIPV4[gate] = addr;
|
||||
else
|
||||
GateIPV6[gate] = addr;
|
||||
mux.unlock();
|
||||
}
|
||||
|
||||
void CCacheManager::updateRptr(const std::string &rptr, const std::string &gate, const std::string &addr)
|
||||
{
|
||||
if (rptr.empty() || gate.empty())
|
||||
return;
|
||||
|
||||
mux.lock();
|
||||
RptrGate[rptr] = gate;
|
||||
if (addr.empty()) {
|
||||
mux.unlock();
|
||||
return;
|
||||
}
|
||||
if (addr.npos == addr.find(':'))
|
||||
GateIPV4[gate] = addr;
|
||||
else
|
||||
GateIPV6[gate] = addr;
|
||||
mux.unlock();
|
||||
}
|
||||
|
||||
void CCacheManager::updateGate(const std::string &G, const std::string &addr)
|
||||
{
|
||||
if (G.empty() || addr.empty())
|
||||
return;
|
||||
std::string gate(G);
|
||||
auto p = gate.find('_');
|
||||
while (gate.npos != p) {
|
||||
gate[p] = ' ';
|
||||
p = gate.find('_');
|
||||
}
|
||||
mux.lock();
|
||||
if (addr.npos == addr.find(':'))
|
||||
GateIPV4[gate] = addr;
|
||||
else
|
||||
GateIPV6[gate] = addr;
|
||||
mux.unlock();
|
||||
}
|
||||
|
||||
// these last three functions are private and not mux locked.
|
||||
std::string CCacheManager::findUserRptr(const std::string &user)
|
||||
{
|
||||
std::string rptr;
|
||||
if (user.empty())
|
||||
return rptr;
|
||||
auto it = UserRptr.find(user);
|
||||
if (it != UserRptr.end())
|
||||
rptr.assign(it->second);
|
||||
return rptr;
|
||||
}
|
||||
|
||||
std::string CCacheManager::findRptrGate(const std::string &rptr)
|
||||
{
|
||||
std::string gate;
|
||||
if (rptr.empty())
|
||||
return gate;
|
||||
auto it = RptrGate.find(rptr);
|
||||
if (it == RptrGate.end()) {
|
||||
gate.assign(rptr);
|
||||
gate[7] = 'G';
|
||||
} else
|
||||
gate.assign(it->second);
|
||||
return gate;
|
||||
}
|
||||
|
||||
std::string CCacheManager::findGateAddr(const std::string &gate)
|
||||
{
|
||||
std::string addr;
|
||||
if (gate.empty())
|
||||
return addr;
|
||||
auto it6 = GateIPV6.find(gate);
|
||||
if (it6 == GateIPV6.end()) {
|
||||
auto it4 = GateIPV4.find(gate);
|
||||
if (it4 == GateIPV4.end()) {
|
||||
return addr;
|
||||
} else {
|
||||
addr.assign(it4->second);
|
||||
}
|
||||
} else
|
||||
addr.assign(it6->second);
|
||||
return addr;
|
||||
}
|
||||
@ -0,0 +1,55 @@
|
||||
/*
|
||||
* Copyright (c) 2020 by Thomas A. Early N7TAE
|
||||
*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <string>
|
||||
#include <mutex>
|
||||
#include <unordered_map>
|
||||
|
||||
class CCacheManager {
|
||||
public:
|
||||
CCacheManager() {}
|
||||
~CCacheManager() {}
|
||||
|
||||
// the bodies of these public functions need to be mux locked to access the maps and the private functions.
|
||||
// for these find functions, if a map value can't be found the returned string will be empty.
|
||||
void findUserData(const std::string &user, std::string &rptr, std::string &gate, std::string &addr);
|
||||
void findRptrData(const std::string &rptr, std::string &gate, std::string &addr);
|
||||
std::string findUserTime(const std::string &user);
|
||||
std::string findUserAddr(const std::string &user);
|
||||
std::string findUserRepeater(const std::string &user);
|
||||
std::string findGateAddress(const std::string &gate);
|
||||
|
||||
void updateUser(const std::string &user, const std::string &rptr, const std::string &gate, const std::string &addr, const std::string &time);
|
||||
void updateRptr(const std::string &rptr, const std::string &gate, const std::string &addr);
|
||||
void updateGate(const std::string &gate, const std::string &addr);
|
||||
|
||||
private:
|
||||
// these three functions aren't mux locked, that's why they're private
|
||||
std::string findUserRptr(const std::string &user);
|
||||
std::string findRptrGate(const std::string &rptr);
|
||||
std::string findGateAddr(const std::string &gate);
|
||||
|
||||
std::unordered_map<std::string, std::string> UserTime;
|
||||
std::unordered_map<std::string, std::string> UserRptr;
|
||||
std::unordered_map<std::string, std::string> RptrGate;
|
||||
std::unordered_map<std::string, std::string> GateIPV4;
|
||||
std::unordered_map<std::string, std::string> GateIPV6;
|
||||
std::mutex mux;
|
||||
};
|
||||
Loading…
Reference in new issue