NatTypeTester/STUN/Utils/AttributeExtensions.cs

using STUN.Enums;
using STUN.Messages;
using STUN.Messages.StunAttributeValues;
using System.Net;

namespace STUN.Utils;

public static class AttributeExtensions
{
	public static StunAttribute BuildChangeRequest(bool changeIp, bool changePort)
	{
		return new StunAttribute
		{
			Type = AttributeType.ChangeRequest,
			Length = 4,
			Value = new ChangeRequestStunAttributeValue { ChangeIp = changeIp, ChangePort = changePort }
		};
	}

	public static StunAttribute BuildMapping(IpFamily family, IPAddress ip, ushort port)
	{
		int length = family switch
		{
			IpFamily.IPv4 => 4,
			IpFamily.IPv6 => 16,
			_ => throw new ArgumentOutOfRangeException(nameof(family), family, null)
		};
		return new StunAttribute
		{
			Type = AttributeType.MappedAddress,
			Length = (ushort)(4 + length),
			Value = new MappedAddressStunAttributeValue
			{
				Family = family,
				Address = ip,
				Port = port
			}
		};
	}

	public static StunAttribute BuildChangeAddress(IpFamily family, IPAddress ip, ushort port)
	{
		int length = family switch
		{
			IpFamily.IPv4 => 4,
			IpFamily.IPv6 => 16,
			_ => throw new ArgumentOutOfRangeException(nameof(family), family, null)
		};
		return new StunAttribute
		{
			Type = AttributeType.ChangedAddress,
			Length = (ushort)(4 + length),
			Value = new ChangedAddressStunAttributeValue
			{
				Family = family,
				Address = ip,
				Port = port
			}
		};
	}

	public static IPEndPoint? GetMappedAddressAttribute(this StunMessage5389 response)
	{
		StunAttribute? mappedAddressAttribute = response.Attributes.FirstOrDefault(t => t.Type == AttributeType.MappedAddress);

		if (mappedAddressAttribute is null)
		{
			return null;
		}

		MappedAddressStunAttributeValue mapped = (MappedAddressStunAttributeValue)mappedAddressAttribute.Value;
		return new IPEndPoint(mapped.Address!, mapped.Port);
	}

	public static IPEndPoint? GetChangedAddressAttribute(this StunMessage5389 response)
	{
		StunAttribute? changedAddressAttribute = response.Attributes.FirstOrDefault(t => t.Type == AttributeType.ChangedAddress);

		if (changedAddressAttribute is null)
		{
			return null;
		}

		ChangedAddressStunAttributeValue address = (ChangedAddressStunAttributeValue)changedAddressAttribute.Value;
		return new IPEndPoint(address.Address!, address.Port);
	}

	public static IPEndPoint? GetXorMappedAddressAttribute(this StunMessage5389 response)
	{
		StunAttribute? mappedAddressAttribute =
			response.Attributes.FirstOrDefault(t => t.Type == AttributeType.XorMappedAddress) ??
			response.Attributes.FirstOrDefault(t => t.Type == AttributeType.MappedAddress);

		if (mappedAddressAttribute is null)
		{
			return null;
		}

		AddressStunAttributeValue mapped = (AddressStunAttributeValue)mappedAddressAttribute.Value;
		return new IPEndPoint(mapped.Address!, mapped.Port);
	}

	public static IPEndPoint? GetOtherAddressAttribute(this StunMessage5389 response)
	{
		StunAttribute? addressAttribute =
			response.Attributes.FirstOrDefault(t => t.Type == AttributeType.OtherAddress) ??
			response.Attributes.FirstOrDefault(t => t.Type == AttributeType.ChangedAddress);

		if (addressAttribute is null)
		{
			return null;
		}

		AddressStunAttributeValue address = (AddressStunAttributeValue)addressAttribute.Value;
		return new IPEndPoint(address.Address!, address.Port);
	}
}