NatTypeTester/UnitTest/StunClient5389TCPTest.cs

using Dns.Net.Clients;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Moq;
using Moq.Protected;
using STUN;
using STUN.Client;
using STUN.Enums;
using STUN.Proxy;
using STUN.StunResult;
using System.Net;

namespace UnitTest;

[TestClass]
public class StunClient5389TCPTest
{
	private readonly DefaultDnsClient _dnsClient = new();

	private static readonly IPEndPoint Any = new(IPAddress.Any, 0);
	private static readonly IPEndPoint LocalAddress1 = IPEndPoint.Parse(@"127.0.0.1:114");
	private static readonly IPEndPoint MappedAddress1 = IPEndPoint.Parse(@"1.1.1.1:114");
	private static readonly IPEndPoint MappedAddress2 = IPEndPoint.Parse(@"1.1.1.1:514");
	private static readonly IPEndPoint ServerAddress = IPEndPoint.Parse(@"2.2.2.2:1919");
	private static readonly IPEndPoint ChangedAddress1 = IPEndPoint.Parse(@"3.3.3.3:23333");
	private static readonly IPEndPoint ChangedAddress2 = IPEndPoint.Parse(@"2.2.2.2:810");
	private static readonly IPEndPoint ChangedAddress3 = IPEndPoint.Parse(@"3.3.3.3:1919");

	[TestMethod]
	public async Task BindingTestSuccessAsync()
	{
		IPAddress ip = await _dnsClient.QueryAsync(@"stun.hot-chilli.net");
		using IStunClient5389 client = new StunClient5389TCP(new IPEndPoint(ip, StunServer.DefaultPort), Any);

		StunResult5389 response = await client.BindingTestAsync();

		Assert.AreEqual(BindingTestResult.Success, response.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, response.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, response.FilteringBehavior);
		Assert.IsNotNull(response.PublicEndPoint);
		Assert.IsNotNull(response.LocalEndPoint);
		Assert.IsNotNull(response.OtherEndPoint);
	}

	[TestMethod]
	public async Task BindingTestFailAsync()
	{
		IPAddress ip = IPAddress.Parse(@"1.1.1.1");
		using IStunClient5389 client = new StunClient5389TCP(new IPEndPoint(ip, StunServer.DefaultPort), Any);

		StunResult5389 response = await client.BindingTestAsync();

		Assert.AreEqual(BindingTestResult.Fail, response.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, response.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, response.FilteringBehavior);
		Assert.IsNull(response.PublicEndPoint);
		Assert.IsNull(response.LocalEndPoint);
		Assert.IsNull(response.OtherEndPoint);
	}

	[TestMethod]
	public async Task TlsBindingTestSuccessAsync()
	{
		Assert.IsTrue(StunServer.TryParse(@"stun.fitauto.ru", out StunServer? stunServer, StunServer.DefaultTlsPort));
		IPAddress ip = await _dnsClient.QueryAsync(stunServer.Hostname);
		ITcpProxy tls = new TlsProxy(stunServer.Hostname);
		using IStunClient5389 client = new StunClient5389TCP(new IPEndPoint(ip, StunServer.DefaultPort), Any, tls);

		StunResult5389 response = await client.BindingTestAsync();

		Assert.AreEqual(BindingTestResult.Success, response.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, response.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, response.FilteringBehavior);
		Assert.IsNotNull(response.PublicEndPoint);
		Assert.IsNotNull(response.LocalEndPoint);
		Assert.IsNotNull(response.OtherEndPoint);
	}

	[Ignore]
	[TestMethod]
	public async Task TestServerAsync()
	{
		const string url = @"https://raw.githubusercontent.com/pradt2/always-online-stun/master/valid_hosts_tcp.txt";
		HttpClient httpClient = new();
		string listRaw = await httpClient.GetStringAsync(url);
		string[] list = listRaw.Split('\n', StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries);

		foreach (string host in list)
		{
			try
			{
				if (!HostnameEndpoint.TryParse(host, out HostnameEndpoint? hostEndpoint, StunServer.DefaultPort))
				{
					continue;
				}

				IPAddress ip = await _dnsClient.QueryAsync(hostEndpoint.Hostname);
				using IStunClient5389 client = new StunClient5389TCP(new IPEndPoint(ip, hostEndpoint.Port), Any);

				await client.QueryAsync();

				if (client.State.MappingBehavior is MappingBehavior.AddressAndPortDependent or MappingBehavior.AddressDependent or MappingBehavior.EndpointIndependent or MappingBehavior.Direct)
				{
					Console.WriteLine(host);
				}
			}
			catch
			{
				// ignored
			}
		}
	}

