367 lines
14 KiB
C#
367 lines
14 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.Threading;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
using TouchSocket.Core;
|
|||
|
|
using TouchSocket.Resources;
|
|||
|
|
using TouchSocket.Sockets;
|
|||
|
|
|
|||
|
|
namespace TouchSocket.Http
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// Http客户端
|
|||
|
|
/// </summary>
|
|||
|
|
public class HttpClient : HttpClientBase
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// Http客户端基类
|
|||
|
|
/// </summary>
|
|||
|
|
public class HttpClientBase : TcpClientBase, IHttpClient
|
|||
|
|
{
|
|||
|
|
#region 字段
|
|||
|
|
|
|||
|
|
private readonly SemaphoreSlim m_semaphoreForRequest = new SemaphoreSlim(1, 1);
|
|||
|
|
private readonly WaitData<HttpResponse> m_waitData = new WaitData<HttpResponse>();
|
|||
|
|
private readonly WaitDataAsync<HttpResponse> m_waitDataAsync = new WaitDataAsync<HttpResponse>();
|
|||
|
|
private bool m_getContent;
|
|||
|
|
|
|||
|
|
#endregion 字段
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public override void Connect(int millisecondsTimeout, CancellationToken token)
|
|||
|
|
{
|
|||
|
|
if (this.Config.GetValue(HttpConfigExtensions.HttpProxyProperty) is HttpProxy httpProxy)
|
|||
|
|
{
|
|||
|
|
var proxyHost = httpProxy.Host;
|
|||
|
|
var credential = httpProxy.Credential;
|
|||
|
|
var remoteHost = this.Config.GetValue(TouchSocketConfigExtension.RemoteIPHostProperty);
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
this.Config.SetRemoteIPHost(proxyHost);
|
|||
|
|
base.Connect(millisecondsTimeout, token);
|
|||
|
|
var request = new HttpRequest();
|
|||
|
|
request.InitHeaders()
|
|||
|
|
.SetHost(remoteHost.Host)
|
|||
|
|
.SetProxyHost(remoteHost.Host)
|
|||
|
|
.AsMethod("CONNECT");
|
|||
|
|
var response = this.Request(request, false, millisecondsTimeout, token);
|
|||
|
|
if (response.IsProxyAuthenticationRequired)
|
|||
|
|
{
|
|||
|
|
if (credential is null)
|
|||
|
|
{
|
|||
|
|
throw new Exception("未指定代理的凭据。");
|
|||
|
|
}
|
|||
|
|
var authHeader = response.Headers.Get(HttpHeaders.ProxyAuthenticate);
|
|||
|
|
if (authHeader.IsNullOrEmpty())
|
|||
|
|
{
|
|||
|
|
throw new Exception("未指定代理身份验证质询。");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var ares = new AuthenticationChallenge(authHeader, credential);
|
|||
|
|
|
|||
|
|
request.Headers.Add(HttpHeaders.ProxyAuthorization, ares.ToString());
|
|||
|
|
if (!response.KeepAlive)
|
|||
|
|
{
|
|||
|
|
base.Close("代理要求关闭连接，随后重写连接。");
|
|||
|
|
base.Connect(millisecondsTimeout, token);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = this.Request(request, false, millisecondsTimeout, token);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (response.StatusCode != 200)
|
|||
|
|
{
|
|||
|
|
throw new Exception(response.StatusMessage);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.Config.SetRemoteIPHost(remoteHost);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
base.Connect(millisecondsTimeout, token);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public override async Task ConnectAsync(int millisecondsTimeout, CancellationToken token)
|
|||
|
|
{
|
|||
|
|
if (this.Config.GetValue(HttpConfigExtensions.HttpProxyProperty) is HttpProxy httpProxy)
|
|||
|
|
{
|
|||
|
|
var proxyHost = httpProxy.Host;
|
|||
|
|
var credential = httpProxy.Credential;
|
|||
|
|
var remoteHost = this.Config.GetValue(TouchSocketConfigExtension.RemoteIPHostProperty);
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
this.Config.SetRemoteIPHost(proxyHost);
|
|||
|
|
await base.ConnectAsync(millisecondsTimeout, token);
|
|||
|
|
var request = new HttpRequest();
|
|||
|
|
request.InitHeaders()
|
|||
|
|
.SetHost(remoteHost.Host)
|
|||
|
|
.SetProxyHost(remoteHost.Host)
|
|||
|
|
.AsMethod("CONNECT");
|
|||
|
|
var response = await this.RequestAsync(request, false, millisecondsTimeout, token);
|
|||
|
|
if (response.IsProxyAuthenticationRequired)
|
|||
|
|
{
|
|||
|
|
if (credential is null)
|
|||
|
|
{
|
|||
|
|
throw new Exception("未指定代理的凭据。");
|
|||
|
|
}
|
|||
|
|
var authHeader = response.Headers.Get(HttpHeaders.ProxyAuthenticate);
|
|||
|
|
if (authHeader.IsNullOrEmpty())
|
|||
|
|
{
|
|||
|
|
throw new Exception("未指定代理身份验证质询。");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var ares = new AuthenticationChallenge(authHeader, credential);
|
|||
|
|
|
|||
|
|
request.Headers.Add(HttpHeaders.ProxyAuthorization, ares.ToString());
|
|||
|
|
if (!response.KeepAlive)
|
|||
|
|
{
|
|||
|
|
base.Close("代理要求关闭连接，随后重写连接。");
|
|||
|
|
await base.ConnectAsync(millisecondsTimeout, token);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response = await this.RequestAsync(request, millisecondsTimeout: millisecondsTimeout);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (response.StatusCode != 200)
|
|||
|
|
{
|
|||
|
|
throw new Exception(response.StatusMessage);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.Config.SetRemoteIPHost(remoteHost);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
await base.ConnectAsync(millisecondsTimeout, token);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public HttpResponse Request(HttpRequest request, bool onlyRequest = false, int millisecondsTimeout = 10 * 1000, CancellationToken token = default)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Wait(token);
|
|||
|
|
this.m_getContent = false;
|
|||
|
|
using (var byteBlock = new ByteBlock())
|
|||
|
|
{
|
|||
|
|
request.Build(byteBlock);
|
|||
|
|
|
|||
|
|
this.Reset(token);
|
|||
|
|
this.DefaultSend(byteBlock);
|
|||
|
|
if (onlyRequest)
|
|||
|
|
{
|
|||
|
|
return default;
|
|||
|
|
}
|
|||
|
|
switch (this.m_waitData.Wait(millisecondsTimeout))
|
|||
|
|
{
|
|||
|
|
case WaitDataStatus.SetRunning:
|
|||
|
|
return this.m_waitData.WaitResult;
|
|||
|
|
|
|||
|
|
case WaitDataStatus.Overtime:
|
|||
|
|
throw new TimeoutException(TouchSocketHttpResource.Overtime.GetDescription());
|
|||
|
|
case WaitDataStatus.Canceled:
|
|||
|
|
throw new OperationCanceledException();
|
|||
|
|
case WaitDataStatus.Default:
|
|||
|
|
case WaitDataStatus.Disposed:
|
|||
|
|
default:
|
|||
|
|
throw new Exception(TouchSocketHttpResource.UnknownError.GetDescription());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Release();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public async Task<HttpResponse> RequestAsync(HttpRequest request, bool onlyRequest = false, int millisecondsTimeout = 10 * 1000, CancellationToken token = default)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
await this.m_semaphoreForRequest.WaitAsync(millisecondsTimeout, token);
|
|||
|
|
this.m_getContent = false;
|
|||
|
|
using (var byteBlock = new ByteBlock())
|
|||
|
|
{
|
|||
|
|
request.Build(byteBlock);
|
|||
|
|
|
|||
|
|
this.Reset(token);
|
|||
|
|
await this.DefaultSendAsync(byteBlock);
|
|||
|
|
if (onlyRequest)
|
|||
|
|
{
|
|||
|
|
return default;
|
|||
|
|
}
|
|||
|
|
switch (await this.m_waitDataAsync.WaitAsync(millisecondsTimeout))
|
|||
|
|
{
|
|||
|
|
case WaitDataStatus.SetRunning:
|
|||
|
|
return this.m_waitDataAsync.WaitResult;
|
|||
|
|
|
|||
|
|
case WaitDataStatus.Overtime:
|
|||
|
|
throw new TimeoutException(TouchSocketHttpResource.Overtime.GetDescription());
|
|||
|
|
case WaitDataStatus.Canceled:
|
|||
|
|
throw new OperationCanceledException();
|
|||
|
|
case WaitDataStatus.Default:
|
|||
|
|
case WaitDataStatus.Disposed:
|
|||
|
|
default:
|
|||
|
|
throw new Exception(TouchSocketHttpResource.UnknownError.GetDescription());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Release();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public HttpResponse RequestContent(HttpRequest request, bool onlyRequest = false, int millisecondsTimeout = 10 * 1000, CancellationToken token = default)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Wait(token);
|
|||
|
|
this.m_getContent = true;
|
|||
|
|
using (var byteBlock = new ByteBlock())
|
|||
|
|
{
|
|||
|
|
request.Build(byteBlock);
|
|||
|
|
|
|||
|
|
this.Reset(token);
|
|||
|
|
|
|||
|
|
this.DefaultSend(byteBlock);
|
|||
|
|
if (onlyRequest)
|
|||
|
|
{
|
|||
|
|
return default;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return this.m_waitData.Wait(millisecondsTimeout) switch
|
|||
|
|
{
|
|||
|
|
WaitDataStatus.SetRunning => this.m_waitData.WaitResult,
|
|||
|
|
WaitDataStatus.Overtime => throw new TimeoutException(TouchSocketHttpResource.Overtime.GetDescription()),
|
|||
|
|
WaitDataStatus.Canceled => throw new OperationCanceledException(),
|
|||
|
|
_ => throw new Exception(TouchSocketHttpResource.UnknownError.GetDescription()),
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Release();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public async Task<HttpResponse> RequestContentAsync(HttpRequest request, bool onlyRequest = false, int millisecondsTimeout = 10 * 1000, CancellationToken token = default)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
await this.m_semaphoreForRequest.WaitAsync(millisecondsTimeout, token);
|
|||
|
|
this.m_getContent = true;
|
|||
|
|
using (var byteBlock = new ByteBlock())
|
|||
|
|
{
|
|||
|
|
request.Build(byteBlock);
|
|||
|
|
|
|||
|
|
this.Reset(token);
|
|||
|
|
|
|||
|
|
await this.DefaultSendAsync(byteBlock);
|
|||
|
|
if (onlyRequest)
|
|||
|
|
{
|
|||
|
|
return default;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return await this.m_waitDataAsync.WaitAsync(millisecondsTimeout) switch
|
|||
|
|
{
|
|||
|
|
WaitDataStatus.SetRunning => this.m_waitData.WaitResult,
|
|||
|
|
WaitDataStatus.Overtime => throw new TimeoutException(TouchSocketHttpResource.Overtime.GetDescription()),
|
|||
|
|
WaitDataStatus.Canceled => throw new OperationCanceledException(),
|
|||
|
|
_ => throw new Exception(TouchSocketHttpResource.UnknownError.GetDescription()),
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
this.m_semaphoreForRequest.Release();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="disposing"></param>
|
|||
|
|
protected override void Dispose(bool disposing)
|
|||
|
|
{
|
|||
|
|
if (disposing)
|
|||
|
|
{
|
|||
|
|
this.m_waitData.SafeDispose();
|
|||
|
|
this.m_waitDataAsync.SafeDispose();
|
|||
|
|
}
|
|||
|
|
base.Dispose(disposing);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="e"></param>
|
|||
|
|
protected override async Task OnConnecting(ConnectingEventArgs e)
|
|||
|
|
{
|
|||
|
|
this.Protocol = Protocol.Http;
|
|||
|
|
this.SetDataHandlingAdapter(new HttpClientDataHandlingAdapter());
|
|||
|
|
await base.OnConnecting(e);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
protected override async Task OnDisconnected(DisconnectEventArgs e)
|
|||
|
|
{
|
|||
|
|
this.m_waitData.Cancel();
|
|||
|
|
this.m_waitDataAsync.Cancel();
|
|||
|
|
await base.OnDisconnected(e);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
protected override Task ReceivedData(ReceivedDataEventArgs e)
|
|||
|
|
{
|
|||
|
|
if (e.RequestInfo is HttpResponse response)
|
|||
|
|
{
|
|||
|
|
if (this.m_getContent)
|
|||
|
|
{
|
|||
|
|
response.TryGetContent(out _);
|
|||
|
|
}
|
|||
|
|
this.Set(response);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return base.ReceivedData(e);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Reset(CancellationToken token)
|
|||
|
|
{
|
|||
|
|
this.m_waitData.Reset();
|
|||
|
|
this.m_waitDataAsync.Reset();
|
|||
|
|
this.m_waitData.SetCancellationToken(token);
|
|||
|
|
this.m_waitDataAsync.SetCancellationToken(token);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Set(HttpResponse response)
|
|||
|
|
{
|
|||
|
|
this.m_waitData.Set(response);
|
|||
|
|
this.m_waitDataAsync.Set(response);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|