mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-12-15 14:51:08 -08:00
wire up pubsub notifications from controller to frontend
This commit is contained in:
Grant Limberg
parent
ee87eb3101
commit
012443acfa
12 changed files with 524 additions and 55 deletions
|
|
@ -1,5 +1,6 @@
|
|||
#include "PubSubWriter.hpp"
|
||||
|
||||
#include "../../osdep/OSUtils.hpp"
|
||||
#include "CtlUtil.hpp"
|
||||
#include "member.pb.h"
|
||||
#include "member_status.pb.h"
|
||||
|
|
@ -15,6 +16,12 @@
|
|||
namespace pubsub = ::google::cloud::pubsub;
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
pbmessages::NetworkChange
|
||||
networkChangeFromJson(std::string controllerID, const nlohmann::json& oldNetwork, const nlohmann::json& newNetwork);
|
||||
pbmessages::MemberChange
|
||||
memberChangeFromJson(std::string controllerID, const nlohmann::json& oldMember, const nlohmann::json& newMember);
|
||||
|
||||
PubSubWriter::PubSubWriter(std::string project, std::string topic, std::string controller_id)
|
||||
: _controller_id(controller_id)
|
||||
, _project(project)
|
||||
|
|
@ -44,12 +51,9 @@ PubSubWriter::~PubSubWriter()
|
|||
{
|
||||
}
|
||||
|
||||
bool PubSubWriter::publishMessage(const std::string& payload, const std::string& frontend)
|
||||
bool PubSubWriter::publishMessage(const std::string& payload)
|
||||
{
|
||||
std::vector<std::pair<std::string, std::string> > attributes;
|
||||
if (! frontend.empty()) {
|
||||
attributes.emplace_back("frontend", frontend);
|
||||
}
|
||||
attributes.emplace_back("controller_id", _controller_id);
|
||||
|
||||
auto msg = pubsub::MessageBuilder {}.SetData(payload).SetAttributes(attributes).Build();
|
||||
|
|
@ -63,30 +67,28 @@ bool PubSubWriter::publishMessage(const std::string& payload, const std::string&
|
|||
return true;
|
||||
}
|
||||
|
||||
bool PubSubWriter::publishNetworkChange(const nlohmann::json& networkJson, const std::string& frontend)
|
||||
bool PubSubWriter::publishNetworkChange(const nlohmann::json& oldNetwork, const nlohmann::json& newNetwork)
|
||||
{
|
||||
pbmessages::NetworkChange nc;
|
||||
// nc.mutable_new_()->CopyFrom(fromJson<pbmessages::NetworkChange_Network>(networkJson));
|
||||
pbmessages::NetworkChange nc = networkChangeFromJson(_controller_id, oldNetwork, newNetwork);
|
||||
std::string payload;
|
||||
if (! nc.SerializeToString(&payload)) {
|
||||
fprintf(stderr, "Failed to serialize NetworkChange protobuf message\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
return publishMessage(payload, frontend);
|
||||
return publishMessage(payload);
|
||||
}
|
||||
|
||||
bool PubSubWriter::publishMemberChange(const nlohmann::json& memberJson, const std::string& frontend)
|
||||
bool PubSubWriter::publishMemberChange(const nlohmann::json& oldMember, const nlohmann::json& newMember)
|
||||
{
|
||||
pbmessages::MemberChange mc;
|
||||
// mc.mutable_new_()->CopyFrom(fromJson<pbmessages::MemberChange_Member>(memberJson));
|
||||
pbmessages::MemberChange mc = memberChangeFromJson(_controller_id, oldMember, newMember);
|
||||
std::string payload;
|
||||
if (! mc.SerializeToString(&payload)) {
|
||||
fprintf(stderr, "Failed to serialize MemberChange protobuf message\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
return publishMessage(payload, frontend);
|
||||
return publishMessage(payload);
|
||||
}
|
||||
|
||||
bool PubSubWriter::publishStatusChange(
|
||||
|
|
@ -122,7 +124,158 @@ bool PubSubWriter::publishStatusChange(
|
|||
return false;
|
||||
}
|
||||
|
||||
return publishMessage(payload, frontend);
|
||||
return publishMessage(payload);
|
||||
}
|
||||
|
||||
pbmessages::NetworkChange_Network* networkFromJson(const nlohmann::json& j)
|
||||
{
|
||||
if (! j.is_object()) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
pbmessages::NetworkChange_Network* n = new pbmessages::NetworkChange_Network();
|
||||
n->set_network_id(j.value("id", ""));
|
||||
n->set_name(j.value("name", ""));
|
||||
n->set_capabilities(OSUtils::jsonDump(j.value("capabilities", "[]"), -1));
|
||||
n->set_creation_time(j.value("creationTime", 0));
|
||||
n->set_enable_broadcast(j.value("enableBroadcast", false));
|
||||
|
||||
for (const auto& p : j["ipAssignmentPools"]) {
|
||||
if (p.is_object()) {
|
||||
auto pool = n->add_assignment_pools();
|
||||
pool->set_start_ip(p.value("ipRangeStart", ""));
|
||||
pool->set_end_ip(p.value("ipRangeEnd", ""));
|
||||
}
|
||||
}
|
||||
|
||||
n->set_mtu(j.value("mtu", 2800));
|
||||
n->set_multicast_limit(j.value("multicastLimit", 32));
|
||||
n->set_is_private(j.value("private", true));
|
||||
n->set_remote_trace_level(j.value("remoteTraceLevel", 0));
|
||||
n->set_remote_trace_target(j.value("remoteTraceTarget", ""));
|
||||
n->set_revision(j.value("revision", 0));
|
||||
|
||||
for (const auto& p : j["routes"]) {
|
||||
if (p.is_object()) {
|
||||
auto r = n->add_routes();
|
||||
r->set_target(p.value("target", ""));
|
||||
r->set_via(p.value("via", ""));
|
||||
}
|
||||
}
|
||||
|
||||
n->set_rules("");
|
||||
n->set_tags(OSUtils::jsonDump(j.value("tags", "[]"), -1));
|
||||
|
||||
pbmessages::NetworkChange_IPV4AssignMode* v4am = new pbmessages::NetworkChange_IPV4AssignMode();
|
||||
if (j["v4AssignMode"].is_object()) {
|
||||
v4am->set_zt(j["v4AssignMode"].value("zt", false));
|
||||
}
|
||||
n->set_allocated_ipv4_assign_mode(v4am);
|
||||
|
||||
pbmessages::NetworkChange_IPV6AssignMode* v6am = new pbmessages::NetworkChange_IPV6AssignMode();
|
||||
if (j["v6AssignMode"].is_object()) {
|
||||
v6am->set_zt(j["v6AssignMode"].value("zt", false));
|
||||
v6am->set_six_plane(j["v6AssignMode"].value("6plane", false));
|
||||
v6am->set_rfc4193(j["v6AssignMode"].value("rfc4193", false));
|
||||
}
|
||||
n->set_allocated_ipv6_assign_mode(v6am);
|
||||
|
||||
nlohmann::json jdns = j.value("dns", nullptr);
|
||||
if (jdns.is_object()) {
|
||||
pbmessages::NetworkChange_DNS* dns = new pbmessages::NetworkChange_DNS();
|
||||
dns->set_domain(jdns.value("domain", ""));
|
||||
for (const auto& s : jdns["servers"]) {
|
||||
if (s.is_string()) {
|
||||
auto server = dns->add_nameservers();
|
||||
*server = s;
|
||||
}
|
||||
}
|
||||
n->set_allocated_dns(dns);
|
||||
}
|
||||
|
||||
n->set_sso_enabled(j.value("ssoEnabled", false));
|
||||
if (j.value("ssoEnabled", false)) {
|
||||
n->set_sso_provider(j.value("provider", ""));
|
||||
n->set_sso_client_id(j.value("clientId", ""));
|
||||
n->set_sso_authorization_endpoint(j.value("authorizationEndpoint", ""));
|
||||
n->set_sso_issuer(j.value("issuer", ""));
|
||||
n->set_sso_provider(j.value("provider", ""));
|
||||
}
|
||||
|
||||
n->set_rules_source("");
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
pbmessages::NetworkChange
|
||||
networkChangeFromJson(std::string controllerID, const nlohmann::json& oldNetwork, const nlohmann::json& newNetwork)
|
||||
{
|
||||
pbmessages::NetworkChange nc;
|
||||
nc.set_allocated_old(networkFromJson(oldNetwork));
|
||||
nc.set_allocated_new_(networkFromJson(newNetwork));
|
||||
nc.set_change_source(pbmessages::NetworkChange_ChangeSource::NetworkChange_ChangeSource_CONTROLLER);
|
||||
|
||||
pbmessages::NetworkChange_NetworkChangeMetadata* metadata = new pbmessages::NetworkChange_NetworkChangeMetadata();
|
||||
metadata->set_controller_id(controllerID);
|
||||
metadata->set_trace_id(""); // TODO: generate a trace ID
|
||||
nc.set_allocated_metadata(metadata);
|
||||
|
||||
return nc;
|
||||
}
|
||||
|
||||
pbmessages::MemberChange_Member* memberFromJson(const nlohmann::json& j)
|
||||
{
|
||||
if (! j.is_object()) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
pbmessages::MemberChange_Member* m = new pbmessages::MemberChange_Member();
|
||||
m->set_network_id(j.value("networkId", ""));
|
||||
m->set_device_id(j.value("id", ""));
|
||||
m->set_identity(j.value("identity", ""));
|
||||
m->set_authorized(j.value("authorized", false));
|
||||
for (const auto& addr : j.value("ipAssignments", nlohmann::json::array())) {
|
||||
if (addr.is_string()) {
|
||||
auto a = m->add_ip_assignments();
|
||||
*a = addr;
|
||||
}
|
||||
}
|
||||
m->set_active_bridge(j.value("activeBridge", false));
|
||||
m->set_tags(OSUtils::jsonDump(j.value("tags", "[]"), -1));
|
||||
m->set_capabilities(OSUtils::jsonDump(j.value("capabilities", "[]"), -1));
|
||||
m->set_creation_time(j.value("creationTime", 0));
|
||||
m->set_no_auto_assign_ips(j.value("noAutoAssignIps", false));
|
||||
m->set_revision(j.value("revision", 0));
|
||||
m->set_last_authorized_time(j.value("lastAuthorizedTime", 0));
|
||||
m->set_last_deauthorized_time(j.value("lastDeauthorizedTime", 0));
|
||||
m->set_last_authorized_credential_type(j.value("lastAuthorizedCredentialType", nullptr));
|
||||
m->set_last_authorized_credential(j.value("lastAuthorizedCredential", nullptr));
|
||||
m->set_version_major(j.value("versionMajor", 0));
|
||||
m->set_version_minor(j.value("versionMinor", 0));
|
||||
m->set_version_rev(j.value("versionRev", 0));
|
||||
m->set_version_protocol(j.value("versionProtocol", 0));
|
||||
m->set_remote_trace_level(j.value("remoteTraceLevel", 0));
|
||||
m->set_remote_trace_target(j.value("remoteTraceTarget", ""));
|
||||
m->set_sso_exempt(j.value("ssoExempt", false));
|
||||
m->set_auth_expiry_time(j.value("authExpiryTime", 0));
|
||||
|
||||
return m;
|
||||
}
|
||||
|
||||
pbmessages::MemberChange
|
||||
memberChangeFromJson(std::string controllerID, const nlohmann::json& oldMember, const nlohmann::json& newMember)
|
||||
{
|
||||
pbmessages::MemberChange mc;
|
||||
mc.set_allocated_old(memberFromJson(oldMember));
|
||||
mc.set_allocated_new_(memberFromJson(newMember));
|
||||
mc.set_change_source(pbmessages::MemberChange_ChangeSource::MemberChange_ChangeSource_CONTROLLER);
|
||||
|
||||
pbmessages::MemberChange_MemberChangeMetadata* metadata = new pbmessages::MemberChange_MemberChangeMetadata();
|
||||
metadata->set_controller_id(controllerID);
|
||||
metadata->set_trace_id(""); // TODO: generate a trace ID
|
||||
mc.set_allocated_metadata(metadata);
|
||||
|
||||
return mc;
|
||||
}
|
||||
|
||||
} // namespace ZeroTier
|
||||
Loading…
Add table
Add a link
Reference in a new issue