| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525 | /* * ZeroTier One - Network Virtualization Everywhere * Copyright (C) 2011-2019  ZeroTier, Inc.  https://www.zerotier.com/ * * 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/>. * * -- * * You can be released from the requirements of the license by purchasing * a commercial license. Buying such a license is mandatory as soon as you * develop commercial closed-source software that incorporates or links * directly against ZeroTier software without disclosing the source code * of your own application. */#ifdef ZT_CONTROLLER_USE_LIBPQ#include "PostgreSQL.hpp"#include "EmbeddedNetworkController.hpp"#include "RabbitMQ.hpp"#include "../version.h"#include <libpq-fe.h>#include <sstream>#include <amqp.h>#include <amqp_tcp_socket.h>using json = nlohmann::json;namespace {static const char *_timestr(){	time_t t = time(0);	char *ts = ctime(&t);	char *p = ts;	if (!p)		return "";	while (*p) {		if (*p == '\n') {			*p = (char)0;			break;		}		++p;	}	return ts;}std::string join(const std::vector<std::string> &elements, const char * const separator){	switch(elements.size()) {	case 0:		return "";	case 1:		return elements[0];	default:		std::ostringstream os;		std::copy(elements.begin(), elements.end()-1, std::ostream_iterator<std::string>(os, separator));		os << *elements.rbegin();		return os.str();	}}}using namespace ZeroTier;PostgreSQL::PostgreSQL(EmbeddedNetworkController *const nc, const Identity &myId, const char *path, int listenPort, MQConfig *mqc)    : DB(nc, myId, path)    , _ready(0)	, _connected(1)    , _run(1)    , _waitNoticePrinted(false)	, _listenPort(listenPort)	, _mqc(mqc){	_connString = std::string(path) + " application_name=controller_" +_myAddressStr;	_readyLock.lock();	_heartbeatThread = std::thread(&PostgreSQL::heartbeat, this);	_membersDbWatcher = std::thread(&PostgreSQL::membersDbWatcher, this);	_networksDbWatcher = std::thread(&PostgreSQL::networksDbWatcher, this);	for (int i = 0; i < ZT_CENTRAL_CONTROLLER_COMMIT_THREADS; ++i) {		_commitThread[i] = std::thread(&PostgreSQL::commitThread, this);	}	_onlineNotificationThread = std::thread(&PostgreSQL::onlineNotificationThread, this);}PostgreSQL::~PostgreSQL(){	_run = 0;	std::this_thread::sleep_for(std::chrono::milliseconds(100));		_heartbeatThread.join();	_membersDbWatcher.join();	_networksDbWatcher.join();	for (int i = 0; i < ZT_CENTRAL_CONTROLLER_COMMIT_THREADS; ++i) {		_commitThread[i].join();	}	_onlineNotificationThread.join();}bool PostgreSQL::waitForReady(){	while (_ready < 2) {		if (!_waitNoticePrinted) {			_waitNoticePrinted = true;			fprintf(stderr, "[%s] NOTICE: %.10llx controller PostgreSQL waiting for initial data download..." ZT_EOL_S, ::_timestr(), (unsigned long long)_myAddress.toInt());		}		_readyLock.lock();		_readyLock.unlock();	}	return true;}bool PostgreSQL::isReady(){	return ((_ready == 2)&&(_connected));}void PostgreSQL::save(nlohmann::json *orig, nlohmann::json &record){	try {		if (!record.is_object()) {			return;		}		waitForReady();		if (orig) {			if (*orig != record) {				record["revision"] = OSUtils::jsonInt(record["revision"],0ULL) + 1;				_commitQueue.post(new nlohmann::json(record));			}		} else {			record["revision"] = 1;			_commitQueue.post(new nlohmann::json(record));		}	} catch (std::exception &e) {		fprintf(stderr, "Error on PostgreSQL::save: %s\n", e.what());	} catch (...) {		fprintf(stderr, "Unknown error on PostgreSQL::save\n");	}}void PostgreSQL::eraseNetwork(const uint64_t networkId){	char tmp2[24];	waitForReady();	Utils::hex(networkId, tmp2);	json *tmp = new json();	(*tmp)["id"] = tmp2;	(*tmp)["objtype"] = "_delete_network";	_commitQueue.post(tmp);}void PostgreSQL::eraseMember(const uint64_t networkId, const uint64_t memberId) {	char tmp2[24];	json *tmp = new json();	Utils::hex(networkId, tmp2);	(*tmp)["nwid"] = tmp2;	Utils::hex(memberId, tmp2);	(*tmp)["id"] = tmp2;	(*tmp)["objtype"] = "_delete_member";	_commitQueue.post(tmp);}void PostgreSQL::nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress &physicalAddress){	std::lock_guard<std::mutex> l(_lastOnline_l);	std::pair<int64_t, InetAddress> &i = _lastOnline[std::pair<uint64_t,uint64_t>(networkId, memberId)];	i.first = OSUtils::now();	if (physicalAddress) {		i.second = physicalAddress;	}}void PostgreSQL::initializeNetworks(PGconn *conn){	try {		if (PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));			exit(1);		}		const char *params[1] = {			_myAddressStr.c_str()		};		PGresult *res = PQexecParams(conn, "SELECT id, EXTRACT(EPOCH FROM creation_time AT TIME ZONE 'UTC')*1000, capabilities, "			"enable_broadcast, EXTRACT(EPOCH FROM last_modified AT TIME ZONE 'UTC')*1000, mtu, multicast_limit, name, private, remote_trace_level, "			"remote_trace_target, revision, rules, tags, v4_assign_mode, v6_assign_mode FROM ztc_network "			"WHERE deleted = false AND controller_id = $1",			1,			NULL,			params,			NULL,			NULL,			0);				if (PQresultStatus(res) != PGRES_TUPLES_OK) {			fprintf(stderr, "Networks Initialization Failed: %s", PQerrorMessage(conn));			PQclear(res);			exit(1);		}		int numRows = PQntuples(res);		for (int i = 0; i < numRows; ++i) {			json empty;			json config;			const char *nwidparam[1] = {				PQgetvalue(res, i, 0)			};			config["id"] = PQgetvalue(res, i, 0);			config["nwid"] = PQgetvalue(res, i, 0);			try {				config["creationTime"] = std::stoull(PQgetvalue(res, i, 1));			} catch (std::exception &e) {				config["creationTime"] = 0ULL;				//fprintf(stderr, "Error converting creation time: %s\n", PQgetvalue(res, i, 1));			}			config["capabilities"] = json::parse(PQgetvalue(res, i, 2));			config["enableBroadcast"] = (strcmp(PQgetvalue(res, i, 3),"t")==0);			try {				config["lastModified"] = std::stoull(PQgetvalue(res, i, 4));			} catch (std::exception &e) {				config["lastModified"] = 0ULL;				//fprintf(stderr, "Error converting last modified: %s\n", PQgetvalue(res, i, 4));			}			try {				config["mtu"] = std::stoi(PQgetvalue(res, i, 5));			} catch (std::exception &e) {				config["mtu"] = 2800;			}			try {				config["multicastLimit"] = std::stoi(PQgetvalue(res, i, 6));			} catch (std::exception &e) {				config["multicastLimit"] = 64;			}			config["name"] = PQgetvalue(res, i, 7);			config["private"] = (strcmp(PQgetvalue(res, i, 8),"t")==0);			try {				config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 9));			} catch (std::exception &e) {				config["remoteTraceLevel"] = 0;			}			config["remoteTraceTarget"] = PQgetvalue(res, i, 10);			try {				config["revision"] = std::stoull(PQgetvalue(res, i, 11));			} catch (std::exception &e) {				config["revision"] = 0ULL;				//fprintf(stderr, "Error converting revision: %s\n", PQgetvalue(res, i, 11));			}			config["rules"] = json::parse(PQgetvalue(res, i, 12));			config["tags"] = json::parse(PQgetvalue(res, i, 13));			config["v4AssignMode"] = json::parse(PQgetvalue(res, i, 14));			config["v6AssignMode"] = json::parse(PQgetvalue(res, i, 15));			config["objtype"] = "network";			config["ipAssignmentPools"] = json::array();			config["routes"] = json::array();			PGresult *r2 = PQexecParams(conn,				"SELECT host(ip_range_start), host(ip_range_end) FROM ztc_network_assignment_pool WHERE network_id = $1",				1,				NULL,				nwidparam,				NULL,				NULL,				0);						if (PQresultStatus(r2) != PGRES_TUPLES_OK) {				fprintf(stderr, "ERROR: Error retreiving IP pools for network: %s\n", PQresultErrorMessage(r2));				PQclear(r2);				PQclear(res);				exit(1);			}			int n = PQntuples(r2);			for (int j = 0; j < n; ++j) {				json ip;				ip["ipRangeStart"] = PQgetvalue(r2, j, 0);				ip["ipRangeEnd"] = PQgetvalue(r2, j, 1);				config["ipAssignmentPools"].push_back(ip);			}			PQclear(r2);			r2 = PQexecParams(conn,				"SELECT host(address), bits, host(via) FROM ztc_network_route WHERE network_id = $1",				1,				NULL,				nwidparam,				NULL,				NULL,				0);			if (PQresultStatus(r2) != PGRES_TUPLES_OK) {				fprintf(stderr, "ERROR: Error retreiving routes for network: %s\n", PQresultErrorMessage(r2));				PQclear(r2);				PQclear(res);				exit(1);			}			n = PQntuples(r2);			for (int j = 0; j < n; ++j) {				std::string addr = PQgetvalue(r2, j, 0);				std::string bits = PQgetvalue(r2, j, 1);				std::string via = PQgetvalue(r2, j, 2);				json route;				route["target"] = addr + "/" + bits;				if (via == "NULL") {					route["via"] = nullptr;				} else {					route["via"] = via;				}				config["routes"].push_back(route);			}			PQclear(r2);						_networkChanged(empty, config, false);		}		PQclear(res);		if (++this->_ready == 2) {			if (_waitNoticePrinted) {				fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());			}			_readyLock.unlock();		}	} catch (std::exception &e) {		fprintf(stderr, "ERROR: Error initializing networks: %s", e.what());		exit(-1);	}}void PostgreSQL::initializeMembers(PGconn *conn){	try {		if (PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));			exit(1);		}		const char *params[1] = {			_myAddressStr.c_str()		};		PGresult *res = PQexecParams(conn,			"SELECT m.id, m.network_id, m.active_bridge, m.authorized, m.capabilities, EXTRACT(EPOCH FROM m.creation_time AT TIME ZONE 'UTC')*1000, m.identity, "			"	EXTRACT(EPOCH FROM m.last_authorized_time AT TIME ZONE 'UTC')*1000, "			"	EXTRACT(EPOCH FROM m.last_deauthorized_time AT TIME ZONE 'UTC')*1000, "			"	m.remote_trace_level, m.remote_trace_target, m.tags, m.v_major, m.v_minor, m.v_rev, m.v_proto, "			"	m.no_auto_assign_ips, m.revision "			"FROM ztc_member m "			"INNER JOIN ztc_network n "			"	ON n.id = m.network_id "			"WHERE n.controller_id = $1 AND m.deleted = false",			1,			NULL,			params,			NULL,			NULL,			0);		if (PQresultStatus(res) != PGRES_TUPLES_OK) {			fprintf(stderr, "Member Initialization Failed: %s", PQerrorMessage(conn));			PQclear(res);			exit(1);		}		int numRows = PQntuples(res);		for (int i = 0; i < numRows; ++i) {			json empty;			json config;			std::string memberId(PQgetvalue(res, i, 0));			std::string networkId(PQgetvalue(res, i, 1));			std::string ctime = PQgetvalue(res, i, 5);			config["id"] = memberId;			config["nwid"] = networkId;			config["activeBridge"] = (strcmp(PQgetvalue(res, i, 2), "t") == 0);			config["authorized"] = (strcmp(PQgetvalue(res, i, 3), "t") == 0);			try {				config["capabilities"] = json::parse(PQgetvalue(res, i, 4));			} catch (std::exception &e) {				config["capabilities"] = json::array();			}			try {				config["creationTime"] = std::stoull(PQgetvalue(res, i, 5));			} catch (std::exception &e) {				config["creationTime"] = 0ULL;				//fprintf(stderr, "Error upding creation time (member): %s\n", PQgetvalue(res, i, 5));			}			config["identity"] = PQgetvalue(res, i, 6);			try {				config["lastAuthorizedTime"] = std::stoull(PQgetvalue(res, i, 7));			} catch(std::exception &e) {				config["lastAuthorizedTime"] = 0ULL;				//fprintf(stderr, "Error updating last auth time (member): %s\n", PQgetvalue(res, i, 7));			}			try {				config["lastDeauthorizedTime"] = std::stoull(PQgetvalue(res, i, 8));			} catch( std::exception &e) {				config["lastDeauthorizedTime"] = 0ULL;				//fprintf(stderr, "Error updating last deauth time (member): %s\n", PQgetvalue(res, i, 8));			}			try {				config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 9));			} catch (std::exception &e) {				config["remoteTraceLevel"] = 0;			}			config["remoteTraceTarget"] = PQgetvalue(res, i, 10);			try {				config["tags"] = json::parse(PQgetvalue(res, i, 11));			} catch (std::exception &e) {				config["tags"] = json::array();			}			try {				config["vMajor"] = std::stoi(PQgetvalue(res, i, 12));			} catch(std::exception &e) {				config["vMajor"] = -1;			}			try {				config["vMinor"] = std::stoi(PQgetvalue(res, i, 13));			} catch (std::exception &e) {				config["vMinor"] = -1;			}			try {				config["vRev"] = std::stoi(PQgetvalue(res, i, 14));			} catch (std::exception &e) {				config["vRev"] = -1;			}			try {				config["vProto"] = std::stoi(PQgetvalue(res, i, 15));			} catch (std::exception &e) {				config["vProto"] = -1;			}			config["noAutoAssignIps"] = (strcmp(PQgetvalue(res, i, 16), "t") == 0);			try {				config["revision"] = std::stoull(PQgetvalue(res, i, 17));			} catch (std::exception &e) {				config["revision"] = 0ULL;				//fprintf(stderr, "Error updating revision (member): %s\n", PQgetvalue(res, i, 17));			}			config["objtype"] = "member";			config["ipAssignments"] = json::array();			const char *p2[2] = {				memberId.c_str(),				networkId.c_str()			};			PGresult *r2 = PQexecParams(conn,				"SELECT DISTINCT address FROM ztc_member_ip_assignment WHERE member_id = $1 AND network_id = $2",				2,				NULL,				p2,				NULL,				NULL,				0);			if (PQresultStatus(r2) != PGRES_TUPLES_OK) {				fprintf(stderr, "Member Initialization Failed: %s", PQerrorMessage(conn));				PQclear(r2);				PQclear(res);				exit(1);			}			int n = PQntuples(r2);			for (int j = 0; j < n; ++j) {				config["ipAssignments"].push_back(PQgetvalue(r2, j, 0));			}			_memberChanged(empty, config, false);		}		PQclear(res);		if (++this->_ready == 2) {			if (_waitNoticePrinted) {				fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());			}			_readyLock.unlock();		}	} catch (std::exception &e) {		fprintf(stderr, "ERROR: Error initializing members: %s\n", e.what());		exit(-1);	}}void PostgreSQL::heartbeat(){	char publicId[1024];	char hostnameTmp[1024];	_myId.toString(false,publicId);	if (gethostname(hostnameTmp, sizeof(hostnameTmp))!= 0) {		hostnameTmp[0] = (char)0;	} else {		for (int i = 0; i < sizeof(hostnameTmp); ++i) {			if ((hostnameTmp[i] == '.')||(hostnameTmp[i] == 0)) {				hostnameTmp[i] = (char)0;				break;			}		}	}	const char *controllerId = _myAddressStr.c_str();	const char *publicIdentity = publicId;	const char *hostname = hostnameTmp;	PGconn *conn = getPgConn();	if (PQstatus(conn) == CONNECTION_BAD) {		fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(conn));		PQfinish(conn);		exit(1);	}	while (_run == 1) {		if(PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "%s heartbeat thread lost connection to Database\n", _myAddressStr.c_str());			PQfinish(conn);			exit(6);		}		if (conn) {			std::string major = std::to_string(ZEROTIER_ONE_VERSION_MAJOR);			std::string minor = std::to_string(ZEROTIER_ONE_VERSION_MINOR);			std::string rev = std::to_string(ZEROTIER_ONE_VERSION_REVISION);			std::string build = std::to_string(ZEROTIER_ONE_VERSION_BUILD);			std::string now = std::to_string(OSUtils::now());			std::string host_port = std::to_string(_listenPort);			std::string use_rabbitmq = (_mqc != NULL) ? "true" : "false";			const char *values[10] = {				controllerId,				hostname,				now.c_str(),				publicIdentity,				major.c_str(),				minor.c_str(),				rev.c_str(),				build.c_str(),				host_port.c_str(),				use_rabbitmq.c_str()			};			PGresult *res = PQexecParams(conn,				"INSERT INTO ztc_controller (id, cluster_host, last_alive, public_identity, v_major, v_minor, v_rev, v_build, host_port, use_rabbitmq) " 				"VALUES ($1, $2, TO_TIMESTAMP($3::double precision/1000), $4, $5, $6, $7, $8, $9, $10) "				"ON CONFLICT (id) DO UPDATE SET cluster_host = EXCLUDED.cluster_host, last_alive = EXCLUDED.last_alive, "				"public_identity = EXCLUDED.public_identity, v_major = EXCLUDED.v_major, v_minor = EXCLUDED.v_minor, "				"v_rev = EXCLUDED.v_rev, v_build = EXCLUDED.v_rev, host_port = EXCLUDED.host_port, "				"use_rabbitmq = EXCLUDED.use_rabbitmq",				10,       // number of parameters				NULL,    // oid field.   ignore				values,  // values for substitution				NULL, // lengths in bytes of each value				NULL,  // binary?				0);			if (PQresultStatus(res) != PGRES_COMMAND_OK) {				fprintf(stderr, "Heartbeat Update Failed: %s\n", PQresultErrorMessage(res));			}			PQclear(res);		}		std::this_thread::sleep_for(std::chrono::milliseconds(1000));	}	PQfinish(conn);	conn = NULL;}void PostgreSQL::membersDbWatcher(){	PGconn *conn = getPgConn(NO_OVERRIDE);	if (PQstatus(conn) == CONNECTION_BAD) {		fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(conn));		PQfinish(conn);		exit(1);	}	initializeMembers(conn);	if (this->_mqc != NULL) {		PQfinish(conn);		conn = NULL;		_membersWatcher_RabbitMQ();	} else {		_membersWatcher_Postgres(conn);		PQfinish(conn);		conn = NULL;	}	if (_run == 1) {		fprintf(stderr, "ERROR: %s membersDbWatcher should still be running! Exiting Controller.\n", _myAddressStr.c_str());		exit(9);	}	fprintf(stderr, "Exited membersDbWatcher\n");}void PostgreSQL::_membersWatcher_Postgres(PGconn *conn) {	char buf[11] = {0};	std::string cmd = "LISTEN member_" + std::string(_myAddress.toString(buf));	PGresult *res = PQexec(conn, cmd.c_str());	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK) {		fprintf(stderr, "LISTEN command failed: %s\n", PQresultErrorMessage(res));		PQclear(res);		PQfinish(conn);		exit(1);	}	PQclear(res); res = NULL;	while(_run == 1) {		if (PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "ERROR: Member Watcher lost connection to Postgres.");			exit(-1);		}		PGnotify *notify = NULL;		PQconsumeInput(conn);		while ((notify = PQnotifies(conn)) != NULL) {			//fprintf(stderr, "ASYNC NOTIFY of '%s' id:%s received\n", notify->relname, notify->extra);			try {				json tmp(json::parse(notify->extra));				json &ov = tmp["old_val"];				json &nv = tmp["new_val"];				json oldConfig, newConfig;				if (ov.is_object()) oldConfig = ov;				if (nv.is_object()) newConfig = nv;				if (oldConfig.is_object() || newConfig.is_object()) {					_memberChanged(oldConfig,newConfig,(this->_ready>=2));				}			} catch (...) {} // ignore bad records			free(notify);		}		std::this_thread::sleep_for(std::chrono::milliseconds(10));	}}void PostgreSQL::_membersWatcher_RabbitMQ() {	char buf[11] = {0};	std::string qname = "member_"+ std::string(_myAddress.toString(buf));	RabbitMQ rmq(_mqc, qname.c_str());	try {		rmq.init();	} catch (std::runtime_error &e) {		fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());		exit(11);	}	while (_run == 1) {		try {			std::string msg = rmq.consume();			// fprintf(stderr, "Got Member Update: %s\n", msg.c_str());			json tmp(json::parse(msg));			json &ov = tmp["old_val"];			json &nv = tmp["new_val"];			json oldConfig, newConfig;			if (ov.is_object()) oldConfig = ov;			if (nv.is_object()) newConfig = nv;			if (oldConfig.is_object() || newConfig.is_object()) {				_memberChanged(oldConfig,newConfig,(this->_ready>=2));			}		} catch (std::runtime_error &e) {			fprintf(stderr, "RABBITMQ ERROR member change: %s\n", e.what());			break;		} catch(std::exception &e ) {			fprintf(stderr, "RABBITMQ ERROR member change: %s\n", e.what());		} catch(...) {			fprintf(stderr, "RABBITMQ ERROR member change: unknown error\n");        }	}}void PostgreSQL::networksDbWatcher(){	PGconn *conn = getPgConn(NO_OVERRIDE);	if (PQstatus(conn) == CONNECTION_BAD) {		fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(conn));		PQfinish(conn);		exit(1);	}	initializeNetworks(conn);	if (this->_mqc != NULL) {		PQfinish(conn);		conn = NULL;		_networksWatcher_RabbitMQ();	} else {		_networksWatcher_Postgres(conn);		PQfinish(conn);		conn = NULL;	}		if (_run == 1) {		fprintf(stderr, "ERROR: %s networksDbWatcher should still be running! Exiting Controller.\n", _myAddressStr.c_str());		exit(8);	}	fprintf(stderr, "Exited membersDbWatcher\n");}void PostgreSQL::_networksWatcher_Postgres(PGconn *conn) {	char buf[11] = {0};	std::string cmd = "LISTEN network_" + std::string(_myAddress.toString(buf));	PGresult *res = PQexec(conn, cmd.c_str());	if (!res || PQresultStatus(res) != PGRES_COMMAND_OK) {		fprintf(stderr, "LISTEN command failed: %s\n", PQresultErrorMessage(res));		PQclear(res);		PQfinish(conn);		exit(1);	}	PQclear(res); res = NULL;	while(_run == 1) {		if (PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "ERROR: Network Watcher lost connection to Postgres.");			exit(-1);		}		PGnotify *notify = NULL;		PQconsumeInput(conn);		while ((notify = PQnotifies(conn)) != NULL) {			//fprintf(stderr, "ASYNC NOTIFY of '%s' id:%s received\n", notify->relname, notify->extra);			try {				json tmp(json::parse(notify->extra));				json &ov = tmp["old_val"];				json &nv = tmp["new_val"];				json oldConfig, newConfig;				if (ov.is_object()) oldConfig = ov;				if (nv.is_object()) newConfig = nv;				if (oldConfig.is_object()||newConfig.is_object()) {					_networkChanged(oldConfig,newConfig,(this->_ready >= 2));				}			} catch (...) {} // ignore bad records			free(notify);		}		std::this_thread::sleep_for(std::chrono::milliseconds(10));	}}void PostgreSQL::_networksWatcher_RabbitMQ() {	char buf[11] = {0};	std::string qname = "network_"+ std::string(_myAddress.toString(buf));	RabbitMQ rmq(_mqc, qname.c_str());	try {		rmq.init();	} catch (std::runtime_error &e) {		fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());		exit(11);	}	while (_run == 1) {		try {			std::string msg = rmq.consume();			// fprintf(stderr, "Got network update: %s\n", msg.c_str());			json tmp(json::parse(msg));			json &ov = tmp["old_val"];			json &nv = tmp["new_val"];			json oldConfig, newConfig;			if (ov.is_object()) oldConfig = ov;			if (nv.is_object()) newConfig = nv;			if (oldConfig.is_object()||newConfig.is_object()) {				_networkChanged(oldConfig,newConfig,(this->_ready >= 2));			}		} catch (std::runtime_error &e) {			fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());			break;		} catch (std::exception &e) {			fprintf(stderr, "RABBITMQ ERROR network watcher: %s\n", e.what());		} catch(...) {			fprintf(stderr, "RABBITMQ ERROR network watcher: unknown error\n");		}	}}void PostgreSQL::commitThread(){	PGconn *conn = getPgConn();	if (PQstatus(conn) == CONNECTION_BAD) {		fprintf(stderr, "ERROR: Connection to database failed: %s\n", PQerrorMessage(conn));		PQfinish(conn);		exit(1);	}	json *config = nullptr;	while(_commitQueue.get(config)&(_run == 1)) {		if (!config) {			continue;		}		if (PQstatus(conn) == CONNECTION_BAD) {			fprintf(stderr, "ERROR: Connection to database failed: %s\n", PQerrorMessage(conn));			PQfinish(conn);			delete config;			exit(1);		}		try { 			const std::string objtype = (*config)["objtype"];			if (objtype == "member") {				try {					std::string memberId = (*config)["id"];					std::string networkId = (*config)["nwid"];					std::string identity = (*config)["identity"];					std::string target = "NULL";					if (!(*config)["remoteTraceTarget"].is_null()) {						target = (*config)["remoteTraceTarget"];					}					std::string caps = OSUtils::jsonDump((*config)["capabilities"], -1);					std::string lastAuthTime = std::to_string((long long)(*config)["lastAuthorizedTime"]);					std::string lastDeauthTime = std::to_string((long long)(*config)["lastDeauthorizedTime"]);					std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);					std::string rev = std::to_string((unsigned long long)(*config)["revision"]);					std::string tags = OSUtils::jsonDump((*config)["tags"], -1);					std::string vmajor = std::to_string((int)(*config)["vMajor"]);					std::string vminor = std::to_string((int)(*config)["vMinor"]);					std::string vrev = std::to_string((int)(*config)["vRev"]);					std::string vproto = std::to_string((int)(*config)["vProto"]);					const char *values[19] = {						memberId.c_str(),						networkId.c_str(),						((*config)["activeBridge"] ? "true" : "false"),						((*config)["authorized"] ? "true" : "false"),						caps.c_str(),						identity.c_str(),						lastAuthTime.c_str(),						lastDeauthTime.c_str(),						((*config)["noAutoAssignIps"] ? "true" : "false"),						rtraceLevel.c_str(),						(target == "NULL") ? NULL : target.c_str(),						rev.c_str(),						tags.c_str(),						vmajor.c_str(),						vminor.c_str(),						vrev.c_str(),						vproto.c_str()					};					PGresult *res = PQexecParams(conn,						"INSERT INTO ztc_member (id, network_id, active_bridge, authorized, capabilities, "						"identity, last_authorized_time, last_deauthorized_time, no_auto_assign_ips, "						"remote_trace_level, remote_trace_target, revision, tags, v_major, v_minor, v_rev, v_proto) "						"VALUES ($1, $2, $3, $4, $5, $6, "						"TO_TIMESTAMP($7::double precision/1000), TO_TIMESTAMP($8::double precision/1000), "						"$9, $10, $11, $12, $13, $14, $15, $16, $17) ON CONFLICT (network_id, id) DO UPDATE SET "						"active_bridge = EXCLUDED.active_bridge, authorized = EXCLUDED.authorized, capabilities = EXCLUDED.capabilities, "						"identity = EXCLUDED.identity, last_authorized_time = EXCLUDED.last_authorized_time, "						"last_deauthorized_time = EXCLUDED.last_deauthorized_time, no_auto_assign_ips = EXCLUDED.no_auto_assign_ips, "						"remote_trace_level = EXCLUDED.remote_trace_level, remote_trace_target = EXCLUDED.remote_trace_target, "						"revision = EXCLUDED.revision+1, tags = EXCLUDED.tags, v_major = EXCLUDED.v_major, "						"v_minor = EXCLUDED.v_minor, v_rev = EXCLUDED.v_rev, v_proto = EXCLUDED.v_proto",						17,						NULL,						values,						NULL,						NULL,						0);										if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error updating member: %s\n", PQresultErrorMessage(res));						fprintf(stderr, "%s", OSUtils::jsonDump(*config, 2).c_str());						PQclear(res);						delete config;						config = nullptr;						continue;					}					PQclear(res);					res = PQexec(conn, "BEGIN");					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error beginning transaction: %s\n", PQresultErrorMessage(res));						PQclear(res);						delete config;						config = nullptr;						continue;					}					PQclear(res);					const char *v2[2] = {						memberId.c_str(),						networkId.c_str()					};					res = PQexecParams(conn,						"DELETE FROM ztc_member_ip_assignment WHERE member_id = $1 AND network_id = $2",						2,						NULL,						v2,						NULL,						NULL,						0);					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error updating IP address assignments: %s\n", PQresultErrorMessage(res));						PQclear(res);						PQclear(PQexec(conn, "ROLLBACK"));;						delete config;						config = nullptr;						continue;					}					PQclear(res);					std::vector<std::string> assignments;					for (auto i = (*config)["ipAssignments"].begin(); i != (*config)["ipAssignments"].end(); ++i) {						std::string addr = *i;						if (std::find(assignments.begin(), assignments.end(), addr) != assignments.end()) {							continue;						}						const char *v3[3] = {							memberId.c_str(),							networkId.c_str(),							addr.c_str()						};						res = PQexecParams(conn,							"INSERT INTO ztc_member_ip_assignment (member_id, network_id, address) VALUES ($1, $2, $3)",							3,							NULL,							v3,							NULL,							NULL,							0);												if (PQresultStatus(res) != PGRES_COMMAND_OK) {							fprintf(stderr, "ERROR: Error setting IP addresses for member: %s\n", PQresultErrorMessage(res));							PQclear(res);							PQclear(PQexec(conn, "ROLLBACK"));							break;;						}					}					res = PQexec(conn, "COMMIT");					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error committing ip address data: %s\n", PQresultErrorMessage(res));					}					PQclear(res);					const uint64_t nwidInt = OSUtils::jsonIntHex((*config)["nwid"], 0ULL);					const uint64_t memberidInt = OSUtils::jsonIntHex((*config)["id"], 0ULL);					if (nwidInt && memberidInt) {						nlohmann::json nwOrig;						nlohmann::json memOrig;						nlohmann::json memNew(*config);												get(nwidInt, nwOrig, memberidInt, memOrig);										_memberChanged(memOrig, memNew, (this->_ready>=2));					} else {						fprintf(stderr, "Can't notify of change.  Error parsing nwid or memberid: %lu-%lu\n", nwidInt, memberidInt);					}				} catch (std::exception &e) {					fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());				}			} else if (objtype == "network") {				try {					std::string id = (*config)["id"];					std::string controllerId = _myAddressStr.c_str();					std::string name = (*config)["name"];					std::string remoteTraceTarget("NULL");					if (!(*config)["remoteTraceTarget"].is_null()) {						remoteTraceTarget = (*config)["remoteTraceTarget"];					}					std::string rulesSource = (*config)["rulesSource"];					std::string caps = OSUtils::jsonDump((*config)["capabilitles"], -1);					std::string now = std::to_string(OSUtils::now());					std::string mtu = std::to_string((int)(*config)["mtu"]);					std::string mcastLimit = std::to_string((int)(*config)["multicastLimit"]);					std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);					std::string rules = OSUtils::jsonDump((*config)["rules"], -1);					std::string tags = OSUtils::jsonDump((*config)["tags"], -1);					std::string v4mode = OSUtils::jsonDump((*config)["v4AssignMode"],-1);					std::string v6mode = OSUtils::jsonDump((*config)["v6AssignMode"], -1);					bool enableBroadcast = (*config)["enableBroadcast"];					bool isPrivate = (*config)["private"];					const char *values[16] = {						id.c_str(),						controllerId.c_str(),						caps.c_str(),						enableBroadcast ? "true" : "false",						now.c_str(),						mtu.c_str(),						mcastLimit.c_str(),						name.c_str(),						isPrivate ? "true" : "false",						rtraceLevel.c_str(),						(remoteTraceTarget == "NULL" ? NULL : remoteTraceTarget.c_str()),						rules.c_str(),						rulesSource.c_str(),						tags.c_str(),						v4mode.c_str(),						v6mode.c_str(),					};					PGresult *res = PQexecParams(conn,						"UPDATE ztc_network SET controller_id = $2, capabilities = $3, enable_broadcast = $4, "						"last_updated = $5, mtu = $6, multicast_limit = $7, name = $8, private = $9, "						"remote_trace_level = $10, remote_trace_target = $11, rules = $12, rules_source = $13, "						"tags = $14, v4_assign_mode = $15, v6_assign_mode = $16 "						"WHERE id = $1",						16,						NULL,						values,						NULL,						NULL,						0);										if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error updating network record: %s\n", PQresultErrorMessage(res));						PQclear(res);						delete config;						config = nullptr;						continue;					}					PQclear(res);					res = PQexec(conn, "BEGIN");					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error beginnning transaction: %s\n", PQresultErrorMessage(res));						PQclear(res);						delete config;						config = nullptr;						continue;					}					PQclear(res);					const char *params[1] = {						id.c_str()					};					res = PQexecParams(conn, 						"DELETE FROM ztc_network_assignment_pool WHERE network_id = $1",						1,						NULL,						params,						NULL,						NULL,						0);					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error updating assignment pool: %s\n", PQresultErrorMessage(res));						PQclear(res);						PQclear(PQexec(conn, "ROLLBACK"));						delete config;						config = nullptr;						continue;					}					PQclear(res);					auto pool = (*config)["ipAssignmentPools"];					bool err = false;					for (auto i = pool.begin(); i != pool.end(); ++i) {						std::string start = (*i)["ipRangeStart"];						std::string end = (*i)["ipRangeEnd"];						const char *p[3] = {							id.c_str(),							start.c_str(),							end.c_str()						};						res = PQexecParams(conn,							"INSERT INTO ztc_network_assignment_pool (network_id, ip_range_start, ip_range_end) "							"VALUES ($1, $2, $3)",							3,							NULL,							p,							NULL,							NULL,							0);						if (PQresultStatus(res) != PGRES_COMMAND_OK) {							fprintf(stderr, "ERROR: Error updating assignment pool: %s\n", PQresultErrorMessage(res));							PQclear(res);							err = true;							break;						}						PQclear(res);					}					if (err) {						PQclear(PQexec(conn, "ROLLBACK"));						delete config;						config = nullptr;						continue;					}					res = PQexecParams(conn, 						"DELETE FROM ztc_network_route WHERE network_id = $1",						1,						NULL,						params,						NULL,						NULL,						0);					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error updating routes: %s\n", PQresultErrorMessage(res));						PQclear(res);						PQclear(PQexec(conn, "ROLLBACK"));						delete config;						config = nullptr;						continue;					}					auto routes = (*config)["routes"];					err = false;					for (auto i = routes.begin(); i != routes.end(); ++i) {						std::string t = (*i)["target"];						std::vector<std::string> target;						std::istringstream f(t);						std::string s;						while(std::getline(f, s, '/')) {							target.push_back(s);						}						if (target.empty() || target.size() != 2) {							continue;						}						std::string targetAddr = target[0];						std::string targetBits = target[1];						std::string via = "NULL";						if (!(*i)["via"].is_null()) {							via = (*i)["via"];						}						const char *p[4] = {							id.c_str(),							targetAddr.c_str(),							targetBits.c_str(),							(via == "NULL" ? NULL : via.c_str()),						};						res = PQexecParams(conn,							"INSERT INTO ztc_network_route (network_id, address, bits, via) VALUES ($1, $2, $3, $4)",							4,							NULL,							p,							NULL,							NULL,							0);						if (PQresultStatus(res) != PGRES_COMMAND_OK) {							fprintf(stderr, "ERROR: Error updating routes: %s\n", PQresultErrorMessage(res));							PQclear(res);							err = true;							break;						}						PQclear(res);					}					if (err) {						PQclear(PQexec(conn, "ROLLBACK"));						delete config;						config = nullptr;						continue;					}					res = PQexec(conn, "COMMIT");					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error committing network update: %s\n", PQresultErrorMessage(res));					}					PQclear(res);					const uint64_t nwidInt = OSUtils::jsonIntHex((*config)["nwid"], 0ULL);					if (nwidInt) {						nlohmann::json nwOrig;						nlohmann::json nwNew(*config);						get(nwidInt, nwOrig);						_networkChanged(nwOrig, nwNew, true);					} else {						fprintf(stderr, "Can't notify network changed: %lu\n", nwidInt);					}				} catch (std::exception &e) {					fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());				}			} else if (objtype == "trace") {				fprintf(stderr, "ERROR: Trace not yet implemented");			} else if (objtype == "_delete_network") {				try {					std::string networkId = (*config)["nwid"];					const char *values[1] = {						networkId.c_str()					};					PGresult * res = PQexecParams(conn,						"UPDATE ztc_network SET deleted = true WHERE id = $1",						1,						NULL,						values,						NULL,						NULL,						0);										if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error deleting network: %s\n", PQresultErrorMessage(res));					}					PQclear(res);				} catch (std::exception &e) {					fprintf(stderr, "ERROR: Error deleting network: %s\n", e.what());				}			} else if (objtype == "_delete_member") {				try {					std::string memberId = (*config)["id"];					std::string networkId = (*config)["nwid"];					const char *values[2] = {						memberId.c_str(),						networkId.c_str()					};					PGresult *res = PQexecParams(conn,						"UPDATE ztc_member SET hidden = true, deleted = true WHERE id = $1 AND network_id = $2",						2,						NULL,						values,						NULL,						NULL,						0);					if (PQresultStatus(res) != PGRES_COMMAND_OK) {						fprintf(stderr, "ERROR: Error deleting member: %s\n", PQresultErrorMessage(res));					}					PQclear(res);				} catch (std::exception &e) {					fprintf(stderr, "ERROR: Error deleting member: %s\n", e.what());				}			} else {				fprintf(stderr, "ERROR: unknown objtype");			}		} catch (std::exception &e) {			fprintf(stderr, "ERROR: Error getting objtype: %s\n", e.what());		}		delete config;		config = nullptr;		std::this_thread::sleep_for(std::chrono::milliseconds(10));	}	PQfinish(conn);	if (_run == 1) {		fprintf(stderr, "ERROR: %s commitThread should still be running! Exiting Controller.\n", _myAddressStr.c_str());		exit(7);	}}void PostgreSQL::onlineNotificationThread(){	PGconn *conn = getPgConn();	if (PQstatus(conn) == CONNECTION_BAD) {		fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(conn));		PQfinish(conn);		exit(1);	}	_connected = 1;	int64_t	lastUpdatedNetworkStatus = 0;	std::unordered_map< std::pair<uint64_t,uint64_t>,int64_t,_PairHasher > lastOnlineCumulative;		while (_run == 1) {		if (PQstatus(conn) != CONNECTION_OK) {			fprintf(stderr, "ERROR: Online Notification thread lost connection to Postgres.");			PQfinish(conn);			exit(5);		}		// map used to send notifications to front end		std::unordered_map<std::string, std::vector<std::string>> updateMap;		std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > lastOnline;		{			std::lock_guard<std::mutex> l(_lastOnline_l);			lastOnline.swap(_lastOnline);		}		PGresult *res = NULL;		std::stringstream memberUpdate;		memberUpdate << "INSERT INTO ztc_member_status (network_id, member_id, address, last_updated) VALUES ";		bool firstRun = true;		bool memberAdded = false;		for (auto i=lastOnline.begin(); i != lastOnline.end(); ++i) {			uint64_t nwid_i = i->first.first;			char nwidTmp[64];			char memTmp[64];			char ipTmp[64];			OSUtils::ztsnprintf(nwidTmp,sizeof(nwidTmp), "%.16llx", nwid_i);			OSUtils::ztsnprintf(memTmp,sizeof(memTmp), "%.10llx", i->first.second);			auto found = _networks.find(nwid_i);			if (found == _networks.end()) {				continue; // skip members trying to join non-existant networks			}			std::vector<std::string> &members = updateMap[networkId];			members.push_back(memberId);			lastOnlineCumulative[i->first] = i->second.first;			std::string networkId(nwidTmp);			std::string memberId(memTmp);						const char *qvals[2] = {				networkId.c_str(),				memberId.c_str()			};			res = PQexecParams(conn,				"SELECT id, network_id FROM ztc_member WHERE network_id = $1 AND id = $2",				2,				NULL,				qvals,				NULL,				NULL,				0);			if (PQresultStatus(res) != PGRES_TUPLES_OK) {				fprintf(stderr, "Member count failed: %s", PQerrorMessage(conn));				PQclear(res);				continue;			}			int nrows = PQntuples(res);			PQclear(res);			if (nrows == 1) {				int64_t ts = i->second.first;				std::string ipAddr = i->second.second.toIpString(ipTmp);				std::string timestamp = std::to_string(ts);								if (firstRun) {					firstRun = false;				} else {					memberUpdate << ", ";				}				memberUpdate << "('" << networkId << "', '" << memberId << '", ' << ipAddr << "', TO_TIMESTAMP(" << timestamp << "::double precision/1000))";				memberAdded = true;			} else if (nrows > 1) {				fprintf(stderr, "nrows > 1?!?");				continue;			} else {				continue;			}		}		memberUpdate << " ON CONFLICT (network_id, member_id) DO UPDATE SET address = EXCLUDED.address, last_updated = EXCLUDED.last_updated;";		if (memberAdded) {			res = PQexec(conn, memberUpdate.str().c_str());			if (res != PGRES_COMMAND_OK) {				fprintf(stderr, "Multiple insert failed: %s", PQerrorMessage(conn));			}			PQclear(res);		}		const int64_t now = OSUtils::now();		if ((now - lastUpdatedNetworkStatus) > 10000) {			lastUpdatedNetworkStatus = now;			std::vector<std::pair<uint64_t, std::shared_ptr<_Network>>> networks;			{				std::lock_guard<std::mutex> l(_networks_l);				for (auto i = _networks.begin(); i != _networks.end(); ++i) {					networks.push_back(*i);				}			}			std::stringstream networkUpdate;			networkUpdate << "INSERT INTO ztc_network_status (network_id, bridge_count, authorized_member_count, online_member_count, total_member_count, last_modified) VALUES ";			bool nwFirstRun = true;			bool networkAdded = false;			for (auto i = networks.begin(); i != networks.end(); ++i) {				char tmp[64];				Utils::hex(i->first, tmp);				std::string networkId(tmp);				std::vector<std::string> &_notUsed = updateMap[networkId];				(void)_notUsed;				uint64_t authMemberCount = 0;				uint64_t totalMemberCount = 0;				uint64_t onlineMemberCount = 0;				uint64_t bridgeCount = 0;				uint64_t ts = now;				{					std::lock_guard<std::mutex> l2(i->second->lock);					authMemberCount = i->second->authorizedMembers.size();					totalMemberCount = i->second->members.size();					bridgeCount = i->second->activeBridgeMembers.size();					for (auto m=i->second->members.begin(); m != i->second->members.end(); ++m) {						auto lo = lastOnlineCumulative.find(std::pair<uint64_t,uint64_t>(i->first, m->first));						if (lo != lastOnlineCumulative.end()) {							if ((now - lo->second) <= (ZT_NETWORK_AUTOCONF_DELAY * 2)) {								++onlineMemberCount;							} else {								lastOnlineCumulative.erase(lo);							}						}					}				}				char *nvals[1] = {					networkId.c_str();				};				res = PQExecParams(conn,					"SELECT id FROM ztc_network WHERE id = $1",					1,					NULL,					nvals,					NULL,					NULL,					0);				if (PQresultStatus(res) != PGRES_TUPLES_OK) {					fprintf(stderr, "Network lookup failed: %s", PQerrorMessage(conn));					PQclear(res);					continue;				}				int nrows = PQntuples(res);				PQclear(res);				if (nrows == 1) {					std::string bc = std::to_string(bridgeCount);					std::string amc = std::to_string(authMemberCount);					std::string omc = std::to_string(onlineMemberCount);					std::string tmc = std::to_string(totalMemberCount);					std::string timestamp = std::to_string(ts);					if (nwFirstRun) {						nwFirstRun = false;					} else {						networkUpdate << ", ";					}					networkUpdate << "('" << networkId << "', " << bc << ", " << amc << ", " << omc << ", " << tmc << ", "							<< "TO_TIMESTAMP(" << timestamp << "::double precision/1000))";					networkAdded = true;				} else if (nrows > 1) {					fprintf(stderr, "Number of networks > 1?!?!?");					continue;				} else {					continue;				}			}			networkUpdate << " ON CONFLICT (network_id) DO UPDATE SET bridge_count = EXCLUDED.bridge_count, "					<< "authorized_member_count = EXCLUDED.authorized_member_count, online_member_count = EXCLUDED.online_member_count, "					<< "total_member_count = EXCLUDED.total_member_count, last_modified = EXCLUDED.last_modified";			if (networkAdded) {				res = PQExec(conn, networkUpdate.str().c_str());				if (res != PGRES_COMMAND_OK) {					fprintf(stderr, "Error during multiple network upsert: %s", PQresultErrorMessage(res));				}				PQclear(res);			}		}		// for (auto it = updateMap.begin(); it != updateMap.end(); ++it) {		// 	std::string networkId = it->first;		// 	std::vector<std::string> members = it->second;		// 	std::stringstream queryBuilder;		// 	std::string membersStr = ::join(members, ",");		// 	queryBuilder << "NOTIFY controller, '" << networkId << ":" << membersStr << "'";		// 	std::string query = queryBuilder.str();		// 	PGresult *res = PQexec(conn,query.c_str());		// 	if (PQresultStatus(res) != PGRES_COMMAND_OK) {		// 		fprintf(stderr, "ERROR: Error sending NOTIFY: %s\n", PQresultErrorMessage(res));		// 	}		// 	PQclear(res);		// }		std::this_thread::sleep_for(std::chrono::milliseconds(0));	}	fprintf(stderr, "%s: Fell out of run loop in onlineNotificationThread\n", _myAddressStr.c_str());	PQfinish(conn);	if (_run == 1) {		fprintf(stderr, "ERROR: %s onlineNotificationThread should still be running! Exiting Controller.\n", _myAddressStr.c_str());		exit(6);	}}PGconn *PostgreSQL::getPgConn(OverrideMode m) {	if (m == ALLOW_PGBOUNCER_OVERRIDE) {		char *connStr = getenv("PGBOUNCER_CONNSTR");		if (connStr != NULL) {			fprintf(stderr, "PGBouncer Override\n");			std::string conn(connStr);			conn += " application_name=controller-"; 			conn += _myAddressStr.c_str();			return PQconnectdb(conn.c_str());		}	}	return PQconnectdb(_connString.c_str());}#endif //ZT_CONTROLLER_USE_LIBPQ
 |