《ASCE1885的源码分析》の简单的进程封装类
一个简单的进程封装类，该类允许我们新建一个远程进程，并对其进行控制。
进程类CProcess的头文件如下：
class CProcess{
public:
PROCESS_INFORMATION ProcessInfo; //进程结构信息
STARTUPINFO StartupInfo; //启动信息
bool KillProcess(UINT); //强制杀死进程，不推荐
bool KillThread(UINT); //强制杀死线程，不推荐
bool LoadProcess(char *, char *, char *); //创建远程进程
bool Pause(void); //挂起主线程
bool Resume(void); //重启主线程
bool UnloadProcess(void); //释放进程和线程资源
bool WaitOnExit(LPDWORD); //等待远程进程的终止
};
进程类CProcess实现文件：
#include <Windows.h>
#include <stdio.h>
#include <stdlib.h>
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include "CProcess.H"
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::LoadProcess(char *ExeName, char *CommandLine, char *CurrentDirectory)
{
ZeroMemory(&StartupInfo, sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo, sizeof(PROCESS_INFORMATION));
StartupInfo.cb = sizeof(STARTUPINFO); //必须的
//Create a suspended Process
BOOL Result = CreateProcess(
ExeName, // pointer to name of executable module
CommandLine, // pointer to command line string
NULL, // pointer to process security attributes
NULL, // pointer to thread security attributes
TRUE, // handle inheritance flag
NORMAL_PRIORITY_CLASS | CREATE_SUSPENDED, // creation flags
NULL, // pointer to new environment block
CurrentDirectory, // pointer to current directory name
&StartupInfo, // pointer to STARTUPINFO
&ProcessInfo // pointer to PROCESS_INFORMATION
);
if(Result)
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::UnloadProcess(void)
{
if(CloseHandle(ProcessInfo.hProcess) && CloseHandle(ProcessInfo.hThread))
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::Pause(void)
{
if(SuspendThread(ProcessInfo.hThread))
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::Resume(void)
{
if(ResumeThread(ProcessInfo.hThread))
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::WaitOnExit(LPDWORD xReturnCode)
{
DWORD ReturnCode = 0;
do{
if(!GetExitCodeProcess(ProcessInfo.hProcess, &ReturnCode))
return false;
Sleep(1);
} while(ReturnCode == STILL_ACTIVE);
xReturnCode = &ReturnCode;
return true;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::KillProcess(UINT ExitCode)
{
if(TerminateProcess(ProcessInfo.hProcess, ExitCode))
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
bool CProcess::KillThread(UINT ExitCode)
{
if(TerminateThread(ProcessInfo.hThread, ExitCode))
return true;
else return false;
}
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
最后就是测试主函数了：
#include <Windows.h>
#include <stdio.h>
#include <stdlib.h>
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include "CProcess/CProcess.H"
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
int main(void)
{
CProcess ASCEProcess;
DWORD ReturnCode = 0;
//调用另外一个进程，初始是挂起状态
if(!ASCEProcess.LoadProcess("D://Program Files//Foxmail//Foxmail.exe", NULL, NULL))
{
printf("Failed to execute the other application/n");
system("pause");
return 0;
}
//重新启动进程
ASCEProcess.Resume();
//等待远程进程的终止
printf("Waiting on remote process to terminate./n");
ASCEProcess.WaitOnExit(&ReturnCode);
//释放资源
ASCEProcess.UnloadProcess();
system("pause");
return 0;
}