7 changed files with 472 additions and 0 deletions
@ -0,0 +1,151 @@ |
|||
using System; |
|||
using System.Collections.Concurrent; |
|||
using System.Collections.Generic; |
|||
using System.Data; |
|||
using System.Text; |
|||
using System.Text.Json; |
|||
using System.Text.RegularExpressions; |
|||
using System.Xml.Linq; |
|||
using TouchSocket.Core; |
|||
using TouchSocket.Http; |
|||
using TouchSocket.Http.WebSockets; |
|||
using TouchSocket.Sockets; |
|||
|
|||
|
|||
namespace ForwardingServer |
|||
{ |
|||
public class AsyncHttpServer |
|||
{ |
|||
public static HttpService httpservice = new HttpService(); |
|||
|
|||
public static async Task HttpMain() |
|||
{ |
|||
await httpservice.SetupAsync(new TouchSocketConfig()//加载配置
|
|||
.SetListenIPHosts(7791) |
|||
.ConfigureContainer(a => |
|||
{ |
|||
a.AddConsoleLogger(); |
|||
}) |
|||
.ConfigurePlugins(a => |
|||
{ |
|||
a.Add<MyHttpPlug1>(); |
|||
//default插件应该最后添加,其作用是
|
|||
//1、为找不到的路由返回404
|
|||
//2、处理 header 为Option的探视跨域请求。
|
|||
a.UseDefaultHttpServicePlugin(); |
|||
})); |
|||
await httpservice.StartAsync(); |
|||
|
|||
Console.WriteLine("HttpService:Run"); |
|||
} |
|||
|
|||
public class MyHttpPlug1 : PluginBase, IHttpPlugin |
|||
{ |
|||
public async Task OnHttpRequest(IHttpSessionClient client, HttpContextEventArgs e) |
|||
{ |
|||
#region HttpContext生命周期
|
|||
//var request = e.Context.Request;//http请求体
|
|||
var response = e.Context.Response;//http响应体
|
|||
#endregion
|
|||
|
|||
var bodyString = await e.Context.Request.GetBodyAsync(); |
|||
|
|||
//获取路径
|
|||
var URL = e.Context.Request.URL; |
|||
|
|||
try |
|||
{ |
|||
string dat; |
|||
DataRow[] rows = DATA.MachineTable.Select("Enterprise='/" + URL + "'"); |
|||
if (rows.Length > 0) |
|||
{ |
|||
string? s_ = rows.First().Field<string?>("ID")!; |
|||
|
|||
if (s_ != null) |
|||
{ |
|||
// var deserialized = JsonSerializer.Deserialize<Dictionary<string, object>>(bodyString)!;
|
|||
|
|||
// if (deserialized.TryGetValue("Terminal", out var val) && val != null) { }
|
|||
// using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(10));
|
|||
|
|||
await AsyncTcpServer.service.SendAsync(s_, bodyString); |
|||
|
|||
DATA.Person deserializedPerson = JsonSerializer.Deserialize<DATA.Person>(bodyString)!; |
|||
|
|||
DATA.UserTable.Rows.Add(client.Id, client.IP, deserializedPerson.User, |
|||
"", deserializedPerson.Terminal, "", ""); |
|||
|
|||
// 创建一个 秒的延时任务
|
|||
Task delayTask = Task.Delay(15000); |
|||
// 创建一个监测任务,不断检查变量
|
|||
Task monitorTask = Task.Run(async () => |
|||
{ |
|||
while (!(DATA.UserTable.Select("Terminal='" + deserializedPerson.Terminal + "'"). |
|||
First().Field<string>("RDAT")!.Length > 1)) |
|||
{ |
|||
await Task.Delay(500); |
|||
} |
|||
}); |
|||
Task winner = await Task.WhenAny(delayTask, monitorTask); |
|||
|
|||
DataRow[] dataRows = DATA.UserTable.Select("Terminal='" + deserializedPerson.Terminal + "'"); |
|||
if (winner == monitorTask) |
|||
{ |
|||
dat = dataRows.First().Field<string>("RDAT")!; |
|||
} |
|||
else { dat = "970"; } |
|||
foreach (DataRow row in dataRows) |
|||
{ |
|||
DATA.UserTable.Rows.Remove(row); |
|||
} |
|||
} |
|||
else { dat = "999"; } |
|||
} |
|||
else |
|||
{ |
|||
dat = "980"; |
|||
} |
|||
|
|||
e.Context.Response.Headers.Add("SUNLIGHT", "SUNLIGHT"); |
|||
e.Context.Response.SetStatus(200, "success"); |
|||
e.Context.Response.SetContent(Encoding.UTF8.GetBytes(dat.ToCharArray())); |
|||
await e.Context.Response.AnswerAsync(); |
|||
} |
|||
catch (TimeoutException) |
|||
{ |
|||
Console.WriteLine("等待TCP服务端响应超时"); |
|||
} |
|||
catch (Exception ex) |
|||
{//错误
|
|||
Console.WriteLine("[" + DateTime.Now.ToString("yyyy-MM-DD HH:mm:ss:fff") + "] Http:" + ex.ToString()); |
|||
//client.SendAsync("Target instruction parsing error manager will close connection");
|
|||
} |
|||
|
|||
/* if (request.IsGet() && request.UrlEquals("/success")) |
|||
{ |
|||
//直接响应文字
|
|||
await response |
|||
.SetStatus(200, "success") |
|||
.FromText("Success") |
|||
.AnswerAsync();//直接回应
|
|||
Console.WriteLine("处理/success"); |
|||
return; |
|||
} |
|||
|
|||
//无法处理,调用下一个插件
|
|||
await e.InvokeNext();*/ |
|||
} |
|||
} |
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
} |
|||
} |
|||
@ -0,0 +1,111 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Data; |
|||
using System.Security.Cryptography.Pkcs; |
|||
using System.Text; |
|||
using System.Text.Json; |
|||
using System.Windows.Markup; |
|||
using System.Xml.Linq; |
|||
using TouchSocket.Core; |
|||
using TouchSocket.Sockets; |
|||
using static System.Runtime.InteropServices.JavaScript.JSType; |
|||
|
|||
namespace ForwardingServer |
|||
{ |
|||
public class AsyncTcpServer |
|||
{ |
|||
public class Personm |
|||
{ |
|||
public required string SERVEREnterprise { get; set; } |
|||
public required string SERVERMachineName { get; set; } |
|||
public required string SERVERNAME { get; set; } |
|||
public required string SERVERUSER { get; set; } |
|||
public required string SERRVERPASWORD { get; set; } |
|||
public required string MachineVer { get; set; } |
|||
} |
|||
public static TcpService service = new TcpService(); |
|||
public static async Task TcpMain() |
|||
{ |
|||
//TcpService service = new TcpService();
|
|||
service.Connecting = (client, e) => { return EasyTask.CompletedTask; };//有客户端正在连接
|
|||
service.Connected = (client, e) => {return EasyTask.CompletedTask;};//有客户端成功连接
|
|||
service.Closing = (client, e) => {return EasyTask.CompletedTask; };//有客户端正在断开连接,只有当主动断开时才有效。
|
|||
service.Closed = (client, e) => { |
|||
//离线记录
|
|||
DataRow[] dat = DATA.MachineTable.Select("ID='"+client.Id+"'"); |
|||
foreach (DataRow row in dat) |
|||
{ |
|||
DATA.MachineTable.Rows.Remove(row); |
|||
} |
|||
Console.WriteLine("[" + DateTime.Now.ToString("yyyy-MM-DD HH:mm:ss:fff") + "] NOT ID " + client.Id + " IP " + client.IP); |
|||
|
|||
return EasyTask.CompletedTask; |
|||
};//有客户端断开连接
|
|||
service.Received = (client, e) =>{ |
|||
string DAT = e.Memory.Span.ToString(Encoding.UTF8); |
|||
try |
|||
{ |
|||
if (DAT.Length > 5) |
|||
{ |
|||
DataRow[] r_ = DATA.MachineTable.Select("ID='" + client.Id + "'"); |
|||
if (r_.Length > 0) |
|||
{ |
|||
int? s_ = r_.First().Field<int?>("Start"); |
|||
|
|||
if (s_ == 30) |
|||
{ |
|||
if (DAT.StartsWith("{") || DAT.StartsWith("[")) |
|||
{ |
|||
DATA.Person deserializedPerson = JsonSerializer.Deserialize<DATA.Person>(DAT)!; |
|||
//转发逻辑
|
|||
var dr = DATA.UserTable.Select("Terminal='" + deserializedPerson.Terminal + "'"); |
|||
if (dr.Length > 0) |
|||
{ |
|||
DataRow drEmployee = dr.First(); |
|||
drEmployee.BeginEdit(); |
|||
drEmployee["RDAT"] = DAT; |
|||
drEmployee.EndEdit(); |
|||
drEmployee.AcceptChanges(); |
|||
} |
|||
} |
|||
|
|||
} |
|||
} |
|||
else |
|||
{ |
|||
Personm deserializedPerson = JsonSerializer.Deserialize<Personm>(DAT)!; |
|||
DATA.MachineTable.Rows.Add(client.Id, client.IP, 30, "/" + deserializedPerson.SERVEREnterprise, |
|||
deserializedPerson.SERVERMachineName, deserializedPerson.MachineVer); |
|||
Console.WriteLine("["+DateTime.Now.ToString("yyyy-MM-DD HH:mm:ss:fff")+"] LINK ID " + client.Id + " IP " + client.IP + "\n DATA =" + DAT); |
|||
} |
|||
} |
|||
} |
|||
catch (Exception ex) |
|||
{//错误
|
|||
Console.WriteLine("Tcp:" + ex.ToString()); |
|||
} |
|||
|
|||
return EasyTask.CompletedTask; |
|||
}; |
|||
|
|||
|
|||
await service.SetupAsync(new TouchSocketConfig()//载入配置
|
|||
// .SetMaxBufferSize(1024 * 1024 * 100)
|
|||
//.SetMinBufferSize(1024 * 1024)
|
|||
.SetSingleStreamDataHandlingAdapter(() => new PeriodPackageAdapter())//以超时周期和包
|
|||
.SetListenIPHosts(new IPHost[] { new IPHost(7794), new IPHost(7795) })//同时监听两个地址
|
|||
.ConfigureContainer(a =>//容器的配置顺序应该在最前面
|
|||
{ |
|||
}) |
|||
.ConfigurePlugins(a => |
|||
{ |
|||
}) |
|||
); |
|||
|
|||
await service.StartAsync();//启动
|
|||
|
|||
Console.WriteLine("TcpServer:Run"); |
|||
} |
|||
|
|||
} |
|||
} |
|||
@ -0,0 +1,142 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Net.Sockets; |
|||
using System.Text; |
|||
using TouchSocket.Core; |
|||
using TouchSocket.Http; |
|||
using TouchSocket.Http.WebSockets; |
|||
using TouchSocket.Sockets; |
|||
using static System.Runtime.InteropServices.JavaScript.JSType; |
|||
|
|||
namespace ForwardingServer |
|||
{ |
|||
public class AsyncWebSocketServer |
|||
{ |
|||
|
|||
public static HttpService service = new HttpService(); |
|||
public static async Task WebSocketMain() |
|||
{ |
|||
//var service = new HttpService();
|
|||
|
|||
var config = new TouchSocketConfig(); |
|||
config.SetListenIPHosts(7792) |
|||
.ConfigureContainer(a => |
|||
{ |
|||
a.AddConsoleLogger(); |
|||
}) |
|||
.ConfigurePlugins(a => |
|||
{ |
|||
//添加WebSocket功能
|
|||
a.UseWebSocket(options => |
|||
{ |
|||
options.SetVerifyConnection(async (client, context) => |
|||
{ |
|||
if (!context.Request.IsUpgrade()) |
|||
{ |
|||
//如果不包含升级协议的header,就直接返回false。
|
|||
return false; |
|||
} |
|||
|
|||
// 在此处添加验证逻辑,例如:匹配特定的URL
|
|||
if (true) // context.Request.UrlEquals("/ws"))
|
|||
{ |
|||
var url = context.Request.URL; |
|||
|
|||
return true; // 允许连接
|
|||
} |
|||
|
|||
//如果url不匹配,则可以直接拒绝。
|
|||
//await context.Response
|
|||
//.SetStatus(403, "url不正确")
|
|||
//.AnswerAsync();
|
|||
return false; |
|||
}); |
|||
|
|||
|
|||
}); |
|||
|
|||
|
|||
a.Add<MyWebSocketReceivePlugin>(); |
|||
a.Add<MyWebSocketClosedPlugin>(); |
|||
}); |
|||
|
|||
//加载配置
|
|||
await service.SetupAsync(config); |
|||
|
|||
//启动服务
|
|||
await service.StartAsync(); |
|||
|
|||
Console.WriteLine("WebSocketService:Run"); |
|||
} |
|||
|
|||
public async Task WebSocketSendAsync(string id,string dat) { |
|||
var clients = service.Clients; |
|||
foreach (var client in clients) |
|||
{ |
|||
if (client.Protocol == Protocol.WebSocket)//先判断是不是websocket协议
|
|||
{ |
|||
if (client.Id == id)//再按指定id发送,或者直接广播发送
|
|||
{ |
|||
var webSocket = client.WebSocket; |
|||
await webSocket.SendAsync(dat); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
|
|||
private class MyWebSocketClosedPlugin : PluginBase, IWebSocketClosedPlugin |
|||
{ |
|||
public async Task OnWebSocketClosed(IWebSocket webSocket, ClosedEventArgs e) |
|||
{ |
|||
var ClientId = (webSocket as IIdClient)!.Id; |
|||
|
|||
} |
|||
} |
|||
private class MyWebSocketReceivePlugin : PluginBase, IWebSocketReceivedPlugin |
|||
{ |
|||
public async Task OnWebSocketReceived(IWebSocket webSocket, WSDataFrameEventArgs e) |
|||
{ |
|||
//处理接收的数据
|
|||
var dataFrame = e.DataFrame; |
|||
switch (dataFrame.Opcode) |
|||
{ |
|||
case WSDataType.Cont: |
|||
//处理中继包
|
|||
break; |
|||
|
|||
case WSDataType.Text: |
|||
//处理文本包
|
|||
Console.WriteLine(dataFrame.ToText()); |
|||
break; |
|||
|
|||
case WSDataType.Binary: |
|||
//处理二进制包
|
|||
var data = dataFrame.PayloadData; |
|||
Console.WriteLine($"收到二进制数据,长度:{data.Length}"); |
|||
break; |
|||
|
|||
case WSDataType.Close: |
|||
//处理关闭包
|
|||
break; |
|||
|
|||
case WSDataType.Ping: |
|||
//处理Ping包
|
|||
break; |
|||
|
|||
case WSDataType.Pong: |
|||
//处理Pong包
|
|||
break; |
|||
|
|||
default: |
|||
//处理其他包
|
|||
break; |
|||
} |
|||
await e.InvokeNext(); |
|||
} |
|||
|
|||
} |
|||
|
|||
|
|||
} |
|||
} |
|||
@ -0,0 +1,41 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Data; |
|||
using System.Text; |
|||
|
|||
namespace ForwardingServer |
|||
{ |
|||
public class DATA |
|||
{ |
|||
public class Person |
|||
{ |
|||
public required string User { get; set; } |
|||
public required string Password { get; set; } |
|||
public required string IP { get; set; } |
|||
public required string Terminal { get; set; } |
|||
public required string Dat { get; set; } |
|||
} |
|||
|
|||
public static DataTable MachineTable = new(); |
|||
public static DataTable UserTable = new(); |
|||
|
|||
public static void Data() |
|||
{ |
|||
MachineTable.Columns.Add("ID", typeof(string)); |
|||
MachineTable.Columns.Add("IP", typeof(string)); |
|||
MachineTable.Columns.Add("Start", typeof(int)); |
|||
MachineTable.Columns.Add("Enterprise", typeof(string)); |
|||
MachineTable.Columns.Add("MachineName", typeof(string)); |
|||
MachineTable.Columns.Add("MachineVer", typeof(string)); |
|||
|
|||
|
|||
UserTable.Columns.Add("ID", typeof(string)); |
|||
UserTable.Columns.Add("IP", typeof(string)); |
|||
UserTable.Columns.Add("User", typeof(string)); |
|||
UserTable.Columns.Add("Enterprise", typeof(string)); |
|||
UserTable.Columns.Add("Terminal", typeof(string)); |
|||
UserTable.Columns.Add("MachineName", typeof(string)); |
|||
UserTable.Columns.Add("RDAT", typeof(string)); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,15 @@ |
|||
<Project Sdk="Microsoft.NET.Sdk"> |
|||
|
|||
<PropertyGroup> |
|||
<OutputType>Exe</OutputType> |
|||
<TargetFramework>net10.0</TargetFramework> |
|||
<ImplicitUsings>enable</ImplicitUsings> |
|||
<Nullable>enable</Nullable> |
|||
</PropertyGroup> |
|||
|
|||
<ItemGroup> |
|||
<PackageReference Include="TouchSocket" Version="4.2.11" /> |
|||
<PackageReference Include="TouchSocket.Http" Version="4.2.11" /> |
|||
</ItemGroup> |
|||
|
|||
</Project> |
|||
@ -0,0 +1,3 @@ |
|||
<Solution> |
|||
<Project Path="ForwardingServer.csproj" /> |
|||
</Solution> |
|||
@ -0,0 +1,9 @@ |
|||
using ForwardingServer; |
|||
|
|||
DATA.Data(); |
|||
|
|||
await AsyncTcpServer.TcpMain(); |
|||
await AsyncHttpServer.HttpMain(); |
|||
//await AsyncWebSocketServer.WebSocketMain();
|
|||
Console.WriteLine("主程序启动,按任意键停止"); |
|||
await Task.Run(() =>Console.ReadKey()); |
|||
Loading…
Reference in new issue