#include <winpr/crt.h>
#include <winpr/thread.h>
#include <winpr/collections.h>

static void* message_echo_pipe_client_thread(void* arg)
{
	int index = 0;
	wMessagePipe* pipe = (wMessagePipe*) arg;

	while (index < 100)
	{
		wMessage message;
		int count;

		if (!MessageQueue_Post(pipe->In, NULL, 0, (void*) (size_t) index, NULL))
			break;

		if (!MessageQueue_Wait(pipe->Out))
			break;

		if (!MessageQueue_Peek(pipe->Out, &message, TRUE))
			break;

		if (message.id == WMQ_QUIT)
			break;

		count = (int) (size_t) message.wParam;

		if (count != index)
			printf("Echo count mismatch: Actual: %d, Expected: %d\n", count, index);

		index++;
	}

	MessageQueue_PostQuit(pipe->In, 0);

	return NULL;
}

static void* message_echo_pipe_server_thread(void* arg)
{
	wMessage message;
	wMessagePipe* pipe;

	pipe = (wMessagePipe*) arg;

	while (MessageQueue_Wait(pipe->In))
	{
		if (MessageQueue_Peek(pipe->In, &message, TRUE))
		{
			if (message.id == WMQ_QUIT)
				break;

			if (!MessageQueue_Dispatch(pipe->Out, &message))
				break;
		}
	}

	return NULL;
}

int TestMessagePipe(int argc, char* argv[])
{
	HANDLE ClientThread = NULL;
	HANDLE ServerThread = NULL;
	wMessagePipe* EchoPipe = NULL;
	int ret = 1;

	if (!(EchoPipe = MessagePipe_New()))
	{
		printf("failed to create message pipe\n");
		goto out;
	}

	if (!(ClientThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) message_echo_pipe_client_thread, (void*) EchoPipe, 0, NULL)))
	{
		printf("failed to create client thread\n");
		goto out;
	}

	if (!(ServerThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) message_echo_pipe_server_thread, (void*) EchoPipe, 0, NULL)))
	{
		printf("failed to create server thread\n");
		goto out;
	}

	WaitForSingleObject(ClientThread, INFINITE);
	WaitForSingleObject(ServerThread, INFINITE);

	ret = 0;

out:
	if (EchoPipe)
		MessagePipe_Free(EchoPipe);
	if (ClientThread)
		CloseHandle(ClientThread);
	if (ServerThread)
		CloseHandle(ServerThread);

	return ret;
}