DnscryptProxyService.cs 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. using DNS.Client;
  2. using DNS.Protocol;
  3. using System;
  4. using System.Diagnostics;
  5. using System.Linq;
  6. using System.Threading;
  7. using System.Threading.Tasks;
  8. namespace FastGithub.DnscryptProxy
  9. {
  10. /// <summary>
  11. /// DnscryptProxy服务
  12. /// </summary>
  13. sealed class DnscryptProxyService
  14. {
  15. private const string name = "dnscrypt-proxy";
  16. private const string testDomain = "api.github.com";
  17. private readonly FastGithubConfig fastGithubConfig;
  18. /// <summary>
  19. /// 获取相关进程
  20. /// </summary>
  21. public Process? Process { get; private set; }
  22. /// <summary>
  23. /// 获取服务控制状态
  24. /// </summary>
  25. public ControllState ControllState { get; private set; } = ControllState.None;
  26. /// <summary>
  27. /// DnscryptProxy服务
  28. /// </summary>
  29. /// <param name="fastGithubConfig"></param>
  30. public DnscryptProxyService(FastGithubConfig fastGithubConfig)
  31. {
  32. this.fastGithubConfig = fastGithubConfig;
  33. }
  34. /// <summary>
  35. /// 启动dnscrypt-proxy
  36. /// </summary>
  37. /// <param name="cancellationToken"></param>
  38. /// <returns></returns>
  39. public async Task StartAsync(CancellationToken cancellationToken)
  40. {
  41. this.ControllState = ControllState.Started;
  42. var tomlPath = $"{name}.toml";
  43. await TomlUtil.SetListensAsync(tomlPath, this.fastGithubConfig.PureDns, cancellationToken);
  44. foreach (var process in Process.GetProcessesByName(name))
  45. {
  46. process.Kill();
  47. }
  48. if (OperatingSystem.IsWindows())
  49. {
  50. StartDnscryptProxy("-service install")?.WaitForExit();
  51. StartDnscryptProxy("-service start")?.WaitForExit();
  52. this.Process = Process.GetProcessesByName(name).FirstOrDefault(item => item.SessionId == 0);
  53. }
  54. else
  55. {
  56. this.Process = StartDnscryptProxy(string.Empty);
  57. }
  58. }
  59. /// <summary>
  60. /// 等待dns服务OK
  61. /// </summary>
  62. /// <param name="cancellationToken"></param>
  63. /// <returns></returns>
  64. public async Task WaitForDnsOKAsync(CancellationToken cancellationToken)
  65. {
  66. var process = this.Process;
  67. if (process == null || process.HasExited || this.ControllState != ControllState.Started)
  68. {
  69. return;
  70. }
  71. using var processExitTokenSource = new CancellationTokenSource();
  72. process.EnableRaisingEvents = true;
  73. process.Exited += (s, e) => processExitTokenSource.Cancel();
  74. using var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, processExitTokenSource.Token);
  75. await this.WaitForDnsOKCoreAsync(linkedTokenSource.Token);
  76. }
  77. /// <summary>
  78. /// 等待dns服务OK
  79. /// </summary>
  80. /// <param name="cancellationToken"></param>
  81. /// <returns></returns>
  82. private async Task WaitForDnsOKCoreAsync(CancellationToken cancellationToken)
  83. {
  84. while (true)
  85. {
  86. try
  87. {
  88. using var timeoutTokenSource = new CancellationTokenSource(TimeSpan.FromSeconds(1d));
  89. using var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, timeoutTokenSource.Token);
  90. var dnsClient = new DnsClient(this.fastGithubConfig.PureDns);
  91. await dnsClient.Lookup(testDomain, RecordType.A, linkedTokenSource.Token);
  92. break;
  93. }
  94. catch (Exception)
  95. {
  96. cancellationToken.ThrowIfCancellationRequested();
  97. }
  98. }
  99. }
  100. /// <summary>
  101. /// 停止dnscrypt-proxy
  102. /// </summary>
  103. public void Stop()
  104. {
  105. this.ControllState = ControllState.Stopped;
  106. if (OperatingSystem.IsWindows())
  107. {
  108. StartDnscryptProxy("-service stop")?.WaitForExit();
  109. StartDnscryptProxy("-service uninstall")?.WaitForExit();
  110. }
  111. if (this.Process != null && this.Process.HasExited == false)
  112. {
  113. this.Process.Kill();
  114. }
  115. }
  116. /// <summary>
  117. /// 启动DnscryptProxy进程
  118. /// </summary>
  119. /// <param name="arguments"></param>
  120. private static Process? StartDnscryptProxy(string arguments)
  121. {
  122. return Process.Start(new ProcessStartInfo
  123. {
  124. FileName = OperatingSystem.IsWindows() ? $"{name}.exe" : name,
  125. Arguments = arguments,
  126. UseShellExecute = true,
  127. CreateNoWindow = true,
  128. WindowStyle = ProcessWindowStyle.Hidden
  129. });
  130. }
  131. /// <summary>
  132. /// 转换为字符串
  133. /// </summary>
  134. /// <returns></returns>
  135. public override string ToString()
  136. {
  137. return name;
  138. }
  139. }
  140. }