123 lines
2.9 KiB
C++
123 lines
2.9 KiB
C++
|
//=======================================================================
|
||
|
|
//
|
||
|
|
// Copyright (c) 1998-2000 Microsoft Corporation. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// File: WaitUtil.h
|
||
|
|
//
|
||
|
|
// Description:
|
||
|
|
//
|
||
|
|
// IU wait message utility library
|
||
|
|
//
|
||
|
|
//=======================================================================
|
||
|
|
|
||
|
|
#include "WaitUtil.h"
|
||
|
|
|
||
|
|
DWORD WaitAndPumpMessages(DWORD nCount, LPHANDLE pHandles, DWORD dwWakeMask)
|
||
|
|
{
|
||
|
|
DWORD dwWaitResult;
|
||
|
|
MSG msg;
|
||
|
|
|
||
|
|
while (TRUE)
|
||
|
|
{
|
||
|
|
dwWaitResult = MsgWaitForMultipleObjects(nCount, pHandles, FALSE, 1000, dwWakeMask);
|
||
|
|
if (dwWaitResult <= WAIT_OBJECT_0 + nCount - 1)
|
||
|
|
{
|
||
|
|
return dwWaitResult;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (WAIT_OBJECT_0 + nCount == dwWaitResult)
|
||
|
|
{
|
||
|
|
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
|
||
|
|
{
|
||
|
|
TranslateMessage(&msg);
|
||
|
|
DispatchMessage(&msg);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return dwWaitResult;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD MyMsgWaitForMultipleObjects(DWORD nCount, LPHANDLE pHandles, BOOL fWaitAll, DWORD dwMilliseconds, DWORD dwWakeMask)
|
||
|
|
{
|
||
|
|
DWORD dwTickStart;
|
||
|
|
DWORD dwWaitResult;
|
||
|
|
DWORD dwLoopMS = 250; // default 250 ms timeout for MsgWaitForMultipleObjects
|
||
|
|
MSG msg;
|
||
|
|
|
||
|
|
dwTickStart = GetTickCount();
|
||
|
|
|
||
|
|
if (dwLoopMS > dwMilliseconds)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Never wait more than dwMilliseconds
|
||
|
|
//
|
||
|
|
dwLoopMS = dwMilliseconds;
|
||
|
|
}
|
||
|
|
|
||
|
|
while (TRUE)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Empty message queue before calling MsgWaitForMultipleObjects or any
|
||
|
|
// existing messages will not be processed until a new message arrives
|
||
|
|
// in the queue.
|
||
|
|
//
|
||
|
|
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
|
||
|
|
{
|
||
|
|
switch (msg.message)
|
||
|
|
{
|
||
|
|
case WM_QUIT:
|
||
|
|
case WM_CLOSE:
|
||
|
|
case WM_DESTROY:
|
||
|
|
{
|
||
|
|
// if the message is one that indicates we're trying to close down, we'll signal the abort
|
||
|
|
// and leave.
|
||
|
|
dwWaitResult = ERROR_REQUEST_ABORTED;
|
||
|
|
return dwWaitResult;
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
TranslateMessage(&msg);
|
||
|
|
DispatchMessage(&msg);
|
||
|
|
}
|
||
|
|
|
||
|
|
dwWaitResult = MsgWaitForMultipleObjects(nCount, pHandles, fWaitAll, dwLoopMS, dwWakeMask);
|
||
|
|
|
||
|
|
if (dwWaitResult <= WAIT_OBJECT_0 + nCount - 1)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// One (or all depending on fWaitAll) of the objects is signaled, return dwWaitResult
|
||
|
|
//
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// NOTE: we ignore WAIT_ABANDONED_0 + n cases and just time out since our callers
|
||
|
|
// don't handle this special case.
|
||
|
|
//
|
||
|
|
|
||
|
|
//
|
||
|
|
// Stop pumping messages after dwMilliseconds
|
||
|
|
//
|
||
|
|
// Timer wraparound handled by unsigned subtract
|
||
|
|
//
|
||
|
|
if (GetTickCount() - dwTickStart >= dwMilliseconds)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// No need to continue, even if caused by new message (WAIT_OBJECT_0 + nCount == dwWaitResult),
|
||
|
|
// we have reached our dwMilliseconds timeout
|
||
|
|
//
|
||
|
|
dwWaitResult = WAIT_TIMEOUT;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Otherwise continue, WAIT_TIMEOUT from MsgWaitForMultipleObjects is only case left
|
||
|
|
//
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
return dwWaitResult;
|
||
|
|
}
|