diff --git a/dlls/ntdll/tests/file.c b/dlls/ntdll/tests/file.c index f182719609f..eb3bc8c9919 100644 --- a/dlls/ntdll/tests/file.c +++ b/dlls/ntdll/tests/file.c @@ -1,6 +1,7 @@ /* Unit test suite for Ntdll file functions * * Copyright 2007 Jeff Latimer + * Copyright 2007 Andrey Turkin * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public @@ -33,6 +34,10 @@ #include "wine/test.h" #include "winternl.h" +#ifndef IO_COMPLETION_ALL_ACCESS +#define IO_COMPLETION_ALL_ACCESS 0x001F0003 +#endif + static VOID (WINAPI *pRtlInitUnicodeString)( PUNICODE_STRING, LPCWSTR ); static VOID (WINAPI *pRtlFreeUnicodeString)(PUNICODE_STRING); static NTSTATUS (WINAPI *pNtCreateMailslotFile)( PHANDLE, ULONG, POBJECT_ATTRIBUTES, PIO_STATUS_BLOCK, @@ -48,6 +53,13 @@ static NTSTATUS (WINAPI *pNtWriteFile)(HANDLE hFile, HANDLE hEvent, PLARGE_INTEGER offset, PULONG key); static NTSTATUS (WINAPI *pNtClose)( PHANDLE ); +static NTSTATUS (WINAPI *pNtCreateIoCompletion)(PHANDLE, ACCESS_MASK, POBJECT_ATTRIBUTES, ULONG); +static NTSTATUS (WINAPI *pNtOpenIoCompletion)(PHANDLE, ACCESS_MASK, POBJECT_ATTRIBUTES); +static NTSTATUS (WINAPI *pNtQueryIoCompletion)(HANDLE, IO_COMPLETION_INFORMATION_CLASS, PVOID, ULONG, PULONG); +static NTSTATUS (WINAPI *pNtRemoveIoCompletion)(HANDLE, PULONG_PTR, PULONG_PTR, PIO_STATUS_BLOCK, PLARGE_INTEGER); +static NTSTATUS (WINAPI *pNtSetIoCompletion)(HANDLE, ULONG_PTR, ULONG_PTR, NTSTATUS, ULONG); +static NTSTATUS (WINAPI *pNtSetInformationFile)(HANDLE, PIO_STATUS_BLOCK, PVOID, ULONG, FILE_INFORMATION_CLASS); + static inline BOOL is_signaled( HANDLE obj ) { return WaitForSingleObject( obj, 0 ) == 0; @@ -79,6 +91,41 @@ static HANDLE create_temp_file( ULONG flags ) return (handle == INVALID_HANDLE_VALUE) ? 0 : handle; } +#define CVALUE_FIRST 0xfffabbcc +#define CKEY_FIRST 0x1030341 +#define CKEY_SECOND 0x132E46 + +ULONG_PTR completionKey; +IO_STATUS_BLOCK ioSb; +ULONG_PTR completionValue; + +static long get_pending_msgs(HANDLE h) +{ + NTSTATUS res; + ULONG a, req; + + res = pNtQueryIoCompletion( h, IoCompletionBasicInformation, (PVOID)&a, sizeof(a), &req ); + ok( res == STATUS_SUCCESS, "NtQueryIoCompletion failed: %x\n", res ); + if (res != STATUS_SUCCESS) return -1; + ok( req == sizeof(a), "Unexpected response size: %x\n", req ); + return a; +} + +static BOOL get_msg(HANDLE h) +{ + LARGE_INTEGER timeout = {{-10000000*3}}; + DWORD res = pNtRemoveIoCompletion( h, &completionKey, &completionValue, &ioSb, &timeout); + ok( res == STATUS_SUCCESS, "NtRemoveIoCompletion failed: %x\n", res ); + if (res != STATUS_SUCCESS) + { + completionKey = completionValue = 0; + memset(&ioSb, 0, sizeof(ioSb)); + return FALSE; + } + return TRUE; +} + + static void WINAPI apc( void *arg, IO_STATUS_BLOCK *iosb, ULONG reserved ) { int *count = arg; @@ -424,6 +471,109 @@ static void nt_mailslot_test(void) pRtlFreeUnicodeString(&str); } +static void test_iocp_setcompletion(HANDLE h) +{ + NTSTATUS res; + long count; + + res = pNtSetIoCompletion( h, CKEY_FIRST, CVALUE_FIRST, STATUS_INVALID_DEVICE_REQUEST, 3 ); + ok( res == STATUS_SUCCESS, "NtSetIoCompletion failed: %x\n", res ); + + count = get_pending_msgs(h); + ok( count == 1, "Unexpected msg count: %ld\n", count ); + + if (get_msg(h)) + { + ok( completionKey == CKEY_FIRST, "Invalid completion key: %lx\n", completionKey ); + ok( ioSb.Information == 3, "Invalid ioSb.Information: %ld\n", ioSb.Information ); + ok( U(ioSb).Status == STATUS_INVALID_DEVICE_REQUEST, "Invalid ioSb.Status: %x\n", U(ioSb).Status); + ok( completionValue == CVALUE_FIRST, "Invalid completion value: %lx\n", completionValue ); + } + + count = get_pending_msgs(h); + ok( !count, "Unexpected msg count: %ld\n", count ); +} + +static void test_iocp_fileio(HANDLE h) +{ + static const char pipe_name[] = "\\\\.\\pipe\\iocompletiontestnamedpipe"; + + HANDLE hPipeSrv = CreateNamedPipeA( pipe_name, PIPE_ACCESS_INBOUND | FILE_FLAG_OVERLAPPED, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT, 4, 1024, 1024, 1000, NULL ); + HANDLE hPipeClt = CreateFileA( pipe_name, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_FLAG_NO_BUFFERING | FILE_FLAG_OVERLAPPED, NULL ); + ok( hPipeSrv != INVALID_HANDLE_VALUE && hPipeClt != INVALID_HANDLE_VALUE, "Cannot create or connect to pipe\n" ); + if (hPipeSrv != INVALID_HANDLE_VALUE && hPipeClt != INVALID_HANDLE_VALUE) + { + OVERLAPPED o = {0,}; + BYTE buf[3]; + DWORD read; + long count; + FILE_COMPLETION_INFORMATION fci = {h, CKEY_SECOND}; + IO_STATUS_BLOCK iosb; + + NTSTATUS res = pNtSetInformationFile( hPipeSrv, &iosb, &fci, sizeof(fci), FileCompletionInformation ); + ok( res == STATUS_SUCCESS, "NtSetInformationFile failed: %x\n", res ); + ok( iosb.Status == STATUS_SUCCESS, "iosb.Status invalid: %x\n", iosb.Status ); + + count = get_pending_msgs(h); + ok( !count, "Unexpected msg count: %ld\n", count ); + ReadFile( hPipeSrv, buf, 3, &read, &o); + count = get_pending_msgs(h); + ok( !count, "Unexpected msg count: %ld\n", count ); + WriteFile( hPipeClt, buf, 3, &read, NULL ); + + todo_wine { + if (get_msg(h)) + { + ok( completionKey == CKEY_SECOND, "Invalid completion key: %lx\n", completionKey ); + ok( ioSb.Information == 3, "Invalid ioSb.Information: %ld\n", ioSb.Information ); + ok( U(ioSb).Status == STATUS_SUCCESS, "Invalid ioSb.Status: %x\n", U(ioSb).Status); + ok( completionValue == (ULONG_PTR)&o, "Invalid completion value: %lx\n", completionValue ); + } + } + count = get_pending_msgs(h); + ok( !count, "Unexpected msg count: %ld\n", count ); + + WriteFile( hPipeClt, buf, 2, &read, NULL ); + count = get_pending_msgs(h); + ok( !count, "Unexpected msg count: %ld\n", count ); + ReadFile( hPipeSrv, buf, 2, &read, &o); + count = get_pending_msgs(h); + todo_wine { + ok( count == 1, "Unexpected msg count: %ld\n", count ); + } + todo_wine { + if (get_msg(h)) + { + ok( completionKey == CKEY_SECOND, "Invalid completion key: %lx\n", completionKey ); + ok( ioSb.Information == 2, "Invalid ioSb.Information: %ld\n", ioSb.Information ); + ok( U(ioSb).Status == STATUS_SUCCESS, "Invalid ioSb.Status: %x\n", U(ioSb).Status); + ok( completionValue == (ULONG_PTR)&o, "Invalid completion value: %lx\n", completionValue ); + } + } + } + + CloseHandle( hPipeSrv ); + CloseHandle( hPipeClt ); +} + +static void test_iocompletion(void) +{ + HANDLE h = INVALID_HANDLE_VALUE; + NTSTATUS res; + + res = pNtCreateIoCompletion( &h, IO_COMPLETION_ALL_ACCESS, NULL, 0); + + ok( res == 0, "NtCreateIoCompletion anonymous failed: %x\n", res ); + ok( h && h != INVALID_HANDLE_VALUE, "Invalid handle returned\n" ); + + if ( h && h != INVALID_HANDLE_VALUE) + { + test_iocp_setcompletion(h); + test_iocp_fileio(h); + pNtClose(h); + } +} + START_TEST(file) { HMODULE hntdll = GetModuleHandleA("ntdll.dll"); @@ -439,7 +589,14 @@ START_TEST(file) pNtReadFile = (void *)GetProcAddress(hntdll, "NtReadFile"); pNtWriteFile = (void *)GetProcAddress(hntdll, "NtWriteFile"); pNtClose = (void *)GetProcAddress(hntdll, "NtClose"); + pNtCreateIoCompletion = (void *)GetProcAddress(hntdll, "NtCreateIoCompletion"); + pNtOpenIoCompletion = (void *)GetProcAddress(hntdll, "NtOpenIoCompletion"); + pNtQueryIoCompletion = (void *)GetProcAddress(hntdll, "NtQueryIoCompletion"); + pNtRemoveIoCompletion = (void *)GetProcAddress(hntdll, "NtRemoveIoCompletion"); + pNtSetIoCompletion = (void *)GetProcAddress(hntdll, "NtSetIoCompletion"); + pNtSetInformationFile = (void *)GetProcAddress(hntdll, "NtSetInformationFile"); read_file_test(); nt_mailslot_test(); + test_iocompletion(); }