巧妙解决 Windows 中进程重复启动难题：WaitForInputIdle 拯救记事本进程

windows

2024-03-06 19:59:55

在 Windows 中巧妙监控和重启进程：终结重复启动的烦恼

前言

在 Windows 环境中，监控和管理进程是系统维护和应用程序开发的关键任务。本文将深入探究一种常见问题及其巧妙解决方法：如何监控进程并仅在该进程终止后才重新启动它。

问题陈述：记事本进程的意外重启

假设我们的目标是创建一个程序，监控记事本进程（notepad.exe），并在其关闭后重新启动。然而，我们面临着一个棘手的难题：在程序启动后，回调函数意外地立即被调用，导致记事本进程重复启动。

深入问题根源

深入分析问题后，我们发现关键在于记事本进程的启动过程。RegisterWaitForSingleObject 函数需要一个进程句柄来监控该进程。然而，在程序启动之初，记事本进程尚未完全启动，注册会导致回调函数被错误调用。

巧妙解决方案：waitForInputIdle 拯救现场

为了解决这个问题，我们需要在注册 RegisterWaitForSingleObject 之前，等待记事本进程完成启动。这就是 WaitForInputIdle 函数发挥作用的地方。它允许我们暂停程序，直到进程完成初始化。

修改后的代码：等待进程启动后再监控

修改后的代码如下：

// ...

shellExecInfo = ShellExecEx(nullptr, "open", notepadExePath, "", "", SW_SHOWNORMAL, SEE_MASK_NOCLOSEPROCESS);

auto code = reinterpret_cast<INT_PTR>(shellExecInfo.hInstApp);
if (code <= 32)
{
    cout << "main(): Error Starting Notepad:" << GetLastError() << endl;
    return 1;
}

DWORD w365ProcessId = GetProcessId(shellExecInfo.hProcess);
cout << "main(): Process started successfully pid:" << w365ProcessId << endl;

// 等待 notepad.exe 完成初始化。
WaitForInputIdle(shellExecInfo.hProcess, INFINITE);

if (!RegisterWaitForSingleObject(
    &hWaitHandle,
    shellExecInfo.hProcess,
    ProcessWaitCallback,
    NULL,
    INFINITE,
    WT_EXECUTEONLYONCE
))
{
    cout << "main(): Error registering wait operation: " << GetLastError() << std::endl;
}

// ...