223 lines
8.3 KiB
C#
223 lines
8.3 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.Text;
|
||
using System.Threading.Tasks;
|
||
using TouchSocket.Core;
|
||
using TouchSocket.Sockets;
|
||
|
||
namespace TouchSocket.Http
|
||
{
|
||
/// <summary>
|
||
/// Http客户端数据处理适配器
|
||
/// </summary>
|
||
public class HttpClientDataHandlingAdapter : NormalDataHandlingAdapter
|
||
{
|
||
/// <summary>
|
||
/// 缓存数据，如果需要手动释放，请先判断，然后到调用<see cref="IDisposable.Dispose"/>后，再置空；
|
||
/// </summary>
|
||
protected ByteBlock tempByteBlock;
|
||
|
||
private static readonly byte[] m_rnCode = Encoding.UTF8.GetBytes("\r\n");
|
||
private ITcpClientBase m_client;
|
||
private HttpResponse m_httpResponse;
|
||
|
||
private long m_surLen;
|
||
|
||
private Task m_task;
|
||
|
||
/// <summary>
|
||
/// <inheritdoc/>
|
||
/// </summary>
|
||
public override bool CanSplicingSend => false;
|
||
|
||
/// <inheritdoc/>
|
||
public override void OnLoaded(object owner)
|
||
{
|
||
if (!(owner is ITcpClientBase clientBase))
|
||
{
|
||
throw new Exception($"此适配器必须适用于{nameof(ITcpClientBase)}");
|
||
}
|
||
this.m_client = clientBase;
|
||
base.OnLoaded(owner);
|
||
}
|
||
|
||
/// <summary>
|
||
/// <inheritdoc/>
|
||
/// </summary>
|
||
/// <param name="byteBlock"></param>
|
||
protected override void PreviewReceived(ByteBlock byteBlock)
|
||
{
|
||
if (this.tempByteBlock == null)
|
||
{
|
||
byteBlock.Pos = 0;
|
||
this.Single(byteBlock, false);
|
||
}
|
||
else
|
||
{
|
||
this.tempByteBlock.Write(byteBlock.Buffer, 0, byteBlock.Len);
|
||
var block = this.tempByteBlock;
|
||
this.tempByteBlock = null;
|
||
block.Pos = 0;
|
||
this.Single(block, true);
|
||
}
|
||
}
|
||
|
||
private void Cache(ByteBlock byteBlock)
|
||
{
|
||
if (byteBlock.CanReadLen > 0)
|
||
{
|
||
this.tempByteBlock = new ByteBlock();
|
||
this.tempByteBlock.Write(byteBlock.Buffer, byteBlock.Pos, byteBlock.CanReadLen);
|
||
if (this.tempByteBlock.Len > this.MaxPackageSize)
|
||
{
|
||
this.OnError(default, "缓存的数据长度大于设定值的情况下未收到解析信号", true, true);
|
||
}
|
||
}
|
||
}
|
||
|
||
private FilterResult ReadChunk(ByteBlock byteBlock)
|
||
{
|
||
var position = byteBlock.Pos;
|
||
var index = byteBlock.Buffer.IndexOfFirst(byteBlock.Pos, byteBlock.CanReadLen, m_rnCode);
|
||
if (index > 0)
|
||
{
|
||
var headerLength = index - byteBlock.Pos;
|
||
var hex = Encoding.ASCII.GetString(byteBlock.Buffer, byteBlock.Pos, headerLength - 1);
|
||
var count = hex.ByHexStringToInt32();
|
||
byteBlock.Pos += headerLength + 1;
|
||
|
||
if (count >= 0)
|
||
{
|
||
if (count > byteBlock.CanReadLen)
|
||
{
|
||
byteBlock.Pos = position;
|
||
return FilterResult.Cache;
|
||
}
|
||
|
||
this.m_httpResponse.InternalInput(byteBlock.Buffer, byteBlock.Pos, count);
|
||
byteBlock.Pos += count;
|
||
byteBlock.Pos += 2;
|
||
return FilterResult.GoOn;
|
||
}
|
||
else
|
||
{
|
||
byteBlock.Pos += 2;
|
||
return FilterResult.Success;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
return FilterResult.Cache;
|
||
}
|
||
}
|
||
|
||
private Task RunGoReceived(HttpResponse response)
|
||
{
|
||
return Task.Run(() =>
|
||
{
|
||
this.GoReceived(null, response);
|
||
});
|
||
}
|
||
|
||
private void Single(ByteBlock byteBlock, bool dis)
|
||
{
|
||
try
|
||
{
|
||
while (byteBlock.CanReadLen > 0)
|
||
{
|
||
if (this.m_httpResponse == null)
|
||
{
|
||
this.m_httpResponse = new HttpResponse(this.m_client);
|
||
if (this.m_httpResponse.ParsingHeader(byteBlock, byteBlock.CanReadLen))
|
||
{
|
||
byteBlock.Pos++;
|
||
if (this.m_httpResponse.IsChunk || this.m_httpResponse.ContentLength > byteBlock.CanReadLen)
|
||
{
|
||
this.m_surLen = this.m_httpResponse.ContentLength;
|
||
this.m_task = this.RunGoReceived(this.m_httpResponse);
|
||
}
|
||
else
|
||
{
|
||
byteBlock.Read(out var buffer, (int)this.m_httpResponse.ContentLength);
|
||
this.m_httpResponse.SetContent(buffer);
|
||
this.GoReceived(null, this.m_httpResponse);
|
||
this.m_httpResponse = null;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
this.Cache(byteBlock);
|
||
this.m_httpResponse = null;
|
||
this.m_task?.Wait();
|
||
this.m_task = null;
|
||
return;
|
||
}
|
||
}
|
||
if (this.m_httpResponse != null)
|
||
{
|
||
if (this.m_httpResponse.IsChunk)
|
||
{
|
||
switch (this.ReadChunk(byteBlock))
|
||
{
|
||
case FilterResult.Cache:
|
||
this.Cache(byteBlock);
|
||
return;
|
||
|
||
case FilterResult.Success:
|
||
this.m_httpResponse = null;
|
||
this.m_task?.Wait();
|
||
this.m_task = null;
|
||
break;
|
||
|
||
case FilterResult.GoOn:
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
else if (this.m_surLen > 0)
|
||
{
|
||
if (byteBlock.CanRead)
|
||
{
|
||
var len = (int)Math.Min(this.m_surLen, byteBlock.CanReadLen);
|
||
this.m_httpResponse.InternalInput(byteBlock.Buffer, byteBlock.Pos, len);
|
||
this.m_surLen -= len;
|
||
byteBlock.Pos += len;
|
||
if (this.m_surLen == 0)
|
||
{
|
||
this.m_httpResponse.InternalInput(new byte[0], 0, 0);
|
||
this.m_httpResponse = null;
|
||
this.m_task?.Wait();
|
||
this.m_task = null;
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
this.m_httpResponse = null;
|
||
this.m_task?.Wait();
|
||
this.m_task = null;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
finally
|
||
{
|
||
if (dis)
|
||
{
|
||
byteBlock.Dispose();
|
||
}
|
||
}
|
||
}
|
||
}
|
||
} |