121 lines
3.1 KiB
C
121 lines
3.1 KiB
C
|
|
#include <winpr/crt.h>
|
|
#include <winpr/sysinfo.h>
|
|
#include <winpr/file.h>
|
|
#include <winpr/synch.h>
|
|
|
|
#define FIRE_COUNT 5
|
|
#define TIMER_COUNT 5
|
|
|
|
struct apc_data
|
|
{
|
|
DWORD TimerId;
|
|
DWORD FireCount;
|
|
DWORD DueTime;
|
|
DWORD Period;
|
|
UINT32 StartTime;
|
|
DWORD MaxFireCount;
|
|
HANDLE CompletionEvent;
|
|
};
|
|
typedef struct apc_data APC_DATA;
|
|
|
|
static VOID CALLBACK TimerRoutine(PVOID lpParam, BOOLEAN TimerOrWaitFired)
|
|
{
|
|
WINPR_UNUSED(TimerOrWaitFired);
|
|
|
|
if (!lpParam)
|
|
return;
|
|
|
|
APC_DATA* apcData = (APC_DATA*)lpParam;
|
|
const UINT32 CurrentTime = GetTickCount();
|
|
const INT64 TimerTime = 1ll * CurrentTime - apcData->StartTime;
|
|
const INT64 expectedTime =
|
|
1ll * apcData->DueTime + (1ll * apcData->Period * apcData->FireCount);
|
|
|
|
apcData->FireCount++;
|
|
|
|
printf("TimerRoutine: TimerId: %" PRIu32 " FireCount: %" PRIu32 " ActualTime: %" PRId64
|
|
" ExpectedTime: %" PRId64 " Discrepancy: %" PRId64 "\n",
|
|
apcData->TimerId, apcData->FireCount, TimerTime, expectedTime, TimerTime - expectedTime);
|
|
|
|
Sleep(11);
|
|
|
|
if (apcData->FireCount == apcData->MaxFireCount)
|
|
{
|
|
(void)SetEvent(apcData->CompletionEvent);
|
|
}
|
|
}
|
|
|
|
int TestSynchTimerQueue(int argc, char* argv[])
|
|
{
|
|
HANDLE hTimers[TIMER_COUNT] = WINPR_C_ARRAY_INIT;
|
|
APC_DATA apcData[TIMER_COUNT] = WINPR_C_ARRAY_INIT;
|
|
|
|
WINPR_UNUSED(argc);
|
|
WINPR_UNUSED(argv);
|
|
|
|
HANDLE hTimerQueue = CreateTimerQueue();
|
|
|
|
if (!hTimerQueue)
|
|
{
|
|
printf("CreateTimerQueue failed (%" PRIu32 ")\n", GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
for (DWORD index = 0; index < TIMER_COUNT; index++)
|
|
{
|
|
apcData[index].TimerId = index;
|
|
apcData[index].StartTime = GetTickCount();
|
|
apcData[index].DueTime = (index * 10) + 50;
|
|
apcData[index].Period = 100;
|
|
apcData[index].FireCount = 0;
|
|
apcData[index].MaxFireCount = FIRE_COUNT;
|
|
|
|
if (!(apcData[index].CompletionEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr)))
|
|
{
|
|
printf("Failed to create apcData[%" PRIu32 "] event (%" PRIu32 ")\n", index,
|
|
GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
if (!CreateTimerQueueTimer(&hTimers[index], hTimerQueue, TimerRoutine, &apcData[index],
|
|
apcData[index].DueTime, apcData[index].Period, 0))
|
|
{
|
|
printf("CreateTimerQueueTimer failed (%" PRIu32 ")\n", GetLastError());
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
for (DWORD index = 0; index < TIMER_COUNT; index++)
|
|
{
|
|
if (WaitForSingleObject(apcData[index].CompletionEvent, 2000) != WAIT_OBJECT_0)
|
|
{
|
|
printf("Failed to wait for timer queue timer #%" PRIu32 " (%" PRIu32 ")\n", index,
|
|
GetLastError());
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
for (DWORD index = 0; index < TIMER_COUNT; index++)
|
|
{
|
|
/**
|
|
* Note: If the CompletionEvent parameter is INVALID_HANDLE_VALUE, the function waits
|
|
* for any running timer callback functions to complete before returning.
|
|
*/
|
|
if (!DeleteTimerQueueTimer(hTimerQueue, hTimers[index], INVALID_HANDLE_VALUE))
|
|
{
|
|
printf("DeleteTimerQueueTimer failed (%" PRIu32 ")\n", GetLastError());
|
|
return -1;
|
|
}
|
|
(void)CloseHandle(apcData[index].CompletionEvent);
|
|
}
|
|
|
|
if (!DeleteTimerQueue(hTimerQueue))
|
|
{
|
|
printf("DeleteTimerQueue failed (%" PRIu32 ")\n", GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|