granicus.if.org Git - icinga2/blob - lib/icinga/command.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   3 #include "icinga/command.hpp"
   4 #include "icinga/command-ti.cpp"
   5 #include "icinga/macroprocessor.hpp"
   6 #include "base/exception.hpp"
   7 #include "base/objectlock.hpp"
   8
   9 using namespace icinga;
  10
  11 REGISTER_TYPE(Command);
  12
  13 void Command::Validate(int types, const ValidationUtils& utils)
  14 {
  15         ObjectImpl<Command>::Validate(types, utils);
  16
  17         Dictionary::Ptr arguments = GetArguments();
  18
  19         if (!(types & FAConfig))
  20                 return;
  21
  22         if (arguments) {
  23                 if (!GetCommandLine().IsObjectType<Array>())
  24                         BOOST_THROW_EXCEPTION(ValidationError(this, { "command" }, "Attribute 'command' must be an array if the 'arguments' attribute is set."));
  25
  26                 ObjectLock olock(arguments);
  27                 for (const Dictionary::Pair& kv : arguments) {
  28                         const Value& arginfo = kv.second;
  29                         Value argval;
  30
  31                         if (arginfo.IsObjectType<Dictionary>()) {
  32                                 Dictionary::Ptr argdict = arginfo;
  33
  34                                 if (argdict->Contains("value")) {
  35                                         Value argvalue = argdict->Get("value");
  36
  37                                         if (argvalue.IsString() && !MacroProcessor::ValidateMacroString(argvalue))
  38                                                 BOOST_THROW_EXCEPTION(ValidationError(this, { "arguments", kv.first, "value" }, "Validation failed: Closing $ not found in macro format string '" + argvalue + "'."));
  39                                 }
  40
  41                                 if (argdict->Contains("set_if")) {
  42                                         Value argsetif = argdict->Get("set_if");
  43
  44                                         if (argsetif.IsString() && !MacroProcessor::ValidateMacroString(argsetif))
  45                                                 BOOST_THROW_EXCEPTION(ValidationError(this, { "arguments", kv.first, "set_if" }, "Closing $ not found in macro format string '" + argsetif + "'."));
  46                                 }
  47                         } else if (arginfo.IsString()) {
  48                                 if (!MacroProcessor::ValidateMacroString(arginfo))
  49                                         BOOST_THROW_EXCEPTION(ValidationError(this, { "arguments", kv.first }, "Closing $ not found in macro format string '" + arginfo + "'."));
  50                         }
  51                 }
  52         }
  53
  54         Dictionary::Ptr env = GetEnv();
  55
  56         if (env) {
  57                 ObjectLock olock(env);
  58                 for (const Dictionary::Pair& kv : env) {
  59                         const Value& envval = kv.second;
  60
  61                         if (!envval.IsString() || envval.IsEmpty())
  62                                 continue;
  63
  64                         if (!MacroProcessor::ValidateMacroString(envval))
  65                                 BOOST_THROW_EXCEPTION(ValidationError(this, { "env", kv.first }, "Closing $ not found in macro format string '" + envval + "'."));
  66                 }
  67         }
  68 }