LocalForward: when binding IPv6Any set Socket to DualMode. (#277)

tmds · Dec 14, 2024 · 75e0ab6 · 75e0ab6
1 parent ea3a348
commit 75e0ab6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/Tmds.Ssh/LocalForward.cs b/src/Tmds.Ssh/LocalForward.cs
@@ -81,6 +81,11 @@ private void Start(EndPoint bindEP, string remoteEndPoint, Func<CancellationToke
             if (bindEP is IPEndPoint ipEndPoint)
             {
                 _serverSocket = new Socket(ipEndPoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
+
+                if (ipEndPoint.Address.Equals(IPAddress.IPv6Any))
+                {
+                    _serverSocket.DualMode = true;
+                }
             }
             else if (bindEP is UnixDomainSocketEndPoint unixEndPoint)
             {

diff --git a/test/Tmds.Ssh.Tests/LocalForwardTests.cs b/test/Tmds.Ssh.Tests/LocalForwardTests.cs
@@ -129,5 +129,17 @@ public async Task StopsWhenClientDisconnects()
 
         using var socket = new Socket(endPoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
         await Assert.ThrowsAnyAsync<SocketException>(async () => await socket.ConnectAsync(endPoint));
-    } 
+    }
+
+    [Fact]
+    public async Task IPv6IsDualMode()
+    {
+        using var client = await _sshServer.CreateClientAsync();
+
+        using var localForward = await client.StartForwardTcpAsync(new IPEndPoint(IPAddress.IPv6Any, 0), "nowhere", 10);
+
+        using var socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
+
+        await socket.ConnectAsync(new IPEndPoint(IPAddress.Loopback, (localForward.EndPoint as IPEndPoint)!.Port));
+    }
 }