From ae0af5f7e1f996d44013aab8430cb30e0e0d018a Mon Sep 17 00:00:00 2001
From: Brennan <brecon@microsoft.com>
Date: Fri, 10 Jan 2025 16:56:49 -0800
Subject: [PATCH] Fix crashing InMemory.FunctionalTests

---
 .../TestTransport/InMemoryTransportConnection.cs              | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/Servers/Kestrel/test/InMemory.FunctionalTests/TestTransport/InMemoryTransportConnection.cs b/src/Servers/Kestrel/test/InMemory.FunctionalTests/TestTransport/InMemoryTransportConnection.cs
index b6f510a722a4..086694908360 100644
--- a/src/Servers/Kestrel/test/InMemory.FunctionalTests/TestTransport/InMemoryTransportConnection.cs
+++ b/src/Servers/Kestrel/test/InMemory.FunctionalTests/TestTransport/InMemoryTransportConnection.cs
@@ -63,13 +63,11 @@ public override void Abort(ConnectionAbortedException abortReason)
 
     public void OnClosed()
     {
-        if (_isClosed)
+        if (Interlocked.CompareExchange(ref _isClosed, true, false) == true)
         {
             return;
         }
 
-        _isClosed = true;
-
         ThreadPool.UnsafeQueueUserWorkItem(state =>
         {
             state._connectionClosedTokenSource.Cancel();