From 193b540aa750446bf67dcad132773541214c296d Mon Sep 17 00:00:00 2001
From: brain
Date: Tue, 31 Oct 2006 20:32:01 +0000
Subject: Socket error state stuff

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@5607 e03df62e-2008-0410-955e-edbf42e46eb7
---
 src/socketengine_epoll.cpp | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

(limited to 'src/socketengine_epoll.cpp')

diff --git a/src/socketengine_epoll.cpp b/src/socketengine_epoll.cpp
index c3c0c89a2..618f30592 100644
--- a/src/socketengine_epoll.cpp
+++ b/src/socketengine_epoll.cpp
@@ -130,10 +130,25 @@ int EPollEngine::GetRemainingFds()
 
 int EPollEngine::DispatchEvents()
 {
+	socklen_t codesize;
+	int errcode;
 	int i = epoll_wait(EngineHandle, events, MAX_DESCRIPTORS, 150);
 	for (int j = 0; j < i; j++)
 	{
 		ServerInstance->Log(DEBUG,"Handle %s event on fd %d",events[j].events & EPOLLOUT ? "write" : "read", events[j].data.fd);
+		if (events[j].events & EPOLLHUP)
+		{
+			ref[events[j].data.fd]->HandleEvent(EVENT_ERROR, 0);
+			continue;
+		}
+		if (events[j].events & EPOLLERR)
+		{
+			/* Get error number */
+			if (getsockopt(events[j].data.fd, SOL_SOCKET, SO_ERROR, &errcode, &codesize) < 0)
+				errcode = errno;
+			ref[events[j].data.fd]->HandleEvent(EVENT_ERROR, errcode);
+			continue;
+		}
 		if (events[j].events & EPOLLOUT)
 		{
 			struct epoll_event ev;
@@ -144,11 +159,13 @@ int EPollEngine::DispatchEvents()
 			{
 				ServerInstance->Log(DEBUG,"epoll: Could not reset fd %d!", events[j].data.fd);
 			}
-			ref[events[j].data.fd]->HandleEvent(EVENT_WRITE);
+			if (ref[events[j].data.fd])
+				ref[events[j].data.fd]->HandleEvent(EVENT_WRITE);
 		}
 		else
 		{
-			ref[events[j].data.fd]->HandleEvent(EVENT_READ);
+			if (ref[events[j].data.fd])
+				ref[events[j].data.fd]->HandleEvent(EVENT_READ);
 		}
 	}
 
-- 
cgit v1.3.1-10-gc9f91