一个基于socket的资源共享平台的实现（四）

现在我们的整个系统基本可用了，还缺少什么？

—— 客户端自动升级。

 

当我们把整个系统部署下去之后，如果打补丁、升级版本等，都需要让用户重新下载的话，肯定用户会觉得很厌烦。

所以我们应该要开发一个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下的界面