mirror of https://github.com/Icinga/icinga2.git
1146 lines
32 KiB
C++
1146 lines
32 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
|
|
* *
|
|
* This program is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU General Public License *
|
|
* as published by the Free Software Foundation; either version 2 *
|
|
* of the License, or (at your option) any later version. *
|
|
* *
|
|
* This program is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program; if not, write to the Free Software Foundation *
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
******************************************************************************/
|
|
|
|
#include "icinga/compatutility.hpp"
|
|
#include "icinga/checkcommand.hpp"
|
|
#include "icinga/eventcommand.hpp"
|
|
#include "icinga/pluginutility.hpp"
|
|
#include "icinga/service.hpp"
|
|
#include "base/utility.hpp"
|
|
#include "base/dynamictype.hpp"
|
|
#include "base/objectlock.hpp"
|
|
#include "base/convert.hpp"
|
|
#include <boost/foreach.hpp>
|
|
#include <boost/algorithm/string/replace.hpp>
|
|
#include <boost/algorithm/string/join.hpp>
|
|
|
|
using namespace icinga;
|
|
|
|
/* command */
|
|
String CompatUtility::GetCommandLine(const Command::Ptr& command)
|
|
{
|
|
Value commandLine = command->GetCommandLine();
|
|
|
|
String result;
|
|
if (commandLine.IsObjectType<Array>()) {
|
|
Array::Ptr args = commandLine;
|
|
|
|
ObjectLock olock(args);
|
|
String arg;
|
|
BOOST_FOREACH(arg, args) {
|
|
// This is obviously incorrect for non-trivial cases.
|
|
result += " \"" + EscapeString(arg) + "\"";
|
|
}
|
|
} else if (!commandLine.IsEmpty()) {
|
|
result = EscapeString(Convert::ToString(commandLine));
|
|
} else {
|
|
result = "<internal>";
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
String CompatUtility::GetCommandNamePrefix(const Command::Ptr command)
|
|
{
|
|
if (!command)
|
|
return Empty;
|
|
|
|
String prefix;
|
|
if (command->GetType() == DynamicType::GetByName("CheckCommand"))
|
|
prefix = "check_";
|
|
else if (command->GetType() == DynamicType::GetByName("NotificationCommand"))
|
|
prefix = "notification_";
|
|
else if (command->GetType() == DynamicType::GetByName("EventCommand"))
|
|
prefix = "event_";
|
|
|
|
return prefix;
|
|
}
|
|
|
|
String CompatUtility::GetCommandName(const Command::Ptr command)
|
|
{
|
|
if (!command)
|
|
return Empty;
|
|
|
|
return GetCommandNamePrefix(command) + command->GetName();
|
|
}
|
|
|
|
/* host */
|
|
String CompatUtility::GetHostAlias(const Host::Ptr& host)
|
|
{
|
|
if (!host->GetDisplayName().IsEmpty())
|
|
return host->GetName();
|
|
else
|
|
return host->GetDisplayName();
|
|
}
|
|
|
|
int CompatUtility::GetHostNotifyOnDown(const Host::Ptr& host)
|
|
{
|
|
unsigned long notification_state_filter = GetCheckableNotificationStateFilter(host);
|
|
|
|
if (notification_state_filter & (1<<ServiceCritical) ||
|
|
notification_state_filter & (1<<ServiceWarning))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetHostNotifyOnUnreachable(const Host::Ptr& host)
|
|
{
|
|
unsigned long notification_state_filter = GetCheckableNotificationStateFilter(host);
|
|
|
|
if (notification_state_filter & (1<<ServiceUnknown))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* service */
|
|
String CompatUtility::GetCheckableCommandArgs(const Checkable::Ptr& checkable)
|
|
{
|
|
CheckCommand::Ptr command = checkable->GetCheckCommand();
|
|
|
|
Dictionary::Ptr args = make_shared<Dictionary>();
|
|
|
|
if (command) {
|
|
Host::Ptr host;
|
|
Service::Ptr service;
|
|
tie(host, service) = GetHostService(checkable);
|
|
String command_line = GetCommandLine(command);
|
|
|
|
Dictionary::Ptr command_vars = command->GetVars();
|
|
|
|
if (command_vars) {
|
|
BOOST_FOREACH(Dictionary::Pair kv, command_vars) {
|
|
String macro = "$" + kv.first + "$"; // this is too simple
|
|
if (command_line.Contains(macro))
|
|
args->Set(kv.first, kv.second);
|
|
|
|
}
|
|
}
|
|
|
|
Dictionary::Ptr host_vars = host->GetVars();
|
|
|
|
if (host_vars) {
|
|
BOOST_FOREACH(Dictionary::Pair kv, host_vars) {
|
|
String macro = "$" + kv.first + "$"; // this is too simple
|
|
if (command_line.Contains(macro))
|
|
args->Set(kv.first, kv.second);
|
|
macro = "$host.vars." + kv.first + "$";
|
|
if (command_line.Contains(macro))
|
|
args->Set(kv.first, kv.second);
|
|
}
|
|
}
|
|
|
|
if (service) {
|
|
Dictionary::Ptr service_vars = service->GetVars();
|
|
|
|
if (service_vars) {
|
|
BOOST_FOREACH(Dictionary::Pair kv, service_vars) {
|
|
String macro = "$" + kv.first + "$"; // this is too simple
|
|
if (command_line.Contains(macro))
|
|
args->Set(kv.first, kv.second);
|
|
macro = "$service.vars." + kv.first + "$";
|
|
if (command_line.Contains(macro))
|
|
args->Set(kv.first, kv.second);
|
|
}
|
|
}
|
|
}
|
|
|
|
String arg_string;
|
|
BOOST_FOREACH(Dictionary::Pair kv, args) {
|
|
arg_string += Convert::ToString(kv.first) + "=" + Convert::ToString(kv.second) + "!";
|
|
}
|
|
return arg_string;
|
|
}
|
|
|
|
return Empty;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableCheckType(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnableActiveChecks() ? 0 : 1);
|
|
}
|
|
|
|
double CompatUtility::GetCheckableCheckInterval(const Checkable::Ptr& checkable)
|
|
{
|
|
return checkable->GetCheckInterval() / 60.0;
|
|
}
|
|
|
|
double CompatUtility::GetCheckableRetryInterval(const Checkable::Ptr& checkable)
|
|
{
|
|
return checkable->GetRetryInterval() / 60.0;
|
|
}
|
|
|
|
String CompatUtility::GetCheckableCheckPeriod(const Checkable::Ptr& checkable)
|
|
{
|
|
TimePeriod::Ptr check_period = checkable->GetCheckPeriod();
|
|
if (check_period)
|
|
return check_period->GetName();
|
|
else
|
|
return "24x7";
|
|
}
|
|
|
|
int CompatUtility::GetCheckableHasBeenChecked(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetLastCheckResult() ? 1 : 0);
|
|
}
|
|
|
|
|
|
int CompatUtility::GetCheckableProblemHasBeenAcknowledged(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetAcknowledgement() != AcknowledgementNone ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableAcknowledgementType(const Checkable::Ptr& checkable)
|
|
{
|
|
return static_cast<int>(checkable->GetAcknowledgement());
|
|
}
|
|
|
|
int CompatUtility::GetCheckablePassiveChecksEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnablePassiveChecks() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableActiveChecksEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnableActiveChecks() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableEventHandlerEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEventCommand() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableFlapDetectionEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnableFlapping() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableIsFlapping(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->IsFlapping() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableIsReachable(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->IsReachable() ? 1 : 0);
|
|
}
|
|
|
|
String CompatUtility::GetCheckablePercentStateChange(const Checkable::Ptr& checkable)
|
|
{
|
|
return Convert::ToString(checkable->GetFlappingCurrent());
|
|
}
|
|
|
|
int CompatUtility::GetCheckableProcessPerformanceData(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnablePerfdata() ? 1 : 0);
|
|
}
|
|
|
|
String CompatUtility::GetCheckableEventHandler(const Checkable::Ptr& checkable)
|
|
{
|
|
String event_command_str;
|
|
EventCommand::Ptr eventcommand = checkable->GetEventCommand();
|
|
|
|
if (eventcommand)
|
|
event_command_str = eventcommand->GetName();
|
|
|
|
return event_command_str;
|
|
}
|
|
|
|
String CompatUtility::GetCheckableCheckCommand(const Checkable::Ptr& checkable)
|
|
{
|
|
String check_command_str;
|
|
CheckCommand::Ptr checkcommand = checkable->GetCheckCommand();
|
|
|
|
if (checkcommand)
|
|
check_command_str = checkcommand->GetName();
|
|
|
|
return check_command_str;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableIsVolatile(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetVolatile() ? 1 : 0);
|
|
}
|
|
|
|
double CompatUtility::GetCheckableLowFlapThreshold(const Checkable::Ptr& checkable)
|
|
{
|
|
return checkable->GetFlappingThreshold();
|
|
}
|
|
|
|
double CompatUtility::GetCheckableHighFlapThreshold(const Checkable::Ptr& checkable)
|
|
{
|
|
return checkable->GetFlappingThreshold();
|
|
}
|
|
|
|
int CompatUtility::GetCheckableFreshnessChecksEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetCheckInterval() > 0 ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableFreshnessThreshold(const Checkable::Ptr& checkable)
|
|
{
|
|
return static_cast<int>(checkable->GetCheckInterval());
|
|
}
|
|
|
|
double CompatUtility::GetCheckableStaleness(const Checkable::Ptr& checkable)
|
|
{
|
|
if (checkable->HasBeenChecked() && checkable->GetLastCheck() > 0)
|
|
return (Utility::GetTime() - checkable->GetLastCheck()) / (checkable->GetCheckInterval() * 3600);
|
|
|
|
return 0.0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableIsAcknowledged(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->IsAcknowledged() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNoMoreNotifications(const Checkable::Ptr& checkable)
|
|
{
|
|
if (CompatUtility::GetCheckableNotificationNotificationInterval(checkable) == 0 && !checkable->GetVolatile())
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableInCheckPeriod(const Checkable::Ptr& checkable)
|
|
{
|
|
TimePeriod::Ptr timeperiod = checkable->GetCheckPeriod();
|
|
|
|
/* none set means always checked */
|
|
if (!timeperiod)
|
|
return 1;
|
|
|
|
return (timeperiod->IsInside(Utility::GetTime()) ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableInNotificationPeriod(const Checkable::Ptr& checkable)
|
|
{
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
ObjectLock olock(notification);
|
|
|
|
TimePeriod::Ptr timeperiod = notification->GetPeriod();
|
|
|
|
/* first notification wins */
|
|
if (timeperiod)
|
|
return (timeperiod->IsInside(Utility::GetTime()) ? 1 : 0);
|
|
}
|
|
|
|
/* none set means always notified */
|
|
return 1;
|
|
}
|
|
|
|
/* vars attr */
|
|
|
|
bool CompatUtility::IsLegacyAttribute(CustomVarObject::Ptr const& object, const String& name)
|
|
{
|
|
if ((name == "address" ||
|
|
name == "address6") &&
|
|
object->GetType() == DynamicType::GetByName("Host"))
|
|
return true;
|
|
|
|
if ((name == "address1" ||
|
|
name == "address2" ||
|
|
name == "address3" ||
|
|
name == "address4" ||
|
|
name == "address5" ||
|
|
name == "address6" ||
|
|
name == "email" ||
|
|
name == "pager") &&
|
|
object->GetType() == DynamicType::GetByName("User"))
|
|
return true;
|
|
|
|
if ((name == "notes" ||
|
|
name == "action_url" ||
|
|
name == "notes_url" ||
|
|
name == "icon_image" ||
|
|
name == "icon_image_alt") &&
|
|
(object->GetType() == DynamicType::GetByName("Host") ||
|
|
object->GetType() == DynamicType::GetByName("Service")))
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
Dictionary::Ptr CompatUtility::GetCustomAttributeConfig(const CustomVarObject::Ptr& object)
|
|
{
|
|
Dictionary::Ptr vars = object->GetVars();
|
|
|
|
Dictionary::Ptr varsvars = make_shared<Dictionary>();
|
|
|
|
if (!vars)
|
|
return Dictionary::Ptr();
|
|
|
|
ObjectLock olock(vars);
|
|
BOOST_FOREACH(const Dictionary::Pair& kv, vars) {
|
|
if (!kv.first.IsEmpty()) {
|
|
if (!IsLegacyAttribute(object, kv.first))
|
|
varsvars->Set(kv.first, kv.second);
|
|
}
|
|
}
|
|
|
|
return varsvars;
|
|
}
|
|
|
|
String CompatUtility::GetCustomAttributeConfig(const CustomVarObject::Ptr& object, const String& name)
|
|
{
|
|
Dictionary::Ptr vars = object->GetVars();
|
|
|
|
if (!vars)
|
|
return Empty;
|
|
|
|
return vars->Get(name);
|
|
}
|
|
|
|
Array::Ptr CompatUtility::GetModifiedAttributesList(const CustomVarObject::Ptr& object)
|
|
{
|
|
Array::Ptr mod_attr_list = make_shared<Array>();
|
|
|
|
if (object->GetType() != DynamicType::GetByName("Host") &&
|
|
object->GetType() != DynamicType::GetByName("Service") &&
|
|
object->GetType() != DynamicType::GetByName("User") &&
|
|
object->GetType() != DynamicType::GetByName("CheckCommand") &&
|
|
object->GetType() != DynamicType::GetByName("EventCommand") &&
|
|
object->GetType() != DynamicType::GetByName("NotificationCommand"))
|
|
return mod_attr_list;
|
|
|
|
int flags = object->GetModifiedAttributes();
|
|
|
|
if ((flags & ModAttrNotificationsEnabled))
|
|
mod_attr_list->Add("notifications_enabled");
|
|
|
|
if ((flags & ModAttrActiveChecksEnabled))
|
|
mod_attr_list->Add("active_checks_enabled");
|
|
|
|
if ((flags & ModAttrPassiveChecksEnabled))
|
|
mod_attr_list->Add("passive_checks_enabled");
|
|
|
|
if ((flags & ModAttrFlapDetectionEnabled))
|
|
mod_attr_list->Add("flap_detection_enabled");
|
|
|
|
if ((flags & ModAttrEventHandlerEnabled))
|
|
mod_attr_list->Add("event_handler_enabled");
|
|
|
|
if ((flags & ModAttrPerformanceDataEnabled))
|
|
mod_attr_list->Add("performance_data_enabled");
|
|
|
|
if ((flags & ModAttrNormalCheckInterval))
|
|
mod_attr_list->Add("check_interval");
|
|
|
|
if ((flags & ModAttrRetryCheckInterval))
|
|
mod_attr_list->Add("retry_interval");
|
|
|
|
if ((flags & ModAttrEventHandlerCommand))
|
|
mod_attr_list->Add("event_handler_command");
|
|
|
|
if ((flags & ModAttrCheckCommand))
|
|
mod_attr_list->Add("check_command");
|
|
|
|
if ((flags & ModAttrMaxCheckAttempts))
|
|
mod_attr_list->Add("max_check_attemps");
|
|
|
|
if ((flags & ModAttrCheckTimeperiod))
|
|
mod_attr_list->Add("check_timeperiod");
|
|
|
|
if ((flags & ModAttrCustomVariable))
|
|
mod_attr_list->Add("custom_variable");
|
|
|
|
return mod_attr_list;
|
|
}
|
|
|
|
/* notifications */
|
|
int CompatUtility::GetCheckableNotificationsEnabled(const Checkable::Ptr& checkable)
|
|
{
|
|
return (checkable->GetEnableNotifications() ? 1 : 0);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotificationLastNotification(const Checkable::Ptr& checkable)
|
|
{
|
|
double last_notification = 0.0;
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
if (notification->GetLastNotification() > last_notification)
|
|
last_notification = notification->GetLastNotification();
|
|
}
|
|
|
|
return static_cast<int>(last_notification);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotificationNextNotification(const Checkable::Ptr& checkable)
|
|
{
|
|
double next_notification = 0.0;
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
if (next_notification == 0 || notification->GetNextNotification() < next_notification)
|
|
next_notification = notification->GetNextNotification();
|
|
}
|
|
|
|
return static_cast<int>(next_notification);
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotificationNotificationNumber(const Checkable::Ptr& checkable)
|
|
{
|
|
int notification_number = 0;
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
if (notification->GetNotificationNumber() > notification_number)
|
|
notification_number = notification->GetNotificationNumber();
|
|
}
|
|
|
|
return notification_number;
|
|
}
|
|
|
|
double CompatUtility::GetCheckableNotificationNotificationInterval(const Checkable::Ptr& checkable)
|
|
{
|
|
double notification_interval = -1;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
if (notification_interval == -1 || notification->GetInterval() < notification_interval)
|
|
notification_interval = notification->GetInterval();
|
|
}
|
|
|
|
if (notification_interval == -1)
|
|
notification_interval = 60;
|
|
|
|
return notification_interval / 60.0;
|
|
}
|
|
|
|
String CompatUtility::GetCheckableNotificationNotificationPeriod(const Checkable::Ptr& checkable)
|
|
{
|
|
TimePeriod::Ptr notification_period;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
|
|
if (notification->GetPeriod())
|
|
notification_period = notification->GetPeriod();
|
|
}
|
|
|
|
if (!notification_period)
|
|
return Empty;
|
|
|
|
return notification_period->GetName();
|
|
}
|
|
|
|
String CompatUtility::GetCheckableNotificationNotificationOptions(const Checkable::Ptr& checkable)
|
|
{
|
|
unsigned long notification_type_filter = 0;
|
|
unsigned long notification_state_filter = 0;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
notification_type_filter = notification->GetTypeFilter();
|
|
notification_state_filter = notification->GetStateFilter();
|
|
}
|
|
|
|
std::vector<String> notification_options;
|
|
|
|
/* notification state filters */
|
|
if (notification_state_filter & (1<<ServiceWarning)) {
|
|
notification_options.push_back("w");
|
|
}
|
|
if (notification_state_filter & (1<<ServiceUnknown)) {
|
|
notification_options.push_back("u");
|
|
}
|
|
if (notification_state_filter & (1<<ServiceCritical)) {
|
|
notification_options.push_back("c");
|
|
}
|
|
|
|
/* notification type filters */
|
|
if (notification_type_filter & (1<<NotificationRecovery)) {
|
|
notification_options.push_back("r");
|
|
}
|
|
if (notification_type_filter & (1<<NotificationFlappingStart) ||
|
|
notification_type_filter & (1<<NotificationFlappingEnd)) {
|
|
notification_options.push_back("f");
|
|
}
|
|
if (notification_type_filter & (1<<NotificationDowntimeStart) ||
|
|
notification_type_filter & (1<<NotificationDowntimeEnd) ||
|
|
notification_type_filter & (1<<NotificationDowntimeRemoved)) {
|
|
notification_options.push_back("s");
|
|
}
|
|
|
|
return boost::algorithm::join(notification_options, ",");
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotificationTypeFilter(const Checkable::Ptr& checkable)
|
|
{
|
|
unsigned long notification_type_filter = 0;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
ObjectLock olock(notification);
|
|
|
|
notification_type_filter = notification->GetTypeFilter();
|
|
}
|
|
|
|
return notification_type_filter;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotificationStateFilter(const Checkable::Ptr& checkable)
|
|
{
|
|
unsigned long notification_state_filter = 0;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
ObjectLock olock(notification);
|
|
|
|
notification_state_filter = notification->GetStateFilter();
|
|
}
|
|
|
|
return notification_state_filter;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnWarning(const Checkable::Ptr& checkable)
|
|
{
|
|
if (GetCheckableNotificationStateFilter(checkable) & (1<<ServiceWarning))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnCritical(const Checkable::Ptr& checkable)
|
|
{
|
|
if (GetCheckableNotificationStateFilter(checkable) & (1<<ServiceCritical))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnUnknown(const Checkable::Ptr& checkable)
|
|
{
|
|
if (GetCheckableNotificationStateFilter(checkable) & (1<<ServiceUnknown))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnRecovery(const Checkable::Ptr& checkable)
|
|
{
|
|
if (GetCheckableNotificationTypeFilter(checkable) & (1<<NotificationRecovery))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnFlapping(const Checkable::Ptr& checkable)
|
|
{
|
|
unsigned long notification_type_filter = GetCheckableNotificationTypeFilter(checkable);
|
|
|
|
if (notification_type_filter & (1<<NotificationFlappingStart) ||
|
|
notification_type_filter & (1<<NotificationFlappingEnd))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int CompatUtility::GetCheckableNotifyOnDowntime(const Checkable::Ptr& checkable)
|
|
{
|
|
unsigned long notification_type_filter = GetCheckableNotificationTypeFilter(checkable);
|
|
|
|
if (notification_type_filter & (1<<NotificationDowntimeStart) ||
|
|
notification_type_filter & (1<<NotificationDowntimeEnd) ||
|
|
notification_type_filter & (1<<NotificationDowntimeRemoved))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
std::set<User::Ptr> CompatUtility::GetCheckableNotificationUsers(const Checkable::Ptr& checkable)
|
|
{
|
|
/* Service -> Notifications -> (Users + UserGroups -> Users) */
|
|
std::set<User::Ptr> allUsers;
|
|
std::set<User::Ptr> users;
|
|
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
ObjectLock olock(notification);
|
|
|
|
users = notification->GetUsers();
|
|
|
|
std::copy(users.begin(), users.end(), std::inserter(allUsers, allUsers.begin()));
|
|
|
|
BOOST_FOREACH(const UserGroup::Ptr& ug, notification->GetUserGroups()) {
|
|
std::set<User::Ptr> members = ug->GetMembers();
|
|
std::copy(members.begin(), members.end(), std::inserter(allUsers, allUsers.begin()));
|
|
}
|
|
}
|
|
|
|
return allUsers;
|
|
}
|
|
|
|
std::set<UserGroup::Ptr> CompatUtility::GetCheckableNotificationUserGroups(const Checkable::Ptr& checkable)
|
|
{
|
|
std::set<UserGroup::Ptr> usergroups;
|
|
/* Service -> Notifications -> UserGroups */
|
|
BOOST_FOREACH(const Notification::Ptr& notification, checkable->GetNotifications()) {
|
|
ObjectLock olock(notification);
|
|
|
|
BOOST_FOREACH(const UserGroup::Ptr& ug, notification->GetUserGroups()) {
|
|
usergroups.insert(ug);
|
|
}
|
|
}
|
|
|
|
return usergroups;
|
|
}
|
|
|
|
String CompatUtility::GetCheckResultOutput(const CheckResult::Ptr& cr)
|
|
{
|
|
if (!cr)
|
|
return Empty;
|
|
|
|
String output;
|
|
|
|
String raw_output = cr->GetOutput();
|
|
|
|
/*
|
|
* replace semi-colons with colons in output
|
|
* semi-colon is used as delimiter in various interfaces
|
|
*/
|
|
boost::algorithm::replace_all(raw_output, ";", ":");
|
|
|
|
size_t line_end = raw_output.Find("\n");
|
|
|
|
return raw_output.SubStr(0, line_end);
|
|
}
|
|
|
|
String CompatUtility::GetCheckResultLongOutput(const CheckResult::Ptr& cr)
|
|
{
|
|
if (!cr)
|
|
return Empty;
|
|
|
|
String long_output;
|
|
String output;
|
|
|
|
String raw_output = cr->GetOutput();
|
|
|
|
/*
|
|
* replace semi-colons with colons in output
|
|
* semi-colon is used as delimiter in various interfaces
|
|
*/
|
|
boost::algorithm::replace_all(raw_output, ";", ":");
|
|
|
|
size_t line_end = raw_output.Find("\n");
|
|
|
|
if (line_end > 0 && line_end != String::NPos) {
|
|
long_output = raw_output.SubStr(line_end+1, raw_output.GetLength());
|
|
return EscapeString(long_output);
|
|
}
|
|
|
|
return Empty;
|
|
}
|
|
|
|
String CompatUtility::GetCheckResultPerfdata(const CheckResult::Ptr& cr)
|
|
{
|
|
if (!cr)
|
|
return String();
|
|
|
|
return PluginUtility::FormatPerfdata(cr->GetPerformanceData());
|
|
}
|
|
|
|
String CompatUtility::EscapeString(const String& str)
|
|
{
|
|
String result = str;
|
|
boost::algorithm::replace_all(result, "\n", "\\n");
|
|
return result;
|
|
}
|
|
|
|
std::pair<unsigned long, unsigned long> CompatUtility::ConvertTimestamp(double time)
|
|
{
|
|
unsigned long time_sec = static_cast<long>(time);
|
|
unsigned long time_usec = (time - time_sec) * 1000 * 1000;
|
|
|
|
return std::make_pair(time_sec, time_usec);
|
|
}
|
|
|
|
int CompatUtility::MapNotificationReasonType(NotificationType type)
|
|
{
|
|
switch (type) {
|
|
case NotificationDowntimeStart:
|
|
return 5;
|
|
case NotificationDowntimeEnd:
|
|
return 6;
|
|
case NotificationDowntimeRemoved:
|
|
return 7;
|
|
case NotificationCustom:
|
|
return 8;
|
|
case NotificationAcknowledgement:
|
|
return 1;
|
|
case NotificationProblem:
|
|
return 0;
|
|
case NotificationRecovery:
|
|
return 0;
|
|
case NotificationFlappingStart:
|
|
return 2;
|
|
case NotificationFlappingEnd:
|
|
return 3;
|
|
default:
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
int CompatUtility::MapExternalCommandType(const String& name)
|
|
{
|
|
if (name == "NONE")
|
|
return 0;
|
|
if (name == "ADD_HOST_COMMENT")
|
|
return 1;
|
|
if (name == "DEL_HOST_COMMENT")
|
|
return 2;
|
|
if (name == "ADD_SVC_COMMENT")
|
|
return 3;
|
|
if (name == "DEL_SVC_COMMENT")
|
|
return 4;
|
|
if (name == "ENABLE_SVC_CHECK")
|
|
return 5;
|
|
if (name == "DISABLE_SVC_CHECK")
|
|
return 6;
|
|
if (name == "SCHEDULE_SVC_CHECK")
|
|
return 7;
|
|
if (name == "DELAY_SVC_NOTIFICATION")
|
|
return 9;
|
|
if (name == "DELAY_HOST_NOTIFICATION")
|
|
return 10;
|
|
if (name == "DISABLE_NOTIFICATIONS")
|
|
return 11;
|
|
if (name == "ENABLE_NOTIFICATIONS")
|
|
return 12;
|
|
if (name == "RESTART_PROCESS")
|
|
return 13;
|
|
if (name == "SHUTDOWN_PROCESS")
|
|
return 14;
|
|
if (name == "ENABLE_HOST_SVC_CHECKS")
|
|
return 15;
|
|
if (name == "DISABLE_HOST_SVC_CHECKS")
|
|
return 16;
|
|
if (name == "SCHEDULE_HOST_SVC_CHECKS")
|
|
return 17;
|
|
if (name == "DELAY_HOST_SVC_NOTIFICATIONS")
|
|
return 19;
|
|
if (name == "DEL_ALL_HOST_COMMENTS")
|
|
return 20;
|
|
if (name == "DEL_ALL_SVC_COMMENTS")
|
|
return 21;
|
|
if (name == "ENABLE_SVC_NOTIFICATIONS")
|
|
return 22;
|
|
if (name == "DISABLE_SVC_NOTIFICATIONS")
|
|
return 23;
|
|
if (name == "ENABLE_HOST_NOTIFICATIONS")
|
|
return 24;
|
|
if (name == "DISABLE_HOST_NOTIFICATIONS")
|
|
return 25;
|
|
if (name == "ENABLE_ALL_NOTIFICATIONS_BEYOND_HOST")
|
|
return 26;
|
|
if (name == "DISABLE_ALL_NOTIFICATIONS_BEYOND_HOST")
|
|
return 27;
|
|
if (name == "ENABLE_HOST_SVC_NOTIFICATIONS")
|
|
return 28;
|
|
if (name == "DISABLE_HOST_SVC_NOTIFICATIONS")
|
|
return 29;
|
|
if (name == "PROCESS_SERVICE_CHECK_RESULT")
|
|
return 30;
|
|
if (name == "SAVE_STATE_INFORMATION")
|
|
return 31;
|
|
if (name == "READ_STATE_INFORMATION")
|
|
return 32;
|
|
if (name == "ACKNOWLEDGE_HOST_PROBLEM")
|
|
return 33;
|
|
if (name == "ACKNOWLEDGE_SVC_PROBLEM")
|
|
return 34;
|
|
if (name == "START_EXECUTING_SVC_CHECKS")
|
|
return 35;
|
|
if (name == "STOP_EXECUTING_SVC_CHECKS")
|
|
return 36;
|
|
if (name == "START_ACCEPTING_PASSIVE_SVC_CHECKS")
|
|
return 37;
|
|
if (name == "STOP_ACCEPTING_PASSIVE_SVC_CHECKS")
|
|
return 38;
|
|
if (name == "ENABLE_PASSIVE_SVC_CHECKS")
|
|
return 39;
|
|
if (name == "DISABLE_PASSIVE_SVC_CHECKS")
|
|
return 40;
|
|
if (name == "ENABLE_EVENT_HANDLERS")
|
|
return 41;
|
|
if (name == "DISABLE_EVENT_HANDLERS")
|
|
return 42;
|
|
if (name == "ENABLE_HOST_EVENT_HANDLER")
|
|
return 43;
|
|
if (name == "DISABLE_HOST_EVENT_HANDLER")
|
|
return 44;
|
|
if (name == "ENABLE_SVC_EVENT_HANDLER")
|
|
return 45;
|
|
if (name == "DISABLE_SVC_EVENT_HANDLER")
|
|
return 46;
|
|
if (name == "ENABLE_HOST_CHECK")
|
|
return 47;
|
|
if (name == "DISABLE_HOST_CHECK")
|
|
return 48;
|
|
if (name == "START_OBSESSING_OVER_SVC_CHECKS")
|
|
return 49;
|
|
if (name == "STOP_OBSESSING_OVER_SVC_CHECKS")
|
|
return 50;
|
|
if (name == "REMOVE_HOST_ACKNOWLEDGEMENT")
|
|
return 51;
|
|
if (name == "REMOVE_SVC_ACKNOWLEDGEMENT")
|
|
return 52;
|
|
if (name == "SCHEDULE_FORCED_HOST_SVC_CHECKS")
|
|
return 53;
|
|
if (name == "SCHEDULE_FORCED_SVC_CHECK")
|
|
return 54;
|
|
if (name == "SCHEDULE_HOST_DOWNTIME")
|
|
return 55;
|
|
if (name == "SCHEDULE_SVC_DOWNTIME")
|
|
return 56;
|
|
if (name == "ENABLE_HOST_FLAP_DETECTION")
|
|
return 57;
|
|
if (name == "DISABLE_HOST_FLAP_DETECTION")
|
|
return 58;
|
|
if (name == "ENABLE_SVC_FLAP_DETECTION")
|
|
return 59;
|
|
if (name == "DISABLE_SVC_FLAP_DETECTION")
|
|
return 60;
|
|
if (name == "ENABLE_FLAP_DETECTION")
|
|
return 61;
|
|
if (name == "DISABLE_FLAP_DETECTION")
|
|
return 62;
|
|
if (name == "ENABLE_HOSTGROUP_SVC_NOTIFICATIONS")
|
|
return 63;
|
|
if (name == "DISABLE_HOSTGROUP_SVC_NOTIFICATIONS")
|
|
return 64;
|
|
if (name == "ENABLE_HOSTGROUP_HOST_NOTIFICATIONS")
|
|
return 65;
|
|
if (name == "DISABLE_HOSTGROUP_HOST_NOTIFICATIONS")
|
|
return 66;
|
|
if (name == "ENABLE_HOSTGROUP_SVC_CHECKS")
|
|
return 67;
|
|
if (name == "DISABLE_HOSTGROUP_SVC_CHECKS")
|
|
return 68;
|
|
if (name == "CANCEL_HOST_DOWNTIME")
|
|
return 69;
|
|
if (name == "CANCEL_SVC_DOWNTIME")
|
|
return 70;
|
|
if (name == "CANCEL_ACTIVE_HOST_DOWNTIME")
|
|
return 71;
|
|
if (name == "CANCEL_PENDING_HOST_DOWNTIME")
|
|
return 72;
|
|
if (name == "CANCEL_ACTIVE_SVC_DOWNTIME")
|
|
return 73;
|
|
if (name == "CANCEL_PENDING_SVC_DOWNTIME")
|
|
return 74;
|
|
if (name == "CANCEL_ACTIVE_HOST_SVC_DOWNTIME")
|
|
return 75;
|
|
if (name == "CANCEL_PENDING_HOST_SVC_DOWNTIME")
|
|
return 76;
|
|
if (name == "FLUSH_PENDING_COMMANDS")
|
|
return 77;
|
|
if (name == "DEL_HOST_DOWNTIME")
|
|
return 78;
|
|
if (name == "DEL_SVC_DOWNTIME")
|
|
return 79;
|
|
if (name == "ENABLE_FAILURE_PREDICTION")
|
|
return 80;
|
|
if (name == "DISABLE_FAILURE_PREDICTION")
|
|
return 81;
|
|
if (name == "ENABLE_PERFORMANCE_DATA")
|
|
return 82;
|
|
if (name == "DISABLE_PERFORMANCE_DATA")
|
|
return 83;
|
|
if (name == "SCHEDULE_HOSTGROUP_HOST_DOWNTIME")
|
|
return 84;
|
|
if (name == "SCHEDULE_HOSTGROUP_SVC_DOWNTIME")
|
|
return 85;
|
|
if (name == "SCHEDULE_HOST_SVC_DOWNTIME")
|
|
return 86;
|
|
if (name == "PROCESS_HOST_CHECK_RESULT")
|
|
return 87;
|
|
if (name == "START_EXECUTING_HOST_CHECKS")
|
|
return 88;
|
|
if (name == "STOP_EXECUTING_HOST_CHECKS")
|
|
return 89;
|
|
if (name == "START_ACCEPTING_PASSIVE_HOST_CHECKS")
|
|
return 90;
|
|
if (name == "STOP_ACCEPTING_PASSIVE_HOST_CHECKS")
|
|
return 91;
|
|
if (name == "ENABLE_PASSIVE_HOST_CHECKS")
|
|
return 92;
|
|
if (name == "DISABLE_PASSIVE_HOST_CHECKS")
|
|
return 93;
|
|
if (name == "START_OBSESSING_OVER_HOST_CHECKS")
|
|
return 94;
|
|
if (name == "STOP_OBSESSING_OVER_HOST_CHECKS")
|
|
return 95;
|
|
if (name == "SCHEDULE_HOST_CHECK")
|
|
return 96;
|
|
if (name == "SCHEDULE_FORCED_HOST_CHECK")
|
|
return 98;
|
|
if (name == "START_OBSESSING_OVER_SVC")
|
|
return 99;
|
|
if (name == "STOP_OBSESSING_OVER_SVC")
|
|
return 100;
|
|
if (name == "START_OBSESSING_OVER_HOST")
|
|
return 101;
|
|
if (name == "STOP_OBSESSING_OVER_HOST")
|
|
return 102;
|
|
if (name == "ENABLE_HOSTGROUP_HOST_CHECKS")
|
|
return 103;
|
|
if (name == "DISABLE_HOSTGROUP_HOST_CHECKS")
|
|
return 104;
|
|
if (name == "ENABLE_HOSTGROUP_PASSIVE_SVC_CHECKS")
|
|
return 105;
|
|
if (name == "DISABLE_HOSTGROUP_PASSIVE_SVC_CHECKS")
|
|
return 106;
|
|
if (name == "ENABLE_HOSTGROUP_PASSIVE_HOST_CHECKS")
|
|
return 107;
|
|
if (name == "DISABLE_HOSTGROUP_PASSIVE_HOST_CHECKS")
|
|
return 108;
|
|
if (name == "ENABLE_SERVICEGROUP_SVC_NOTIFICATIONS")
|
|
return 109;
|
|
if (name == "DISABLE_SERVICEGROUP_SVC_NOTIFICATIONS")
|
|
return 110;
|
|
if (name == "ENABLE_SERVICEGROUP_HOST_NOTIFICATIONS")
|
|
return 111;
|
|
if (name == "DISABLE_SERVICEGROUP_HOST_NOTIFICATIONS")
|
|
return 112;
|
|
if (name == "ENABLE_SERVICEGROUP_SVC_CHECKS")
|
|
return 113;
|
|
if (name == "DISABLE_SERVICEGROUP_SVC_CHECKS")
|
|
return 114;
|
|
if (name == "ENABLE_SERVICEGROUP_HOST_CHECKS")
|
|
return 115;
|
|
if (name == "DISABLE_SERVICEGROUP_HOST_CHECKS")
|
|
return 116;
|
|
if (name == "ENABLE_SERVICEGROUP_PASSIVE_SVC_CHECKS")
|
|
return 117;
|
|
if (name == "DISABLE_SERVICEGROUP_PASSIVE_SVC_CHECKS")
|
|
return 118;
|
|
if (name == "ENABLE_SERVICEGROUP_PASSIVE_HOST_CHECKS")
|
|
return 119;
|
|
if (name == "DISABLE_SERVICEGROUP_PASSIVE_HOST_CHECKS")
|
|
return 120;
|
|
if (name == "SCHEDULE_SERVICEGROUP_HOST_DOWNTIME")
|
|
return 121;
|
|
if (name == "SCHEDULE_SERVICEGROUP_SVC_DOWNTIME")
|
|
return 122;
|
|
if (name == "CHANGE_GLOBAL_HOST_EVENT_HANDLER")
|
|
return 123;
|
|
if (name == "CHANGE_GLOBAL_SVC_EVENT_HANDLER")
|
|
return 124;
|
|
if (name == "CHANGE_HOST_EVENT_HANDLER")
|
|
return 125;
|
|
if (name == "CHANGE_SVC_EVENT_HANDLER")
|
|
return 126;
|
|
if (name == "CHANGE_HOST_CHECK_COMMAND")
|
|
return 127;
|
|
if (name == "CHANGE_SVC_CHECK_COMMAND")
|
|
return 128;
|
|
if (name == "CHANGE_NORMAL_HOST_CHECK_INTERVAL")
|
|
return 129;
|
|
if (name == "CHANGE_NORMAL_SVC_CHECK_INTERVAL")
|
|
return 130;
|
|
if (name == "CHANGE_RETRY_SVC_CHECK_INTERVAL")
|
|
return 131;
|
|
if (name == "CHANGE_MAX_HOST_CHECK_ATTEMPTS")
|
|
return 132;
|
|
if (name == "CHANGE_MAX_SVC_CHECK_ATTEMPTS")
|
|
return 133;
|
|
if (name == "SCHEDULE_AND_PROPAGATE_TRIGGERED_HOST_DOWNTIME")
|
|
return 134;
|
|
if (name == "ENABLE_HOST_AND_CHILD_NOTIFICATIONS")
|
|
return 135;
|
|
if (name == "DISABLE_HOST_AND_CHILD_NOTIFICATIONS")
|
|
return 136;
|
|
if (name == "SCHEDULE_AND_PROPAGATE_HOST_DOWNTIME")
|
|
return 137;
|
|
if (name == "ENABLE_SERVICE_FRESHNESS_CHECKS")
|
|
return 138;
|
|
if (name == "DISABLE_SERVICE_FRESHNESS_CHECKS")
|
|
return 139;
|
|
if (name == "ENABLE_HOST_FRESHNESS_CHECKS")
|
|
return 140;
|
|
if (name == "DISABLE_HOST_FRESHNESS_CHECKS")
|
|
return 141;
|
|
if (name == "SET_HOST_NOTIFICATION_NUMBER")
|
|
return 142;
|
|
if (name == "SET_SVC_NOTIFICATION_NUMBER")
|
|
return 143;
|
|
if (name == "CHANGE_HOST_CHECK_TIMEPERIOD")
|
|
return 144;
|
|
if (name == "CHANGE_SVC_CHECK_TIMEPERIOD")
|
|
return 145;
|
|
if (name == "PROCESS_FILE")
|
|
return 146;
|
|
if (name == "CHANGE_CUSTOM_HOST_VAR")
|
|
return 147;
|
|
if (name == "CHANGE_CUSTOM_SVC_VAR")
|
|
return 148;
|
|
if (name == "CHANGE_CUSTOM_CONTACT_VAR")
|
|
return 149;
|
|
if (name == "ENABLE_CONTACT_HOST_NOTIFICATIONS")
|
|
return 150;
|
|
if (name == "DISABLE_CONTACT_HOST_NOTIFICATIONS")
|
|
return 151;
|
|
if (name == "ENABLE_CONTACT_SVC_NOTIFICATIONS")
|
|
return 152;
|
|
if (name == "DISABLE_CONTACT_SVC_NOTIFICATIONS")
|
|
return 153;
|
|
if (name == "ENABLE_CONTACTGROUP_HOST_NOTIFICATIONS")
|
|
return 154;
|
|
if (name == "DISABLE_CONTACTGROUP_HOST_NOTIFICATIONS")
|
|
return 155;
|
|
if (name == "ENABLE_CONTACTGROUP_SVC_NOTIFICATIONS")
|
|
return 156;
|
|
if (name == "DISABLE_CONTACTGROUP_SVC_NOTIFICATIONS")
|
|
return 157;
|
|
if (name == "CHANGE_RETRY_HOST_CHECK_INTERVAL")
|
|
return 158;
|
|
if (name == "SEND_CUSTOM_HOST_NOTIFICATION")
|
|
return 159;
|
|
if (name == "SEND_CUSTOM_SVC_NOTIFICATION")
|
|
return 160;
|
|
if (name == "CHANGE_HOST_NOTIFICATION_TIMEPERIOD")
|
|
return 161;
|
|
if (name == "CHANGE_SVC_NOTIFICATION_TIMEPERIOD")
|
|
return 162;
|
|
if (name == "CHANGE_CONTACT_HOST_NOTIFICATION_TIMEPERIOD")
|
|
return 163;
|
|
if (name == "CHANGE_CONTACT_SVC_NOTIFICATION_TIMEPERIOD")
|
|
return 164;
|
|
if (name == "CHANGE_HOST_MODATTR")
|
|
return 165;
|
|
if (name == "CHANGE_SVC_MODATTR")
|
|
return 166;
|
|
if (name == "CHANGE_CONTACT_MODATTR")
|
|
return 167;
|
|
if (name == "CHANGE_CONTACT_MODHATTR")
|
|
return 168;
|
|
if (name == "CHANGE_CONTACT_MODSATTR")
|
|
return 169;
|
|
if (name == "SYNC_STATE_INFORMATION")
|
|
return 170;
|
|
if (name == "DEL_DOWNTIME_BY_HOST_NAME")
|
|
return 171;
|
|
if (name == "DEL_DOWNTIME_BY_HOSTGROUP_NAME")
|
|
return 172;
|
|
if (name == "DEL_DOWNTIME_BY_START_TIME_COMMENT")
|
|
return 173;
|
|
if (name == "ACKNOWLEDGE_HOST_PROBLEM_EXPIRE")
|
|
return 174;
|
|
if (name == "ACKNOWLEDGE_SVC_PROBLEM_EXPIRE")
|
|
return 175;
|
|
if (name == "DISABLE_NOTIFICATIONS_EXPIRE_TIME")
|
|
return 176;
|
|
if (name == "CUSTOM_COMMAND")
|
|
return 999;
|
|
|
|
return 0;
|
|
}
|