WIN平台下应用程序利用批处理更新自身

今天又碰到需要做一个简单的自动升级的客户端程序，

客户端程序在拿到升级包并且解压之后，怎样自更新并再启动呢？

这里我封装了一个类，利用批处理，将升级文件覆盖掉本地文件，然后重启动应用程序。

 

#ifndef KILL_MYSELF_H_ #define KILL_MYSELF_H_ #include <iostream> #include <vector> class UpdateMyself { public: UpdateMyself() { m_Files.clear(); } //设置更新目录 void SetDir( std::string dir ) { m_UpdateDir = dir; } //增加更新文件 void AddFile( std::string filename ) { m_Files.push_back( filename ); } //设置启动应用名 void SetStartApp( std::string appname ) { m_StartApp = appname; } //执行 void Execute() { //建立更新批处理文件 FILE *fp; fp = fopen( "update.bat" , "w+" ); fprintf( fp , "@echo off/n" ); //覆盖本地文件 for( std::vector<std::string>::iterator iter = m_Files.begin(); iter != m_Files.end(); ++iter ) { std::string filename = *iter; fprintf( fp , "del %s/n", filename.c_str() ); fprintf( fp , "copy .//%s//%s %s/n", m_UpdateDir.c_str(), filename.c_str(), filename.c_str() ); fprintf( fp , "del .//%s//%s/n", m_UpdateDir.c_str(), filename.c_str() ); } //自杀 fprintf( fp , ":begin/n" ); fprintf( fp , "del %s/n", m_StartApp.c_str() ); fprintf( fp , "if exist %s goto begin/n", m_StartApp.c_str() ); fprintf( fp , "copy .//%s//%s %s/n", m_UpdateDir.c_str(), m_StartApp.c_str(), m_StartApp.c_str() ); //fprintf( fp , "rd/q %s/n", m_UpdateDir.c_str() ); fprintf( fp , "del .//%s//%s/n", m_UpdateDir.c_str(), m_StartApp.c_str() ); fprintf( fp , "start %s/n", m_StartApp.c_str() ); fprintf( fp , "del %%0%%/n"); fclose(fp); WinExec("update.bat",SW_SHOW); exit(0); } private: std::vector<std::string> m_Files; std::string m_UpdateDir; std::string m_StartApp; }; #endif

 

假设我们下载之后的更新文件都位于update目录中，我只需要

 

UpdateMyself test;

test.SetDir("update");

test.AddFile(...);  把要更新的文件名加进去

test.SetStartApp(...); 设置更新之后启动的应用程序名称

test.Execute();

 

就可以完成更新了。