FreeRDP/winpr/libwinpr/synch/test/TestSynchWaitableTimerAPC.c
2019-04-05 09:14:35 +02:00

108 lines
2.3 KiB
C

#include <winpr/crt.h>
#include <winpr/synch.h>
#include <winpr/sysinfo.h>
static int g_Count = 0;
static HANDLE g_Event = NULL;
struct apc_data
{
UINT32 StartTime;
};
typedef struct apc_data APC_DATA;
static VOID CALLBACK TimerAPCProc(LPVOID lpArg, DWORD dwTimerLowValue, DWORD dwTimerHighValue)
{
APC_DATA* apcData;
UINT32 CurrentTime = GetTickCount();
WINPR_UNUSED(dwTimerLowValue);
WINPR_UNUSED(dwTimerHighValue);
if (!lpArg)
return;
apcData = (APC_DATA*) lpArg;
printf("TimerAPCProc: time: %"PRIu32"\n", CurrentTime - apcData->StartTime);
g_Count++;
if (g_Count >= 5)
{
SetEvent(g_Event);
}
}
int TestSynchWaitableTimerAPC(int argc, char* argv[])
{
int status = -1;
HANDLE hTimer = NULL;
BOOL bSuccess;
LARGE_INTEGER due;
APC_DATA apcData = { 0 };
WINPR_UNUSED(argc);
WINPR_UNUSED(argv);
g_Event = CreateEvent(NULL, TRUE, FALSE, NULL);
if (!g_Event)
{
printf("Failed to create event\n");
goto cleanup;
}
hTimer = CreateWaitableTimer(NULL, FALSE, NULL);
if (!hTimer)
goto cleanup;
due.QuadPart = -15000000LL; /* 1.5 seconds */
apcData.StartTime = GetTickCount();
bSuccess = SetWaitableTimer(hTimer, &due, 2000, TimerAPCProc, &apcData, FALSE);
if (!bSuccess)
goto cleanup;
/**
* See Remarks at https://msdn.microsoft.com/en-us/library/windows/desktop/ms686786(v=vs.85).aspx
* The SetWaitableTimer completion routine is executed by the thread that activates the timer
* using SetWaitableTimer. However, the thread must be in an ALERTABLE state.
*/
/**
* Note: On WIN32 we need to use WaitForSingleObjectEx with parameter bAlertable = TRUE
* However, WinPR currently (May 2016) does not have a working WaitForSingleObjectEx implementation
* but its non-WIN32 WaitForSingleObject implementations seem to be alertable by WinPR's
* timer implementations.
**/
for (;;)
{
DWORD rc;
#ifdef _WIN32
rc = WaitForSingleObjectEx(g_Event, INFINITE, TRUE);
#else
rc = WaitForSingleObject(g_Event, INFINITE);
#endif
if (rc == WAIT_OBJECT_0)
break;
if (rc == 0x000000C0L) /* WAIT_IO_COMPLETION */
continue;
printf("Failed to wait for completion event (%"PRIu32")\n", GetLastError());
goto cleanup;
}
status = 0;
cleanup:
if (hTimer)
CloseHandle(hTimer);
if (g_Event)
CloseHandle(g_Event);
return status;
}