fix: More compatible behavior for getting the current executable dir

Mrs4s · Sep 11, 2022 · 532f55f · 532f55f
1 parent 8389d91
commit 532f55f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/internal/base/flag.go b/internal/base/flag.go
@@ -140,7 +140,8 @@ func ResetWorkingDir() {
 			args = append(args, os.Args[i])
 		}
 	}
-	p, _ := filepath.Abs(os.Args[0])
+	ex, _ := os.Executable()
+	p, _ := filepath.Abs(ex)
 	_, err := os.Stat(p)
 	if !(err == nil || errors.Is(err, os.ErrExist)) {
 		log.Fatalf("重置工作目录时出现错误: 无法找到路径 %v", p)

diff --git a/server/daemon.go b/server/daemon.go
@@ -5,6 +5,7 @@ package server
 import (
 	"os"
 	"os/exec"
+	"path/filepath"
 	"strconv"
 	"strings"
 
@@ -28,7 +29,9 @@ func Daemon() {
 		execArgs = append(execArgs, args[i])
 	}
 
-	proc := exec.Command(os.Args[0], execArgs...)
+	ex, _ := os.Executable()
+	p, _ := filepath.Abs(ex)
+	proc := exec.Command(p, execArgs...)
 	err := proc.Start()
 	if err != nil {
 		panic(err)