summaryrefslogtreecommitdiff
path: root/zto/controller/JSONDB.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'zto/controller/JSONDB.hpp')
-rw-r--r--zto/controller/JSONDB.hpp127
1 files changed, 83 insertions, 44 deletions
diff --git a/zto/controller/JSONDB.hpp b/zto/controller/JSONDB.hpp
index 00eeb8d..530f963 100644
--- a/zto/controller/JSONDB.hpp
+++ b/zto/controller/JSONDB.hpp
@@ -28,86 +28,125 @@
#include <stdexcept>
#include <vector>
#include <algorithm>
+#include <unordered_map>
#include "../node/Constants.hpp"
#include "../node/Utils.hpp"
+#include "../node/InetAddress.hpp"
+#include "../node/Mutex.hpp"
#include "../ext/json/json.hpp"
#include "../osdep/OSUtils.hpp"
+#include "../osdep/Http.hpp"
+#include "../osdep/Thread.hpp"
namespace ZeroTier {
/**
- * Hierarchical JSON store that persists into the filesystem
+ * Hierarchical JSON store that persists into the filesystem or via HTTP
*/
class JSONDB
{
public:
- JSONDB(const std::string &basePath) :
- _basePath(basePath)
+ struct NetworkSummaryInfo
{
- _reload(_basePath,std::string());
- }
+ NetworkSummaryInfo() : authorizedMemberCount(0),activeMemberCount(0),totalMemberCount(0),mostRecentDeauthTime(0) {}
+ std::vector<Address> activeBridges;
+ std::vector<InetAddress> allocatedIps;
+ unsigned long authorizedMemberCount;
+ unsigned long activeMemberCount;
+ unsigned long totalMemberCount;
+ uint64_t mostRecentDeauthTime;
+ };
- inline void reload()
- {
- _db.clear();
- _reload(_basePath,std::string());
- }
+ JSONDB(const std::string &basePath);
+ ~JSONDB();
bool writeRaw(const std::string &n,const std::string &obj);
- bool put(const std::string &n,const nlohmann::json &obj);
+ bool hasNetwork(const uint64_t networkId) const;
+
+ bool getNetwork(const uint64_t networkId,nlohmann::json &config) const;
- inline bool put(const std::string &n1,const std::string &n2,const nlohmann::json &obj) { return this->put((n1 + "/" + n2),obj); }
- inline bool put(const std::string &n1,const std::string &n2,const std::string &n3,const nlohmann::json &obj) { return this->put((n1 + "/" + n2 + "/" + n3),obj); }
- inline bool put(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4,const nlohmann::json &obj) { return this->put((n1 + "/" + n2 + "/" + n3 + "/" + n4),obj); }
- inline bool put(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4,const std::string &n5,const nlohmann::json &obj) { return this->put((n1 + "/" + n2 + "/" + n3 + "/" + n4 + "/" + n5),obj); }
+ bool getNetworkSummaryInfo(const uint64_t networkId,NetworkSummaryInfo &ns) const;
- const nlohmann::json &get(const std::string &n);
+ /**
+ * @return Bit mask: 0 == none, 1 == network only, 3 == network and member
+ */
+ int getNetworkAndMember(const uint64_t networkId,const uint64_t nodeId,nlohmann::json &networkConfig,nlohmann::json &memberConfig,NetworkSummaryInfo &ns) const;
- inline const nlohmann::json &get(const std::string &n1,const std::string &n2) { return this->get((n1 + "/" + n2)); }
- inline const nlohmann::json &get(const std::string &n1,const std::string &n2,const std::string &n3) { return this->get((n1 + "/" + n2 + "/" + n3)); }
- inline const nlohmann::json &get(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4) { return this->get((n1 + "/" + n2 + "/" + n3 + "/" + n4)); }
- inline const nlohmann::json &get(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4,const std::string &n5) { return this->get((n1 + "/" + n2 + "/" + n3 + "/" + n4 + "/" + n5)); }
+ bool getNetworkMember(const uint64_t networkId,const uint64_t nodeId,nlohmann::json &memberConfig) const;
- void erase(const std::string &n);
+ void saveNetwork(const uint64_t networkId,const nlohmann::json &networkConfig);
- inline void erase(const std::string &n1,const std::string &n2) { this->erase(n1 + "/" + n2); }
- inline void erase(const std::string &n1,const std::string &n2,const std::string &n3) { this->erase(n1 + "/" + n2 + "/" + n3); }
- inline void erase(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4) { this->erase(n1 + "/" + n2 + "/" + n3 + "/" + n4); }
- inline void erase(const std::string &n1,const std::string &n2,const std::string &n3,const std::string &n4,const std::string &n5) { this->erase(n1 + "/" + n2 + "/" + n3 + "/" + n4 + "/" + n5); }
+ void saveNetworkMember(const uint64_t networkId,const uint64_t nodeId,const nlohmann::json &memberConfig);
+
+ nlohmann::json eraseNetwork(const uint64_t networkId);
+
+ nlohmann::json eraseNetworkMember(const uint64_t networkId,const uint64_t nodeId,bool recomputeSummaryInfo = true);
+
+ std::vector<uint64_t> networkIds() const
+ {
+ std::vector<uint64_t> r;
+ Mutex::Lock _l(_networks_m);
+ for(std::unordered_map<uint64_t,_NW>::const_iterator n(_networks.begin());n!=_networks.end();++n)
+ r.push_back(n->first);
+ return r;
+ }
template<typename F>
- inline void filter(const std::string &prefix,F func)
+ inline void eachMember(const uint64_t networkId,F func)
{
- for(std::map<std::string,_E>::iterator i(_db.lower_bound(prefix));i!=_db.end();) {
- if ((i->first.length() >= prefix.length())&&(!memcmp(i->first.data(),prefix.data(),prefix.length()))) {
- if (!func(i->first,get(i->first))) {
- std::map<std::string,_E>::iterator i2(i); ++i2;
- this->erase(i->first);
- i = i2;
- } else ++i;
- } else break;
+ Mutex::Lock _l(_networks_m);
+ std::unordered_map<uint64_t,_NW>::const_iterator i(_networks.find(networkId));
+ if (i != _networks.end()) {
+ for(std::unordered_map< uint64_t,std::vector<uint8_t> >::const_iterator m(i->second.members.begin());m!=i->second.members.end();++m) {
+ try {
+ func(networkId,m->first,nlohmann::json::from_msgpack(m->second));
+ } catch ( ... ) {}
+ }
}
}
- inline bool operator==(const JSONDB &db) const { return ((_basePath == db._basePath)&&(_db == db._db)); }
- inline bool operator!=(const JSONDB &db) const { return (!(*this == db)); }
+ template<typename F>
+ inline void eachId(F func)
+ {
+ Mutex::Lock _l(_networks_m);
+ for(std::unordered_map<uint64_t,_NW>::const_iterator i(_networks.begin());i!=_networks.end();++i) {
+ for(std::unordered_map< uint64_t,std::vector<uint8_t> >::const_iterator m(i->second.members.begin());m!=i->second.members.end();++m) {
+ try {
+ func(i->first,m->first);
+ } catch ( ... ) {}
+ }
+ }
+ }
+
+ void threadMain()
+ throw();
private:
- void _reload(const std::string &p,const std::string &b);
- bool _isValidObjectName(const std::string &n);
+ bool _load(const std::string &p);
+ void _recomputeSummaryInfo(const uint64_t networkId);
std::string _genPath(const std::string &n,bool create);
- struct _E
+ std::string _basePath;
+ InetAddress _httpAddr;
+
+ Thread _summaryThread;
+ std::vector<uint64_t> _summaryThreadToDo;
+ volatile bool _summaryThreadRun;
+ Mutex _summaryThread_m;
+
+ struct _NW
{
- nlohmann::json obj;
- inline bool operator==(const _E &e) const { return (obj == e.obj); }
- inline bool operator!=(const _E &e) const { return (obj != e.obj); }
+ _NW() : summaryInfoLastComputed(0) {}
+ std::vector<uint8_t> config;
+ NetworkSummaryInfo summaryInfo;
+ uint64_t summaryInfoLastComputed;
+ std::unordered_map< uint64_t,std::vector<uint8_t> > members;
};
- std::string _basePath;
- std::map<std::string,_E> _db;
+ std::unordered_map<uint64_t,_NW> _networks;
+ Mutex _networks_m;
};
} // namespace ZeroTier
generated by cgit v1.2.3 (git 2.39.1) at 2025-12-30 16:23:41 +0000