Improve config validation for arrays of object names

fixes #12556
Icinga · Aug 26, 2016 · dac0ff9 · dac0ff9
1 parent 4f46e59
commit dac0ff9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/lib/remote/zone.cpp b/lib/remote/zone.cpp
@@ -41,7 +41,10 @@ void Zone::OnAllConfigLoaded(void)
 	if (endpoints) {
 		ObjectLock olock(endpoints);
 		for (const String& endpoint : endpoints) {
-			Endpoint::GetByName(endpoint)->SetCachedZone(this);
+			Endpoint::Ptr ep = Endpoint::GetByName(endpoint);
+
+			if (ep)
+				ep->SetCachedZone(this);
 		}
 	}
 

diff --git a/tools/mkclass/classcompiler.cpp b/tools/mkclass/classcompiler.cpp
@@ -535,7 +535,14 @@ void ClassCompiler::HandleClass(const Klass& klass, const ClassDebugInfo&)
 		std::string ftype = FieldTypeToIcingaName(field, true);
 
 		if (field.Type.IsName) {
-			m_Impl << "\t" << "if (!avalue.IsEmpty() && !utils.ValidateName(\"" << field.Type.TypeName << "\", avalue))" << std::endl
+			m_Impl << "\t" << "if (";
+
+			if (field.Type.ArrayRank > 0)
+				m_Impl << "avalue.IsEmpty() || ";
+			else
+				m_Impl << "!avalue.IsEmpty() && ";
+
+			m_Impl << "!utils.ValidateName(\"" << field.Type.TypeName << "\", avalue))" << std::endl
 			       << "\t\t" << "BOOST_THROW_EXCEPTION(ValidationError(dynamic_cast<ConfigObject *>(this), boost::assign::list_of(\"" << field.Name << "\"), \"Object '\" + avalue + \"' of type '" << field.Type.TypeName
 			       << "' does not exist.\"));" << std::endl;
 		} else if (field.Type.ArrayRank > 0 && (ftype == "Number" || ftype == "Boolean")) {