115 lines
4.2 KiB
C#
115 lines
4.2 KiB
C#
using FastGithub.Configuration;
|
||
using FastGithub.ReverseProxy;
|
||
using Microsoft.AspNetCore.Connections;
|
||
using Microsoft.AspNetCore.Hosting;
|
||
using Microsoft.AspNetCore.Server.Kestrel.Core;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Logging;
|
||
using System;
|
||
using System.Net;
|
||
|
||
namespace FastGithub
|
||
{
|
||
/// <summary>
|
||
/// Kestrel扩展
|
||
/// </summary>
|
||
public static class KestrelServerOptionsExtensions
|
||
{
|
||
/// <summary>
|
||
/// 无限制
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
public static void NoLimit(this KestrelServerOptions kestrel)
|
||
{
|
||
kestrel.Limits.MaxRequestBodySize = null;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 监听http代理
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
public static void ListenHttpProxy(this KestrelServerOptions kestrel)
|
||
{
|
||
var httpPort = kestrel.ApplicationServices.GetRequiredService<PortService>().HttpProxyPort;
|
||
if (LocalMachine.CanListenTcp(httpPort) == false)
|
||
{
|
||
throw new FastGithubException("tcp端口{httpsPort}已经被其它进程占用，请在配置文件更换一个端口");
|
||
}
|
||
|
||
kestrel.Listen(IPAddress.Any, httpPort);
|
||
kestrel.GetLogger().LogInformation($"已监听tcp端口{httpPort}，http代理启动完成");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 监听ssh反向代理
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
public static void ListenSshReverseProxy(this KestrelServerOptions kestrel)
|
||
{
|
||
const int SSH_PORT = 22;
|
||
if (LocalMachine.CanListenTcp(SSH_PORT) == true)
|
||
{
|
||
kestrel.Listen(IPAddress.Any, SSH_PORT, listen => listen.UseConnectionHandler<SshReverseProxyHandler>());
|
||
kestrel.GetLogger().LogInformation($"已监听tcp端口{SSH_PORT}，github的ssh代理启动完成");
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 监听http反向代理
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
public static void ListenHttpReverseProxy(this KestrelServerOptions kestrel)
|
||
{
|
||
const int HTTP_PORT = 80;
|
||
if (LocalMachine.CanListenTcp(HTTP_PORT) == true)
|
||
{
|
||
kestrel.Listen(IPAddress.Any, HTTP_PORT);
|
||
kestrel.GetLogger().LogInformation($"已监听tcp端口{HTTP_PORT}，http反向代理启动完成");
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 监听https反向代理
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
/// <exception cref="FastGithubException"></exception>
|
||
public static void ListenHttpsReverseProxy(this KestrelServerOptions kestrel)
|
||
{
|
||
var httpsPort = kestrel.ApplicationServices.GetRequiredService<PortService>().HttpsReverseProxyPort;
|
||
if (OperatingSystem.IsWindows())
|
||
{
|
||
TcpTable.KillPortOwner(httpsPort);
|
||
}
|
||
|
||
if (LocalMachine.CanListenTcp(httpsPort) == false)
|
||
{
|
||
throw new FastGithubException($"tcp端口{httpsPort}已经被其它进程占用");
|
||
}
|
||
|
||
var certService = kestrel.ApplicationServices.GetRequiredService<CertService>();
|
||
certService.CreateCaCertIfNotExists();
|
||
certService.InstallAndTrustCaCert();
|
||
|
||
kestrel.Listen(IPAddress.Any, httpsPort,
|
||
listen => listen.UseHttps(https =>
|
||
https.ServerCertificateSelector = (ctx, domain) =>
|
||
certService.GetOrCreateServerCert(domain)));
|
||
|
||
var logger = kestrel.GetLogger();
|
||
logger.LogInformation($"已监听tcp端口{httpsPort}，https反向代理启动完成");
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 获取日志
|
||
/// </summary>
|
||
/// <param name="kestrel"></param>
|
||
/// <returns></returns>
|
||
private static ILogger GetLogger(this KestrelServerOptions kestrel)
|
||
{
|
||
var loggerFactory = kestrel.ApplicationServices.GetRequiredService<ILoggerFactory>();
|
||
return loggerFactory.CreateLogger($"{nameof(FastGithub)}.{nameof(ReverseProxy)}");
|
||
}
|
||
}
|
||
}
|