/******************************************************************************
* Icinga 2 *
- * Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/) *
+ * Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
#include "icinga/service.h"
#include "icinga/hostgroup.h"
#include "icinga/icingaapplication.h"
+#include "icinga/pluginutility.h"
#include "base/dynamictype.h"
#include "base/objectlock.h"
#include "base/logger_fwd.h"
#include "base/timer.h"
#include "base/convert.h"
+#include "base/utility.h"
#include "base/scriptfunction.h"
#include "base/debug.h"
+#include "base/serializer.h"
#include "config/configitembuilder.h"
#include "config/configcompilercontext.h"
-#include <boost/tuple/tuple.hpp>
-#include <boost/smart_ptr/make_shared.hpp>
#include <boost/foreach.hpp>
using namespace icinga;
REGISTER_TYPE(Host);
-void Host::Start(void)
+void Host::OnConfigLoaded(void)
{
- DynamicObject::Start();
+ DynamicObject::OnConfigLoaded();
ASSERT(!OwnsLock());
}
}
-void Host::OnConfigLoaded(void)
-{
- UpdateSlaveServices();
-}
-
void Host::Stop(void)
{
DynamicObject::Stop();
// TODO: unregister slave services/notifications?
}
-String Host::GetDisplayName(void) const
-{
- if (!m_DisplayName.IsEmpty())
- return m_DisplayName;
- else
- return GetName();
-}
-
-Array::Ptr Host::GetGroups(void) const
-{
- return m_HostGroups;
-}
-
-Dictionary::Ptr Host::GetMacros(void) const
-{
- return m_Macros;
-}
-
-Array::Ptr Host::GetHostDependencies(void) const
-{
- return m_HostDependencies;;
-}
-
-Array::Ptr Host::GetServiceDependencies(void) const
-{
- return m_ServiceDependencies;
-}
-
-String Host::GetCheck(void) const
-{
- return m_Check;
-}
-
-Dictionary::Ptr Host::GetNotificationDescriptions(void) const
-{
- return m_NotificationDescriptions;
-}
-
-bool Host::IsReachable(void) const
-{
- ASSERT(!OwnsLock());
-
- std::set<Service::Ptr> parentServices = GetParentServices();
-
- BOOST_FOREACH(const Service::Ptr& service, parentServices) {
- ObjectLock olock(service);
-
- /* ignore pending services */
- if (!service->GetLastCheckResult())
- continue;
-
- /* ignore soft states */
- if (service->GetStateType() == StateTypeSoft)
- continue;
-
- /* ignore services states OK and Warning */
- if (service->GetState() == StateOK ||
- service->GetState() == StateWarning)
- continue;
-
- return false;
- }
-
- std::set<Host::Ptr> parentHosts = GetParentHosts();
-
- BOOST_FOREACH(const Host::Ptr& host, parentHosts) {
- Service::Ptr hc = host->GetCheckService();
-
- /* ignore hosts that don't have a check */
- if (!hc)
- continue;
-
- ObjectLock olock(hc);
-
- /* ignore soft states */
- if (hc->GetStateType() == StateTypeSoft)
- continue;
-
- /* ignore hosts that are up */
- if (hc->GetState() == StateOK)
- continue;
-
- return false;
- }
-
- return true;
-}
-
-void Host::UpdateSlaveServices(void)
-{
- ASSERT(!OwnsLock());
-
- ConfigItem::Ptr item = ConfigItem::GetObject("Host", GetName());
-
- /* Don't create slave services unless we own this object */
- if (!item)
- return;
-
- if (!m_ServiceDescriptions)
- return;
-
- ObjectLock olock(m_ServiceDescriptions);
- String svcname;
- Value svcdesc;
- BOOST_FOREACH(boost::tie(svcname, svcdesc), m_ServiceDescriptions) {
- if (svcdesc.IsScalar())
- svcname = svcdesc;
-
- std::ostringstream namebuf;
- namebuf << GetName() << ":" << svcname;
- String name = namebuf.str();
-
- std::vector<String> path;
- path.push_back("services");
- path.push_back(svcname);
-
- DebugInfo di;
- item->GetLinkedExpressionList()->FindDebugInfoPath(path, di);
-
- if (di.Path.IsEmpty())
- di = item->GetDebugInfo();
-
- ConfigItemBuilder::Ptr builder = boost::make_shared<ConfigItemBuilder>(di);
- builder->SetType("Service");
- builder->SetName(name);
- builder->AddExpression("host_name", OperatorSet, GetName());
- builder->AddExpression("display_name", OperatorSet, svcname);
- builder->AddExpression("short_name", OperatorSet, svcname);
-
- if (!svcdesc.IsObjectType<Dictionary>())
- BOOST_THROW_EXCEPTION(std::invalid_argument("Service description must be either a string or a dictionary."));
-
- Dictionary::Ptr service = svcdesc;
-
- Array::Ptr templates = service->Get("templates");
-
- if (templates) {
- ObjectLock olock(templates);
-
- BOOST_FOREACH(const Value& tmpl, templates) {
- builder->AddParent(tmpl);
- }
- }
-
- /* Clone attributes from the service expression list. */
- ExpressionList::Ptr svc_exprl = boost::make_shared<ExpressionList>();
- item->GetLinkedExpressionList()->ExtractPath(path, svc_exprl);
-
- std::vector<String> dpath;
- dpath.push_back("templates");
- svc_exprl->ErasePath(dpath);
-
- builder->AddExpressionList(svc_exprl);
-
- ConfigItem::Ptr serviceItem = builder->Compile();
- serviceItem->Register();
- DynamicObject::Ptr dobj = serviceItem->Commit();
- dobj->OnConfigLoaded();
- }
-}
-
std::set<Service::Ptr> Host::GetServices(void) const
{
boost::mutex::scoped_lock lock(m_ServicesMutex);
std::set<Service::Ptr> services;
- Service::WeakPtr wservice;
- BOOST_FOREACH(boost::tie(boost::tuples::ignore, wservice), m_Services) {
- Service::Ptr service = wservice.lock();
-
- if (!service)
- continue;
-
- services.insert(service);
+ typedef std::pair<String, Service::Ptr> ServicePair;
+ BOOST_FOREACH(const ServicePair& kv, m_Services) {
+ services.insert(kv.second);
}
return services;
return GetServices().size();
}
-Service::Ptr Host::GetServiceByShortName(const Value& name) const
+Service::Ptr Host::GetServiceByShortName(const Value& name)
{
if (name.IsScalar()) {
{
return Service::GetByNamePair(dict->Get("host"), dict->Get("service"));
} else {
- BOOST_THROW_EXCEPTION(std::invalid_argument("Host/Service name pair is invalid: " + name.Serialize()));
- }
-}
-
-std::set<Host::Ptr> Host::GetParentHosts(void) const
-{
- std::set<Host::Ptr> parents;
-
- Array::Ptr dependencies = GetHostDependencies();
-
- if (dependencies) {
- ObjectLock olock(dependencies);
-
- BOOST_FOREACH(const Value& value, dependencies) {
- if (value == GetName())
- continue;
-
- Host::Ptr host = GetByName(value);
-
- if (!host)
- continue;
-
- parents.insert(host);
- }
+ BOOST_THROW_EXCEPTION(std::invalid_argument("Host/Service name pair is invalid: " + JsonSerialize(name)));
}
-
- return parents;
}
-std::set<Host::Ptr> Host::GetChildHosts(void) const
+HostState Host::CalculateState(ServiceState state)
{
- std::set<Host::Ptr> childs;
-
- BOOST_FOREACH(const Host::Ptr& host, DynamicType::GetObjects<Host>()) {
- Array::Ptr dependencies = host->GetHostDependencies();
-
- if (dependencies) {
- ObjectLock olock(dependencies);
-
- BOOST_FOREACH(const Value& value, dependencies) {
- if (value == GetName())
- childs.insert(host);
- }
- }
- }
-
- return childs;
-
-}
-
-Service::Ptr Host::GetCheckService(void) const
-{
- String host_check = GetCheck();
-
- if (host_check.IsEmpty())
- return Service::Ptr();
-
- return GetServiceByShortName(host_check);
-}
-
-std::set<Service::Ptr> Host::GetParentServices(void) const
-{
- std::set<Service::Ptr> parents;
-
- Array::Ptr dependencies = GetServiceDependencies();
-
- if (dependencies) {
- ObjectLock olock(dependencies);
-
- BOOST_FOREACH(const Value& value, dependencies) {
- parents.insert(GetServiceByShortName(value));
- }
- }
-
- return parents;
-}
-
-HostState Host::CalculateState(ServiceState state, bool reachable)
-{
- if (!reachable)
- return HostUnreachable;
-
switch (state) {
case StateOK:
case StateWarning:
{
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;
- }
-
+ return CalculateState(GetStateRaw());
}
HostState Host::GetLastState(void) const
{
ASSERT(!OwnsLock());
- if (!IsReachable())
- return HostUnreachable;
-
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return HostUp;
-
- switch (hc->GetLastState()) {
- case StateOK:
- case StateWarning:
- return HostUp;
- default:
- return HostDown;
- }
+ return CalculateState(GetLastStateRaw());
}
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;
- }
+ return CalculateState(GetLastHardStateRaw());
}
double Host::GetLastStateUp(void) const
{
ASSERT(!OwnsLock());
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return 0;
-
- if (hc->GetLastStateOK() > hc->GetLastStateWarning())
- return hc->GetLastStateOK();
+ if (GetLastStateOK() > GetLastStateWarning())
+ return GetLastStateOK();
else
- return hc->GetLastStateWarning();
+ return GetLastStateWarning();
}
double Host::GetLastStateDown(void) const
{
ASSERT(!OwnsLock());
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return 0;
-
- return hc->GetLastStateCritical();
+ return GetLastStateCritical();
}
-double Host::GetLastStateUnreachable(void) const
+HostState Host::StateFromString(const String& state)
{
- ASSERT(!OwnsLock());
-
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return 0;
-
- return hc->GetLastStateUnreachable();
+ if (state == "UP")
+ return HostUp;
+ else
+ return HostDown;
}
-double Host::GetLastStateChange(void) const
+String Host::StateToString(HostState state)
{
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return IcingaApplication::GetInstance()->GetStartTime();
-
- return hc->GetLastStateChange();
+ switch (state) {
+ case HostUp:
+ return "UP";
+ case HostDown:
+ return "DOWN";
+ default:
+ return "INVALID";
+ }
}
-
-double Host::GetLastHardStateChange(void) const
+StateType Host::StateTypeFromString(const String& type)
{
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return IcingaApplication::GetInstance()->GetStartTime();
-
- return hc->GetLastHardStateChange();
+ if (type == "SOFT")
+ return StateTypeSoft;
+ else
+ return StateTypeHard;
}
-StateType Host::GetLastStateType(void) const
+String Host::StateTypeToString(StateType type)
{
- Service::Ptr hc = GetCheckService();
-
- if (!hc)
- return StateTypeHard;
-
- return hc->GetLastStateType();
+ if (type == StateTypeSoft)
+ return "SOFT";
+ else
+ return "HARD";
}
-StateType Host::GetStateType(void) const
+bool Host::ResolveMacro(const String& macro, const CheckResult::Ptr&, String *result) const
{
- Service::Ptr hc = GetCheckService();
+ String key;
+ Dictionary::Ptr vars;
- if (!hc)
- return StateTypeHard;
+ /* special treatment for address macros providing name fallback */
+ if (macro == "address" || macro == "address6") {
+ vars = GetVars();
- return hc->GetStateType();
-}
+ String value;
+ if (vars && vars->Contains(macro))
+ value = vars->Get(macro);
-String Host::StateToString(HostState state)
-{
- switch (state) {
- case HostUp:
- return "UP";
- case HostDown:
- return "DOWN";
- case HostUnreachable:
- return "UNREACHABLE";
- default:
- return "INVALID";
+ if (value.IsEmpty()) {
+ *result = GetName();
+ return true;
+ } else {
+ *result = value;
+ return true;
+ }
}
-}
+ else if (macro == "host.vars.address" || macro == "host.vars.address6") {
+ key = macro.SubStr(10);
+ vars = GetVars();
-bool Host::ResolveMacro(const String& macro, const Dictionary::Ptr&, String *result) const
-{
- if (macro == "HOSTNAME") {
- *result = GetName();
- return true;
- }
- else if (macro == "HOSTDISPLAYNAME" || macro == "HOSTALIAS") {
- *result = GetDisplayName();
- return true;
+ String value;
+ if (vars && vars->Contains(key))
+ value = vars->Get(key);
+
+ if (value.IsEmpty()) {
+ *result = GetName();
+ return true;
+ } else {
+ *result = value;
+ return true;
+ }
}
- Service::Ptr hc = GetCheckService();
- Dictionary::Ptr hccr;
-
- if (hc) {
- ServiceState state = hc->GetState();
- bool reachable = IsReachable();
-
- if (macro == "HOSTSTATE") {
- 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);
+ /* require prefix for object macros */
+ if (macro.SubStr(0, 5) == "host.") {
+ key = macro.SubStr(5);
+
+ if (key.SubStr(0, 5) == "vars.") {
+ vars = GetVars();
+ String vars_key = key.SubStr(5);
+
+ if (vars && vars->Contains(vars_key)) {
+ *result = vars->Get(vars_key);
+ return true;
}
+ }
+ else if (key == "name") {
+ *result = GetName();
+ return true;
+ }
+ else if (key == "displaymane") {
+ *result = GetDisplayName();
+ return true;
+ }
+
+ CheckResult::Ptr cr = GetLastCheckResult();
+ if (key == "state") {
+ *result = StateToString(GetState());
return true;
- } else if (macro == "HOSTSTATEID") {
- *result = Convert::ToString(state);
+ } else if (key == "stateid") {
+ *result = Convert::ToString(GetState());
return true;
- } else if (macro == "HOSTSTATETYPE") {
- *result = Service::StateTypeToString(hc->GetStateType());
+ } else if (key == "statetype") {
+ *result = StateTypeToString(GetStateType());
return true;
- } else if (macro == "HOSTATTEMPT") {
- *result = Convert::ToString(hc->GetCurrentCheckAttempt());
+ } else if (key == "attempt") {
+ *result = Convert::ToString(GetCheckAttempt());
return true;
- } else if (macro == "MAXHOSTATTEMPT") {
- *result = Convert::ToString(hc->GetMaxCheckAttempts());
+ } else if (key == "maxattempt") {
+ *result = Convert::ToString(GetMaxCheckAttempts());
return true;
- } else if (macro == "LASTHOSTSTATE") {
+ } else if (key == "laststate") {
*result = StateToString(GetLastState());
return true;
- } else if (macro == "LASTHOSTSTATEID") {
+ } else if (key == "laststateid") {
*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;
- }
-
- 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));
+ } else if (key == "laststatetype") {
+ *result = StateTypeToString(GetLastStateType());
return true;
- } else if (macro == "HOSTOUTPUT") {
- *result = hccr->Get("output");
+ } else if (key == "laststatechange") {
+ *result = Convert::ToString((long)GetLastStateChange());
return true;
- } else if (macro == "HOSTPERFDATA") {
- *result = hccr->Get("performance_data_raw");
+ } else if (key == "durationsec") {
+ *result = Convert::ToString((long)(Utility::GetTime() - GetLastStateChange()));
return true;
- } else if (macro == "LASTHOSTCHECK") {
- *result = Convert::ToString((long)hccr->Get("schedule_start"));
- return true;
- }
- }
+ } else if (key == "checkcommand") {
+ CheckCommand::Ptr commandObj = GetCheckCommand();
- Dictionary::Ptr macros = GetMacros();
+ if (commandObj)
+ *result = commandObj->GetName();
+ else
+ *result = "";
- String name = macro;
+ return true;
+ } else if (key == "totalservices" || key == "totalservicesok" || key == "totalserviceswarning"
+ || key == "totalservicesunknown" || key == "totalservicescritical") {
+ int filter = -1;
+ int count = 0;
+
+ if (key == "totalservicesok")
+ filter = StateOK;
+ else if (key == "totalserviceswarning")
+ filter = StateWarning;
+ else if (key == "totalservicesunknown")
+ filter = StateUnknown;
+ else if (key == "totalservicescritical")
+ filter = StateCritical;
+
+ BOOST_FOREACH(const Service::Ptr& service, GetServices()) {
+ if (filter != -1 && service->GetState() != filter)
+ continue;
+
+ count++;
+ }
+
+ *result = Convert::ToString(count);
+ return true;
+ }
- if (name == "HOSTADDRESS")
- name = "address";
- else if (macro == "HOSTADDRESS6")
- name = "address6";
- if (macros && macros->Contains(name)) {
- *result = macros->Get(name);
- return true;
- }
+ if (cr) {
+ if (key == "latency") {
+ *result = Convert::ToString(Service::CalculateLatency(cr));
+ return true;
+ } else if (key == "executiontime") {
+ *result = Convert::ToString(Service::CalculateExecutionTime(cr));
+ return true;
+ } else if (key == "output") {
+ *result = cr->GetOutput();
+ return true;
+ } else if (key == "perfdata") {
+ *result = PluginUtility::FormatPerfdata(cr->GetPerformanceData());
+ return true;
+ } else if (key == "lastcheck") {
+ *result = Convert::ToString((long)cr->GetScheduleStart());
+ return true;
+ }
+ }
+ } else {
+ vars = GetVars();
- if (macro == "HOSTADDRESS" || macro == "HOSTADDRESS6") {
- *result = GetName();
- return true;
+ if (vars && vars->Contains(macro)) {
+ *result = vars->Get(macro);
+ return true;
+ }
}
return false;
}
-
-void Host::InternalSerialize(const Dictionary::Ptr& bag, int attributeTypes) const
-{
- DynamicObject::InternalSerialize(bag, attributeTypes);
-
- if (attributeTypes & Attribute_Config) {
- bag->Set("display_name", m_DisplayName);
- bag->Set("hostgroups", m_HostGroups);
- bag->Set("macros", m_Macros);
- bag->Set("hostdependencies", m_HostDependencies);
- bag->Set("servicedependencies", m_ServiceDependencies);
- bag->Set("check", m_Check);
- bag->Set("services", m_ServiceDescriptions);
- bag->Set("notifications", m_NotificationDescriptions);
- }
-}
-
-void Host::InternalDeserialize(const Dictionary::Ptr& bag, int attributeTypes)
-{
- DynamicObject::InternalDeserialize(bag, attributeTypes);
-
- if (attributeTypes & Attribute_Config) {
- m_DisplayName = bag->Get("display_name");
- m_HostGroups = bag->Get("hostgroups");
- m_Macros = bag->Get("macros");
- m_HostDependencies = bag->Get("hostdependencies");
- m_ServiceDependencies = bag->Get("servicedependencies");
- m_Check = bag->Get("check");
- m_ServiceDescriptions = bag->Get("services");
- m_NotificationDescriptions = bag->Get("notifications");
- }
-}