icinga2/base/tcpserver.cpp

64 lines
1.1 KiB
C++
Raw Normal View History

2012-03-28 13:24:49 +02:00
#include "i2-base.h"
using namespace icinga;
TCPServer::TCPServer(void)
{
2012-04-24 14:02:15 +02:00
m_ClientFactory = bind(&TCPClientFactory, RoleInbound);
2012-03-28 13:24:49 +02:00
}
2012-04-24 14:02:15 +02:00
void TCPServer::SetClientFactory(function<TCPClient::Ptr()> clientFactory)
2012-03-28 13:24:49 +02:00
{
m_ClientFactory = clientFactory;
}
2012-04-24 14:02:15 +02:00
function<TCPClient::Ptr()> TCPServer::GetFactoryFunction(void) const
2012-03-28 13:24:49 +02:00
{
return m_ClientFactory;
}
void TCPServer::Start(void)
{
TCPSocket::Start();
2012-04-03 11:13:17 +02:00
OnReadable += bind_weak(&TCPServer::ReadableEventHandler, shared_from_this());
2012-03-28 13:24:49 +02:00
}
void TCPServer::Listen(void)
{
int rc = listen(GetFD(), SOMAXCONN);
if (rc < 0) {
Close();
return;
}
2012-03-28 13:24:49 +02:00
}
2012-04-18 15:22:25 +02:00
int TCPServer::ReadableEventHandler(const EventArgs& ea)
2012-03-28 13:24:49 +02:00
{
int fd;
sockaddr_in addr;
socklen_t addrlen = sizeof(addr);
fd = accept(GetFD(), (sockaddr *)&addr, &addrlen);
2012-04-22 16:45:31 +02:00
if (fd < 0) {
HandleSocketError();
return 0;
2012-04-18 15:22:25 +02:00
}
NewClientEventArgs nea;
nea.Source = shared_from_this();
nea.Client = static_pointer_cast<TCPSocket>(m_ClientFactory());
nea.Client->SetFD(fd);
nea.Client->Start();
2012-03-28 13:24:49 +02:00
OnNewClient(nea);
return 0;
}
bool TCPServer::WantsToRead(void) const
{
return true;
}