#include <stdlib.h>

#include <winpr/crt.h>

#include <winpr/synch.h>

#include <winpr/thread.h>

static DWORD WINAPI test_thread(LPVOID arg)

{

	long timeout = rand();

	timeout %= 1000;

	timeout += 100;

	Sleep(timeout);

	ExitThread(0);

	return 0;

}

static int start_threads(DWORD count, HANDLE *threads)

{

	DWORD i;

	for (i=0; i

	{

		threads[i] = CreateThread(NULL, 0, test_thread, NULL, 0, NULL);

		if (!threads[i])

		{

			printf("CreateThread [%"PRIu32"] failure\n", i);

			return -1;

		}

	}

	return 0;

}

static int close_threads(DWORD count, HANDLE *threads)

{

	DWORD i;

	for (i=0; i

	{

		if (!CloseHandle(threads[i]))

		{

			printf("CloseHandle [%"PRIu32"] failure\n", i);

			return -1;

		}

	}

	return 0;

}

int TestSynchMultipleThreads(int argc, char *argv[])

{

#define THREADS 24

	DWORD rc = 0, ev, i;

	HANDLE threads[THREADS];

	/* WaitForAll, timeout */

	if (start_threads(THREADS, threads))

		return 1;

	if (WaitForMultipleObjects(THREADS, threads, TRUE, 50) != WAIT_TIMEOUT)

	{

		printf("WaitForMultipleObjects bWaitAll, timeout 50 failed\n");

		rc = 2;

	}

	if (WaitForMultipleObjects(THREADS, threads, TRUE, INFINITE) != WAIT_OBJECT_0)

	{

		printf("WaitForMultipleObjects bWaitAll, INFINITE failed\n");

		rc = 3;

	}

	if (close_threads(THREADS, threads))

		return 4;

	/* WaitOne, infinite */

	if (rc)

		return rc;

	if (start_threads(THREADS, threads))

		return 5;

	ev = WaitForMultipleObjects(THREADS, threads, FALSE, INFINITE);

	if ((ev < WAIT_OBJECT_0) || (ev > (WAIT_OBJECT_0 + THREADS)))

	{

		printf("WaitForMultipleObjects INFINITE failed\n");

		rc = 6;

	}

	if (WaitForMultipleObjects(THREADS, threads, TRUE, INFINITE) != WAIT_OBJECT_0)

	{

		printf("WaitForMultipleObjects bWaitAll, INFINITE failed\n");

		rc = 7;

	}

	if (close_threads(THREADS, threads))

		return 8;

	if (rc)

		return rc;

	/* WaitOne, timeout */

	if (start_threads(THREADS, threads))

		return 9;

	if (WaitForMultipleObjects(THREADS, threads, FALSE, 50) != WAIT_TIMEOUT)

	{

		printf("WaitForMultipleObjects timeout 50 failed\n");

		rc = 10;

	}

	if (WaitForMultipleObjects(THREADS, threads, TRUE, INFINITE) != WAIT_OBJECT_0)

	{

		printf("WaitForMultipleObjects bWaitAll, INFINITE failed\n");

		rc = 11;

	}

	if (close_threads(THREADS, threads))

		return 12;

	if (rc)

		return 13;

	/* WaitOne, timeout, multiple joins */

	if (start_threads(THREADS, threads))

		return 14;

	for (i = 0; i < THREADS; i++)

	{

		if (WaitForMultipleObjects(THREADS, threads, FALSE, 0) != WAIT_TIMEOUT)

		{

			printf("WaitForMultipleObjects timeout 50 failed\n");

			rc = 15;

		}

	}

	if (WaitForMultipleObjects(THREADS, threads, TRUE, INFINITE) != WAIT_OBJECT_0)

	{

		printf("WaitForMultipleObjects bWaitAll, INFINITE failed\n");

		rc = 16;

	}

	if (close_threads(THREADS, threads))

		rc = 17;

	if (rc)

		return rc;

	/* Thread detach test */

	if (start_threads(THREADS, threads))

		return 18;

	if (close_threads(THREADS, threads))

		return 19;

	return 0;

}