mirror of https://github.com/Icinga/icinga2.git
94 lines
3.1 KiB
C++
94 lines
3.1 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2018 Icinga Development Team (https://www.icinga.com/) *
|
|
* *
|
|
* 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 "config/configcompilercontext.hpp"
|
|
#include "base/singleton.hpp"
|
|
#include "base/json.hpp"
|
|
#include "base/netstring.hpp"
|
|
#include "base/exception.hpp"
|
|
#include "base/application.hpp"
|
|
|
|
using namespace icinga;
|
|
|
|
ConfigCompilerContext *ConfigCompilerContext::GetInstance()
|
|
{
|
|
return Singleton<ConfigCompilerContext>::GetInstance();
|
|
}
|
|
|
|
void ConfigCompilerContext::OpenObjectsFile(const String& filename)
|
|
{
|
|
m_ObjectsPath = filename;
|
|
|
|
auto *fp = new std::fstream();
|
|
try {
|
|
m_ObjectsTempFile = Utility::CreateTempFile(filename + ".XXXXXX", 0600, *fp);
|
|
} catch (const std::exception& ex) {
|
|
Log(LogCritical, "cli", "Could not create temporary objects file: " + DiagnosticInformation(ex, false));
|
|
Application::Exit(1);
|
|
}
|
|
|
|
if (!*fp)
|
|
BOOST_THROW_EXCEPTION(std::runtime_error("Could not open '" + m_ObjectsTempFile + "' file"));
|
|
|
|
m_ObjectsFP = fp;
|
|
}
|
|
|
|
void ConfigCompilerContext::WriteObject(const Dictionary::Ptr& object)
|
|
{
|
|
if (!m_ObjectsFP)
|
|
return;
|
|
|
|
String json = JsonEncode(object);
|
|
|
|
{
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
NetString::WriteStringToStream(*m_ObjectsFP, json);
|
|
}
|
|
}
|
|
|
|
void ConfigCompilerContext::CancelObjectsFile()
|
|
{
|
|
delete m_ObjectsFP;
|
|
m_ObjectsFP = nullptr;
|
|
|
|
#ifdef _WIN32
|
|
_unlink(m_ObjectsTempFile.CStr());
|
|
#else /* _WIN32 */
|
|
unlink(m_ObjectsTempFile.CStr());
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
void ConfigCompilerContext::FinishObjectsFile()
|
|
{
|
|
delete m_ObjectsFP;
|
|
m_ObjectsFP = nullptr;
|
|
|
|
#ifdef _WIN32
|
|
_unlink(m_ObjectsPath.CStr());
|
|
#endif /* _WIN32 */
|
|
|
|
if (rename(m_ObjectsTempFile.CStr(), m_ObjectsPath.CStr()) < 0) {
|
|
BOOST_THROW_EXCEPTION(posix_error()
|
|
<< boost::errinfo_api_function("rename")
|
|
<< boost::errinfo_errno(errno)
|
|
<< boost::errinfo_file_name(m_ObjectsTempFile));
|
|
}
|
|
}
|
|
|