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.
99 lines
2.7 KiB
99 lines
2.7 KiB
/* (c) ZeroTier, Inc. |
|
* See LICENSE.txt in nonfree/ |
|
*/ |
|
|
|
#include "DB.hpp" |
|
|
|
#ifdef ZT_CONTROLLER_USE_LIBPQ |
|
|
|
#ifndef ZT_CONTROLLER_CV2_HPP |
|
#define ZT_CONTROLLER_CV2_HPP |
|
|
|
#define ZT_CENTRAL_CONTROLLER_COMMIT_THREADS 4 |
|
|
|
#include "../../node/Metrics.hpp" |
|
#include "ConnectionPool.hpp" |
|
#include "PostgreSQL.hpp" |
|
|
|
#include <memory> |
|
#include <pqxx/pqxx> |
|
#include <redis++/redis++.h> |
|
|
|
namespace ZeroTier { |
|
class CV2 : public DB { |
|
friend class MemberNotificationReceiver<CV2>; |
|
friend class NetworkNotificationReceiver<CV2>; |
|
|
|
public: |
|
CV2(const Identity& myId, const char* path, int listenPort); |
|
virtual ~CV2(); |
|
|
|
virtual bool waitForReady(); |
|
virtual bool isReady(); |
|
virtual bool save(nlohmann::json& record, bool notifyListeners); |
|
virtual void eraseNetwork(const uint64_t networkId); |
|
virtual void eraseMember(const uint64_t networkId, const uint64_t memberId); |
|
virtual void nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress& physicalAddress); |
|
virtual void nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress& physicalAddress, const char* osArch); |
|
virtual AuthInfo getSSOAuthInfo(const nlohmann::json& member, const std::string& redirectURL); |
|
|
|
virtual bool ready() |
|
{ |
|
return _ready == 2; |
|
} |
|
|
|
protected: |
|
struct _PairHasher { |
|
inline std::size_t operator()(const std::pair<uint64_t, uint64_t>& p) const |
|
{ |
|
return (std::size_t)(p.first ^ p.second); |
|
} |
|
}; |
|
virtual void _memberChanged(nlohmann::json& old, nlohmann::json& memberConfig, bool notifyListeners); |
|
|
|
virtual void _networkChanged(nlohmann::json& old, nlohmann::json& networkConfig, bool notifyListeners); |
|
|
|
private: |
|
void initializeNetworks(); |
|
void initializeMembers(); |
|
void heartbeat(); |
|
void membersDbWatcher(); |
|
void networksDbWatcher(); |
|
|
|
void commitThread(); |
|
void onlineNotificationThread(); |
|
|
|
// void notifyNewMember(const std::string &networkID, const std::string &memberID); |
|
|
|
enum OverrideMode { ALLOW_PGBOUNCER_OVERRIDE = 0, NO_OVERRIDE = 1 }; |
|
|
|
std::shared_ptr<ConnectionPool<PostgresConnection> > _pool; |
|
|
|
const Identity _myId; |
|
const Address _myAddress; |
|
std::string _myAddressStr; |
|
std::string _connString; |
|
|
|
BlockingQueue<std::pair<nlohmann::json, bool> > _commitQueue; |
|
|
|
std::thread _heartbeatThread; |
|
std::thread _membersDbWatcher; |
|
std::thread _networksDbWatcher; |
|
std::thread _commitThread[ZT_CENTRAL_CONTROLLER_COMMIT_THREADS]; |
|
std::thread _onlineNotificationThread; |
|
|
|
std::unordered_map<std::pair<uint64_t, uint64_t>, NodeOnlineRecord, _PairHasher> _lastOnline; |
|
|
|
mutable std::mutex _lastOnline_l; |
|
mutable std::mutex _readyLock; |
|
std::atomic<int> _ready, _connected, _run; |
|
mutable volatile bool _waitNoticePrinted; |
|
|
|
int _listenPort; |
|
uint8_t _ssoPsk[48]; |
|
}; |
|
|
|
} // namespace ZeroTier |
|
|
|
#endif // ZT_CONTROLLER_CV2_HPP |
|
#endif // ZT_CONTROLLER_USE_LIBPQ
|
|
|