Added limit of 255 open sockets

Author: ccw808

sockets/include/CSocket.h

@@ -44,6 +44,7 @@ class CSocket
     bool IsConnected  ();
 
     void* GetUserdata ();
+    static int GetTotalOpenSocketCount ();
 
 private:
     bool ProcessTargetLocation (const string& strHost, const unsigned short& usPort);
@@ -61,6 +62,7 @@ class CSocket
     struct sockaddr_in m_sSockAddr;
 
     SOCKET             m_pSocket;
+    static int         ms_iTotalOpenSocketCount;
 };
 
 #endif

sockets/src/CFunctions.cpp

@@ -25,6 +25,14 @@ int CFunctions::sockOpen(lua_State* luaVM)
         // Make sure host is a string, and port is a number
         if (lua_type(luaVM, 1) == LUA_TSTRING && lua_type(luaVM, 2) == LUA_TNUMBER)
         {
+
+            if ( CSocket::GetTotalOpenSocketCount () >= 255 )
+            {
+                pModuleManager->ErrorPrintf("[Sockets] Can't sockOpen as 255 sockets already open");
+                lua_pushboolean(luaVM, false);
+                return 1;
+            }
+
             // Put the host in a string, and the port in an unsigned short
             string strHost        = lua_tostring(luaVM, 1);
             unsigned short usPort = static_cast < unsigned short > ( lua_tonumber ( luaVM, 2 ) );

sockets/src/CSocket.cpp

@@ -7,6 +7,7 @@
 ******************************************************/
 
 #include "CSocket.h"
+int CSocket::ms_iTotalOpenSocketCount = 0;
 
 CSocket::CSocket(lua_State *luaVM, const string& strHost, const unsigned short& usPort)
 {
@@ -29,6 +30,8 @@ CSocket::CSocket(lua_State *luaVM, const string& strHost, const unsigned short&
     if (m_pSocket == ERR_INVALID_SOCKET)
         return;
 
+    ms_iTotalOpenSocketCount++;
+
     // Set the socket to non-blocking mode
     SetNonBlocking();
 
@@ -214,9 +217,11 @@ void CSocket::CloseSocket()
     shutdown(m_pSocket, 1);
 
 #ifdef WIN32
-    closesocket(m_pSocket);
+    if ( closesocket(m_pSocket) == 0 )
+        ms_iTotalOpenSocketCount--;
 #else
-    close(m_pSocket);
+    if ( close(m_pSocket) == 0 )
+        ms_iTotalOpenSocketCount--;
 #endif
 
     // Unset the socket variable, so there's no mistaking there
@@ -258,3 +263,8 @@ void CSocket::TriggerEvent(const string eventName, const string arg1)
 
     args.Call(m_pLuaVM, "triggerEvent");
 }
+
+int CSocket::GetTotalOpenSocketCount()
+{
+    return ms_iTotalOpenSocketCount;
+}