1 /*
2 * Modules
3 *
4 * Copyright 1995 Alexandre Julliard
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 "config.h"
22 #include "wine/port.h"
23
24 #include <fcntl.h>
25 #include <stdarg.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <sys/types.h>
30 #ifdef HAVE_UNISTD_H
31 # include <unistd.h>
32 #endif
33 #include "ntstatus.h"
34 #define WIN32_NO_STATUS
35 #include "winerror.h"
36 #include "windef.h"
37 #include "winbase.h"
38 #include "winternl.h"
39 #include "kernel_private.h"
40
41 #include "wine/exception.h"
42 #include "wine/debug.h"
43 #include "wine/unicode.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(module);
46
47 #define NE_FFLAGS_LIBMODULE 0x8000
48
49 static WCHAR *dll_directory; /* extra path for SetDllDirectoryW */
50
51 static CRITICAL_SECTION dlldir_section;
52 static CRITICAL_SECTION_DEBUG critsect_debug =
53 {
54 0, 0, &dlldir_section,
55 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
56 0, 0, { (DWORD_PTR)(__FILE__ ": dlldir_section") }
57 };
58 static CRITICAL_SECTION dlldir_section = { &critsect_debug, -1, 0, 0, 0, 0 };
59
60
61 /****************************************************************************
62 * GetDllDirectoryA (KERNEL32.@)
63 */
64 DWORD WINAPI GetDllDirectoryA( DWORD buf_len, LPSTR buffer )
65 {
66 DWORD len;
67
68 RtlEnterCriticalSection( &dlldir_section );
69 len = dll_directory ? FILE_name_WtoA( dll_directory, strlenW(dll_directory), NULL, 0 ) : 0;
70 if (buffer && buf_len > len)
71 {
72 if (dll_directory) FILE_name_WtoA( dll_directory, -1, buffer, buf_len );
73 else *buffer = 0;
74 }
75 else
76 {
77 len++; /* for terminating null */
78 if (buffer) *buffer = 0;
79 }
80 RtlLeaveCriticalSection( &dlldir_section );
81 return len;
82 }
83
84
85 /****************************************************************************
86 * GetDllDirectoryW (KERNEL32.@)
87 */
88 DWORD WINAPI GetDllDirectoryW( DWORD buf_len, LPWSTR buffer )
89 {
90 DWORD len;
91
92 RtlEnterCriticalSection( &dlldir_section );
93 len = dll_directory ? strlenW( dll_directory ) : 0;
94 if (buffer && buf_len > len)
95 {
96 if (dll_directory) memcpy( buffer, dll_directory, (len + 1) * sizeof(WCHAR) );
97 else *buffer = 0;
98 }
99 else
100 {
101 len++; /* for terminating null */
102 if (buffer) *buffer = 0;
103 }
104 RtlLeaveCriticalSection( &dlldir_section );
105 return len;
106 }
107
108
109 /****************************************************************************
110 * SetDllDirectoryA (KERNEL32.@)
111 */
112 BOOL WINAPI SetDllDirectoryA( LPCSTR dir )
113 {
114 WCHAR *dirW;
115 BOOL ret;
116
117 if (!(dirW = FILE_name_AtoW( dir, TRUE ))) return FALSE;
118 ret = SetDllDirectoryW( dirW );
119 HeapFree( GetProcessHeap(), 0, dirW );
120 return ret;
121 }
122
123
124 /****************************************************************************
125 * SetDllDirectoryW (KERNEL32.@)
126 */
127 BOOL WINAPI SetDllDirectoryW( LPCWSTR dir )
128 {
129 WCHAR *newdir = NULL;
130
131 if (dir)
132 {
133 DWORD len = (strlenW(dir) + 1) * sizeof(WCHAR);
134 if (!(newdir = HeapAlloc( GetProcessHeap(), 0, len )))
135 {
136 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
137 return FALSE;
138 }
139 memcpy( newdir, dir, len );
140 }
141
142 RtlEnterCriticalSection( &dlldir_section );
143 HeapFree( GetProcessHeap(), 0, dll_directory );
144 dll_directory = newdir;
145 RtlLeaveCriticalSection( &dlldir_section );
146 return TRUE;
147 }
148
149
150 /****************************************************************************
151 * DisableThreadLibraryCalls (KERNEL32.@)
152 *
153 * Inform the module loader that thread notifications are not required for a dll.
154 *
155 * PARAMS
156 * hModule [I] Module handle to skip calls for
157 *
158 * RETURNS
159 * Success: TRUE. Thread attach and detach notifications will not be sent
160 * to hModule.
161 * Failure: FALSE. Use GetLastError() to determine the cause.
162 *
163 * NOTES
164 * This is typically called from the dll entry point of a dll during process
165 * attachment, for dlls that do not need to process thread notifications.
166 */
167 BOOL WINAPI DisableThreadLibraryCalls( HMODULE hModule )
168 {
169 NTSTATUS nts = LdrDisableThreadCalloutsForDll( hModule );
170 if (nts == STATUS_SUCCESS) return TRUE;
171
172 SetLastError( RtlNtStatusToDosError( nts ) );
173 return FALSE;
174 }
175
176
177 /* Check whether a file is an OS/2 or a very old Windows executable
178 * by testing on import of KERNEL.
179 *
180 * FIXME: is reading the module imports the only way of discerning
181 * old Windows binaries from OS/2 ones ? At least it seems so...
182 */
183 static DWORD MODULE_Decide_OS2_OldWin(HANDLE hfile, const IMAGE_DOS_HEADER *mz, const IMAGE_OS2_HEADER *ne)
184 {
185 DWORD currpos = SetFilePointer( hfile, 0, NULL, SEEK_CUR);
186 DWORD ret = BINARY_OS216;
187 LPWORD modtab = NULL;
188 LPSTR nametab = NULL;
189 DWORD len;
190 int i;
191
192 /* read modref table */
193 if ( (SetFilePointer( hfile, mz->e_lfanew + ne->ne_modtab, NULL, SEEK_SET ) == -1)
194 || (!(modtab = HeapAlloc( GetProcessHeap(), 0, ne->ne_cmod*sizeof(WORD))))
195 || (!(ReadFile(hfile, modtab, ne->ne_cmod*sizeof(WORD), &len, NULL)))
196 || (len != ne->ne_cmod*sizeof(WORD)) )
197 goto broken;
198
199 /* read imported names table */
200 if ( (SetFilePointer( hfile, mz->e_lfanew + ne->ne_imptab, NULL, SEEK_SET ) == -1)
201 || (!(nametab = HeapAlloc( GetProcessHeap(), 0, ne->ne_enttab - ne->ne_imptab)))
202 || (!(ReadFile(hfile, nametab, ne->ne_enttab - ne->ne_imptab, &len, NULL)))
203 || (len != ne->ne_enttab - ne->ne_imptab) )
204 goto broken;
205
206 for (i=0; i < ne->ne_cmod; i++)
207 {
208 LPSTR module = &nametab[modtab[i]];
209 TRACE("modref: %.*s\n", module[0], &module[1]);
210 if (!(strncmp(&module[1], "KERNEL", module[0])))
211 { /* very old Windows file */
212 MESSAGE("This seems to be a very old (pre-3.0) Windows executable. Expect crashes, especially if this is a real-mode binary !\n");
213 ret = BINARY_WIN16;
214 goto good;
215 }
216 }
217
218 broken:
219 ERR("Hmm, an error occurred. Is this binary file broken?\n");
220
221 good:
222 HeapFree( GetProcessHeap(), 0, modtab);
223 HeapFree( GetProcessHeap(), 0, nametab);
224 SetFilePointer( hfile, currpos, NULL, SEEK_SET); /* restore filepos */
225 return ret;
226 }
227
228 /***********************************************************************
229 * MODULE_GetBinaryType
230 */
231 void MODULE_get_binary_info( HANDLE hfile, struct binary_info *info )
232 {
233 union
234 {
235 struct
236 {
237 unsigned char magic[4];
238 unsigned char class;
239 unsigned char data;
240 unsigned char version;
241 unsigned char ignored[9];
242 unsigned short type;
243 unsigned short machine;
244 } elf;
245 struct
246 {
247 unsigned int magic;
248 unsigned int cputype;
249 unsigned int cpusubtype;
250 unsigned int filetype;
251 } macho;
252 IMAGE_DOS_HEADER mz;
253 } header;
254
255 DWORD len;
256
257 memset( info, 0, sizeof(*info) );
258
259 /* Seek to the start of the file and read the header information. */
260 if (SetFilePointer( hfile, 0, NULL, SEEK_SET ) == -1) return;
261 if (!ReadFile( hfile, &header, sizeof(header), &len, NULL ) || len != sizeof(header)) return;
262
263 if (!memcmp( header.elf.magic, "\177ELF", 4 ))
264 {
265 if (header.elf.class == 2) info->flags |= BINARY_FLAG_64BIT;
266 /* FIXME: we don't bother to check byte order, architecture, etc. */
267 switch(header.elf.type)
268 {
269 case 2: info->type = BINARY_UNIX_EXE; break;
270 case 3: info->type = BINARY_UNIX_LIB; break;
271 }
272 }
273 /* Mach-o File with Endian set to Big Endian or Little Endian */
274 else if (header.macho.magic == 0xfeedface || header.macho.magic == 0xcefaedfe)
275 {
276 if ((header.macho.cputype >> 24) == 1) info->flags |= BINARY_FLAG_64BIT;
277 switch(header.macho.filetype)
278 {
279 case 2: info->type = BINARY_UNIX_EXE; break;
280 case 8: info->type = BINARY_UNIX_LIB; break;
281 }
282 }
283 /* Not ELF, try DOS */
284 else if (header.mz.e_magic == IMAGE_DOS_SIGNATURE)
285 {
286 union
287 {
288 IMAGE_OS2_HEADER os2;
289 IMAGE_NT_HEADERS32 nt;
290 } ext_header;
291
292 /* We do have a DOS image so we will now try to seek into
293 * the file by the amount indicated by the field
294 * "Offset to extended header" and read in the
295 * "magic" field information at that location.
296 * This will tell us if there is more header information
297 * to read or not.
298 */
299 info->type = BINARY_DOS;
300 if (SetFilePointer( hfile, header.mz.e_lfanew, NULL, SEEK_SET ) == -1) return;
301 if (!ReadFile( hfile, &ext_header, sizeof(ext_header), &len, NULL ) || len < 4) return;
302
303 /* Reading the magic field succeeded so
304 * we will try to determine what type it is.
305 */
306 if (!memcmp( &ext_header.nt.Signature, "PE\0\0", 4 ))
307 {
308 if (len >= sizeof(ext_header.nt.FileHeader))
309 {
310 info->type = BINARY_PE;
311 if (ext_header.nt.FileHeader.Characteristics & IMAGE_FILE_DLL)
312 info->flags |= BINARY_FLAG_DLL;
313 if (len < sizeof(ext_header.nt)) /* clear remaining part of header if missing */
314 memset( (char *)&ext_header.nt + len, 0, sizeof(ext_header.nt) - len );
315 switch (ext_header.nt.OptionalHeader.Magic)
316 {
317 case IMAGE_NT_OPTIONAL_HDR32_MAGIC:
318 info->res_start = (void *)(ULONG_PTR)ext_header.nt.OptionalHeader.ImageBase;
319 info->res_end = (void *)((ULONG_PTR)ext_header.nt.OptionalHeader.ImageBase +
320 ext_header.nt.OptionalHeader.SizeOfImage);
321 break;
322 case IMAGE_NT_OPTIONAL_HDR64_MAGIC:
323 info->flags |= BINARY_FLAG_64BIT;
324 break;
325 }
326 }
327 }
328 else if (!memcmp( &ext_header.os2.ne_magic, "NE", 2 ))
329 {
330 /* This is a Windows executable (NE) header. This can
331 * mean either a 16-bit OS/2 or a 16-bit Windows or even a
332 * DOS program (running under a DOS extender). To decide
333 * which, we'll have to read the NE header.
334 */
335 if (len >= sizeof(ext_header.os2))
336 {
337 if (ext_header.os2.ne_flags & NE_FFLAGS_LIBMODULE) info->flags |= BINARY_FLAG_DLL;
338 switch ( ext_header.os2.ne_exetyp )
339 {
340 case 1: info->type = BINARY_OS216; break; /* OS/2 */
341 case 2: info->type = BINARY_WIN16; break; /* Windows */
342 case 3: info->type = BINARY_DOS; break; /* European MS-DOS 4.x */
343 case 4: info->type = BINARY_WIN16; break; /* Windows 386; FIXME: is this 32bit??? */
344 case 5: info->type = BINARY_DOS; break; /* BOSS, Borland Operating System Services */
345 /* other types, e.g. 0 is: "unknown" */
346 default: info->type = MODULE_Decide_OS2_OldWin(hfile, &header.mz, &ext_header.os2); break;
347 }
348 }
349 }
350 }
351 }
352
353 /***********************************************************************
354 * GetBinaryTypeW [KERNEL32.@]
355 *
356 * Determine whether a file is executable, and if so, what kind.
357 *
358 * PARAMS
359 * lpApplicationName [I] Path of the file to check
360 * lpBinaryType [O] Destination for the binary type
361 *
362 * RETURNS
363 * TRUE, if the file is an executable, in which case lpBinaryType is set.
364 * FALSE, if the file is not an executable or if the function fails.
365 *
366 * NOTES
367 * The type of executable is a property that determines which subsystem an
368 * executable file runs under. lpBinaryType can be set to one of the following
369 * values:
370 * SCS_32BIT_BINARY: A Win32 based application
371 * SCS_DOS_BINARY: An MS-Dos based application
372 * SCS_WOW_BINARY: A Win16 based application
373 * SCS_PIF_BINARY: A PIF file that executes an MS-Dos based app
374 * SCS_POSIX_BINARY: A POSIX based application ( Not implemented )
375 * SCS_OS216_BINARY: A 16bit OS/2 based application
376 *
377 * To find the binary type, this function reads in the files header information.
378 * If extended header information is not present it will assume that the file
379 * is a DOS executable. If extended header information is present it will
380 * determine if the file is a 16 or 32 bit Windows executable by checking the
381 * flags in the header.
382 *
383 * ".com" and ".pif" files are only recognized by their file name extension,
384 * as per native Windows.
385 */
386 BOOL WINAPI GetBinaryTypeW( LPCWSTR lpApplicationName, LPDWORD lpBinaryType )
387 {
388 BOOL ret = FALSE;
389 HANDLE hfile;
390 struct binary_info binary_info;
391
392 TRACE("%s\n", debugstr_w(lpApplicationName) );
393
394 /* Sanity check.
395 */
396 if ( lpApplicationName == NULL || lpBinaryType == NULL )
397 return FALSE;
398
399 /* Open the file indicated by lpApplicationName for reading.
400 */
401 hfile = CreateFileW( lpApplicationName, GENERIC_READ, FILE_SHARE_READ,
402 NULL, OPEN_EXISTING, 0, 0 );
403 if ( hfile == INVALID_HANDLE_VALUE )
404 return FALSE;
405
406 /* Check binary type
407 */
408 MODULE_get_binary_info( hfile, &binary_info );
409 switch (binary_info.type)
410 {
411 case BINARY_UNKNOWN:
412 {
413 static const WCHAR comW[] = { '.','C','O','M',0 };
414 static const WCHAR pifW[] = { '.','P','I','F',0 };
415 const WCHAR *ptr;
416
417 /* try to determine from file name */
418 ptr = strrchrW( lpApplicationName, '.' );
419 if (!ptr) break;
420 if (!strcmpiW( ptr, comW ))
421 {
422 *lpBinaryType = SCS_DOS_BINARY;
423 ret = TRUE;
424 }
425 else if (!strcmpiW( ptr, pifW ))
426 {
427 *lpBinaryType = SCS_PIF_BINARY;
428 ret = TRUE;
429 }
430 break;
431 }
432 case BINARY_PE:
433 *lpBinaryType = (binary_info.flags & BINARY_FLAG_64BIT) ? SCS_64BIT_BINARY : SCS_32BIT_BINARY;
434 ret = TRUE;
435 break;
436 case BINARY_WIN16:
437 *lpBinaryType = SCS_WOW_BINARY;
438 ret = TRUE;
439 break;
440 case BINARY_OS216:
441 *lpBinaryType = SCS_OS216_BINARY;
442 ret = TRUE;
443 break;
444 case BINARY_DOS:
445 *lpBinaryType = SCS_DOS_BINARY;
446 ret = TRUE;
447 break;
448 case BINARY_UNIX_EXE:
449 case BINARY_UNIX_LIB:
450 ret = FALSE;
451 break;
452 }
453
454 CloseHandle( hfile );
455 return ret;
456 }
457
458 /***********************************************************************
459 * GetBinaryTypeA [KERNEL32.@]
460 * GetBinaryType [KERNEL32.@]
461 *
462 * See GetBinaryTypeW.
463 */
464 BOOL WINAPI GetBinaryTypeA( LPCSTR lpApplicationName, LPDWORD lpBinaryType )
465 {
466 ANSI_STRING app_nameA;
467 NTSTATUS status;
468
469 TRACE("%s\n", debugstr_a(lpApplicationName));
470
471 /* Sanity check.
472 */
473 if ( lpApplicationName == NULL || lpBinaryType == NULL )
474 return FALSE;
475
476 RtlInitAnsiString(&app_nameA, lpApplicationName);
477 status = RtlAnsiStringToUnicodeString(&NtCurrentTeb()->StaticUnicodeString,
478 &app_nameA, FALSE);
479 if (!status)
480 return GetBinaryTypeW(NtCurrentTeb()->StaticUnicodeString.Buffer, lpBinaryType);
481
482 SetLastError(RtlNtStatusToDosError(status));
483 return FALSE;
484 }
485
486 /***********************************************************************
487 * GetModuleHandleExA (KERNEL32.@)
488 */
489 BOOL WINAPI GetModuleHandleExA( DWORD flags, LPCSTR name, HMODULE *module )
490 {
491 WCHAR *nameW;
492
493 if (!name || (flags & GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS))
494 return GetModuleHandleExW( flags, (LPCWSTR)name, module );
495
496 if (!(nameW = FILE_name_AtoW( name, FALSE ))) return FALSE;
497 return GetModuleHandleExW( flags, nameW, module );
498 }
499
500 /***********************************************************************
501 * GetModuleHandleExW (KERNEL32.@)
502 */
503 BOOL WINAPI GetModuleHandleExW( DWORD flags, LPCWSTR name, HMODULE *module )
504 {
505 NTSTATUS status = STATUS_SUCCESS;
506 HMODULE ret;
507 ULONG magic;
508
509 /* if we are messing with the refcount, grab the loader lock */
510 if ((flags & GET_MODULE_HANDLE_EX_FLAG_PIN) ||
511 !(flags & GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT))
512 LdrLockLoaderLock( 0, NULL, &magic );
513
514 if (!name)
515 {
516 ret = NtCurrentTeb()->Peb->ImageBaseAddress;
517 }
518 else if (flags & GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS)
519 {
520 void *dummy;
521 if (!(ret = RtlPcToFileHeader( (void *)name, &dummy ))) status = STATUS_DLL_NOT_FOUND;
522 }
523 else
524 {
525 UNICODE_STRING wstr;
526 RtlInitUnicodeString( &wstr, name );
527 status = LdrGetDllHandle( NULL, 0, &wstr, &ret );
528 }
529
530 if (status == STATUS_SUCCESS)
531 {
532 if (flags & GET_MODULE_HANDLE_EX_FLAG_PIN)
533 FIXME( "should pin refcount for %p\n", ret );
534 else if (!(flags & GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT))
535 LdrAddRefDll( 0, ret );
536 }
537 else SetLastError( RtlNtStatusToDosError( status ) );
538
539 if ((flags & GET_MODULE_HANDLE_EX_FLAG_PIN) ||
540 !(flags & GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT))
541 LdrUnlockLoaderLock( 0, magic );
542
543 if (module) *module = ret;
544 return (status == STATUS_SUCCESS);
545 }
546
547 /***********************************************************************
548 * GetModuleHandleA (KERNEL32.@)
549 *
550 * Get the handle of a dll loaded into the process address space.
551 *
552 * PARAMS
553 * module [I] Name of the dll
554 *
555 * RETURNS
556 * Success: A handle to the loaded dll.
557 * Failure: A NULL handle. Use GetLastError() to determine the cause.
558 */
559 HMODULE WINAPI GetModuleHandleA(LPCSTR module)
560 {
561 HMODULE ret;
562
563 if (!GetModuleHandleExA( GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, module, &ret )) ret = 0;
564 return ret;
565 }
566
567 /***********************************************************************
568 * GetModuleHandleW (KERNEL32.@)
569 *
570 * Unicode version of GetModuleHandleA.
571 */
572 HMODULE WINAPI GetModuleHandleW(LPCWSTR module)
573 {
574 HMODULE ret;
575
576 if (!GetModuleHandleExW( GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, module, &ret )) ret = 0;
577 return ret;
578 }
579
580
581 /***********************************************************************
582 * GetModuleFileNameA (KERNEL32.@)
583 *
584 * Get the file name of a loaded module from its handle.
585 *
586 * RETURNS
587 * Success: The length of the file name, excluding the terminating NUL.
588 * Failure: 0. Use GetLastError() to determine the cause.
589 *
590 * NOTES
591 * This function always returns the long path of hModule (as opposed to
592 * GetModuleFileName16() which returns short paths when the modules version
593 * field is < 4.0).
594 * The function doesn't write a terminating '\0' if the buffer is too
595 * small.
596 */
597 DWORD WINAPI GetModuleFileNameA(
598 HMODULE hModule, /* [in] Module handle (32 bit) */
599 LPSTR lpFileName, /* [out] Destination for file name */
600 DWORD size ) /* [in] Size of lpFileName in characters */
601 {
602 LPWSTR filenameW = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) );
603 DWORD len;
604
605 if (!filenameW)
606 {
607 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
608 return 0;
609 }
610 if ((len = GetModuleFileNameW( hModule, filenameW, size )))
611 {
612 len = FILE_name_WtoA( filenameW, len, lpFileName, size );
613 if (len < size)
614 lpFileName[len] = '\0';
615 else
616 SetLastError( ERROR_INSUFFICIENT_BUFFER );
617 }
618 HeapFree( GetProcessHeap(), 0, filenameW );
619 return len;
620 }
621
622 /***********************************************************************
623 * GetModuleFileNameW (KERNEL32.@)
624 *
625 * Unicode version of GetModuleFileNameA.
626 */
627 DWORD WINAPI GetModuleFileNameW( HMODULE hModule, LPWSTR lpFileName, DWORD size )
628 {
629 ULONG magic, len = 0;
630 LDR_MODULE *pldr;
631 NTSTATUS nts;
632 WIN16_SUBSYSTEM_TIB *win16_tib;
633
634 if (!hModule && ((win16_tib = NtCurrentTeb()->Tib.SubSystemTib)) && win16_tib->exe_name)
635 {
636 len = min(size, win16_tib->exe_name->Length / sizeof(WCHAR));
637 memcpy( lpFileName, win16_tib->exe_name->Buffer, len * sizeof(WCHAR) );
638 if (len < size) lpFileName[len] = '\0';
639 goto done;
640 }
641
642 LdrLockLoaderLock( 0, NULL, &magic );
643
644 if (!hModule) hModule = NtCurrentTeb()->Peb->ImageBaseAddress;
645 nts = LdrFindEntryForAddress( hModule, &pldr );
646 if (nts == STATUS_SUCCESS)
647 {
648 len = min(size, pldr->FullDllName.Length / sizeof(WCHAR));
649 memcpy(lpFileName, pldr->FullDllName.Buffer, len * sizeof(WCHAR));
650 if (len < size)
651 lpFileName[len] = '\0';
652 else
653 SetLastError( ERROR_INSUFFICIENT_BUFFER );
654 }
655 else SetLastError( RtlNtStatusToDosError( nts ) );
656
657 LdrUnlockLoaderLock( 0, magic );
658 done:
659 TRACE( "%s\n", debugstr_wn(lpFileName, len) );
660 return len;
661 }
662
663
664 /***********************************************************************
665 * get_dll_system_path
666 */
667 static const WCHAR *get_dll_system_path(void)
668 {
669 static WCHAR *cached_path;
670
671 if (!cached_path)
672 {
673 WCHAR *p, *path;
674 int len = 3;
675
676 len += 2 * GetSystemDirectoryW( NULL, 0 );
677 len += GetWindowsDirectoryW( NULL, 0 );
678 p = path = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
679 *p++ = '.';
680 *p++ = ';';
681 GetSystemDirectoryW( p, path + len - p);
682 p += strlenW(p);
683 /* if system directory ends in "32" add 16-bit version too */
684 if (p[-2] == '3' && p[-1] == '2')
685 {
686 *p++ = ';';
687 GetSystemDirectoryW( p, path + len - p);
688 p += strlenW(p) - 2;
689 }
690 *p++ = ';';
691 GetWindowsDirectoryW( p, path + len - p);
692 cached_path = path;
693 }
694 return cached_path;
695 }
696
697 /******************************************************************
698 * get_module_path_end
699 *
700 * Returns the end of the directory component of the module path.
701 */
702 static inline const WCHAR *get_module_path_end(const WCHAR *module)
703 {
704 const WCHAR *p;
705 const WCHAR *mod_end = module;
706 if (!module) return mod_end;
707
708 if ((p = strrchrW( mod_end, '\\' ))) mod_end = p;
709 if ((p = strrchrW( mod_end, '/' ))) mod_end = p;
710 if (mod_end == module + 2 && module[1] == ':') mod_end++;
711 if (mod_end == module && module[0] && module[1] == ':') mod_end += 2;
712
713 return mod_end;
714 }
715
716 /******************************************************************
717 * MODULE_get_dll_load_path
718 *
719 * Compute the load path to use for a given dll.
720 * Returned pointer must be freed by caller.
721 */
722 WCHAR *MODULE_get_dll_load_path( LPCWSTR module )
723 {
724 static const WCHAR pathW[] = {'P','A','T','H',0};
725
726 const WCHAR *system_path = get_dll_system_path();
727 const WCHAR *mod_end = NULL;
728 UNICODE_STRING name, value;
729 WCHAR *p, *ret;
730 int len = 0, path_len = 0;
731
732 /* adjust length for module name */
733
734 if (module)
735 mod_end = get_module_path_end( module );
736 /* if module is NULL or doesn't contain a path, fall back to directory
737 * process was loaded from */
738 if (module == mod_end)
739 {
740 module = NtCurrentTeb()->Peb->ProcessParameters->ImagePathName.Buffer;
741 mod_end = get_module_path_end( module );
742 }
743 len += (mod_end - module) + 1;
744
745 len += strlenW( system_path ) + 2;
746
747 /* get the PATH variable */
748
749 RtlInitUnicodeString( &name, pathW );
750 value.Length = 0;
751 value.MaximumLength = 0;
752 value.Buffer = NULL;
753 if (RtlQueryEnvironmentVariable_U( NULL, &name, &value ) == STATUS_BUFFER_TOO_SMALL)
754 path_len = value.Length;
755
756 RtlEnterCriticalSection( &dlldir_section );
757 if (dll_directory) len += strlenW(dll_directory) + 1;
758 if ((p = ret = HeapAlloc( GetProcessHeap(), 0, path_len + len * sizeof(WCHAR) )))
759 {
760 if (module)
761 {
762 memcpy( ret, module, (mod_end - module) * sizeof(WCHAR) );
763 p += (mod_end - module);
764 *p++ = ';';
765 }
766 if (dll_directory)
767 {
768 strcpyW( p, dll_directory );
769 p += strlenW(p);
770 *p++ = ';';
771 }
772 }
773 RtlLeaveCriticalSection( &dlldir_section );
774 if (!ret) return NULL;
775
776 strcpyW( p, system_path );
777 p += strlenW(p);
778 *p++ = ';';
779 value.Buffer = p;
780 value.MaximumLength = path_len;
781
782 while (RtlQueryEnvironmentVariable_U( NULL, &name, &value ) == STATUS_BUFFER_TOO_SMALL)
783 {
784 WCHAR *new_ptr;
785
786 /* grow the buffer and retry */
787 path_len = value.Length;
788 if (!(new_ptr = HeapReAlloc( GetProcessHeap(), 0, ret, path_len + len * sizeof(WCHAR) )))
789 {
790 HeapFree( GetProcessHeap(), 0, ret );
791 return NULL;
792 }
793 value.Buffer = new_ptr + (value.Buffer - ret);
794 value.MaximumLength = path_len;
795 ret = new_ptr;
796 }
797 value.Buffer[value.Length / sizeof(WCHAR)] = 0;
798 return ret;
799 }
800
801
802 /******************************************************************
803 * load_library_as_datafile
804 */
805 static BOOL load_library_as_datafile( LPCWSTR name, HMODULE* hmod)
806 {
807 static const WCHAR dotDLL[] = {'.','d','l','l',0};
808
809 WCHAR filenameW[MAX_PATH];
810 HANDLE hFile = INVALID_HANDLE_VALUE;
811 HANDLE mapping;
812 HMODULE module;
813
814 *hmod = 0;
815
816 if (SearchPathW( NULL, name, dotDLL, sizeof(filenameW) / sizeof(filenameW[0]),
817 filenameW, NULL ))
818 {
819 hFile = CreateFileW( filenameW, GENERIC_READ, FILE_SHARE_READ,
820 NULL, OPEN_EXISTING, 0, 0 );
821 }
822 if (hFile == INVALID_HANDLE_VALUE) return FALSE;
823
824 mapping = CreateFileMappingW( hFile, NULL, PAGE_READONLY, 0, 0, NULL );
825 CloseHandle( hFile );
826 if (!mapping) return FALSE;
827
828 module = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
829 CloseHandle( mapping );
830 if (!module) return FALSE;
831
832 /* make sure it's a valid PE file */
833 if (!RtlImageNtHeader(module))
834 {
835 UnmapViewOfFile( module );
836 return FALSE;
837 }
838 *hmod = (HMODULE)((char *)module + 1); /* set low bit of handle to indicate datafile module */
839 return TRUE;
840 }
841
842
843 /******************************************************************
844 * load_library
845 *
846 * Helper for LoadLibraryExA/W.
847 */
848 static HMODULE load_library( const UNICODE_STRING *libname, DWORD flags )
849 {
850 NTSTATUS nts;
851 HMODULE hModule;
852 WCHAR *load_path;
853
854 load_path = MODULE_get_dll_load_path( flags & LOAD_WITH_ALTERED_SEARCH_PATH ? libname->Buffer : NULL );
855
856 if (flags & LOAD_LIBRARY_AS_DATAFILE)
857 {
858 ULONG magic;
859
860 LdrLockLoaderLock( 0, NULL, &magic );
861 if (!(nts = LdrGetDllHandle( load_path, flags, libname, &hModule )))
862 {
863 LdrAddRefDll( 0, hModule );
864 LdrUnlockLoaderLock( 0, magic );
865 goto done;
866 }
867 LdrUnlockLoaderLock( 0, magic );
868
869 /* The method in load_library_as_datafile allows searching for the
870 * 'native' libraries only
871 */
872 if (load_library_as_datafile( libname->Buffer, &hModule )) goto done;
873 flags |= DONT_RESOLVE_DLL_REFERENCES; /* Just in case */
874 /* Fallback to normal behaviour */
875 }
876
877 nts = LdrLoadDll( load_path, flags, libname, &hModule );
878 if (nts != STATUS_SUCCESS)
879 {
880 hModule = 0;
881 SetLastError( RtlNtStatusToDosError( nts ) );
882 }
883 done:
884 HeapFree( GetProcessHeap(), 0, load_path );
885 return hModule;
886 }
887
888
889 /******************************************************************
890 * LoadLibraryExA (KERNEL32.@)
891 *
892 * Load a dll file into the process address space.
893 *
894 * PARAMS
895 * libname [I] Name of the file to load
896 * hfile [I] Reserved, must be 0.
897 * flags [I] Flags for loading the dll
898 *
899 * RETURNS
900 * Success: A handle to the loaded dll.
901 * Failure: A NULL handle. Use GetLastError() to determine the cause.
902 *
903 * NOTES
904 * The HFILE parameter is not used and marked reserved in the SDK. I can
905 * only guess that it should force a file to be mapped, but I rather
906 * ignore the parameter because it would be extremely difficult to
907 * integrate this with different types of module representations.
908 */
909 HMODULE WINAPI DECLSPEC_HOTPATCH LoadLibraryExA(LPCSTR libname, HANDLE hfile, DWORD flags)
910 {
911 WCHAR *libnameW;
912
913 if (!(libnameW = FILE_name_AtoW( libname, FALSE ))) return 0;
914 return LoadLibraryExW( libnameW, hfile, flags );
915 }
916
917 /***********************************************************************
918 * LoadLibraryExW (KERNEL32.@)
919 *
920 * Unicode version of LoadLibraryExA.
921 */
922 HMODULE WINAPI DECLSPEC_HOTPATCH LoadLibraryExW(LPCWSTR libnameW, HANDLE hfile, DWORD flags)
923 {
924 UNICODE_STRING wstr;
925 HMODULE res;
926
927 if (!libnameW)
928 {
929 SetLastError(ERROR_INVALID_PARAMETER);
930 return 0;
931 }
932 RtlInitUnicodeString( &wstr, libnameW );
933 if (wstr.Buffer[wstr.Length/sizeof(WCHAR) - 1] != ' ')
934 return load_library( &wstr, flags );
935
936 /* Library name has trailing spaces */
937 RtlCreateUnicodeString( &wstr, libnameW );
938 while (wstr.Length > sizeof(WCHAR) &&
939 wstr.Buffer[wstr.Length/sizeof(WCHAR) - 1] == ' ')
940 {
941 wstr.Length -= sizeof(WCHAR);
942 }
943 wstr.Buffer[wstr.Length/sizeof(WCHAR)] = '\0';
944 res = load_library( &wstr, flags );
945 RtlFreeUnicodeString( &wstr );
946 return res;
947 }
948
949 /***********************************************************************
950 * LoadLibraryA (KERNEL32.@)
951 *
952 * Load a dll file into the process address space.
953 *
954 * PARAMS
955 * libname [I] Name of the file to load
956 *
957 * RETURNS
958 * Success: A handle to the loaded dll.
959 * Failure: A NULL handle. Use GetLastError() to determine the cause.
960 *
961 * NOTES
962 * See LoadLibraryExA().
963 */
964 HMODULE WINAPI DECLSPEC_HOTPATCH LoadLibraryA(LPCSTR libname)
965 {
966 return LoadLibraryExA(libname, 0, 0);
967 }
968
969 /***********************************************************************
970 * LoadLibraryW (KERNEL32.@)
971 *
972 * Unicode version of LoadLibraryA.
973 */
974 HMODULE WINAPI DECLSPEC_HOTPATCH LoadLibraryW(LPCWSTR libnameW)
975 {
976 return LoadLibraryExW(libnameW, 0, 0);
977 }
978
979 /***********************************************************************
980 * FreeLibrary (KERNEL32.@)
981 *
982 * Free a dll loaded into the process address space.
983 *
984 * PARAMS
985 * hLibModule [I] Handle to the dll returned by LoadLibraryA().
986 *
987 * RETURNS
988 * Success: TRUE. The dll is removed if it is not still in use.
989 * Failure: FALSE. Use GetLastError() to determine the cause.
990 */
991 BOOL WINAPI DECLSPEC_HOTPATCH FreeLibrary(HINSTANCE hLibModule)
992 {
993 BOOL retv = FALSE;
994 NTSTATUS nts;
995
996 if (!hLibModule)
997 {
998 SetLastError( ERROR_INVALID_HANDLE );
999 return FALSE;
1000 }
1001
1002 if ((ULONG_PTR)hLibModule & 1)
1003 {
1004 /* this is a LOAD_LIBRARY_AS_DATAFILE module */
1005 char *ptr = (char *)hLibModule - 1;
1006 return UnmapViewOfFile( ptr );
1007 }
1008
1009 if ((nts = LdrUnloadDll( hLibModule )) == STATUS_SUCCESS) retv = TRUE;
1010 else SetLastError( RtlNtStatusToDosError( nts ) );
1011
1012 return retv;
1013 }
1014
1015 /***********************************************************************
1016 * GetProcAddress (KERNEL32.@)
1017 *
1018 * Find the address of an exported symbol in a loaded dll.
1019 *
1020 * PARAMS
1021 * hModule [I] Handle to the dll returned by LoadLibraryA().
1022 * function [I] Name of the symbol, or an integer ordinal number < 16384
1023 *
1024 * RETURNS
1025 * Success: A pointer to the symbol in the process address space.
1026 * Failure: NULL. Use GetLastError() to determine the cause.
1027 */
1028 FARPROC WINAPI GetProcAddress( HMODULE hModule, LPCSTR function )
1029 {
1030 NTSTATUS nts;
1031 FARPROC fp;
1032
1033 if (!hModule) hModule = NtCurrentTeb()->Peb->ImageBaseAddress;
1034
1035 if (HIWORD(function))
1036 {
1037 ANSI_STRING str;
1038
1039 RtlInitAnsiString( &str, function );
1040 nts = LdrGetProcedureAddress( hModule, &str, 0, (void**)&fp );
1041 }
1042 else
1043 nts = LdrGetProcedureAddress( hModule, NULL, LOWORD(function), (void**)&fp );
1044 if (nts != STATUS_SUCCESS)
1045 {
1046 SetLastError( RtlNtStatusToDosError( nts ) );
1047 fp = NULL;
1048 }
1049 return fp;
1050 }
1051
1052 /***********************************************************************
1053 * DelayLoadFailureHook (KERNEL32.@)
1054 */
1055 FARPROC WINAPI DelayLoadFailureHook( LPCSTR name, LPCSTR function )
1056 {
1057 ULONG_PTR args[2];
1058
1059 if ((ULONG_PTR)function >> 16)
1060 ERR( "failed to delay load %s.%s\n", name, function );
1061 else
1062 ERR( "failed to delay load %s.%u\n", name, LOWORD(function) );
1063 args[0] = (ULONG_PTR)name;
1064 args[1] = (ULONG_PTR)function;
1065 RaiseException( EXCEPTION_WINE_STUB, EH_NONCONTINUABLE, 2, args );
1066 return NULL;
1067 }
1068
1069
1070 #ifdef __i386__
1071
1072 /***********************************************************************
1073 * __wine_dll_register_16 (KERNEL32.@)
1074 *
1075 * No longer used.
1076 */
1077 void __wine_dll_register_16( const IMAGE_DOS_HEADER *header, const char *file_name )
1078 {
1079 ERR( "loading old style 16-bit dll %s no longer supported\n", file_name );
1080 }
1081
1082
1083 /***********************************************************************
1084 * __wine_dll_unregister_16 (KERNEL32.@)
1085 *
1086 * No longer used.
1087 */
1088 void __wine_dll_unregister_16( const IMAGE_DOS_HEADER *header )
1089 {
1090 }
1091
1092 #endif
1093
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.