如何编写自我替换/更新二进制文件？

Question

我正在尝试编写一个可以查找URL的C程序,并且可以使用它的新版本,它应该能够更新自己.

我试过的方法:

1. forkout一个新的进程下载新的二进制文件说BINARY.tmp,我用来forkout的代码是:

   int
   forkout_cmd(char *cmdstr) {
     pid_t pid;
     char *cmd[4];
   
     cmd[0] = "/bin/bash";
     cmd[1] = "-c";
     cmd[2] = cmdstr;
     cmd[3] = NULL;
   
     pid = vfork();
     if( pid == -1 ) {
       logmsg("Forking for upgradation failed.");
       return -1;
     }else if( pid == 0 ){
       /* we are in child process */
       execvp(cmd[0], cmd);
       logmsg("execl failed while executing upgradation job.");
     }else{
       /* need not to wait for the child to complete. */
       wait(NULL);
     }
       return 0;
   }
   

2. 新进程尝试覆盖原始BINARY

   例如,您可能会考虑分叉的例程:

   forkout_cmd("wget -O BINARY.tmp https://someurl.com/BINARY_LATEST; /bin/mv -f BINARY.tmp BINARY");
   

但是,覆盖失败,因为原始二进制文件仍在执行中,因此在磁盘上忙,有人可以在这里提供一些建议来克服这个问题.

提前致谢.

Answer 1

将当前运行的二进制文件重命名为其他内容,编写新的二进制文件,运行它,然后再删除重命名的二进制文件.

Answer 2

我将保存binary.tmp到与可执行文件相同的目录,验证其校验和/签名(无论100%确定没有发生错误),然后以原子方式将其重命名为可执行文件的名称.

在Linux下,这可以在程序运行时完成,没有任何问题(您只是更改链接,底层文件在映射到它时仍然存在,直到程序关闭或重新启动).

在任何情况下,我都不会重命名原始文件,甚至不会覆盖它.这是不安全的,没有必要.在触摸原始文件之前,您可以执行所有可能在临时文件上失败的"不安全"操作.如果在原子重命名中出现任何问题,您仍然可以使用原始文件.

然后提示用户重新启动程序(如果是交互式的)并完成.