mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-09-09 08:12:52 +02:00
7f3b15011e
Muuuuch easier to use external dependencies now Also tried out conan and vcpkg. Ran into dependency issues when solving for packages to install with conan. vcpkg is just obtuse as all hell to install and not easy to integrate
110 lines
No EOL
3.1 KiB
C++
110 lines
No EOL
3.1 KiB
C++
/*
|
|
* Copyright (c)2025 ZeroTier, Inc.
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file in the project's root directory.
|
|
*
|
|
* Change Date: 2026-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2.0 of the Apache License.
|
|
*/
|
|
/****/
|
|
|
|
#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 <sw/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;
|
|
}
|
|
|
|
virtual void _memberChanged(nlohmann::json& old, nlohmann::json& memberConfig, bool notifyListeners);
|
|
|
|
virtual void _networkChanged(nlohmann::json& old, nlohmann::json& networkConfig, bool notifyListeners);
|
|
|
|
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);
|
|
}
|
|
};
|
|
|
|
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
|