2015-02-24 20:28:58 +00:00
|
|
|
/*
|
|
|
|
* ZeroTier One - Network Virtualization Everywhere
|
|
|
|
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
* --
|
|
|
|
*
|
|
|
|
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
|
|
|
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
|
|
|
*
|
|
|
|
* If you would like to embed ZeroTier into a commercial application or
|
|
|
|
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
|
|
|
* LLC. Start here: http://www.zerotier.com/
|
|
|
|
*/
|
|
|
|
|
2015-04-15 22:12:09 +00:00
|
|
|
#ifndef ZT_SQLITENETWORKCONTROLLER_HPP
|
|
|
|
#define ZT_SQLITENETWORKCONTROLLER_HPP
|
2015-02-24 20:28:58 +00:00
|
|
|
|
|
|
|
#include <stdint.h>
|
2015-03-12 21:03:53 +00:00
|
|
|
|
|
|
|
#include <sqlite3.h>
|
|
|
|
|
2015-02-24 20:28:58 +00:00
|
|
|
#include <string>
|
|
|
|
#include <map>
|
|
|
|
#include <vector>
|
|
|
|
|
2015-02-24 22:17:57 +00:00
|
|
|
#include "../node/Constants.hpp"
|
2015-04-15 22:12:09 +00:00
|
|
|
#include "../node/NetworkController.hpp"
|
2015-02-24 22:17:57 +00:00
|
|
|
#include "../node/Mutex.hpp"
|
2015-11-03 23:52:10 +00:00
|
|
|
#include "../osdep/Thread.hpp"
|
2015-04-21 23:41:35 +00:00
|
|
|
|
2015-09-15 17:59:23 +00:00
|
|
|
// Number of in-memory last log entries to maintain per user
|
|
|
|
#define ZT_SQLITENETWORKCONTROLLER_IN_MEMORY_LOG_SIZE 32
|
|
|
|
|
2016-02-22 23:48:27 +00:00
|
|
|
// How long do circuit tests last before they're forgotten?
|
|
|
|
#define ZT_SQLITENETWORKCONTROLLER_CIRCUIT_TEST_TIMEOUT 60000
|
2015-10-08 20:25:38 +00:00
|
|
|
|
2015-02-24 22:17:57 +00:00
|
|
|
namespace ZeroTier {
|
2015-02-24 20:28:58 +00:00
|
|
|
|
2015-10-06 22:56:18 +00:00
|
|
|
class Node;
|
|
|
|
|
2015-05-16 23:09:28 +00:00
|
|
|
class SqliteNetworkController : public NetworkController
|
2015-02-24 20:28:58 +00:00
|
|
|
{
|
|
|
|
public:
|
2015-10-06 22:56:18 +00:00
|
|
|
SqliteNetworkController(Node *node,const char *dbPath,const char *circuitTestPath);
|
2015-04-15 22:12:09 +00:00
|
|
|
virtual ~SqliteNetworkController();
|
2015-03-13 20:53:15 +00:00
|
|
|
|
2015-04-15 22:12:09 +00:00
|
|
|
virtual NetworkController::ResultCode doNetworkConfigRequest(
|
2015-02-24 20:28:58 +00:00
|
|
|
const InetAddress &fromAddr,
|
2015-04-16 01:32:25 +00:00
|
|
|
const Identity &signingId,
|
2015-03-18 23:10:48 +00:00
|
|
|
const Identity &identity,
|
2015-02-24 20:28:58 +00:00
|
|
|
uint64_t nwid,
|
2016-06-21 14:32:58 +00:00
|
|
|
const Dictionary<ZT_NETWORKCONFIG_DICT_CAPACITY> &metaData,
|
2016-06-16 23:05:57 +00:00
|
|
|
NetworkConfig &nc);
|
2015-02-24 20:28:58 +00:00
|
|
|
|
2015-05-16 23:09:28 +00:00
|
|
|
unsigned int handleControlPlaneHttpGET(
|
2015-04-21 23:41:35 +00:00
|
|
|
const std::vector<std::string> &path,
|
|
|
|
const std::map<std::string,std::string> &urlArgs,
|
|
|
|
const std::map<std::string,std::string> &headers,
|
|
|
|
const std::string &body,
|
|
|
|
std::string &responseBody,
|
|
|
|
std::string &responseContentType);
|
2015-05-16 23:09:28 +00:00
|
|
|
unsigned int handleControlPlaneHttpPOST(
|
2015-04-21 23:41:35 +00:00
|
|
|
const std::vector<std::string> &path,
|
|
|
|
const std::map<std::string,std::string> &urlArgs,
|
|
|
|
const std::map<std::string,std::string> &headers,
|
|
|
|
const std::string &body,
|
|
|
|
std::string &responseBody,
|
|
|
|
std::string &responseContentType);
|
2015-05-16 23:09:28 +00:00
|
|
|
unsigned int handleControlPlaneHttpDELETE(
|
2015-04-21 23:41:35 +00:00
|
|
|
const std::vector<std::string> &path,
|
|
|
|
const std::map<std::string,std::string> &urlArgs,
|
|
|
|
const std::map<std::string,std::string> &headers,
|
|
|
|
const std::string &body,
|
|
|
|
std::string &responseBody,
|
|
|
|
std::string &responseContentType);
|
|
|
|
|
2015-11-03 23:52:10 +00:00
|
|
|
// threadMain() for backup thread -- do not call directly
|
|
|
|
void threadMain()
|
|
|
|
throw();
|
|
|
|
|
2015-02-24 20:28:58 +00:00
|
|
|
private:
|
2016-06-09 18:02:42 +00:00
|
|
|
/* deprecated
|
2015-06-29 21:52:09 +00:00
|
|
|
enum IpAssignmentType {
|
|
|
|
// IP assignment is a static IP address
|
|
|
|
ZT_IP_ASSIGNMENT_TYPE_ADDRESS = 0,
|
|
|
|
// IP assignment is a network -- a route via this interface, not an address
|
|
|
|
ZT_IP_ASSIGNMENT_TYPE_NETWORK = 1
|
|
|
|
};
|
2016-06-09 18:02:42 +00:00
|
|
|
*/
|
2015-06-29 21:52:09 +00:00
|
|
|
|
2015-05-17 00:12:29 +00:00
|
|
|
unsigned int _doCPGet(
|
|
|
|
const std::vector<std::string> &path,
|
|
|
|
const std::map<std::string,std::string> &urlArgs,
|
|
|
|
const std::map<std::string,std::string> &headers,
|
|
|
|
const std::string &body,
|
|
|
|
std::string &responseBody,
|
|
|
|
std::string &responseContentType);
|
|
|
|
|
2015-10-08 20:25:38 +00:00
|
|
|
static void _circuitTestCallback(ZT_Node *node,ZT_CircuitTest *test,const ZT_CircuitTestReport *report);
|
|
|
|
|
2015-10-06 22:56:18 +00:00
|
|
|
Node *_node;
|
2015-11-03 23:52:10 +00:00
|
|
|
Thread _backupThread;
|
|
|
|
volatile bool _backupThreadRun;
|
2016-06-24 13:53:23 +00:00
|
|
|
volatile bool _backupNeeded;
|
2015-03-12 21:03:53 +00:00
|
|
|
std::string _dbPath;
|
2015-10-06 22:56:18 +00:00
|
|
|
std::string _circuitTestPath;
|
2015-07-21 17:39:29 +00:00
|
|
|
std::string _instanceId;
|
2015-07-22 21:01:49 +00:00
|
|
|
|
2015-10-08 20:25:38 +00:00
|
|
|
// Circuit tests outstanding
|
2016-01-26 20:42:44 +00:00
|
|
|
struct _CircuitTestEntry
|
|
|
|
{
|
|
|
|
ZT_CircuitTest *test;
|
|
|
|
std::string jsonResults;
|
|
|
|
};
|
|
|
|
std::map< uint64_t,_CircuitTestEntry > _circuitTests;
|
2015-10-08 20:25:38 +00:00
|
|
|
|
2016-02-04 02:10:56 +00:00
|
|
|
// Last request time by address, for rate limitation
|
2016-02-04 22:39:43 +00:00
|
|
|
std::map< std::pair<uint64_t,uint64_t>,uint64_t > _lastRequestTime;
|
2016-02-04 02:10:56 +00:00
|
|
|
|
2015-03-12 21:03:53 +00:00
|
|
|
sqlite3 *_db;
|
2015-02-24 22:17:57 +00:00
|
|
|
|
2015-03-18 23:10:48 +00:00
|
|
|
sqlite3_stmt *_sGetNetworkById;
|
2015-04-21 23:41:35 +00:00
|
|
|
sqlite3_stmt *_sGetMember;
|
2015-03-18 23:10:48 +00:00
|
|
|
sqlite3_stmt *_sCreateMember;
|
|
|
|
sqlite3_stmt *_sGetNodeIdentity;
|
2015-09-10 21:37:34 +00:00
|
|
|
sqlite3_stmt *_sCreateOrReplaceNode;
|
2015-03-18 23:10:48 +00:00
|
|
|
sqlite3_stmt *_sGetEtherTypesFromRuleTable;
|
|
|
|
sqlite3_stmt *_sGetActiveBridges;
|
|
|
|
sqlite3_stmt *_sGetIpAssignmentsForNode;
|
|
|
|
sqlite3_stmt *_sGetIpAssignmentPools;
|
|
|
|
sqlite3_stmt *_sCheckIfIpIsAllocated;
|
|
|
|
sqlite3_stmt *_sAllocateIp;
|
2015-05-16 19:50:42 +00:00
|
|
|
sqlite3_stmt *_sDeleteIpAllocations;
|
2015-04-21 00:47:12 +00:00
|
|
|
sqlite3_stmt *_sGetRelays;
|
2015-04-21 23:41:35 +00:00
|
|
|
sqlite3_stmt *_sListNetworks;
|
|
|
|
sqlite3_stmt *_sListNetworkMembers;
|
|
|
|
sqlite3_stmt *_sGetMember2;
|
|
|
|
sqlite3_stmt *_sGetIpAssignmentPools2;
|
|
|
|
sqlite3_stmt *_sListRules;
|
2015-04-22 01:08:33 +00:00
|
|
|
sqlite3_stmt *_sCreateRule;
|
|
|
|
sqlite3_stmt *_sCreateNetwork;
|
|
|
|
sqlite3_stmt *_sGetNetworkRevision;
|
2015-04-24 19:29:31 +00:00
|
|
|
sqlite3_stmt *_sSetNetworkRevision;
|
2015-04-22 02:49:04 +00:00
|
|
|
sqlite3_stmt *_sGetIpAssignmentsForNode2;
|
|
|
|
sqlite3_stmt *_sDeleteRelaysForNetwork;
|
|
|
|
sqlite3_stmt *_sCreateRelay;
|
2015-04-23 01:06:26 +00:00
|
|
|
sqlite3_stmt *_sDeleteIpAssignmentPoolsForNetwork;
|
|
|
|
sqlite3_stmt *_sDeleteRulesForNetwork;
|
|
|
|
sqlite3_stmt *_sCreateIpAssignmentPool;
|
2015-06-19 19:19:42 +00:00
|
|
|
sqlite3_stmt *_sUpdateMemberAuthorized;
|
|
|
|
sqlite3_stmt *_sUpdateMemberActiveBridge;
|
2016-06-28 19:44:47 +00:00
|
|
|
sqlite3_stmt *_sUpdateMemberHistory;
|
2015-05-16 20:42:53 +00:00
|
|
|
sqlite3_stmt *_sDeleteMember;
|
2015-09-11 22:02:26 +00:00
|
|
|
sqlite3_stmt *_sDeleteAllNetworkMembers;
|
2016-06-28 19:44:47 +00:00
|
|
|
sqlite3_stmt *_sGetActiveNodesOnNetwork;
|
2015-06-13 09:34:31 +00:00
|
|
|
sqlite3_stmt *_sDeleteNetwork;
|
2016-06-10 15:26:27 +00:00
|
|
|
sqlite3_stmt *_sCreateRoute;
|
|
|
|
sqlite3_stmt *_sGetRoutes;
|
|
|
|
sqlite3_stmt *_sDeleteRoutes;
|
2015-07-17 00:34:03 +00:00
|
|
|
sqlite3_stmt *_sIncrementMemberRevisionCounter;
|
2015-07-17 17:47:21 +00:00
|
|
|
sqlite3_stmt *_sGetConfig;
|
|
|
|
sqlite3_stmt *_sSetConfig;
|
2015-03-18 23:10:48 +00:00
|
|
|
|
2015-03-12 21:03:53 +00:00
|
|
|
Mutex _lock;
|
2015-02-24 20:28:58 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace ZeroTier
|
|
|
|
|
|
|
|
#endif
|