1 /*
2 * Unit tests for module/DLL/library API
3 *
4 * Copyright (c) 2004 Eric Pouech
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include "wine/test.h"
22 #include <windows.h>
23
24 static BOOL is_unicode_enabled = TRUE;
25
26 static BOOL cmpStrAW(const char* a, const WCHAR* b, DWORD lenA, DWORD lenB)
27 {
28 WCHAR aw[1024];
29
30 DWORD len = MultiByteToWideChar( AreFileApisANSI() ? CP_ACP : CP_OEMCP, 0,
31 a, lenA, aw, sizeof(aw) / sizeof(aw[0]) );
32 if (len != lenB) return FALSE;
33 return memcmp(aw, b, len * sizeof(WCHAR)) == 0;
34 }
35
36 static void testGetModuleFileName(const char* name)
37 {
38 HMODULE hMod;
39 char bufA[MAX_PATH];
40 WCHAR bufW[MAX_PATH];
41 DWORD len1A, len1W = 0, len2A, len2W = 0;
42
43 hMod = (name) ? GetModuleHandle(name) : NULL;
44
45 /* first test, with enough space in buffer */
46 memset(bufA, '-', sizeof(bufA));
47 len1A = GetModuleFileNameA(hMod, bufA, sizeof(bufA));
48 ok(len1A > 0, "Getting module filename for handle %p\n", hMod);
49
50 if (is_unicode_enabled)
51 {
52 memset(bufW, '-', sizeof(bufW));
53 len1W = GetModuleFileNameW(hMod, bufW, sizeof(bufW) / sizeof(WCHAR));
54 ok(len1W > 0, "Getting module filename for handle %p\n", hMod);
55 }
56
57 ok(len1A == strlen(bufA), "Unexpected length of GetModuleFilenameA (%d/%d)\n", len1A, lstrlenA(bufA));
58
59 if (is_unicode_enabled)
60 {
61 ok(len1W == lstrlenW(bufW), "Unexpected length of GetModuleFilenameW (%d/%d)\n", len1W, lstrlenW(bufW));
62 ok(cmpStrAW(bufA, bufW, len1A, len1W), "Comparing GetModuleFilenameAW results\n");
63 }
64
65 /* second test with a buffer too small */
66 memset(bufA, '-', sizeof(bufA));
67 len2A = GetModuleFileNameA(hMod, bufA, len1A / 2);
68 ok(len2A > 0, "Getting module filename for handle %p\n", hMod);
69
70 if (is_unicode_enabled)
71 {
72 memset(bufW, '-', sizeof(bufW));
73 len2W = GetModuleFileNameW(hMod, bufW, len1W / 2);
74 ok(len2W > 0, "Getting module filename for handle %p\n", hMod);
75 ok(cmpStrAW(bufA, bufW, len2A, len2W), "Comparing GetModuleFilenameAW results with buffer too small\n" );
76 ok(len1W / 2 == len2W, "Correct length in GetModuleFilenameW with buffer too small (%d/%d)\n", len1W / 2, len2W);
77 }
78
79 ok(len1A / 2 == len2A ||
80 len1A / 2 == len2A + 1, /* Win9x */
81 "Correct length in GetModuleFilenameA with buffer too small (%d/%d)\n", len1A / 2, len2A);
82 }
83
84 static void testGetModuleFileName_Wrong(void)
85 {
86 char bufA[MAX_PATH];
87 WCHAR bufW[MAX_PATH];
88
89 /* test wrong handle */
90 if (is_unicode_enabled)
91 {
92 bufW[0] = '*';
93 ok(GetModuleFileNameW((void*)0xffffffff, bufW, sizeof(bufW) / sizeof(WCHAR)) == 0, "Unexpected success in module handle\n");
94 ok(bufW[0] == '*', "When failing, buffer shouldn't be written to\n");
95 }
96
97 bufA[0] = '*';
98 ok(GetModuleFileNameA((void*)0xffffffff, bufA, sizeof(bufA)) == 0, "Unexpected success in module handle\n");
99 ok(bufA[0] == '*' ||
100 bufA[0] == 0 /* Win9x */,
101 "When failing, buffer shouldn't be written to\n");
102 }
103
104 static void testLoadLibraryA(void)
105 {
106 HMODULE hModule, hModule1;
107 FARPROC fp;
108
109 SetLastError(0xdeadbeef);
110 hModule = LoadLibraryA("kernel32.dll");
111 ok( hModule != NULL, "kernel32.dll should be loadable\n");
112 ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %d\n", GetLastError());
113
114 fp = GetProcAddress(hModule, "CreateFileA");
115 ok( fp != NULL, "CreateFileA should be there\n");
116 ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %d\n", GetLastError());
117
118 SetLastError(0xdeadbeef);
119 hModule1 = LoadLibraryA("kernel32 ");
120 /* Only winNT does this */
121 if (GetLastError() != ERROR_DLL_NOT_FOUND)
122 {
123 ok( hModule1 != NULL, "\"kernel32 \" should be loadable\n");
124 ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %d\n", GetLastError());
125 ok( hModule == hModule1, "Loaded wrong module\n");
126 FreeLibrary(hModule1);
127 }
128 FreeLibrary(hModule);
129 }
130
131 static void testNestedLoadLibraryA(void)
132 {
133 static const char dllname[] = "shell32.dll";
134 char path1[MAX_PATH], path2[MAX_PATH];
135 HMODULE hModule1, hModule2, hModule3;
136
137 /* This is not really a Windows conformance test, but more a Wine
138 * regression test. Wine's builtin dlls can be loaded from multiple paths,
139 * and this test tries to make sure that Wine does not get confused and
140 * really unloads the Unix .so file at the right time. Failure to do so
141 * will result in the dll being unloadable.
142 * This test must be done with a dll that can be unloaded, which means:
143 * - it must not already be loaded
144 * - it must not have a 16-bit counterpart
145 */
146 GetWindowsDirectory(path1, sizeof(path1));
147 strcat(path1, "\\system\\");
148 strcat(path1, dllname);
149 hModule1 = LoadLibraryA(path1);
150 if (!hModule1)
151 {
152 /* We must be on Windows NT, so we cannot test */
153 return;
154 }
155
156 GetWindowsDirectory(path2, sizeof(path2));
157 strcat(path2, "\\system32\\");
158 strcat(path2, dllname);
159 hModule2 = LoadLibraryA(path2);
160 if (!hModule2)
161 {
162 /* We must be on Windows 9x, so we cannot test */
163 ok(FreeLibrary(hModule1), "FreeLibrary() failed\n");
164 return;
165 }
166
167 /* The first LoadLibrary() call may have registered the dll under the
168 * system32 path. So load it, again, under the '...\system\...' path so
169 * Wine does not immediately notice that it is already loaded.
170 */
171 hModule3 = LoadLibraryA(path1);
172 ok(hModule3 != NULL, "LoadLibrary(%s) failed\n", path1);
173
174 /* Now fully unload the dll */
175 ok(FreeLibrary(hModule3), "FreeLibrary() failed\n");
176 ok(FreeLibrary(hModule2), "FreeLibrary() failed\n");
177 ok(FreeLibrary(hModule1), "FreeLibrary() failed\n");
178 ok(GetModuleHandle(dllname) == NULL, "%s was not fully unloaded\n", dllname);
179
180 /* Try to load the dll again, if refcounting is ok, this should work */
181 hModule1 = LoadLibraryA(path1);
182 ok(hModule1 != NULL, "LoadLibrary(%s) failed\n", path1);
183 if (hModule1 != NULL)
184 ok(FreeLibrary(hModule1), "FreeLibrary() failed\n");
185 }
186
187 static void testLoadLibraryA_Wrong(void)
188 {
189 HMODULE hModule;
190
191 /* Try to load a nonexistent dll */
192 SetLastError(0xdeadbeef);
193 hModule = LoadLibraryA("non_ex_pv.dll");
194 ok( !hModule, "non_ex_pv.dll should be not loadable\n");
195 ok( GetLastError() == ERROR_MOD_NOT_FOUND || GetLastError() == ERROR_DLL_NOT_FOUND,
196 "Expected ERROR_MOD_NOT_FOUND or ERROR_DLL_NOT_FOUND (win9x), got %d\n", GetLastError());
197
198 /* Just in case */
199 FreeLibrary(hModule);
200 }
201
202 static void testGetProcAddress_Wrong(void)
203 {
204 FARPROC fp;
205
206 SetLastError(0xdeadbeef);
207 fp = GetProcAddress(NULL, "non_ex_call");
208 ok( !fp, "non_ex_call should not be found\n");
209 ok( GetLastError() == ERROR_PROC_NOT_FOUND || GetLastError() == ERROR_INVALID_HANDLE,
210 "Expected ERROR_PROC_NOT_FOUND or ERROR_INVALID_HANDLE(win9x), got %d\n", GetLastError());
211
212 SetLastError(0xdeadbeef);
213 fp = GetProcAddress((HMODULE)0xdeadbeef, "non_ex_call");
214 ok( !fp, "non_ex_call should not be found\n");
215 ok( GetLastError() == ERROR_MOD_NOT_FOUND || GetLastError() == ERROR_INVALID_HANDLE,
216 "Expected ERROR_MOD_NOT_FOUND or ERROR_INVALID_HANDLE(win9x), got %d\n", GetLastError());
217 }
218
219 static void testLoadLibraryEx(void)
220 {
221 CHAR path[MAX_PATH];
222 HMODULE hmodule;
223 HANDLE hfile;
224
225 hfile = CreateFileA("testfile.dll", GENERIC_READ | GENERIC_WRITE,
226 FILE_SHARE_READ | FILE_SHARE_WRITE,
227 NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
228 ok(hfile != INVALID_HANDLE_VALUE, "Expected a valid file handle\n");
229
230 /* NULL lpFileName */
231 if (is_unicode_enabled)
232 {
233 SetLastError(0xdeadbeef);
234 hmodule = LoadLibraryExA(NULL, NULL, 0);
235 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
236 ok(GetLastError() == ERROR_MOD_NOT_FOUND ||
237 GetLastError() == ERROR_INVALID_PARAMETER, /* win9x */
238 "Expected ERROR_MOD_NOT_FOUND or ERROR_INVALID_PARAMETER, got %d\n",
239 GetLastError());
240 }
241 else
242 win_skip("NULL filename crashes on WinMe\n");
243
244 /* empty lpFileName */
245 SetLastError(0xdeadbeef);
246 hmodule = LoadLibraryExA("", NULL, 0);
247 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
248 ok(GetLastError() == ERROR_MOD_NOT_FOUND ||
249 GetLastError() == ERROR_DLL_NOT_FOUND, /* win9x */
250 "Expected ERROR_MOD_NOT_FOUND or ERROR_DLL_NOT_FOUND, got %d\n",
251 GetLastError());
252
253 /* hFile is non-NULL */
254 SetLastError(0xdeadbeef);
255 hmodule = LoadLibraryExA("testfile.dll", hfile, 0);
256 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
257 todo_wine
258 {
259 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
260 GetLastError() == ERROR_INVALID_PARAMETER || /* win2k3 */
261 GetLastError() == ERROR_FILE_NOT_FOUND, /* win9x */
262 "Unexpected last error, got %d\n", GetLastError());
263 }
264
265 SetLastError(0xdeadbeef);
266 hmodule = LoadLibraryExA("testfile.dll", (HANDLE)0xdeadbeef, 0);
267 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
268 todo_wine
269 {
270 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
271 GetLastError() == ERROR_INVALID_PARAMETER || /* win2k3 */
272 GetLastError() == ERROR_FILE_NOT_FOUND, /* win9x */
273 "Unexpected last error, got %d\n", GetLastError());
274 }
275
276 /* try to open a file that is locked */
277 SetLastError(0xdeadbeef);
278 hmodule = LoadLibraryExA("testfile.dll", NULL, 0);
279 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
280 todo_wine
281 {
282 ok(GetLastError() == ERROR_SHARING_VIOLATION ||
283 GetLastError() == ERROR_FILE_NOT_FOUND, /* win9x */
284 "Expected ERROR_SHARING_VIOLATION or ERROR_FILE_NOT_FOUND, got %d\n",
285 GetLastError());
286 }
287
288 /* lpFileName does not matter */
289 if (is_unicode_enabled)
290 {
291 SetLastError(0xdeadbeef);
292 hmodule = LoadLibraryExA(NULL, hfile, 0);
293 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
294 ok(GetLastError() == ERROR_MOD_NOT_FOUND ||
295 GetLastError() == ERROR_INVALID_PARAMETER, /* win2k3 */
296 "Expected ERROR_MOD_NOT_FOUND or ERROR_INVALID_PARAMETER, got %d\n",
297 GetLastError());
298 }
299
300 CloseHandle(hfile);
301
302 /* load empty file */
303 SetLastError(0xdeadbeef);
304 hmodule = LoadLibraryExA("testfile.dll", NULL, LOAD_LIBRARY_AS_DATAFILE);
305 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
306 todo_wine
307 {
308 ok(GetLastError() == ERROR_FILE_INVALID ||
309 GetLastError() == ERROR_BAD_FORMAT, /* win9x */
310 "Expected ERROR_FILE_INVALID or ERROR_BAD_FORMAT, got %d\n",
311 GetLastError());
312 }
313
314 DeleteFileA("testfile.dll");
315
316 GetSystemDirectoryA(path, MAX_PATH);
317 if (path[lstrlenA(path) - 1] != '\\')
318 lstrcatA(path, "\\");
319 lstrcatA(path, "kernel32.dll");
320
321 /* load kernel32.dll with an absolute path */
322 SetLastError(0xdeadbeef);
323 hmodule = LoadLibraryExA(path, NULL, LOAD_LIBRARY_AS_DATAFILE);
324 ok(hmodule != 0, "Expected valid module handle\n");
325 ok(GetLastError() == 0xdeadbeef ||
326 GetLastError() == ERROR_SUCCESS, /* win9x */
327 "Expected 0xdeadbeef or ERROR_SUCCESS, got %d\n", GetLastError());
328
329 /* try invalid file handle */
330 SetLastError(0xdeadbeef);
331 hmodule = LoadLibraryExA(path, (HANDLE)0xdeadbeef, 0);
332 if (!hmodule) /* succeeds on xp and older */
333 ok(GetLastError() == ERROR_INVALID_PARAMETER, "wrong error %u\n", GetLastError());
334
335 CloseHandle(hmodule);
336
337 /* load kernel32.dll with no path */
338 SetLastError(0xdeadbeef);
339 hmodule = LoadLibraryExA("kernel32.dll", NULL, LOAD_LIBRARY_AS_DATAFILE);
340 ok(hmodule != 0, "Expected valid module handle\n");
341 ok(GetLastError() == 0xdeadbeef ||
342 GetLastError() == ERROR_SUCCESS, /* win9x */
343 "Expected 0xdeadbeef or ERROR_SUCCESS, got %d\n", GetLastError());
344
345 CloseHandle(hmodule);
346
347 GetCurrentDirectoryA(MAX_PATH, path);
348 if (path[lstrlenA(path) - 1] != '\\')
349 lstrcatA(path, "\\");
350 lstrcatA(path, "kernel32.dll");
351
352 /* load kernel32.dll with an absolute path that does not exist */
353 SetLastError(0xdeadbeef);
354 hmodule = LoadLibraryExA(path, NULL, LOAD_LIBRARY_AS_DATAFILE);
355 todo_wine
356 {
357 ok(hmodule == 0, "Expected 0, got %p\n", hmodule);
358 }
359 ok(GetLastError() == ERROR_FILE_NOT_FOUND ||
360 broken(GetLastError() == ERROR_INVALID_HANDLE), /* nt4 */
361 "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
362 }
363
364 START_TEST(module)
365 {
366 WCHAR filenameW[MAX_PATH];
367
368 /* Test if we can use GetModuleFileNameW */
369
370 SetLastError(0xdeadbeef);
371 GetModuleFileNameW(NULL, filenameW, MAX_PATH);
372 if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
373 {
374 win_skip("GetModuleFileNameW not existing on this platform, skipping W-calls\n");
375 is_unicode_enabled = FALSE;
376 }
377
378 testGetModuleFileName(NULL);
379 testGetModuleFileName("kernel32.dll");
380 testGetModuleFileName_Wrong();
381
382 testLoadLibraryA();
383 testNestedLoadLibraryA();
384 testLoadLibraryA_Wrong();
385 testGetProcAddress_Wrong();
386 testLoadLibraryEx();
387 }
388
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.