treat max receive len as a long

Author: mcmonkey4eva

src/Core/Settings.cs

@@ -236,6 +236,9 @@ public string GetExternalUrl()
 
         [ConfigComment("Maximum megabytes that can be sent as a single message from a client to the Swarm server.\nSet this lower to limit above, set this higher to allow very large file uploads.")]
         public int MaxNetworkRequestMegabytes = 200;
+
+        /// <summary>Converts <see cref="MaxNetworkRequestMegabytes"/> to bytes as a long.</summary>
+        public long MaxReceiveBytes => MaxNetworkRequestMegabytes * (1024L * 1024);
     }
 
     /// <summary>Settings related to file paths.</summary>

src/Utils/Utilities.cs

@@ -311,7 +311,7 @@ public static async Task<byte[]> ReceiveData(this WebSocket socket, long maxByte
         {
             result = await socket.ReceiveAsync(buffer, limit);
             ms.Write(buffer, 0, result.Count);
-            if (ms.Length > maxBytes)
+            if (ms.Length > maxBytes || ms.Length >= int.MaxValue)
             {
                 throw new IOException($"Received too much data! (over {maxBytes} bytes)");
             }

src/WebAPI/API.cs

@@ -51,7 +51,7 @@ void Error(string message)
             if (context.WebSockets.IsWebSocketRequest)
             {
                 socket = await context.WebSockets.AcceptWebSocketAsync();
-                input = await socket.ReceiveJson(TimeSpan.FromMinutes(1), Program.ServerSettings.Network.MaxNetworkRequestMegabytes * (1024 * 1024));
+                input = await socket.ReceiveJson(TimeSpan.FromMinutes(1), Program.ServerSettings.Network.MaxReceiveBytes);
             }
             else if (context.Request.Method == "POST")
             {
@@ -61,13 +61,14 @@ void Error(string message)
                     context.Response.Redirect("/Error/BasicAPI");
                     return;
                 }
-                if (context.Request.ContentLength <= 0 || context.Request.ContentLength >= Program.ServerSettings.Network.MaxNetworkRequestMegabytes * (1024 * 1024))
+                // Note: int32 size limit due to array allocation. Can't singular read direct more than 2 gig.
+                if (!context.Request.ContentLength.HasValue || context.Request.ContentLength <= 0 || context.Request.ContentLength >= Program.ServerSettings.Network.MaxReceiveBytes || context.Request.ContentLength >= int.MaxValue)
                 {
                     Error($"Request has invalid content length: {context.Request.ContentLength}");
                     context.Response.Redirect("/Error/BasicAPI");
                     return;
                 }
-                byte[] rawData = new byte[(int)context.Request.ContentLength];
+                byte[] rawData = new byte[(int)context.Request.ContentLength.Value];
                 await context.Request.Body.ReadExactlyAsync(rawData, 0, rawData.Length);
                 input = JObject.Parse(Encoding.UTF8.GetString(rawData));
             }

src/WebAPI/T2IAPI.cs

@@ -100,7 +100,7 @@ static int guessBatchSize(JObject input)
                 int batchOffset = images * guessBatchSize(rawInput);
                 while (!cancelTok.IsCancellationRequested)
                 {
-                    byte[] rec = await socket.ReceiveData(Program.ServerSettings.Network.MaxNetworkRequestMegabytes * (1024 * 1024), linked.Token);
+                    byte[] rec = await socket.ReceiveData(Program.ServerSettings.Network.MaxReceiveBytes, linked.Token);
                     Volatile.Write(ref retain, true);
                     if (socket.State != WebSocketState.Open || cancelTok.IsCancellationRequested || Volatile.Read(ref ended))
                     {