第一份2.0-LTS代码上传
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
// 优先级2:embed 解压到临时目录
|
||||
// 优先级2:embed 解压
|
||||
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.toml(v2.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 {
|
||||
|
||||
Reference in New Issue
Block a user