ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ntdll/tests: Add tests for kernel32 threadpool I/O functions. 

Signed-off-by: Zebediah Figura <z.figura12@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2020-04-11 00:16:27 -05:00 committed by Alexandre Julliard
parent 75e8c4493a
commit 320a025882
1 changed files with 189 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

226
dlls/ntdll/tests/threadpool.c
View File

@ -22,7 +22,6 @@
#define NONAMELESSUNION #define NONAMELESSUNION
#include "ntdll_test.h" #include "ntdll_test.h"
static HMODULE hntdll = 0;
static NTSTATUS (WINAPI *pTpAllocCleanupGroup)(TP_CLEANUP_GROUP **); static NTSTATUS (WINAPI *pTpAllocCleanupGroup)(TP_CLEANUP_GROUP **);
static NTSTATUS (WINAPI *pTpAllocIoCompletion)(TP_IO **,HANDLE,PTP_IO_CALLBACK,void *,TP_CALLBACK_ENVIRON *); static NTSTATUS (WINAPI *pTpAllocIoCompletion)(TP_IO **,HANDLE,PTP_IO_CALLBACK,void *,TP_CALLBACK_ENVIRON *);
static NTSTATUS (WINAPI *pTpAllocPool)(TP_POOL **,PVOID); static NTSTATUS (WINAPI *pTpAllocPool)(TP_POOL **,PVOID);
@ -52,51 +51,58 @@ static VOID (WINAPI *pTpWaitForTimer)(TP_TIMER *,BOOL);
static VOID (WINAPI *pTpWaitForWait)(TP_WAIT *,BOOL); static VOID (WINAPI *pTpWaitForWait)(TP_WAIT *,BOOL);
static VOID (WINAPI *pTpWaitForWork)(TP_WORK *,BOOL); static VOID (WINAPI *pTpWaitForWork)(TP_WORK *,BOOL);
#define NTDLL_GET_PROC(func) \ static void (WINAPI *pCancelThreadpoolIo)(TP_IO *);
static void (WINAPI *pCloseThreadpoolIo)(TP_IO *);
static TP_IO *(WINAPI *pCreateThreadpoolIo)(HANDLE, PTP_WIN32_IO_CALLBACK, void *, TP_CALLBACK_ENVIRON *);
static void (WINAPI *pStartThreadpoolIo)(TP_IO *);
static void (WINAPI *pWaitForThreadpoolIoCallbacks)(TP_IO *, BOOL);
#define GET_PROC(func) \
do \ do \
{ \ { \
p ## func = (void *)GetProcAddress(hntdll, #func); \ p ## func = (void *)GetProcAddress(module, #func); \
if (!p ## func) trace("Failed to get address for %s\n", #func); \ if (!p ## func) trace("Failed to get address for %s\n", #func); \
} \ } \
while (0) while (0)
static BOOL init_threadpool(void) static BOOL init_threadpool(void)
{ {
hntdll = GetModuleHandleA("ntdll"); HMODULE module = GetModuleHandleA("ntdll");
if (!hntdll) GET_PROC(TpAllocCleanupGroup);
{ GET_PROC(TpAllocIoCompletion);
win_skip("Could not load ntdll\n"); GET_PROC(TpAllocPool);
return FALSE; GET_PROC(TpAllocTimer);
} GET_PROC(TpAllocWait);
GET_PROC(TpAllocWork);
GET_PROC(TpCallbackMayRunLong);
GET_PROC(TpCallbackReleaseSemaphoreOnCompletion);
GET_PROC(TpCancelAsyncIoOperation);
GET_PROC(TpDisassociateCallback);
GET_PROC(TpIsTimerSet);
GET_PROC(TpPostWork);
GET_PROC(TpReleaseCleanupGroup);
GET_PROC(TpReleaseCleanupGroupMembers);
GET_PROC(TpReleaseIoCompletion);
GET_PROC(TpReleasePool);
GET_PROC(TpReleaseTimer);
GET_PROC(TpReleaseWait);
GET_PROC(TpReleaseWork);
GET_PROC(TpSetPoolMaxThreads);
GET_PROC(TpSetTimer);
GET_PROC(TpSetWait);
GET_PROC(TpSimpleTryPost);
GET_PROC(TpStartAsyncIoOperation);
GET_PROC(TpWaitForIoCompletion);
GET_PROC(TpWaitForTimer);
GET_PROC(TpWaitForWait);
GET_PROC(TpWaitForWork);
NTDLL_GET_PROC(TpAllocCleanupGroup); module = GetModuleHandleA("kernel32");
NTDLL_GET_PROC(TpAllocIoCompletion); GET_PROC(CancelThreadpoolIo);
NTDLL_GET_PROC(TpAllocPool); GET_PROC(CloseThreadpoolIo);
NTDLL_GET_PROC(TpAllocTimer); GET_PROC(CreateThreadpoolIo);
NTDLL_GET_PROC(TpAllocWait); GET_PROC(StartThreadpoolIo);
NTDLL_GET_PROC(TpAllocWork); GET_PROC(WaitForThreadpoolIoCallbacks);
NTDLL_GET_PROC(TpCancelAsyncIoOperation);
NTDLL_GET_PROC(TpCallbackMayRunLong);
NTDLL_GET_PROC(TpCallbackReleaseSemaphoreOnCompletion);
NTDLL_GET_PROC(TpDisassociateCallback);
NTDLL_GET_PROC(TpIsTimerSet);
NTDLL_GET_PROC(TpPostWork);
NTDLL_GET_PROC(TpReleaseCleanupGroup);
NTDLL_GET_PROC(TpReleaseCleanupGroupMembers);
NTDLL_GET_PROC(TpReleaseIoCompletion);
NTDLL_GET_PROC(TpReleasePool);
NTDLL_GET_PROC(TpReleaseTimer);
NTDLL_GET_PROC(TpReleaseWait);
NTDLL_GET_PROC(TpReleaseWork);
NTDLL_GET_PROC(TpSetPoolMaxThreads);
NTDLL_GET_PROC(TpSetTimer);
NTDLL_GET_PROC(TpSetWait);
NTDLL_GET_PROC(TpSimpleTryPost);
NTDLL_GET_PROC(TpStartAsyncIoOperation);
NTDLL_GET_PROC(TpWaitForIoCompletion);
NTDLL_GET_PROC(TpWaitForTimer);
NTDLL_GET_PROC(TpWaitForWait);
NTDLL_GET_PROC(TpWaitForWork);
if (!pTpAllocPool) if (!pTpAllocPool)
{ {
@ -2084,6 +2090,151 @@ static void test_tp_io(void)
pTpReleasePool(pool); pTpReleasePool(pool);
} }
static void CALLBACK kernel32_io_cb(TP_CALLBACK_INSTANCE *instance, void *userdata,
void *ovl, ULONG ret, ULONG_PTR length, TP_IO *io)
{
struct io_cb_ctx *ctx = userdata;
++ctx->count;
ctx->ovl = ovl;
ctx->ret = ret;
ctx->length = length;
ctx->io = io;
}
static void test_kernel32_tp_io(void)
{
TP_CALLBACK_ENVIRON environment = {.Version = 1};
OVERLAPPED ovl = {}, ovl2 = {};
HANDLE client, server, thread;
struct io_cb_ctx userdata;
char in[1], in2[1];
const char out[1];
NTSTATUS status;
DWORD ret_size;
TP_POOL *pool;
TP_IO *io;
BOOL ret;
ovl.hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
status = pTpAllocPool(&pool, NULL);
ok(!status, "failed to allocate pool, status %#x\n", status);
server = CreateNamedPipeA("\\\\.\\pipe\\wine_tp_test",
PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED, 0, 1, 1024, 1024, 0, NULL);
ok(server != INVALID_HANDLE_VALUE, "Failed to create server pipe, error %u.\n", GetLastError());
client = CreateFileA("\\\\.\\pipe\\wine_tp_test", GENERIC_READ | GENERIC_WRITE,
0, NULL, OPEN_EXISTING, 0, 0);
ok(client != INVALID_HANDLE_VALUE, "Failed to create client pipe, error %u.\n", GetLastError());
environment.Pool = pool;
io = NULL;
io = pCreateThreadpoolIo(server, kernel32_io_cb, &userdata, &environment);
todo_wine ok(!!io, "expected non-NULL TP_IO\n");
if (!io)
return;
pWaitForThreadpoolIoCallbacks(io, FALSE);
userdata.count = 0;
pStartThreadpoolIo(io);
thread = CreateThread(NULL, 0, io_wait_thread, io, 0, NULL);
ok(WaitForSingleObject(thread, 100) == WAIT_TIMEOUT, "TpWaitForIoCompletion() should not return\n");
ret = ReadFile(server, in, sizeof(in), NULL, &ovl);
ok(!ret, "wrong ret %d\n", ret);
ok(GetLastError() == ERROR_IO_PENDING, "wrong error %u\n", GetLastError());
ret = WriteFile(client, out, sizeof(out), &ret_size, NULL);
ok(ret, "WriteFile() failed, error %u\n", GetLastError());
pWaitForThreadpoolIoCallbacks(io, FALSE);
ok(userdata.count == 1, "callback ran %u times\n", userdata.count);
ok(userdata.ovl == &ovl, "expected %p, got %p\n", &ovl, userdata.ovl);
ok(userdata.ret == ERROR_SUCCESS, "got status %#x\n", userdata.ret);
ok(userdata.length == 1, "got length %lu\n", userdata.length);
ok(userdata.io == io, "expected %p, got %p\n", io, userdata.io);
ok(!WaitForSingleObject(thread, 1000), "wait timed out\n");
CloseHandle(thread);
userdata.count = 0;
pStartThreadpoolIo(io);
pStartThreadpoolIo(io);
ret = ReadFile(server, in, sizeof(in), NULL, &ovl);
ok(!ret, "wrong ret %d\n", ret);
ok(GetLastError() == ERROR_IO_PENDING, "wrong error %u\n", GetLastError());
ret = ReadFile(server, in2, sizeof(in2), NULL, &ovl2);
ok(!ret, "wrong ret %d\n", ret);
ok(GetLastError() == ERROR_IO_PENDING, "wrong error %u\n", GetLastError());
ret = WriteFile(client, out, sizeof(out), &ret_size, NULL);
ok(ret, "WriteFile() failed, error %u\n", GetLastError());
ret = WriteFile(client, out, sizeof(out), &ret_size, NULL);
ok(ret, "WriteFile() failed, error %u\n", GetLastError());
pWaitForThreadpoolIoCallbacks(io, FALSE);
ok(userdata.count == 2, "callback ran %u times\n", userdata.count);
ok(userdata.ret == STATUS_SUCCESS, "got status %#x\n", userdata.ret);
ok(userdata.length == 1, "got length %lu\n", userdata.length);
ok(userdata.io == io, "expected %p, got %p\n", io, userdata.io);
userdata.count = 0;
pStartThreadpoolIo(io);
pWaitForThreadpoolIoCallbacks(io, TRUE);
ok(!userdata.count, "callback ran %u times\n", userdata.count);
pStartThreadpoolIo(io);
ret = WriteFile(client, out, sizeof(out), &ret_size, NULL);
ok(ret, "WriteFile() failed, error %u\n", GetLastError());
ret = ReadFile(server, in, sizeof(in), NULL, &ovl);
ok(ret, "wrong ret %d\n", ret);
pWaitForThreadpoolIoCallbacks(io, FALSE);
ok(userdata.count == 1, "callback ran %u times\n", userdata.count);
ok(userdata.ovl == &ovl, "expected %p, got %p\n", &ovl, userdata.ovl);
ok(userdata.ret == ERROR_SUCCESS, "got status %#x\n", userdata.ret);
ok(userdata.length == 1, "got length %lu\n", userdata.length);
ok(userdata.io == io, "expected %p, got %p\n", io, userdata.io);
userdata.count = 0;
pStartThreadpoolIo(io);
ret = ReadFile(server, NULL, 1, NULL, &ovl);
ok(!ret, "wrong ret %d\n", ret);
ok(GetLastError() == ERROR_NOACCESS, "wrong error %u\n", GetLastError());
pCancelThreadpoolIo(io);
pWaitForThreadpoolIoCallbacks(io, FALSE);
ok(!userdata.count, "callback ran %u times\n", userdata.count);
userdata.count = 0;
pStartThreadpoolIo(io);
ret = ReadFile(server, in, sizeof(in), NULL, &ovl);
ok(!ret, "wrong ret %d\n", ret);
ok(GetLastError() == ERROR_IO_PENDING, "wrong error %u\n", GetLastError());
ret = CancelIo(server);
ok(ret, "CancelIo() failed, error %u\n", GetLastError());
pWaitForThreadpoolIoCallbacks(io, FALSE);
ok(userdata.count == 1, "callback ran %u times\n", userdata.count);
ok(userdata.ovl == &ovl, "expected %p, got %p\n", &ovl, userdata.ovl);
ok(userdata.ret == ERROR_OPERATION_ABORTED, "got status %#x\n", userdata.ret);
ok(!userdata.length, "got length %lu\n", userdata.length);
ok(userdata.io == io, "expected %p, got %p\n", io, userdata.io);
CloseHandle(ovl.hEvent);
CloseHandle(client);
CloseHandle(server);
pCloseThreadpoolIo(io);
pTpReleasePool(pool);
}
START_TEST(threadpool) START_TEST(threadpool)
{ {
test_RtlQueueWorkItem(); test_RtlQueueWorkItem();
@ -2104,4 +2255,5 @@ START_TEST(threadpool)
test_tp_wait(); test_tp_wait();
test_tp_multi_wait(); test_tp_multi_wait();
test_tp_io(); test_tp_io();
test_kernel32_tp_io();
} }