icinga2/base/socket.cpp

#include "i2-base.h"

using namespace icinga;

list<Socket::WeakPtr> Socket::Sockets;

Socket::Socket(void)
{
	m_FD = INVALID_SOCKET;
}

Socket::~Socket(void)
{
	Close(true);
}

void Socket::Start(void)
{
	OnException += bind_weak(&Socket::ExceptionEventHandler, shared_from_this());

	Sockets.push_front(static_pointer_cast<Socket>(shared_from_this()));
}

void Socket::Stop(void)
{
	Sockets.remove_if(weak_ptr_eq_raw<Socket>(this));
}

void Socket::SetFD(SOCKET fd)
{
	unsigned long lTrue = 1;

	if (fd != INVALID_SOCKET)
		ioctlsocket(fd, FIONBIO, &lTrue);

	m_FD = fd;
}

SOCKET Socket::GetFD(void) const
{
	return m_FD;
}

void Socket::Close(void)
{
	Close(false);
}

void Socket::Close(bool from_dtor)
{
	if (m_FD != INVALID_SOCKET) {
		closesocket(m_FD);
		m_FD = INVALID_SOCKET;

		/* nobody can possibly have a valid event subscription when the destructor has been called */
		if (!from_dtor) {
			EventArgs::Ptr ea = make_shared<EventArgs>();
			ea->Source = shared_from_this();
			OnClosed(ea);
		}
	}

	if (!from_dtor)
		Stop();
}

string Socket::FormatErrorCode(int code)
{
	char *message;
	string result = "Unknown socket error.";

#ifdef _WIN32
	if (FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, code, 0, (char *)&message, 0, NULL) != 0) {
		result = string(message);
		LocalFree(message);
	}
#else /* _WIN32 */
	if (code != 0)
		message = strerror(code);

	result = string(message);
#endif /* _WIN32 */

	return result;
}

int Socket::ExceptionEventHandler(EventArgs::Ptr ea)
{
	int opt;
	socklen_t optlen = sizeof(opt);

	int rc = getsockopt(GetFD(), SOL_SOCKET, SO_ERROR, (char *)&opt, &optlen);

	if (rc < 0) {
		Close();
		return 0;
	}

	if (opt != 0) {
		SocketErrorEventArgs::Ptr ea = make_shared<SocketErrorEventArgs>();
		ea->Code = opt;
		ea->Message = FormatErrorCode(opt);
		Close();

	}

	return 0;
}

void Socket::CloseAllSockets(void)
{
	for (list<Socket::WeakPtr>::iterator i = Sockets.begin(); i != Sockets.end(); ) {
		Socket::Ptr socket = i->lock();

		i++;

		if (socket == NULL)
			continue;

		socket->Close();
	}
}

bool Socket::WantsToRead(void) const
{
	return false;
}

bool Socket::WantsToWrite(void) const
{
	return false;
}