icinga2/lib/icinga/host.cpp

645 lines
14 KiB
C++
Raw Normal View History

/******************************************************************************
* Icinga 2 *
2014-01-09 00:32:11 +01:00
* Copyright (C) 2012-present 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. *
******************************************************************************/
2013-03-17 20:19:29 +01:00
#include "icinga/host.h"
#include "icinga/service.h"
#include "icinga/hostgroup.h"
#include "icinga/icingaapplication.h"
#include "icinga/pluginutility.h"
2013-03-16 21:18:53 +01:00
#include "base/dynamictype.h"
#include "base/objectlock.h"
#include "base/logger_fwd.h"
2013-03-18 11:02:18 +01:00
#include "base/timer.h"
#include "base/convert.h"
2013-10-18 11:30:55 +02:00
#include "base/utility.h"
2013-03-25 18:36:15 +01:00
#include "base/scriptfunction.h"
2013-08-28 08:18:58 +02:00
#include "base/debug.h"
#include "base/serializer.h"
2013-03-17 20:19:29 +01:00
#include "config/configitembuilder.h"
#include "config/configcompilercontext.h"
2013-03-16 21:18:53 +01:00
#include <boost/foreach.hpp>
using namespace icinga;
2013-03-01 12:07:52 +01:00
REGISTER_TYPE(Host);
void Host::Start(void)
2013-02-26 10:13:54 +01:00
{
DynamicObject::Start();
2013-02-26 10:13:54 +01:00
ASSERT(!OwnsLock());
2013-02-20 19:52:25 +01:00
Array::Ptr groups = GetGroups();
if (groups) {
2013-09-09 13:52:37 +02:00
ObjectLock olock(groups);
BOOST_FOREACH(const String& name, groups) {
HostGroup::Ptr hg = HostGroup::GetByName(name);
if (hg)
hg->AddMember(GetSelf());
}
}
}
void Host::OnConfigLoaded(void)
{
UpdateSlaveServices();
}
void Host::Stop(void)
{
DynamicObject::Stop();
2013-03-02 09:07:47 +01:00
Array::Ptr groups = GetGroups();
if (groups) {
2013-09-09 13:52:37 +02:00
ObjectLock olock(groups);
BOOST_FOREACH(const String& name, groups) {
HostGroup::Ptr hg = HostGroup::GetByName(name);
if (hg)
hg->RemoveMember(GetSelf());
}
}
// TODO: unregister slave services/notifications?
}
2013-03-02 09:07:47 +01:00
bool Host::IsReachable(void) const
{
ASSERT(!OwnsLock());
2013-03-02 09:07:47 +01:00
2013-03-16 21:18:53 +01:00
std::set<Service::Ptr> parentServices = GetParentServices();
2013-02-19 23:02:08 +01:00
BOOST_FOREACH(const Service::Ptr& service, parentServices) {
ObjectLock olock(service);
2013-02-07 20:29:35 +01:00
/* ignore pending services */
if (!service->GetLastCheckResult())
continue;
2013-02-07 20:29:35 +01:00
/* ignore soft states */
if (service->GetStateType() == StateTypeSoft)
continue;
2013-02-07 20:29:35 +01:00
/* ignore services states OK and Warning */
if (service->GetState() == StateOK ||
service->GetState() == StateWarning)
continue;
return false;
}
2013-03-16 21:18:53 +01:00
std::set<Host::Ptr> parentHosts = GetParentHosts();
2013-02-19 23:02:08 +01:00
BOOST_FOREACH(const Host::Ptr& host, parentHosts) {
Service::Ptr hc = host->GetCheckService();
2013-03-02 09:07:47 +01:00
/* ignore hosts that don't have a check */
2013-03-02 09:07:47 +01:00
if (!hc)
continue;
ObjectLock olock(hc);
2013-03-04 15:52:42 +01:00
/* ignore soft states */
if (hc->GetStateType() == StateTypeSoft)
continue;
2013-02-19 23:02:08 +01:00
/* ignore hosts that are up */
2013-03-02 09:07:47 +01:00
if (hc->GetState() == StateOK)
2013-02-19 23:02:08 +01:00
continue;
2013-02-19 12:17:31 +01:00
2013-02-19 23:02:08 +01:00
return false;
}
2013-02-19 12:17:31 +01:00
2013-02-19 23:02:08 +01:00
return true;
}
2013-03-02 09:07:47 +01:00
void Host::UpdateSlaveServices(void)
{
ASSERT(!OwnsLock());
2013-10-26 09:41:45 +02:00
Dictionary::Ptr service_descriptions = GetServiceDescriptions();
if (!service_descriptions ||service_descriptions->GetLength() == 0)
return;
ConfigItem::Ptr item = ConfigItem::GetObject("Host", GetName());
2013-10-26 09:41:45 +02:00
ObjectLock olock(service_descriptions);
BOOST_FOREACH(const Dictionary::Pair& kv, service_descriptions) {
std::ostringstream namebuf;
namebuf << GetName() << "!" << kv.first;
String name = namebuf.str();
2013-09-24 13:13:14 +02:00
std::vector<String> path;
path.push_back("services");
path.push_back(kv.first);
2013-09-24 13:13:14 +02:00
ExpressionList::Ptr exprl;
{
ObjectLock ilock(item);
exprl = item->GetLinkedExpressionList();
}
2013-09-24 13:13:14 +02:00
DebugInfo di;
exprl->FindDebugInfoPath(path, di);
2013-09-24 13:13:14 +02:00
if (di.Path.IsEmpty())
di = item->GetDebugInfo();
ConfigItemBuilder::Ptr builder = make_shared<ConfigItemBuilder>(di);
builder->SetType("Service");
builder->SetName(name);
2013-09-25 09:19:25 +02:00
builder->AddExpression("host", OperatorSet, GetName());
builder->AddExpression("display_name", OperatorSet, kv.first);
builder->AddExpression("short_name", OperatorSet, kv.first);
if (!kv.second.IsObjectType<Dictionary>())
BOOST_THROW_EXCEPTION(std::invalid_argument("Service description must be either a string or a dictionary."));
Dictionary::Ptr service = kv.second;
Array::Ptr templates = service->Get("templates");
2013-03-01 12:07:52 +01:00
if (templates) {
ObjectLock olock(templates);
BOOST_FOREACH(const Value& tmpl, templates) {
builder->AddParent(tmpl);
}
}
2013-03-01 12:07:52 +01:00
/* Clone attributes from the service expression list. */
ExpressionList::Ptr svc_exprl = make_shared<ExpressionList>();
exprl->ExtractPath(path, svc_exprl);
2013-09-24 13:13:14 +02:00
builder->AddExpressionList(svc_exprl);
ConfigItem::Ptr serviceItem = builder->Compile();
serviceItem->Register();
DynamicObject::Ptr dobj = serviceItem->Commit();
2013-08-29 19:25:34 +02:00
dobj->OnConfigLoaded();
}
}
2013-03-16 21:18:53 +01:00
std::set<Service::Ptr> Host::GetServices(void) const
{
boost::mutex::scoped_lock lock(m_ServicesMutex);
std::set<Service::Ptr> services;
typedef std::pair<String, Service::Ptr> ServicePair;
BOOST_FOREACH(const ServicePair& kv, m_Services) {
services.insert(kv.second);
}
return services;
}
void Host::AddService(const Service::Ptr& service)
{
boost::mutex::scoped_lock lock(m_ServicesMutex);
m_Services[service->GetShortName()] = service;
}
2013-07-09 16:59:31 +02:00
void Host::RemoveService(const Service::Ptr& service)
2013-02-27 15:23:25 +01:00
{
boost::mutex::scoped_lock lock(m_ServicesMutex);
2013-02-27 16:04:49 +01:00
m_Services.erase(service->GetShortName());
2013-02-27 15:23:25 +01:00
}
int Host::GetTotalServices(void) const
{
return GetServices().size();
}
2013-03-02 09:07:47 +01:00
Service::Ptr Host::GetServiceByShortName(const Value& name) const
2013-02-07 20:29:35 +01:00
{
if (name.IsScalar()) {
{
boost::mutex::scoped_lock lock(m_ServicesMutex);
2013-02-26 10:13:54 +01:00
std::map<String, Service::Ptr>::const_iterator it = m_Services.find(name);
if (it != m_Services.end())
return it->second;
}
2013-02-27 15:23:25 +01:00
return Service::Ptr();
} else if (name.IsObjectType<Dictionary>()) {
Dictionary::Ptr dict = name;
2013-02-27 15:23:25 +01:00
String short_name;
2013-03-02 09:07:47 +01:00
return Service::GetByNamePair(dict->Get("host"), dict->Get("service"));
} else {
BOOST_THROW_EXCEPTION(std::invalid_argument("Host/Service name pair is invalid: " + JsonSerialize(name)));
}
2013-02-07 20:29:35 +01:00
}
2013-03-16 21:18:53 +01:00
std::set<Host::Ptr> Host::GetParentHosts(void) const
2013-02-07 20:29:35 +01:00
{
2013-03-16 21:18:53 +01:00
std::set<Host::Ptr> parents;
2013-02-07 20:29:35 +01:00
2013-03-14 12:17:46 +01:00
Array::Ptr dependencies = GetHostDependencies();
2013-02-07 20:29:35 +01:00
if (dependencies) {
2013-03-02 09:07:47 +01:00
ObjectLock olock(dependencies);
2013-03-14 12:17:46 +01:00
BOOST_FOREACH(const Value& value, dependencies) {
2013-03-02 09:07:47 +01:00
if (value == GetName())
2013-02-07 20:29:35 +01:00
continue;
2013-02-27 15:23:25 +01:00
Host::Ptr host = GetByName(value);
parents.insert(host);
2013-02-07 20:29:35 +01:00
}
}
return parents;
}
2013-07-19 16:11:17 +02:00
std::set<Host::Ptr> Host::GetChildHosts(void) const
{
2013-10-26 09:41:45 +02:00
std::set<Host::Ptr> children;
2013-07-19 16:11:17 +02:00
BOOST_FOREACH(const Host::Ptr& host, DynamicType::GetObjects<Host>()) {
2013-07-19 16:11:17 +02:00
Array::Ptr dependencies = host->GetHostDependencies();
if (dependencies) {
ObjectLock olock(dependencies);
BOOST_FOREACH(const Value& value, dependencies) {
if (value == GetName())
2013-10-26 09:41:45 +02:00
children.insert(host);
2013-07-19 16:11:17 +02:00
}
}
}
2013-10-26 09:41:45 +02:00
return children;
2013-07-19 16:11:17 +02:00
}
Service::Ptr Host::GetCheckService(void) const
2013-02-07 20:29:35 +01:00
{
String host_check = GetCheck();
2013-02-27 15:23:25 +01:00
if (host_check.IsEmpty())
return Service::Ptr();
2013-03-02 09:07:47 +01:00
return GetServiceByShortName(host_check);
2013-02-07 20:29:35 +01:00
}
2013-03-16 21:18:53 +01:00
std::set<Service::Ptr> Host::GetParentServices(void) const
2013-02-07 20:29:35 +01:00
{
2013-03-16 21:18:53 +01:00
std::set<Service::Ptr> parents;
2013-02-07 20:29:35 +01:00
2013-03-14 12:17:46 +01:00
Array::Ptr dependencies = GetServiceDependencies();
2013-02-07 20:29:35 +01:00
if (dependencies) {
2013-03-02 09:07:47 +01:00
ObjectLock olock(dependencies);
2013-03-14 12:17:46 +01:00
BOOST_FOREACH(const Value& value, dependencies) {
2013-03-02 09:07:47 +01:00
parents.insert(GetServiceByShortName(value));
2013-02-07 20:29:35 +01:00
}
}
return parents;
}
2013-03-19 13:04:30 +01:00
HostState Host::CalculateState(ServiceState state, bool reachable)
2013-03-07 12:04:20 +01:00
{
2013-03-19 13:04:30 +01:00
if (!reachable)
2013-03-07 12:04:20 +01:00
return HostUnreachable;
2013-03-19 13:04:30 +01:00
switch (state) {
2013-03-07 12:04:20 +01:00
case StateOK:
case StateWarning:
return HostUp;
default:
return HostDown;
}
}
HostState Host::GetState(void) const
{
ASSERT(!OwnsLock());
if (!IsReachable())
return HostUnreachable;
Service::Ptr hc = GetCheckService();
if (!hc)
return HostUp;
switch (hc->GetState()) {
case StateOK:
case StateWarning:
return HostUp;
default:
return HostDown;
}
}
2013-03-07 12:04:20 +01:00
HostState Host::GetLastState(void) const
{
ASSERT(!OwnsLock());
2013-03-07 12:04:20 +01:00
if (!IsReachable())
return HostUnreachable;
Service::Ptr hc = GetCheckService();
2013-03-07 12:04:20 +01:00
if (!hc)
return HostUp;
switch (hc->GetLastState()) {
case StateOK:
case StateWarning:
return HostUp;
default:
return HostDown;
}
}
HostState Host::GetLastHardState(void) const
{
ASSERT(!OwnsLock());
if (!IsReachable())
return HostUnreachable;
Service::Ptr hc = GetCheckService();
if (!hc)
return HostUp;
switch (hc->GetLastHardState()) {
case StateOK:
case StateWarning:
return HostUp;
default:
return HostDown;
}
}
double Host::GetLastStateUp(void) const
{
ASSERT(!OwnsLock());
Service::Ptr hc = GetCheckService();
if (!hc)
return 0;
if (hc->GetLastStateOK() > hc->GetLastStateWarning())
return hc->GetLastStateOK();
else
return hc->GetLastStateWarning();
}
double Host::GetLastStateDown(void) const
{
ASSERT(!OwnsLock());
Service::Ptr hc = GetCheckService();
if (!hc)
return 0;
return hc->GetLastStateCritical();
}
double Host::GetLastStateUnreachable(void) const
{
ASSERT(!OwnsLock());
Service::Ptr hc = GetCheckService();
if (!hc)
return 0;
return hc->GetLastStateUnreachable();
}
double Host::GetLastStateChange(void) const
{
Service::Ptr hc = GetCheckService();
if (!hc)
2013-10-26 09:41:45 +02:00
return Application::GetStartTime();
return hc->GetLastStateChange();
}
double Host::GetLastHardStateChange(void) const
{
Service::Ptr hc = GetCheckService();
if (!hc)
2013-10-26 09:41:45 +02:00
return Application::GetStartTime();
return hc->GetLastHardStateChange();
}
2013-03-07 12:04:20 +01:00
StateType Host::GetLastStateType(void) const
{
Service::Ptr hc = GetCheckService();
2013-03-07 12:04:20 +01:00
if (!hc)
return StateTypeHard;
return hc->GetLastStateType();
}
StateType Host::GetStateType(void) const
{
Service::Ptr hc = GetCheckService();
if (!hc)
return StateTypeHard;
return hc->GetStateType();
}
2013-10-29 13:44:43 +01:00
HostState Host::StateFromString(const String& state)
{
if (state == "UP")
return HostUp;
else if (state == "DOWN")
return HostDown;
else if (state == "UNREACHABLE")
return HostUnreachable;
else
return HostUnreachable;
}
2013-03-19 13:04:30 +01:00
String Host::StateToString(HostState state)
2013-03-07 12:04:20 +01:00
{
switch (state) {
case HostUp:
return "UP";
case HostDown:
return "DOWN";
case HostUnreachable:
return "UNREACHABLE";
default:
return "INVALID";
}
}
2013-10-29 13:44:43 +01:00
StateType Host::StateTypeFromString(const String& type)
{
if (type == "SOFT")
return StateTypeSoft;
else
return StateTypeHard;
}
String Host::StateTypeToString(StateType type)
{
if (type == StateTypeSoft)
return "SOFT";
else
return "HARD";
}
bool Host::ResolveMacro(const String& macro, const CheckResult::Ptr&, String *result) const
{
if (macro == "HOSTNAME") {
*result = GetName();
return true;
}
else if (macro == "HOSTDISPLAYNAME" || macro == "HOSTALIAS") {
*result = GetDisplayName();
return true;
}
2013-02-24 01:10:34 +01:00
Service::Ptr hc = GetCheckService();
CheckResult::Ptr hccr;
2013-02-24 01:10:34 +01:00
if (hc) {
2013-03-19 13:04:30 +01:00
ServiceState state = hc->GetState();
bool reachable = IsReachable();
if (macro == "HOSTSTATE") {
2013-08-30 17:02:21 +02:00
HostState hstate = CalculateState(state, reachable);
switch (hstate) {
case HostUnreachable:
*result = "UNREACHABLE";
break;
case HostUp:
*result = "UP";
break;
case HostDown:
*result = "DOWN";
break;
default:
ASSERT(0);
}
return true;
} else if (macro == "HOSTSTATEID") {
*result = Convert::ToString(state);
return true;
} else if (macro == "HOSTSTATETYPE") {
*result = Service::StateTypeToString(hc->GetStateType());
return true;
} else if (macro == "HOSTATTEMPT") {
2013-10-26 09:41:45 +02:00
*result = Convert::ToString(hc->GetCheckAttempt());
return true;
} else if (macro == "MAXHOSTATTEMPT") {
*result = Convert::ToString(hc->GetMaxCheckAttempts());
return true;
} else if (macro == "LASTHOSTSTATE") {
*result = StateToString(GetLastState());
return true;
} else if (macro == "LASTHOSTSTATEID") {
*result = Convert::ToString(GetLastState());
return true;
} else if (macro == "LASTHOSTSTATETYPE") {
*result = Service::StateTypeToString(GetLastStateType());
return true;
} else if (macro == "LASTHOSTSTATECHANGE") {
*result = Convert::ToString((long)hc->GetLastStateChange());
return true;
} else if (macro == "HOSTDURATIONSEC") {
*result = Convert::ToString((long)(Utility::GetTime() - hc->GetLastStateChange()));
return true;
}
2013-03-07 12:04:20 +01:00
hccr = hc->GetLastCheckResult();
}
if (hccr) {
if (macro == "HOSTLATENCY") {
*result = Convert::ToString(Service::CalculateLatency(hccr));
return true;
} else if (macro == "HOSTEXECUTIONTIME") {
*result = Convert::ToString(Service::CalculateExecutionTime(hccr));
return true;
} else if (macro == "HOSTOUTPUT") {
*result = hccr->GetOutput();
return true;
} else if (macro == "HOSTPERFDATA") {
*result = PluginUtility::FormatPerfdata(hccr->GetPerformanceData());
return true;
} else if (macro == "LASTHOSTCHECK") {
*result = Convert::ToString((long)hccr->GetScheduleStart());
return true;
}
}
2013-02-24 01:10:34 +01:00
if (macro.SubStr(0, 5) == "_HOST") {
Dictionary::Ptr custom = GetCustom();
*result = custom ? custom->Get(macro.SubStr(5)) : "";
return true;
}
Dictionary::Ptr macros = GetMacros();
2013-03-07 12:04:20 +01:00
String name = macro;
if (name == "HOSTADDRESS")
name = "address";
else if (macro == "HOSTADDRESS6")
name = "address6";
if (macros && macros->Contains(name)) {
*result = macros->Get(name);
return true;
}
if (macro == "HOSTADDRESS" || macro == "HOSTADDRESS6") {
*result = GetName();
return true;
2013-02-24 01:10:34 +01:00
}
return false;
}