Milestone 5: deliver embedded RDP sessions and lifecycle hardening
This commit is contained in:
105
third_party/FreeRDP/winpr/libwinpr/utils/test/TestMessagePipe.c
vendored
Normal file
105
third_party/FreeRDP/winpr/libwinpr/utils/test/TestMessagePipe.c
vendored
Normal file
@@ -0,0 +1,105 @@
|
||||
|
||||
#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;
|
||||
}
|
||||
Reference in New Issue
Block a user