Fix QUIC stream handle leak (#56550)

dotnet · Jul 30, 2021 · acb6b9b · acb6b9b
1 parent 58232e0
commit acb6b9b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/libraries/System.Net.Quic/src/System/Net/Quic/Implementations/MsQuic/MsQuicStream.cs b/src/libraries/System.Net.Quic/src/System/Net/Quic/Implementations/MsQuic/MsQuicStream.cs
@@ -1334,6 +1334,13 @@ private static uint HandleEventConnectionClose(State state)
                     ExceptionDispatchInfo.SetCurrentStackTrace(GetConnectionAbortedException(state)));
             }
 
+            // Dispose was called before complete event.
+            bool releaseHandles = Interlocked.Exchange(ref state.ShutdownDone, 2) == 1;
+            if (releaseHandles)
+            {
+                state.Cleanup();
+            }
+
             return MsQuicStatusCodes.Success;
         }