mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 14:18:41 +00:00
431 lines
14 KiB
C
431 lines
14 KiB
C
![]() |
/* Unit test suite for Ntdll NamedPipe API functions
|
||
|
*
|
||
|
* Copyright 2011 Bernhard Loos
|
||
|
*
|
||
|
* This library is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* This library is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Lesser General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
* License along with this library; if not, write to the Free Software
|
||
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
||
|
*/
|
||
|
|
||
|
#include <stdio.h>
|
||
|
#include <stdarg.h>
|
||
|
|
||
|
#include "ntstatus.h"
|
||
|
#define WIN32_NO_STATUS
|
||
|
#include "windef.h"
|
||
|
#include "winbase.h"
|
||
|
#include "winuser.h"
|
||
|
#include "winreg.h"
|
||
|
#include "winnls.h"
|
||
|
#include "wine/test.h"
|
||
|
#include "winternl.h"
|
||
|
#include "winioctl.h"
|
||
|
|
||
|
#ifndef __WINE_WINTERNL_H
|
||
|
|
||
|
typedef struct {
|
||
|
ULONG NamedPipeType;
|
||
|
ULONG NamedPipeConfiguration;
|
||
|
ULONG MaximumInstances;
|
||
|
ULONG CurrentInstances;
|
||
|
ULONG InboundQuota;
|
||
|
ULONG ReadDataAvailable;
|
||
|
ULONG OutboundQuota;
|
||
|
ULONG WriteQuotaAvailable;
|
||
|
ULONG NamedPipeState;
|
||
|
ULONG NamedPipeEnd;
|
||
|
} FILE_PIPE_LOCAL_INFORMATION;
|
||
|
|
||
|
#ifndef FILE_SYNCHRONOUS_IO_ALERT
|
||
|
#define FILE_SYNCHRONOUS_IO_ALERT 0x10
|
||
|
#endif
|
||
|
|
||
|
#ifndef FILE_SYNCHRONOUS_IO_NONALERT
|
||
|
#define FILE_SYNCHRONOUS_IO_NONALERT 0x20
|
||
|
#endif
|
||
|
|
||
|
#ifndef FSCTL_PIPE_LISTEN
|
||
|
#define FSCTL_PIPE_LISTEN CTL_CODE(FILE_DEVICE_NAMED_PIPE, 2, METHOD_BUFFERED, FILE_ANY_ACCESS)
|
||
|
#endif
|
||
|
#endif
|
||
|
|
||
|
static NTSTATUS (WINAPI *pNtFsControlFile) (HANDLE handle, HANDLE event, PIO_APC_ROUTINE apc, PVOID apc_context, PIO_STATUS_BLOCK io, ULONG code, PVOID in_buffer, ULONG in_size, PVOID out_buffer, ULONG out_size);
|
||
|
static NTSTATUS (WINAPI *pNtCreateNamedPipeFile) (PHANDLE handle, ULONG access,
|
||
|
POBJECT_ATTRIBUTES attr, PIO_STATUS_BLOCK iosb,
|
||
|
ULONG sharing, ULONG dispo, ULONG options,
|
||
|
ULONG pipe_type, ULONG read_mode,
|
||
|
ULONG completion_mode, ULONG max_inst,
|
||
|
ULONG inbound_quota, ULONG outbound_quota,
|
||
|
PLARGE_INTEGER timeout);
|
||
|
static NTSTATUS (WINAPI *pNtQueryInformationFile) (IN HANDLE FileHandle, OUT PIO_STATUS_BLOCK IoStatusBlock, OUT PVOID FileInformation, IN ULONG Length, IN FILE_INFORMATION_CLASS FileInformationClass);
|
||
|
static NTSTATUS (WINAPI *pNtCancelIoFile) (HANDLE hFile, PIO_STATUS_BLOCK io_status);
|
||
|
static void (WINAPI *pRtlInitUnicodeString) (PUNICODE_STRING target, PCWSTR source);
|
||
|
|
||
|
static HANDLE (WINAPI *pOpenThread)(DWORD dwDesiredAccess, BOOL bInheritHandle, DWORD dwThreadId);
|
||
|
static DWORD (WINAPI *pQueueUserAPC)(PAPCFUNC pfnAPC, HANDLE hThread, ULONG_PTR dwData);
|
||
|
|
||
|
|
||
|
static BOOL init_func_ptrs(void)
|
||
|
{
|
||
|
HMODULE module = GetModuleHandle("ntdll.dll");
|
||
|
|
||
|
#define loadfunc(name) if (!(p##name = (void *)GetProcAddress(module, #name))) { \
|
||
|
trace("GetProcAddress(%s) failed\n", #name); \
|
||
|
return FALSE; \
|
||
|
}
|
||
|
|
||
|
loadfunc(NtFsControlFile)
|
||
|
loadfunc(NtCreateNamedPipeFile)
|
||
|
loadfunc(NtQueryInformationFile)
|
||
|
loadfunc(NtCancelIoFile)
|
||
|
loadfunc(RtlInitUnicodeString)
|
||
|
|
||
|
/* not fatal */
|
||
|
module = GetModuleHandle("kernel32.dll");
|
||
|
pOpenThread = (void *)GetProcAddress(module, "OpenThread");
|
||
|
pQueueUserAPC = (void *)GetProcAddress(module, "QueueUserAPC");
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
static const WCHAR testpipe[] = { '\\', '\\', '.', '\\', 'p', 'i', 'p', 'e', '\\',
|
||
|
't', 'e', 's', 't', 'p', 'i', 'p', 'e', 0 };
|
||
|
static const WCHAR testpipe_nt[] = { '\\', '?', '?', '\\', 'p', 'i', 'p', 'e', '\\',
|
||
|
't', 'e', 's', 't', 'p', 'i', 'p', 'e', 0 };
|
||
|
|
||
|
static NTSTATUS create_pipe(PHANDLE handle, ULONG sharing, ULONG options)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
OBJECT_ATTRIBUTES attr;
|
||
|
UNICODE_STRING name;
|
||
|
LARGE_INTEGER timeout;
|
||
|
NTSTATUS res;
|
||
|
|
||
|
pRtlInitUnicodeString(&name, testpipe_nt);
|
||
|
|
||
|
attr.Length = sizeof(attr);
|
||
|
attr.RootDirectory = 0;
|
||
|
attr.ObjectName = &name;
|
||
|
attr.Attributes = 0x40; /*case insensitive */
|
||
|
attr.SecurityDescriptor = NULL;
|
||
|
attr.SecurityQualityOfService = NULL;
|
||
|
|
||
|
timeout.QuadPart = -100000000000ll;
|
||
|
|
||
|
res = pNtCreateNamedPipeFile(handle, FILE_READ_ATTRIBUTES | SYNCHRONIZE, &attr, &iosb, sharing, 2 /*FILE_CREATE*/,
|
||
|
options, 1, 0, 0, 0xFFFFFFFF, 500, 500, &timeout);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
static void test_create_invalid(void)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
OBJECT_ATTRIBUTES attr;
|
||
|
UNICODE_STRING name;
|
||
|
LARGE_INTEGER timeout;
|
||
|
NTSTATUS res;
|
||
|
HANDLE handle, handle2;
|
||
|
FILE_PIPE_LOCAL_INFORMATION info;
|
||
|
|
||
|
pRtlInitUnicodeString(&name, testpipe_nt);
|
||
|
|
||
|
attr.Length = sizeof(attr);
|
||
|
attr.RootDirectory = 0;
|
||
|
attr.ObjectName = &name;
|
||
|
attr.Attributes = 0x40; /*case insensitive */
|
||
|
attr.SecurityDescriptor = NULL;
|
||
|
attr.SecurityQualityOfService = NULL;
|
||
|
|
||
|
timeout.QuadPart = -100000000000ll;
|
||
|
|
||
|
/* create a pipe with sharing = 0 */
|
||
|
res = pNtCreateNamedPipeFile(&handle, FILE_READ_ATTRIBUTES | SYNCHRONIZE, &attr, &iosb, 0, 2 /*FILE_CREATE*/,
|
||
|
0, 1, 0, 0, 0xFFFFFFFF, 500, 500, &timeout);
|
||
|
todo_wine ok(res == STATUS_INVALID_PARAMETER, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
if (!res)
|
||
|
CloseHandle(handle);
|
||
|
|
||
|
/* create a pipe without r/w access */
|
||
|
res = pNtCreateNamedPipeFile(&handle, SYNCHRONIZE, &attr, &iosb, FILE_SHARE_READ | FILE_SHARE_WRITE, 2 /*FILE_CREATE*/,
|
||
|
0, 1, 0, 0, 0xFFFFFFFF, 500, 500, &timeout);
|
||
|
ok(!res, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
|
||
|
res = pNtQueryInformationFile(handle, &iosb, &info, sizeof(info), (FILE_INFORMATION_CLASS)24);
|
||
|
todo_wine ok(res == STATUS_ACCESS_DENIED, "NtQueryInformationFile returned %x\n", res);
|
||
|
|
||
|
/* test FILE_CREATE creation disposition */
|
||
|
res = pNtCreateNamedPipeFile(&handle2, SYNCHRONIZE, &attr, &iosb, FILE_SHARE_READ | FILE_SHARE_WRITE, 2 /*FILE_CREATE*/,
|
||
|
0, 1, 0, 0, 0xFFFFFFFF, 500, 500, &timeout);
|
||
|
todo_wine ok(res == STATUS_ACCESS_DENIED, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
if (!res)
|
||
|
CloseHandle(handle2);
|
||
|
|
||
|
CloseHandle(handle);
|
||
|
}
|
||
|
|
||
|
static BOOL ioapc_called;
|
||
|
static void CALLBACK ioapc(void *arg, PIO_STATUS_BLOCK io, ULONG reserved)
|
||
|
{
|
||
|
ioapc_called = TRUE;
|
||
|
}
|
||
|
|
||
|
static NTSTATUS listen_pipe(HANDLE hPipe, HANDLE hEvent, PIO_STATUS_BLOCK iosb, BOOL use_apc)
|
||
|
{
|
||
|
int dummy;
|
||
|
|
||
|
ioapc_called = FALSE;
|
||
|
|
||
|
return pNtFsControlFile(hPipe, hEvent, use_apc ? &ioapc: NULL, use_apc ? &dummy: NULL, iosb, FSCTL_PIPE_LISTEN, 0, 0, 0, 0);
|
||
|
}
|
||
|
|
||
|
static void test_overlapped(void)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
HANDLE hEvent;
|
||
|
HANDLE hPipe;
|
||
|
HANDLE hClient;
|
||
|
NTSTATUS res;
|
||
|
|
||
|
hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
|
||
|
ok(hEvent != INVALID_HANDLE_VALUE, "can't create event, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = create_pipe(&hPipe, FILE_SHARE_READ | FILE_SHARE_WRITE, 0 /* OVERLAPPED */);
|
||
|
ok(!res, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
|
||
|
memset(&iosb, 0x55, sizeof(iosb));
|
||
|
|
||
|
/* try with event and apc */
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
ok(res == STATUS_PENDING, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
hClient = CreateFileW(testpipe, GENERIC_READ | GENERIC_WRITE, 0, 0, OPEN_EXISTING, 0, 0);
|
||
|
ok(hClient != INVALID_HANDLE_VALUE, "can't open pipe, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
ok(U(iosb).Status == 0, "Wrong iostatus %x\n", U(iosb).Status);
|
||
|
ok(WaitForSingleObject(hEvent, 0) == 0, "hEvent not signaled\n");
|
||
|
|
||
|
ok(!ioapc_called, "IOAPC ran too early\n");
|
||
|
|
||
|
SleepEx(0, TRUE); /* alertable wait state */
|
||
|
|
||
|
ok(ioapc_called, "IOAPC didn't run\n");
|
||
|
|
||
|
CloseHandle(hEvent);
|
||
|
CloseHandle(hPipe);
|
||
|
CloseHandle(hClient);
|
||
|
}
|
||
|
|
||
|
static BOOL userapc_called;
|
||
|
static void CALLBACK userapc(ULONG_PTR dwParam)
|
||
|
{
|
||
|
userapc_called = TRUE;
|
||
|
}
|
||
|
|
||
|
static BOOL open_succeeded;
|
||
|
static DWORD WINAPI thread(PVOID main_thread)
|
||
|
{
|
||
|
HANDLE h;
|
||
|
|
||
|
Sleep(400);
|
||
|
|
||
|
if (main_thread) {
|
||
|
userapc_called = FALSE;
|
||
|
ok(pQueueUserAPC(&userapc, main_thread, 0), "can't queue user apc, GetLastError: %x\n", GetLastError());
|
||
|
CloseHandle(main_thread);
|
||
|
}
|
||
|
|
||
|
Sleep(400);
|
||
|
|
||
|
h = CreateFileW(testpipe, GENERIC_WRITE | GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);
|
||
|
|
||
|
if (h != INVALID_HANDLE_VALUE) {
|
||
|
open_succeeded = TRUE;
|
||
|
Sleep(100);
|
||
|
CloseHandle(h);
|
||
|
} else
|
||
|
open_succeeded = FALSE;
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static void test_alertable(void)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
HANDLE hEvent;
|
||
|
HANDLE hPipe;
|
||
|
NTSTATUS res;
|
||
|
HANDLE hThread;
|
||
|
|
||
|
memset(&iosb, 0x55, sizeof(iosb));
|
||
|
|
||
|
hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
|
||
|
ok(hEvent != INVALID_HANDLE_VALUE, "can't create event, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = create_pipe(&hPipe, FILE_SHARE_READ | FILE_SHARE_WRITE, FILE_SYNCHRONOUS_IO_ALERT);
|
||
|
ok(!res, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
|
||
|
/* queue an user apc before calling listen */
|
||
|
userapc_called = FALSE;
|
||
|
ok(pQueueUserAPC(&userapc, GetCurrentThread(), 0), "can't queue user apc, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
todo_wine ok(res == STATUS_CANCELLED, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
todo_wine ok(userapc_called, "user apc didn't run\n");
|
||
|
ok(U(iosb).Status == 0x55555555, "iosb.Status got changed to %x\n", U(iosb).Status);
|
||
|
todo_wine ok(WaitForSingleObjectEx(hEvent, 0, TRUE) == WAIT_TIMEOUT, "hEvent signaled\n");
|
||
|
ok(!ioapc_called, "IOAPC ran\n");
|
||
|
|
||
|
/* queue an user apc from a different thread */
|
||
|
hThread = CreateThread(NULL, 0, &thread, pOpenThread(MAXIMUM_ALLOWED, FALSE, GetCurrentThreadId()), 0, 0);
|
||
|
ok(hThread != INVALID_HANDLE_VALUE, "can't create thread, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
/* wine_todo: the earlier NtFsControlFile call gets cancelled after the pipe gets set into listen state
|
||
|
instead of before, so this NtFsControlFile will fail STATUS_INVALID_HANDLE */
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
todo_wine ok(res == STATUS_CANCELLED, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
ok(userapc_called, "user apc didn't run\n");
|
||
|
todo_wine ok(U(iosb).Status == 0x55555555, "iosb.Status got changed to %x\n", U(iosb).Status);
|
||
|
ok(WaitForSingleObjectEx(hEvent, 0, TRUE) == WAIT_TIMEOUT, "hEvent signaled\n");
|
||
|
ok(!ioapc_called, "IOAPC ran\n");
|
||
|
|
||
|
WaitForSingleObject(hThread, INFINITE);
|
||
|
|
||
|
SleepEx(0, TRUE); /* get rid of the userapc, if NtFsControlFile failed */
|
||
|
|
||
|
ok(open_succeeded, "couldn't open client side pipe\n");
|
||
|
|
||
|
CloseHandle(hThread);
|
||
|
DisconnectNamedPipe(hPipe);
|
||
|
|
||
|
/* finally try without an apc */
|
||
|
hThread = CreateThread(NULL, 0, &thread, 0, 0, 0);
|
||
|
ok(hThread != INVALID_HANDLE_VALUE, "can't create thread, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
todo_wine ok(!res, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
ok(open_succeeded, "couldn't open client side pipe\n");
|
||
|
ok(!U(iosb).Status, "Wrong iostatus %x\n", U(iosb).Status);
|
||
|
todo_wine ok(WaitForSingleObject(hEvent, 0) == 0, "hEvent not signaled\n");
|
||
|
|
||
|
WaitForSingleObject(hThread, INFINITE);
|
||
|
CloseHandle(hThread);
|
||
|
CloseHandle(hEvent);
|
||
|
CloseHandle(hPipe);
|
||
|
}
|
||
|
|
||
|
static void test_nonalertable(void)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
HANDLE hEvent;
|
||
|
HANDLE hPipe;
|
||
|
NTSTATUS res;
|
||
|
HANDLE hThread;
|
||
|
|
||
|
memset(&iosb, 0x55, sizeof(iosb));
|
||
|
|
||
|
hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
|
||
|
ok(hEvent != INVALID_HANDLE_VALUE, "can't create event, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = create_pipe(&hPipe, FILE_SHARE_READ | FILE_SHARE_WRITE, FILE_SYNCHRONOUS_IO_NONALERT);
|
||
|
ok(!res, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
|
||
|
hThread = CreateThread(NULL, 0, &thread, 0, 0, 0);
|
||
|
ok(hThread != INVALID_HANDLE_VALUE, "can't create thread, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
userapc_called = FALSE;
|
||
|
ok(pQueueUserAPC(&userapc, GetCurrentThread(), 0), "can't queue user apc, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
todo_wine ok(!res, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
ok(open_succeeded, "couldn't open client side pipe\n");
|
||
|
todo_wine ok(!U(iosb).Status, "Wrong iostatus %x\n", U(iosb).Status);
|
||
|
todo_wine ok(WaitForSingleObject(hEvent, 0) == 0, "hEvent not signaled\n");
|
||
|
|
||
|
ok(!ioapc_called, "IOAPC ran too early\n");
|
||
|
ok(!userapc_called, "user apc ran too early\n");
|
||
|
|
||
|
SleepEx(0, TRUE); /* alertable wait state */
|
||
|
|
||
|
ok(ioapc_called, "IOAPC didn't run\n");
|
||
|
ok(userapc_called, "user apc didn't run\n");
|
||
|
|
||
|
WaitForSingleObject(hThread, INFINITE);
|
||
|
CloseHandle(hThread);
|
||
|
CloseHandle(hEvent);
|
||
|
CloseHandle(hPipe);
|
||
|
}
|
||
|
|
||
|
static void test_cancelio(void)
|
||
|
{
|
||
|
IO_STATUS_BLOCK iosb;
|
||
|
IO_STATUS_BLOCK cancel_sb;
|
||
|
HANDLE hEvent;
|
||
|
HANDLE hPipe;
|
||
|
NTSTATUS res;
|
||
|
|
||
|
hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
|
||
|
ok(hEvent != INVALID_HANDLE_VALUE, "can't create event, GetLastError: %x\n", GetLastError());
|
||
|
|
||
|
res = create_pipe(&hPipe, FILE_SHARE_READ | FILE_SHARE_WRITE, 0 /* OVERLAPPED */);
|
||
|
ok(!res, "NtCreateNamedPipeFile returned %x\n", res);
|
||
|
|
||
|
memset(&iosb, 0x55, sizeof(iosb));
|
||
|
|
||
|
res = listen_pipe(hPipe, hEvent, &iosb, TRUE);
|
||
|
ok(res == STATUS_PENDING, "NtFsControlFile returned %x\n", res);
|
||
|
|
||
|
res = pNtCancelIoFile(hPipe, &cancel_sb);
|
||
|
todo_wine ok(!res, "NtCancelIoFile returned %x\n", res);
|
||
|
|
||
|
todo_wine {
|
||
|
ok(U(iosb).Status == STATUS_CANCELLED, "Wrong iostatus %x\n", U(iosb).Status);
|
||
|
ok(WaitForSingleObject(hEvent, 0) == 0, "hEvent not signaled\n");
|
||
|
}
|
||
|
|
||
|
ok(!ioapc_called, "IOAPC ran too early\n");
|
||
|
|
||
|
SleepEx(0, TRUE); /* alertable wait state */
|
||
|
|
||
|
ok(ioapc_called, "IOAPC didn't run\n");
|
||
|
|
||
|
CloseHandle(hEvent);
|
||
|
CloseHandle(hPipe);
|
||
|
}
|
||
|
|
||
|
START_TEST(pipe)
|
||
|
{
|
||
|
if (!init_func_ptrs())
|
||
|
return;
|
||
|
|
||
|
trace("starting invalid create tests\n");
|
||
|
test_create_invalid();
|
||
|
|
||
|
trace("starting overlapped tests\n");
|
||
|
test_overlapped();
|
||
|
|
||
|
if (!pOpenThread || !pQueueUserAPC)
|
||
|
return;
|
||
|
|
||
|
trace("starting alertable tests\n");
|
||
|
test_alertable();
|
||
|
|
||
|
trace("starting nonalertable tests\n");
|
||
|
test_nonalertable();
|
||
|
|
||
|
trace("starting cancelio tests\n");
|
||
|
test_cancelio();
|
||
|
}
|