mirror of https://github.com/Icinga/icinga2.git
134 lines
2.2 KiB
C++
134 lines
2.2 KiB
C++
#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(ea->Code);
|
|
OnError(ea);
|
|
|
|
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;
|
|
}
|