/****************************************************************************** * Icinga 2 * * Copyright (C) 2012 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/externalcommandprocessor.h" #include "icinga/host.h" #include "icinga/service.h" #include "icinga/user.h" #include "icinga/hostgroup.h" #include "icinga/servicegroup.h" #include "icinga/pluginchecktask.h" #include "base/convert.h" #include "base/logger_fwd.h" #include "base/objectlock.h" #include "base/application.h" #include #include #include #include #include using namespace icinga; boost::once_flag ExternalCommandProcessor::m_InitializeOnce = BOOST_ONCE_INIT; boost::mutex ExternalCommandProcessor::m_Mutex; std::map ExternalCommandProcessor::m_Commands; /** * @threadsafety Always. */ void ExternalCommandProcessor::Execute(const String& line) { if (line.IsEmpty()) return; if (line[0] != '[') BOOST_THROW_EXCEPTION(std::invalid_argument("Missing timestamp in command: " + line)); size_t pos = line.FindFirstOf("]"); if (pos == String::NPos) BOOST_THROW_EXCEPTION(std::invalid_argument("Missing timestamp in command: " + line)); String timestamp = line.SubStr(1, pos - 1); String args = line.SubStr(pos + 2, String::NPos); double ts = Convert::ToDouble(timestamp); if (ts == 0) BOOST_THROW_EXCEPTION(std::invalid_argument("Invalid timestamp in command: " + line)); std::vector argv; boost::algorithm::split(argv, args, boost::is_any_of(";")); if (argv.empty()) BOOST_THROW_EXCEPTION(std::invalid_argument("Missing arguments in command: " + line)); std::vector argvExtra(argv.begin() + 1, argv.end()); Execute(ts, argv[0], argvExtra); } /** * @threadsafety Always. */ void ExternalCommandProcessor::Execute(double time, const String& command, const std::vector& arguments) { boost::call_once(m_InitializeOnce, &ExternalCommandProcessor::Initialize); Callback callback; { boost::mutex::scoped_lock lock(m_Mutex); std::map::iterator it; it = m_Commands.find(command); if (it == m_Commands.end()) BOOST_THROW_EXCEPTION(std::invalid_argument("The external command '" + command + "' does not exist.")); callback = it->second; } callback(time, arguments); } /** * @threadsafety Always. */ void ExternalCommandProcessor::Initialize(void) { RegisterCommand("PROCESS_HOST_CHECK_RESULT", &ExternalCommandProcessor::ProcessHostCheckResult); RegisterCommand("PROCESS_SERVICE_CHECK_RESULT", &ExternalCommandProcessor::ProcessServiceCheckResult); RegisterCommand("SCHEDULE_HOST_CHECK", &ExternalCommandProcessor::ScheduleHostCheck); RegisterCommand("SCHEDULE_FORCED_HOST_CHECK", &ExternalCommandProcessor::ScheduleForcedHostCheck); RegisterCommand("SCHEDULE_SVC_CHECK", &ExternalCommandProcessor::ScheduleSvcCheck); RegisterCommand("SCHEDULE_FORCED_SVC_CHECK", &ExternalCommandProcessor::ScheduleForcedSvcCheck); RegisterCommand("ENABLE_HOST_CHECK", &ExternalCommandProcessor::EnableHostCheck); RegisterCommand("DISABLE_HOST_CHECK", &ExternalCommandProcessor::DisableHostCheck); RegisterCommand("ENABLE_SVC_CHECK", &ExternalCommandProcessor::EnableSvcCheck); RegisterCommand("DISABLE_SVC_CHECK", &ExternalCommandProcessor::DisableSvcCheck); RegisterCommand("SHUTDOWN_PROCESS", &ExternalCommandProcessor::ShutdownProcess); RegisterCommand("SCHEDULE_FORCED_HOST_SVC_CHECKS", &ExternalCommandProcessor::ScheduleForcedHostSvcChecks); RegisterCommand("SCHEDULE_HOST_SVC_CHECKS", &ExternalCommandProcessor::ScheduleHostSvcChecks); RegisterCommand("ENABLE_HOST_SVC_CHECKS", &ExternalCommandProcessor::EnableHostSvcChecks); RegisterCommand("DISABLE_HOST_SVC_CHECKS", &ExternalCommandProcessor::DisableHostSvcChecks); RegisterCommand("ACKNOWLEDGE_SVC_PROBLEM", &ExternalCommandProcessor::AcknowledgeSvcProblem); RegisterCommand("ACKNOWLEDGE_SVC_PROBLEM_EXPIRE", &ExternalCommandProcessor::AcknowledgeSvcProblemExpire); RegisterCommand("REMOVE_SVC_ACKNOWLEDGEMENT", &ExternalCommandProcessor::RemoveSvcAcknowledgement); RegisterCommand("ACKNOWLEDGE_HOST_PROBLEM", &ExternalCommandProcessor::AcknowledgeHostProblem); RegisterCommand("ACKNOWLEDGE_HOST_PROBLEM_EXPIRE", &ExternalCommandProcessor::AcknowledgeHostProblemExpire); RegisterCommand("REMOVE_HOST_ACKNOWLEDGEMENT", &ExternalCommandProcessor::RemoveHostAcknowledgement); RegisterCommand("ENABLE_HOSTGROUP_SVC_CHECKS", &ExternalCommandProcessor::EnableHostgroupSvcChecks); RegisterCommand("DISABLE_HOSTGROUP_SVC_CHECKS", &ExternalCommandProcessor::DisableHostgroupSvcChecks); RegisterCommand("ENABLE_SERVICEGROUP_SVC_CHECKS", &ExternalCommandProcessor::EnableServicegroupSvcChecks); RegisterCommand("DISABLE_SERVICEGROUP_SVC_CHECKS", &ExternalCommandProcessor::DisableServicegroupSvcChecks); RegisterCommand("ENABLE_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::EnablePassiveHostChecks); RegisterCommand("DISABLE_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::DisablePassiveHostChecks); RegisterCommand("ENABLE_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::EnablePassiveSvcChecks); RegisterCommand("DISABLE_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::DisablePassiveSvcChecks); RegisterCommand("ENABLE_SERVICEGROUP_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::EnableServicegroupPassiveSvcChecks); RegisterCommand("DISABLE_SERVICEGROUP_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::DisableServicegroupPassiveSvcChecks); RegisterCommand("ENABLE_HOSTGROUP_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::EnableHostgroupPassiveSvcChecks); RegisterCommand("DISABLE_HOSTGROUP_PASSIVE_SVC_CHECKS", &ExternalCommandProcessor::DisableHostgroupPassiveSvcChecks); RegisterCommand("PROCESS_FILE", &ExternalCommandProcessor::ProcessFile); RegisterCommand("SCHEDULE_SVC_DOWNTIME", &ExternalCommandProcessor::ScheduleSvcDowntime); RegisterCommand("DEL_SVC_DOWNTIME", &ExternalCommandProcessor::DelSvcDowntime); RegisterCommand("SCHEDULE_HOST_DOWNTIME", &ExternalCommandProcessor::ScheduleHostDowntime); RegisterCommand("DEL_HOST_DOWNTIME", &ExternalCommandProcessor::DelHostDowntime); RegisterCommand("SCHEDULE_HOST_SVC_DOWNTIME", &ExternalCommandProcessor::ScheduleHostSvcDowntime); RegisterCommand("SCHEDULE_HOSTGROUP_HOST_DOWNTIME", &ExternalCommandProcessor::ScheduleHostgroupHostDowntime); RegisterCommand("SCHEDULE_HOSTGROUP_SVC_DOWNTIME", &ExternalCommandProcessor::ScheduleHostgroupSvcDowntime); RegisterCommand("SCHEDULE_SERVICEGROUP_HOST_DOWNTIME", &ExternalCommandProcessor::ScheduleServicegroupHostDowntime); RegisterCommand("SCHEDULE_SERVICEGROUP_SVC_DOWNTIME", &ExternalCommandProcessor::ScheduleServicegroupSvcDowntime); RegisterCommand("ADD_HOST_COMMENT", &ExternalCommandProcessor::AddHostComment); RegisterCommand("DEL_HOST_COMMENT", &ExternalCommandProcessor::DelHostComment); RegisterCommand("ADD_SVC_COMMENT", &ExternalCommandProcessor::AddSvcComment); RegisterCommand("DEL_SVC_COMMENT", &ExternalCommandProcessor::DelSvcComment); RegisterCommand("DEL_ALL_HOST_COMMENTS", &ExternalCommandProcessor::DelAllHostComments); RegisterCommand("DEL_ALL_SVC_COMMENTS", &ExternalCommandProcessor::DelAllSvcComments); RegisterCommand("SEND_CUSTOM_HOST_NOTIFICATION", &ExternalCommandProcessor::SendCustomHostNotification); RegisterCommand("SEND_CUSTOM_SVC_NOTIFICATION", &ExternalCommandProcessor::SendCustomSvcNotification); RegisterCommand("DELAY_HOST_NOTIFICATION", &ExternalCommandProcessor::DelayHostNotification); RegisterCommand("DELAY_SVC_NOTIFICATION", &ExternalCommandProcessor::DelaySvcNotification); RegisterCommand("ENABLE_HOST_NOTIFICATIONS", &ExternalCommandProcessor::EnableHostNotifications); RegisterCommand("DISABLE_HOST_NOTIFICATIONS", &ExternalCommandProcessor::DisableHostNotifications); RegisterCommand("ENABLE_SVC_NOTIFICATIONS", &ExternalCommandProcessor::EnableSvcNotifications); RegisterCommand("DISABLE_SVC_NOTIFICATIONS", &ExternalCommandProcessor::DisableSvcNotifications); } /** * @threadsafety Always. */ void ExternalCommandProcessor::RegisterCommand(const String& command, const ExternalCommandProcessor::Callback& callback) { boost::mutex::scoped_lock lock(m_Mutex); m_Commands[command] = callback; } void ExternalCommandProcessor::ProcessHostCheckResult(double time, const std::vector& arguments) { if (arguments.size() < 3) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 3 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Service::Ptr hc = host->GetHostCheckService(); if (!hc->GetEnablePassiveChecks()) BOOST_THROW_EXCEPTION(std::invalid_argument("Got passive check result for host '" + arguments[0] + "' which has passive checks disabled.")); int exitStatus = Convert::ToDouble(arguments[1]); Dictionary::Ptr result = PluginCheckTask::ParseCheckOutput(arguments[2]); result->Set("state", PluginCheckTask::ExitStatusToState(exitStatus)); result->Set("schedule_start", time); result->Set("schedule_end", time); result->Set("execution_start", time); result->Set("execution_end", time); result->Set("active", 0); Log(LogInformation, "icinga", "Processing passive check result for host '" + arguments[0] + "'"); hc->ProcessCheckResult(result); { ObjectLock olock(hc); /* Reschedule the next check. The side effect of this is that for as long * as we receive passive results for a service we won't execute any * active checks. */ hc->SetNextCheck(Utility::GetTime() + hc->GetCheckInterval()); } } void ExternalCommandProcessor::ProcessServiceCheckResult(double time, const std::vector& arguments) { if (arguments.size() < 4) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 4 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); if (!service->GetEnablePassiveChecks()) BOOST_THROW_EXCEPTION(std::invalid_argument("Got passive check result for service '" + arguments[1] + "' which has passive checks disabled.")); int exitStatus = Convert::ToDouble(arguments[2]); Dictionary::Ptr result = PluginCheckTask::ParseCheckOutput(arguments[3]); result->Set("state", PluginCheckTask::ExitStatusToState(exitStatus)); result->Set("schedule_start", time); result->Set("schedule_end", time); result->Set("execution_start", time); result->Set("execution_end", time); result->Set("active", 0); Log(LogInformation, "icinga", "Processing passive check result for service '" + arguments[1] + "'"); service->ProcessCheckResult(result); { ObjectLock olock(service); /* Reschedule the next check. The side effect of this is that for as long * as we receive passive results for a service we won't execute any * active checks. */ service->SetNextCheck(Utility::GetTime() + service->GetCheckInterval()); } } void ExternalCommandProcessor::ScheduleHostCheck(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Service::Ptr hc = host->GetHostCheckService(); double planned_check = Convert::ToDouble(arguments[1]); if (planned_check > hc->GetNextCheck()) { Log(LogInformation, "icinga", "Ignoring reschedule request for host '" + arguments[0] + "' (next check is already sooner than requested check time)"); return; } Log(LogInformation, "icinga", "Rescheduling next check for host '" + arguments[0] + "'"); { ObjectLock olock(hc); hc->SetNextCheck(planned_check); } } void ExternalCommandProcessor::ScheduleForcedHostCheck(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Service::Ptr hc = host->GetHostCheckService(); Log(LogInformation, "icinga", "Rescheduling next check for host '" + arguments[0] + "'"); { ObjectLock olock(hc); hc->SetForceNextCheck(true); hc->SetNextCheck(Convert::ToDouble(arguments[1])); } } void ExternalCommandProcessor::ScheduleSvcCheck(double, const std::vector& arguments) { if (arguments.size() < 3) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 3 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); double planned_check = Convert::ToDouble(arguments[2]); if (planned_check > service->GetNextCheck()) { Log(LogInformation, "icinga", "Ignoring reschedule request for service '" + arguments[1] + "' (next check is already sooner than requested check time)"); return; } Log(LogInformation, "icinga", "Rescheduling next check for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetNextCheck(planned_check); } } void ExternalCommandProcessor::ScheduleForcedSvcCheck(double, const std::vector& arguments) { if (arguments.size() < 3) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 3 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Rescheduling next check for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetForceNextCheck(true); service->SetNextCheck(Convert::ToDouble(arguments[2])); } } void ExternalCommandProcessor::EnableHostCheck(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Enabling active checks for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnableActiveChecks(true); } } void ExternalCommandProcessor::DisableHostCheck(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Disabling active checks for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnableActiveChecks(false); } } void ExternalCommandProcessor::EnableSvcCheck(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Enabling active checks for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(true); } } void ExternalCommandProcessor::DisableSvcCheck(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Disabling active checks for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(false); } } void ExternalCommandProcessor::ShutdownProcess(double, const std::vector&) { Log(LogInformation, "icinga", "Shutting down Icinga via external command."); Application::RequestShutdown(); } void ExternalCommandProcessor::ScheduleForcedHostSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); double planned_check = Convert::ToDouble(arguments[1]); Host::Ptr host = Host::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Rescheduling next check for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetNextCheck(planned_check); service->SetForceNextCheck(true); } } } void ExternalCommandProcessor::ScheduleHostSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); double planned_check = Convert::ToDouble(arguments[1]); Host::Ptr host = Host::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { if (planned_check > service->GetNextCheck()) { Log(LogInformation, "icinga", "Ignoring reschedule request for service '" + service->GetName() + "' (next check is already sooner than requested check time)"); continue; } Log(LogInformation, "icinga", "Rescheduling next check for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetNextCheck(planned_check); } } } void ExternalCommandProcessor::EnableHostSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Enabling active checks for service '" + service->GetName() + "'"); service->SetEnableActiveChecks(true); } } void ExternalCommandProcessor::DisableHostSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Disabling active checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(false); } } } void ExternalCommandProcessor::AcknowledgeSvcProblem(double, const std::vector& arguments) { if (arguments.size() < 7) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 7 arguments.")); bool sticky = Convert::ToBool(arguments[2]); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); if (service->GetState() == StateOK) BOOST_THROW_EXCEPTION(std::invalid_argument("The service '" + arguments[1] + "' is OK.")); Log(LogInformation, "icinga", "Setting acknowledgement for service '" + service->GetName() + "'"); service->AcknowledgeProblem(sticky ? AcknowledgementSticky : AcknowledgementNormal); } void ExternalCommandProcessor::AcknowledgeSvcProblemExpire(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); bool sticky = Convert::ToBool(arguments[2]); double timestamp = Convert::ToDouble(arguments[5]); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); if (service->GetState() == StateOK) BOOST_THROW_EXCEPTION(std::invalid_argument("The service '" + arguments[1] + "' is OK.")); Log(LogInformation, "icinga", "Setting timed acknowledgement for service '" + service->GetName() + "'"); service->AcknowledgeProblem(sticky ? AcknowledgementSticky : AcknowledgementNormal, timestamp); } void ExternalCommandProcessor::RemoveSvcAcknowledgement(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Removing acknowledgement for service '" + service->GetName() + "'"); service->ClearAcknowledgement(); } void ExternalCommandProcessor::AcknowledgeHostProblem(double, const std::vector& arguments) { if (arguments.size() < 6) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 6 arguments.")); bool sticky = Convert::ToBool(arguments[1]); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Setting acknowledgement for host '" + host->GetName() + "'"); Service::Ptr service = host->GetHostCheckService(); if (service) { if (service->GetState() == StateOK) BOOST_THROW_EXCEPTION(std::invalid_argument("The host '" + arguments[0] + "' is OK.")); service->AcknowledgeProblem(sticky ? AcknowledgementSticky : AcknowledgementNormal); } } void ExternalCommandProcessor::AcknowledgeHostProblemExpire(double, const std::vector& arguments) { if (arguments.size() < 7) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 7 arguments.")); bool sticky = Convert::ToBool(arguments[1]); double timestamp = Convert::ToDouble(arguments[4]); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Setting timed acknowledgement for host '" + host->GetName() + "'"); Service::Ptr service = host->GetHostCheckService(); if (service) { if (service->GetState() == StateOK) BOOST_THROW_EXCEPTION(std::invalid_argument("The host '" + arguments[0] + "' is OK.")); service->AcknowledgeProblem(sticky ? AcknowledgementSticky : AcknowledgementNormal, timestamp); } } void ExternalCommandProcessor::RemoveHostAcknowledgement(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Removing acknowledgement for host '" + host->GetName() + "'"); Service::Ptr service = host->GetHostCheckService(); if (service) service->ClearAcknowledgement(); } void ExternalCommandProcessor::EnableHostgroupSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Enabling active checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(true); } } } } void ExternalCommandProcessor::DisableHostgroupSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Disabling active checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(false); } } } } void ExternalCommandProcessor::EnableServicegroupSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Log(LogInformation, "icinga", "Enabling active checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(true); } } } void ExternalCommandProcessor::DisableServicegroupSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Log(LogInformation, "icinga", "Disabling active checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnableActiveChecks(false); } } } void ExternalCommandProcessor::EnablePassiveHostChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Enabling passive checks for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnablePassiveChecks(true); } } void ExternalCommandProcessor::DisablePassiveHostChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Disabling passive checks for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnablePassiveChecks(false); } } void ExternalCommandProcessor::EnablePassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Enabling passive checks for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(true); } } void ExternalCommandProcessor::DisablePassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Disabling passive checks for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(false); } } void ExternalCommandProcessor::EnableServicegroupPassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Log(LogInformation, "icinga", "Enabling passive checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(true); } } } void ExternalCommandProcessor::DisableServicegroupPassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Log(LogInformation, "icinga", "Disabling passive checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(true); } } } void ExternalCommandProcessor::EnableHostgroupPassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Enabling passive checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(true); } } } } void ExternalCommandProcessor::DisableHostgroupPassiveSvcChecks(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Disabling passive checks for service '" + service->GetName() + "'"); { ObjectLock olock(service); service->SetEnablePassiveChecks(false); } } } } void ExternalCommandProcessor::ProcessFile(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); String file = arguments[0]; bool del = Convert::ToBool(arguments[1]); std::ifstream ifp; ifp.exceptions(std::ifstream::badbit); ifp.open(file.CStr(), std::ifstream::in); while(ifp.good()) { std::string line; std::getline(ifp, line); try { Log(LogInformation, "compat", "Executing external command: " + line); Execute(line); } catch (const std::exception& ex) { std::ostringstream msgbuf; msgbuf << "External command failed: " << boost::diagnostic_information(ex); Log(LogWarning, "icinga", msgbuf.str()); } } ifp.close(); if (del) (void) unlink(file.CStr()); } void ExternalCommandProcessor::ScheduleSvcDowntime(double, const std::vector& arguments) { if (arguments.size() < 9) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 9 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[5]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); Log(LogInformation, "icinga", "Creating downtime for service " + service->GetName()); (void) service->AddDowntime(arguments[7], arguments[8], Convert::ToDouble(arguments[2]), Convert::ToDouble(arguments[3]), Convert::ToBool(arguments[4]), triggeredBy, Convert::ToDouble(arguments[6])); } void ExternalCommandProcessor::DelSvcDowntime(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); int id = Convert::ToLong(arguments[0]); Log(LogInformation, "icinga", "Removing downtime ID " + arguments[0]); String rid = Service::GetDowntimeIDFromLegacyID(id); Service::RemoveDowntime(rid); } void ExternalCommandProcessor::ScheduleHostDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); Log(LogInformation, "icinga", "Creating downtime for host " + host->GetName()); Service::Ptr service = host->GetHostCheckService(); if (service) { (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } void ExternalCommandProcessor::DelHostDowntime(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); int id = Convert::ToLong(arguments[0]); Log(LogInformation, "icinga", "Removing downtime ID " + arguments[0]); String rid = Service::GetDowntimeIDFromLegacyID(id); Service::RemoveDowntime(rid); } void ExternalCommandProcessor::ScheduleHostSvcDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { Log(LogInformation, "icinga", "Creating downtime for service " + service->GetName()); (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } void ExternalCommandProcessor::ScheduleHostgroupHostDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { Log(LogInformation, "icinga", "Creating downtime for host " + host->GetName()); Service::Ptr service = host->GetHostCheckService(); if (service) { (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } } void ExternalCommandProcessor::ScheduleHostgroupSvcDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); /* Note: we can't just directly create downtimes for all the services by iterating * over all hosts in the host group - otherwise we might end up creating multiple * downtimes for some services. */ std::set services; BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) { BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) { services.insert(service); } } BOOST_FOREACH(const Service::Ptr& service, services) { Log(LogInformation, "icinga", "Creating downtime for service " + service->GetName()); (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } void ExternalCommandProcessor::ScheduleServicegroupHostDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); /* Note: we can't just directly create downtimes for all the hosts by iterating * over all services in the service group - otherwise we might end up creating multiple * downtimes for some hosts. */ std::set services; BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Host::Ptr host = service->GetHost(); Service::Ptr hcService = host->GetHostCheckService(); if (hcService) services.insert(hcService); } BOOST_FOREACH(const Service::Ptr& service, services) { Log(LogInformation, "icinga", "Creating downtime for service " + service->GetName()); (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } void ExternalCommandProcessor::ScheduleServicegroupSvcDowntime(double, const std::vector& arguments) { if (arguments.size() < 8) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 8 arguments.")); ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]); String triggeredBy; int triggeredByLegacy = Convert::ToLong(arguments[4]); if (triggeredByLegacy != 0) triggeredBy = Service::GetDowntimeIDFromLegacyID(triggeredByLegacy); BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) { Log(LogInformation, "icinga", "Creating downtime for service " + service->GetName()); (void) service->AddDowntime(arguments[6], arguments[7], Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]), Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5])); } } void ExternalCommandProcessor::AddHostComment(double, const std::vector& arguments) { if (arguments.size() < 4) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 4 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Creating comment for host " + host->GetName()); Service::Ptr service = host->GetHostCheckService(); if (service) (void) service->AddComment(CommentUser, arguments[2], arguments[3], 0); } void ExternalCommandProcessor::DelHostComment(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); int id = Convert::ToLong(arguments[0]); Log(LogInformation, "icinga", "Removing comment ID " + arguments[0]); String rid = Service::GetCommentIDFromLegacyID(id); Service::RemoveComment(rid); } void ExternalCommandProcessor::AddSvcComment(double, const std::vector& arguments) { if (arguments.size() < 5) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 5 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Creating comment for service " + service->GetName()); (void) service->AddComment(CommentUser, arguments[3], arguments[4], 0); } void ExternalCommandProcessor::DelSvcComment(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); int id = Convert::ToLong(arguments[0]); Log(LogInformation, "icinga", "Removing comment ID " + arguments[0]); String rid = Service::GetCommentIDFromLegacyID(id); Service::RemoveComment(rid); } void ExternalCommandProcessor::DelAllHostComments(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Removing all comments for host " + host->GetName()); Service::Ptr service = host->GetHostCheckService(); if (service) service->RemoveAllComments(); } void ExternalCommandProcessor::DelAllSvcComments(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Removing all comments for service " + service->GetName()); service->RemoveAllComments(); } void ExternalCommandProcessor::SendCustomHostNotification(double, const std::vector& arguments) { if (arguments.size() < 4) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 4 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Sending custom notification for host " + host->GetName()); Service::Ptr service = host->GetHostCheckService(); if (service) service->RequestNotifications(NotificationCustom, service->GetLastCheckResult()); } void ExternalCommandProcessor::SendCustomSvcNotification(double, const std::vector& arguments) { if (arguments.size() < 5) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 5 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Sending custom notification for service " + service->GetName()); service->RequestNotifications(NotificationCustom, service->GetLastCheckResult()); } void ExternalCommandProcessor::DelayHostNotification(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Delaying notifications for host " + host->GetName()); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; BOOST_FOREACH(const Notification::Ptr& notification, hc->GetNotifications()) { ObjectLock olock(notification); notification->SetNextNotification(Convert::ToDouble(arguments[1])); } } void ExternalCommandProcessor::DelaySvcNotification(double, const std::vector& arguments) { if (arguments.size() < 3) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 3 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Delaying notifications for service " + service->GetName()); BOOST_FOREACH(const Notification::Ptr& notification, service->GetNotifications()) { ObjectLock olock(notification); notification->SetNextNotification(Convert::ToDouble(arguments[2])); } } void ExternalCommandProcessor::EnableHostNotifications(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Enabling notifications for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnableNotifications(true); } } void ExternalCommandProcessor::DisableHostNotifications(double, const std::vector& arguments) { if (arguments.size() < 1) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 argument.")); Host::Ptr host = Host::GetByName(arguments[0]); Log(LogInformation, "icinga", "Disabling notifications for host '" + arguments[0] + "'"); Service::Ptr hc = host->GetHostCheckService(); if (!hc) return; { ObjectLock olock(hc); hc->SetEnableNotifications(false); } } void ExternalCommandProcessor::EnableSvcNotifications(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Enabling notifications for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnableNotifications(true); } } void ExternalCommandProcessor::DisableSvcNotifications(double, const std::vector& arguments) { if (arguments.size() < 2) BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 2 arguments.")); Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]); Log(LogInformation, "icinga", "Disabling notifications for service '" + arguments[1] + "'"); { ObjectLock olock(service); service->SetEnableNotifications(false); } }