mirror of
				https://github.com/Icinga/icinga2.git
				synced 2025-11-03 21:25:56 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			1545 lines
		
	
	
		
			52 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			1545 lines
		
	
	
		
			52 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/******************************************************************************
 | 
						|
 * 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 "base/utility.h"
 | 
						|
#include <fstream>
 | 
						|
#include <boost/algorithm/string/classification.hpp>
 | 
						|
#include <boost/foreach.hpp>
 | 
						|
#include <boost/exception/diagnostic_information.hpp>
 | 
						|
#include <boost/algorithm/string/split.hpp>
 | 
						|
 | 
						|
using namespace icinga;
 | 
						|
 | 
						|
boost::once_flag ExternalCommandProcessor::m_InitializeOnce = BOOST_ONCE_INIT;
 | 
						|
boost::mutex ExternalCommandProcessor::m_Mutex;
 | 
						|
std::map<String, ExternalCommandProcessor::Callback> ExternalCommandProcessor::m_Commands;
 | 
						|
 | 
						|
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<String> 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<String> argvExtra(argv.begin() + 1, argv.end());
 | 
						|
	Execute(ts, argv[0], argvExtra);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::Execute(double time, const String& command, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	boost::call_once(m_InitializeOnce, &ExternalCommandProcessor::Initialize);
 | 
						|
 | 
						|
	Callback callback;
 | 
						|
 | 
						|
	{
 | 
						|
		boost::mutex::scoped_lock lock(m_Mutex);
 | 
						|
 | 
						|
		std::map<String, ExternalCommandProcessor::Callback>::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);
 | 
						|
}
 | 
						|
 | 
						|
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("RESTART_PROCESS", &ExternalCommandProcessor::RestartProcess);
 | 
						|
	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("DISABLE_HOST_FLAP_DETECTION", &ExternalCommandProcessor::DisableHostFlapping);
 | 
						|
	RegisterCommand("ENABLE_HOST_FLAP_DETECTION", &ExternalCommandProcessor::EnableHostFlapping);
 | 
						|
	RegisterCommand("DISABLE_SVC_FLAP_DETECTION", &ExternalCommandProcessor::DisableSvcFlapping);
 | 
						|
	RegisterCommand("ENABLE_SVC_FLAP_DETECTION", &ExternalCommandProcessor::EnableSvcFlapping);
 | 
						|
	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);
 | 
						|
	RegisterCommand("DISABLE_HOSTGROUP_HOST_CHECKS", &ExternalCommandProcessor::DisableHostgroupHostChecks);
 | 
						|
	RegisterCommand("DISABLE_HOSTGROUP_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::DisableHostgroupPassiveHostChecks);
 | 
						|
	RegisterCommand("DISABLE_SERVICEGROUP_HOST_CHECKS", &ExternalCommandProcessor::DisableServicegroupHostChecks);
 | 
						|
	RegisterCommand("DISABLE_SERVICEGROUP_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::DisableServicegroupPassiveHostChecks);
 | 
						|
	RegisterCommand("ENABLE_HOSTGROUP_HOST_CHECKS", &ExternalCommandProcessor::EnableHostgroupHostChecks);
 | 
						|
	RegisterCommand("ENABLE_HOSTGROUP_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::EnableHostgroupPassiveHostChecks);
 | 
						|
	RegisterCommand("ENABLE_SERVICEGROUP_HOST_CHECKS", &ExternalCommandProcessor::EnableServicegroupHostChecks);
 | 
						|
	RegisterCommand("ENABLE_SERVICEGROUP_PASSIVE_HOST_CHECKS", &ExternalCommandProcessor::EnableServicegroupPassiveHostChecks);
 | 
						|
}
 | 
						|
 | 
						|
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<String>& 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<String>& 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<String>& 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();
 | 
						|
 | 
						|
	if (!hc) {
 | 
						|
		Log(LogInformation, "icinga", "Ignoring request request for host '" +
 | 
						|
		    arguments[0] + "' (does not have a host check)");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	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] + "'");
 | 
						|
 | 
						|
	if (planned_check < Utility::GetTime())
 | 
						|
		planned_check = Utility::GetTime();
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(hc);
 | 
						|
 | 
						|
		hc->SetNextCheck(planned_check);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleForcedHostCheck(double, const std::vector<String>& 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();
 | 
						|
 | 
						|
	if (!hc) {
 | 
						|
		Log(LogInformation, "icinga", "Ignoring request request for host '" +
 | 
						|
		    arguments[0] + "' (does not have a host check)");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	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<String>& 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] + "'");
 | 
						|
 | 
						|
	if (planned_check < Utility::GetTime())
 | 
						|
		planned_check = Utility::GetTime();
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(service);
 | 
						|
 | 
						|
		service->SetNextCheck(planned_check);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleForcedSvcCheck(double, const std::vector<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>&)
 | 
						|
{
 | 
						|
	Log(LogInformation, "icinga", "Shutting down Icinga via external command.");
 | 
						|
	Application::RequestShutdown();
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::RestartProcess(double, const std::vector<String>&)
 | 
						|
{
 | 
						|
	Log(LogInformation, "icinga", "Restarting Icinga via external command.");
 | 
						|
	Application::RequestRestart();
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleForcedHostSvcChecks(double, const std::vector<String>& 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<String>& 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]);
 | 
						|
 | 
						|
	if (planned_check < Utility::GetTime())
 | 
						|
		planned_check = Utility::GetTime();
 | 
						|
 | 
						|
	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<String>& 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<String>& 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<String>& 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->AddComment(CommentAcknowledgement, arguments[5], arguments[6], 0);
 | 
						|
	service->AcknowledgeProblem(arguments[5], arguments[6], sticky ? AcknowledgementSticky : AcknowledgementNormal);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::AcknowledgeSvcProblemExpire(double, const std::vector<String>& 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->AddComment(CommentAcknowledgement, arguments[6], arguments[7], 0);
 | 
						|
	service->AcknowledgeProblem(arguments[6], arguments[7], sticky ? AcknowledgementSticky : AcknowledgementNormal, timestamp);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::RemoveSvcAcknowledgement(double, const std::vector<String>& 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<String>& 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->AddComment(CommentAcknowledgement, arguments[4], arguments[5], 0);
 | 
						|
		service->AcknowledgeProblem(arguments[4], arguments[5], sticky ? AcknowledgementSticky : AcknowledgementNormal);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::AcknowledgeHostProblemExpire(double, const std::vector<String>& 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->AddComment(CommentAcknowledgement, arguments[5], arguments[6], 0);
 | 
						|
		service->AcknowledgeProblem(arguments[5], arguments[6], sticky ? AcknowledgementSticky : AcknowledgementNormal, timestamp);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::RemoveHostAcknowledgement(double, const std::vector<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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());
 | 
						|
	String comment_id = service->AddComment(CommentDowntime, arguments[7], arguments[8], Convert::ToDouble(arguments[3]));
 | 
						|
	(void) service->AddDowntime(comment_id,
 | 
						|
	    Convert::ToDouble(arguments[2]), Convert::ToDouble(arguments[3]),
 | 
						|
	    Convert::ToBool(arguments[4]), triggeredBy, Convert::ToDouble(arguments[6]));
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DelSvcDowntime(double, const std::vector<String>& 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, true);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleHostDowntime(double, const std::vector<String>& 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) {
 | 
						|
		String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
		(void) service->AddDowntime(comment_id,
 | 
						|
		    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
		    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DelHostDowntime(double, const std::vector<String>& 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, true);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleHostSvcDowntime(double, const std::vector<String>& 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());
 | 
						|
		String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
		(void) service->AddDowntime(comment_id,
 | 
						|
		    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
		    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleHostgroupHostDowntime(double, const std::vector<String>& 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) {
 | 
						|
			String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
			(void) service->AddDowntime(comment_id,
 | 
						|
			    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
			    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleHostgroupSvcDowntime(double, const std::vector<String>& 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<Service::Ptr> 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());
 | 
						|
		String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
		(void) service->AddDowntime(comment_id,
 | 
						|
		    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
		    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleServicegroupHostDowntime(double, const std::vector<String>& 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<Service::Ptr> 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());
 | 
						|
		String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
		(void) service->AddDowntime(comment_id,
 | 
						|
		    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
		    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::ScheduleServicegroupSvcDowntime(double, const std::vector<String>& 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());
 | 
						|
		String comment_id = service->AddComment(CommentDowntime, arguments[6], arguments[7], Convert::ToDouble(arguments[2]));
 | 
						|
		(void) service->AddDowntime(comment_id,
 | 
						|
		    Convert::ToDouble(arguments[1]), Convert::ToDouble(arguments[2]),
 | 
						|
		    Convert::ToBool(arguments[3]), triggeredBy, Convert::ToDouble(arguments[5]));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::AddHostComment(double, const std::vector<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 4)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 4 arguments."));
 | 
						|
 | 
						|
	Host::Ptr host = Host::GetByName(arguments[0]);
 | 
						|
	int options = Convert::ToLong(arguments[1]);
 | 
						|
 | 
						|
	Log(LogInformation, "icinga", "Sending custom notification for host " + host->GetName());
 | 
						|
	Service::Ptr service = host->GetHostCheckService();
 | 
						|
	if (service) {
 | 
						|
		if (options & 2) {
 | 
						|
			ObjectLock olock(service);
 | 
						|
			service->SetForceNextNotification(true);
 | 
						|
		}
 | 
						|
 | 
						|
		Service::OnNotificationsRequested(service, NotificationCustom, service->GetLastCheckResult(), arguments[2], arguments[3]);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::SendCustomSvcNotification(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 5)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 5 arguments."));
 | 
						|
 | 
						|
	Service::Ptr service = Service::GetByNamePair(arguments[0], arguments[1]);
 | 
						|
	int options = Convert::ToLong(arguments[2]);
 | 
						|
 | 
						|
	Log(LogInformation, "icinga", "Sending custom notification for service " + service->GetName());
 | 
						|
 | 
						|
	if (options & 2) {
 | 
						|
		ObjectLock olock(service);
 | 
						|
		service->SetForceNextNotification(true);
 | 
						|
	}
 | 
						|
 | 
						|
	Service::OnNotificationsRequested(service, NotificationCustom, service->GetLastCheckResult(), arguments[3], arguments[4]);
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DelayHostNotification(double, const std::vector<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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<String>& 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);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableHostgroupHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) {
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Disabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnableActiveChecks(false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableHostgroupPassiveHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) {
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Disabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnablePassiveChecks(false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableServicegroupHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) {
 | 
						|
		Host::Ptr host = service->GetHost();
 | 
						|
 | 
						|
		if (!host)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Disabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnableActiveChecks(false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableServicegroupPassiveHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) {
 | 
						|
		Host::Ptr host = service->GetHost();
 | 
						|
 | 
						|
		if (!host)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Disabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnablePassiveChecks(false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableHostgroupHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	HostGroup::Ptr hg = HostGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) {
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Enabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnableActiveChecks(true);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableHostgroupPassiveHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableServicegroupHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) {
 | 
						|
		Host::Ptr host = service->GetHost();
 | 
						|
 | 
						|
		if (!host)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Enabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnableActiveChecks(true);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableServicegroupPassiveHostChecks(double, const std::vector<String>& arguments)
 | 
						|
{
 | 
						|
	if (arguments.size() < 1)
 | 
						|
		BOOST_THROW_EXCEPTION(std::invalid_argument("Expected 1 arguments."));
 | 
						|
 | 
						|
	ServiceGroup::Ptr sg = ServiceGroup::GetByName(arguments[0]);
 | 
						|
 | 
						|
	BOOST_FOREACH(const Service::Ptr& service, sg->GetMembers()) {
 | 
						|
		Host::Ptr host = service->GetHost();
 | 
						|
 | 
						|
		if (!host)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
		if (!hc)
 | 
						|
			continue;
 | 
						|
 | 
						|
		Log(LogInformation, "icinga", "Enabling active checks for host '" + host->GetName() + "'");
 | 
						|
 | 
						|
		{
 | 
						|
			ObjectLock olock(hc);
 | 
						|
 | 
						|
			hc->SetEnablePassiveChecks(false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableHostFlapping(double, const std::vector<String>& 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 flapping detection for host '" + arguments[0] + "'");
 | 
						|
	Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
	if (!hc)
 | 
						|
		return;
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(hc);
 | 
						|
 | 
						|
		hc->SetEnableFlapping(true);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableHostFlapping(double, const std::vector<String>& 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 flapping detection for host '" + arguments[0] + "'");
 | 
						|
	Service::Ptr hc = host->GetHostCheckService();
 | 
						|
 | 
						|
	if (!hc)
 | 
						|
		return;
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(hc);
 | 
						|
 | 
						|
		hc->SetEnableFlapping(false);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::EnableSvcFlapping(double, const std::vector<String>& 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 flapping detection for service '" + arguments[1] + "'");
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(service);
 | 
						|
 | 
						|
		service->SetEnableFlapping(true);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void ExternalCommandProcessor::DisableSvcFlapping(double, const std::vector<String>& 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 flapping detection for service '" + arguments[1] + "'");
 | 
						|
 | 
						|
	{
 | 
						|
		ObjectLock olock(service);
 | 
						|
 | 
						|
		service->SetEnableFlapping(false);
 | 
						|
	}
 | 
						|
}
 |