NatTypeTester/STUN/Utils/NetUtils.cs

using System;
using System.Diagnostics;
using System.Linq;
using System.Net;
using System.Net.NetworkInformation;
using System.Net.Sockets;
using System.Threading.Tasks;

namespace STUN.Utils
{
	public static class NetUtils
	{
		public static IPEndPoint? ParseEndpoint(string? str)
		{
			if (str is null)
			{
				return null;
			}

			var ipPort = str.Trim().Split(':');
			if (ipPort.Length < 2)
			{
				return null;
			}

			IPAddress? ip = null;
			if (ipPort.Length == 2 && IPAddress.TryParse(ipPort[0], out ip))
			{
				if (!IPAddress.TryParse(ipPort[0], out ip))
				{
					return null;
				}
			}
			else if (ipPort.Length > 2)
			{
				var ipStr = string.Join(@":", ipPort, 0, ipPort.Length - 1);
				if (!ipStr.StartsWith(@"[") || !ipStr.EndsWith(@"]") || !IPAddress.TryParse(ipStr, out ip))
				{
					return null;
				}
			}

			if (ip != null && ushort.TryParse(ipPort.Last(), out var port))
			{
				return new IPEndPoint(ip, port);
			}

			return null;
		}

		public static TcpState GetState(this TcpClient tcpClient)
		{
			var foo = IPGlobalProperties
				.GetIPGlobalProperties()
				.GetActiveTcpConnections()
				.SingleOrDefault(x => x.LocalEndPoint.Equals(tcpClient.Client.LocalEndPoint));
			return foo?.State ?? TcpState.Unknown;
		}

		public static async Task<(IPAddress, int, IPEndPoint)> ReceiveMessageFromAsync(this Socket client, EndPoint receive, byte[] array, SocketFlags flag)
		{
			var tcs = new TaskCompletionSource<(IPAddress, int, IPEndPoint)>(TaskCreationOptions.RunContinuationsAsynchronously);
			_ = Task.Run(() =>
			{
				try
				{
					var length = client.ReceiveMessageFrom(array, 0, array.Length, ref flag, ref receive, out var ipPacketInformation);

					var local = ipPacketInformation.Address;

					Debug.WriteLine($@"{(IPEndPoint)receive} => {local} {length} 字节");
					tcs.SetResult((local, length, (IPEndPoint)receive));
				}
				catch (Exception ex)
				{
					tcs.SetException(ex);
				}
			});

			return await tcs.Task;
		}
	}
}