1 files changed, 166 insertions, 0 deletions
diff --git a/lib/compat/checkresultreader.cpp b/lib/compat/checkresultreader.cpp
new file mode 100644
index 0000000..e4516a3
--- /dev/null
+++ b/lib/compat/checkresultreader.cpp
@@ -0,0 +1,166 @@
+/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
+
+#include "icinga/compatutility.hpp"
+#include "compat/checkresultreader.hpp"
+#include "compat/checkresultreader-ti.cpp"
+#include "icinga/service.hpp"
+#include "icinga/pluginutility.hpp"
+#include "icinga/icingaapplication.hpp"
+#include "base/configtype.hpp"
+#include "base/objectlock.hpp"
+#include "base/logger.hpp"
+#include "base/convert.hpp"
+#include "base/application.hpp"
+#include "base/utility.hpp"
+#include "base/exception.hpp"
+#include "base/context.hpp"
+#include "base/statsfunction.hpp"
+#include <fstream>
+
+using namespace icinga;
+
+REGISTER_TYPE(CheckResultReader);
+
+REGISTER_STATSFUNCTION(CheckResultReader, &CheckResultReader::StatsFunc);
+
+void CheckResultReader::StatsFunc(const Dictionary::Ptr& status, const Array::Ptr&)
+{
+	DictionaryData nodes;
+
+	for (const CheckResultReader::Ptr& checkresultreader : ConfigType::GetObjectsByType<CheckResultReader>()) {
+		nodes.emplace_back(checkresultreader->GetName(), 1); //add more stats
+	}
+
+	status->Set("checkresultreader", new Dictionary(std::move(nodes)));
+}
+
+/**
+ * @threadsafety Always.
+ */
+void CheckResultReader::Start(bool runtimeCreated)
+{
+	ObjectImpl<CheckResultReader>::Start(runtimeCreated);
+
+	Log(LogInformation, "CheckResultReader")
+		<< "'" << GetName() << "' started.";
+
+	Log(LogWarning, "CheckResultReader")
+		<< "This feature is DEPRECATED and may be removed in future releases. Check the roadmap at https://github.com/Icinga/icinga2/milestones";
+
+#ifndef _WIN32
+	m_ReadTimer = new Timer();
+	m_ReadTimer->OnTimerExpired.connect([this](const Timer * const&) { ReadTimerHandler(); });
+	m_ReadTimer->SetInterval(5);
+	m_ReadTimer->Start();
+#endif /* _WIN32 */
+}
+
+/**
+ * @threadsafety Always.
+ */
+void CheckResultReader::Stop(bool runtimeRemoved)
+{
+	Log(LogInformation, "CheckResultReader")
+		<< "'" << GetName() << "' stopped.";
+
+	ObjectImpl<CheckResultReader>::Stop(runtimeRemoved);
+}
+
+/**
+ * @threadsafety Always.
+ */
+void CheckResultReader::ReadTimerHandler() const
+{
+	CONTEXT("Processing check result files in '" + GetSpoolDir() + "'");
+
+	Utility::Glob(GetSpoolDir() + "/c??????.ok", [this](const String& path) { ProcessCheckResultFile(path); }, GlobFile);
+}
+
+void CheckResultReader::ProcessCheckResultFile(const String& path) const
+{
+	CONTEXT("Processing check result file '" + path + "'");
+
+	String crfile = String(path.Begin(), path.End() - 3); /* Remove the ".ok" extension. */
+
+	std::ifstream fp;
+	fp.exceptions(std::ifstream::badbit);
+	fp.open(crfile.CStr());
+
+	std::map<String, String> attrs;
+
+	while (fp.good()) {
+		std::string line;
+		std::getline(fp, line);
+
+		if (line.empty() || line[0] == '#')
+			continue; /* Ignore comments and empty lines. */
+
+		size_t pos = line.find_first_of('=');
+
+		if (pos == std::string::npos)
+			continue; /* Ignore invalid lines. */
+
+		String key = line.substr(0, pos);
+		String value = line.substr(pos + 1);
+
+		attrs[key] = value;
+	}
+
+	/* Remove the checkresult files. */
+	Utility::Remove(path);
+
+	Utility::Remove(crfile);
+
+	Checkable::Ptr checkable;
+
+	Host::Ptr host = Host::GetByName(attrs["host_name"]);
+
+	if (!host) {
+		Log(LogWarning, "CheckResultReader")
+			<< "Ignoring checkresult file for host '" << attrs["host_name"] << "': Host does not exist.";
+
+		return;
+	}
+
+	if (attrs.find("service_description") != attrs.end()) {
+		Service::Ptr service = host->GetServiceByShortName(attrs["service_description"]);
+
+		if (!service) {
+			Log(LogWarning, "CheckResultReader")
+				<< "Ignoring checkresult file for host '" << attrs["host_name"]
+				<< "', service '" << attrs["service_description"] << "': Service does not exist.";
+
+			return;
+		}
+
+		checkable = service;
+	} else
+		checkable = host;
+
+	CheckResult::Ptr result = new CheckResult();
+	String output = CompatUtility::UnEscapeString(attrs["output"]);
+	std::pair<String, Value> co = PluginUtility::ParseCheckOutput(output);
+	result->SetOutput(co.first);
+	result->SetPerformanceData(PluginUtility::SplitPerfdata(co.second));
+	result->SetState(PluginUtility::ExitStatusToState(Convert::ToLong(attrs["return_code"])));
+
+	if (attrs.find("start_time") != attrs.end())
+		result->SetExecutionStart(Convert::ToDouble(attrs["start_time"]));
+	else
+		result->SetExecutionStart(Utility::GetTime());
+
+	if (attrs.find("finish_time") != attrs.end())
+		result->SetExecutionEnd(Convert::ToDouble(attrs["finish_time"]));
+	else
+		result->SetExecutionEnd(result->GetExecutionStart());
+
+	checkable->ProcessCheckResult(result);
+
+	Log(LogDebug, "CheckResultReader")
+		<< "Processed checkresult file for object '" << checkable->GetName() << "'";
+
+	/* Reschedule the next check. The side effect of this is that for as long
+	 * as we receive check result files for a host/service we won't execute any
+	 * active checks. */
+	checkable->SetNextCheck(Utility::GetTime() + checkable->GetCheckInterval());
+}