HttpRequest.cs

#region License
/*
 * HttpRequest.cs
 *
 * The MIT License
 *
 * Copyright (c) 2012-2015 sta.blockhead
 * Copyright © 2016 Nivloc Enterprises Ltd
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#endregion

#region Contributors
/*
 * Contributors:
 * - David Burhans
 */
#endregion

using System;
using System.Collections.Specialized;
#if SSHARP
using Crestron.SimplSharp.CrestronIO;
#else
using System.IO;
#endif
using System.Text;
using WebSocketSharp.Net;

namespace WebSocketSharp
	{
	internal class HttpRequest : HttpBase
		{
		#region Private Fields

		private CookieCollection _cookies;
		private string _method;
		private string _uri;

		#endregion

		#region Private Constructors

		private HttpRequest (string method, string uri, Version version, NameValueCollection headers)
			: base (version, headers)
			{
			_method = method;
			_uri = uri;
			}

		#endregion

		#region Internal Constructors

		internal HttpRequest (string method, string uri)
			: this (method, uri, HttpVersion.Version11, new NameValueCollection ())
			{
			Headers["User-Agent"] = "websocket-sharp/1.0";
			}

		#endregion

		#region Public Properties

		public AuthenticationResponse AuthenticationResponse
			{
			get
				{
				var res = Headers["Authorization"];
				return res != null && res.Length > 0
					   ? AuthenticationResponse.Parse (res)
					   : null;
				}
			}

		public CookieCollection Cookies
			{
			get
				{
				if (_cookies == null)
					_cookies = Headers.GetCookies (false);

				return _cookies;
				}
			}

		public string HttpMethod
			{
			get
				{
				return _method;
				}
			}

		public bool IsWebSocketRequest
			{
			get
				{
				return _method == "GET"
						 && ProtocolVersion > HttpVersion.Version10
						 && Headers.Upgrades ("websocket");
				}
			}

		public string RequestUri
			{
			get
				{
				return _uri;
				}
			}

		#endregion

		#region Internal Methods

		internal static HttpRequest CreateConnectRequest (Uri uri)
			{
			var host = uri.DnsSafeHost;
			var port = uri.Port;
			var authority = String.Format ("{0}:{1}", host, port);
			var req = new HttpRequest ("CONNECT", authority);
			req.Headers["Host"] = port == 80 ? host : authority;

			return req;
			}

		internal static HttpRequest CreateWebSocketRequest (Uri uri)
			{
			var req = new HttpRequest ("GET", uri.PathAndQuery);

			var headers = req.Headers;

			// Only includes a port number in the Host header value if it's non-default.
			// See: https://tools.ietf.org/html/rfc6455#page-17
			var port = uri.Port;
			var schm = uri.Scheme;
			headers["Host"] = (port == 80 && schm == "ws") || (port == 443 && schm == "wss")
									? uri.DnsSafeHost
									: uri.Authority;

			headers["Upgrade"] = "websocket";
			headers["Connection"] = "Upgrade";

			return req;
			}

		internal HttpResponse GetResponse (Stream stream, int millisecondsTimeout)
			{
			var buff = ToByteArray ();
			stream.Write (buff, 0, buff.Length);

			return Read<HttpResponse> (stream, HttpResponse.Parse, millisecondsTimeout);
			}

		internal static HttpRequest Parse (string[] headerParts)
			{
			var requestLine = headerParts[0].Split (new[] { ' ' }, 3);
			if (requestLine.Length != 3)
				throw new ArgumentException ("Invalid request line: " + headerParts[0]);

			var headers = new WebHeaderCollection ();
			for (int i = 1; i < headerParts.Length; i++)
				headers.InternalSet (headerParts[i], false);

			return new HttpRequest (
			  requestLine[0], requestLine[1], new Version (requestLine[2].Substring (5)), headers);
			}

		internal static HttpRequest Read (Stream stream, int millisecondsTimeout)
			{
			return Read<HttpRequest> (stream, Parse, millisecondsTimeout);
			}

		#endregion

		#region Public Methods

		public void SetCookies (CookieCollection cookies)
			{
			if (cookies == null || cookies.Count == 0)
				return;

			var buff = new StringBuilder (64);
			foreach (var cookie in cookies.Sorted)
				if (!cookie.Expired)
					buff.AppendFormat ("{0}; ", cookie.ToString ());

			var len = buff.Length;
			if (len > 2)
				{
				buff.Length = len - 2;
				Headers["Cookie"] = buff.ToString ();
				}
			}

		public override string ToString ()
			{
			var output = new StringBuilder (64);
			output.AppendFormat ("{0} {1} HTTP/{2}{3}", _method, _uri, ProtocolVersion, CrLf);

			var headers = Headers;
			foreach (var key in headers.AllKeys)
				output.AppendFormat ("{0}: {1}{2}", key, headers[key], CrLf);

			output.Append (CrLf);

			var entity = EntityBody;
			if (entity.Length > 0)
				output.Append (entity);

			return output.ToString ();
			}

		#endregion
		}
	}