现在我们的整个系统基本可用了,还缺少什么?
—— 客户端自动升级。
当我们把整个系统部署下去之后,如果打补丁、升级版本等,都需要让用户重新下载的话,肯定用户会觉得很厌烦。
所以我们应该要开发一个loader,客户端能够自动检测更新、升级,并可以在服务器端打包升级流及升级流信息下发。
在此系统中,我直接是服务器下发一条升级信息(带版本号),然后与客户端当前版本号进行配对,若高于客户端,则开始客户端更新升级。
这里由于我们的客户端很小,所以我直接将客户端exe通过TCP传输下来。客户端接收完毕之后,自我关闭,换成更新过的exe启动。
这里有个小伎俩,一个exe程序如何升级自身?
当时我上网找了许多,都觉得太麻烦,于是自己想了一个“歪门邪道”,不过真的很好用,这里跟大家分享一下。
假设我们正在运行的客户端程序叫做 1.exe,我们通过TCP拿到的升级后的程序叫做 2.exe。
我们此时需要在1.exe中将自身结束,将2.exe改名成1.exe并且启动它。问题就出在这一步,既然已经把自己都结束了,如何去讲2.exe改名和启动呢?
——通过bat。
我们在1.exe运行过程中动态创建一个批处理文件,其干以下事情:
start:
若存在1.exe ,删除之,若删除不了,返回start;
将2.exe改名为1.exe;
启动1.exe;
将自己删掉;
注意最后在1.exe中我们需要异步启动bat(很显然,不然死循环了),这样就可以实现我们所说的功能啦。
部分代码如下:
void CNetShareDlg::UpDateClient(char* Buffer) { wchar_t* pVer = NetShareIndexManager::AnsiToUnicode(Buffer + 2); CString Version(pVer); delete pVer; NSConfig con = ObjCfg.Get(); string s1( Buffer + 2); char* p = NetShareIndexManager::UnicodeToAnsi(con.Version); string s2(p); delete p; if( s1 != s2 && gDownloadingBusy == false ) //版本不相同,启动自动升级 { //发送从服务器下载信息 string sendbuffer; sendbuffer.push_back(NS_UDP_CLIENT_GETEXE); sendbuffer.push_back('\n'); SendToServer(sendbuffer); //从服务器下载文件 gDownloadingBusy = true; cpplib::resource::MutexLock MyMutex(FileDownloadMutex); TCP_Client::Instance()->SetFileSize(gDownloadingFileSize); TCP_Client::Instance()->GetFile("update.dat",TCP_TRANCESPORT_PORT); //自杀,重启 KillTimer(TIME_EVENT_LIVING);//不再发送心跳包 AfxMessageBox(_T("自动更新成功,将启动新版本")); //获取自身文件名 TCHAR filename[1024]; ::GetModuleFileName( NULL, filename, 1024 ); CString str( filename ); int pos = str.ReverseFind(_T('\\')); CString MyFileName = str.Mid(pos + 1,str.GetLength() - pos); char* p = NetShareIndexManager::UnicodeToAnsi(MyFileName.GetBuffer(0)); //建立更新批处理文件 FILE *fp; fp = fopen( "update.bat" , "w+" ); fprintf( fp , "@echo off\n" ); fprintf( fp , ":begin\n" ); fprintf( fp , "del %s\n", p ); fprintf( fp , "if exist %s goto begin\n", p ); fprintf( fp , "copy update.dat %s\n", p ); fprintf( fp , "del update.dat\n" ); fprintf( fp , "start %s\n", p ); fprintf( fp , "del %%0%%\n"); fclose(fp); delete p; con.Version = Version; ObjCfg.Save( con );//保存新版本号 WinExec("update.bat",SW_SHOW); gDownloadingBusy = false; exit(0); } }
最后看看我们的丑陋的MFC下的界面