tyler.durden
/
icinga2
mirror of https://github.com/Icinga/icinga2.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
icinga2/icinga-app/icinga.cpp

286 lines
8.4 KiB
C++
Raw Blame History

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012 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 <i2-icinga.h>
#ifndef _WIN32
# include "icinga-version.h"
# define ICINGA_VERSION VERSION ", " GIT_MESSAGE
# include <ltdl.h>
#endif /* _WIN32 */
using namespace icinga;
namespace po = boost::program_options;
static po::variables_map g_AppParams;
static String g_ConfigUnit;
#ifndef _WIN32
static bool g_ReloadConfig = false;
static Timer::Ptr g_ReloadConfigTimer;
#endif /* _WIN32 */
static bool LoadConfigFiles(bool validateOnly)
{
ConfigCompilerContext context;
ConfigCompilerContext::SetContext(&context);
BOOST_FOREACH(const String& configPath, g_AppParams["config"].as<vector<String> >()) {
ConfigCompiler::CompileFile(configPath);
}
String name, fragment;
BOOST_FOREACH(tie(name, fragment), ConfigFragmentRegistry::GetInstance()->GetItems()) {
ConfigCompiler::CompileText(name, fragment);
}
ConfigCompilerContext::SetContext(NULL);
bool hasError = false;
BOOST_FOREACH(const ConfigCompilerError& error, context.GetErrors()) {
if (!error.Warning) {
hasError = true;
break;
}
}
/* Don't validate if we have already encountered at least one error. */
if (!hasError)
context.Validate();
hasError = false;
BOOST_FOREACH(const ConfigCompilerError& error, context.GetErrors()) {
if (error.Warning) {
Logger::Write(LogWarning, "icinga-app", "Config warning: " + error.Message);
} else {
hasError = true;
Logger::Write(LogCritical, "icinga-app", "Config error: " + error.Message);
}
}
if (hasError)
return false;
if (validateOnly)
return true;
context.ActivateItems();
if (!g_ConfigUnit.IsEmpty()) {
/* ActivateItems has taken care of replacing all previous items
* with new versions - which are automatically in a different
* compilation unit. This UnloadUnit() call takes care of
* removing all left-over items from the previous config. */
ConfigItem::UnloadUnit(g_ConfigUnit);
}
g_ConfigUnit = context.GetUnit();
return true;
}
#ifndef _WIN32
static void ReloadConfigTimerHandler(void)
{
if (g_ReloadConfig) {
Logger::Write(LogInformation, "icinga-app", "Received SIGHUP. Reloading config files.");
LoadConfigFiles(false);
g_ReloadConfig = false;
}
}
static void SigHupHandler(int signum)
{
ASSERT(signum == SIGHUP);
g_ReloadConfig = true;
}
#endif /* _WIN32 */
/**
* Entry point for the Icinga application.
*
* @params argc Number of command line arguments.
* @params argv Command line arguments.
* @returns The application's exit status.
*/
int main(int argc, char **argv)
{
#ifndef _WIN32
LTDL_SET_PRELOADED_SYMBOLS();
#endif /* _WIN32 */
#ifndef _WIN32
lt_dlinit();
#endif /* _WIN32 */
/* Set command-line arguments. */
Application::SetArgC(argc);
Application::SetArgV(argv);
/* Install exception handlers to make debugging easier. */
Application::InstallExceptionHandlers();
#ifdef ICINGA_PREFIX
Application::SetPrefixDir(ICINGA_PREFIX);
#endif /* ICINGA_PREFIX */
#ifdef ICINGA_LOCALSTATEDIR
Application::SetLocalStateDir(ICINGA_LOCALSTATEDIR);
#endif /* ICINGA_LOCALSTATEDIR */
#ifdef ICINGA_PKGLIBDIR
Application::SetPkgLibDir(ICINGA_PKGLIBDIR);
#endif /* ICINGA_PKGLIBDIR */
#ifdef ICINGA_PKGDATADIR
Application::SetPkgDataDir(ICINGA_PKGDATADIR);
#endif /* ICINGA_PKGDATADIR */
po::options_description desc("Supported options");
desc.add_options()
("help", "show this help message")
("version,V", "show version information")
("library,l", po::value<vector<String> >(), "load a library")
("include,I", po::value<vector<String> >(), "add include search directory")
("config,c", po::value<vector<String> >(), "parse a configuration file")
("validate,v", "exit after validating the configuration")
("debug,x", "enable debugging")
;
try {
po::store(po::parse_command_line(argc, argv, desc), g_AppParams);
} catch (const exception& ex) {
stringstream msgbuf;
msgbuf << "Error while parsing command-line options: " << ex.what();
Logger::Write(LogCritical, "icinga-app", msgbuf.str());
return EXIT_FAILURE;
}
po::notify(g_AppParams);
if (g_AppParams.count("debug"))
Application::SetDebugging(true);
if (g_AppParams.count("help") || g_AppParams.count("version")) {
String appName = Utility::BaseName(argv[0]);
if (appName.GetLength() > 3 && appName.SubStr(0, 3) == "lt-")
appName = appName.SubStr(3, appName.GetLength() - 3);
std::cout << appName << " " << "- The Icinga 2 networking monitoring daemon.";
if (g_AppParams.count("version")) {
#ifndef _WIN32
std::cout << " (Version: " << ICINGA_VERSION << ")";
#endif /* _WIN32 */
std::cout << std::endl
<< "Copyright (c) 2012-2013 Icinga Development Team (http://www.icinga.org)" << std::endl
<< "License GPLv2+: GNU GPL version 2 or later <http://gnu.org/licenses/gpl2.html>" << std::endl
<< "This is free software: you are free to change and redistribute it." << std::endl
<< "There is NO WARRANTY, to the extent permitted by law.";
}
std::cout << std::endl;
if (g_AppParams.count("version"))
return EXIT_SUCCESS;
}
if (g_AppParams.count("help")) {
std::cout << std::endl
<< desc << std::endl
<< "Report bugs at <https://dev.icinga.org/>" << std::endl
<< "Icinga home page: <http://www.icinga.org/>" << std::endl;
return EXIT_SUCCESS;
}
Logger::Write(LogInformation, "icinga-app", "Icinga application loader"
#ifndef _WIN32
" (version: " ICINGA_VERSION ")"
#endif /* _WIN32 */
);
String searchDir = Application::GetPkgLibDir();
Logger::Write(LogInformation, "base", "Adding library search dir: " + searchDir);
#ifdef _WIN32
SetDllDirectory(searchDir.CStr());
#else /* _WIN32 */
lt_dladdsearchdir(searchDir.CStr());
#endif /* _WIN32 */
(void) Utility::LoadExtensionLibrary("icinga");
if (g_AppParams.count("library")) {
BOOST_FOREACH(const String& libraryName, g_AppParams["library"].as<vector<String> >()) {
(void) Utility::LoadExtensionLibrary(libraryName);
}
}
ConfigCompiler::AddIncludeSearchDir(Application::GetPkgDataDir());
if (g_AppParams.count("include")) {
BOOST_FOREACH(const String& includePath, g_AppParams["include"].as<vector<String> >()) {
ConfigCompiler::AddIncludeSearchDir(includePath);
}
}
if (g_AppParams.count("config") == 0) {
Logger::Write(LogCritical, "icinga-app", "You need to specify at least one config file (using the --config option).");
return EXIT_FAILURE;
}
bool validateOnly = g_AppParams.count("validate");
if (!LoadConfigFiles(validateOnly))
return EXIT_FAILURE;
if (validateOnly) {
Logger::Write(LogInformation, "icinga-app", "Finished validating the configuration file(s).");
return EXIT_SUCCESS;
}
Application::Ptr app = Application::GetInstance();
if (!app)
BOOST_THROW_EXCEPTION(runtime_error("Configuration must create an Application object."));
#ifndef _WIN32
struct sigaction sa;
memset(&sa, 0, sizeof(sa));
sa.sa_handler = &SigHupHandler;
sigaction(SIGHUP, &sa, NULL);
g_ReloadConfigTimer = boost::make_shared<Timer>();
g_ReloadConfigTimer->SetInterval(1);
g_ReloadConfigTimer->OnTimerExpired.connect(boost::bind(&ReloadConfigTimerHandler));
g_ReloadConfigTimer->Start();
#endif /* _WIN32 */
return app->Run();
}
Reference in New Issue View Git Blame Copy Permalink