dohertj2/natsnet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(batch3): verify client proxy protocol feature group

Browse Source
This commit is contained in:
Joseph Doherty
2026-02-28 07:24:45 -05:00
parent 6ba5670f83
commit 6b67c83c0e
5 changed files with 189 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

119
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/Protocol/ProxyProtocolTests.cs
View File

@@ -18,6 +18,8 @@ using System.Net;
using System.Net.Sockets;
using Shouldly;
using Xunit;
using ZB.MOM.NatsNet.Server;
using ZB.MOM.NatsNet.Server.Internal;
using ZB.MOM.NatsNet.Server.Protocol;
namespace ZB.MOM.NatsNet.Server.Tests.Protocol;
@@ -427,4 +429,121 @@ public sealed class ProxyProtocolTests
Should.Throw<InvalidDataException>(() =>
ProxyProtocolParser.ReadProxyProtoHeader(stream));
}
[Fact]
public void DetectProxyProtoVersion_WhenV1Header_ReturnsVersionAndPrefix()
{
var header = BuildProxyV1Header("TCP4", "127.0.0.1", "10.0.0.1", 12345, 4222);
using var stream = new MemoryStream(header);
var (version, firstBytes) = ClientConnection.DetectProxyProtoVersion(stream);
version.ShouldBe(1);
System.Text.Encoding.ASCII.GetString(firstBytes).ShouldBe("PROXY ");
}
[Fact]
public void ReadProxyProtoV1Header_WhenPrefixConsumed_ParsesV1Payload()
{
var header = BuildProxyV1Header("TCP4", "192.168.1.50", "10.0.0.1", 12345, 4222);
using var stream = new MemoryStream(header[6..]);
var addr = ClientConnection.ReadProxyProtoV1Header(stream);
addr.ShouldNotBeNull();
addr!.SrcIp.ToString().ShouldBe("192.168.1.50");
addr.SrcPort.ShouldBe((ushort)12345);
}
[Fact]
public void ReadProxyProtoHeader_WhenV2Header_ParsesAddress()
{
var header = BuildProxyV2Header("192.168.1.60", "10.0.0.1", 2222, 4222, 0x10);
using var stream = new MemoryStream(header);
var addr = ClientConnection.ReadProxyProtoHeader(stream);
addr.ShouldNotBeNull();
addr!.SrcIp.ToString().ShouldBe("192.168.1.60");
addr.SrcPort.ShouldBe((ushort)2222);
}
[Fact]
public void ReadProxyProtoV2Header_WhenValidHeader_ParsesAddress()
{
var header = BuildProxyV2Header("2001:db8::11", "2001:db8::22", 3001, 4222, 0x20);
using var stream = new MemoryStream(header);
var addr = ClientConnection.ReadProxyProtoV2Header(stream);
addr.ShouldNotBeNull();
addr!.SrcIp.ToString().ShouldBe("2001:db8::11");
addr.SrcPort.ShouldBe((ushort)3001);
}
[Fact]
public void ParseProxyProtoV2Header_WhenIPv4Family_ParsesAddress()
{
var header = new byte[] { 0x21, 0x11, 0x00, 0x0C };
var addrData = new byte[12];
IPAddress.Parse("172.16.1.10").GetAddressBytes().CopyTo(addrData, 0);
IPAddress.Parse("172.16.1.1").GetAddressBytes().CopyTo(addrData, 4);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(8, 2), 5000);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(10, 2), 4222);
using var stream = new MemoryStream(addrData);
var addr = ClientConnection.ParseProxyProtoV2Header(stream, header);
addr.ShouldNotBeNull();
addr!.SrcIp.ToString().ShouldBe("172.16.1.10");
addr.SrcPort.ShouldBe((ushort)5000);
}
[Fact]
public void ParseIPv4Addr_WhenValidPayload_ParsesAddress()
{
var addrData = new byte[12];
IPAddress.Parse("192.0.2.20").GetAddressBytes().CopyTo(addrData, 0);
IPAddress.Parse("192.0.2.10").GetAddressBytes().CopyTo(addrData, 4);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(8, 2), 7000);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(10, 2), 4222);
using var stream = new MemoryStream(addrData);
var addr = ClientConnection.ParseIPv4Addr(stream, (ushort)addrData.Length);
addr.SrcIp.ToString().ShouldBe("192.0.2.20");
addr.SrcPort.ShouldBe((ushort)7000);
}
[Fact]
public void ParseIPv6Addr_WhenValidPayload_ParsesAddress()
{
var addrData = new byte[36];
IPAddress.Parse("2001:db8::20").GetAddressBytes().CopyTo(addrData, 0);
IPAddress.Parse("2001:db8::10").GetAddressBytes().CopyTo(addrData, 16);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(32, 2), 8000);
BinaryPrimitives.WriteUInt16BigEndian(addrData.AsSpan(34, 2), 4222);
using var stream = new MemoryStream(addrData);
var addr = ClientConnection.ParseIPv6Addr(stream, (ushort)addrData.Length);
addr.SrcIp.ToString().ShouldBe("2001:db8::20");
addr.SrcPort.ShouldBe((ushort)8000);
}
[Fact]
public void RemoteAddr_WhenProxyAddressPresent_ReturnsProxyEndpoint()
{
var client = new ClientConnection(ClientKind.Client);
client.SetProxyRemoteAddress(new ProxyProtocolAddress(
IPAddress.Parse("203.0.113.10"), 4444, IPAddress.Parse("10.0.0.1"), 4222));
var remote = client.RemoteAddr();
remote.ShouldNotBeNull();
remote.ShouldBeOfType<IPEndPoint>();
var endpoint = (IPEndPoint)remote;
endpoint.Address.ToString().ShouldBe("203.0.113.10");
endpoint.Port.ShouldBe(4444);
}
}