mirror of
https://github.com/Icinga/icinga2.git
synced 2025-08-21 01:28:21 +02:00
ddafaa0add
Note that even when passing `nullptr` as target zone to `RelayMessage()`, the cluster message will still be sent to the parent zone. These incoming messages will now be rejected by the parent nodes. At the moment, there's no way to only send within the local zone.
1054 lines
32 KiB
C++
1054 lines
32 KiB
C++
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
|
|
|
|
#include "icinga/clusterevents.hpp"
|
|
#include "icinga/service.hpp"
|
|
#include "remote/apilistener.hpp"
|
|
#include "remote/endpoint.hpp"
|
|
#include "remote/messageorigin.hpp"
|
|
#include "remote/zone.hpp"
|
|
#include "remote/apifunction.hpp"
|
|
#include "remote/eventqueue.hpp"
|
|
#include "base/application.hpp"
|
|
#include "base/configtype.hpp"
|
|
#include "base/utility.hpp"
|
|
#include "base/perfdatavalue.hpp"
|
|
#include "base/exception.hpp"
|
|
#include "base/initialize.hpp"
|
|
#include "base/serializer.hpp"
|
|
#include "base/json.hpp"
|
|
#include <fstream>
|
|
|
|
using namespace icinga;
|
|
|
|
INITIALIZE_ONCE(&ClusterEvents::StaticInitialize);
|
|
|
|
REGISTER_APIFUNCTION(CheckResult, event, &ClusterEvents::CheckResultAPIHandler);
|
|
REGISTER_APIFUNCTION(SetNextCheck, event, &ClusterEvents::NextCheckChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetLastCheckStarted, event, &ClusterEvents::LastCheckStartedChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetSuppressedNotifications, event, &ClusterEvents::SuppressedNotificationsChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetSuppressedNotificationTypes, event, &ClusterEvents::SuppressedNotificationTypesChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetNextNotification, event, &ClusterEvents::NextNotificationChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetForceNextCheck, event, &ClusterEvents::ForceNextCheckChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetForceNextNotification, event, &ClusterEvents::ForceNextNotificationChangedAPIHandler);
|
|
REGISTER_APIFUNCTION(SetAcknowledgement, event, &ClusterEvents::AcknowledgementSetAPIHandler);
|
|
REGISTER_APIFUNCTION(ClearAcknowledgement, event, &ClusterEvents::AcknowledgementClearedAPIHandler);
|
|
REGISTER_APIFUNCTION(ExecuteCommand, event, &ClusterEvents::ExecuteCommandAPIHandler);
|
|
REGISTER_APIFUNCTION(SendNotifications, event, &ClusterEvents::SendNotificationsAPIHandler);
|
|
REGISTER_APIFUNCTION(NotificationSentUser, event, &ClusterEvents::NotificationSentUserAPIHandler);
|
|
REGISTER_APIFUNCTION(NotificationSentToAllUsers, event, &ClusterEvents::NotificationSentToAllUsersAPIHandler);
|
|
|
|
void ClusterEvents::StaticInitialize()
|
|
{
|
|
Checkable::OnNewCheckResult.connect(&ClusterEvents::CheckResultHandler);
|
|
Checkable::OnNextCheckChanged.connect(&ClusterEvents::NextCheckChangedHandler);
|
|
Checkable::OnLastCheckStartedChanged.connect(&ClusterEvents::LastCheckStartedChangedHandler);
|
|
Checkable::OnSuppressedNotificationsChanged.connect(&ClusterEvents::SuppressedNotificationsChangedHandler);
|
|
Notification::OnSuppressedNotificationsChanged.connect(&ClusterEvents::SuppressedNotificationTypesChangedHandler);
|
|
Notification::OnNextNotificationChanged.connect(&ClusterEvents::NextNotificationChangedHandler);
|
|
Checkable::OnForceNextCheckChanged.connect(&ClusterEvents::ForceNextCheckChangedHandler);
|
|
Checkable::OnForceNextNotificationChanged.connect(&ClusterEvents::ForceNextNotificationChangedHandler);
|
|
Checkable::OnNotificationsRequested.connect(&ClusterEvents::SendNotificationsHandler);
|
|
Checkable::OnNotificationSentToUser.connect(&ClusterEvents::NotificationSentUserHandler);
|
|
Checkable::OnNotificationSentToAllUsers.connect(&ClusterEvents::NotificationSentToAllUsersHandler);
|
|
|
|
Checkable::OnAcknowledgementSet.connect(&ClusterEvents::AcknowledgementSetHandler);
|
|
Checkable::OnAcknowledgementCleared.connect(&ClusterEvents::AcknowledgementClearedHandler);
|
|
}
|
|
|
|
Dictionary::Ptr ClusterEvents::MakeCheckResultMessage(const Checkable::Ptr& checkable, const CheckResult::Ptr& cr)
|
|
{
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::CheckResult");
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
else {
|
|
Value agent_service_name = checkable->GetExtension("agent_service_name");
|
|
|
|
if (!agent_service_name.IsEmpty())
|
|
params->Set("service", agent_service_name);
|
|
}
|
|
params->Set("cr", Serialize(cr));
|
|
|
|
message->Set("params", params);
|
|
|
|
return message;
|
|
}
|
|
|
|
void ClusterEvents::CheckResultHandler(const Checkable::Ptr& checkable, const CheckResult::Ptr& cr, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Dictionary::Ptr message = MakeCheckResultMessage(checkable, cr);
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::CheckResultAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'check result' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
CheckResult::Ptr cr;
|
|
Array::Ptr vperf;
|
|
|
|
if (params->Contains("cr")) {
|
|
cr = new CheckResult();
|
|
Dictionary::Ptr vcr = params->Get("cr");
|
|
|
|
if (vcr && vcr->Contains("performance_data")) {
|
|
vperf = vcr->Get("performance_data");
|
|
|
|
if (vperf)
|
|
vcr->Remove("performance_data");
|
|
|
|
Deserialize(cr, vcr, true);
|
|
}
|
|
}
|
|
|
|
if (!cr)
|
|
return Empty;
|
|
|
|
ArrayData rperf;
|
|
|
|
if (vperf) {
|
|
ObjectLock olock(vperf);
|
|
for (const Value& vp : vperf) {
|
|
Value p;
|
|
|
|
if (vp.IsObjectType<Dictionary>()) {
|
|
PerfdataValue::Ptr val = new PerfdataValue();
|
|
Deserialize(val, vp, true);
|
|
rperf.push_back(val);
|
|
} else
|
|
rperf.push_back(vp);
|
|
}
|
|
}
|
|
|
|
cr->SetPerformanceData(new Array(std::move(rperf)));
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable) && endpoint != checkable->GetCommandEndpoint()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'check result' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
if (!checkable->IsPaused() && Zone::GetLocalZone() == checkable->GetZone() && endpoint == checkable->GetCommandEndpoint())
|
|
checkable->ProcessCheckResult(cr);
|
|
else
|
|
checkable->ProcessCheckResult(cr, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::NextCheckChangedHandler(const Checkable::Ptr& checkable, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("next_check", checkable->GetNextCheck());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetNextCheck");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::NextCheckChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'next check changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'next check changed' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
double nextCheck = params->Get("next_check");
|
|
|
|
if (nextCheck < Application::GetStartTime() + 60)
|
|
return Empty;
|
|
|
|
checkable->SetNextCheck(params->Get("next_check"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::LastCheckStartedChangedHandler(const Checkable::Ptr& checkable, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("last_check_started", checkable->GetLastCheckStarted());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetLastCheckStarted");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::LastCheckStartedChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'last_check_started changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'last_check_started changed' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->SetLastCheckStarted(params->Get("last_check_started"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::SuppressedNotificationsChangedHandler(const Checkable::Ptr& checkable, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("suppressed_notifications", checkable->GetSuppressedNotifications());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetSuppressedNotifications");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, nullptr, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::SuppressedNotificationsChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'suppressed notifications changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && origin->FromZone != Zone::GetLocalZone()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'suppressed notifications changed' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->SetSuppressedNotifications(params->Get("suppressed_notifications"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::SuppressedNotificationTypesChangedHandler(const Notification::Ptr& notification, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("notification", notification->GetName());
|
|
params->Set("suppressed_notifications", notification->GetSuppressedNotifications());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetSuppressedNotificationTypes");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, nullptr, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::SuppressedNotificationTypesChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'suppressed notifications changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
auto notification (Notification::GetByName(params->Get("notification")));
|
|
|
|
if (!notification)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && origin->FromZone != Zone::GetLocalZone()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'suppressed notification types changed' message for notification '" << notification->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
notification->SetSuppressedNotifications(params->Get("suppressed_notifications"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::NextNotificationChangedHandler(const Notification::Ptr& notification, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("notification", notification->GetName());
|
|
params->Set("next_notification", notification->GetNextNotification());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetNextNotification");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, notification, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::NextNotificationChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'next notification changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Notification::Ptr notification = Notification::GetByName(params->Get("notification"));
|
|
|
|
if (!notification)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(notification)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'next notification changed' message for notification '" << notification->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
double nextNotification = params->Get("next_notification");
|
|
|
|
if (nextNotification < Utility::GetTime())
|
|
return Empty;
|
|
|
|
notification->SetNextNotification(nextNotification, false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::ForceNextCheckChangedHandler(const Checkable::Ptr& checkable, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("forced", checkable->GetForceNextCheck());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetForceNextCheck");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::ForceNextCheckChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'force next check changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'force next check' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->SetForceNextCheck(params->Get("forced"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::ForceNextNotificationChangedHandler(const Checkable::Ptr& checkable, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("forced", checkable->GetForceNextNotification());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetForceNextNotification");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::ForceNextNotificationChangedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'force next notification changed' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'force next notification' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->SetForceNextNotification(params->Get("forced"), false, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::AcknowledgementSetHandler(const Checkable::Ptr& checkable,
|
|
const String& author, const String& comment, AcknowledgementType type,
|
|
bool notify, bool persistent, double changeTime, double expiry, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("author", author);
|
|
params->Set("comment", comment);
|
|
params->Set("acktype", type);
|
|
params->Set("notify", notify);
|
|
params->Set("persistent", persistent);
|
|
params->Set("expiry", expiry);
|
|
params->Set("change_time", changeTime);
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::SetAcknowledgement");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::AcknowledgementSetAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'acknowledgement set' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'acknowledgement set' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
ObjectLock oLock (checkable);
|
|
|
|
if (checkable->IsAcknowledged()) {
|
|
Log(LogWarning, "ClusterEvents")
|
|
<< "Discarding 'acknowledgement set' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Checkable is already acknowledged.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->AcknowledgeProblem(params->Get("author"), params->Get("comment"),
|
|
static_cast<AcknowledgementType>(static_cast<int>(params->Get("acktype"))),
|
|
params->Get("notify"), params->Get("persistent"), params->Get("change_time"), params->Get("expiry"), origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::AcknowledgementClearedHandler(const Checkable::Ptr& checkable, const String& removedBy, double changeTime, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("author", removedBy);
|
|
params->Set("change_time", changeTime);
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::ClearAcknowledgement");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, checkable, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::AcknowledgementClearedAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'acknowledgement cleared' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && !origin->FromZone->CanAccessObject(checkable)) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'acknowledgement cleared' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
checkable->ClearAcknowledgement(params->Get("author"), params->Get("change_time"), origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
Value ClusterEvents::ExecuteCommandAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
EnqueueCheck(origin, params);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::SendNotificationsHandler(const Checkable::Ptr& checkable, NotificationType type,
|
|
const CheckResult::Ptr& cr, const String& author, const String& text, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Dictionary::Ptr message = MakeCheckResultMessage(checkable, cr);
|
|
message->Set("method", "event::SendNotifications");
|
|
|
|
Dictionary::Ptr params = message->Get("params");
|
|
params->Set("type", type);
|
|
params->Set("author", author);
|
|
params->Set("text", text);
|
|
|
|
listener->RelayMessage(origin, nullptr, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::SendNotificationsAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'send notification' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && origin->FromZone != Zone::GetLocalZone()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'send custom notification' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
CheckResult::Ptr cr;
|
|
Array::Ptr vperf;
|
|
|
|
if (params->Contains("cr")) {
|
|
cr = new CheckResult();
|
|
Dictionary::Ptr vcr = params->Get("cr");
|
|
|
|
if (vcr && vcr->Contains("performance_data")) {
|
|
vperf = vcr->Get("performance_data");
|
|
|
|
if (vperf)
|
|
vcr->Remove("performance_data");
|
|
|
|
Deserialize(cr, vcr, true);
|
|
}
|
|
}
|
|
|
|
NotificationType type = static_cast<NotificationType>(static_cast<int>(params->Get("type")));
|
|
String author = params->Get("author");
|
|
String text = params->Get("text");
|
|
|
|
Checkable::OnNotificationsRequested(checkable, type, cr, author, text, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::NotificationSentUserHandler(const Notification::Ptr& notification, const Checkable::Ptr& checkable, const User::Ptr& user,
|
|
NotificationType notificationType, const CheckResult::Ptr& cr, const String& author, const String& commentText, const String& command,
|
|
const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("notification", notification->GetName());
|
|
params->Set("user", user->GetName());
|
|
params->Set("type", notificationType);
|
|
params->Set("cr", Serialize(cr));
|
|
params->Set("author", author);
|
|
params->Set("text", commentText);
|
|
params->Set("command", command);
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::NotificationSentUser");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, nullptr, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::NotificationSentUserAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'sent notification to user' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && origin->FromZone != Zone::GetLocalZone()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'send notification to user' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
CheckResult::Ptr cr;
|
|
Array::Ptr vperf;
|
|
|
|
if (params->Contains("cr")) {
|
|
cr = new CheckResult();
|
|
Dictionary::Ptr vcr = params->Get("cr");
|
|
|
|
if (vcr && vcr->Contains("performance_data")) {
|
|
vperf = vcr->Get("performance_data");
|
|
|
|
if (vperf)
|
|
vcr->Remove("performance_data");
|
|
|
|
Deserialize(cr, vcr, true);
|
|
}
|
|
}
|
|
|
|
NotificationType type = static_cast<NotificationType>(static_cast<int>(params->Get("type")));
|
|
String author = params->Get("author");
|
|
String text = params->Get("text");
|
|
|
|
Notification::Ptr notification = Notification::GetByName(params->Get("notification"));
|
|
|
|
if (!notification)
|
|
return Empty;
|
|
|
|
User::Ptr user = User::GetByName(params->Get("user"));
|
|
|
|
if (!user)
|
|
return Empty;
|
|
|
|
String command = params->Get("command");
|
|
|
|
Checkable::OnNotificationSentToUser(notification, checkable, user, type, cr, author, text, command, origin);
|
|
|
|
return Empty;
|
|
}
|
|
|
|
void ClusterEvents::NotificationSentToAllUsersHandler(const Notification::Ptr& notification, const Checkable::Ptr& checkable, const std::set<User::Ptr>& users,
|
|
NotificationType notificationType, const CheckResult::Ptr& cr, const String& author, const String& commentText, const MessageOrigin::Ptr& origin)
|
|
{
|
|
ApiListener::Ptr listener = ApiListener::GetInstance();
|
|
|
|
if (!listener)
|
|
return;
|
|
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
|
|
Dictionary::Ptr params = new Dictionary();
|
|
params->Set("host", host->GetName());
|
|
if (service)
|
|
params->Set("service", service->GetShortName());
|
|
params->Set("notification", notification->GetName());
|
|
|
|
ArrayData ausers;
|
|
for (const User::Ptr& user : users) {
|
|
ausers.push_back(user->GetName());
|
|
}
|
|
params->Set("users", new Array(std::move(ausers)));
|
|
|
|
params->Set("type", notificationType);
|
|
params->Set("cr", Serialize(cr));
|
|
params->Set("author", author);
|
|
params->Set("text", commentText);
|
|
|
|
params->Set("last_notification", notification->GetLastNotification());
|
|
params->Set("next_notification", notification->GetNextNotification());
|
|
params->Set("notification_number", notification->GetNotificationNumber());
|
|
params->Set("last_problem_notification", notification->GetLastProblemNotification());
|
|
params->Set("no_more_notifications", notification->GetNoMoreNotifications());
|
|
|
|
Dictionary::Ptr message = new Dictionary();
|
|
message->Set("jsonrpc", "2.0");
|
|
message->Set("method", "event::NotificationSentToAllUsers");
|
|
message->Set("params", params);
|
|
|
|
listener->RelayMessage(origin, nullptr, message, true);
|
|
}
|
|
|
|
Value ClusterEvents::NotificationSentToAllUsersAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Endpoint::Ptr endpoint = origin->FromClient->GetEndpoint();
|
|
|
|
if (!endpoint) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'sent notification to all users' message from '" << origin->FromClient->GetIdentity() << "': Invalid endpoint origin (client not allowed).";
|
|
return Empty;
|
|
}
|
|
|
|
Host::Ptr host = Host::GetByName(params->Get("host"));
|
|
|
|
if (!host)
|
|
return Empty;
|
|
|
|
Checkable::Ptr checkable;
|
|
|
|
if (params->Contains("service"))
|
|
checkable = host->GetServiceByShortName(params->Get("service"));
|
|
else
|
|
checkable = host;
|
|
|
|
if (!checkable)
|
|
return Empty;
|
|
|
|
if (origin->FromZone && origin->FromZone != Zone::GetLocalZone()) {
|
|
Log(LogNotice, "ClusterEvents")
|
|
<< "Discarding 'sent notification to all users' message for checkable '" << checkable->GetName()
|
|
<< "' from '" << origin->FromClient->GetIdentity() << "': Unauthorized access.";
|
|
return Empty;
|
|
}
|
|
|
|
CheckResult::Ptr cr;
|
|
Array::Ptr vperf;
|
|
|
|
if (params->Contains("cr")) {
|
|
cr = new CheckResult();
|
|
Dictionary::Ptr vcr = params->Get("cr");
|
|
|
|
if (vcr && vcr->Contains("performance_data")) {
|
|
vperf = vcr->Get("performance_data");
|
|
|
|
if (vperf)
|
|
vcr->Remove("performance_data");
|
|
|
|
Deserialize(cr, vcr, true);
|
|
}
|
|
}
|
|
|
|
NotificationType type = static_cast<NotificationType>(static_cast<int>(params->Get("type")));
|
|
String author = params->Get("author");
|
|
String text = params->Get("text");
|
|
|
|
Notification::Ptr notification = Notification::GetByName(params->Get("notification"));
|
|
|
|
if (!notification)
|
|
return Empty;
|
|
|
|
Array::Ptr ausers = params->Get("users");
|
|
|
|
if (!ausers)
|
|
return Empty;
|
|
|
|
std::set<User::Ptr> users;
|
|
|
|
{
|
|
ObjectLock olock(ausers);
|
|
for (const String& auser : ausers) {
|
|
User::Ptr user = User::GetByName(auser);
|
|
|
|
if (!user)
|
|
continue;
|
|
|
|
users.insert(user);
|
|
}
|
|
}
|
|
|
|
notification->SetLastNotification(params->Get("last_notification"));
|
|
notification->SetNextNotification(params->Get("next_notification"));
|
|
notification->SetNotificationNumber(params->Get("notification_number"));
|
|
notification->SetLastProblemNotification(params->Get("last_problem_notification"));
|
|
notification->SetNoMoreNotifications(params->Get("no_more_notifications"));
|
|
|
|
ArrayData notifiedProblemUsers;
|
|
for (const User::Ptr& user : users) {
|
|
notifiedProblemUsers.push_back(user->GetName());
|
|
}
|
|
|
|
notification->SetNotifiedProblemUsers(new Array(std::move(notifiedProblemUsers)));
|
|
|
|
Checkable::OnNotificationSentToAllUsers(notification, checkable, users, type, cr, author, text, origin);
|
|
|
|
return Empty;
|
|
}
|