TestSynchInit.c

#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/crypto.h>
#include <winpr/synch.h>
#include <winpr/thread.h>
#include <winpr/interlocked.h>
 
#define TEST_NUM_THREADS 100
#define TEST_NUM_FAILURES 10
 
static INIT_ONCE initOnceTest = INIT_ONCE_STATIC_INIT;
 
static HANDLE hStartEvent = NULL;
static LONG* pErrors = NULL;
static LONG* pTestThreadFunctionCalls = NULL;
static LONG* pTestOnceFunctionCalls = NULL;
static LONG* pInitOnceExecuteOnceCalls = NULL;
 
static UINT32 prand(UINT32 max)
{
  UINT32 tmp = 0;
  if (max <= 1)
    return 1;
  winpr_RAND(&tmp, sizeof(tmp));
  return tmp % (max - 1) + 1;
}
 
static BOOL CALLBACK TestOnceFunction(PINIT_ONCE once, PVOID param, PVOID* context)
{
  LONG calls = InterlockedIncrement(pTestOnceFunctionCalls) - 1;
 
  WINPR_UNUSED(once);
  WINPR_UNUSED(param);
  WINPR_UNUSED(context);
 
  /* simulate execution time */
  Sleep(30 + prand(40));
 
  if (calls < TEST_NUM_FAILURES)
  {
    /* simulated error */
    return FALSE;
  }
  if (calls == TEST_NUM_FAILURES)
  {
    return TRUE;
  }
  (void)fprintf(stderr, "%s: error: called again after success\n", __func__);
  InterlockedIncrement(pErrors);
  return FALSE;
}
 
static DWORD WINAPI TestThreadFunction(LPVOID lpParam)
{
  LONG calls = 0;
  BOOL ok = 0;
 
  WINPR_UNUSED(lpParam);
 
  InterlockedIncrement(pTestThreadFunctionCalls);
  if (WaitForSingleObject(hStartEvent, INFINITE) != WAIT_OBJECT_0)
  {
    (void)fprintf(stderr, "%s: error: failed to wait for start event\n", __func__);
    InterlockedIncrement(pErrors);
    return 0;
  }
 
  ok = InitOnceExecuteOnce(&initOnceTest, TestOnceFunction, NULL, NULL);
  calls = InterlockedIncrement(pInitOnceExecuteOnceCalls);
  if (!ok && calls > TEST_NUM_FAILURES)
  {
    (void)fprintf(stderr, "%s: InitOnceExecuteOnce failed unexpectedly\n", __func__);
    InterlockedIncrement(pErrors);
  }
  return 0;
}
 
int TestSynchInit(int argc, char* argv[])
{
  HANDLE hThreads[TEST_NUM_THREADS];
  DWORD dwCreatedThreads = 0;
  BOOL result = FALSE;
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  pErrors = winpr_aligned_malloc(sizeof(LONG), sizeof(LONG));
  pTestThreadFunctionCalls = winpr_aligned_malloc(sizeof(LONG), sizeof(LONG));
  pTestOnceFunctionCalls = winpr_aligned_malloc(sizeof(LONG), sizeof(LONG));
  pInitOnceExecuteOnceCalls = winpr_aligned_malloc(sizeof(LONG), sizeof(LONG));
 
  if (!pErrors || !pTestThreadFunctionCalls || !pTestOnceFunctionCalls ||
      !pInitOnceExecuteOnceCalls)
  {
    (void)fprintf(stderr, "error: _aligned_malloc failed\n");
    goto out;
  }
 
  *pErrors = 0;
  *pTestThreadFunctionCalls = 0;
  *pTestOnceFunctionCalls = 0;
  *pInitOnceExecuteOnceCalls = 0;
 
  if (!(hStartEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
  {
    (void)fprintf(stderr, "error creating start event\n");
    InterlockedIncrement(pErrors);
    goto out;
  }
 
  for (DWORD i = 0; i < TEST_NUM_THREADS; i++)
  {
    if (!(hThreads[i] = CreateThread(NULL, 0, TestThreadFunction, NULL, 0, NULL)))
    {
      (void)fprintf(stderr, "error creating thread #%" PRIu32 "\n", i);
      InterlockedIncrement(pErrors);
      goto out;
    }
    dwCreatedThreads++;
  }
 
  Sleep(100);
  (void)SetEvent(hStartEvent);
 
  for (DWORD i = 0; i < dwCreatedThreads; i++)
  {
    if (WaitForSingleObject(hThreads[i], INFINITE) != WAIT_OBJECT_0)
    {
      (void)fprintf(stderr, "error: error waiting for thread #%" PRIu32 "\n", i);
      InterlockedIncrement(pErrors);
      goto out;
    }
  }
 
  if (*pErrors == 0 && *pTestThreadFunctionCalls == TEST_NUM_THREADS &&
      *pInitOnceExecuteOnceCalls == TEST_NUM_THREADS &&
      *pTestOnceFunctionCalls == TEST_NUM_FAILURES + 1)
  {
    result = TRUE;
  }
 
out:
  (void)fprintf(stderr, "Test result:              %s\n", result ? "OK" : "ERROR");
  (void)fprintf(stderr, "Error count:              %" PRId32 "\n", pErrors ? *pErrors : -1);
  (void)fprintf(stderr, "Threads created:          %" PRIu32 "\n", dwCreatedThreads);
  (void)fprintf(stderr, "TestThreadFunctionCalls:  %" PRId32 "\n",
                pTestThreadFunctionCalls ? *pTestThreadFunctionCalls : -1);
  (void)fprintf(stderr, "InitOnceExecuteOnceCalls: %" PRId32 "\n",
                pInitOnceExecuteOnceCalls ? *pInitOnceExecuteOnceCalls : -1);
  (void)fprintf(stderr, "TestOnceFunctionCalls:    %" PRId32 "\n",
                pTestOnceFunctionCalls ? *pTestOnceFunctionCalls : -1);
 
  winpr_aligned_free(pErrors);
  winpr_aligned_free(pTestThreadFunctionCalls);
  winpr_aligned_free(pTestOnceFunctionCalls);
  winpr_aligned_free(pInitOnceExecuteOnceCalls);
 
  (void)CloseHandle(hStartEvent);
 
  for (DWORD i = 0; i < dwCreatedThreads; i++)
  {
    (void)CloseHandle(hThreads[i]);
  }
 
  return (result ? 0 : 1);
}