TestMessagePipe.c

 
#include <winpr/crt.h>
#include <winpr/thread.h>
#include <winpr/collections.h>
 
static DWORD WINAPI message_echo_pipe_client_thread(LPVOID arg)
{
  int index = 0;
  wMessagePipe* pipe = (wMessagePipe*)arg;
 
  while (index < 100)
  {
    wMessage message = WINPR_C_ARRAY_INIT;
    int count = -1;
 
    if (!MessageQueue_Post(pipe->In, nullptr, 0, (void*)(size_t)index, nullptr))
      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 0;
}
 
static DWORD WINAPI message_echo_pipe_server_thread(LPVOID arg)
{
  wMessage message = WINPR_C_ARRAY_INIT;
  wMessagePipe* 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 0;
}
 
int TestMessagePipe(int argc, char* argv[])
{
  HANDLE ClientThread = nullptr;
  HANDLE ServerThread = nullptr;
  wMessagePipe* EchoPipe = nullptr;
  int ret = 1;
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  if (!(EchoPipe = MessagePipe_New()))
  {
    printf("failed to create message pipe\n");
    goto out;
  }
 
  if (!(ClientThread = CreateThread(nullptr, 0, message_echo_pipe_client_thread, (void*)EchoPipe,
                                    0, nullptr)))
  {
    printf("failed to create client thread\n");
    goto out;
  }
 
  if (!(ServerThread = CreateThread(nullptr, 0, message_echo_pipe_server_thread, (void*)EchoPipe,
                                    0, nullptr)))
  {
    printf("failed to create server thread\n");
    goto out;
  }
 
  (void)WaitForSingleObject(ClientThread, INFINITE);
  (void)WaitForSingleObject(ServerThread, INFINITE);
 
  ret = 0;
 
out:
  if (EchoPipe)
    MessagePipe_Free(EchoPipe);
  if (ClientThread)
    (void)CloseHandle(ClientThread);
  if (ServerThread)
    (void)CloseHandle(ServerThread);
 
  return ret;
}