// ProcessController.cpp: implementation of the CProcessController class.
|
//
|
//////////////////////////////////////////////////////////////////////
|
|
|
//#include "ProcessControl.h"
|
#include "stdafx.h"
|
#include "ProcessController.h"
|
|
#include <tlhelp32.h>
|
#include <stdio.h>
|
|
#ifdef _DEBUG
|
#undef THIS_FILE
|
static char THIS_FILE[]=__FILE__;
|
#define new DEBUG_NEW
|
#endif
|
|
//////////////////////////////////////////////////////////////////////
|
// Construction/Destruction
|
//////////////////////////////////////////////////////////////////////
|
|
CProcessController::CProcessController()
|
{
|
|
}
|
|
DWORD CProcessController::TerminateProcessByName(CString ProcessName)
|
{
|
HANDLE hProcessSnap = NULL;
|
DWORD Return = FALSE;
|
PROCESSENTRY32 pe32 = {0};
|
|
ProcessName.MakeLower();
|
|
hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|
if (hProcessSnap == INVALID_HANDLE_VALUE)
|
return (DWORD)INVALID_HANDLE_VALUE;
|
|
pe32.dwSize = sizeof(PROCESSENTRY32);
|
|
if (Process32First(hProcessSnap, &pe32))
|
{
|
DWORD Code = 0;
|
DWORD dwPriorityClass;
|
|
do
|
{
|
HANDLE hProcess;
|
|
// Get the actual priority class.
|
hProcess = OpenProcess (PROCESS_ALL_ACCESS,
|
FALSE, pe32.th32ProcessID);
|
dwPriorityClass = GetPriorityClass (hProcess);
|
|
CString Temp = pe32.szExeFile;
|
Temp.MakeLower();
|
|
if (Temp == ProcessName)
|
{
|
if (TerminateProcess(hProcess, 0))
|
{
|
GetExitCodeProcess(hProcess, &Code);
|
}
|
else
|
{
|
return GetLastError();
|
}
|
}
|
|
CloseHandle (hProcess);
|
}
|
while (Process32Next(hProcessSnap, &pe32));
|
Return = TRUE;
|
}
|
else
|
Return = FALSE; // could not walk the list of processes
|
|
// Do not forget to clean up the snapshot object.
|
|
CloseHandle (hProcessSnap);
|
|
return Return;
|
}
|
|
int CProcessController::GetProcessList(PPROCESSLIST pProcessList, int MaxNum)
|
{
|
int Index = 0;
|
HANDLE hProcessSnap = NULL;
|
DWORD Return = FALSE;
|
PROCESSENTRY32 pe32 = {0};
|
|
hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|
if (hProcessSnap == INVALID_HANDLE_VALUE)
|
return 0;
|
|
pe32.dwSize = sizeof(PROCESSENTRY32);
|
|
if (Process32First(hProcessSnap, &pe32))
|
{
|
DWORD Code = 0;
|
DWORD dwPriorityClass;
|
|
do
|
{
|
HANDLE hProcess;
|
|
// Get the actual priority class.
|
hProcess = OpenProcess (PROCESS_ALL_ACCESS,
|
FALSE, pe32.th32ProcessID);
|
dwPriorityClass = GetPriorityClass (hProcess);
|
|
if (Index < MaxNum)
|
{
|
pProcessList[Index].ProcessName = pe32.szExeFile;
|
Index++;
|
}
|
|
CloseHandle (hProcess);
|
|
}
|
while (Process32Next(hProcessSnap, &pe32));
|
}
|
else
|
return 0;
|
// could not walk the list of processes
|
|
// Do not forget to clean up the snapshot object.
|
|
CloseHandle (hProcessSnap);
|
|
return Index;
|
}
|
|
CProcessController::~CProcessController()
|
{
|
|
}
|
