/****************************************************************************** * 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 "livestatus/component.h" #include "base/dynamictype.h" #include "base/logger_fwd.h" #include "base/tcpsocket.h" #include "base/networkstream.h" #include "base/application.h" #include using namespace icinga; using namespace livestatus; REGISTER_TYPE(LivestatusComponent); LivestatusComponent::LivestatusComponent(const Dictionary::Ptr& serializedUpdate) : DynamicObject(serializedUpdate) { RegisterAttribute("socket_path", Attribute_Config, &m_SocketPath); } /** * Starts the component. */ void LivestatusComponent::Start(void) { //#ifndef _WIN32 // UnixSocket::Ptr socket = boost::make_shared(); // socket->Bind(GetSocketPath()); //#else /* _WIN32 */ TcpSocket::Ptr socket = boost::make_shared(); socket->Bind("6558", AF_INET); //#endif /* _WIN32 */ m_Listener = socket; boost::thread thread(boost::bind(&LivestatusComponent::ServerThreadProc, this, socket)); thread.detach(); } String LivestatusComponent::GetSocketPath(void) const { Value socketPath = m_SocketPath; if (socketPath.IsEmpty()) return Application::GetLocalStateDir() + "/run/icinga2/livestatus"; else return socketPath; } void LivestatusComponent::ServerThreadProc(const Socket::Ptr& server) { server->Listen(); for (;;) { Socket::Ptr client = server->Accept(); Log(LogInformation, "livestatus", "Client connected"); boost::thread thread(boost::bind(&LivestatusComponent::ClientThreadProc, this, client)); thread.detach(); } } void LivestatusComponent::ClientThreadProc(const Socket::Ptr& client) { Stream::Ptr stream = boost::make_shared(client); for (;;) { String line; bool read_line = false; std::vector lines; while (stream->ReadLine(&line)) { read_line = true; if (line.GetLength() > 0) lines.push_back(line); else break; } Query::Ptr query = boost::make_shared(lines); query->Execute(stream); } }