OpenRA/OpenRA.Game/Server/UPnP.cs

#region Copyright & License Information
/*
 * Copyright 2008-2009 http://www.codeproject.com/Members/Harold-Aptroot
 * Source: http://www.codeproject.com/Articles/27992/NAT-Traversal-with-UPnP-in-C
 * This file is licensed under A Public Domain dedication.
 * For more information, see http://creativecommons.org/licenses/publicdomain/
 */
#endregion

using System;
using System.Collections.Generic;
using System.Text;
using System.Net.Sockets;
using System.Net;
using System.Xml;
using System.IO;

namespace UPnP
{
	public class NAT
	{
		static string _serviceUrl;

		public static bool Discover()
		{
			Socket s = new Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp);
			s.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Broadcast, 1);
			s.ReceiveTimeout = 3000; //3 seconds
			string req = "M-SEARCH * HTTP/1.1\r\n" +
			"HOST: 239.255.255.250:1900\r\n" +
			"ST:upnp:rootdevice\r\n" +
			"MAN:\"ssdp:discover\"\r\n" +
			"MX:3\r\n\r\n";
			byte[] data = Encoding.ASCII.GetBytes(req);
			IPEndPoint ipe = new IPEndPoint(IPAddress.Broadcast, 1900);
			byte[] buffer = new byte[0x1000];

			try
			{
				s.SendTo(data, ipe);
				int length = 0;
				do
				{
					length = s.Receive(buffer);

					string resp = Encoding.ASCII.GetString(buffer, 0, length).ToLower();
					if (resp.Contains("upnp:rootdevice"))
					{
						resp = resp.Substring(resp.ToLower().IndexOf("location:") + 9);
						resp = resp.Substring(0, resp.IndexOf("\r")).Trim();
						if (!string.IsNullOrEmpty(_serviceUrl = GetServiceUrl(resp)))
						{
							s.Close();
							return true;
						}
					}
				} while (length > 0);
				s.Close();
				return false;
			}
			catch
			{
				s.Close();
				return false;
			}
		}

		private static String GetServiceUrl(string resp)
		{
			XmlDocument desc = new XmlDocument();
			HttpWebRequest r = (HttpWebRequest)WebRequest.Create(resp);
			r.KeepAlive = false;
			using (WebResponse wres = r.GetResponse())
			{
				using (Stream ress = wres.GetResponseStream())
				{
					desc.Load(ress);
					XmlNamespaceManager nsMgr = new XmlNamespaceManager(desc.NameTable);
					nsMgr.AddNamespace("tns", "urn:schemas-upnp-org:device-1-0");
					XmlNode typen = desc.SelectSingleNode("//tns:device/tns:deviceType/text()", nsMgr);
					if (!typen.Value.Contains("InternetGatewayDevice"))
						return null;
					XmlNode node = desc.SelectSingleNode("//tns:service[tns:serviceType=\"urn:schemas-upnp-org:service:WANIPConnection:1\"]/tns:controlURL/text()", nsMgr);
					if (node == null)
						return null;
					Uri respUri = new Uri(resp);
					Uri combinedUri = new Uri(respUri, node.Value);
					return combinedUri.AbsoluteUri;
				}
			}
		}

		public static bool ForwardPort(int port, ProtocolType protocol, string description)
		{
			if (string.IsNullOrEmpty(_serviceUrl))
				throw new Exception("No UPnP service available or Discover() has not been called");
			string body = String.Format("<u:AddPortMapping xmlns:u=\"urn:schemas-upnp-org:service:WANIPConnection:1\">"+
						    "<NewRemoteHost></NewRemoteHost><NewExternalPort>{0}</NewExternalPort>"+
						    "<NewProtocol>{1}</NewProtocol><NewInternalPort>{0}</NewInternalPort>" +
						    "<NewInternalClient>{2}</NewInternalClient><NewEnabled>1</NewEnabled>" +
						    "<NewPortMappingDescription>{3}</NewPortMappingDescription>"+
						    "<NewLeaseDuration>0</NewLeaseDuration></u:AddPortMapping>",
						    port, protocol.ToString().ToUpper(),Dns.GetHostAddresses(Dns.GetHostName())[0], description);
			if (SOAPRequest(_serviceUrl, body, "AddPortMapping") != null)
				return true;
			else
				return false;
		}

		public static bool DeleteForwardingRule(int port, ProtocolType protocol)
		{
			if (string.IsNullOrEmpty(_serviceUrl))
				throw new Exception("No UPnP service available or Discover() has not been called");
			string body = String.Format("<u:DeletePortMapping xmlns:u=\"urn:schemas-upnp-org:service:WANIPConnection:1\">" +
						    "<NewRemoteHost></NewRemoteHost><NewExternalPort>{0}</NewExternalPort>"+
						    "<NewProtocol>{1}</NewProtocol></u:DeletePortMapping>", port, protocol.ToString().ToUpper() );
			if (SOAPRequest(_serviceUrl, body, "DeletePortMapping") != null)
				return true;
			else
				return false;
		}

		public static IPAddress GetExternalIP()
		{
			if (string.IsNullOrEmpty(_serviceUrl))
				throw new Exception("No UPnP service available or Discover() has not been called");
			XmlDocument xdoc = SOAPRequest(_serviceUrl, "<u:GetExternalIPAddress xmlns:u=\"urn:schemas-upnp-org:service:WANIPConnection:1\">" +
						       "</u:GetExternalIPAddress>", "GetExternalIPAddress");
			XmlNamespaceManager nsMgr = new XmlNamespaceManager(xdoc.NameTable);
			nsMgr.AddNamespace("tns", "urn:schemas-upnp-org:device-1-0");
			string IP = xdoc.SelectSingleNode("//NewExternalIPAddress/text()", nsMgr).Value;
			return IPAddress.Parse(IP);
		}

		private static XmlDocument SOAPRequest(string url, string soap, string function)
		{
			string body = "<?xml version=\"1.0\"?>" +
				      "<s:Envelope xmlns:s=\"http://schemas.xmlsoap.org/soap/envelope/\" s:encodingStyle=\"http://schemas.xmlsoap.org/soap/encoding/\">" +
				      "<s:Body>" +
				      soap +
				      "</s:Body>" +
				      "</s:Envelope>";
			HttpWebRequest r = (HttpWebRequest)WebRequest.Create(url);
			r.KeepAlive = false;
			r.Method = "POST";
			byte[] b = Encoding.UTF8.GetBytes(body);
			r.Headers.Add("SOAPACTION", "\"urn:schemas-upnp-org:service:WANIPConnection:1#" + function + "\"");
			r.ContentType = "text/xml; charset=\"utf-8\"";
			r.ContentLength = b.Length;
			Stream newStream = r.GetRequestStream();
			newStream.Write(b, 0, b.Length);
			XmlDocument resp = new XmlDocument();
			using (WebResponse wres = r.GetResponse())
			{
				using (Stream ress = wres.GetResponseStream())
				{
					resp.Load(ress);
					return resp;
				}
			}
		}
	}
}