netch/Netch/Controllers/NTTController.cs

using System;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using Netch.Interfaces;
using Netch.Utils;
using Serilog;

namespace Netch.Controllers
{
    public class NTTController : Guard, IController
    {
        public NTTController() : base("NTT.exe")
        {
        }

        public override string Name => "NTT";

        /// <summary>
        ///     启动 NatTypeTester
        /// </summary>
        /// <returns></returns>
        public async Task<(string? result, string? localEnd, string? publicEnd)> StartAsync()
        {
            string? localEnd = null, publicEnd = null, result = null, bindingTest = null;

            try
            {
                Instance.StartInfo.Arguments = $" {Global.Settings.STUN_Server} {Global.Settings.STUN_Server_Port}";
                Instance.Start();

                var output = await Instance.StandardOutput.ReadToEndAsync();
                var error = await Instance.StandardError.ReadToEndAsync();

                try
                {
                    await File.WriteAllTextAsync(Path.Combine(Global.NetchDir, $"logging\\{Name}.log"), $"{output}\r\n{error}");
                }
                catch (Exception e)
                {
                    Log.Warning(e, "写入 {Name} 日志错误", Name);
                }

                if (output.IsNullOrWhiteSpace())
                {
                    if (error.IsNullOrWhiteSpace())
                    {
                        Log.Warning("NTT no output");
                        return (null, null, null);
                    }

                    var errorFirst = error.GetLines().First();
                    return (errorFirst.SplitTrimEntries(':').Last(), null, null);
                }

                foreach (var line in output.Split('\n'))
                {
                    var str = line.SplitTrimEntries(':');
                    if (str.Length < 2)
                        continue;

                    var key = str[0];
                    var value = str[1];
                    switch (key)
                    {
                        case "Other address is":
                        case "Nat mapping behavior":
                        case "Nat filtering behavior":
                            break;
                        case "Binding test":
                            bindingTest = value;
                            break;
                        case "Local address":
                            localEnd = value;
                            break;
                        case "Mapped address":
                            publicEnd = value;
                            break;
                        case "result":
                            result = value;
                            break;
                    }
                }

                if (bindingTest == "Fail")
                    result = "Fail";

                return (result, localEnd, publicEnd);
            }
            catch (Exception e)
            {
                Log.Error(e, "{Name} 控制器启动异常", Name);
                try
                {
                    await StopAsync();
                }
                catch
                {
                    // ignored
                }

                return (null, null, null);
            }
        }
    }
}