254 lines
5.9 KiB
C#
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
|
|
}
|
|
}
|