sc
/
ForwardingServer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

添加项目文件。

master
Administrator 1 month ago
committed by 忱 沈
parent
0f75ba6bd4
commit
5d6cebc982
7 changed files with 472 additions and 0 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 151
      AsyncHttpServer.cs
  2. 111
      AsyncTcpServer.cs
  3. 142
      AsyncWebSocketServer.cs
  4. 41
      DATA.cs
  5. 15
      ForwardingServer.csproj
  6. 3
      ForwardingServer.slnx
  7. 9
      Program.cs

151
AsyncHttpServer.cs
View File

@ -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();*/
}
}
}
}

111
AsyncTcpServer.cs
View File

@ -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");
}
}
}

142
AsyncWebSocketServer.cs
View File

@ -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();
}
}
}
}

41
DATA.cs
View File

@ -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));
}
}
}

15
ForwardingServer.csproj
View File

@ -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>

3
ForwardingServer.slnx
View File

@ -0,0 +1,3 @@
<Solution>
<Project Path="ForwardingServer.csproj" />
</Solution>

9
Program.cs
View File

@ -0,0 +1,9 @@
using ForwardingServer;
DATA.Data();
await AsyncTcpServer.TcpMain();
await AsyncHttpServer.HttpMain();
//await AsyncWebSocketServer.WebSocketMain();
Console.WriteLine("主程序启动，按任意键停止");
await Task.Run(() =>Console.ReadKey());
Write Preview
Loading…
Cancel
Save