C++程序自动重启的实现代码.docxVIP

第

C++程序自动重启的实现代码

目录一、自动重启的原理二、自动重启的实现三、相关知识点3.1CommandLineToArg3.2LocalFree3.3OpenProcess3.4WaitForSingleObject3.5GetExitCodeProcess3.6GetModuleFileName3.7GetStartupInfo3.8CreateProcess拓展

一、自动重启的原理

我不知道为什么很多程序员觉得自动重启很low，就像我始终不明白为什么有些人一听见我说重新编译一下就笑，难道不是重新编译一下大部分问题就解决了吗？

自动重启原理很简单，用一个进程监控另一个进程，挂了就再启动一个。细节也不算多，主要是正确判断进程状态和启动方式，其实最大的工作量是程序恢复时应该如何回到原来的状态，这意味着程序要随时保存状态。

只要你能做到用户无感，你在背后做了什么用户在意吗？

二、自动重启的实现

如果是UNIX，用fork然后监控子进程，挂了就再fork，一个循环就解决问题了。

windows上麻烦一些，监控进程，控制台程序：

#includestdafx.h

#includeshellapi.h

#includestdio.h

#includestring

usingnamespacestd;

boolbDebug=false;

intAPIENTRYwWinMain(_In_HINSTANCEhInstance,

_In_opt_HINSTANCEhPrevInstance,

_In_LPWSTRlpCmdLine,

_In_intnCmdShow)

UNREFERENCED_PARAMETER(hPrevInstance);

//TODO:在此处放置代码。

if(bDebug)MessageBox(NULL,lpCmdLine,TEXT(启动),0);

wchar_tbuf[256];

DWORDpid=0;

LPWSTR*szArglist;

intnArgs;

inti;

//从命令行获取要监控的进程的PID，参数-pid后的下一个参数

szArglist=CommandLineToArgvW(lpCmdLine,nArgs);//注意，如果没有参数，会返回程序名，如果有参数，则不包括程序名（或许是个BUG）

if(NULL==szArglist)

MessageBox(NULL,lpCmdLine,TEXT(CommandLineToArgvW失败),0);

return0;

else

wsprintf(buf,TEXT(参数个数%d),nArgs);

if(bDebug)MessageBox(NULL,buf,TEXT(),0);

for(i=0;inArgs;++i)

if(bDebug)MessageBox(NULL,szArglist[i],TEXT(CommandLineToArgvW),0);

if(0==_tcsicmp(szArglist[i],TEXT(-pid))i+1nArgs)

pid=_wtol(szArglist[i+1]);

wsprintf(buf,TEXT(%u),pid);

if(bDebug)MessageBox(NULL,buf,szArglist[i+1],0);

LocalFree(szArglist);

wsprintf(buf,TEXT(%u),pid);

if(bDebug)MessageBox(NULL,buf,TEXT(pid),0);

//打开进程以供监控

HANDLEhandle=OpenProcess(PROCESS_QUERY_INFORMATION|SYNCHRONIZE,FALSE,pid);

//等待进程结束

DWORDstate=WaitForSingleObject(handle,INFINITE);

if(WAIT_OBJECT_0==state)

DWORDexitCode;

if(!GetExitCodeProcess(handle,exitCode))//获得退出码

MessageBox(NULL,TEXT(GetExitCodeProcess出错),TEXT(