granicus.if.org Git - icinga2/blob - lib/methods/dummychecktask.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   3 #ifndef _WIN32
   4 #       include <stdlib.h>
   5 #endif /* _WIN32 */
   6 #include "methods/dummychecktask.hpp"
   7 #include "icinga/icingaapplication.hpp"
   8 #include "icinga/pluginutility.hpp"
   9 #include "base/utility.hpp"
  10 #include "base/perfdatavalue.hpp"
  11 #include "base/convert.hpp"
  12 #include "base/function.hpp"
  13 #include "base/logger.hpp"
  14
  15 using namespace icinga;
  16
  17 REGISTER_FUNCTION_NONCONST(Internal, DummyCheck, &DummyCheckTask::ScriptFunc, "checkable:cr:resolvedMacros:useResolvedMacros");
  18
  19 void DummyCheckTask::ScriptFunc(const Checkable::Ptr& checkable, const CheckResult::Ptr& cr,
  20         const Dictionary::Ptr& resolvedMacros, bool useResolvedMacros)
  21 {
  22         REQUIRE_NOT_NULL(checkable);
  23         REQUIRE_NOT_NULL(cr);
  24
  25         CheckCommand::Ptr command = checkable->GetCheckCommand();
  26
  27         Host::Ptr host;
  28         Service::Ptr service;
  29         tie(host, service) = GetHostService(checkable);
  30
  31         MacroProcessor::ResolverList resolvers;
  32         if (service)
  33                 resolvers.emplace_back("service", service);
  34         resolvers.emplace_back("host", host);
  35         resolvers.emplace_back("command", command);
  36         resolvers.emplace_back("icinga", IcingaApplication::GetInstance());
  37
  38         int dummyState = MacroProcessor::ResolveMacros("$dummy_state$", resolvers, checkable->GetLastCheckResult(),
  39                 nullptr, MacroProcessor::EscapeCallback(), resolvedMacros, useResolvedMacros);
  40
  41         String dummyText = MacroProcessor::ResolveMacros("$dummy_text$", resolvers, checkable->GetLastCheckResult(),
  42                 nullptr, MacroProcessor::EscapeCallback(), resolvedMacros, useResolvedMacros);
  43
  44         if (resolvedMacros && !useResolvedMacros)
  45                 return;
  46
  47         /* Parse output and performance data. */
  48         std::pair<String, String> co = PluginUtility::ParseCheckOutput(dummyText);
  49
  50         double now = Utility::GetTime();
  51
  52         cr->SetOutput(co.first);
  53         cr->SetPerformanceData(PluginUtility::SplitPerfdata(co.second));
  54         cr->SetState(PluginUtility::ExitStatusToState(dummyState));
  55         cr->SetExitStatus(dummyState);
  56         cr->SetExecutionStart(now);
  57         cr->SetExecutionEnd(now);
  58         cr->SetCommand(command->GetName());
  59
  60         checkable->ProcessCheckResult(cr);
  61 }