434 lines
16 KiB
C#
434 lines
16 KiB
C#
|
using System;
|
|||
|
|
using System.Net;
|
|||
|
|
using System.Net.Sockets;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Threading;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
using Ultron.Proxy.Utils;
|
|||
|
|
using Ultron.Proxy.Interfaces;
|
|||
|
|
|
|||
|
|
namespace Ultron.Proxy
|
|||
|
|
{
|
|||
|
|
//+------------------------+
|
|||
|
|
//| NAT |
|
|||
|
|
//| |
|
|||
|
|
//| |
|
|||
|
|
//| +----------+ | +-----------+
|
|||
|
|
//| | | | | |
|
|||
|
|
//| | client |------------> provider |
|
|||
|
|
//| | | | | |
|
|||
|
|
//| +----+-----+ | +------^----+
|
|||
|
|
//| | | |
|
|||
|
|
//| | | |
|
|||
|
|
//| | | |
|
|||
|
|
//| +----V-----+ | |
|
|||
|
|
//| | | | |
|
|||
|
|
//| | IIS | | |
|
|||
|
|
//| | | | |
|
|||
|
|
//| +----------+ | +------+-------+
|
|||
|
|
//| | | |
|
|||
|
|
//| | | consumer |
|
|||
|
|
//| | | |
|
|||
|
|
//+------------------------+ +--------------+
|
|||
|
|
public class ServerHost
|
|||
|
|
{
|
|||
|
|
//服务端代理转发端口
|
|||
|
|
public static int ClientServicePort = 9973;
|
|||
|
|
//服务端配置通讯端口
|
|||
|
|
public static int ConfigServicePort = 12307;
|
|||
|
|
//远端管理端口
|
|||
|
|
public static int WebManagementPort = 0;
|
|||
|
|
|
|||
|
|
public ClientConnectionManager ConnectionManager = null;
|
|||
|
|
|
|||
|
|
//inject
|
|||
|
|
internal static ILogger Logger;
|
|||
|
|
|
|||
|
|
public ServerHost(ILogger logger)
|
|||
|
|
{
|
|||
|
|
Logger = logger;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//必须设置远程端口才可以通信
|
|||
|
|
public ServerHost SetWebPort(int port)
|
|||
|
|
{
|
|||
|
|
WebManagementPort = port;
|
|||
|
|
return this;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async Task Start()
|
|||
|
|
{
|
|||
|
|
TaskScheduler.UnobservedTaskException += TaskScheduler_UnobservedTaskException;
|
|||
|
|
CancellationTokenSource ctsConfig = new CancellationTokenSource();
|
|||
|
|
CancellationTokenSource ctsHttp = new CancellationTokenSource();
|
|||
|
|
CancellationTokenSource ctsConsumer = new CancellationTokenSource();
|
|||
|
|
|
|||
|
|
//1.反向连接池配置
|
|||
|
|
ConnectionManager = ClientConnectionManager.GetInstance();
|
|||
|
|
//注册客户端发生连接时的事件
|
|||
|
|
ConnectionManager.AppTcpClientMapConfigConnected += ConnectionManager_AppAdded;
|
|||
|
|
Logger.Debug("Ultron.Proxy server started");
|
|||
|
|
|
|||
|
|
//2.开启http服务
|
|||
|
|
if (WebManagementPort > 0)
|
|||
|
|
StartHttpService(ctsHttp);
|
|||
|
|
|
|||
|
|
//3.开启配置服务
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
await StartConfigService(ctsConfig);
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
Logger.Debug(ex.Message);
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
Logger.Debug("all closed");
|
|||
|
|
ctsConfig.Cancel();
|
|||
|
|
//listenerConsumer.Stop();
|
|||
|
|
}
|
|||
|
|
////4.通过已配置的端口集合开启侦听
|
|||
|
|
//foreach (var kv in ConnectionManager.PortAppMap)
|
|||
|
|
//{
|
|||
|
|
// ListenConsumeAsync(kv.Key, ctsConsumer.Token);
|
|||
|
|
//}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void TaskScheduler_UnobservedTaskException(object sender, UnobservedTaskExceptionEventArgs e)
|
|||
|
|
{
|
|||
|
|
Logger.Error(e.Exception.ToString(), e.Exception);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#region HTTPServer
|
|||
|
|
private async Task StartHttpService(CancellationTokenSource ctsHttp)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
HttpListener listener = new HttpListener();
|
|||
|
|
listener.Prefixes.Add($"http://127.0.0.1:{WebManagementPort}/");
|
|||
|
|
//TcpListener listenerConfigService = new TcpListener(IPAddress.Any, WebManagementPort);
|
|||
|
|
Logger.Debug("Listening HTTP request on port " + WebManagementPort.ToString() + "...");
|
|||
|
|
await AcceptHttpRequest(listener, ctsHttp);
|
|||
|
|
}
|
|||
|
|
catch (HttpListenerException ex)
|
|||
|
|
{
|
|||
|
|
Logger.Debug("Please run this program in administrator mode." + ex);
|
|||
|
|
ServerHost.Logger.Error(ex.ToString(), ex);
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
Logger.Debug(ex);
|
|||
|
|
ServerHost.Logger.Error(ex.ToString(), ex);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private async Task AcceptHttpRequest(HttpListener httpService, CancellationTokenSource ctsHttp)
|
|||
|
|
{
|
|||
|
|
httpService.Start();
|
|||
|
|
while (true)
|
|||
|
|
{
|
|||
|
|
var client = await httpService.GetContextAsync();
|
|||
|
|
ProcessHttpRequestAsync(client);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private async Task ProcessHttpRequestAsync(HttpListenerContext context)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
|
|||
|
|
var request = context.Request;
|
|||
|
|
var response = context.Response;
|
|||
|
|
|
|||
|
|
response.ContentEncoding = Encoding.UTF8;
|
|||
|
|
response.ContentType = "text/html;charset=utf-8";
|
|||
|
|
|
|||
|
|
//getJson
|
|||
|
|
StringBuilder json = new StringBuilder("[ ");
|
|||
|
|
foreach (var app in this.ConnectionManager.PortAppMap)
|
|||
|
|
{
|
|||
|
|
json.Append("{ ");
|
|||
|
|
json.Append(KV2Json("port", app.Key)).C();
|
|||
|
|
json.Append(KV2Json("clientId", app.Value.ClientIdAppId.ClientID)).C();
|
|||
|
|
json.Append(KV2Json("appId", app.Value.ClientIdAppId.AppID)).C();
|
|||
|
|
|
|||
|
|
//反向连接
|
|||
|
|
json.Append(KV2Json("revconns"));
|
|||
|
|
json.Append("[ ");
|
|||
|
|
foreach (var reverseClient in app.Value.ReverseClients)
|
|||
|
|
{
|
|||
|
|
json.Append("{ ");
|
|||
|
|
if (reverseClient.Connected)
|
|||
|
|
{
|
|||
|
|
json.Append(KV2Json("lEndPoint", reverseClient.Client.LocalEndPoint.ToString())).C();
|
|||
|
|
json.Append(KV2Json("rEndPoint", reverseClient.Client.RemoteEndPoint.ToString()));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//json.Append(KV2Json("p", c)).C();
|
|||
|
|
//json.Append(KV2Json("port", ca.Key));
|
|||
|
|
json.Append("}");
|
|||
|
|
json.C();
|
|||
|
|
}
|
|||
|
|
json.D();
|
|||
|
|
json.Append("]").C(); ;
|
|||
|
|
|
|||
|
|
//隧道状态
|
|||
|
|
json.Append(KV2Json("tunnels"));
|
|||
|
|
json.Append("[ ");
|
|||
|
|
foreach (var tunnel in app.Value.Tunnels)
|
|||
|
|
{
|
|||
|
|
json.Append("{ ");
|
|||
|
|
if (tunnel.ClientServerClient.Connected)
|
|||
|
|
|
|||
|
|
json.Append(KV2Json("clientServerClient", tunnel.ClientServerClient?.Client.LocalEndPoint.ToString())).C();
|
|||
|
|
if (tunnel.ConsumerClient.Connected)
|
|||
|
|
json.Append(KV2Json("consumerClient", tunnel.ConsumerClient?.Client.LocalEndPoint.ToString())).C();
|
|||
|
|
|
|||
|
|
json.D();
|
|||
|
|
//json.Append(KV2Json("p", c)).C();
|
|||
|
|
//json.Append(KV2Json("port", ca.Key));
|
|||
|
|
json.Append("}");
|
|||
|
|
json.C();
|
|||
|
|
}
|
|||
|
|
json.D();
|
|||
|
|
json.Append("]");
|
|||
|
|
json.Append("}").C();
|
|||
|
|
}
|
|||
|
|
json.D();
|
|||
|
|
json.Append("]");
|
|||
|
|
await response.OutputStream.WriteAsync(HtmlUtil.GetContent(json.ToString()));
|
|||
|
|
//await response.OutputStream.WriteAsync(HtmlUtil.GetContent(request.RawUrl));
|
|||
|
|
response.OutputStream.Close();
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Logger.Error(e.Message, e);
|
|||
|
|
throw;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private string KV2Json(string key)
|
|||
|
|
{
|
|||
|
|
return "\"" + key + "\":";
|
|||
|
|
}
|
|||
|
|
private string KV2Json(string key, object value)
|
|||
|
|
{
|
|||
|
|
return "\"" + key + "\":\"" + value.ToString() + "\"";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#endregion
|
|||
|
|
|
|||
|
|
|
|||
|
|
private async Task StartConfigService(CancellationTokenSource accepting)
|
|||
|
|
{
|
|||
|
|
TcpListener listenerConfigService = new TcpListener(IPAddress.Any, ConfigServicePort);
|
|||
|
|
|
|||
|
|
|
|||
|
|
Logger.Debug("Listening config request on port " + ConfigServicePort.ToString() + "...");
|
|||
|
|
var taskResultConfig = AcceptConfigRequest(listenerConfigService);
|
|||
|
|
|
|||
|
|
await taskResultConfig; //block here to hold open the server
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 有连接连上则开始侦听新的端口
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="sender"></param>
|
|||
|
|
/// <param name="e"></param>
|
|||
|
|
private void ConnectionManager_AppAdded(object sender, AppChangedEventArgs e)
|
|||
|
|
{
|
|||
|
|
Logger.Debug("AppTcpClientMapReverseConnected事件已触发");
|
|||
|
|
int port = 0;
|
|||
|
|
foreach (var kv in ConnectionManager.PortAppMap)
|
|||
|
|
{
|
|||
|
|
if (kv.Value.ClientIdAppId.AppID == e.App.AppID &&
|
|||
|
|
kv.Value.ClientIdAppId.ClientID == e.App.ClientID) port = kv.Key;
|
|||
|
|
}
|
|||
|
|
if (port == 0) throw new Exception("app未注册");
|
|||
|
|
var ct = new CancellationToken();
|
|||
|
|
|
|||
|
|
|
|||
|
|
ListenConsumeAsync(port, ct);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#region 配置
|
|||
|
|
//配置服务，客户端可以通过这个服务接收现有的空闲端口
|
|||
|
|
//accept a config request.
|
|||
|
|
//request:
|
|||
|
|
// 2 1 1
|
|||
|
|
// clientid appid nouse
|
|||
|
|
//
|
|||
|
|
//response:
|
|||
|
|
// 2 1 1 ...N
|
|||
|
|
// clientid appid port
|
|||
|
|
private async Task AcceptConfigRequest(TcpListener listenerConfigService)
|
|||
|
|
{
|
|||
|
|
listenerConfigService.Start(100);
|
|||
|
|
while (true)
|
|||
|
|
{
|
|||
|
|
var client = await listenerConfigService.AcceptTcpClientAsync();
|
|||
|
|
ProcessConfigRequestAsync(client);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private async Task ProcessConfigRequestAsync(TcpClient client)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
//长度固定4个字节
|
|||
|
|
int configRequestLength = 3;
|
|||
|
|
byte[] appRequestBytes = new byte[configRequestLength];
|
|||
|
|
Logger.Debug("config request received.");
|
|||
|
|
var nstream = client.GetStream();
|
|||
|
|
|
|||
|
|
//1.读取配置请求1
|
|||
|
|
int resultByte = await nstream.ReadAsync(appRequestBytes);
|
|||
|
|
Logger.Debug("appRequestBytes received.");
|
|||
|
|
if (resultByte == 0)
|
|||
|
|
{
|
|||
|
|
CloseClient(client);
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
//2.根据配置请求1获取更多配置信息
|
|||
|
|
int appCount = (int)appRequestBytes[2];
|
|||
|
|
byte[] consumerPortBytes = new byte[appCount * 2];
|
|||
|
|
int resultByte2 = await nstream.ReadAsync(consumerPortBytes);
|
|||
|
|
Logger.Debug("consumerPortBytes received.");
|
|||
|
|
if (resultByte2 == 0)
|
|||
|
|
{
|
|||
|
|
CloseClient(client);
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//3.分配配置ID，并且写回给客户端
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
byte[] arrangedIds = ConnectionManager.ArrageConfigIds(appRequestBytes, consumerPortBytes);
|
|||
|
|
Logger.Debug("apprequest arranged");
|
|||
|
|
await nstream.WriteAsync(arrangedIds);
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{ Logger.Debug(ex.ToString()); }
|
|||
|
|
|
|||
|
|
|
|||
|
|
Logger.Debug("arrangedIds written.");
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Logger.Debug(e);
|
|||
|
|
throw;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
#endregion
|
|||
|
|
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 同时侦听来自consumer的链接和到provider的链接
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="consumerlistener"></param>
|
|||
|
|
/// <param name="ct"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
async Task ListenConsumeAsync(int consumerPort, CancellationToken ct)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
var consumerlistener = new TcpListener(IPAddress.Any, consumerPort);
|
|||
|
|
consumerlistener.Start(1000);
|
|||
|
|
//给两个listen，同时监听3端
|
|||
|
|
var clientCounter = 0;
|
|||
|
|
while (!ct.IsCancellationRequested)
|
|||
|
|
{
|
|||
|
|
//目标的代理服务联通了，才去处理consumer端的请求。
|
|||
|
|
Logger.Debug("listening serviceClient....Port:" + consumerPort);
|
|||
|
|
TcpClient consumerClient = await consumerlistener.AcceptTcpClientAsync();
|
|||
|
|
//记录tcp隧道，消费端
|
|||
|
|
TcpTunnel tunnel = new TcpTunnel();
|
|||
|
|
tunnel.ConsumerClient = consumerClient;
|
|||
|
|
ClientConnectionManager.GetInstance().PortAppMap[consumerPort].Tunnels.Add(tunnel);
|
|||
|
|
Logger.Debug("consumer已连接：" + consumerClient.Client.RemoteEndPoint.ToString());
|
|||
|
|
//消费端连接成功,连接
|
|||
|
|
|
|||
|
|
|
|||
|
|
//需要端口
|
|||
|
|
TcpClient s2pClient = await ConnectionManager.GetClient(consumerPort);
|
|||
|
|
//记录tcp隧道，客户端
|
|||
|
|
tunnel.ClientServerClient = s2pClient;
|
|||
|
|
//✳关键过程✳
|
|||
|
|
//连接完之后发送一个字节过去促使客户端建立转发隧道
|
|||
|
|
await s2pClient.GetStream().WriteAsync(new byte[] { 1 }, 0, 1);
|
|||
|
|
clientCounter++;
|
|||
|
|
|
|||
|
|
TcpTransferAsync(consumerlistener, consumerClient, s2pClient, clientCounter, ct);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Logger.Debug(e);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#region datatransfer
|
|||
|
|
//3端互相传输数据
|
|||
|
|
async Task TcpTransferAsync(TcpListener consumerlistener, TcpClient consumerClient, TcpClient providerClient,
|
|||
|
|
int clientIndex,
|
|||
|
|
CancellationToken ct)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
ServerHost.Logger.Debug($"New client ({clientIndex}) connected");
|
|||
|
|
|
|||
|
|
CancellationTokenSource transfering = new CancellationTokenSource();
|
|||
|
|
|
|||
|
|
var providerStream = providerClient.GetStream();
|
|||
|
|
var consumerStream = consumerClient.GetStream();
|
|||
|
|
Task taskC2PLooping = ToStaticTransfer(transfering.Token, consumerStream, providerStream);
|
|||
|
|
Task taskP2CLooping = StreamTransfer(transfering.Token, providerStream, consumerStream);
|
|||
|
|
|
|||
|
|
//任何一端传输中断或者故障，则关闭所有连接
|
|||
|
|
var comletedTask = await Task.WhenAny(taskC2PLooping, taskP2CLooping);
|
|||
|
|
//comletedTask.
|
|||
|
|
Logger.Debug($"Transferring ({clientIndex}) STOPPED");
|
|||
|
|
consumerClient.Close();
|
|||
|
|
providerClient.Close();
|
|||
|
|
transfering.Cancel();
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Logger.Debug(e);
|
|||
|
|
throw;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private async Task StreamTransfer(CancellationToken ct, NetworkStream fromStream, NetworkStream toStream)
|
|||
|
|
{
|
|||
|
|
await fromStream.CopyToAsync(toStream, ct);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private async Task ToStaticTransfer(CancellationToken ct, NetworkStream fromStream, NetworkStream toStream, Func<byte[], Task<bool>> beforeTransferHandle = null)
|
|||
|
|
{
|
|||
|
|
await fromStream.CopyToAsync(toStream, ct);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void CloseClient(TcpClient client)
|
|||
|
|
{
|
|||
|
|
Logger.Debug("invalid request,Closing client:" + client.Client.RemoteEndPoint.ToString());
|
|||
|
|
client.Close();
|
|||
|
|
Logger.Debug("Closed client:" + client.Client.RemoteEndPoint.ToString());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endregion
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|