mirror of
https://github.com/netchx/netch.git
synced 2026-03-14 17:43:18 +08:00
383 lines
14 KiB
C#
383 lines
14 KiB
C#
using Netch.Forms;
|
||
using System;
|
||
using System.Diagnostics;
|
||
using System.IO;
|
||
using System.ServiceProcess;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace Netch.Controllers
|
||
{
|
||
public class NFController
|
||
{
|
||
/// <summary>
|
||
/// 流量变动事件
|
||
/// </summary>
|
||
public event BandwidthUpdateHandler OnBandwidthUpdated;
|
||
|
||
/// <summary>
|
||
/// 流量变动处理器
|
||
/// </summary>
|
||
/// <param name="upload">上传</param>
|
||
/// <param name="download">下载</param>
|
||
public delegate void BandwidthUpdateHandler(long upload, long download);
|
||
|
||
/// <summary>
|
||
/// 进程实例
|
||
/// </summary>
|
||
public Process Instance;
|
||
|
||
/// <summary>
|
||
/// 当前状态
|
||
/// </summary>
|
||
public Models.State State = Models.State.Waiting;
|
||
|
||
// 生成驱动文件路径
|
||
public string driverPath = string.Format("{0}\\drivers\\netfilter2.sys", Environment.SystemDirectory);
|
||
|
||
/// <summary>
|
||
/// 启动
|
||
/// </summary>
|
||
/// <param name="server">服务器</param>
|
||
/// <param name="mode">模式</param>
|
||
/// <param name="StopServiceAndRestart">先停止驱动服务再重新启动</param>
|
||
/// <returns>是否成功</returns>
|
||
public bool Start(Models.Server server, Models.Mode mode, bool StopServiceAndRestart)
|
||
{
|
||
if (!StopServiceAndRestart)
|
||
MainForm.Instance.StatusText($"{Utils.i18N.Translate("Status")}{Utils.i18N.Translate(": ")}{Utils.i18N.Translate("Starting Redirector")}");
|
||
|
||
// 检查驱动是否存在
|
||
if (File.Exists(driverPath))
|
||
{
|
||
// 生成系统版本
|
||
var version = $"{Environment.OSVersion.Version.Major.ToString()}.{Environment.OSVersion.Version.Minor.ToString()}";
|
||
var driverName = "";
|
||
|
||
switch (version)
|
||
{
|
||
case "10.0":
|
||
driverName = "Win-10.sys";
|
||
break;
|
||
case "6.3":
|
||
case "6.2":
|
||
driverName = "Win-8.sys";
|
||
break;
|
||
case "6.1":
|
||
case "6.0":
|
||
driverName = "Win-7.sys";
|
||
break;
|
||
default:
|
||
Utils.Logging.Info($"不支持的系统版本：{version}");
|
||
return false;
|
||
}
|
||
|
||
//检查驱动版本号
|
||
FileVersionInfo SystemfileVerInfo = System.Diagnostics.FileVersionInfo.GetVersionInfo(driverPath);
|
||
FileVersionInfo BinFileVerInfo = System.Diagnostics.FileVersionInfo.GetVersionInfo(string.Format("bin\\{0}", driverName));
|
||
|
||
if (!SystemfileVerInfo.FileVersion.Equals(BinFileVerInfo.FileVersion))
|
||
{
|
||
Utils.Logging.Info("开始更新驱动");
|
||
//需要更新驱动
|
||
try
|
||
{
|
||
var service = new ServiceController("netfilter2");
|
||
if (service.Status == ServiceControllerStatus.Running)
|
||
{
|
||
service.Stop();
|
||
service.WaitForStatus(ServiceControllerStatus.Stopped);
|
||
}
|
||
nfapinet.NFAPI.nf_unRegisterDriver("netfilter2");
|
||
|
||
//删除老驱动
|
||
File.Delete(driverPath);
|
||
if (!InstallDriver())
|
||
return false;
|
||
|
||
Utils.Logging.Info($"驱动更新完毕，当前驱动版本:{BinFileVerInfo.FileVersion}");
|
||
}
|
||
catch (Exception)
|
||
{
|
||
Utils.Logging.Info($"更新驱动出错");
|
||
}
|
||
|
||
}
|
||
|
||
}
|
||
else
|
||
{
|
||
if (!InstallDriver())
|
||
return false;
|
||
}
|
||
|
||
try
|
||
{
|
||
//启动驱动服务
|
||
var service = new ServiceController("netfilter2");
|
||
if (service.Status == ServiceControllerStatus.Running && StopServiceAndRestart)
|
||
{
|
||
//防止其他程序占用 重置NF百万ID限制
|
||
service.Stop();
|
||
service.WaitForStatus(ServiceControllerStatus.Stopped);
|
||
MainForm.Instance.StatusText($"{Utils.i18N.Translate("Status")}{Utils.i18N.Translate(": ")}{Utils.i18N.Translate("Starting netfilter2 Service")}");
|
||
service.Start();
|
||
}
|
||
else if (service.Status == ServiceControllerStatus.Stopped)
|
||
{
|
||
MainForm.Instance.StatusText($"{Utils.i18N.Translate("Status")}{Utils.i18N.Translate(": ")}{Utils.i18N.Translate("Starting netfilter2 Service")}");
|
||
service.Start();
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utils.Logging.Info(e.ToString());
|
||
|
||
var result = nfapinet.NFAPI.nf_registerDriver("netfilter2");
|
||
if (result != nfapinet.NF_STATUS.NF_STATUS_SUCCESS)
|
||
{
|
||
Utils.Logging.Info($"注册驱动失败，返回值：{result}");
|
||
return false;
|
||
}
|
||
}
|
||
|
||
var processes = "";
|
||
|
||
//开启进程白名单模式
|
||
if (!Global.Settings.ProcessBypassMode)
|
||
{
|
||
processes += "NTT.exe,";
|
||
}
|
||
|
||
foreach (var proc in mode.Rule)
|
||
{
|
||
processes += proc;
|
||
processes += ",";
|
||
}
|
||
processes = processes.Substring(0, processes.Length - 1);
|
||
|
||
Instance = MainController.GetProcess();
|
||
var fallback = "";
|
||
|
||
if (Global.Settings.UseRedirector2)
|
||
{
|
||
if (!File.Exists("bin\\Redirector2.exe"))
|
||
{
|
||
return false;
|
||
}
|
||
Instance.StartInfo.FileName = "bin\\Redirector2.exe";
|
||
|
||
|
||
if (server.Type != "Socks5")
|
||
{
|
||
fallback += $" 127.0.0.1:{Global.Settings.Socks5LocalPort}";
|
||
fallback += $" \"{processes}\"";
|
||
}
|
||
else
|
||
{
|
||
var result = Utils.DNS.Lookup(server.Hostname);
|
||
if (result == null)
|
||
{
|
||
Utils.Logging.Info("无法解析服务器 IP 地址");
|
||
return false;
|
||
}
|
||
|
||
fallback += $" {result}:{server.Port}";
|
||
fallback += $" \"{processes}\"";
|
||
|
||
if (!string.IsNullOrWhiteSpace(server.Username) && !string.IsNullOrWhiteSpace(server.Password))
|
||
{
|
||
fallback += $" \"{server.Username}\"";
|
||
fallback += $" \"{server.Password}\"";
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
if (!File.Exists("bin\\Redirector.exe"))
|
||
{
|
||
return false;
|
||
}
|
||
Instance.StartInfo.FileName = "bin\\Redirector.exe";
|
||
|
||
//开启进程白名单模式
|
||
if (Global.Settings.ProcessBypassMode)
|
||
{
|
||
processes += ",Shadowsocks.exe";
|
||
processes += ",ShadowsocksR.exe";
|
||
processes += ",Privoxy.exe";
|
||
processes += ",simple-obfs.exe";
|
||
processes += ",v2ray.exe,v2ctl.exe,v2ray-plugin.exe";
|
||
fallback += " -bypass true ";
|
||
}
|
||
else
|
||
{
|
||
fallback += " -bypass false ";
|
||
}
|
||
|
||
if (server.Type != "Socks5")
|
||
{
|
||
fallback += $"-r 127.0.0.1:{Global.Settings.Socks5LocalPort} -p \"{processes}\"";
|
||
}
|
||
else
|
||
{
|
||
var result = Utils.DNS.Lookup(server.Hostname);
|
||
if (result == null)
|
||
{
|
||
Utils.Logging.Info("无法解析服务器 IP 地址");
|
||
return false;
|
||
}
|
||
|
||
fallback += $"-r {result}:{server.Port} -p \"{processes}\"";
|
||
|
||
if (!string.IsNullOrWhiteSpace(server.Username) && !string.IsNullOrWhiteSpace(server.Password))
|
||
{
|
||
fallback += $" -username \"{server.Username}\" -password \"{server.Password}\"";
|
||
}
|
||
}
|
||
}
|
||
|
||
if (File.Exists("logging\\redirector.log"))
|
||
File.Delete("logging\\redirector.log");
|
||
|
||
Instance.StartInfo.Arguments = fallback + $" -tcport {Global.Settings.RedirectorTCPPort}";
|
||
Utils.Logging.Info(Instance.StartInfo.Arguments);
|
||
Instance.OutputDataReceived += OnOutputDataReceived;
|
||
Instance.ErrorDataReceived += OnOutputDataReceived;
|
||
State = Models.State.Starting;
|
||
Instance.Start();
|
||
Instance.BeginOutputReadLine();
|
||
Instance.BeginErrorReadLine();
|
||
|
||
for (var i = 0; i < 10; i++)
|
||
{
|
||
Thread.Sleep(1000);
|
||
|
||
if (State == Models.State.Started)
|
||
{
|
||
Utils.Logging.Info($"成功启动Redirector耗时:{i + 1}秒");
|
||
return true;
|
||
}
|
||
else
|
||
{
|
||
Utils.Logging.Info($"Redirector启动中，已耗时:{i + 1}秒");
|
||
}
|
||
}
|
||
|
||
Utils.Logging.Info("NF 进程启动超时");
|
||
Stop();
|
||
return false;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 停止
|
||
/// </summary>
|
||
public void Stop()
|
||
{
|
||
try
|
||
{
|
||
if (Instance != null && !Instance.HasExited)
|
||
{
|
||
Instance.Kill();
|
||
Instance.WaitForExit();
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utils.Logging.Info(e.ToString());
|
||
}
|
||
}
|
||
public bool InstallDriver()
|
||
{
|
||
|
||
Utils.Logging.Info("安装驱动中");
|
||
// 生成系统版本
|
||
var version = $"{Environment.OSVersion.Version.Major.ToString()}.{Environment.OSVersion.Version.Minor.ToString()}";
|
||
|
||
// 检查系统版本并复制对应驱动
|
||
try
|
||
{
|
||
switch (version)
|
||
{
|
||
case "10.0":
|
||
File.Copy("bin\\Win-10.sys", driverPath);
|
||
Utils.Logging.Info("已复制 Win10 驱动");
|
||
break;
|
||
case "6.3":
|
||
case "6.2":
|
||
File.Copy("bin\\Win-8.sys", driverPath);
|
||
Utils.Logging.Info("已复制 Win8 驱动");
|
||
break;
|
||
case "6.1":
|
||
case "6.0":
|
||
File.Copy("bin\\Win-7.sys", driverPath);
|
||
Utils.Logging.Info("已复制 Win7 驱动");
|
||
break;
|
||
default:
|
||
Utils.Logging.Info($"不支持的系统版本：{version}");
|
||
return false;
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Utils.Logging.Info("复制驱动文件失败");
|
||
Utils.Logging.Info(e.ToString());
|
||
return false;
|
||
}
|
||
MainForm.Instance.StatusText($"{Utils.i18N.Translate("Status")}{Utils.i18N.Translate(": ")}{Utils.i18N.Translate("Register driver")}");
|
||
// 注册驱动文件
|
||
var result = nfapinet.NFAPI.nf_registerDriver("netfilter2");
|
||
if (result != nfapinet.NF_STATUS.NF_STATUS_SUCCESS)
|
||
{
|
||
Utils.Logging.Info($"注册驱动失败，返回值：{result}");
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
public void OnOutputDataReceived(object sender, DataReceivedEventArgs e)
|
||
{
|
||
if (!string.IsNullOrWhiteSpace(e.Data))
|
||
{
|
||
File.AppendAllText("logging\\redirector.log", string.Format("{0}\r\n", e.Data));
|
||
|
||
if (State == Models.State.Starting)
|
||
{
|
||
if (Instance.HasExited)
|
||
{
|
||
State = Models.State.Stopped;
|
||
}
|
||
else if (e.Data.Contains("Start") || e.Data.Contains("Redirect to"))
|
||
{
|
||
State = Models.State.Started;
|
||
}
|
||
else if (e.Data.Contains("Failed") || e.Data.Contains("Unable"))
|
||
{
|
||
State = Models.State.Stopped;
|
||
}
|
||
}
|
||
else if (State == Models.State.Started)
|
||
{
|
||
if (e.Data.StartsWith("[Application][Bandwidth]"))
|
||
{
|
||
var splited = e.Data.Replace("[Application][Bandwidth]", "").Trim().Split(',');
|
||
if (splited.Length == 2)
|
||
{
|
||
var uploadSplited = splited[0].Split(':');
|
||
var downloadSplited = splited[1].Split(':');
|
||
|
||
if (uploadSplited.Length == 2 && downloadSplited.Length == 2)
|
||
{
|
||
if (long.TryParse(uploadSplited[1], out var upload) && long.TryParse(downloadSplited[1], out var download))
|
||
{
|
||
Task.Run(() => OnBandwidthUpdated(upload, download));
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|