FastGithub/FastGithub.DomainResolve/DnscryptProxy.cs
2021-08-20 09:39:48 +08:00

113 lines
3.3 KiB
C#

using System;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net;
using System.Threading;
using System.Threading.Tasks;
namespace FastGithub.DomainResolve
{
/// <summary>
/// DnscryptProxy服务
/// </summary>
sealed class DnscryptProxy
{
private const string PATH = "dnscryptproxy";
private const string Name = "dnscrypt-proxy";
/// <summary>
/// 相关进程
/// </summary>
private Process? process;
/// <summary>
/// 获取监听的节点
/// </summary>
public IPEndPoint EndPoint { get; }
/// <summary>
/// DnscryptProxy服务
/// </summary>
/// <param name="endPoint">监听的节点</param>
public DnscryptProxy(IPEndPoint endPoint)
{
this.EndPoint = endPoint;
}
/// <summary>
/// 启动dnscrypt-proxy
/// </summary>
/// <param name="cancellationToken"></param>
/// <returns></returns>
public async Task StartAsync(CancellationToken cancellationToken)
{
var tomlPath = Path.Combine(PATH, $"{Name}.toml");
await TomlUtil.SetListensAsync(tomlPath, this.EndPoint, cancellationToken);
await TomlUtil.SetEdnsClientSubnetAsync(tomlPath, cancellationToken);
foreach (var process in Process.GetProcessesByName(Name))
{
process.Kill();
process.WaitForExit();
}
if (OperatingSystem.IsWindows())
{
StartDnscryptProxy("-service uninstall")?.WaitForExit();
StartDnscryptProxy("-service install")?.WaitForExit();
StartDnscryptProxy("-service start")?.WaitForExit();
this.process = Process.GetProcessesByName(Name).FirstOrDefault(item => item.SessionId == 0);
}
else
{
this.process = StartDnscryptProxy(string.Empty);
}
}
/// <summary>
/// 停止dnscrypt-proxy
/// </summary>
public void Stop()
{
if (OperatingSystem.IsWindows())
{
StartDnscryptProxy("-service stop")?.WaitForExit();
StartDnscryptProxy("-service uninstall")?.WaitForExit();
}
if (this.process != null && this.process.HasExited == false)
{
this.process.Kill();
}
}
/// <summary>
/// 启动DnscryptProxy进程
/// </summary>
/// <param name="arguments"></param>
private static Process? StartDnscryptProxy(string arguments)
{
var fileName = OperatingSystem.IsWindows() ? $"{Name}.exe" : Name;
return Process.Start(new ProcessStartInfo
{
FileName = Path.Combine(PATH, fileName),
Arguments = arguments,
WorkingDirectory = PATH,
UseShellExecute = false,
CreateNoWindow = true,
WindowStyle = ProcessWindowStyle.Hidden
});
}
/// <summary>
/// 转换为字符串
/// </summary>
/// <returns></returns>
public override string ToString()
{
return Name;
}
}
}