最新消息:

在NSIS中实现安装时取消并回滚(2)——写个线程控制插件实现它

nsis 永恒心锁 1095浏览 0评论

需要自己完成这个功能。下面就来介绍一种相对比较简单的实现方式。

功能难点在上一篇中已经有描述,这里就不赘述了。基本原理就是在安装过程中,使用插件中把后台的释放文件的线程挂起,问用户是不是真要取消安装,是的话就使用安装过程中生成的Log文件进行回滚,不是的话就让被挂起的线程继续运行。

大致要解释如下几个问题

1. 编写线程控制插件。

2. 开启安装页面的取消按钮,并自定义取消函数。

3. 与安装日志功能集成,实现回滚操作。

首先是实现线程控制插件。这个插件有如下几个功能:

1. 挂起后台所有线程

2. 继续后台所有线程

3. 终止后台所有线程

为什么是后台?因为UI线程要用来弹出一个MessageBox让用户点,如果也挂起了,程序就死了。为什么是所有线程呢?因为如果有多个后台线程,如果那个线程不主动配合调查,想要找出哪个线程是用来释放文件似乎不是很容易。简单起见,就全挂起得了,好在不会有什么问题。更好的是,这个安装程序在安装页面只有两个线程,一个UI线程,一个后台释放文件。

还有一点要注意的就是TerminateThread函数是异步的。最好多调用一个WaitForSingleObject来保证函数返回时,线程的确已经被和谐了才行。(不调似乎也没有什么问题)

代码就不贴了,可以从这里下载。关于如何编写NSIS插件可以参考制作NSIS命令行窗口输出插件这篇文章。

然后就是开启安装页面的取消按钮,在上一篇在NSIS中实现安装时取消并回滚(1)——现状提到的几个帖子有示例的。代码如下:(仅适用于MUI2)

!define MUI_PAGE_CUSTOMFUNCTION_PRE OnInstFilesPagePre ;必须放在下面的语句前面
!insertmacro MUI_PAGE_INSTFILES
Function OnInstFilesPagePre
GetDlgItem $0 $HWNDPARENT 2
EnableWindow $0 1
FunctionEnd

自定义取消函数的代码如下:

!define MUI_CUSTOMFUNCTION_ABORT OnUserAbort
Function OnUserAbort
…
nsExec::ExecToLog 'cmd /c copy /Y "$INSTDIR\install.log" "$INSTDIR\temp.log"'
Call CreateLogFromFile
Call RemoveDirectoriesFromLog
…
FunctionEnd

函数中给出的代码就是实现回滚操作部分的代码。在调用回滚函数(其实就是卸载时调用的根据日志删除文件的函数)前要把install.log复制一份,然后用复制的日志文件进行回滚。为什么要这样呢?因为后台的释放文件的线程被Terminate掉了,而就是那个线程打开的日志文件并写入安装日志,线程被和谐的,那么释放文件句柄的代码就不会被执行。说白了就是资源泄露了。后果就是再次打开这个文件的时候会失败,就算是只读也不行。用NSIS脚本举个例子。

FileOpen $0 “txt.txt” “w”
FileOpen $1 “txt.txt” “r” ;失败

第一次打开写,可以正确打开,第二次打开,虽然只读也会失败。但是如果像下面这样两次都是读。两次都是可以成功的。

FileOpen $0 “txt.txt” “r”
FileOpen $1 “txt.txt” “r” ;成功

上面只列出了功能实现的比较关键的代码。完整的代码可以从这里下载。如果要正确编译成安装包,还要下载这个头文件。

到此为止,基本在NSIS中实现了取消安装并回滚。但是这种实现有下面一些入缺陷。(都快成定律了,解决一个问题,引出三个问题。)

1. 回滚时install.log文件不能被删除。因为在上面已经解释了。

2. 回滚时,滚动条不回滚。其实应该是可以实现的。大家自己试下吧,应该不难。

3. 回滚后程序直接退出,而不会显示安装失败页面之类的页面。因为我们的回滚操作就是在User Abort里做的嘛。使用User Abort的后果就是直接退出了。应该也是有方法绕过去的

永恒心锁,版权所有丨如未注明,均为转载丨本网站采用BY-NC-SA协议进行授权

转载请注明:永恒心锁-分享互联网 » 在NSIS中实现安装时取消并回滚(2)——写个线程控制插件实现它

您必须 登录 才能发表评论!