2018-08-31 21:58:15 +00:00
|
|
|
/*
|
2019-08-23 16:23:39 +00:00
|
|
|
* Copyright (c)2019 ZeroTier, Inc.
|
2018-08-31 21:58:15 +00:00
|
|
|
*
|
2019-08-23 16:23:39 +00:00
|
|
|
* Use of this software is governed by the Business Source License included
|
|
|
|
* in the LICENSE.TXT file in the project's root directory.
|
2018-08-31 21:58:15 +00:00
|
|
|
*
|
2020-08-20 19:51:39 +00:00
|
|
|
* Change Date: 2025-01-01
|
2018-08-31 21:58:15 +00:00
|
|
|
*
|
2019-08-23 16:23:39 +00:00
|
|
|
* 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.
|
2018-08-31 21:58:15 +00:00
|
|
|
*/
|
2019-08-23 16:23:39 +00:00
|
|
|
/****/
|
2019-08-06 13:51:23 +00:00
|
|
|
|
|
|
|
#include "DB.hpp"
|
2019-08-06 12:51:50 +00:00
|
|
|
|
2018-08-31 21:58:15 +00:00
|
|
|
#ifdef ZT_CONTROLLER_USE_LIBPQ
|
|
|
|
|
|
|
|
#ifndef ZT_CONTROLLER_LIBPQ_HPP
|
|
|
|
#define ZT_CONTROLLER_LIBPQ_HPP
|
|
|
|
|
2018-12-06 21:08:31 +00:00
|
|
|
#define ZT_CENTRAL_CONTROLLER_COMMIT_THREADS 4
|
|
|
|
|
2021-06-02 18:44:00 +00:00
|
|
|
#include "ConnectionPool.hpp"
|
|
|
|
#include <pqxx/pqxx>
|
|
|
|
|
2020-05-12 18:56:19 +00:00
|
|
|
#include <memory>
|
2020-05-11 23:02:49 +00:00
|
|
|
#include <redis++/redis++.h>
|
|
|
|
|
2018-08-31 21:58:15 +00:00
|
|
|
extern "C" {
|
2019-08-06 12:51:50 +00:00
|
|
|
typedef struct pg_conn PGconn;
|
2018-08-31 21:58:15 +00:00
|
|
|
}
|
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
namespace ZeroTier {
|
2018-08-31 21:58:15 +00:00
|
|
|
|
2020-05-11 22:03:56 +00:00
|
|
|
struct RedisConfig;
|
|
|
|
|
2021-06-02 18:44:00 +00:00
|
|
|
|
|
|
|
class PostgresConnection : public Connection {
|
|
|
|
public:
|
|
|
|
virtual ~PostgresConnection() {
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<pqxx::connection> c;
|
|
|
|
int a;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
class PostgresConnFactory : public ConnectionFactory {
|
|
|
|
public:
|
|
|
|
PostgresConnFactory(std::string &connString)
|
|
|
|
: m_connString(connString)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual std::shared_ptr<Connection> create() {
|
|
|
|
auto c = std::shared_ptr<PostgresConnection>(new PostgresConnection());
|
|
|
|
c->c = std::make_shared<pqxx::connection>(m_connString);
|
|
|
|
return std::static_pointer_cast<Connection>(c);
|
|
|
|
}
|
|
|
|
private:
|
|
|
|
std::string m_connString;
|
|
|
|
};
|
|
|
|
|
|
|
|
class PostgreSQL;
|
|
|
|
|
|
|
|
class MemberNotificationReceiver : public pqxx::notification_receiver {
|
|
|
|
public:
|
|
|
|
MemberNotificationReceiver(PostgreSQL *p, pqxx::connection &c, const std::string &channel);
|
2021-06-02 20:46:54 +00:00
|
|
|
virtual ~MemberNotificationReceiver() {
|
|
|
|
fprintf(stderr, "MemberNotificationReceiver destroyed\n");
|
|
|
|
}
|
2021-06-02 18:44:00 +00:00
|
|
|
|
|
|
|
virtual void operator() (const std::string &payload, int backendPid);
|
|
|
|
private:
|
|
|
|
PostgreSQL *_psql;
|
|
|
|
};
|
|
|
|
|
|
|
|
class NetworkNotificationReceiver : public pqxx::notification_receiver {
|
|
|
|
public:
|
|
|
|
NetworkNotificationReceiver(PostgreSQL *p, pqxx::connection &c, const std::string &channel);
|
2021-06-02 20:46:54 +00:00
|
|
|
virtual ~NetworkNotificationReceiver() {
|
|
|
|
fprintf(stderr, "NetworkNotificationReceiver destroyed\n");
|
|
|
|
};
|
2021-06-02 18:44:00 +00:00
|
|
|
|
|
|
|
virtual void operator() (const std::string &payload, int packend_pid);
|
|
|
|
private:
|
|
|
|
PostgreSQL *_psql;
|
|
|
|
};
|
|
|
|
|
2018-08-31 21:58:15 +00:00
|
|
|
/**
|
|
|
|
* A controller database driver that talks to PostgreSQL
|
|
|
|
*
|
|
|
|
* This is for use with ZeroTier Central. Others are free to build and use it
|
2021-06-02 18:44:00 +00:00
|
|
|
* but be aware that we might change it at any time.
|
2018-08-31 21:58:15 +00:00
|
|
|
*/
|
|
|
|
class PostgreSQL : public DB
|
|
|
|
{
|
2021-06-02 18:44:00 +00:00
|
|
|
friend class MemberNotificationReceiver;
|
|
|
|
friend class NetworkNotificationReceiver;
|
2018-08-31 21:58:15 +00:00
|
|
|
public:
|
2020-05-11 22:03:56 +00:00
|
|
|
PostgreSQL(const Identity &myId, const char *path, int listenPort, RedisConfig *rc);
|
2019-08-06 12:51:50 +00:00
|
|
|
virtual ~PostgreSQL();
|
2018-08-31 21:58:15 +00:00
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
virtual bool waitForReady();
|
|
|
|
virtual bool isReady();
|
2019-08-06 15:42:54 +00:00
|
|
|
virtual bool save(nlohmann::json &record,bool notifyListeners);
|
2019-08-06 12:51:50 +00:00
|
|
|
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);
|
2021-05-28 21:08:24 +00:00
|
|
|
virtual void updateMemberOnLoad(const uint64_t networkId, const uint64_t memberId, nlohmann::json &member);
|
2018-08-31 21:58:15 +00:00
|
|
|
|
|
|
|
protected:
|
2019-08-06 12:51:50 +00:00
|
|
|
struct _PairHasher
|
2018-08-31 21:58:15 +00:00
|
|
|
{
|
|
|
|
inline std::size_t operator()(const std::pair<uint64_t,uint64_t> &p) const { return (std::size_t)(p.first ^ p.second); }
|
|
|
|
};
|
2021-06-02 20:46:11 +00:00
|
|
|
virtual void _memberChanged(nlohmann::json &old,nlohmann::json &memberConfig,bool notifyListeners) {
|
|
|
|
fprintf(stderr, "PostgreSQL::_memberChanged\n");
|
2021-06-02 18:44:00 +00:00
|
|
|
DB::_memberChanged(old, memberConfig, notifyListeners);
|
|
|
|
}
|
|
|
|
|
2021-06-02 20:46:11 +00:00
|
|
|
virtual void _networkChanged(nlohmann::json &old,nlohmann::json &networkConfig,bool notifyListeners) {
|
|
|
|
fprintf(stderr, "PostgreSQL::_networkChanged\n");
|
|
|
|
DB::_networkChanged(old, networkConfig, notifyListeners);
|
2021-06-02 18:44:00 +00:00
|
|
|
}
|
2018-08-31 21:58:15 +00:00
|
|
|
|
|
|
|
private:
|
2021-06-02 18:44:00 +00:00
|
|
|
void initializeNetworks();
|
|
|
|
void initializeMembers();
|
2019-08-06 12:51:50 +00:00
|
|
|
void heartbeat();
|
|
|
|
void membersDbWatcher();
|
2021-06-02 18:44:00 +00:00
|
|
|
void _membersWatcher_Postgres();
|
2019-08-06 12:51:50 +00:00
|
|
|
void networksDbWatcher();
|
2021-06-02 18:44:00 +00:00
|
|
|
void _networksWatcher_Postgres();
|
2019-08-06 12:51:50 +00:00
|
|
|
|
2020-05-12 18:56:19 +00:00
|
|
|
void _membersWatcher_Redis();
|
2020-05-11 18:48:05 +00:00
|
|
|
void _networksWatcher_Redis();
|
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
void commitThread();
|
|
|
|
void onlineNotificationThread();
|
2020-05-29 02:22:07 +00:00
|
|
|
void onlineNotification_Postgres();
|
2020-05-14 00:23:27 +00:00
|
|
|
void onlineNotification_Redis();
|
|
|
|
void _doRedisUpdate(sw::redis::Transaction &tx, std::string &controllerId,
|
|
|
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > &lastOnline);
|
2019-08-06 12:51:50 +00:00
|
|
|
|
|
|
|
enum OverrideMode {
|
|
|
|
ALLOW_PGBOUNCER_OVERRIDE = 0,
|
|
|
|
NO_OVERRIDE = 1
|
|
|
|
};
|
2018-12-03 23:19:15 +00:00
|
|
|
|
2021-06-02 18:44:00 +00:00
|
|
|
std::shared_ptr<ConnectionPool<PostgresConnection> > _pool;
|
2018-12-03 23:19:15 +00:00
|
|
|
|
2019-08-06 15:42:54 +00:00
|
|
|
const Identity _myId;
|
|
|
|
const Address _myAddress;
|
|
|
|
std::string _myAddressStr;
|
2019-08-06 12:51:50 +00:00
|
|
|
std::string _connString;
|
2018-08-31 21:58:15 +00:00
|
|
|
|
2019-08-06 15:42:54 +00:00
|
|
|
BlockingQueue< std::pair<nlohmann::json,bool> > _commitQueue;
|
2018-08-31 21:58:15 +00:00
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
std::thread _heartbeatThread;
|
|
|
|
std::thread _membersDbWatcher;
|
|
|
|
std::thread _networksDbWatcher;
|
|
|
|
std::thread _commitThread[ZT_CENTRAL_CONTROLLER_COMMIT_THREADS];
|
|
|
|
std::thread _onlineNotificationThread;
|
2018-08-31 21:58:15 +00:00
|
|
|
|
|
|
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > _lastOnline;
|
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
mutable std::mutex _lastOnline_l;
|
|
|
|
mutable std::mutex _readyLock;
|
|
|
|
std::atomic<int> _ready, _connected, _run;
|
|
|
|
mutable volatile bool _waitNoticePrinted;
|
2019-01-21 19:18:20 +00:00
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
int _listenPort;
|
2021-05-28 21:08:24 +00:00
|
|
|
uint8_t _ssoPsk[48];
|
2020-05-11 22:03:56 +00:00
|
|
|
|
|
|
|
RedisConfig *_rc;
|
2020-05-12 18:56:19 +00:00
|
|
|
std::shared_ptr<sw::redis::Redis> _redis;
|
|
|
|
std::shared_ptr<sw::redis::RedisCluster> _cluster;
|
2018-08-31 21:58:15 +00:00
|
|
|
};
|
|
|
|
|
2019-08-06 12:51:50 +00:00
|
|
|
} // namespace ZeroTier
|
2018-08-31 21:58:15 +00:00
|
|
|
|
|
|
|
#endif // ZT_CONTROLLER_LIBPQ_HPP
|
|
|
|
|
2019-07-27 00:39:00 +00:00
|
|
|
#endif // ZT_CONTROLLER_USE_LIBPQ
|