void/ZeroTierOne
2
0
Fork 0
mirror of https://github.com/zerotier/ZeroTierOne.git synced 2025-06-05 03:53:44 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

.

Browse source
This commit is contained in:
Adam Ierymenko 2019-07-26 16:03:21 -07:00
parent 196bac726c
commit c8c33db1d1
1 changed files with 162 additions and 101 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

263
controller/LFDB.cpp
View file

@ -57,7 +57,6 @@ LFDB::LFDB(EmbeddedNetworkController *const nc,const Identity &myId,const char *
std::string membersSelectorName("com.zerotier.controller.lfdb:"); membersSelectorName.append(controllerAddress); membersSelectorName.append("/network/member"); std::string membersSelectorName("com.zerotier.controller.lfdb:"); membersSelectorName.append(controllerAddress); membersSelectorName.append("/network/member");
httplib::Client htcli(_lfNodeHost.c_str(),_lfNodePort,600); httplib::Client htcli(_lfNodeHost.c_str(),_lfNodePort,600);
std::ostringstream query;
int64_t timeRangeStart = 0; int64_t timeRangeStart = 0;
while (_running) { while (_running) {
{ {
@ -66,129 +65,129 @@ LFDB::LFDB(EmbeddedNetworkController *const nc,const Identity &myId,const char *
if (ns->second.dirty) { if (ns->second.dirty) {
nlohmann::json network; nlohmann::json network;
if (get(ns->first,network)) { if (get(ns->first,network)) {
nlohmann::json newrec; nlohmann::json newrec,selector0;
newrec["Selectors"] = {{ { { "Name",networksSelectorName },{ "Ordinal",ns->first } } }}; selector0["Name"] = networksSelectorName;
selector0["Ordinal"] = ns->first;
newrec["Selectors"].push_back(selector0);
newrec["Value"] = network.dump(); newrec["Value"] = network.dump();
newrec["OwnerPrivate"] = _lfOwnerPrivate; newrec["OwnerPrivate"] = _lfOwnerPrivate;
newrec["MaskingKey"] = controllerAddress; newrec["MaskingKey"] = controllerAddress;
auto resp = htcli.Post("/make",newrec.dump(),"application/json"); newrec["PulseIfUnchanged"] = true;
if (resp->status == 200) { printf("%s\n",newrec.dump().c_str());
ns->second.dirty = false; auto resp = htcli.Post("/makerecord",newrec.dump(),"application/json");
if (resp) {
if (resp->status == 200) {
ns->second.dirty = false;
printf("SET network %.16llx %s\n",ns->first,resp->body.c_str());
} else {
fprintf(stderr,"ERROR: LFDB: %d from node (create/update network): %s" ZT_EOL_S,resp->status,resp->body.c_str());
}
} else { } else {
fprintf(stderr,"ERROR: LFDB: %d from node (create/update network): %s" ZT_EOL_S,resp->status,resp->body.c_str()); fprintf(stderr,"ERROR: LFDB: node is offline" ZT_EOL_S);
} }
} }
} }
for(auto ms=ns->second.members.begin();ms!=ns->second.members.end();++ms) { for(auto ms=ns->second.members.begin();ms!=ns->second.members.end();++ms) {
if ((_storeOnlineState)&&(ms->second.lastOnlineDirty)) { if ((_storeOnlineState)&&(ms->second.lastOnlineDirty)&&(ms->second.lastOnlineAddress)) {
char tmp[1024],tmp2[128];
OSUtils::ztsnprintf(tmp,sizeof(tmp),"com.zerotier.controller.lfdb:%s/network/%.16llx/online",controllerAddress,(unsigned long long)ns->first);
ms->second.lastOnlineAddress.toIpString(tmp2);
nlohmann::json newrec,selector0,selector1,selectors;
selector0["Name"] = tmp;
selector0["Ordinal"] = ms->first;
selector1["Name"] = tmp2;
selector1["Ordinal"] = 0;
selectors.push_back(selector0);
selectors.push_back(selector1);
newrec["Selectors"] = selectors;
newrec["Value"] = tmp2;
newrec["OwnerPrivate"] = _lfOwnerPrivate;
newrec["MaskingKey"] = controllerAddress;
newrec["Timestamp"] = ms->second.lastOnlineTime;
newrec["PulseIfUnchanged"] = true;
auto resp = htcli.Post("/makerecord",newrec.dump(),"application/json");
if (resp) {
if (resp->status == 200) {
ms->second.lastOnlineDirty = false;
printf("SET member online %.16llx %.10llx %s\n",ns->first,ms->first,resp->body.c_str());
} else {
fprintf(stderr,"ERROR: LFDB: %d from node (create/update member): %s" ZT_EOL_S,resp->status,resp->body.c_str());
}
} else {
fprintf(stderr,"ERROR: LFDB: node is offline" ZT_EOL_S);
}
} }
if (ms->second.dirty) { if (ms->second.dirty) {
nlohmann::json network,member; nlohmann::json network,member;
if (get(ns->first,network,ms->first,member)) { if (get(ns->first,network,ms->first,member)) {
nlohmann::json newrec; nlohmann::json newrec,selector0,selector1,selectors;
newrec["Selectors"] = {{ { { "Name",networksSelectorName },{ "Ordinal",ns->first } },{ { "Name",membersSelectorName },{ "Ordinal",ms->first } } }}; selector0["Name"] = networksSelectorName;
selector0["Ordinal"] = ns->first;
selector1["Name"] = membersSelectorName;
selector1["Ordinal"] = ms->first;
selectors.push_back(selector0);
selectors.push_back(selector1);
newrec["Selectors"] = selectors;
newrec["Value"] = member.dump(); newrec["Value"] = member.dump();
newrec["OwnerPrivate"] = _lfOwnerPrivate; newrec["OwnerPrivate"] = _lfOwnerPrivate;
newrec["MaskingKey"] = controllerAddress; newrec["MaskingKey"] = controllerAddress;
auto resp = htcli.Post("/make",newrec.dump(),"application/json"); newrec["PulseIfUnchanged"] = true;
if (resp->status == 200) { auto resp = htcli.Post("/makerecord",newrec.dump(),"application/json");
ms->second.dirty = false; if (resp) {
} else { if (resp->status == 200) {
fprintf(stderr,"ERROR: LFDB: %d from node (create/update member): %s" ZT_EOL_S,resp->status,resp->body.c_str()); ms->second.dirty = false;
} printf("SET member %.16llx %.10llx %s\n",ns->first,ms->first,resp->body.c_str());
} } else {
} fprintf(stderr,"ERROR: LFDB: %d from node (create/update member): %s" ZT_EOL_S,resp->status,resp->body.c_str());
}
}
}
query.clear();
query
<< '{'
<< "\"Ranges\":[{"
<< "\"Name\":\"" << networksSelectorName << "\","
<< "\"Range\":[0,18446744073709551615]"
<< "}],"
<< "\"TimeRange\":[" << timeRangeStart << ",18446744073709551615],"
<< "\"MaskingKey\":\"" << controllerAddress << "\","
<< "\"Owners\":[\"" << _lfOwnerPublic << "\"]"
<< '}';
auto resp = htcli.Post("/query",query.str(),"application/json");
if (resp->status == 200) {
nlohmann::json results(OSUtils::jsonParse(resp->body));
if ((results.is_array())&&(results.size() > 0)) {
for(std::size_t ri=0;ri<results.size();++ri) {
nlohmann::json &rset = results[ri];
if ((rset.is_array())&&(rset.size() > 0)) {
nlohmann::json &result = rset[0];
if (result.is_object()) {
nlohmann::json &record = result["Record"];
if (record.is_object()) {
const std::string recordValue = result["Value"];
nlohmann::json network(OSUtils::jsonParse(recordValue));
if (network.is_object()) {
const std::string idstr = network["id"];
const uint64_t id = Utils::hexStrToU64(idstr.c_str());
if ((id >> 24) == controllerAddressInt) {
std::lock_guard<std::mutex> sl(_state_l);
_NetworkState &ns = _state[id];
if (!ns.dirty) {
nlohmann::json nullJson;
_networkChanged(nullJson,network,false);
}
}
} }
} else {
fprintf(stderr,"ERROR: LFDB: node is offline" ZT_EOL_S);
} }
} }
} }
} }
} }
} else {
fprintf(stderr,"ERROR: LFDB: %d from node: %s" ZT_EOL_S,resp->status,resp->body.c_str());
} }
query.clear(); {
query std::ostringstream query;
<< '{' query
<< "\"Ranges\":[{" << '{'
<< "\"Name\":\"" << networksSelectorName << "\"," << "\"Ranges\":[{"
<< "\"Range\":[0,18446744073709551615]" << "\"Name\":\"" << networksSelectorName << "\","
<< "},{" << "\"Range\":[0,18446744073709551615]"
<< "\"Name\":\"" << membersSelectorName << "\"," << "}],"
<< "\"Range\":[0,18446744073709551615]" << "\"TimeRange\":[" << timeRangeStart << ",18446744073709551615],"
<< "}]," << "\"MaskingKey\":\"" << controllerAddress << "\","
<< "\"TimeRange\":[" << timeRangeStart << ",18446744073709551615]," << "\"Owners\":[\"" << _lfOwnerPublic << "\"]"
<< "\"MaskingKey\":\"" << controllerAddress << "\"," << '}';
<< "\"Owners\":[\"" << _lfOwnerPublic << "\"]" auto resp = htcli.Post("/query",query.str(),"application/json");
<< '}'; if (resp) {
resp = htcli.Post("/query",query.str(),"application/json"); if (resp->status == 200) {
if (resp->status == 200) { nlohmann::json results(OSUtils::jsonParse(resp->body));
nlohmann::json results(OSUtils::jsonParse(resp->body)); if ((results.is_array())&&(results.size() > 0)) {
if ((results.is_array())&&(results.size() > 0)) { for(std::size_t ri=0;ri<results.size();++ri) {
for(std::size_t ri=0;ri<results.size();++ri) { nlohmann::json &rset = results[ri];
nlohmann::json &rset = results[ri]; if ((rset.is_array())&&(rset.size() > 0)) {
if ((rset.is_array())&&(rset.size() > 0)) { nlohmann::json &result = rset[0];
nlohmann::json &result = rset[0]; if (result.is_object()) {
if (result.is_object()) { nlohmann::json &record = result["Record"];
nlohmann::json &record = result["Record"]; if (record.is_object()) {
if (record.is_object()) { const std::string recordValue = result["Value"];
const std::string recordValue = result["Value"]; printf("GET network %s\n",recordValue.c_str());
nlohmann::json member(OSUtils::jsonParse(recordValue)); nlohmann::json network(OSUtils::jsonParse(recordValue));
if (member.is_object()) { if (network.is_object()) {
const std::string nwidstr = member["nwid"]; const std::string idstr = network["id"];
const std::string idstr = member["id"]; const uint64_t id = Utils::hexStrToU64(idstr.c_str());
const uint64_t nwid = Utils::hexStrToU64(nwidstr.c_str()); if ((id >> 24) == controllerAddressInt) {
const uint64_t id = Utils::hexStrToU64(idstr.c_str()); std::lock_guard<std::mutex> sl(_state_l);
if ((id)&&((nwid >> 24) == controllerAddressInt)) { _NetworkState &ns = _state[id];
std::lock_guard<std::mutex> sl(_state_l); if (!ns.dirty) {
auto ns = _state.find(nwid); nlohmann::json nullJson;
if (ns != _state.end()) { _networkChanged(nullJson,network,false);
_MemberState &ms = ns->second.members[id]; }
if (!ms.dirty) {
nlohmann::json nullJson;
_memberChanged(nullJson,member,false);
} }
} }
} }
@ -196,10 +195,72 @@ LFDB::LFDB(EmbeddedNetworkController *const nc,const Identity &myId,const char *
} }
} }
} }
} else {
fprintf(stderr,"ERROR: LFDB: %d from node: %s" ZT_EOL_S,resp->status,resp->body.c_str());
} }
} else {
fprintf(stderr,"ERROR: LFDB: node is offline" ZT_EOL_S);
}
}
{
std::ostringstream query;
query
<< '{'
<< "\"Ranges\":[{"
<< "\"Name\":\"" << networksSelectorName << "\","
<< "\"Range\":[0,18446744073709551615]"
<< "},{"
<< "\"Name\":\"" << membersSelectorName << "\","
<< "\"Range\":[0,18446744073709551615]"
<< "}],"
<< "\"TimeRange\":[" << timeRangeStart << ",18446744073709551615],"
<< "\"MaskingKey\":\"" << controllerAddress << "\","
<< "\"Owners\":[\"" << _lfOwnerPublic << "\"]"
<< '}';
auto resp = htcli.Post("/query",query.str(),"application/json");
if (resp) {
if (resp->status == 200) {
nlohmann::json results(OSUtils::jsonParse(resp->body));
if ((results.is_array())&&(results.size() > 0)) {
for(std::size_t ri=0;ri<results.size();++ri) {
nlohmann::json &rset = results[ri];
if ((rset.is_array())&&(rset.size() > 0)) {
nlohmann::json &result = rset[0];
if (result.is_object()) {
nlohmann::json &record = result["Record"];
if (record.is_object()) {
const std::string recordValue = result["Value"];
printf("GET member %s\n",recordValue.c_str());
nlohmann::json member(OSUtils::jsonParse(recordValue));
if (member.is_object()) {
const std::string nwidstr = member["nwid"];
const std::string idstr = member["id"];
const uint64_t nwid = Utils::hexStrToU64(nwidstr.c_str());
const uint64_t id = Utils::hexStrToU64(idstr.c_str());
if ((id)&&((nwid >> 24) == controllerAddressInt)) {
std::lock_guard<std::mutex> sl(_state_l);
auto ns = _state.find(nwid);
if (ns != _state.end()) {
_MemberState &ms = ns->second.members[id];
if (!ms.dirty) {
nlohmann::json nullJson;
_memberChanged(nullJson,member,false);
}
}
}
}
}
}
}
}
}
} else {
fprintf(stderr,"ERROR: LFDB: %d from node: %s" ZT_EOL_S,resp->status,resp->body.c_str());
}
} else {
fprintf(stderr,"ERROR: LFDB: node is offline" ZT_EOL_S);
} }
} else {
fprintf(stderr,"ERROR: LFDB: %d from node: %s" ZT_EOL_S,resp->status,resp->body.c_str());
} }
timeRangeStart = time(nullptr) - 120; // start next query 2m before now to avoid losing updates timeRangeStart = time(nullptr) - 120; // start next query 2m before now to avoid losing updates