啟動一個進程並獲得其主視窗控制代碼

通過以下兩個函數可以實現啟動一個進程並獲得其主視窗控制代碼。

BOOL CALLBACK CDSUpdate::EnumDSUMWindowsProc(HWND hwnd, LPARAM lParam)

{

    DWORD dwID;

    EnumParam* pEP = (EnumParam*)lParam;

    GetWindowThreadProcessId(hwnd, &dwID);

    if (dwID == pEP->m_dwProcessID)

    {

        pEP->m_hMainWnd = hwnd;

        return false;

    }

    return true;

}

HWND CDSUpdate::RunDSUM(CString sDSUMFile)

{

    EnumParam ep;

    STARTUPINFO si;

    PROCESS_INFORMATION pi;

    CString sFolder;

    ZeroMemory(&si, sizeof(STARTUPINFO));

    si.cb = sizeof(STARTUPINFO);

    si.wShowWindow = SW_SHOW;

    ZeroMemory(&pi, sizeof(PROCESS_INFORMATION));

    UpdateLog(::Message, L"(RunDSUM): Start DSUM.");

    if (!CreateProcess(NULL, (LPTSTR)(LPCTSTR)sDSUMFile, NULL, NULL, FALSE, 0, NULL, sDSUMFile.Left(sDSUMFile.ReverseFind('//')), &si, &pi))

    {

        UpdateLog(::Error, L"(RunDSUM): Start DSUM fail (%d).", GetLastError());

        return NULL;

    }

    // Close process and thread handles.

    CloseHandle(pi.hProcess);

    CloseHandle(pi.hThread);

 

    ep.m_dwProcessID = pi.dwProcessId;

    ep.m_hMainWnd = NULL;

    for (int i = 0; i < 50; i++)     // Retry about 10s.

    {

        Sleep(200);

        EnumWindows(EnumDSUMWindowsProc, (LPARAM)&ep);

        if (ep.m_hMainWnd)

            break;

    }

    if (!ep.m_hMainWnd)

    {

        UpdateLog(::Error, L"(RunDSUM): Couldn't find the hwnd of DSUM.");

        return NULL;

    }

    return ep.m_hMainWnd;

}