Remove unneeded checks on the GetModuleHandle() return value for cases where we are...
[wine] / dlls / ntdll / tests / file.c
1 /* Unit test suite for Ntdll file functions
2  *
3  * Copyright 2007 Jeff Latimer
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  *
19  * NOTES
20  * We use function pointers here as there is no import library for NTDLL on
21  * windows.
22  */
23
24 #include <stdio.h>
25 #include <stdarg.h>
26
27 #include "ntstatus.h"
28 /* Define WIN32_NO_STATUS so MSVC does not give us duplicate macro
29  * definition errors when we get to winnt.h
30  */
31 #define WIN32_NO_STATUS
32
33 #include "wine/test.h"
34 #include "winternl.h"
35
36 static VOID     (WINAPI *pRtlInitUnicodeString)( PUNICODE_STRING, LPCWSTR );
37 static VOID     (WINAPI *pRtlFreeUnicodeString)(PUNICODE_STRING);
38 static NTSTATUS (WINAPI *pNtCreateMailslotFile)( PHANDLE, ULONG, POBJECT_ATTRIBUTES, PIO_STATUS_BLOCK,
39                                        ULONG, ULONG, ULONG, PLARGE_INTEGER );
40 static NTSTATUS (WINAPI *pNtClose)( PHANDLE );
41
42 static void nt_mailslot_test(void)
43 {
44     HANDLE hslot;
45     ACCESS_MASK DesiredAccess;
46     OBJECT_ATTRIBUTES attr;
47
48     ULONG CreateOptions;
49     ULONG MailslotQuota;
50     ULONG MaxMessageSize;
51     LARGE_INTEGER TimeOut;
52     IO_STATUS_BLOCK IoStatusBlock;
53     NTSTATUS rc;
54     UNICODE_STRING str;
55     WCHAR buffer1[] = { '\\','?','?','\\','M','A','I','L','S','L','O','T','\\',
56                         'R',':','\\','F','R','E','D','\0' };
57
58     TimeOut.QuadPart = -1;
59
60     pRtlInitUnicodeString(&str, buffer1);
61     InitializeObjectAttributes(&attr, &str, OBJ_CASE_INSENSITIVE, 0, NULL);
62     DesiredAccess = CreateOptions = MailslotQuota = MaxMessageSize = 0;
63
64     /*
65      * Check for NULL pointer handling
66      */
67     rc = pNtCreateMailslotFile(NULL, DesiredAccess,
68          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
69          &TimeOut);
70     ok( rc == STATUS_ACCESS_VIOLATION, "rc = %x not c0000005 STATUS_ACCESS_VIOLATION\n", rc);
71
72     /*
73      * Test to see if the Timeout can be NULL
74      */
75     rc = pNtCreateMailslotFile(&hslot, DesiredAccess,
76          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
77          NULL);
78     ok( rc == STATUS_SUCCESS, "rc = %x not STATUS_SUCCESS\n", rc);
79     ok( hslot != 0, "Handle is invalid\n");
80
81     if  ( rc == STATUS_SUCCESS ) rc = pNtClose(hslot);
82
83     /*
84      * Test that the length field is checked properly
85      */
86     attr.Length = 0;
87     rc = pNtCreateMailslotFile(&hslot, DesiredAccess,
88          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
89          &TimeOut);
90     todo_wine ok( rc == STATUS_INVALID_PARAMETER, "rc = %x not c000000d STATUS_INVALID_PARAMETER\n", rc);
91
92     if  (rc == STATUS_SUCCESS) pNtClose(hslot);
93
94     attr.Length = sizeof(OBJECT_ATTRIBUTES)+1;
95     rc = pNtCreateMailslotFile(&hslot, DesiredAccess,
96          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
97          &TimeOut);
98     todo_wine ok( rc == STATUS_INVALID_PARAMETER, "rc = %x not c000000d STATUS_INVALID_PARAMETER\n", rc);
99
100     if  (rc == STATUS_SUCCESS) pNtClose(hslot);
101
102     /*
103      * Test handling of a NULL unicode string in ObjectName
104      */
105     InitializeObjectAttributes(&attr, &str, OBJ_CASE_INSENSITIVE, 0, NULL);
106     attr.ObjectName = NULL;
107     rc = pNtCreateMailslotFile(&hslot, DesiredAccess,
108          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
109          &TimeOut);
110     ok( rc == STATUS_OBJECT_PATH_SYNTAX_BAD, "rc = %x not c000003b STATUS_OBJECT_PATH_SYNTAX_BAD\n", rc);
111
112     if  (rc == STATUS_SUCCESS) pNtClose(hslot);
113
114     /*
115      * Test a valid call
116      */
117     InitializeObjectAttributes(&attr, &str, OBJ_CASE_INSENSITIVE, 0, NULL);
118     rc = pNtCreateMailslotFile(&hslot, DesiredAccess,
119          &attr, &IoStatusBlock, CreateOptions, MailslotQuota, MaxMessageSize,
120          &TimeOut);
121     ok( rc == STATUS_SUCCESS, "Create MailslotFile failed rc = %x %u\n", rc, GetLastError());
122     ok( hslot != 0, "Handle is invalid\n");
123
124     rc = pNtClose(hslot);
125     ok( rc == STATUS_SUCCESS, "NtClose failed\n");
126
127     pRtlFreeUnicodeString(&str);
128 }
129
130 START_TEST(file)
131 {
132     HMODULE hntdll = GetModuleHandleA("ntdll.dll");
133     if (!hntdll)
134     {
135         skip("not running on NT, skipping test\n");
136         return;
137     }
138
139     pRtlFreeUnicodeString   = (void *)GetProcAddress(hntdll, "RtlFreeUnicodeString");
140     pRtlInitUnicodeString   = (void *)GetProcAddress(hntdll, "RtlInitUnicodeString");
141     pNtCreateMailslotFile   = (void *)GetProcAddress(hntdll, "NtCreateMailslotFile");
142     pNtClose                = (void *)GetProcAddress(hntdll, "NtClose");
143
144     nt_mailslot_test();
145 }