using System;
using System.Net;
using System.Net.Sockets;

namespace NatTypeTester_Console.Net
{
	public static class NetUtils
	{
		#region static method CompareArray

		/// <summary>
		/// Compares if specified array items equals.
		/// </summary>
		/// <param name="array1">Array 1.</param>
		/// <param name="array2">Array 2</param>
		/// <returns>Returns true if both arrays are equal.</returns>
		public static bool CompareArray(Array array1, Array array2)
		{
			return CompareArray(array1, array2, array2.Length);
		}

		/// <summary>
		/// Compares if specified array items equals.
		/// </summary>
		/// <param name="array1">Array 1.</param>
		/// <param name="array2">Array 2</param>
		/// <param name="array2Count">Number of bytes in array 2 used for compare.</param>
		/// <returns>Returns true if both arrays are equal.</returns>
		public static bool CompareArray(Array array1, Array array2, int array2Count)
		{
			if (array1 == null && array2 == null)
			{
				return true;
			}
			if (array1 == null)
			{
				return false;
			}
			if (array2 == null)
			{
				return false;
			}
			if (array1.Length != array2Count)
			{
				return false;
			}

			for (var i = 0; i < array1.Length; i++)
			{
				if (!array1.GetValue(i).Equals(array2.GetValue(i)))
				{
					return false;
				}
			}

			return true;
		}

		#endregion

		#region static method IsPrivateIP

		/// <summary>
		/// Gets if specified IP address is private LAN IP address. For example 192.168.x.x is private ip.
		/// </summary>
		/// <param name="ip">IP address to check.</param>
		/// <returns>Returns true if IP is private IP.</returns>
		/// <exception cref="ArgumentNullException">Is raised when <b>ip</b> is null reference.</exception>
		public static bool IsPrivateIP(IPAddress ip)
		{
			if (ip == null)
			{
				throw new ArgumentNullException(nameof(ip));
			}

			if (ip.AddressFamily == AddressFamily.InterNetwork)
			{
				var ipBytes = ip.GetAddressBytes();

				/* Private IPs:
					First Octet = 192 AND Second Octet = 168 (Example: 192.168.X.X) 
					First Octet = 172 AND (Second Octet >= 16 AND Second Octet <= 31) (Example: 172.16.X.X - 172.31.X.X)
					First Octet = 10 (Example: 10.X.X.X)
					First Octet = 169 AND Second Octet = 254 (Example: 169.254.X.X)

				*/

				if (ipBytes[0] == 192 && ipBytes[1] == 168)
				{
					return true;
				}
				if (ipBytes[0] == 172 && ipBytes[1] >= 16 && ipBytes[1] <= 31)
				{
					return true;
				}
				if (ipBytes[0] == 10)
				{
					return true;
				}
				if (ipBytes[0] == 169 && ipBytes[1] == 254)
				{
					return true;
				}
			}

			return false;
		}

		#endregion

		#region static method CreateSocket

		/// <summary>
		/// Creates new socket for the specified end point.
		/// </summary>
		/// <param name="localEP">Local end point.</param>
		/// <param name="protocolType">Protocol type.</param>
		/// <returns>Return newly created socket.</returns>
		/// <exception cref="ArgumentNullException">Is raised when <b>localEP</b> is null reference.</exception>
		public static Socket CreateSocket(IPEndPoint localEP, ProtocolType protocolType)
		{
			if (localEP == null)
			{
				throw new ArgumentNullException(nameof(localEP));
			}

			var socketType = SocketType.Stream;
			if (protocolType == ProtocolType.Udp)
			{
				socketType = SocketType.Dgram;
			}

			if (localEP.AddressFamily == AddressFamily.InterNetwork)
			{
				var socket = new Socket(AddressFamily.InterNetwork, socketType, protocolType);
				socket.Bind(localEP);

				return socket;
			}

			if (localEP.AddressFamily == AddressFamily.InterNetworkV6)
			{
				var socket = new Socket(AddressFamily.InterNetworkV6, socketType, protocolType);
				socket.Bind(localEP);

				return socket;
			}

			throw new ArgumentException(@"Invalid IPEndPoint address family.");
		}

		#endregion

	}
}