Service: be handled while host is down

[icinga2] / lib / icinga / pluginutility.cpp

diff --git a/lib/icinga/pluginutility.cpp b/lib/icinga/pluginutility.cpp
index 2e1e393f2b99ae7ce431c13fc8f196226861f8a4..d19399870bb5007941f71e633923e87689c3f69f 100644 (file)
--- a/lib/icinga/pluginutility.cpp
+++ b/lib/icinga/pluginutility.cpp
@@ -1,42 +1,22 @@
-/******************************************************************************
- * Icinga 2                                                                   *
- * Copyright (C) 2012-2015 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.             *
- ******************************************************************************/
+/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
 
 #include "icinga/pluginutility.hpp"
 #include "icinga/macroprocessor.hpp"
-#include "icinga/perfdatavalue.hpp"
 #include "base/logger.hpp"
 #include "base/utility.hpp"
+#include "base/perfdatavalue.hpp"
 #include "base/convert.hpp"
 #include "base/process.hpp"
 #include "base/objectlock.hpp"
 #include "base/exception.hpp"
-#include <boost/algorithm/string/classification.hpp>
-#include <boost/algorithm/string/split.hpp>
 #include <boost/algorithm/string/trim.hpp>
-#include <boost/foreach.hpp>
 
 using namespace icinga;
 
 void PluginUtility::ExecuteCommand(const Command::Ptr& commandObj, const Checkable::Ptr& checkable,
-    const CheckResult::Ptr& cr, const MacroProcessor::ResolverList& macroResolvers,
-    const Dictionary::Ptr& resolvedMacros, bool useResolvedMacros,
-    const boost::function<void(const Value& commandLine, const ProcessResult&)>& callback)
+       const CheckResult::Ptr& cr, const MacroProcessor::ResolverList& macroResolvers,
+       const Dictionary::Ptr& resolvedMacros, bool useResolvedMacros, int timeout,
+       const std::function<void(const Value& commandLine, const ProcessResult&)>& callback)
 {
        Value raw_command = commandObj->GetCommandLine();
        Dictionary::Ptr raw_arguments = commandObj->GetArguments();
@@ -45,7 +25,7 @@ void PluginUtility::ExecuteCommand(const Command::Ptr& commandObj, const Checkab
 
        try {
                command = MacroProcessor::ResolveArguments(raw_command, raw_arguments,
-                   macroResolvers, cr, resolvedMacros, useResolvedMacros);
+                       macroResolvers, cr, resolvedMacros, useResolvedMacros);
        } catch (const std::exception& ex) {
                String message = DiagnosticInformation(ex);
 
@@ -70,12 +50,19 @@ void PluginUtility::ExecuteCommand(const Command::Ptr& commandObj, const Checkab
 
        if (env) {
                ObjectLock olock(env);
-               BOOST_FOREACH(const Dictionary::Pair& kv, env) {
+               for (const Dictionary::Pair& kv : env) {
                        String name = kv.second;
 
+                       String missingMacro;
                        Value value = MacroProcessor::ResolveMacros(name, macroResolvers, cr,
-                           NULL, MacroProcessor::EscapeCallback(), resolvedMacros,
-                           useResolvedMacros);
+                               &missingMacro, MacroProcessor::EscapeCallback(), resolvedMacros,
+                               useResolvedMacros);
+
+#ifdef I2_DEBUG
+                       if (!missingMacro.IsEmpty())
+                               Log(LogDebug, "PluginUtility")
+                                       << "Macro '" << name << "' is not defined.";
+#endif /* I2_DEBUG */
 
                        if (value.IsObjectType<Array>())
                                value = Utility::Join(value, ';');
@@ -88,8 +75,11 @@ void PluginUtility::ExecuteCommand(const Command::Ptr& commandObj, const Checkab
                return;
 
        Process::Ptr process = new Process(Process::PrepareCommand(command), envMacros);
-       process->SetTimeout(commandObj->GetTimeout());
-       process->Run(boost::bind(callback, command, _1));
+
+       process->SetTimeout(timeout);
+       process->SetAdjustPriority(true);
+
+       process->Run(std::bind(callback, command, _1));
 }
 
 ServiceState PluginUtility::ExitStatusToState(int exitStatus)
@@ -111,10 +101,9 @@ std::pair<String, String> PluginUtility::ParseCheckOutput(const String& output)
        String text;
        String perfdata;
 
-       std::vector<String> lines;
-       boost::algorithm::split(lines, output, boost::is_any_of("\r\n"));
+       std::vector<String> lines = output.Split("\r\n");
 
-       BOOST_FOREACH (const String& line, lines) {
+       for (const String& line : lines) {
                size_t delim = line.FindFirstOf("|");
 
                if (!text.IsEmpty())
@@ -139,7 +128,7 @@ std::pair<String, String> PluginUtility::ParseCheckOutput(const String& output)
 
 Array::Ptr PluginUtility::SplitPerfdata(const String& perfdata)
 {
-       Array::Ptr result = new Array();
+       ArrayData result;
 
        size_t begin = 0;
        String multi_prefix;
@@ -176,7 +165,7 @@ Array::Ptr PluginUtility::SplitPerfdata(const String& perfdata)
                else
                        pdv = label + "=" + value;
 
-               result->Add(pdv);
+               result.emplace_back(std::move(pdv));
 
                if (multi_index != String::NPos)
                        multi_prefix = label.SubStr(0, multi_index);
@@ -184,7 +173,7 @@ Array::Ptr PluginUtility::SplitPerfdata(const String& perfdata)
                begin = spq + 1;
        }
 
-       return result;
+       return new Array(std::move(result));
 }
 
 String PluginUtility::FormatPerfdata(const Array::Ptr& perfdata)
@@ -197,7 +186,7 @@ String PluginUtility::FormatPerfdata(const Array::Ptr& perfdata)
        ObjectLock olock(perfdata);
 
        bool first = true;
-       BOOST_FOREACH(const Value& pdv, perfdata) {
+       for (const Value& pdv : perfdata) {
                if (!first)
                        result << " ";
                else