VC++.模块遍历2

#include <Psapi.h>
#include <vector>
BOOL TraversalModule2(OUT std::vector<HMODULE> &vec, IN DWORD dwPid)
{
    vec.clear();
    try
    {
        HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, 0, dwPid);
        if (hProcess)
        {
            BYTE *pbt = 0;
            DWORD dwSize = 0, dw = 0x1000;
            BOOL b = 0;
            for (; dw < INT_MAX && !b; dw *= 2)
            {
                if (pbt)
                {
                    delete pbt;
                    pbt = 0;
                }
                pbt = new BYTE[dw];
                if (pbt)
                {
                    b = EnumProcessModules(hProcess, (HMODULE*)pbt, dw, &dwSize);
                }
            }
            if (pbt && b)
            {
                HMODULE *pdw = (HMODULE*)pbt;
                for (DWORD i = 0; i < dwSize / sizeof(HMODULE); ++i)
                {
                    vec.push_back(pdw[i]);
                }
                delete pbt;
                pbt = 0;
            }
            CloseHandle(hProcess);
        }
    }
    catch (...)
    {
        OutputDebugStringA(__FUNCTION__);
        return 0;
    }
    return vec.size() > 0;
}

int main()
{
    STARTUPINFO si = { 0 };
    si.cb = sizeof(si);
    PROCESS_INFORMATION pi = { 0 };
    BOOL b = CreateProcess(_T("C:/Using/winmine.exe"), 0, 0, 0, 0, 0, 0, 0, &si, &pi);
    if (b)
    {
        Sleep(1000);
        std::vector<HMODULE> vec;
        std::cout << TraversalModule2(vec, pi.dwProcessId) << endl;
        for each (HMODULE var in vec)
        {
            printf("%08X\t", var);
            printf("\r\n");
        }
        TerminateProcess(OpenProcess(PROCESS_ALL_ACCESS, 0, pi.dwProcessId), 0);
    }

    return 0;
}