记录Window 编程常用的一些小功能函数

#include <Windows.h>
#include <tlhelp32.h>
#include <Psapi.h>
#include <stdio.h>
#pragma comment(lib,"psapi.lib")

//通过进程名获取PID

bool GetPidByProcessName(char* ProcessName,DWORD &PID)
{
	bool ret = false;
	PROCESSENTRY32 pe32;
	pe32.dwSize = sizeof(pe32);
	HANDLE hSnap = NULL;
	hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);
	if(INVALID_HANDLE_VALUE == hSnap)
	{
		printf("创建进程快照失败![]\n");
		return false;
	}
	BOOL bMore = Process32First(hSnap,&pe32);
	while(bMore)
	{
		if(stricmp(pe32.szExeFile,ProcessName) == 0)
		{
			PID = pe32.th32ProcessID;
			ret = true;
			break;
		}
		bMore = Process32Next(hSnap,&pe32);
	}

	return ret;
}

//通过进程名获取全路径]
bool GetPathByProcessName(char* ProcessName,char* FullPath)
{
	bool ret = false;
	PROCESSENTRY32 pe32;
	pe32.dwSize = sizeof(pe32);
	HANDLE hProcess = INVALID_HANDLE_VALUE;
	HANDLE hSnap = INVALID_HANDLE_VALUE;
	hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);
	if(INVALID_HANDLE_VALUE == hSnap)
	{
		printf("创建进程快照失败!\n");
		return false;
	}
	BOOL bMore = Process32First(hSnap,&pe32);
	while(bMore)
	{
		if(stricmp(pe32.szExeFile,ProcessName) == 0)
		{
			hProcess = OpenProcess(PROCESS_ALL_ACCESS,false,pe32.th32ProcessID);
			if(hProcess == INVALID_HANDLE_VALUE)
			{
				printf("打开进程PID失败!\n");
			}
			char Path[256] = {0};
			GetModuleFileNameEx(hProcess,NULL,Path,sizeof(Path));
			strcpy(FullPath,Path);
			ret = true;
			break;
		}
		bMore = Process32Next(hSnap,&pe32);
	}

	return ret;
}

//通过进程名获取TID
bool GetTidByProcessName(char* ProcessName,DWORD &ThreadId)
{
	bool ret = false;
	DWORD ProcessId = 0;
	HANDLE hProcess = INVALID_HANDLE_VALUE;
	THREADENTRY32 pe32;
	pe32.dwSize = sizeof(pe32);
	GetPidByProcessName(ProcessName,ProcessId);
	HANDLE hSnap = INVALID_HANDLE_VALUE;
	hProcess = OpenProcess(PROCESS_CREATE_THREAD | PROCESS_VM_OPERATION	| PROCESS_VM_WRITE, FALSE, ProcessId);
	hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD,ProcessId);
	if(INVALID_HANDLE_VALUE == hSnap)
	{
		printf("创建线程快照失败!\n");
		return false;
	}
	BOOL bMore = Thread32First(hSnap,&pe32);
	while(bMore)
	{
		if(pe32.th32OwnerProcessID == ProcessId)
		{
			ThreadId = pe32.th32ThreadID;
			ret = true;
			break;
		}
		bMore = Thread32Next(hSnap,&pe32);
	}

	return ret;
}

int main()
{
	char *ProcessName = "QQ.exe";
	char Path[256] = {0};
	DWORD PID = 0;
	DWORD TID = 0;
	GetPidByProcessName(ProcessName,PID);
	GetPathByProcessName(ProcessName,Path);
	GetTidByProcessName(ProcessName,TID);
	printf("PID = %d, Path = %s, TID = %d \n",PID,Path,TID);
	return 0;
}