第一份2.0-LTS代码上传

This commit is contained in:
2026-07-26 21:55:42 +08:00
parent 542f265a35
commit e64a8687d2
8 changed files with 286 additions and 115 deletions
+25 -22
View File
@@ -27,10 +27,10 @@ var (
frpcPathMutex sync.Mutex
)
// getFrpcPath 获取 frpc 路径,优先级:
// 1. 当前目录 ./bin/<platform> (便于用户手动管理 frpc 版本)
// 2. embed 解压到 /tmp (便于开箱即用)
// 3. 系统 PATH 中的 frpc (降级方案)
// ============================================================
// 获取 frpc 路径
// ============================================================
func getFrpcPath() (string, error) {
frpcPathMutex.Lock()
defer frpcPathMutex.Unlock()
@@ -68,7 +68,7 @@ func getFrpcPath() (string, error) {
return localPath, nil
}
// 优先级2embed 解压到临时目录
// 优先级2embed 解压
data, err := embeddedFrpc.ReadFile("bin/" + fileName)
if err == nil {
tmpPath := filepath.Join(os.TempDir(), "frpc")
@@ -92,10 +92,13 @@ func getFrpcPath() (string, error) {
return path, nil
}
return "", fmt.Errorf("未找到 frpc 文件 (本地 bin/、embed、PATH 均无)")
return "", fmt.Errorf("未找到 frpc 文件")
}
// GenerateFrpcConfig 生成 frpc.toml
// ============================================================
// 生成 frpc.tomlv2.0 支持 wireProtocol
// ============================================================
func GenerateFrpcConfig() error {
cfg, err := GetGlobalConfig()
if err != nil {
@@ -114,14 +117,24 @@ func GenerateFrpcConfig() error {
}
}
// 构建模板数据
data := struct {
*GlobalConfig
Proxies []Proxy
Proxies []Proxy
WireProtocolLine string // v2 协议配置行(为空则不输出)
}{
GlobalConfig: cfg,
Proxies: activeProxies,
}
// v2.0: 如果启用 v2 协议,生成配置行
if cfg.WireProtocolV2 {
data.WireProtocolLine = `wireProtocol = "v2"`
} else {
data.WireProtocolLine = "" // 不输出,使用默认 v1
}
// 读取模板
var tmplContent string
if _, err := os.Stat("frpc.tmpl"); err == nil {
content, readErr := os.ReadFile("frpc.tmpl")
@@ -151,7 +164,10 @@ func GenerateFrpcConfig() error {
return nil
}
// isFrpcRunning 通过 PID 文件检查 frpc 是否在运行
// ============================================================
// frpc 进程管理
// ============================================================
func isFrpcRunning() bool {
pidData, err := os.ReadFile("./frpc.pid")
if err != nil {
@@ -163,7 +179,6 @@ func isFrpcRunning() bool {
}
if runtime.GOOS == "windows" {
// Windows 用 tasklist 检查
cmd := exec.Command("tasklist", "/FI", "PID eq", strconv.Itoa(pid))
output, err := cmd.CombinedOutput()
if err != nil {
@@ -172,7 +187,6 @@ func isFrpcRunning() bool {
return strings.Contains(string(output), strconv.Itoa(pid))
}
// Linux/Unix: 用 kill 0 检查进程是否存在
process, err := os.FindProcess(pid)
if err != nil {
return false
@@ -180,7 +194,6 @@ func isFrpcRunning() bool {
return process.Signal(syscall.Signal(0)) == nil
}
// StartFrpc 启动 frpc(独立进程,脱离父进程)
func StartFrpc() error {
frpcPath, err := getFrpcPath()
if err != nil {
@@ -197,16 +210,12 @@ func StartFrpc() error {
return nil
}
// 清理旧的 PID 文件
os.Remove("./frpc.pid")
cmd := exec.Command(frpcPath, "-c", "./frpc.toml")
setWindowHide(cmd)
// ✅ 设置进程属性(Unix: Setsid 脱离父进程,Windows: 不处理)
setSysProcAttr(cmd)
// 把输出重定向到日志文件
logFile, err := os.OpenFile("./frpc.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
if err != nil {
return fmt.Errorf("打开日志文件失败: %w", err)
@@ -218,7 +227,6 @@ func StartFrpc() error {
return fmt.Errorf("启动 frpc 失败: %w", err)
}
// ✅ 保存 PID 到文件
if err := os.WriteFile("./frpc.pid", []byte(fmt.Sprintf("%d", cmd.Process.Pid)), 0644); err != nil {
return fmt.Errorf("保存 PID 失败: %w", err)
}
@@ -226,7 +234,6 @@ func StartFrpc() error {
return nil
}
// StopFrpc 停止 frpc(通过 PID 文件精准杀进程)
func StopFrpc() error {
if runtime.GOOS == "windows" {
cmd := exec.Command("taskkill", "/F", "/IM", "frpc.exe")
@@ -237,10 +244,8 @@ func StopFrpc() error {
return nil
}
// Linux: 读取 PID 文件精准杀进程
pidData, err := os.ReadFile("./frpc.pid")
if err != nil {
// 降级到 pkill
cmd := exec.Command("pkill", "-f", "frpc")
if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "no process") {
return fmt.Errorf("停止 frpc 失败: %w", err)
@@ -263,12 +268,10 @@ func StopFrpc() error {
return nil
}
// GetFrpcStatus 获取 frpc 运行状态(通过 PID 文件)
func GetFrpcStatus() (bool, error) {
return isFrpcRunning(), nil
}
// ReloadFrpc 热加载,失败时自动降级为重启
func ReloadFrpc() error {
running := isFrpcRunning()
if !running {