From 024f972eb1ffc67bb32b2f083955199cbd6ecdde Mon Sep 17 00:00:00 2001
From: brain <brain@e03df62e-2008-0410-955e-edbf42e46eb7>
Date: Thu, 29 Dec 2005 14:03:14 +0000
Subject: Added counters for MAX_DESCRIPTORS etc which allows the socket engine
 to refuse new fd's once you hit the OS limit

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@2696 e03df62e-2008-0410-955e-edbf42e46eb7
---
 src/inspircd.cpp     |  6 +-----
 src/socketengine.cpp | 31 ++++++++++++++++++++++++++++++-
 2 files changed, 31 insertions(+), 6 deletions(-)

(limited to 'src')

diff --git a/src/inspircd.cpp b/src/inspircd.cpp
index 6f1889bbf..fbcfc12d1 100644
--- a/src/inspircd.cpp
+++ b/src/inspircd.cpp
@@ -474,13 +474,9 @@ int InspIRCd::Run()
 		 * descriptors in its list... dns, modules, users,
 		 * servers... so its nice and easy, just one call.
 		 */
-		numberactive = SE->Wait(activefds);
-
-		if (!numberactive)
+		if (!(numberactive = SE->Wait(activefds)))
 			continue;
 
-		log(DEBUG,"%d active fds this time around",numberactive);
-
 		/**
 		 * Now process each of the fd's. For users, we have a fast
 		 * lookup table which can find a user by file descriptor, so
diff --git a/src/socketengine.cpp b/src/socketengine.cpp
index f910802bb..3e0b3814d 100644
--- a/src/socketengine.cpp
+++ b/src/socketengine.cpp
@@ -41,6 +41,7 @@ SocketEngine::SocketEngine()
 #ifdef USE_KQUEUE
 	EngineHandle = kqueue();
 #endif
+	CurrentSetSize = 0;
 }
 
 SocketEngine::~SocketEngine()
@@ -99,7 +100,8 @@ bool SocketEngine::AddFd(int fd, bool readable, char type)
 		return false;
 	}
 #endif
-return true;
+	CurrentSetSize++;
+	return true;
 }
 
 bool SocketEngine::DelFd(int fd)
@@ -138,10 +140,37 @@ bool SocketEngine::DelFd(int fd)
 		return false;
 	}
 #endif
+	CurrentSetSize--;
 	ref[fd] = 0;
 	return true;
 }
 
+int SocketEngine::GetMaxFds()
+{
+#ifdef USE_SELECT
+	return FD_SETSIZE;
+#endif
+#ifdef USE_KQUEUE
+	return MAX_DESCRIPTORS;
+#endif
+#ifdef USE_EPOLL
+	return MAX_DESCRIPTORS;
+#endif
+}
+
+int SocketEngine::GetRemainingFds()
+{
+#ifdef USE_SELECT
+	return FD_SETSIZE - CurrentSetSize;
+#endif
+#ifdef USE_KQUEUE
+	return MAX_DESCRIPTORS - CurrentSetSize;
+#endif
+#ifdef USE_EPOLL
+	return MAX_DESCRIPTORS - CurrentSetSize;
+#endif
+}
+
 int SocketEngine::Wait(int* fdlist)
 {
 	int result = 0;
-- 
cgit v1.2.3