summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGrant Limberg <grant.limberg@zerotier.com>2018-09-05 11:49:07 -0700
committerGrant Limberg <grant.limberg@zerotier.com>2018-09-05 11:49:07 -0700
commitd701f4e3311fc75f07a60a81bc922eb038b9df8d (patch)
tree7368d7e1ccf753817850d0c72c88edb566c18ed9
parent92abc92463789bde771fb03985fcdd5ce9784e50 (diff)
downloadinfinitytier-d701f4e3311fc75f07a60a81bc922eb038b9df8d.tar.gz
infinitytier-d701f4e3311fc75f07a60a81bc922eb038b9df8d.zip
Added some exception hadling. Dont want to live *too* dangerously here
-rw-r--r--controller/PostgreSQL.cpp1033
1 files changed, 533 insertions, 500 deletions
diff --git a/controller/PostgreSQL.cpp b/controller/PostgreSQL.cpp
index dc4f400d..420b0cac 100644
--- a/controller/PostgreSQL.cpp
+++ b/controller/PostgreSQL.cpp
@@ -153,58 +153,20 @@ void PostgreSQL::nodeIsOnline(const uint64_t networkId, const uint64_t memberId,
void PostgreSQL::initializeNetworks(PGconn *conn)
{
- if (PQstatus(conn) != CONNECTION_OK) {
- fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));
- exit(1);
- }
+ 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);
- }
+ const char *params[1] = {
+ _myAddressStr.c_str()
+ };
- int numRows = PQntuples(res);
- for (int i = 0; i < numRows; ++i) {
- json empty;
- json config;
- config["nwid"] = PQgetvalue(res, i, 0);
- config["creationTime"] = std::stoull(PQgetvalue(res, i, 1));
- config["capabilities"] = json::parse(PQgetvalue(res, i, 2));
- config["enableBroadcast"] = (strcmp(PQgetvalue(res, i, 3),"true")==0);
- config["lastModified"] = std::stoull(PQgetvalue(res, i, 4));
- config["mtu"] = std::stoi(PQgetvalue(res, i, 5));
- config["multicastLimit"] = std::stoi(PQgetvalue(res, i, 6));
- config["name"] = PQgetvalue(res, i, 7);
- config["private"] = (strcmp(PQgetvalue(res, i, 8),"true")==0);
- config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 9));
- config["remoteTraceTarget"] = PQgetvalue(res, i, 10);
- config["revision"] = std::stoull(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",
+ 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,
@@ -212,169 +174,217 @@ void PostgreSQL::initializeNetworks(PGconn *conn)
NULL,
0);
- if (PQresultStatus(r2) != PGRES_TUPLES_OK) {
- fprintf(stderr, "ERROR: Error retreiving IP pools for network: %s\n", PQresultErrorMessage(r2));
- PQclear(r2);
+ if (PQresultStatus(res) != PGRES_TUPLES_OK) {
+ fprintf(stderr, "Networks Initialization Failed: %s", PQerrorMessage(conn));
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);
- }
+ int numRows = PQntuples(res);
+ for (int i = 0; i < numRows; ++i) {
+ json empty;
+ json config;
+ config["nwid"] = PQgetvalue(res, i, 0);
+ config["creationTime"] = std::stoull(PQgetvalue(res, i, 1));
+ config["capabilities"] = json::parse(PQgetvalue(res, i, 2));
+ config["enableBroadcast"] = (strcmp(PQgetvalue(res, i, 3),"true")==0);
+ config["lastModified"] = std::stoull(PQgetvalue(res, i, 4));
+ config["mtu"] = std::stoi(PQgetvalue(res, i, 5));
+ config["multicastLimit"] = std::stoi(PQgetvalue(res, i, 6));
+ config["name"] = PQgetvalue(res, i, 7);
+ config["private"] = (strcmp(PQgetvalue(res, i, 8),"true")==0);
+ config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 9));
+ config["remoteTraceTarget"] = PQgetvalue(res, i, 10);
+ config["revision"] = std::stoull(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,
+ params,
+ 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);
+ }
- PQclear(r2);
+ 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);
- r2 = PQexecParams(conn,
- "SELECT host(address), bits, host(via) FROM ztc_network_route WHERE network_id = $1",
- 1,
- NULL,
- params,
- NULL,
- NULL,
- 0);
+ config["ipAssignmentPools"].push_back(ip);
+ }
- 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);
+ r2 = PQexecParams(conn,
+ "SELECT host(address), bits, host(via) FROM ztc_network_route WHERE network_id = $1",
+ 1,
+ NULL,
+ params,
+ NULL,
+ NULL,
+ 0);
- json route;
- route["target"] = addr + "/" + bits;
+ if (PQresultStatus(r2) != PGRES_TUPLES_OK) {
+ fprintf(stderr, "ERROR: Error retreiving routes for network: %s\n", PQresultErrorMessage(r2));
+ PQclear(r2);
+ PQclear(res);
+ exit(1);
+ }
- if (via == "NULL") {
- route["via"] = nullptr;
- } else {
- route["via"] = via;
+ 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);
}
- config["routes"].push_back(route);
- }
- PQclear(r2);
-
- _networkChanged(empty, config, false);
- }
+ PQclear(r2);
+
+ _networkChanged(empty, config, false);
+ }
- PQclear(res);
+ 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());
+ 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();
}
- _readyLock.unlock();
+ } catch (std::exception &e) {
+ fprintf(stderr, "ERROR: Error initializing networks: %s", e.what());
+ exit(-1);
}
}
void PostgreSQL::initializeMembers(PGconn *conn)
{
- 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);
- }
+ try {
+ if (PQstatus(conn) != CONNECTION_OK) {
+ fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));
+ 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));
- config["id"] = memberId;
- config["nwid"] = networkId;
- config["activeBridge"] = (strcmp(PQgetvalue(res, i, 3), "true") == 0);
- config["authorized"] = (strcmp(PQgetvalue(res, i, 4), "true") == 0);
- config["capabilities"] = json::parse(PQgetvalue(res, i, 5));
- config["creationTime"] = std::stoull(PQgetvalue(res, i, 6));
- config["identity"] = PQgetvalue(res, i, 7);
- config["lastAuthorizedTime"] = std::stoull(PQgetvalue(res, i, 8));
- config["lastDeauthorizedTime"] = std::stoull(PQgetvalue(res, i, 9));
- config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 10));
- config["remoteTraceTarget"] = PQgetvalue(res, i, 11);
- config["tags"] = json::parse(PQgetvalue(res, i, 12));
- config["vMajor"] = std::stoi(PQgetvalue(res, i, 13));
- config["vMinor"] = std::stoi(PQgetvalue(res, i, 14));
- config["vRev"] = std::stoi(PQgetvalue(res, i, 15));
- config["vProto"] = std::stoi(PQgetvalue(res, i, 16));
- config["noAutoAssignIps"] = (strcmp(PQgetvalue(res, i, 17), "true") == 0);
- config["revision"] = std::stoull(PQgetvalue(res, i, 18));
- config["objtype"] = "member";
- config["ipAssignments"] = json::array();
- const char *p2[2] = {
- memberId.c_str(),
- networkId.c_str()
+ const char *params[1] = {
+ _myAddressStr.c_str()
};
- PGresult *r2 = PQexecParams(conn,
- "SELECT address FROM ztc_member_ip_assignment WHERE member_id = $1 AND network_id = $2",
- 2,
+ 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,
- p2,
+ params,
NULL,
NULL,
0);
- if (PQresultStatus(r2) != PGRES_TUPLES_OK) {
+ if (PQresultStatus(res) != 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));
- }
+ 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));
+ config["id"] = memberId;
+ config["nwid"] = networkId;
+ config["activeBridge"] = (strcmp(PQgetvalue(res, i, 3), "true") == 0);
+ config["authorized"] = (strcmp(PQgetvalue(res, i, 4), "true") == 0);
+ config["capabilities"] = json::parse(PQgetvalue(res, i, 5));
+ config["creationTime"] = std::stoull(PQgetvalue(res, i, 6));
+ config["identity"] = PQgetvalue(res, i, 7);
+ config["lastAuthorizedTime"] = std::stoull(PQgetvalue(res, i, 8));
+ config["lastDeauthorizedTime"] = std::stoull(PQgetvalue(res, i, 9));
+ config["remoteTraceLevel"] = std::stoi(PQgetvalue(res, i, 10));
+ config["remoteTraceTarget"] = PQgetvalue(res, i, 11);
+ config["tags"] = json::parse(PQgetvalue(res, i, 12));
+ config["vMajor"] = std::stoi(PQgetvalue(res, i, 13));
+ config["vMinor"] = std::stoi(PQgetvalue(res, i, 14));
+ config["vRev"] = std::stoi(PQgetvalue(res, i, 15));
+ config["vProto"] = std::stoi(PQgetvalue(res, i, 16));
+ config["noAutoAssignIps"] = (strcmp(PQgetvalue(res, i, 17), "true") == 0);
+ config["revision"] = std::stoull(PQgetvalue(res, i, 18));
+ config["objtype"] = "member";
+ config["ipAssignments"] = json::array();
+ const char *p2[2] = {
+ memberId.c_str(),
+ networkId.c_str()
+ };
- _memberChanged(empty, config, false);
- }
+ PGresult *r2 = PQexecParams(conn,
+ "SELECT 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);
+ }
- PQclear(res);
+ int n = PQntuples(r2);
+ for (int j = 0; j < n; ++j) {
+ config["ipAssignments"].push_back(PQgetvalue(r2, j, 0));
+ }
- 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());
+ _memberChanged(empty, config, false);
}
- _readyLock.unlock();
+
+ 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);
}
}
@@ -565,7 +575,7 @@ void PostgreSQL::commitThread()
{
PGconn *conn = PQconnectdb(_path.c_str());
if (PQstatus(conn) == CONNECTION_BAD) {
- fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(conn));
+ fprintf(stderr, "ERROR: Connection to database failed: %s\n", PQerrorMessage(conn));
PQfinish(conn);
exit(1);
}
@@ -575,362 +585,385 @@ void PostgreSQL::commitThread()
if (!config) {
continue;
}
- const std::string objtype = (*config)["objtype"];
- if (objtype == "member") {
- std::string memberId = (*config)["id"];
- std::string networkId = (*config)["nwid"];
- std::string name = (*config)["name"];
- std::string description = (*config)["description"];
- std::string identity = (*config)["identity"];
- std::string target = "NULL";
- if (!(*config)["remoteTraceTarget"].is_null()) {
- target = (*config)["remoteTraceTarget"];
- }
- const char *values[19] = {
- memberId.c_str(),
- networkId.c_str(),
- ((*config)["activeBridge"] ? "true" : "false"),
- ((*config)["authorized"] ? "true" : "false"),
- OSUtils::jsonDump((*config)["capabilities"], -1).c_str(),
- name.c_str(),
- description.c_str(),
- identity.c_str(),
- std::to_string((long long)(*config)["lastAuthorizedTime"]).c_str(),
- std::to_string((long long)(*config)["lastDeauthorizedTime"]).c_str(),
- ((*config)["noAutoAssignIps"] ? "true" : "false"),
- std::to_string((int)(*config)["remoteTraceLevel"]).c_str(),
- (target == "NULL") ? NULL : target.c_str(),
- std::to_string((unsigned long long)(*config)["revision"]).c_str(),
- OSUtils::jsonDump((*config)["tags"], -1).c_str(),
- std::to_string((int)(*config)["vMajor"]).c_str(),
- std::to_string((int)(*config)["vMinor"]).c_str(),
- std::to_string((int)(*config)["vRev"]).c_str(),
- std::to_string((int)(*config)["vProto"]).c_str()
- };
+ if (PQstatus(conn) == CONNECTION_BAD) {
+ fprintf(stderr, "ERROR: Connection to database failed: %s\n", PQerrorMessage(conn));
+ PQfinish(conn);
+ 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 name = (*config)["name"];
+ std::string description = (*config)["description"];
+ std::string identity = (*config)["identity"];
+ std::string target = "NULL";
+ if (!(*config)["remoteTraceTarget"].is_null()) {
+ target = (*config)["remoteTraceTarget"];
+ }
+ const char *values[19] = {
+ memberId.c_str(),
+ networkId.c_str(),
+ ((*config)["activeBridge"] ? "true" : "false"),
+ ((*config)["authorized"] ? "true" : "false"),
+ OSUtils::jsonDump((*config)["capabilities"], -1).c_str(),
+ name.c_str(),
+ description.c_str(),
+ identity.c_str(),
+ std::to_string((long long)(*config)["lastAuthorizedTime"]).c_str(),
+ std::to_string((long long)(*config)["lastDeauthorizedTime"]).c_str(),
+ ((*config)["noAutoAssignIps"] ? "true" : "false"),
+ std::to_string((int)(*config)["remoteTraceLevel"]).c_str(),
+ (target == "NULL") ? NULL : target.c_str(),
+ std::to_string((unsigned long long)(*config)["revision"]).c_str(),
+ OSUtils::jsonDump((*config)["tags"], -1).c_str(),
+ std::to_string((int)(*config)["vMajor"]).c_str(),
+ std::to_string((int)(*config)["vMinor"]).c_str(),
+ std::to_string((int)(*config)["vRev"]).c_str(),
+ std::to_string((int)(*config)["vProto"]).c_str()
+ };
+
+ PGresult *res = PQexecParams(conn,
+ "INSERT INTO ztc_member (id, network_id, active_bridge, authorized, capabilities, "
+ "name, description, 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, $7, $8, "
+ "TO_TIMESTAMP($9::double precision/1000), TO_TIMESTAMP($10::double precision/1000), "
+ "$11, $12, $13, $14, $15, $16, $17, $18, $19) ON CONFLICT (network_id, id) DO UPDATE SET "
+ "active_bridge = EXCLUDED.active_bridge, authorized = EXCDLUDED.authorized, capabilities = EXCLUDED.capabilities, "
+ "name = EXCLUDED.name, description = EXCLUDED.description, "
+ "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",
+ 20,
+ NULL,
+ values,
+ NULL,
+ NULL,
+ 0);
+
+ if (PQresultStatus(res) != PGRES_COMMAND_OK) {
+ fprintf(stderr, "ERROR: Error updating member: %s\n", PQresultErrorMessage(res));
+ PQclear(res);
+ continue;
+ }
- PGresult *res = PQexecParams(conn,
- "INSERT INTO ztc_member (id, network_id, active_bridge, authorized, capabilities, "
- "name, description, 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, $7, $8, "
- "TO_TIMESTAMP($9::double precision/1000), TO_TIMESTAMP($10::double precision/1000), "
- "$11, $12, $13, $14, $15, $16, $17, $18, $19) ON CONFLICT (network_id, id) DO UPDATE SET "
- "active_bridge = EXCLUDED.active_bridge, authorized = EXCDLUDED.authorized, capabilities = EXCLUDED.capabilities, "
- "name = EXCLUDED.name, description = EXCLUDED.description, "
- "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",
- 20,
- NULL,
- values,
- NULL,
- NULL,
- 0);
-
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error updating member: %s\n", PQresultErrorMessage(res));
- PQclear(res);
- continue;
- }
+ PQclear(res);
- PQclear(res);
+ res = PQexec(conn, "BEGIN");
+ if (PQresultStatus(res) != PGRES_COMMAND_OK) {
+ fprintf(stderr, "ERROR: Error beginning transaction: %s\n", PQresultErrorMessage(res));
+ PQclear(res);
+ continue;
+ }
- res = PQexec(conn, "BEGIN");
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error beginning transaction: %s\n", PQresultErrorMessage(res));
- PQclear(res);
- continue;
- }
+ PQclear(res);
- PQclear(res);
+ const char *v2[2] = {
+ memberId.c_str(),
+ networkId.c_str()
+ };
- 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);
- 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"));;
+ continue;
+ }
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error updating IP address assignments: %s\n", PQresultErrorMessage(res));
- PQclear(res);
- PQclear(PQexec(conn, "ROLLBACK"));;
- continue;
- }
+ PQclear(res);
- PQclear(res);
+ for (auto i = (*config)["ipAssignments"].begin(); i != (*config)["ipAssignments"].end(); ++i) {
+ std::string addr = *i;
+ 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"));
+ continue;
+ }
+ }
- for (auto i = (*config)["ipAssignments"].begin(); i != (*config)["ipAssignments"].end(); ++i) {
- std::string addr = *i;
- const char *v3[3] = {
- memberId.c_str(),
- networkId.c_str(),
- addr.c_str()
- };
+ res = PQexec(conn, "COMMIT");
+ if (PQresultStatus(res) != PGRES_COMMAND_OK) {
+ fprintf(stderr, "ERROR: Error committing ip address data: %s\n", PQresultErrorMessage(res));
+ }
- 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"));
- continue;
+ } catch (std::exception &e) {
+ fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());
}
- }
-
- res = PQexec(conn, "COMMIT");
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error committing ip address data: %s\n", PQresultErrorMessage(res));
- }
-
- PQclear(res);
- } else if (objtype == "network") {
- 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"];
- const char *values[16] = {
- id.c_str(),
- controllerId.c_str(),
- OSUtils::jsonDump((*config)["capabilitles"], -1).c_str(),
- ((*config)["enableBroadcast"] ? "true" : "false"),
- std::to_string(OSUtils::now()).c_str(),
- std::to_string((int)(*config)["mtu"]).c_str(),
- std::to_string((int)(*config)["multicastLimit"]).c_str(),
- name.c_str(),
- ((*config)["private"] ? "true" : "false"),
- std::to_string((int)(*config)["remoteTraceLevel"]).c_str(),
- (remoteTraceTarget == "NULL" ? NULL : remoteTraceTarget.c_str()),
- OSUtils::jsonDump((*config)["rules"], -1).c_str(),
- rulesSource.c_str(),
- OSUtils::jsonDump((*config)["tags"], -1).c_str(),
- OSUtils::jsonDump((*config)["v4AssignMode"],-1).c_str(),
- OSUtils::jsonDump((*config)["v6AssignMode"], -1).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);
- 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);
- continue;
- }
+ } 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"];
+ const char *values[16] = {
+ id.c_str(),
+ controllerId.c_str(),
+ OSUtils::jsonDump((*config)["capabilitles"], -1).c_str(),
+ ((*config)["enableBroadcast"] ? "true" : "false"),
+ std::to_string(OSUtils::now()).c_str(),
+ std::to_string((int)(*config)["mtu"]).c_str(),
+ std::to_string((int)(*config)["multicastLimit"]).c_str(),
+ name.c_str(),
+ ((*config)["private"] ? "true" : "false"),
+ std::to_string((int)(*config)["remoteTraceLevel"]).c_str(),
+ (remoteTraceTarget == "NULL" ? NULL : remoteTraceTarget.c_str()),
+ OSUtils::jsonDump((*config)["rules"], -1).c_str(),
+ rulesSource.c_str(),
+ OSUtils::jsonDump((*config)["tags"], -1).c_str(),
+ OSUtils::jsonDump((*config)["v4AssignMode"],-1).c_str(),
+ OSUtils::jsonDump((*config)["v6AssignMode"], -1).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);
+ continue;
+ }
- PQclear(res);
+ 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"));
- continue;
- }
+ res = PQexec(conn, "BEGIN");
+ if (PQresultStatus(res) != PGRES_COMMAND_OK) {
+ fprintf(stderr, "ERROR: Error beginnning transaction: %s\n", PQresultErrorMessage(res));
+ PQclear(res);
+ continue;
+ }
- PQclear(res);
+ 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()
- };
+ 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"));
+ continue;
+ }
- 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"));
- continue;
- }
- res = PQexecParams(conn,
- "DELETE FROM ztc_network_route WHERE network_id = $1",
- 1,
- NULL,
- params,
- NULL,
- NULL,
- 0);
+ 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"));
+ continue;
+ }
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error updating routes: %s\n", PQresultErrorMessage(res));
- PQclear(res);
- PQclear(PQexec(conn, "ROLLBACK"));
- 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"));
+ 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()),
- };
+ 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"];
+ }
- res = PQexecParams(conn,
- "INSERT INTO ztc_network_route (network_id, address, bits, via) VALUES ($1, $2, $3, $4)",
- 4,
- NULL,
- p,
- NULL,
- NULL,
- 0);
+ 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"));
+ continue;
+ }
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error updating routes: %s\n", PQresultErrorMessage(res));
+ res = PQexec(conn, "COMMIT");
+ if (PQresultStatus(res) != PGRES_COMMAND_OK) {
+ fprintf(stderr, "ERROR: Error committing network update: %s\n", PQresultErrorMessage(res));
+ }
PQclear(res);
- err = true;
- break;
+ } catch (std::exception &e) {
+ fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());
}
- PQclear(res);
- }
- if (err) {
- PQclear(PQexec(conn, "ROLLBAcK"));
- continue;
- }
-
- res = PQexec(conn, "COMMIT");
- if (PQresultStatus(res) != PGRES_COMMAND_OK) {
- fprintf(stderr, "ERROR: Error committing network update: %s\n", PQresultErrorMessage(res));
- }
- PQclear(res);
- } else if (objtype == "trace") {
- fprintf(stderr, "ERROR: Trace not yet implemented");
- } else if (objtype == "_delete_network") {
-
- 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);
- } else if (objtype == "_delete_member") {
- std::string memberId = (*config)["id"];
- std::string networkId = (*config)["nwid"];
+ } 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));
+ }
- const char *values[2] = {
- memberId.c_str(),
- networkId.c_str()
- };
+ 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);
- 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));
+ }
- 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");
}
-
- PQclear(res);
- } else {
- fprintf(stderr, "ERROR: unknown objtype");
+ } catch (std::exception &e) {
+ fprintf(stderr, "ERROR: Error getting objtype: %s\n", e.what());
}
-
delete config;
std::this_thread::sleep_for(std::chrono::milliseconds(10));