mirror of https://github.com/Icinga/icinga2.git
127 lines
3.6 KiB
C++
127 lines
3.6 KiB
C++
/******************************************************************************
|
|
* 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-compatido.h"
|
|
|
|
using namespace icinga;
|
|
|
|
/**
|
|
* Constructor for the IdoSocket class.
|
|
*
|
|
* @param role The role of the underlying TCP client.
|
|
*/
|
|
IdoSocket::IdoSocket(TcpClientRole role)
|
|
: TcpClient(role)
|
|
{
|
|
/*
|
|
* we currently do not receive any data from the ido socket,
|
|
* this is just data output - so we do not need to bind
|
|
* a local instance of our datahandler in case of a new
|
|
* signal telling about new data
|
|
*/
|
|
OnDataAvailable.connect(boost::bind(&IdoSocket::DataAvailableHandler, this));
|
|
|
|
/*
|
|
* what to do on disconnect
|
|
*/
|
|
OnClosed.connect(boost::bind(&IdoSocket::ClientClosedHandler, this));
|
|
|
|
}
|
|
|
|
/**
|
|
* Sends a message to the ido socket
|
|
*
|
|
* @param message The message.
|
|
*/
|
|
void IdoSocket::SendMessage(const String& message)
|
|
{
|
|
/*
|
|
* write our message to the send queue
|
|
* as we inherit all the functionality
|
|
* of the tcpclient class
|
|
*/
|
|
Write(message.CStr(), message.GetLength());
|
|
}
|
|
|
|
|
|
/**
|
|
* Handles closed client connect
|
|
*/
|
|
void IdoSocket::ClientClosedHandler(void)
|
|
{
|
|
try {
|
|
CheckException();
|
|
} catch (const exception& ex) {
|
|
stringstream message;
|
|
message << "Error occured for ido socket: " << ex.what();
|
|
|
|
Logger::Write(LogWarning, "compatido", message.str());
|
|
}
|
|
|
|
Logger::Write(LogWarning, "compatido", "Lost connection to ido socket");
|
|
|
|
SetReconnect(true);
|
|
|
|
OnDisconnected(GetSelf());
|
|
}
|
|
|
|
|
|
/**
|
|
* Set reconnect vstate
|
|
*
|
|
* @aparam enable Enables the reconnect.
|
|
*/
|
|
void IdoSocket::SetReconnect(bool reconnect)
|
|
{
|
|
m_Reconnect = reconnect;
|
|
}
|
|
|
|
/**
|
|
* Get reconnect state
|
|
*
|
|
* @returns reconnect The reconnect variable
|
|
*/
|
|
bool IdoSocket::GetReconnect(void)
|
|
{
|
|
return m_Reconnect;
|
|
}
|
|
|
|
/**
|
|
* Processes inbound data.
|
|
* Currently not used, as we do not receive data from ido sockets
|
|
*/
|
|
void IdoSocket::DataAvailableHandler(void)
|
|
{
|
|
return;
|
|
}
|
|
|
|
/**
|
|
* Factory function for ido socket clients.
|
|
*
|
|
* @param fd The file descriptor.
|
|
* @param role The role of the underlying TCP client.
|
|
* @returns A new ido socket client.
|
|
*/
|
|
IdoSocket::Ptr icinga::IdoSocketFactory(SOCKET fd, TcpClientRole role)
|
|
{
|
|
IdoSocket::Ptr client = boost::make_shared<IdoSocket>(role);
|
|
client->SetFD(fd);
|
|
return client;
|
|
}
|