mirror of https://github.com/Icinga/icinga2.git
130 lines
4.6 KiB
C++
130 lines
4.6 KiB
C++
/******************************************************************************
|
|
* 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. *
|
|
******************************************************************************/
|
|
|
|
#include "icinga/command.hpp"
|
|
#include "icinga/macroprocessor.hpp"
|
|
#include "base/function.hpp"
|
|
#include "base/exception.hpp"
|
|
#include "base/objectlock.hpp"
|
|
#include <boost/foreach.hpp>
|
|
|
|
using namespace icinga;
|
|
|
|
REGISTER_TYPE(Command);
|
|
REGISTER_SCRIPTFUNCTION(ValidateCommandAttributes, &Command::ValidateAttributes);
|
|
REGISTER_SCRIPTFUNCTION(ValidateCommandArguments, &Command::ValidateArguments);
|
|
REGISTER_SCRIPTFUNCTION(ValidateEnvironmentVariables, &Command::ValidateEnvironmentVariables);
|
|
|
|
int Command::GetModifiedAttributes(void) const
|
|
{
|
|
int attrs = 0;
|
|
|
|
if (GetOverrideVars())
|
|
attrs |= ModAttrCustomVariable;
|
|
|
|
return attrs;
|
|
}
|
|
|
|
void Command::SetModifiedAttributes(int flags, const MessageOrigin& origin)
|
|
{
|
|
if ((flags & ModAttrCustomVariable) == 0) {
|
|
SetOverrideVars(Empty);
|
|
OnVarsChanged(this, GetVars(), origin);
|
|
}
|
|
}
|
|
|
|
void Command::ValidateAttributes(const String& location, const Command::Ptr& object)
|
|
{
|
|
if (object->GetArguments() != Empty && !object->GetCommandLine().IsObjectType<Array>()) {
|
|
BOOST_THROW_EXCEPTION(ScriptError("Validation failed for " +
|
|
location + ": Attribute 'command' must be an array if the 'arguments' attribute is set.", object->GetDebugInfo()));
|
|
}
|
|
}
|
|
|
|
void Command::ValidateArguments(const String& location, const Command::Ptr& object)
|
|
{
|
|
Dictionary::Ptr arguments = object->GetArguments();
|
|
|
|
if (!arguments)
|
|
return;
|
|
|
|
ObjectLock olock(arguments);
|
|
BOOST_FOREACH(const Dictionary::Pair& kv, arguments) {
|
|
const Value& arginfo = kv.second;
|
|
Value argval;
|
|
|
|
if (arginfo.IsObjectType<Dictionary>()) {
|
|
Dictionary::Ptr argdict = arginfo;
|
|
|
|
if (argdict->Contains("value")) {
|
|
String argvalue = argdict->Get("value");
|
|
|
|
if (!MacroProcessor::ValidateMacroString(argvalue)) {
|
|
BOOST_THROW_EXCEPTION(ScriptError("Validation failed for " +
|
|
location + ": Closing $ not found in macro format string '" + argvalue + "'.", object->GetDebugInfo()));
|
|
}
|
|
}
|
|
|
|
if (argdict->Contains("set_if")) {
|
|
String argsetif = argdict->Get("set_if");
|
|
|
|
if (!MacroProcessor::ValidateMacroString(argsetif)) {
|
|
BOOST_THROW_EXCEPTION(ScriptError("Validation failed for " +
|
|
location + ": Closing $ not found in macro format string '" + argsetif + "'.", object->GetDebugInfo()));
|
|
}
|
|
}
|
|
} else if (arginfo.IsObjectType<Function>()) {
|
|
continue; /* we cannot evaluate macros in functions */
|
|
} else {
|
|
argval = arginfo;
|
|
|
|
if (argval.IsEmpty())
|
|
continue;
|
|
|
|
String argstr = argval;
|
|
|
|
if (!MacroProcessor::ValidateMacroString(argstr)) {
|
|
BOOST_THROW_EXCEPTION(ScriptError("Validation failed for " +
|
|
location + ": Closing $ not found in macro format string '" + argstr + "'.", object->GetDebugInfo()));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void Command::ValidateEnvironmentVariables(const String& location, const Command::Ptr& object)
|
|
{
|
|
Dictionary::Ptr env = object->GetEnv();
|
|
|
|
if (!env)
|
|
return;
|
|
|
|
ObjectLock olock(env);
|
|
BOOST_FOREACH(const Dictionary::Pair& kv, env) {
|
|
const Value& envval = kv.second;
|
|
|
|
if (!envval.IsString() || envval.IsEmpty())
|
|
continue;
|
|
|
|
if (!MacroProcessor::ValidateMacroString(envval)) {
|
|
BOOST_THROW_EXCEPTION(ScriptError("Validation failed for " +
|
|
location + ": Closing $ not found in macro format string '" + envval + "'.", object->GetDebugInfo()));
|
|
}
|
|
}
|
|
}
|