Patch dos bug / vuln (#164)

RiptideNetworking · Jan 7, 2025 · ffd0fec · ffd0fec
1 parent 10f4db9
commit ffd0fec
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/RiptideNetworking/RiptideNetworking/Transports/Udp/UdpPeer.cs b/RiptideNetworking/RiptideNetworking/Transports/Udp/UdpPeer.cs
@@ -126,6 +126,8 @@ private void Receive()
                             isRunning = false;
                             break;
                         case SocketError.ConnectionReset:
+                            tryReceiveMore = true;
+                            break;
                         case SocketError.MessageSize:
                         case SocketError.TimedOut:
                             break;
@@ -160,7 +162,7 @@ internal void Send(byte[] dataBuffer, int numBytes, IPEndPoint toEndPoint)
                 if (isRunning)
                     socket.SendTo(dataBuffer, numBytes, SocketFlags.None, toEndPoint);
             }
-            catch(SocketException)
+            catch (SocketException)
             {
                 // May want to consider triggering a disconnect here (perhaps depending on the type
                 // of SocketException)? Timeout should catch disconnections, but disconnecting