C++ Windows 下 根據進程名獲取進程ID 以及該進程下所有窗口的句柄

轉載自博客： https://www.cnblogs.com/tangxin-blog/p/6077746.html

 

#include <windows.h>
#include <stdint.h>
#include <tlhelp32.h>
#include <stdio.h>
#include <iostream>
#include <vector>

typedef struct EnumHWndsArg
{
    std::vector<HWND> *vecHWnds;
    DWORD dwProcessId;
}EnumHWndsArg, *LPEnumHWndsArg;

HANDLE GetProcessHandleByID(int nID)//通過進程ID獲取進程句柄
{
    return OpenProcess(PROCESS_ALL_ACCESS, FALSE, nID);
}

DWORD GetProcessIDByName(const char* pName)
{
    HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (INVALID_HANDLE_VALUE == hSnapshot) {
        return NULL;
    }
    PROCESSENTRY32 pe = { sizeof(pe) };
    for (BOOL ret = Process32First(hSnapshot, &pe); ret; ret = Process32Next(hSnapshot, &pe)) {
        if (strcmp(pe.szExeFile, pName) == 0) {
            CloseHandle(hSnapshot);
            return pe.th32ProcessID;
        }
        //printf("%-6d %s\n", pe.th32ProcessID, pe.szExeFile);
    }
    CloseHandle(hSnapshot);
    return 0;
}

BOOL CALLBACK lpEnumFunc(HWND hwnd, LPARAM lParam)
{
    EnumHWndsArg *pArg = (LPEnumHWndsArg)lParam;
    DWORD  processId;
    GetWindowThreadProcessId(hwnd, &processId);
    if (processId == pArg->dwProcessId)
    {
        pArg->vecHWnds->push_back(hwnd);
        //printf("%p\n", hwnd);
    }
    return TRUE;
}

void GetHWndsByProcessID(DWORD processID, std::vector<HWND> &vecHWnds)
{
    EnumHWndsArg wi;
    wi.dwProcessId = processID;
    wi.vecHWnds = &vecHWnds;
    EnumWindows(lpEnumFunc, (LPARAM)&wi);
}

int32_t main()
{
    DWORD pid = GetProcessIDByName("notepad++.exe");
    printf("pid = %u\n", pid);
    if (pid != 0)
    {
        std::vector<HWND> vecHWnds;
        GetHWndsByProcessID(pid, vecHWnds);
        printf("vecHWnds.size() = %u\n", vecHWnds.size());
        for (const HWND &h : vecHWnds)
        {
            HWND parent = GetParent(h);
            if (parent == NULL)
            {
                printf("%p --->Main Wnd\n", h);
            }
            else
            {
                printf("%p %p\n", h, parent);
            }
        }
    }
    getchar();
    return S_OK;
}

error C2664: “strcmp”: 不能將參數 1 從“WCHAR [260]”轉換爲“const char *”

參考博客：https://blog.csdn.net/bruce135lee/article/details/80060834