345 lines
9.9 KiB
C#
345 lines
9.9 KiB
C#
|
//------------------------------------------------------------------------------
|
|||
|
|
// 此代码版权（除特别声明或在XREF结尾的命名空间的代码）归作者本人若汝棋茗所有
|
|||
|
|
// 源代码使用协议遵循本仓库的开源协议及附加协议，若本仓库没有设置，则按MIT开源协议授权
|
|||
|
|
// CSDN博客：https://blog.csdn.net/qq_40374647
|
|||
|
|
// 哔哩哔哩视频：https://space.bilibili.com/94253567
|
|||
|
|
// Gitee源代码仓库：https://gitee.com/RRQM_Home
|
|||
|
|
// Github源代码仓库：https://github.com/RRQM
|
|||
|
|
// API首页：https://touchsocket.net/
|
|||
|
|
// 交流QQ群：234762506
|
|||
|
|
// 感谢您的下载和使用
|
|||
|
|
//------------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
using System;
|
|||
|
|
using System.Reflection;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using TouchSocket.Core;
|
|||
|
|
using TouchSocket.Sockets;
|
|||
|
|
|
|||
|
|
namespace TouchSocket.Http
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// Http基础头部
|
|||
|
|
/// </summary>
|
|||
|
|
public abstract class HttpBase : BlockReader, IRequestInfo
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 服务器版本
|
|||
|
|
/// </summary>
|
|||
|
|
public static readonly string ServerVersion = Assembly.GetExecutingAssembly().GetName().Version.ToString();
|
|||
|
|
|
|||
|
|
private static readonly byte[] m_rnrnCode = Encoding.UTF8.GetBytes("\r\n\r\n");
|
|||
|
|
|
|||
|
|
private readonly InternalHttpHeader m_headers;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 构造函数
|
|||
|
|
/// </summary>
|
|||
|
|
public HttpBase()
|
|||
|
|
{
|
|||
|
|
this.ReadTimeout = 1000 * 30;
|
|||
|
|
this.m_headers = new InternalHttpHeader();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 能否写入。
|
|||
|
|
/// </summary>
|
|||
|
|
public abstract bool CanWrite { get; }
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 客户端
|
|||
|
|
/// </summary>
|
|||
|
|
public abstract ITcpClientBase Client { get; }
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 内容填充完成
|
|||
|
|
/// </summary>
|
|||
|
|
public bool? ContentComplated { get; protected set; } = null;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 内容长度
|
|||
|
|
/// </summary>
|
|||
|
|
public long ContentLength
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
var contentLength = this.m_headers.Get(HttpHeaders.ContentLength);
|
|||
|
|
if (contentLength.IsNullOrEmpty())
|
|||
|
|
{
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
if (long.TryParse(contentLength, out var value))
|
|||
|
|
{
|
|||
|
|
return value;
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.ContentLength, value.ToString());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 保持连接。
|
|||
|
|
/// <para>
|
|||
|
|
/// 一般的，当是http1.1时，如果没有显式的Connection: close，即返回true。当是http1.0时，如果没有显式的Connection: Keep-Alive，即返回false。
|
|||
|
|
/// </para>
|
|||
|
|
/// </summary>
|
|||
|
|
public bool KeepAlive
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
var keepalive = this.Headers.Get(HttpHeaders.Connection);
|
|||
|
|
if (this.ProtocolVersion == "1.0")
|
|||
|
|
{
|
|||
|
|
if (keepalive.IsNullOrEmpty())
|
|||
|
|
{
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return keepalive.Equals("keep-alive", StringComparison.OrdinalIgnoreCase);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
if (keepalive.IsNullOrEmpty())
|
|||
|
|
{
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return keepalive.Equals("keep-alive", StringComparison.OrdinalIgnoreCase);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (this.ProtocolVersion == "1.0")
|
|||
|
|
{
|
|||
|
|
if (value)
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.Connection, "Keep-Alive");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.Connection, "close");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
if (!value)
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.Connection, "close");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 内容类型
|
|||
|
|
/// </summary>
|
|||
|
|
public string ContentType
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return this.m_headers.Get(HttpHeaders.ContentType);
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.ContentType, value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 允许编码
|
|||
|
|
/// </summary>
|
|||
|
|
public string AcceptEncoding
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return this.m_headers.Get(HttpHeaders.AcceptEncoding);
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.AcceptEncoding, value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 可接受MIME类型
|
|||
|
|
/// </summary>
|
|||
|
|
public string Accept
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return this.m_headers.Get(HttpHeaders.Accept);
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
this.m_headers.Add(HttpHeaders.Accept, value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 传递标识
|
|||
|
|
/// </summary>
|
|||
|
|
public object Flag { get; set; }
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 请求头集合
|
|||
|
|
/// </summary>
|
|||
|
|
public IHttpHeader Headers
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return this.m_headers;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 协议名称，默认HTTP
|
|||
|
|
/// </summary>
|
|||
|
|
public Protocol Protocols { get; set; } = Protocol.Http;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// HTTP协议版本，默认1.1
|
|||
|
|
/// </summary>
|
|||
|
|
public string ProtocolVersion { get; set; } = "1.1";
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 请求行
|
|||
|
|
/// </summary>
|
|||
|
|
public string RequestLine { get; private set; }
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="byteBlock"></param>
|
|||
|
|
/// <param name="length"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public bool ParsingHeader(ByteBlock byteBlock, int length)
|
|||
|
|
{
|
|||
|
|
var index = byteBlock.Buffer.IndexOfFirst(byteBlock.Pos, length, m_rnrnCode);
|
|||
|
|
if (index > 0)
|
|||
|
|
{
|
|||
|
|
var headerLength = index - byteBlock.Pos;
|
|||
|
|
this.ReadHeaders(byteBlock.Buffer, byteBlock.Pos, headerLength);
|
|||
|
|
byteBlock.Pos += headerLength;
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 从Request中持续读取数据。
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="buffer"></param>
|
|||
|
|
/// <param name="offset"></param>
|
|||
|
|
/// <param name="count"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public override int Read(byte[] buffer, int offset, int count)
|
|||
|
|
{
|
|||
|
|
return base.Read(buffer, offset, count);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 从内存中读取
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="buffer"></param>
|
|||
|
|
/// <param name="offset"></param>
|
|||
|
|
/// <param name="length"></param>
|
|||
|
|
public void ReadHeaders(byte[] buffer, int offset, int length)
|
|||
|
|
{
|
|||
|
|
var data = Encoding.UTF8.GetString(buffer, offset, length);
|
|||
|
|
var rows = Regex.Split(data, "\r\n");
|
|||
|
|
|
|||
|
|
//Request URL & Method & Version
|
|||
|
|
this.RequestLine = rows[0];
|
|||
|
|
|
|||
|
|
//Request Headers
|
|||
|
|
this.GetRequestHeaders(rows);
|
|||
|
|
this.LoadHeaderProterties();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 设置一次性内容
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="content"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public abstract void SetContent(byte[] content);
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 获取一次性内容。
|
|||
|
|
/// </summary>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public abstract bool TryGetContent(out byte[] content);
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 持续写入内容。
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="buffer"></param>
|
|||
|
|
/// <param name="offset"></param>
|
|||
|
|
/// <param name="count"></param>
|
|||
|
|
public abstract void WriteContent(byte[] buffer, int offset, int count);
|
|||
|
|
|
|||
|
|
internal bool InternalInput(byte[] buffer, int offset, int length)
|
|||
|
|
{
|
|||
|
|
return this.Input(buffer, offset, length);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="disposing"></param>
|
|||
|
|
protected override void Dispose(bool disposing)
|
|||
|
|
{
|
|||
|
|
if (!this.DisposedValue && this.CanRead)
|
|||
|
|
{
|
|||
|
|
this.TryGetContent(out _);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
base.Dispose(disposing);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 读取信息
|
|||
|
|
/// </summary>
|
|||
|
|
protected abstract void LoadHeaderProterties();
|
|||
|
|
|
|||
|
|
private void GetRequestHeaders(string[] rows)
|
|||
|
|
{
|
|||
|
|
this.m_headers.Clear();
|
|||
|
|
if (rows == null || rows.Length <= 0)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
foreach (var item in rows)
|
|||
|
|
{
|
|||
|
|
var kv = item.SplitFirst(':');
|
|||
|
|
if (kv.Length == 2)
|
|||
|
|
{
|
|||
|
|
var key = kv[0].ToLower();
|
|||
|
|
this.m_headers.Add(key, kv[1]);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 重置Http状态。
|
|||
|
|
/// </summary>
|
|||
|
|
public virtual void Reset()
|
|||
|
|
{
|
|||
|
|
base.ResetBlock();
|
|||
|
|
this.m_headers.Clear();
|
|||
|
|
this.ContentComplated = null;
|
|||
|
|
this.RequestLine = default;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|