PowerToys/src/common/utils/processApi.h

#pragma once

#include <vector>
#include <wil/resource.h>
#include <Shlwapi.h>
#include <Psapi.h>
#include <string_view>

inline std::vector<wil::unique_process_handle> getProcessHandlesByName(const std::wstring_view processName, DWORD handleAccess)
{
    std::vector<wil::unique_process_handle> result;
    DWORD bytesRequired;
    std::vector<DWORD> processIds;
    processIds.resize(4096 / sizeof(processIds[0]));
    auto processIdSize = static_cast<DWORD>(size(processIds) * sizeof(processIds[0]));
    EnumProcesses(processIds.data(), processIdSize, &bytesRequired);
    while (bytesRequired == processIdSize)
    {
        processIdSize *= 2;
        processIds.resize(processIdSize / sizeof(processIds[0]));
        EnumProcesses(processIds.data(), processIdSize, &bytesRequired);
    }
    processIds.resize(bytesRequired / sizeof(processIds[0]));

    handleAccess |= PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ;
    for (const DWORD processId : processIds)
    {
        try
        {
            wil::unique_process_handle hProcess{ OpenProcess(handleAccess, FALSE, processId) };
            wchar_t name[MAX_PATH + 1];
            if (!hProcess || !GetProcessImageFileNameW(hProcess.get(), name, MAX_PATH))
            {
                continue;
            }
            if (processName == PathFindFileNameW(name))
            {
                result.push_back(std::move(hProcess));
            }
        }
        catch (...)
        {
        }
    }
    return result;
}