Add support of receive buffer limit & send buffer limit

chronoxor · Jun 4, 2021 · cb854eb · cb854eb
1 parent 4fe5800
commit cb854eb
Show file tree

Hide file tree

Showing 7 changed files with 154 additions and 1 deletion.
diff --git a/source/NetCoreServer/NetCoreServer.csproj b/source/NetCoreServer/NetCoreServer.csproj
@@ -2,7 +2,7 @@
 
   <PropertyGroup>
     <TargetFramework>net5.0</TargetFramework>
-    <Version>5.0.18.0</Version>
+    <Version>5.1.0.0</Version>
     <Authors>Ivan Shynkarenka</Authors>
     <Copyright>Copyright (c) 2019-2021 Ivan Shynkarenka</Copyright>
     <RepositoryUrl>https://github.com/chronoxor/NetCoreServer</RepositoryUrl>

diff --git a/source/NetCoreServer/SslClient.cs b/source/NetCoreServer/SslClient.cs
@@ -104,10 +104,18 @@ public SslClient(SslContext context, IPEndPoint endpoint)
         /// </remarks>
         public bool OptionNoDelay { get; set; }
         /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -519,6 +527,13 @@ public virtual bool SendAsync(byte[] buffer, long offset, long size)
 
             lock (_sendLock)
             {
+                // Check the send buffer limit
+                if (((_sendBufferMain.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+                    return false;
+                }
+
                 // Fill the main send buffer
                 _sendBufferMain.Append(buffer, offset, size);
 
@@ -866,7 +881,17 @@ private void ProcessReceive(IAsyncResult result)
 
                     // If the receive buffer is full increase its size
                     if (_receiveBuffer.Capacity == size)
+                    {
+                        // Check the receive buffer limit
+                        if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                        {
+                            SendError(SocketError.NoBufferSpaceAvailable);
+                            DisconnectAsync();
+                            return;
+                        }
+
                         _receiveBuffer.Reserve(2 * size);
+                    }
                 }
 
                 _receiving = false;

diff --git a/source/NetCoreServer/SslSession.cs b/source/NetCoreServer/SslSession.cs
@@ -56,11 +56,19 @@ public SslSession(SslServer server)
         /// </summary>
         public long BytesReceived { get; private set; }
 
+        /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
         /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -320,6 +328,13 @@ public virtual bool SendAsync(byte[] buffer, long offset, long size)
 
             lock (_sendLock)
             {
+                // Check the send buffer limit
+                if (((_sendBufferMain.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+                    return false;
+                }
+
                 // Fill the main send buffer
                 _sendBufferMain.Append(buffer, offset, size);
 
@@ -589,7 +604,17 @@ private void ProcessReceive(IAsyncResult result)
 
                     // If the receive buffer is full increase its size
                     if (_receiveBuffer.Capacity == size)
+                    {
+                        // Check the receive buffer limit
+                        if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                        {
+                            SendError(SocketError.NoBufferSpaceAvailable);
+                            Disconnect();
+                            return;
+                        }
+
                         _receiveBuffer.Reserve(2 * size);
+                    }
                 }
 
                 _receiving = false;

diff --git a/source/NetCoreServer/TcpClient.cs b/source/NetCoreServer/TcpClient.cs
@@ -89,10 +89,18 @@ public TcpClient(IPEndPoint endpoint)
         /// </remarks>
         public bool OptionNoDelay { get; set; }
         /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -459,6 +467,13 @@ public virtual bool SendAsync(byte[] buffer, long offset, long size)
 
             lock (_sendLock)
             {
+                // Check the send buffer limit
+                if (((_sendBufferMain.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+                    return false;
+                }
+
                 // Fill the main send buffer
                 _sendBufferMain.Append(buffer, offset, size);
 
@@ -761,7 +776,17 @@ private bool ProcessReceive(SocketAsyncEventArgs e)
 
                 // If the receive buffer is full increase its size
                 if (_receiveBuffer.Capacity == size)
+                {
+                    // Check the receive buffer limit
+                    if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                    {
+                        SendError(SocketError.NoBufferSpaceAvailable);
+                        DisconnectAsync();
+                        return false;
+                    }
+
                     _receiveBuffer.Reserve(2 * size);
+                }
             }
 
             _receiving = false;

diff --git a/source/NetCoreServer/TcpSession.cs b/source/NetCoreServer/TcpSession.cs
@@ -55,11 +55,19 @@ public TcpSession(TcpServer server)
         /// </summary>
         public long BytesReceived { get; private set; }
 
+        /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
         /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -294,6 +302,13 @@ public virtual bool SendAsync(byte[] buffer, long offset, long size)
 
             lock (_sendLock)
             {
+                // Check the send buffer limit
+                if (((_sendBufferMain.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+                    return false;
+                }
+
                 // Fill the main send buffer
                 _sendBufferMain.Append(buffer, offset, size);
 
@@ -543,7 +558,17 @@ private bool ProcessReceive(SocketAsyncEventArgs e)
 
                 // If the receive buffer is full increase its size
                 if (_receiveBuffer.Capacity == size)
+                {
+                    // Check the receive buffer limit
+                    if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                    {
+                        SendError(SocketError.NoBufferSpaceAvailable);
+                        Disconnect();
+                        return false;
+                    }
+
                     _receiveBuffer.Reserve(2 * size);
+                }
             }
 
             _receiving = false;

diff --git a/source/NetCoreServer/UdpClient.cs b/source/NetCoreServer/UdpClient.cs
@@ -100,10 +100,18 @@ public UdpClient(IPEndPoint endpoint)
         /// </summary>
         public bool OptionMulticast { get; set; }
         /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -486,6 +494,13 @@ public virtual bool SendAsync(EndPoint endpoint, byte[] buffer, long offset, lon
             if (size == 0)
                 return true;
 
+            // Check the send buffer limit
+            if (((_sendBuffer.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+            {
+                SendError(SocketError.NoBufferSpaceAvailable);
+                return false;
+            }
+
             // Fill the main send buffer
             _sendBuffer.Append(buffer, offset, size);
 
@@ -694,7 +709,17 @@ private void ProcessReceiveFrom(SocketAsyncEventArgs e)
 
             // If the receive buffer is full increase its size
             if (_receiveBuffer.Capacity == size)
+            {
+                // Check the receive buffer limit
+                if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+                    Disconnect();
+                    return;
+                }
+
                 _receiveBuffer.Reserve(2 * size);
+            }
         }
 
         /// <summary>

diff --git a/source/NetCoreServer/UdpServer.cs b/source/NetCoreServer/UdpServer.cs
@@ -101,10 +101,18 @@ public UdpServer(IPEndPoint endpoint)
         /// </remarks>
         public bool OptionExclusiveAddressUse { get; set; }
         /// <summary>
+        /// Option: receive buffer limit
+        /// </summary>
+        public int OptionReceiveBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: receive buffer size
         /// </summary>
         public int OptionReceiveBufferSize { get; set; } = 8192;
         /// <summary>
+        /// Option: send buffer limit
+        /// </summary>
+        public int OptionSendBufferLimit { get; set; } = 0;
+        /// <summary>
         /// Option: send buffer size
         /// </summary>
         public int OptionSendBufferSize { get; set; } = 8192;
@@ -450,6 +458,13 @@ public virtual bool SendAsync(EndPoint endpoint, byte[] buffer, long offset, lon
             if (size == 0)
                 return true;
 
+            // Check the send buffer limit
+            if (((_sendBuffer.Size + size) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
+            {
+                SendError(SocketError.NoBufferSpaceAvailable);
+                return false;
+            }
+
             // Fill the main send buffer
             _sendBuffer.Append(buffer, offset, size);
 
@@ -660,7 +675,20 @@ private void ProcessReceiveFrom(SocketAsyncEventArgs e)
 
             // If the receive buffer is full increase its size
             if (_receiveBuffer.Capacity == size)
+            {
+                // Check the receive buffer limit
+                if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
+                {
+                    SendError(SocketError.NoBufferSpaceAvailable);
+
+                    // Call the datagram received zero handler
+                    OnReceived(e.RemoteEndPoint, _receiveBuffer.Data, 0, 0);
+
+                    return;
+                }
+
                 _receiveBuffer.Reserve(2 * size);
+            }
         }
 
         /// <summary>