From aea38a0dd702895b7dfd18efd26a0dcead2279fc Mon Sep 17 00:00:00 2001
From: Attila Molnar
Date: Sat, 8 Feb 2014 22:34:11 +0100
Subject: Update SocketEngine::CurrentSetSize in AddFdRef()/DelFdRef()

---
 src/socketengine.cpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/socketengine.cpp')

diff --git a/src/socketengine.cpp b/src/socketengine.cpp
index c5f8031b5..35fe5292f 100644
--- a/src/socketengine.cpp
+++ b/src/socketengine.cpp
@@ -39,6 +39,7 @@ SocketEngine::SocketEngine()
 	TotalEvents = WriteEvents = ReadEvents = ErrorEvents = 0;
 	lastempty = ServerInstance->Time();
 	indata = outdata = 0;
+	CurrentSetSize = 0;
 }
 
 SocketEngine::~SocketEngine()
@@ -103,6 +104,7 @@ bool SocketEngine::AddFdRef(EventHandler* eh)
 	while (static_cast<unsigned int>(fd) >= ref.size())
 		ref.resize(ref.empty() ? 1 : (ref.size() * 2));
 	ref[fd] = eh;
+	CurrentSetSize++;
 	return true;
 }
 
@@ -110,7 +112,10 @@ void SocketEngine::DelFdRef(EventHandler *eh)
 {
 	int fd = eh->GetFd();
 	if (GetRef(fd) == eh)
+	{
 		ref[fd] = NULL;
+		CurrentSetSize--;
+	}
 }
 
 bool SocketEngine::HasFd(int fd)
-- 
cgit v1.3.1-10-gc9f91