netch/Netch/Controllers/TrojanController.cs

using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Threading;
using Netch.Forms;
using Netch.Models;
using Netch.Utils;
using Newtonsoft.Json;

namespace Netch.Controllers
{
    public class TrojanController : ServerClient
    {
        public TrojanController()
        {
            MainName = "Trojan";
            ready = BeforeStartProgress();
        }

        public override bool Start(Server server, Mode mode)
        {
            MainForm.Instance.StatusText(i18N.Translate("Starting Trojan"));

            File.WriteAllText("data\\last.json", JsonConvert.SerializeObject(new Trojan
            {
                local_addr = Global.Settings.LocalAddress,
                local_port = Global.Settings.Socks5LocalPort,
                remote_addr = server.Hostname,
                remote_port = server.Port,
                password = new List<string>
                {
                    server.Password
                }
            }));

            Instance = MainController.GetProcess("bin\\Trojan.exe");
            Instance.StartInfo.Arguments = "-c ..\\data\\last.json";
            Instance.OutputDataReceived += OnOutputDataReceived;
            Instance.ErrorDataReceived += OnOutputDataReceived;

            State = State.Starting;
            Instance.Start();
            Instance.BeginOutputReadLine();
            Instance.BeginErrorReadLine();
            for (var i = 0; i < 1000; i++)
            {
                Thread.Sleep(10);

                if (State == State.Started) return true;

                if (State == State.Stopped)
                {
                    Logging.Info("Trojan 进程启动失败");

                    Stop();
                    return false;
                }
            }

            Logging.Info("Trojan 进程启动超时");
            Stop();
            return false;
        }

        public override void OnOutputDataReceived(object sender, DataReceivedEventArgs e)
        {
            if (!WriteLog(e)) return;
            if (State == State.Starting)
            {
                if (Instance.HasExited)
                    State = State.Stopped;
                else if (e.Data.Contains("started"))
                    State = State.Started;
                else if (e.Data.Contains("exiting")) State = State.Stopped;
            }
        }
    }
}