lxh2875931338/frpc-console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

文件夹挂载逻辑变动

Browse Source
This commit is contained in:
不安的亡命樱
2026-07-30 00:18:06 +08:00
parent 826ce399e7
commit defc9d00d3
5 changed files with 115 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frp.go
+101 -35
View File
@@ -4,6 +4,7 @@ import (
"bytes"
"embed"
"fmt"
"io"
"log"
"os"
"os/exec"
@@ -24,16 +25,12 @@ var FrpcTemplateContent string
var (
cachedFrpcPath string
frpcPathMutex sync.Mutex
frpsPathMutex sync.Mutex
)
// ============================================================
// 获取 frpc 路径
// ============================================================
func getFrpcPath() (string, error) {
frpcPathMutex.Lock()
defer frpcPathMutex.Unlock()
frpsPathMutex.Lock()
defer frpsPathMutex.Unlock()
if cachedFrpcPath != "" {
if _, err := os.Stat(cachedFrpcPath); err == nil {
@@ -61,14 +58,12 @@ func getFrpcPath() (string, error) {
return "", fmt.Errorf("不支持的平台: %s/%s", runtime.GOOS, runtime.GOARCH)
}
// 优先级1：本地 ./bin/
localPath := filepath.Join(".", "bin", fileName)
if _, err := os.Stat(localPath); err == nil {
cachedFrpcPath = localPath
return localPath, nil
}
// 优先级2：embed 解压
data, err := embeddedFrpc.ReadFile("bin/" + fileName)
if err == nil {
tmpPath := filepath.Join(os.TempDir(), "frpc")
@@ -85,7 +80,6 @@ func getFrpcPath() (string, error) {
}
}
// 优先级3：系统 PATH
path, err := exec.LookPath("frpc")
if err == nil {
cachedFrpcPath = path
@@ -95,10 +89,6 @@ func getFrpcPath() (string, error) {
return "", fmt.Errorf("未找到 frpc 文件")
}
// ============================================================
// 生成 frpc.toml（v2.0 支持 wireProtocol）
// ============================================================
func GenerateFrpcConfig() error {
cfg, err := GetGlobalConfig()
if err != nil {
@@ -117,24 +107,21 @@ func GenerateFrpcConfig() error {
}
}
// 构建模板数据
data := struct {
*GlobalConfig
Proxies []Proxy
WireProtocolLine string // v2 协议配置行（为空则不输出）
WireProtocolLine string
}{
GlobalConfig: cfg,
Proxies: activeProxies,
}
// v2.0: 如果启用 v2 协议，生成配置行
if cfg.WireProtocolV2 {
data.WireProtocolLine = `wireProtocol = "v2"`
} else {
data.WireProtocolLine = "" // 不输出，使用默认 v1
data.WireProtocolLine = ""
}
// 读取模板
var tmplContent string
if _, err := os.Stat("frpc.tmpl"); err == nil {
content, readErr := os.ReadFile("frpc.tmpl")
@@ -157,19 +144,20 @@ func GenerateFrpcConfig() error {
return fmt.Errorf("渲染模板失败: %w", err)
}
if err := os.WriteFile("./frpc.toml", buf.Bytes(), 0644); err != nil {
// 所有文件写入 ./data/ 目录
if err := os.MkdirAll("./data", 0755); err != nil {
return fmt.Errorf("创建 data 目录失败: %w", err)
}
if err := os.WriteFile("./data/frpc.toml", buf.Bytes(), 0644); err != nil {
return fmt.Errorf("写入配置文件失败: %w", err)
}
return nil
}
// ============================================================
// frpc 进程管理
// ============================================================
func isFrpcRunning() bool {
pidData, err := os.ReadFile("./frpc.pid")
pidData, err := os.ReadFile("./data/frpc.pid")
if err != nil {
return false
}
@@ -200,7 +188,11 @@ func StartFrpc() error {
return fmt.Errorf("获取 frpc 路径失败: %w", err)
}
if _, err := os.Stat("./frpc.toml"); os.IsNotExist(err) {
if err := os.MkdirAll("./data", 0755); err != nil {
return fmt.Errorf("创建 data 目录失败: %w", err)
}
if _, err := os.Stat("./data/frpc.toml"); os.IsNotExist(err) {
if err := GenerateFrpcConfig(); err != nil {
return fmt.Errorf("生成配置文件失败: %w", err)
}
@@ -210,13 +202,13 @@ func StartFrpc() error {
return nil
}
os.Remove("./frpc.pid")
os.Remove("./data/frpc.pid")
cmd := exec.Command(frpcPath, "-c", "./frpc.toml")
cmd := exec.Command(frpcPath, "-c", "./data/frpc.toml")
setWindowHide(cmd)
setSysProcAttr(cmd)
logFile, err := os.OpenFile("./frpc.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
logFile, err := os.OpenFile("./data/frpc.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
if err != nil {
return fmt.Errorf("打开日志文件失败: %w", err)
}
@@ -227,7 +219,7 @@ func StartFrpc() error {
return fmt.Errorf("启动 frpc 失败: %w", err)
}
if err := os.WriteFile("./frpc.pid", []byte(fmt.Sprintf("%d", cmd.Process.Pid)), 0644); err != nil {
if err := os.WriteFile("./data/frpc.pid", []byte(fmt.Sprintf("%d", cmd.Process.Pid)), 0644); err != nil {
return fmt.Errorf("保存 PID 失败: %w", err)
}
@@ -240,11 +232,11 @@ func StopFrpc() error {
if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "not found") {
return fmt.Errorf("停止 frpc 失败: %w", err)
}
os.Remove("./frpc.pid")
os.Remove("./data/frpc.pid")
return nil
}
pidData, err := os.ReadFile("./frpc.pid")
pidData, err := os.ReadFile("./data/frpc.pid")
if err != nil {
cmd := exec.Command("pkill", "-f", "frpc")
if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "no process") {
@@ -256,7 +248,7 @@ func StopFrpc() error {
pid, _ := strconv.Atoi(strings.TrimSpace(string(pidData)))
process, err := os.FindProcess(pid)
if err != nil {
os.Remove("./frpc.pid")
os.Remove("./data/frpc.pid")
return nil
}
@@ -264,7 +256,7 @@ func StopFrpc() error {
return fmt.Errorf("杀死进程失败: %w", err)
}
os.Remove("./frpc.pid")
os.Remove("./data/frpc.pid")
return nil
}
@@ -283,7 +275,7 @@ func ReloadFrpc() error {
return fmt.Errorf("获取 frpc 路径失败: %w", err)
}
cmd := exec.Command(frpcPath, "reload", "-c", "./frpc.toml")
cmd := exec.Command(frpcPath, "reload", "-c", "./data/frpc.toml")
_, err = cmd.CombinedOutput()
if err != nil {
log.Printf("⚠️ 热加载失败 (%v)，自动降级为重启 frpc", err)
@@ -297,3 +289,77 @@ func ReloadFrpc() error {
}
return nil
}
// readTailLog 读取文件末尾 n 行
func readTailLog(filePath string, n int) ([]string, error) {
file, err := os.Open(filePath)
if err != nil {
return nil, err
}
defer file.Close()
info, err := file.Stat()
if err != nil {
return nil, err
}
fileSize := info.Size()
if fileSize == 0 {
return []string{}, nil
}
const chunkSize = 4096
var lines []string
var leftover []byte
offset := fileSize
for len(lines) < n && offset > 0 {
readSize := chunkSize
if offset < int64(chunkSize) {
readSize = int(offset)
}
offset -= int64(readSize)
buf := make([]byte, readSize)
_, err := file.ReadAt(buf, offset)
if err != nil && err != io.EOF {
return nil, err
}
data := append(buf, leftover...)
leftover = nil
start := 0
for i := len(data) - 1; i >= 0; i-- {
if data[i] == '\n' {
if i+1 < len(data) {
line := string(data[i+1:])
if line != "" {
lines = append([]string{line}, lines...)
if len(lines) >= n {
break
}
}
}
start = i
}
}
if len(lines) < n && start > 0 {
leftover = data[:start]
}
}
if len(lines) < n && len(leftover) > 0 {
parts := strings.Split(string(leftover), "\n")
for i := len(parts) - 1; i >= 0; i-- {
if parts[i] != "" {
lines = append([]string{parts[i]}, lines...)
if len(lines) >= n {
break
}
}
}
}
return lines, nil
}