254 lines
5.9 KiB
C#

#region License
/*
* HttpRequest.cs
*
* The MIT License
*
* Copyright (c) 2012-2024 sta.blockhead
*
* 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;
using System.IO;
using System.Text;
using WebSocketSharp.Net;
namespace WebSocketSharp
{
internal class HttpRequest : HttpBase
{
#region Private Fields
private CookieCollection _cookies;
private string _method;
private string _target;
#endregion
#region Private Constructors
private HttpRequest (
string method,
string target,
Version version,
NameValueCollection headers
)
: base (version, headers)
{
_method = method;
_target = target;
}
#endregion
#region Internal Constructors
internal HttpRequest (string method, string target)
: this (method, target, HttpVersion.Version11, new NameValueCollection ())
{
Headers["User-Agent"] = "websocket-sharp/1.0";
}
#endregion
#region Internal Properties
internal string RequestLine {
get {
var fmt = "{0} {1} HTTP/{2}{3}";
return String.Format (fmt, _method, _target, ProtocolVersion, CrLf);
}
}
#endregion
#region Public Properties
public AuthenticationResponse AuthenticationResponse {
get {
var val = Headers["Authorization"];
return val != null && val.Length > 0
? AuthenticationResponse.Parse (val)
: 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 override string MessageHeader {
get {
return RequestLine + HeaderSection;
}
}
public string RequestTarget {
get {
return _target;
}
}
#endregion
#region Internal Methods
internal static HttpRequest CreateConnectRequest (Uri targetUri)
{
var fmt = "{0}:{1}";
var host = targetUri.DnsSafeHost;
var port = targetUri.Port;
var authority = String.Format (fmt, host, port);
var ret = new HttpRequest ("CONNECT", authority);
ret.Headers["Host"] = port != 80 ? authority : host;
return ret;
}
internal static HttpRequest CreateWebSocketHandshakeRequest (Uri targetUri)
{
var ret = new HttpRequest ("GET", targetUri.PathAndQuery);
var headers = ret.Headers;
var port = targetUri.Port;
var schm = targetUri.Scheme;
var isDefaultPort = (port == 80 && schm == "ws")
|| (port == 443 && schm == "wss");
headers["Host"] = !isDefaultPort
? targetUri.Authority
: targetUri.DnsSafeHost;
headers["Upgrade"] = "websocket";
headers["Connection"] = "Upgrade";
return ret;
}
internal HttpResponse GetResponse (Stream stream, int millisecondsTimeout)
{
WriteTo (stream);
return HttpResponse.ReadResponse (stream, millisecondsTimeout);
}
internal static HttpRequest Parse (string[] messageHeader)
{
var len = messageHeader.Length;
if (len == 0) {
var msg = "An empty request header.";
throw new ArgumentException (msg);
}
var rlParts = messageHeader[0].Split (new[] { ' ' }, 3);
if (rlParts.Length != 3) {
var msg = "It includes an invalid request line.";
throw new ArgumentException (msg);
}
var method = rlParts[0];
var target = rlParts[1];
var ver = rlParts[2].Substring (5).ToVersion ();
var headers = new WebHeaderCollection ();
for (var i = 1; i < len; i++)
headers.InternalSet (messageHeader[i], false);
return new HttpRequest (method, target, ver, headers);
}
internal static HttpRequest ReadRequest (
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)
continue;
buff.AppendFormat ("{0}; ", cookie);
}
var len = buff.Length;
if (len <= 2)
return;
buff.Length = len - 2;
Headers["Cookie"] = buff.ToString ();
}
#endregion
}
}