/******************************************************************************
* Icinga 2 *
- * Copyright (C) 2012 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/host.h"
#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/scriptfunction.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;
-static boost::mutex l_ServiceMutex;
-static std::map<String, std::map<String, Service::WeakPtr> > l_ServicesCache;
-static bool l_ServicesCacheNeedsUpdate = false;
-static Timer::Ptr l_ServicesCacheTimer;
-
-REGISTER_SCRIPTFUNCTION(ValidateServiceDictionary, &Host::ValidateServiceDictionary);
-
REGISTER_TYPE(Host);
-Host::Host(const Dictionary::Ptr& serializedUpdate)
- : DynamicObject(serializedUpdate)
-{
- RegisterAttribute("display_name", Attribute_Config, &m_DisplayName);
- RegisterAttribute("hostgroups", Attribute_Config, &m_HostGroups);
- RegisterAttribute("macros", Attribute_Config, &m_Macros);
- RegisterAttribute("hostdependencies", Attribute_Config, &m_HostDependencies);
- RegisterAttribute("servicedependencies", Attribute_Config, &m_ServiceDependencies);
- RegisterAttribute("hostcheck", Attribute_Config, &m_HostCheck);
-
-}
-
-Host::~Host(void)
-{
- HostGroup::InvalidateMembersCache();
-
- if (m_SlaveServices) {
- ConfigItem::Ptr service;
- BOOST_FOREACH(boost::tie(boost::tuples::ignore, service), m_SlaveServices) {
- service->Unregister();
- }
- }
-}
-
-void Host::OnRegistrationCompleted(void)
-{
- ASSERT(!OwnsLock());
-
- DynamicObject::OnRegistrationCompleted();
-
- Host::InvalidateServicesCache();
- UpdateSlaveServices();
-}
-
-String Host::GetDisplayName(void) const
-{
- if (!m_DisplayName.IsEmpty())
- return m_DisplayName;
- else
- return GetName();
-}
-
-Host::Ptr Host::GetByName(const String& name)
-{
- DynamicObject::Ptr configObject = DynamicObject::GetObject("Host", name);
-
- return dynamic_pointer_cast<Host>(configObject);
-}
-
-Array::Ptr Host::GetGroups(void) const
-{
- return m_HostGroups;;
-}
-
-Dictionary::Ptr Host::GetMacros(void) const
-{
- return m_Macros;
-}
-
-Array::Ptr Host::GetHostDependencies(void) const
+void Host::OnConfigLoaded(void)
{
- return m_HostDependencies;;
-}
-
-Array::Ptr Host::GetServiceDependencies(void) const
-{
- return m_ServiceDependencies;
-}
-
-String Host::GetHostCheck(void) const
-{
- return m_HostCheck;
-}
+ DynamicObject::OnConfigLoaded();
-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;
- }
+ Array::Ptr groups = GetGroups();
- std::set<Host::Ptr> parentHosts = GetParentHosts();
+ if (groups) {
+ ObjectLock olock(groups);
- BOOST_FOREACH(const Host::Ptr& host, parentHosts) {
- Service::Ptr hc = host->GetHostCheckService();
+ BOOST_FOREACH(const String& name, groups) {
+ HostGroup::Ptr hg = HostGroup::GetByName(name);
- /* ignore hosts that don't have a hostcheck */
- 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;
+ if (hg)
+ hg->AddMember(GetSelf());
+ }
}
-
- return true;
}
-void Host::UpdateSlaveServices(void)
+void Host::Stop(void)
{
- ASSERT(!OwnsLock());
-
- ConfigItem::Ptr item = ConfigItem::GetObject("Host", GetName());
-
- /* Don't create slave services unless we own this object */
- if (!item)
- return;
-
- Dictionary::Ptr oldServices = m_SlaveServices;
- Dictionary::Ptr serviceDescs = Get("services");
-
- Dictionary::Ptr newServices = boost::make_shared<Dictionary>();
-
- if (serviceDescs) {
- ObjectLock olock(serviceDescs);
- String svcname;
- Value svcdesc;
- BOOST_FOREACH(boost::tie(svcname, svcdesc), serviceDescs) {
- if (svcdesc.IsScalar())
- svcname = svcdesc;
-
- std::ostringstream namebuf;
- namebuf << GetName() << ":" << svcname;
- String name = namebuf.str();
+ DynamicObject::Stop();
- ConfigItemBuilder::Ptr builder = boost::make_shared<ConfigItemBuilder>(item->GetDebugInfo());
- builder->SetType("Service");
- builder->SetName(name);
- builder->AddExpression("host_name", OperatorSet, GetName());
- builder->AddExpression("display_name", OperatorSet, svcname);
- builder->AddExpression("short_name", OperatorSet, svcname);
+ Array::Ptr groups = GetGroups();
- if (!svcdesc.IsObjectType<Dictionary>())
- BOOST_THROW_EXCEPTION(std::invalid_argument("Service description must be either a string or a dictionary."));
+ if (groups) {
+ ObjectLock olock(groups);
- Dictionary::Ptr service = svcdesc;
+ BOOST_FOREACH(const String& name, groups) {
+ HostGroup::Ptr hg = HostGroup::GetByName(name);
- Array::Ptr templates = service->Get("templates");
-
- if (templates) {
- ObjectLock olock(templates);
-
- BOOST_FOREACH(const Value& tmpl, templates) {
- builder->AddParent(tmpl);
- }
- }
-
- /* Clone attributes from the host object. */
- std::set<String, string_iless> keys;
- keys.insert("check_interval");
- keys.insert("retry_interval");
- keys.insert("servicegroups");
- keys.insert("checkers");
- keys.insert("notification_interval");
- keys.insert("notification_type_filter");
- keys.insert("notification_state_filter");
- keys.insert("check_period");
- keys.insert("servicedependencies");
- keys.insert("hostdependencies");
- keys.insert("export_macros");
-
- ExpressionList::Ptr host_exprl = boost::make_shared<ExpressionList>();
- item->GetLinkedExpressionList()->ExtractFiltered(keys, host_exprl);
- builder->AddExpressionList(host_exprl);
-
- /* Clone attributes from the service expression list. */
- std::vector<String> path;
- path.push_back("services");
- path.push_back(svcname);
-
- ExpressionList::Ptr svc_exprl = boost::make_shared<ExpressionList>();
- item->GetLinkedExpressionList()->ExtractPath(path, svc_exprl);
- builder->AddExpressionList(svc_exprl);
-
- ConfigItem::Ptr serviceItem = builder->Compile();
- DynamicObject::Ptr dobj = serviceItem->Commit();
-
- newServices->Set(name, serviceItem);
+ if (hg)
+ hg->RemoveMember(GetSelf());
}
}
- if (oldServices) {
- ObjectLock olock(oldServices);
-
- ConfigItem::Ptr service;
- BOOST_FOREACH(boost::tie(boost::tuples::ignore, service), oldServices) {
- if (!service)
- continue;
-
- if (!newServices->Contains(service->GetName()))
- service->Unregister();
- }
- }
-
- newServices->Seal();
-
- Set("slave_services", newServices);
-}
-
-void Host::OnAttributeChanged(const String& name)
-{
- ASSERT(!OwnsLock());
-
- if (name == "hostgroups")
- HostGroup::InvalidateMembersCache();
- else if (name == "services") {
- UpdateSlaveServices();
- } else if (name == "notifications") {
- BOOST_FOREACH(const Service::Ptr& service, GetServices()) {
- service->UpdateSlaveNotifications();
- }
- }
+ // TODO: unregister slave services/notifications?
}
std::set<Service::Ptr> Host::GetServices(void) const
{
- std::set<Service::Ptr> services;
-
- boost::mutex::scoped_lock lock(l_ServiceMutex);
+ boost::mutex::scoped_lock lock(m_ServicesMutex);
- Service::WeakPtr wservice;
- BOOST_FOREACH(boost::tie(boost::tuples::ignore, wservice), l_ServicesCache[GetName()]) {
- Service::Ptr service = wservice.lock();
-
- if (!service)
- continue;
-
- services.insert(service);
+ 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::InvalidateServicesCache(void)
+void Host::AddService(const Service::Ptr& service)
{
- {
- boost::mutex::scoped_lock lock(l_ServiceMutex);
-
- if (l_ServicesCacheNeedsUpdate)
- return; /* Someone else has already requested a refresh. */
+ boost::mutex::scoped_lock lock(m_ServicesMutex);
- if (!l_ServicesCacheTimer) {
- l_ServicesCacheTimer = boost::make_shared<Timer>();
- l_ServicesCacheTimer->SetInterval(0.5);
- l_ServicesCacheTimer->OnTimerExpired.connect(boost::bind(&Host::RefreshServicesCache));
- l_ServicesCacheTimer->Start();
- }
-
- l_ServicesCacheNeedsUpdate = true;
- }
+ m_Services[service->GetShortName()] = service;
}
-void Host::RefreshServicesCache(void)
+void Host::RemoveService(const Service::Ptr& service)
{
- {
- boost::mutex::scoped_lock lock(l_ServiceMutex);
-
- if (!l_ServicesCacheNeedsUpdate)
- return;
-
- l_ServicesCacheNeedsUpdate = false;
- }
-
- Log(LogDebug, "icinga", "Updating Host services cache.");
-
- std::map<String, std::map<String, Service::WeakPtr> > newServicesCache;
-
- BOOST_FOREACH(const DynamicObject::Ptr& object, DynamicType::GetObjects("Service")) {
- const Service::Ptr& service = static_pointer_cast<Service>(object);
-
- Host::Ptr host = service->GetHost();
+ boost::mutex::scoped_lock lock(m_ServicesMutex);
- if (!host)
- continue;
-
- // TODO: assert for duplicate short_names
-
- newServicesCache[host->GetName()][service->GetShortName()] = service;
- }
-
- boost::mutex::scoped_lock lock(l_ServiceMutex);
- l_ServicesCache.swap(newServicesCache);
+ m_Services.erase(service->GetShortName());
}
-Value Host::ValidateServiceDictionary(const String& location, const Dictionary::Ptr& attrs)
+int Host::GetTotalServices(void) const
{
- ObjectLock olock(attrs);
-
- String key;
- Value value;
- BOOST_FOREACH(boost::tie(key, value), attrs) {
- std::vector<String> templates;
-
- if (!value.IsObjectType<Dictionary>())
- BOOST_THROW_EXCEPTION(std::invalid_argument("Service description must be a dictionary."));
-
- Dictionary::Ptr serviceDesc = value;
-
- Array::Ptr templatesArray = serviceDesc->Get("templates");
-
- if (templatesArray) {
- ObjectLock tlock(templatesArray);
-
- BOOST_FOREACH(const Value& tmpl, templatesArray) {
- templates.push_back(tmpl);
- }
- }
-
- BOOST_FOREACH(const String& name, templates) {
- ConfigItem::Ptr item;
-
- ConfigCompilerContext *context = ConfigCompilerContext::GetContext();
-
- if (context)
- item = context->GetItem("Service", name);
-
- /* ignore already active objects while we're in the compiler
- * context and linking to existing items is disabled. */
- if (!item && (!context || (context->GetFlags() & CompilerLinkExisting)))
- item = ConfigItem::GetObject("Service", name);
-
- if (!item) {
- ConfigCompilerContext::GetContext()->AddError(false, "Validation failed for " +
- location + ": Template '" + name + "' not found.");
- }
- }
- }
-
- return Empty;
+ return GetServices().size();
}
-Service::Ptr Host::GetServiceByShortName(const Value& name) const
+Service::Ptr Host::GetServiceByShortName(const Value& name)
{
if (name.IsScalar()) {
{
- boost::mutex::scoped_lock lock(l_ServiceMutex);
+ boost::mutex::scoped_lock lock(m_ServicesMutex);
- std::map<String, Service::WeakPtr>& services = l_ServicesCache[GetName()];
- std::map<String, Service::WeakPtr>::iterator it = services.find(name);
+ std::map<String, Service::Ptr>::const_iterator it = m_Services.find(name);
- if (it != services.end()) {
- Service::Ptr service = it->second.lock();
- ASSERT(service);
- return service;
- }
+ if (it != m_Services.end())
+ return it->second;
}
return Service::Ptr();
return Service::GetByNamePair(dict->Get("host"), dict->Get("service"));
} else {
- BOOST_THROW_EXCEPTION(std::invalid_argument("Host/Service name pair is invalid: " + name.Serialize()));
+ BOOST_THROW_EXCEPTION(std::invalid_argument("Host/Service name pair is invalid: " + JsonSerialize(name)));
}
}
-std::set<Host::Ptr> Host::GetParentHosts(void) const
+HostState Host::CalculateState(ServiceState state)
{
- 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);
- }
+ switch (state) {
+ case StateOK:
+ case StateWarning:
+ return HostUp;
+ default:
+ return HostDown;
}
-
- return parents;
}
-Service::Ptr Host::GetHostCheckService(void) const
+HostState Host::GetState(void) const
{
- String host_check = GetHostCheck();
-
- if (host_check.IsEmpty())
- return Service::Ptr();
+ ASSERT(!OwnsLock());
- return GetServiceByShortName(host_check);
+ return CalculateState(GetStateRaw());
}
-std::set<Service::Ptr> Host::GetParentServices(void) const
+HostState Host::GetLastState(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));
- }
- }
+ ASSERT(!OwnsLock());
- return parents;
+ return CalculateState(GetLastStateRaw());
}
-HostState Host::CalculateState(ServiceState state, bool reachable)
+HostState Host::GetLastHardState(void) const
{
- if (!reachable)
- return HostUnreachable;
+ ASSERT(!OwnsLock());
- switch (state) {
- case StateOK:
- case StateWarning:
- return HostUp;
- default:
- return HostDown;
- }
+ return CalculateState(GetLastHardStateRaw());
}
-HostState Host::GetLastState(void) const
+double Host::GetLastStateUp(void) const
{
ASSERT(!OwnsLock());
- if (!IsReachable())
- return HostUnreachable;
-
- Service::Ptr hc = GetHostCheckService();
-
- if (!hc)
- return HostUp;
-
- switch (hc->GetLastState()) {
- case StateOK:
- case StateWarning:
- return HostUp;
- default:
- return HostDown;
- }
+ if (GetLastStateOK() > GetLastStateWarning())
+ return GetLastStateOK();
+ else
+ return GetLastStateWarning();
}
-StateType Host::GetLastStateType(void) const
+double Host::GetLastStateDown(void) const
{
- Service::Ptr hc = GetHostCheckService();
+ ASSERT(!OwnsLock());
- if (!hc)
- return StateTypeHard;
+ return GetLastStateCritical();
+}
- return hc->GetLastStateType();
+HostState Host::StateFromString(const String& state)
+{
+ if (state == "UP")
+ return HostUp;
+ else
+ return HostDown;
}
String Host::StateToString(HostState state)
return "UP";
case HostDown:
return "DOWN";
- case HostUnreachable:
- return "UNREACHABLE";
default:
return "INVALID";
}
}
-bool Host::ResolveMacro(const String& macro, const Dictionary::Ptr&, String *result) const
+StateType Host::StateTypeFromString(const String& type)
+{
+ if (type == "SOFT")
+ return StateTypeSoft;
+ else
+ return StateTypeHard;
+}
+
+String Host::StateTypeToString(StateType type)
{
- if (macro == "HOSTNAME") {
- *result = GetName();
- return true;
+ if (type == StateTypeSoft)
+ return "SOFT";
+ else
+ return "HARD";
+}
+
+bool Host::ResolveMacro(const String& macro, const CheckResult::Ptr&, String *result) const
+{
+ String key;
+ Dictionary::Ptr vars;
+
+ /* special treatment for address macros providing name fallback */
+ if (macro == "address" || macro == "address6") {
+ vars = GetVars();
+
+ String value;
+ if (vars && vars->Contains(macro))
+ value = vars->Get(macro);
+
+ if (value.IsEmpty()) {
+ *result = GetName();
+ return true;
+ } else {
+ *result = value;
+ return true;
+ }
}
- else if (macro == "HOSTDISPLAYNAME" || macro == "HOSTALIAS") {
- *result = GetDisplayName();
- return true;
+ else if (macro == "host.vars.address" || macro == "host.vars.address6") {
+ key = macro.SubStr(10);
+ vars = GetVars();
+
+ 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 = GetHostCheckService();
- Dictionary::Ptr hccr;
+ /* require prefix for object macros */
+ if (macro.SubStr(0, 5) == "host.") {
+ key = macro.SubStr(5);
- if (hc) {
- ServiceState state = hc->GetState();
- bool reachable = IsReachable();
+ if (key.SubStr(0, 5) == "vars.") {
+ vars = GetVars();
+ String vars_key = key.SubStr(5);
- if (macro == "HOSTSTATE") {
- *result = Convert::ToString(CalculateState(state, reachable));
+ 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;
- } else if (macro == "HOSTSTATEID") {
- *result = Convert::ToString(state);
+ }
+
+ CheckResult::Ptr cr = GetLastCheckResult();
+
+ if (key == "state") {
+ *result = StateToString(GetState());
return true;
- } else if (macro == "HOSTSTATETYPE") {
- *result = Service::StateTypeToString(hc->GetStateType());
+ } else if (key == "stateid") {
+ *result = Convert::ToString(GetState());
return true;
- } else if (macro == "HOSTATTEMPT") {
- *result = Convert::ToString(hc->GetCurrentCheckAttempt());
+ } else if (key == "statetype") {
+ *result = StateTypeToString(GetStateType());
return true;
- } else if (macro == "MAXHOSTATTEMPT") {
- *result = Convert::ToString(hc->GetMaxCheckAttempts());
+ } else if (key == "attempt") {
+ *result = Convert::ToString(GetCheckAttempt());
return true;
- } else if (macro == "LASTHOSTSTATE") {
+ } else if (key == "maxattempt") {
+ *result = Convert::ToString(GetMaxCheckAttempts());
+ return true;
+ } 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());
+ } else if (key == "laststatetype") {
+ *result = StateTypeToString(GetLastStateType());
return true;
- } else if (macro == "LASTHOSTSTATECHANGE") {
- *result = Convert::ToString((long)hc->GetLastStateChange());
+ } else if (key == "laststatechange") {
+ *result = Convert::ToString((long)GetLastStateChange());
return true;
- }
+ } else if (key == "durationsec") {
+ *result = Convert::ToString((long)(Utility::GetTime() - GetLastStateChange()));
+ return true;
+ } else if (key == "checkcommand") {
+ CheckCommand::Ptr commandObj = GetCheckCommand();
- hccr = hc->GetLastCheckResult();
- }
+ if (commandObj)
+ *result = commandObj->GetName();
+ else
+ *result = "";
- 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->Get("output");
return true;
- } else if (macro == "HOSTPERFDATA") {
- *result = hccr->Get("performance_data_raw");
- return true;
- } else if (macro == "LASTHOSTCHECK") {
- *result = Convert::ToString((long)hccr->Get("schedule_start"));
- return true;
- }
- }
-
- Dictionary::Ptr macros = GetMacros();
+ } 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++;
+ }
- String name = macro;
+ *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;