一个基于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下的界面