93 lines
4.3 KiB
C#
93 lines
4.3 KiB
C#
#region copyright
|
||
//------------------------------------------------------------------------------
|
||
// 此代码版权声明为全文件覆盖，如有原作者特别声明，会在下方手动补充
|
||
// 此代码版权（除特别声明外的代码）归作者本人Diego所有
|
||
// 源代码使用协议遵循本仓库的开源协议及附加协议
|
||
// Gitee源代码仓库：https://gitee.com/diego2098/ThingsGateway
|
||
// Github源代码仓库：https://github.com/kimdiego2098/ThingsGateway
|
||
// 使用文档：https://diego2098.gitee.io/thingsgateway-docs/
|
||
// QQ群：605534569
|
||
//------------------------------------------------------------------------------
|
||
#endregion
|
||
|
||
//------------------------------------------------------------------------------
|
||
// 此代码版权（除特别声明或在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首页：http://rrqm_home.gitee.io/touchsocket/
|
||
// 交流QQ群：234762506
|
||
// 感谢您的下载和使用
|
||
//------------------------------------------------------------------------------
|
||
//------------------------------------------------------------------------------
|
||
|
||
namespace ThingsGateway.Foundation.Http.WebSockets
|
||
{
|
||
/// <summary>
|
||
/// WebSocketServerExtensions
|
||
/// </summary>
|
||
public static class WebSocketServerExtension
|
||
{
|
||
/// <summary>
|
||
/// 转化Protocol协议标识为<see cref="Protocol.WebSocket"/>
|
||
/// </summary>
|
||
/// <param name="client"></param>
|
||
/// <param name="httpContext">Http上下文</param>
|
||
public static async Task<bool> SwitchProtocolToWebSocket<TClient>(this TClient client, HttpContext httpContext) where TClient : IHttpSocketClient
|
||
{
|
||
if (client.Protocol == Protocol.WebSocket)
|
||
{
|
||
return true;
|
||
}
|
||
if (client.Protocol == Protocol.Http)
|
||
{
|
||
if (WSTools.TryGetResponse(httpContext.Request, httpContext.Response))
|
||
{
|
||
var args = new HttpContextEventArgs(new HttpContext(httpContext.Request, httpContext.Response))
|
||
{
|
||
IsPermitOperation = true
|
||
};
|
||
await client.PluginsManager.RaiseAsync(nameof(IWebSocketHandshakingPlugin.OnWebSocketHandshaking), client, args).ConfigureAwait(false);
|
||
|
||
if (args.Context.Response.Responsed)
|
||
{
|
||
return false;
|
||
}
|
||
if (args.IsPermitOperation)
|
||
{
|
||
client.SetDataHandlingAdapter(new WebSocketDataHandlingAdapter());
|
||
client.Protocol = Protocol.WebSocket;
|
||
client.SetValue(WebSocketFeature.HandshakedProperty, true);//设置握手状态
|
||
|
||
using (var byteBlock = new ByteBlock())
|
||
{
|
||
args.Context.Response.Build(byteBlock);
|
||
await client.DefaultSendAsync(byteBlock).ConfigureAwait(false);
|
||
}
|
||
await client.PluginsManager.RaiseAsync(nameof(IWebSocketHandshakedPlugin.OnWebSocketHandshaked), client, new HttpContextEventArgs(httpContext))
|
||
.ConfigureAwait(false);
|
||
return true;
|
||
}
|
||
else
|
||
{
|
||
args.Context.Response.SetStatus(403, "Forbidden");
|
||
using (var byteBlock = new ByteBlock())
|
||
{
|
||
args.Context.Response.Build(byteBlock);
|
||
await client.DefaultSendAsync(byteBlock).ConfigureAwait(false);
|
||
}
|
||
|
||
client.Close("主动拒绝WebSocket连接");
|
||
}
|
||
}
|
||
else
|
||
{
|
||
client.Close("WebSocket连接协议不正确");
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
}
|
||
} |