Update README to add WS data intercept.

twitchax · Sep 26, 2024 · ef2bf59 · ef2bf59
1 parent bb9d6d4
commit ef2bf59
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -213,6 +213,13 @@ public class MyController : Controller
     private WsProxyOptions _wsOptions = WsProxyOptionsBuilder.Instance
         .WithBufferSize(8192)
         .WithIntercept(context => new ValueTask<bool>(context.WebSockets.WebSocketRequestedProtocols.Contains("interceptedProtocol")))
+        .WithDataIntercept((data, direction, type) =>
+        {
+            if(direction == WsProxyDataDirection.Downstream && System.Text.Encoding.Default.GetString(data.Array).StartsWith("BAD")) 
+                data.Array[0] = (byte)'M';
+
+            return Task.CompletedTask;
+        })
         .WithBeforeConnect((context, wso) =>
         {
             wso.AddSubProtocol("myRandomProto");

diff --git a/src/Test/Ws/WsHelpers.cs b/src/Test/Ws/WsHelpers.cs
@@ -61,7 +61,8 @@ internal static Task RunWsServers(CancellationToken token)
                             .UseWs("ws://localhost:5002/", options => options
                                 .WithBufferSize(8192)
                                 .WithIntercept(context => new ValueTask<bool>(context.WebSockets.WebSocketRequestedProtocols.Contains("interceptedProtocol")))
-                                .WithDataIntercept((data, direction, type) => {
+                                .WithDataIntercept((data, direction, type) =>
+                                {
                                     if(direction == WsProxyDataDirection.Downstream && System.Text.Encoding.Default.GetString(data.Array).StartsWith("[should_be_intercepted]"))
                                         data.Array[0] = (byte)']';