Andrey Turkin : ntdll: Add some I/O completion tests.
Alexandre Julliard
julliard at winehq.org
Thu Nov 15 07:45:05 CST 2007
Module: wine
Branch: master
Commit: 074281ff6c50bfb1be3d7229bc7042683d56137e
URL: http://source.winehq.org/git/wine.git/?a=commit;h=074281ff6c50bfb1be3d7229bc7042683d56137e
Author: Andrey Turkin <andrey.turkin at gmail.com>
Date: Sat Nov 10 01:11:31 2007 +0300
ntdll: Add some I/O completion tests.
---
dlls/ntdll/tests/file.c | 157 +++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 157 insertions(+), 0 deletions(-)
diff --git a/dlls/ntdll/tests/file.c b/dlls/ntdll/tests/file.c
index f182719..eb3bc8c 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();
}
More information about the wine-cvs
mailing list