Added sockets limit.

Author: [email protected]

sockets/CFunctions.cpp

@@ -26,6 +26,13 @@ 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 )
         {
+            // Socket limit exceeded ?
+            if ( CSocketManager::SocketLimitExceeded ( ) )
+            {
+                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/CFunctions.h

@@ -22,10 +22,7 @@ class CFunctions;
 #define __CFUNCTIONS_H
 
 #include <stdio.h>
-
 #include "ml_sockets.h"
-#include "include/ILuaModuleManager.h"
-extern ILuaModuleManager10 *pModuleManager;
 
 class CFunctions
 {

sockets/CSocketManager.cpp

@@ -28,11 +28,25 @@ void CSocketManager::DoPulse()
     }
 }
 
-void CSocketManager::SocketAdd(CSocket*& pSocket, bool bListen )
+bool CSocketManager::SocketLimitExceeded ( void )
+{
+    unsigned int uiSockets = vecSockets.size();
+    if ( uiSockets > MAX_SOCKETS )
+    {
+        pModuleManager->ErrorPrintf ( "[Sockets] Sockets limit reached! (Max: %i)\n", MAX_SOCKETS );
+        return true;
+    }
+
+    return false;
+}
+
+bool CSocketManager::SocketAdd(CSocket*& pSocket, bool bListen )
 {
     // Add the socket to the loop stuff
     sSocketsStorage* pSocketStorage = new sSocketsStorage ( pSocket, bListen );
     vecSockets.push_back ( pSocketStorage );
+
+    return true;
 }
 
 bool CSocketManager::SocketRemove(CSocket*& pSocket)

sockets/CSocketManager.h

@@ -12,7 +12,11 @@
 class CSocket;
 
 #include <vector>
-#include "ml_sockets.h"
+#include "ml_sockets.h"
+#include "include/ILuaModuleManager.h"
+extern ILuaModuleManager10 *pModuleManager;
+
+#define MAX_SOCKETS 255
 
 struct sSocketsStorage
 {
@@ -31,7 +35,8 @@ class CSocketManager
 public:
     static void DoPulse     ();
 
-    static void SocketAdd   ( CSocket*& pSocket, bool bListen = false );
+    static bool SocketLimitExceeded   ( void );
+    static bool SocketAdd   ( CSocket*& pSocket, bool bListen = false );
     static bool SocketRemove( CSocket*& pSocket );
     static bool GetSocket   ( CSocket*& pSocket, void* pUserdata );