mirror of
https://github.com/qtgolang/SunnyNet.git
synced 2026-08-30 22:11:29 -07:00
165 lines
7.4 KiB
Go
165 lines
7.4 KiB
Go
//go:build linux && !android
|
||
// +build linux,!android
|
||
|
||
package Tun // 定义包名为 Tun
|
||
|
||
import ( // 导入所需包
|
||
"bytes" // 用于接收子进程输出
|
||
"fmt" // 字符串格式化
|
||
"io" // IO 接口
|
||
"os" // 文件与进程操作
|
||
"os/exec" // 执行外部命令
|
||
"path/filepath" // 构造临时文件路径
|
||
"regexp" // 正则解析后台 PID
|
||
"strconv" // 字符串转数字
|
||
"strings" // 构建脚本文本
|
||
"syscall" // 进程存活检测
|
||
"time" // 定时控制
|
||
)
|
||
|
||
// 全局脚本文本（由 CreateSh 生成）
|
||
var sh = "" // 存放生成后的 Shell 脚本文本
|
||
|
||
// 监控器状态（保持你原有语义）
|
||
var watchdogStarted bool // 标记监控脚本是否已启动
|
||
var watchdogPid int // 记录后台化脚本 PID
|
||
|
||
// 脚本落盘路径（保持与原代码一致）
|
||
const scriptPath = "/usr/local/bin/SunnyTunCancel.sh" // 清理脚本的绝对路径
|
||
|
||
// writeFileAtomic 原子写文件（避免部分写入导致执行失败）
|
||
func writeFileAtomic(path string, data []byte, perm os.FileMode) error { // 定义原子写文件函数
|
||
dir := filepath.Dir(path) // 获取目录
|
||
tmp := filepath.Join(dir, "."+filepath.Base(path)+".tmp") // 生成临时文件路径
|
||
if err := os.WriteFile(tmp, data, perm); err != nil { // 先写入临时文件
|
||
return err // 返回错误
|
||
}
|
||
return os.Rename(tmp, path) // 重命名替换，达到原子写入效果
|
||
}
|
||
|
||
// CreateSh 生成守护/清理脚本（根据 r 的状态把命令具体化到脚本内）
|
||
func (r *TunRouter) CreateSh() { // 定义 CreateSh 方法
|
||
var b strings.Builder // 使用高效的字符串构建器
|
||
// 写入脚本头与后台化逻辑
|
||
b.WriteString(`#!/bin/bash
|
||
set -u
|
||
set -o pipefail
|
||
|
||
# ==============================
|
||
# 🛠 SunnyTunCancel.sh
|
||
# 功能：后台监控目标进程，当进程退出时自动恢复默认网络配置
|
||
# ==============================
|
||
|
||
# 若未后台化，则后台化自己并仅输出后台 PID 一行
|
||
if [ -z "${DISOWNED-}" ]; then
|
||
export DISOWNED=1
|
||
nohup "$0" "$@" >/dev/null 2>&1 &
|
||
printf "%s\n" "$!"
|
||
exit 0
|
||
fi
|
||
|
||
# 目标 PID（第一个参数）
|
||
TARGET_PID="${1:-}"
|
||
if [ -z "${TARGET_PID}" ]; then
|
||
echo "missing TARGET_PID" >&2
|
||
exit 1
|
||
fi
|
||
|
||
echo "🕓 监控进程 PID: ${TARGET_PID}"
|
||
|
||
# 轮询检测目标进程是否仍在
|
||
while kill -0 "${TARGET_PID}" 2>/dev/null; do
|
||
sleep 1
|
||
done
|
||
|
||
# === 以下为恢复/清理动作 ===
|
||
`)
|
||
// 删除 iptables 规则（如果你进程曾添加过）
|
||
if len(r.iptablesRule) > 0 { // 如果记录了 iptables 规则
|
||
del := append([]string{}, r.iptablesRule...) // 拷贝一份
|
||
if len(del) > 3 && del[3] == "-A" { // 如果第四个参数为 -A
|
||
del[3] = "-D" // 替换为 -D，实现删除
|
||
}
|
||
b.WriteString(strings.Join(del, " ")) // 拼接命令
|
||
b.WriteByte('\n') // 换行
|
||
}
|
||
// 删除策略路由（按优先级顺序）
|
||
b.WriteString("ip rule del priority 220 || true\n") // 删兜底
|
||
b.WriteString("ip rule del priority 120 || true\n") // 删本地子网
|
||
b.WriteString("ip rule del priority 100 || true\n") // 删 fwmark
|
||
b.WriteString("ip rule del priority 90 || true\n") // 删 from hostIP
|
||
|
||
// 清理路由表 200 与 100（仅删我们加的几条，避免误删用户其它配置）
|
||
b.WriteString("ip route flush table 200 || true\n") // 清空 200
|
||
b.WriteString(fmt.Sprintf("ip route del default via %s dev %s table 100 || true\n", r.defGWIP, r.ifaceName)) // 删默认
|
||
b.WriteString(fmt.Sprintf("ip route del %s/32 dev %s table 100 || true\n", r.defGWIP, r.ifaceName)) // 删/32
|
||
b.WriteString(fmt.Sprintf("ip route del %s dev %s table 100 || true\n", r.localCIDR, r.ifaceName)) // 删本地网段
|
||
|
||
// 还原 rp_filter
|
||
b.WriteString("sysctl -w net.ipv4.conf.all.rp_filter=1 || true\n") // all=1
|
||
b.WriteString("sysctl -w net.ipv4.conf.default.rp_filter=1 || true\n") // default=1
|
||
b.WriteString(fmt.Sprintf("sysctl -w net.ipv4.conf.%s.rp_filter=1 || true\n", r.ifaceName)) // iface=1
|
||
b.WriteString(fmt.Sprintf("sysctl -w net.ipv4.conf.%s.rp_filter=1 || true\n", r.tunName)) // tun=1
|
||
// 删除 TUN 接口（双保险）
|
||
b.WriteString(fmt.Sprintf("ip link del %s || true\n", r.tunName)) // 删除 TUN
|
||
// 收尾注释
|
||
b.WriteString("echo '✅ SunnyTun 清理完成'\n") // 提示完成
|
||
// 回填到全局 sh
|
||
sh = b.String() // 将生成好的脚本文本保存到全局变量
|
||
}
|
||
|
||
// startWatchdog 启动路由恢复监控脚本（防止主进程异常退出导致网络不恢复）
|
||
func startWatchdog() { // 定义 startWatchdog 函数
|
||
if watchdogStarted { // 如果已启动
|
||
return // 直接返回
|
||
}
|
||
watchdogStarted = true // 标记已启动
|
||
rou.CreateSh()
|
||
_ = os.Remove(scriptPath) // 先尝试删除旧脚本（忽略错误）
|
||
_ = writeFileAtomic(scriptPath, []byte(sh), 0o777) // 原子写入新脚本并赋可执行权限
|
||
|
||
go func() { // 启动后台 goroutine 持续监控
|
||
mainPid := os.Getpid() // 获取当前主进程 PID（被监控对象）
|
||
for { // 循环监控
|
||
// 若已有后台脚本在运行，则无需拉起
|
||
if watchdogPid > 0 && syscall.Kill(watchdogPid, 0) == nil { // 若 PID 存在
|
||
time.Sleep(time.Second) // 休眠 1 秒后继续检测
|
||
continue // 进入下次循环
|
||
}
|
||
|
||
// 运行脚本：使用 /bin/bash，保障 $EUID 等 Bash 变量可用
|
||
// 兼容你原来的三参形式：传 (PID, defaultGatewayIf, ifaceName)
|
||
// 即使脚本未使用这两个参数也不影响（保持你的外部调用兼容）
|
||
cmd := exec.Command("/bin/bash", scriptPath, strconv.Itoa(mainPid), defaultGatewayIf, ifaceName) // 构造命令
|
||
var buffer bytes.Buffer // 创建输出缓冲区
|
||
cmd.Stdout = &buffer // 捕获标准输出（脚本首次运行仅输出后台 PID）
|
||
cmd.Stderr = io.Discard // 丢弃标准错误（避免污染 PID 解析）
|
||
if err := cmd.Start(); err != nil { // 启动脚本失败
|
||
watchdogPid = 0 // PID 置 0，稍后重试
|
||
time.Sleep(time.Second) // 等待 1 秒
|
||
continue // 下一轮重试
|
||
}
|
||
_ = cmd.Wait() // 等待脚本前台进程退出（它会很快退出，只打印后台 PID）
|
||
// 解析输出中的第一个十进制数字作为 PID（更鲁棒，避免杂讯）
|
||
out := strings.TrimSpace(buffer.String()) // 去掉首尾空白
|
||
re := regexp.MustCompile(`\b\d+\b`) // 匹配第一个整数
|
||
pidStr := re.FindString(out) // 提取 PID 字符串
|
||
if pidStr == "" { // 若未匹配到
|
||
watchdogPid = 0 // 置 0 以便重试
|
||
time.Sleep(time.Second) // 等待 1 秒
|
||
continue // 重试
|
||
}
|
||
pid, err := strconv.Atoi(pidStr) // 转换为整数
|
||
if err != nil || pid <= 0 { // 若解析失败或不合理
|
||
watchdogPid = 0 // 置 0
|
||
time.Sleep(time.Second) // 等待 1 秒
|
||
continue // 重试
|
||
}
|
||
watchdogPid = pid // 记录后台脚本 PID
|
||
|
||
// 正常情况下，此处后台脚本常驻；定期确认其存活
|
||
time.Sleep(time.Second) // 休眠 1 秒再继续下一轮检查
|
||
}
|
||
}() // 启动 goroutine 结束
|
||
}
|