From ab8ececbe70f7c83667d6ebb592fc1df17ad26a4 Mon Sep 17 00:00:00 2001 From: Moritz Warning Date: Sat, 20 Jan 2018 21:55:52 +0100 Subject: [PATCH] Revert "Do not serve controller requests until init is done." This reverts commit f4feccc6265cc480b84c85f897b225714072d4ec. --- controller/JSONDB.cpp | 20 +++++++------------- controller/JSONDB.hpp | 15 +++++++-------- 2 files changed, 14 insertions(+), 21 deletions(-) diff --git a/controller/JSONDB.cpp b/controller/JSONDB.cpp index d3e76fc1..007e0fec 100644 --- a/controller/JSONDB.cpp +++ b/controller/JSONDB.cpp @@ -26,8 +26,7 @@ static const nlohmann::json _EMPTY_JSON(nlohmann::json::object()); static const std::map _ZT_JSONDB_GET_HEADERS; JSONDB::JSONDB(const std::string &basePath) : - _basePath(basePath), - _ready(false) + _basePath(basePath) { if ((_basePath.length() > 7)&&(_basePath.substr(0,7) == "http://")) { // TODO: this doesn't yet support IPv6 since bracketed address notiation isn't supported. @@ -50,7 +49,7 @@ JSONDB::JSONDB(const std::string &basePath) : OSUtils::mkdir(_basePath.c_str()); OSUtils::lockDownFile(_basePath.c_str(),true); // networks might contain auth tokens, etc., so restrict directory permissions } - _ready = _reload(_basePath,std::string()); + _reload(_basePath,std::string()); } bool JSONDB::writeRaw(const std::string &n,const std::string &obj) @@ -84,13 +83,9 @@ bool JSONDB::put(const std::string &n,const nlohmann::json &obj) const nlohmann::json &JSONDB::get(const std::string &n) { - while (!_ready) { - Thread::sleep(250); - _ready = _reload(_basePath,std::string()); - } - if (!_isValidObjectName(n)) return _EMPTY_JSON; + std::map::iterator e(_db.find(n)); if (e != _db.end()) return e->second.obj; @@ -138,7 +133,7 @@ void JSONDB::erase(const std::string &n) _db.erase(n); } -bool JSONDB::_reload(const std::string &p,const std::string &b) +void JSONDB::_reload(const std::string &p,const std::string &b) { if (_httpAddr) { std::string body; @@ -155,11 +150,11 @@ bool JSONDB::_reload(const std::string &p,const std::string &b) _db[tmp].obj = i.value(); } } - return true; } - } catch ( ... ) {} // invalid JSON, so maybe incomplete request + } catch ( ... ) { + // TODO: report error? + } } - return false; } else { std::vector dl(OSUtils::listDirectory(p.c_str(),true)); for(std::vector::const_iterator di(dl.begin());di!=dl.end();++di) { @@ -169,7 +164,6 @@ bool JSONDB::_reload(const std::string &p,const std::string &b) this->_reload((p + ZT_PATH_SEPARATOR + *di),(b + *di + ZT_PATH_SEPARATOR)); } } - return true; } } diff --git a/controller/JSONDB.hpp b/controller/JSONDB.hpp index beafbaf5..c19112ed 100644 --- a/controller/JSONDB.hpp +++ b/controller/JSONDB.hpp @@ -36,7 +36,6 @@ #include "../ext/json/json.hpp" #include "../osdep/OSUtils.hpp" #include "../osdep/Http.hpp" -#include "../osdep/Thread.hpp" namespace ZeroTier { @@ -48,6 +47,12 @@ class JSONDB public: JSONDB(const std::string &basePath); + inline void reload() + { + _db.clear(); + _reload(_basePath,std::string()); + } + bool writeRaw(const std::string &n,const std::string &obj); bool put(const std::string &n,const nlohmann::json &obj); @@ -74,11 +79,6 @@ public: template inline void filter(const std::string &prefix,F func) { - while (!_ready) { - Thread::sleep(250); - _ready = _reload(_basePath,std::string()); - } - for(std::map::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))) { @@ -94,7 +94,7 @@ public: inline bool operator!=(const JSONDB &db) const { return (!(*this == db)); } private: - bool _reload(const std::string &p,const std::string &b); + void _reload(const std::string &p,const std::string &b); bool _isValidObjectName(const std::string &n); std::string _genPath(const std::string &n,bool create); @@ -108,7 +108,6 @@ private: InetAddress _httpAddr; std::string _basePath; std::map _db; - volatile bool _ready; }; } // namespace ZeroTier -- 2.15.1