	[Ignore]
	[TestMethod]
	public async Task TestTlsServerAsync()
	{
		const string url = @"https://raw.githubusercontent.com/pradt2/always-online-stun/master/valid_hosts_tcp.txt";
		HttpClient httpClient = new();
		string listRaw = await httpClient.GetStringAsync(url);
		string[] list = listRaw.Split('\n', StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries);

		foreach (string host in list)
		{
			try
			{
				if (!HostnameEndpoint.TryParse(host, out HostnameEndpoint? hostEndpoint, StunServer.DefaultTlsPort))
				{
					continue;
				}

				IPAddress ip = await _dnsClient.QueryAsync(hostEndpoint.Hostname);
				ITcpProxy proxy = new TlsProxy(hostEndpoint.Hostname);
				using IStunClient5389 client = new StunClient5389TCP(new IPEndPoint(ip, StunServer.DefaultTlsPort), Any, proxy);

				await client.QueryAsync();

				if (client.State.MappingBehavior is MappingBehavior.AddressAndPortDependent or MappingBehavior.AddressDependent or MappingBehavior.EndpointIndependent or MappingBehavior.Direct)
				{
					Console.WriteLine(host);
				}
			}
			catch
			{
				// ignored
			}
		}
	}

	[TestMethod]
	public async Task MappingBehaviorTestFailAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 fail = new() { BindingTestResult = BindingTestResult.Fail };

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(fail);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Fail, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNull(client.State.PublicEndPoint);
		Assert.IsNull(client.State.LocalEndPoint);
		Assert.IsNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestUnsupportedServerAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1
		};
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		await TestAsync();

		StunResult5389 r2 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress2
		};
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r2);
		await TestAsync();

		StunResult5389 r3 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress3
		};
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r3);
		await TestAsync();

		return;

		async Task TestAsync()
		{
			await client.QueryAsync();

			Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
			Assert.AreEqual(MappingBehavior.UnsupportedServer, client.State.MappingBehavior);
			Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
			Assert.IsNotNull(client.State.PublicEndPoint);
			Assert.IsNotNull(client.State.LocalEndPoint);
		}
	}

	[TestMethod]
	public async Task MappingBehaviorTestDirectAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 response = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = MappedAddress1,
			OtherEndPoint = ChangedAddress1
		};

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(response);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Direct, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestEndpointIndependentAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ItExpr.IsAny<IPEndPoint>(), ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.EndpointIndependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTest2FailAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r2 = new() { BindingTestResult = BindingTestResult.Fail };

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ServerAddress, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress3, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r2);
		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Fail, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestAddressDependentAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r2 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r3 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ServerAddress, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress3, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r2);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress1, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r3);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.AddressDependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestAddressAndPortDependentAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r2 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r3 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ServerAddress, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress3, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r2);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress1, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r3);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.AddressAndPortDependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTest3FailAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		StunResult5389 r1 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r2 = new()
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		StunResult5389 r3 = new() { BindingTestResult = BindingTestResult.Fail };

		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ServerAddress, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r1);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress3, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r2);
		mock.Protected().Setup<ValueTask<StunResult5389>>(@"BindingTestBaseAsync", ChangedAddress1, ItExpr.IsAny<CancellationToken>()).ReturnsAsync(r3);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Fail, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.None, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTestAsync()
	{
		Mock<StunClient5389TCP> mock = new(ServerAddress, Any, default!);
		IStunClient5389 client = mock.Object;

		await Assert.ThrowsExceptionAsync<NotSupportedException>(async () => await client.FilteringBehaviorTestAsync());
	}
}