1 /******************************************************************************
3 * Copyright (C) 2012-2018 Icinga Development Team (https://www.icinga.com/) *
5 * This program is free software; you can redistribute it and/or *
6 * modify it under the terms of the GNU General Public License *
7 * as published by the Free Software Foundation; either version 2 *
8 * of the License, or (at your option) any later version. *
10 * This program is distributed in the hope that it will be useful, *
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13 * GNU General Public License for more details. *
15 * You should have received a copy of the GNU General Public License *
16 * along with this program; if not, write to the Free Software Foundation *
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
18 ******************************************************************************/
20 #include "icinga/compatutility.hpp"
21 #include "compat/checkresultreader.hpp"
22 #include "compat/checkresultreader-ti.cpp"
23 #include "icinga/service.hpp"
24 #include "icinga/pluginutility.hpp"
25 #include "icinga/icingaapplication.hpp"
26 #include "base/configtype.hpp"
27 #include "base/objectlock.hpp"
28 #include "base/logger.hpp"
29 #include "base/convert.hpp"
30 #include "base/application.hpp"
31 #include "base/utility.hpp"
32 #include "base/exception.hpp"
33 #include "base/context.hpp"
34 #include "base/statsfunction.hpp"
37 using namespace icinga;
39 REGISTER_TYPE(CheckResultReader);
41 REGISTER_STATSFUNCTION(CheckResultReader, &CheckResultReader::StatsFunc);
43 void CheckResultReader::StatsFunc(const Dictionary::Ptr& status, const Array::Ptr&)
47 for (const CheckResultReader::Ptr& checkresultreader : ConfigType::GetObjectsByType<CheckResultReader>()) {
48 nodes.emplace_back(checkresultreader->GetName(), 1); //add more stats
51 status->Set("checkresultreader", new Dictionary(std::move(nodes)));
55 * @threadsafety Always.
57 void CheckResultReader::Start(bool runtimeCreated)
59 ObjectImpl<CheckResultReader>::Start(runtimeCreated);
61 Log(LogInformation, "CheckResultReader")
62 << "'" << GetName() << "' started.";
64 Log(LogWarning, "CheckResultReader")
65 << "The CheckResultReader feature is DEPRECATED and will be removed in Icinga v2.11.";
68 m_ReadTimer = new Timer();
69 m_ReadTimer->OnTimerExpired.connect(std::bind(&CheckResultReader::ReadTimerHandler, this));
70 m_ReadTimer->SetInterval(5);
76 * @threadsafety Always.
78 void CheckResultReader::Stop(bool runtimeRemoved)
80 Log(LogInformation, "CheckResultReader")
81 << "'" << GetName() << "' stopped.";
83 ObjectImpl<CheckResultReader>::Stop(runtimeRemoved);
87 * @threadsafety Always.
89 void CheckResultReader::ReadTimerHandler() const
91 CONTEXT("Processing check result files in '" + GetSpoolDir() + "'");
93 Utility::Glob(GetSpoolDir() + "/c??????.ok", std::bind(&CheckResultReader::ProcessCheckResultFile, this, _1), GlobFile);
96 void CheckResultReader::ProcessCheckResultFile(const String& path) const
98 CONTEXT("Processing check result file '" + path + "'");
100 String crfile = String(path.Begin(), path.End() - 3); /* Remove the ".ok" extension. */
103 fp.exceptions(std::ifstream::badbit);
104 fp.open(crfile.CStr());
106 std::map<String, String> attrs;
110 std::getline(fp, line);
112 if (line.empty() || line[0] == '#')
113 continue; /* Ignore comments and empty lines. */
115 size_t pos = line.find_first_of('=');
117 if (pos == std::string::npos)
118 continue; /* Ignore invalid lines. */
120 String key = line.substr(0, pos);
121 String value = line.substr(pos + 1);
126 /* Remove the checkresult files. */
127 if (unlink(path.CStr()) < 0)
128 BOOST_THROW_EXCEPTION(posix_error()
129 << boost::errinfo_api_function("unlink")
130 << boost::errinfo_errno(errno)
131 << boost::errinfo_file_name(path));
133 if (unlink(crfile.CStr()) < 0)
134 BOOST_THROW_EXCEPTION(posix_error()
135 << boost::errinfo_api_function("unlink")
136 << boost::errinfo_errno(errno)
137 << boost::errinfo_file_name(crfile));
139 Checkable::Ptr checkable;
141 Host::Ptr host = Host::GetByName(attrs["host_name"]);
144 Log(LogWarning, "CheckResultReader")
145 << "Ignoring checkresult file for host '" << attrs["host_name"] << "': Host does not exist.";
150 if (attrs.find("service_description") != attrs.end()) {
151 Service::Ptr service = host->GetServiceByShortName(attrs["service_description"]);
154 Log(LogWarning, "CheckResultReader")
155 << "Ignoring checkresult file for host '" << attrs["host_name"]
156 << "', service '" << attrs["service_description"] << "': Service does not exist.";
165 CheckResult::Ptr result = new CheckResult();
166 String output = CompatUtility::UnEscapeString(attrs["output"]);
167 std::pair<String, Value> co = PluginUtility::ParseCheckOutput(output);
168 result->SetOutput(co.first);
169 result->SetPerformanceData(PluginUtility::SplitPerfdata(co.second));
170 result->SetState(PluginUtility::ExitStatusToState(Convert::ToLong(attrs["return_code"])));
172 if (attrs.find("start_time") != attrs.end())
173 result->SetExecutionStart(Convert::ToDouble(attrs["start_time"]));
175 result->SetExecutionStart(Utility::GetTime());
177 if (attrs.find("finish_time") != attrs.end())
178 result->SetExecutionEnd(Convert::ToDouble(attrs["finish_time"]));
180 result->SetExecutionEnd(result->GetExecutionStart());
182 checkable->ProcessCheckResult(result);
184 Log(LogDebug, "CheckResultReader")
185 << "Processed checkresult file for object '" << checkable->GetName() << "'";
187 /* Reschedule the next check. The side effect of this is that for as long
188 * as we receive check result files for a host/service we won't execute any
190 checkable->SetNextCheck(Utility::GetTime() + checkable->GetCheckInterval());