Fixes issue with web sockets on Windows (#399)

microsoft · Nov 29, 2023 · dda2477 · dda2477
1 parent a8a1860
commit dda2477
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dev-proxy-plugins/Inspection/WebSocketServer.cs b/dev-proxy-plugins/Inspection/WebSocketServer.cs
@@ -16,6 +16,7 @@ public class WebSocketServer
     private HttpListener? listener;
     private int port;
     private WebSocket? webSocket;
+    static SemaphoreSlim webSocketSemaphore = new SemaphoreSlim(1, 1);
 
     public bool IsConnected => webSocket is not null;
     public event Action<string>? MessageReceived;
@@ -88,7 +89,13 @@ public async Task SendAsync<TMsg>(TMsg message)
             DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull
         });
 
+        // we need a semaphore to avoid multiple simultaneous writes
+        // which aren't allowed
+        await webSocketSemaphore.WaitAsync();
+
         byte[] messageBytes = Encoding.UTF8.GetBytes(messageString);
         await webSocket.SendAsync(new ArraySegment<byte>(messageBytes), WebSocketMessageType.Text, true, CancellationToken.None);
+
+        webSocketSemaphore.Release();
     }
 }