程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> Windows編程 - 開啟/關閉/遍歷程序的類代碼(C++)

Windows編程 - 開啟/關閉/遍歷程序的類代碼(C++)

編輯：C++入門知識

開啟/關閉/遍歷程序的類代碼(C++)

類包含4個函數, 啟動程序, 遍歷所有進程, 關閉程序, 遍歷進程依賴的動態鏈接庫.

示例: Image.exe是預先生成的可執行程序(exe), 啟動程序, 間隔5秒, 關閉程序.

使用方法參加測試程序.

代碼:

/*
 * process.h
 *
 *  Created on: 2014.06.08
 *      Author: Spike
 */

/*vs 2012*/

#ifndef TRAVERSEPROCESSMODEL_H
#define TRAVERSEPROCESSMODEL_H

#include 
#include 
#include 
#include

#include 
#include      //snapshot

class Process 
{
public:
	bool startProcess (const std::string _name);
	bool terminateProcess (const std::string _name);
	bool traverseModels (const std::string _name);
	bool traverseProcesses (std::map& _nameID);
};

#endif //TRAVERSEPROCESSMODEL_H




/*
 * process.cpp
 *
 *  Created on: 2014.06.08
 *      Author: Spike
 */

/*vs 2012*/

#include process.h

using namespace std;

bool Process::startProcess (const std::string name) 
{
	STARTUPINFO si; //參數設置
	memset(&si, 0, sizeof(STARTUPINFO));
	si.cb = sizeof(STARTUPINFO);
	si.dwFlags = STARTF_USESHOWWINDOW;
	si.wShowWindow = SW_SHOW;

	PROCESS_INFORMATION pi; //參數結束

	/*printf(Please enter the name of process to start:);
	std::string name;
	cin >> name;*/

	if (!CreateProcessA(NULL, (LPSTR)name.c_str(), NULL, NULL, FALSE, 0,NULL,NULL,&si,&pi)) {
		cout<& _nameID) 
{
	PROCESSENTRY32 pe32;
	pe32.dwSize = sizeof(pe32);

	HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
	if (hProcessSnap == INVALID_HANDLE_VALUE) {
		std::cout << CreateToolhelp32Snapshot Error! << std::endl;;
		return false;
	}

	BOOL bResult =Process32First(hProcessSnap, &pe32);

	while(bResult) 
	{
		std::string name = pe32.szExeFile;
		int id = pe32.th32ProcessID;

		//std::cout << Process Name: << name <<    << ProcessID: << id<< std::endl;

		_nameID.insert(std::pair(name, id)); //字典存儲
		bResult = Process32Next(hProcessSnap,&pe32);
	}

	CloseHandle(hProcessSnap);

	return true;
}

bool Process::terminateProcess (const std::string _name) 
{
	std::map nameID;

	if (!traverseProcesses(nameID)) {
		cout << Traverse Processes Error! << endl;
	}

	DWORD dwId;

	/*printf(Please enter the name of process to terminal:);
	std::string name;
	cin >> name;*/

	dwId = nameID[_name];

	BOOL bRet = FALSE;

	HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, dwId);
	if(hProcess != NULL)
		bRet = TerminateProcess(hProcess, 0);
	CloseHandle(hProcess);

	if(bRet){
		std::cout << Terminate Process Success! << std::endl;
	} else {
		std::cout << Terminate Process Error! << std::endl;
		return false;
	}

	return true;
}

bool Process::traverseModels (const std::string _name) 
{
	std::map nameID;

	if (!traverseProcesses(nameID)) {
		cout << Traverse Processes Error! << endl;
	}

	DWORD dwId;

	/*printf(Please enter the name of process to traverse processmodels:);
	std::string name;
	cin >> name;*/

	dwId = nameID[_name];

	HANDLE hModuleSnap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, dwId);
	if(hModuleSnap == INVALID_HANDLE_VALUE){
		printf(CreateToolhelp32SnapshotError! 
);
		return false;
	}

	MODULEENTRY32 module32;
	module32.dwSize = sizeof(module32);
	BOOL bResult = Module32First(hModuleSnap, &module32);

	while(bResult){
		std::wcout << Module: << std::left << std::setw(25) << module32.szModule <<   
			<< endl << Path: << module32.szExePath << std::endl;
		bResult = Module32Next(hModuleSnap, &module32);
	}

	CloseHandle(hModuleSnap);

	return true;

}

測試代碼:

/*
 * main.cpp
 *
 *  Created on: 2014.06.08
 *      Author: Spike
 */

/*vs 2012*/

#include process.h

using namespace std;

int main(void)
{
	Process tpm;

	const std::string name = D:/Test/Image.exe;
	const std::string programme = Image.exe;

	if (!tpm.startProcess(name)) {
		cout << Start Processe Error! << endl;
	}

	Sleep(5000);

	if(!tpm.terminateProcess(programme)){
		cout << Terminate Process Error! << endl;
	}

	return 0;
}

輸出: