Fix path of knownhosts file

macdylan · Apr 17, 2023 · 79940b9 · 79940b9
1 parent f5ae9f1
commit 79940b9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -30,7 +30,8 @@ func main() {
 	}()
 
 	// 获取程序所在目录
-	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
+	ex, _ := os.Executable()
+	dir, err := filepath.Abs(filepath.Dir(ex))
 	if err != nil {
 		log.Panicln(err)
 	}