由windows快捷方式的工作目录属性问题导致的bug记录

秦羽 2023-09-05 AM 196℃ 0条

## 背景

在使用go写一个脚本管理程序时，需要将程序加入开机自启动，于是使用`github.com/go-ole/go-ole`库在windows的启动目录（`C:\Users\UserName\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Startup`）下创建程序的快捷方式，从而实现开机启动。

go创建快捷方式的代码如下：

```go
func createShortcut(source string, target string) error {
	var err error
	err = ole.CoInitializeEx(0, ole.COINIT_APARTMENTTHREADED|ole.COINIT_SPEED_OVER_MEMORY)
	if err != nil {
		return err
	}
	defer ole.CoUninitialize()
	oleShellObject, err := oleutil.CreateObject("WScript.Shell")
	if err != nil {
		return err
	}
	defer oleShellObject.Release()
	wShell, err := oleShellObject.QueryInterface(ole.IID_IDispatch)
	if err != nil {
		return err
	}
	defer wShell.Release()
	cs, err := oleutil.CallMethod(wShell, "CreateShortcut", target)
	if err != nil {
		return err
	}
	iDispatch := cs.ToIDispatch()
	_, err = oleutil.PutProperty(iDispatch, "TargetPath", source)
	if err != nil {
		return err
	}

_, err = oleutil.CallMethod(iDispatch, "Save")
	if err != nil {
		return err
	}
	return nil
}
```

## bug描述

创建快捷方式成功后，程序直接手动启动运行正常，但是开机自启动后程序内执行脚本任务时报错并且程序退出，日志显示乱码：
```
鍛戒护鎵ц澶辫触 exit status 2
```

## bug原因

经过排查，发现是在程序中使用`exec.Command`执行脚本时，配置的脚本路径是相对路径：`./shells/python/xxx.py`

** 而使用go创建的快捷方式没有继承程序的工作目录，而windows直接创建的快捷方式会继承目标文件的工作目录 **

所以导致用程序创建的快捷方式执行时，相对路径是启动目录，找不到脚本文件而报错；而windows创建的快捷方式则能找到相对路径的脚本文件，所以正常运行。

## 解决方式

- 方式1：修改配置中的脚本路径，改为绝对路径
- 方式2：通过`os.Executable()`拿到当前exe路径，再通过`path.Join()`和`filepath.Dir()`拼接脚本路径
- 方式3：修改创建快捷方式的代码，增加工作目录设置

修改后的代码如下：

```go
	func createShortcut(source string, target string) error {
		var err error
		err = ole.CoInitializeEx(0, ole.COINIT_APARTMENTTHREADED|ole.COINIT_SPEED_OVER_MEMORY)
		if err != nil {
			return err
		}
		defer ole.CoUninitialize()
		oleShellObject, err := oleutil.CreateObject("WScript.Shell")
		if err != nil {
			return err
		}
		defer oleShellObject.Release()
		wShell, err := oleShellObject.QueryInterface(ole.IID_IDispatch)
		if err != nil {
			return err
		}
		defer wShell.Release()
		cs, err := oleutil.CallMethod(wShell, "CreateShortcut", target)
		if err != nil {
			return err
		}
		iDispatch := cs.ToIDispatch()

// 设置快捷方式的目标路径
		_, err = oleutil.PutProperty(iDispatch, "TargetPath", source)
		if err != nil {
			return err
		}

// 设置快捷方式的工作目录
		_, err = oleutil.PutProperty(iDispatch, "WorkingDirectory", filepath.Dir(source))
		if err != nil {
			return err
		}

_, err = oleutil.CallMethod(iDispatch, "Save")
		if err != nil {
			return err
		}
		return nil
	}

```
	
再次运行程序，正常了

标签: windows, 快捷方式, lnk

非特殊说明，本博所有文章均为博主原创。

如若转载，请注明出处：https://www.qinyu.cc/archives/258.html

上一篇 go在windows下通过创建快捷方式设置exe开机启动

下一篇 yolov5使用gpu训练模型

由windows快捷方式的工作目录属性问题导致的bug记录

评论啦~

栏目分类

标签云

友情链接申请

由windows快捷方式的工作目录属性问题导致的bug记录

 评论啦~

 栏目分类

标签云

友情链接申请

评论啦~

栏目分类

标签云

友情链接申请