loader/module.c

   1 /*
   2  * Modules
   3  *
   4  * Copyright 1995 Alexandre Julliard
   5  */
   6
   7 #include <assert.h>
   8 #include <fcntl.h>
   9 #include <stdlib.h>
  10 #include <stdio.h>
  11 #include <string.h>
  12 #include <sys/types.h>
  13 #include <unistd.h>
  14 #include "wine/winbase16.h"
  15 #include "wine/winestring.h"
  16 #include "winerror.h"
  17 #include "heap.h"
  18 #include "neexe.h"
  19 #include "process.h"
  20 #include "syslevel.h"
  21 #include "selectors.h"
  22 #include "debugtools.h"
  23 #include "callback.h"
  24 #include "loadorder.h"
  25 #include "elfdll.h"
  26 #include "server.h"
  27
  28 DEFAULT_DEBUG_CHANNEL(module);
  29 DECLARE_DEBUG_CHANNEL(win32);
  30
  31
  32 /*************************************************************************
  33  *              MODULE32_LookupHMODULE
  34  * looks for the referenced HMODULE in the current process
  35  * NOTE: Assumes that the process critical section is held!
  36  */
  37 static WINE_MODREF *MODULE32_LookupHMODULE( HMODULE hmod )
  38 {
  39     WINE_MODREF *wm;
  40
  41     if (!hmod)
  42         return PROCESS_Current()->exe_modref;
  43
  44     if (!HIWORD(hmod)) {
  45         ERR("tried to lookup 0x%04x in win32 module handler!\n",hmod);
  46         SetLastError( ERROR_INVALID_HANDLE );
  47         return NULL;
  48     }
  49     for ( wm = PROCESS_Current()->modref_list; wm; wm=wm->next )
  50         if (wm->module == hmod)
  51             return wm;
  52     SetLastError( ERROR_INVALID_HANDLE );
  53     return NULL;
  54 }
  55
  56 /*************************************************************************
  57  *              MODULE_AllocModRef
  58  *
  59  * Allocate a WINE_MODREF structure and add it to the process list
  60  * NOTE: Assumes that the process critical section is held!
  61  */
  62 WINE_MODREF *MODULE_AllocModRef( HMODULE hModule, LPCSTR filename )
  63 {
  64     WINE_MODREF *wm;
  65     DWORD len;
  66
  67     if ((wm = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*wm) )))
  68     {
  69         wm->module = hModule;
  70         wm->tlsindex = -1;
  71
  72         wm->filename = HEAP_strdupA( GetProcessHeap(), 0, filename );
  73         if ((wm->modname = strrchr( wm->filename, '\\' ))) wm->modname++;
  74         else wm->modname = wm->filename;
  75
  76         len = GetShortPathNameA( wm->filename, NULL, 0 );
  77         wm->short_filename = (char *)HeapAlloc( GetProcessHeap(), 0, len+1 );
  78         GetShortPathNameA( wm->filename, wm->short_filename, len+1 );
  79         if ((wm->short_modname = strrchr( wm->short_filename, '\\' ))) wm->short_modname++;
  80         else wm->short_modname = wm->short_filename;
  81
  82         wm->next = PROCESS_Current()->modref_list;
  83         if (wm->next) wm->next->prev = wm;
  84         PROCESS_Current()->modref_list = wm;
  85     }
  86     return wm;
  87 }
  88
  89 /*************************************************************************
  90  *              MODULE_InitDLL
  91  */
  92 static BOOL MODULE_InitDLL( WINE_MODREF *wm, DWORD type, LPVOID lpReserved )
  93 {
  94     BOOL retv = TRUE;
  95
  96     static LPCSTR typeName[] = { "PROCESS_DETACH", "PROCESS_ATTACH",
  97                                  "THREAD_ATTACH", "THREAD_DETACH" };
  98     assert( wm );
  99
 100     /* Skip calls for modules loaded with special load flags */
 101
 102     if (wm->flags & WINE_MODREF_DONT_RESOLVE_REFS) return TRUE;
 103
 104     TRACE("(%s,%s,%p) - CALL\n", wm->modname, typeName[type], lpReserved );
 105
 106     /* Call the initialization routine */
 107     retv = PE_InitDLL( wm->module, type, lpReserved );
 108
 109     /* The state of the module list may have changed due to the call
 110        to PE_InitDLL. We cannot assume that this module has not been
 111        deleted.  */
 112     TRACE("(%p,%s,%p) - RETURN %d\n", wm, typeName[type], lpReserved, retv );
 113
 114     return retv;
 115 }
 116
 117 /*************************************************************************
 118  *              MODULE_DllProcessAttach
 119  *
 120  * Send the process attach notification to all DLLs the given module
 121  * depends on (recursively). This is somewhat complicated due to the fact that
 122  *
 123  * - we have to respect the module dependencies, i.e. modules implicitly
 124  *   referenced by another module have to be initialized before the module
 125  *   itself can be initialized
 126  *
 127  * - the initialization routine of a DLL can itself call LoadLibrary,
 128  *   thereby introducing a whole new set of dependencies (even involving
 129  *   the 'old' modules) at any time during the whole process
 130  *
 131  * (Note that this routine can be recursively entered not only directly
 132  *  from itself, but also via LoadLibrary from one of the called initialization
 133  *  routines.)
 134  *
 135  * Furthermore, we need to rearrange the main WINE_MODREF list to allow
 136  * the process *detach* notifications to be sent in the correct order.
 137  * This must not only take into account module dependencies, but also
 138  * 'hidden' dependencies created by modules calling LoadLibrary in their
 139  * attach notification routine.
 140  *
 141  * The strategy is rather simple: we move a WINE_MODREF to the head of the
 142  * list after the attach notification has returned.  This implies that the
 143  * detach notifications are called in the reverse of the sequence the attach
 144  * notifications *returned*.
 145  *
 146  * NOTE: Assumes that the process critical section is held!
 147  *
 148  */
 149 BOOL MODULE_DllProcessAttach( WINE_MODREF *wm, LPVOID lpReserved )
 150 {
 151     BOOL retv = TRUE;
 152     int i;
 153     assert( wm );
 154
 155     /* prevent infinite recursion in case of cyclical dependencies */
 156     if (    ( wm->flags & WINE_MODREF_MARKER )
 157          || ( wm->flags & WINE_MODREF_PROCESS_ATTACHED ) )
 158         return retv;
 159
 160     TRACE("(%s,%p) - START\n", wm->modname, lpReserved );
 161
 162     /* Tag current MODREF to prevent recursive loop */
 163     wm->flags |= WINE_MODREF_MARKER;
 164
 165     /* Recursively attach all DLLs this one depends on */
 166     for ( i = 0; retv && i < wm->nDeps; i++ )
 167         if ( wm->deps[i] )
 168             retv = MODULE_DllProcessAttach( wm->deps[i], lpReserved );
 169
 170     /* Call DLL entry point */
 171     if ( retv )
 172     {
 173         retv = MODULE_InitDLL( wm, DLL_PROCESS_ATTACH, lpReserved );
 174         if ( retv )
 175             wm->flags |= WINE_MODREF_PROCESS_ATTACHED;
 176     }
 177
 178     /* Re-insert MODREF at head of list */
 179     if ( retv && wm->prev )
 180     {
 181         wm->prev->next = wm->next;
 182         if ( wm->next ) wm->next->prev = wm->prev;
 183
 184         wm->prev = NULL;
 185         wm->next = PROCESS_Current()->modref_list;
 186         PROCESS_Current()->modref_list = wm->next->prev = wm;
 187     }
 188
 189     /* Remove recursion flag */
 190     wm->flags &= ~WINE_MODREF_MARKER;
 191
 192     TRACE("(%s,%p) - END\n", wm->modname, lpReserved );
 193
 194     return retv;
 195 }
 196
 197 /*************************************************************************
 198  *              MODULE_DllProcessDetach
 199  *
 200  * Send DLL process detach notifications.  See the comment about calling
 201  * sequence at MODULE_DllProcessAttach.  Unless the bForceDetach flag
 202  * is set, only DLLs with zero refcount are notified.
 203  */
 204 void MODULE_DllProcessDetach( BOOL bForceDetach, LPVOID lpReserved )
 205 {
 206     WINE_MODREF *wm;
 207
 208     EnterCriticalSection( &PROCESS_Current()->crit_section );
 209
 210     do
 211     {
 212         for ( wm = PROCESS_Current()->modref_list; wm; wm = wm->next )
 213         {
 214             /* Check whether to detach this DLL */
 215             if ( !(wm->flags & WINE_MODREF_PROCESS_ATTACHED) )
 216                 continue;
 217             if ( wm->refCount > 0 && !bForceDetach )
 218                 continue;
 219
 220             /* Call detach notification */
 221             wm->flags &= ~WINE_MODREF_PROCESS_ATTACHED;
 222             MODULE_InitDLL( wm, DLL_PROCESS_DETACH, lpReserved );
 223
 224             /* Restart at head of WINE_MODREF list, as entries might have
 225                been added and/or removed while performing the call ... */
 226             break;
 227         }
 228     } while ( wm );
 229
 230     LeaveCriticalSection( &PROCESS_Current()->crit_section );
 231 }
 232
 233 /*************************************************************************
 234  *              MODULE_DllThreadAttach
 235  *
 236  * Send DLL thread attach notifications. These are sent in the
 237  * reverse sequence of process detach notification.
 238  *
 239  */
 240 void MODULE_DllThreadAttach( LPVOID lpReserved )
 241 {
 242     WINE_MODREF *wm;
 243
 244     EnterCriticalSection( &PROCESS_Current()->crit_section );
 245
 246     for ( wm = PROCESS_Current()->modref_list; wm; wm = wm->next )
 247         if ( !wm->next )
 248             break;
 249
 250     for ( ; wm; wm = wm->prev )
 251     {
 252         if ( !(wm->flags & WINE_MODREF_PROCESS_ATTACHED) )
 253             continue;
 254         if ( wm->flags & WINE_MODREF_NO_DLL_CALLS )
 255             continue;
 256
 257         MODULE_InitDLL( wm, DLL_THREAD_ATTACH, lpReserved );
 258     }
 259
 260     LeaveCriticalSection( &PROCESS_Current()->crit_section );
 261 }
 262
 263 /*************************************************************************
 264  *              MODULE_DllThreadDetach
 265  *
 266  * Send DLL thread detach notifications. These are sent in the
 267  * same sequence as process detach notification.
 268  *
 269  */
 270 void MODULE_DllThreadDetach( LPVOID lpReserved )
 271 {
 272     WINE_MODREF *wm;
 273
 274     EnterCriticalSection( &PROCESS_Current()->crit_section );
 275
 276     for ( wm = PROCESS_Current()->modref_list; wm; wm = wm->next )
 277     {
 278         if ( !(wm->flags & WINE_MODREF_PROCESS_ATTACHED) )
 279             continue;
 280         if ( wm->flags & WINE_MODREF_NO_DLL_CALLS )
 281             continue;
 282
 283         MODULE_InitDLL( wm, DLL_THREAD_DETACH, lpReserved );
 284     }
 285
 286     LeaveCriticalSection( &PROCESS_Current()->crit_section );
 287 }
 288
 289 /****************************************************************************
 290  *              DisableThreadLibraryCalls (KERNEL32.74)
 291  *
 292  * Don't call DllEntryPoint for DLL_THREAD_{ATTACH,DETACH} if set.
 293  */
 294 BOOL WINAPI DisableThreadLibraryCalls( HMODULE hModule )
 295 {
 296     WINE_MODREF *wm;
 297     BOOL retval = TRUE;
 298
 299     EnterCriticalSection( &PROCESS_Current()->crit_section );
 300
 301     wm = MODULE32_LookupHMODULE( hModule );
 302     if ( !wm )
 303         retval = FALSE;
 304     else
 305         wm->flags |= WINE_MODREF_NO_DLL_CALLS;
 306
 307     LeaveCriticalSection( &PROCESS_Current()->crit_section );
 308
 309     return retval;
 310 }
 311
 312
 313 /***********************************************************************
 314  *           MODULE_CreateDummyModule
 315  *
 316  * Create a dummy NE module for Win32 or Winelib.
 317  */
 318 HMODULE MODULE_CreateDummyModule( LPCSTR filename, HMODULE module32 )
 319 {
 320     HMODULE hModule;
 321     NE_MODULE *pModule;
 322     SEGTABLEENTRY *pSegment;
 323     char *pStr,*s;
 324     unsigned int len;
 325     const char* basename;
 326     OFSTRUCT *ofs;
 327     int of_size, size;
 328
 329     /* Extract base filename */
 330     basename = strrchr(filename, '\\');
 331     if (!basename) basename = filename;
 332     else basename++;
 333     len = strlen(basename);
 334     if ((s = strchr(basename, '.'))) len = s - basename;
 335
 336     /* Allocate module */
 337     of_size = sizeof(OFSTRUCT) - sizeof(ofs->szPathName)
 338                     + strlen(filename) + 1;
 339     size = sizeof(NE_MODULE) +
 340                  /* loaded file info */
 341                  of_size +
 342                  /* segment table: DS,CS */
 343                  2 * sizeof(SEGTABLEENTRY) +
 344                  /* name table */
 345                  len + 2 +
 346                  /* several empty tables */
 347                  8;
 348
 349     hModule = GlobalAlloc16( GMEM_MOVEABLE | GMEM_ZEROINIT, size );
 350     if (!hModule) return (HMODULE)11;  /* invalid exe */
 351
 352     FarSetOwner16( hModule, hModule );
 353     pModule = (NE_MODULE *)GlobalLock16( hModule );
 354
 355     /* Set all used entries */
 356     pModule->magic            = IMAGE_OS2_SIGNATURE;
 357     pModule->count            = 1;
 358     pModule->next             = 0;
 359     pModule->flags            = 0;
 360     pModule->dgroup           = 0;
 361     pModule->ss               = 1;
 362     pModule->cs               = 2;
 363     pModule->heap_size        = 0;
 364     pModule->stack_size       = 0;
 365     pModule->seg_count        = 2;
 366     pModule->modref_count     = 0;
 367     pModule->nrname_size      = 0;
 368     pModule->fileinfo         = sizeof(NE_MODULE);
 369     pModule->os_flags         = NE_OSFLAGS_WINDOWS;
 370     pModule->self             = hModule;
 371     pModule->module32         = module32;
 372
 373     /* Set version and flags */
 374     if (module32)
 375     {
 376         pModule->expected_version =
 377             ((PE_HEADER(module32)->OptionalHeader.MajorSubsystemVersion & 0xff) << 8 ) |
 378              (PE_HEADER(module32)->OptionalHeader.MinorSubsystemVersion & 0xff);
 379         pModule->flags |= NE_FFLAGS_WIN32;
 380         if (PE_HEADER(module32)->FileHeader.Characteristics & IMAGE_FILE_DLL)
 381             pModule->flags |= NE_FFLAGS_LIBMODULE | NE_FFLAGS_SINGLEDATA;
 382     }
 383
 384     /* Set loaded file information */
 385     ofs = (OFSTRUCT *)(pModule + 1);
 386     memset( ofs, 0, of_size );
 387     ofs->cBytes = of_size < 256 ? of_size : 255;   /* FIXME */
 388     strcpy( ofs->szPathName, filename );
 389
 390     pSegment = (SEGTABLEENTRY*)((char*)(pModule + 1) + of_size);
 391     pModule->seg_table = (int)pSegment - (int)pModule;
 392     /* Data segment */
 393     pSegment->size    = 0;
 394     pSegment->flags   = NE_SEGFLAGS_DATA;
 395     pSegment->minsize = 0x1000;
 396     pSegment++;
 397     /* Code segment */
 398     pSegment->flags   = 0;
 399     pSegment++;
 400
 401     /* Module name */
 402     pStr = (char *)pSegment;
 403     pModule->name_table = (int)pStr - (int)pModule;
 404     assert(len<256);
 405     *pStr = len;
 406     lstrcpynA( pStr+1, basename, len+1 );
 407     pStr += len+2;
 408
 409     /* All tables zero terminated */
 410     pModule->res_table = pModule->import_table = pModule->entry_table =
 411                 (int)pStr - (int)pModule;
 412
 413     NE_RegisterModule( pModule );
 414     return hModule;
 415 }
 416
 417
 418 /**********************************************************************
 419  *          MODULE_FindModule
 420  *
 421  * Find a (loaded) win32 module depending on path
 422  *
 423  * RETURNS
 424  *      the module handle if found
 425  *      0 if not
 426  */
 427 WINE_MODREF *MODULE_FindModule(
 428         LPCSTR path     /* [in] pathname of module/library to be found */
 429 ) {
 430     WINE_MODREF *wm;
 431     char dllname[260], *p;
 432
 433     /* Append .DLL to name if no extension present */
 434     strcpy( dllname, path );
 435     if (!(p = strrchr( dllname, '.')) || strchr( p, '/' ) || strchr( p, '\\'))
 436             strcat( dllname, ".DLL" );
 437
 438     for ( wm = PROCESS_Current()->modref_list; wm; wm = wm->next )
 439     {
 440         if ( !strcasecmp( dllname, wm->modname ) )
 441             break;
 442         if ( !strcasecmp( dllname, wm->filename ) )
 443             break;
 444         if ( !strcasecmp( dllname, wm->short_modname ) )
 445             break;
 446         if ( !strcasecmp( dllname, wm->short_filename ) )
 447             break;
 448     }
 449
 450     return wm;
 451 }
 452
 453
 454 /* Check whether a file is an OS/2 or a very old Windows executable
 455  * by testing on import of KERNEL.
 456  *
 457  * FIXME: is reading the module imports the only way of discerning
 458  *        old Windows binaries from OS/2 ones ? At least it seems so...
 459  */
 460 static DWORD MODULE_Decide_OS2_OldWin(HANDLE hfile, IMAGE_DOS_HEADER *mz, IMAGE_OS2_HEADER *ne)
 461 {
 462     DWORD currpos = SetFilePointer( hfile, 0, NULL, SEEK_CUR);
 463     DWORD type = SCS_OS216_BINARY;
 464     LPWORD modtab = NULL;
 465     LPSTR nametab = NULL;
 466     DWORD len;
 467     int i;
 468
 469     /* read modref table */
 470     if ( (SetFilePointer( hfile, mz->e_lfanew + ne->ne_modtab, NULL, SEEK_SET ) == -1)
 471       || (!(modtab = HeapAlloc( GetProcessHeap(), 0, ne->ne_cmod*sizeof(WORD))))
 472       || (!(ReadFile(hfile, modtab, ne->ne_cmod*sizeof(WORD), &len, NULL)))
 473       || (len != ne->ne_cmod*sizeof(WORD)) )
 474         goto broken;
 475
 476     /* read imported names table */
 477     if ( (SetFilePointer( hfile, mz->e_lfanew + ne->ne_imptab, NULL, SEEK_SET ) == -1)
 478       || (!(nametab = HeapAlloc( GetProcessHeap(), 0, ne->ne_enttab - ne->ne_imptab)))
 479       || (!(ReadFile(hfile, nametab, ne->ne_enttab - ne->ne_imptab, &len, NULL)))
 480       || (len != ne->ne_enttab - ne->ne_imptab) )
 481         goto broken;
 482
 483     for (i=0; i < ne->ne_cmod; i++)
 484     {
 485         LPSTR module = &nametab[modtab[i]];
 486         TRACE("modref: %.*s\n", module[0], &module[1]);
 487         if (!(strncmp(&module[1], "KERNEL", module[0])))
 488         { /* very old Windows file */
 489             MESSAGE("This seems to be a very old (pre-3.0) Windows executable. Expect crashes, especially if this is a real-mode binary !\n");
 490             type = SCS_WOW_BINARY;
 491             goto good;
 492         }
 493     }
 494
 495 broken:
 496     ERR("Hmm, an error occurred. Is this binary file broken ?\n");
 497
 498 good:
 499     HeapFree( GetProcessHeap(), 0, modtab);
 500     HeapFree( GetProcessHeap(), 0, nametab);
 501     SetFilePointer( hfile, currpos, NULL, SEEK_SET); /* restore filepos */
 502     return type;
 503 }
 504
 505 /***********************************************************************
 506  *           MODULE_GetBinaryType
 507  *
 508  * The GetBinaryType function determines whether a file is executable
 509  * or not and if it is it returns what type of executable it is.
 510  * The type of executable is a property that determines in which
 511  * subsystem an executable file runs under.
 512  *
 513  * Binary types returned:
 514  * SCS_32BIT_BINARY: A Win32 based application
 515  * SCS_DOS_BINARY: An MS-Dos based application
 516  * SCS_WOW_BINARY: A Win16 based application
 517  * SCS_PIF_BINARY: A PIF file that executes an MS-Dos based app
 518  * SCS_POSIX_BINARY: A POSIX based application ( Not implemented )
 519  * SCS_OS216_BINARY: A 16bit OS/2 based application
 520  *
 521  * Returns TRUE if the file is an executable in which case
 522  * the value pointed by lpBinaryType is set.
 523  * Returns FALSE if the file is not an executable or if the function fails.
 524  *
 525  * To do so it opens the file and reads in the header information
 526  * if the extended header information is not present it will
 527  * assume that the file is a DOS executable.
 528  * If the extended header information is present it will
 529  * determine if the file is a 16 or 32 bit Windows executable
 530  * by check the flags in the header.
 531  *
 532  * Note that .COM and .PIF files are only recognized by their
 533  * file name extension; but Windows does it the same way ...
 534  */
 535 BOOL MODULE_GetBinaryType( HANDLE hfile, LPCSTR filename, LPDWORD lpBinaryType )
 536 {
 537     IMAGE_DOS_HEADER mz_header;
 538     char magic[4], *ptr;
 539     DWORD len;
 540
 541     /* Seek to the start of the file and read the DOS header information.
 542      */
 543     if (    SetFilePointer( hfile, 0, NULL, SEEK_SET ) != -1
 544          && ReadFile( hfile, &mz_header, sizeof(mz_header), &len, NULL )
 545          && len == sizeof(mz_header) )
 546     {
 547         /* Now that we have the header check the e_magic field
 548          * to see if this is a dos image.
 549          */
 550         if ( mz_header.e_magic == IMAGE_DOS_SIGNATURE )
 551         {
 552             BOOL lfanewValid = FALSE;
 553             /* We do have a DOS image so we will now try to seek into
 554              * the file by the amount indicated by the field
 555              * "Offset to extended header" and read in the
 556              * "magic" field information at that location.
 557              * This will tell us if there is more header information
 558              * to read or not.
 559              */
 560             /* But before we do we will make sure that header
 561              * structure encompasses the "Offset to extended header"
 562              * field.
 563              */
 564             if ( (mz_header.e_cparhdr<<4) >= sizeof(IMAGE_DOS_HEADER) )
 565                 if ( ( mz_header.e_crlc == 0 ) ||
 566                      ( mz_header.e_lfarlc >= sizeof(IMAGE_DOS_HEADER) ) )
 567                     if (    mz_header.e_lfanew >= sizeof(IMAGE_DOS_HEADER)
 568                          && SetFilePointer( hfile, mz_header.e_lfanew, NULL, SEEK_SET ) != -1
 569                          && ReadFile( hfile, magic, sizeof(magic), &len, NULL )
 570                          && len == sizeof(magic) )
 571                         lfanewValid = TRUE;
 572
 573             if ( !lfanewValid )
 574             {
 575                 /* If we cannot read this "extended header" we will
 576                  * assume that we have a simple DOS executable.
 577                  */
 578                 *lpBinaryType = SCS_DOS_BINARY;
 579                 return TRUE;
 580             }
 581             else
 582             {
 583                 /* Reading the magic field succeeded so
 584                  * we will try to determine what type it is.
 585                  */
 586                 if ( *(DWORD*)magic      == IMAGE_NT_SIGNATURE )
 587                 {
 588                     /* This is an NT signature.
 589                      */
 590                     *lpBinaryType = SCS_32BIT_BINARY;
 591                     return TRUE;
 592                 }
 593                 else if ( *(WORD*)magic == IMAGE_OS2_SIGNATURE )
 594                 {
 595                     /* The IMAGE_OS2_SIGNATURE indicates that the
 596                      * "extended header is a Windows executable (NE)
 597                      * header."  This can mean either a 16-bit OS/2
 598                      * or a 16-bit Windows or even a DOS program
 599                      * (running under a DOS extender).  To decide
 600                      * which, we'll have to read the NE header.
 601                      */
 602
 603                      IMAGE_OS2_HEADER ne;
 604                      if (    SetFilePointer( hfile, mz_header.e_lfanew, NULL, SEEK_SET ) != -1
 605                           && ReadFile( hfile, &ne, sizeof(ne), &len, NULL )
 606                           && len == sizeof(ne) )
 607                      {
 608                          switch ( ne.ne_exetyp )
 609                          {
 610                          case 2:  *lpBinaryType = SCS_WOW_BINARY;   return TRUE;
 611                          case 5:  *lpBinaryType = SCS_DOS_BINARY;   return TRUE;
 612                          default: *lpBinaryType =
 613                                   MODULE_Decide_OS2_OldWin(hfile, &mz_header, &ne);
 614                                   return TRUE;
 615                          }
 616                      }
 617                      /* Couldn't read header, so abort. */
 618                      return FALSE;
 619                 }
 620                 else
 621                 {
 622                     /* Unknown extended header, but this file is nonetheless
 623                        DOS-executable.
 624                      */
 625                     *lpBinaryType = SCS_DOS_BINARY;
 626                     return TRUE;
 627                 }
 628             }
 629         }
 630     }
 631
 632     /* If we get here, we don't even have a correct MZ header.
 633      * Try to check the file extension for known types ...
 634      */
 635     ptr = strrchr( filename, '.' );
 636     if ( ptr && !strchr( ptr, '\\' ) && !strchr( ptr, '/' ) )
 637     {
 638         if ( !strcasecmp( ptr, ".COM" ) )
 639         {
 640             *lpBinaryType = SCS_DOS_BINARY;
 641             return TRUE;
 642         }
 643
 644         if ( !strcasecmp( ptr, ".PIF" ) )
 645         {
 646             *lpBinaryType = SCS_PIF_BINARY;
 647             return TRUE;
 648         }
 649     }
 650
 651     return FALSE;
 652 }
 653
 654 /***********************************************************************
 655  *             GetBinaryTypeA                     [KERNEL32.280]
 656  */
 657 BOOL WINAPI GetBinaryTypeA( LPCSTR lpApplicationName, LPDWORD lpBinaryType )
 658 {
 659     BOOL ret = FALSE;
 660     HANDLE hfile;
 661
 662     TRACE_(win32)("%s\n", lpApplicationName );
 663
 664     /* Sanity check.
 665      */
 666     if ( lpApplicationName == NULL || lpBinaryType == NULL )
 667         return FALSE;
 668
 669     /* Open the file indicated by lpApplicationName for reading.
 670      */
 671     hfile = CreateFileA( lpApplicationName, GENERIC_READ, FILE_SHARE_READ,
 672                          NULL, OPEN_EXISTING, 0, -1 );
 673     if ( hfile == INVALID_HANDLE_VALUE )
 674         return FALSE;
 675
 676     /* Check binary type
 677      */
 678     ret = MODULE_GetBinaryType( hfile, lpApplicationName, lpBinaryType );
 679
 680     /* Close the file.
 681      */
 682     CloseHandle( hfile );
 683
 684     return ret;
 685 }
 686
 687 /***********************************************************************
 688  *             GetBinaryTypeW                      [KERNEL32.281]
 689  */
 690 BOOL WINAPI GetBinaryTypeW( LPCWSTR lpApplicationName, LPDWORD lpBinaryType )
 691 {
 692     BOOL ret = FALSE;
 693     LPSTR strNew = NULL;
 694
 695     TRACE_(win32)("%s\n", debugstr_w(lpApplicationName) );
 696
 697     /* Sanity check.
 698      */
 699     if ( lpApplicationName == NULL || lpBinaryType == NULL )
 700         return FALSE;
 701
 702     /* Convert the wide string to a ascii string.
 703      */
 704     strNew = HEAP_strdupWtoA( GetProcessHeap(), 0, lpApplicationName );
 705
 706     if ( strNew != NULL )
 707     {
 708         ret = GetBinaryTypeA( strNew, lpBinaryType );
 709
 710         /* Free the allocated string.
 711          */
 712         HeapFree( GetProcessHeap(), 0, strNew );
 713     }
 714
 715     return ret;
 716 }
 717
 718
 719 /***********************************************************************
 720  *           WinExec16   (KERNEL.166)
 721  */
 722 HINSTANCE16 WINAPI WinExec16( LPCSTR lpCmdLine, UINT16 nCmdShow )
 723 {
 724     LPCSTR p, args = NULL;
 725     LPCSTR name_beg, name_end;
 726     LPSTR name, cmdline;
 727     int arglen;
 728     HINSTANCE16 ret;
 729     char buffer[MAX_PATH];
 730
 731     if (*lpCmdLine == '"') /* has to be only one and only at beginning ! */
 732     {
 733       name_beg = lpCmdLine+1;
 734       p = strchr ( lpCmdLine+1, '"' );
 735       if (p)
 736       {
 737           name_end = p;
 738           args = strchr ( p, ' ' );
 739       }
 740       else /* yes, even valid with trailing '"' missing */
 741           name_end = lpCmdLine+strlen(lpCmdLine);
 742     }
 743     else
 744     {
 745       name_beg = lpCmdLine;
 746       args = strchr( lpCmdLine, ' ' );
 747       name_end = args ? args : lpCmdLine+strlen(lpCmdLine);
 748     }
 749
 750     if ((name_beg == lpCmdLine) && (!args))
 751     { /* just use the original cmdline string as file name */
 752         name = (LPSTR)lpCmdLine;
 753     }
 754     else
 755     {
 756         if (!(name = HeapAlloc( GetProcessHeap(), 0, name_end - name_beg + 1 )))
 757             return ERROR_NOT_ENOUGH_MEMORY;
 758         memcpy( name, name_beg, name_end - name_beg );
 759         name[name_end - name_beg] = '\0';
 760     }
 761
 762     if (args)
 763     {
 764         args++;
 765         arglen = strlen(args);
 766         cmdline = SEGPTR_ALLOC( 2 + arglen );
 767         cmdline[0] = (BYTE)arglen;
 768         strcpy( cmdline + 1, args );
 769     }
 770     else
 771     {
 772         cmdline = SEGPTR_ALLOC( 2 );
 773         cmdline[0] = cmdline[1] = 0;
 774     }
 775
 776     TRACE("name: '%s', cmdline: '%.*s'\n", name, cmdline[0], &cmdline[1]);
 777
 778     if (SearchPathA( NULL, name, ".exe", sizeof(buffer), buffer, NULL ))
 779     {
 780         LOADPARAMS16 params;
 781         WORD *showCmd = SEGPTR_ALLOC( 2*sizeof(WORD) );
 782         showCmd[0] = 2;
 783         showCmd[1] = nCmdShow;
 784
 785         params.hEnvironment = 0;
 786         params.cmdLine = SEGPTR_GET(cmdline);
 787         params.showCmd = SEGPTR_GET(showCmd);
 788         params.reserved = 0;
 789
 790         ret = LoadModule16( buffer, &params );
 791
 792         SEGPTR_FREE( showCmd );
 793         SEGPTR_FREE( cmdline );
 794     }
 795     else ret = GetLastError();
 796
 797     if (name != lpCmdLine) HeapFree( GetProcessHeap(), 0, name );
 798
 799     if (ret == 21)  /* 32-bit module */
 800     {
 801         SYSLEVEL_ReleaseWin16Lock();
 802         ret = WinExec( lpCmdLine, nCmdShow );
 803         SYSLEVEL_RestoreWin16Lock();
 804     }
 805     return ret;
 806 }
 807
 808 /***********************************************************************
 809  *           WinExec   (KERNEL32.566)
 810  */
 811 HINSTANCE WINAPI WinExec( LPCSTR lpCmdLine, UINT nCmdShow )
 812 {
 813     PROCESS_INFORMATION info;
 814     STARTUPINFOA startup;
 815     HINSTANCE hInstance;
 816     char *cmdline;
 817
 818     memset( &startup, 0, sizeof(startup) );
 819     startup.cb = sizeof(startup);
 820     startup.dwFlags = STARTF_USESHOWWINDOW;
 821     startup.wShowWindow = nCmdShow;
 822
 823     /* cmdline needs to be writeable for CreateProcess */
 824     if (!(cmdline = HEAP_strdupA( GetProcessHeap(), 0, lpCmdLine ))) return 0;
 825
 826     if (CreateProcessA( NULL, cmdline, NULL, NULL, FALSE,
 827                         0, NULL, NULL, &startup, &info ))
 828     {
 829         /* Give 30 seconds to the app to come up */
 830         if (Callout.WaitForInputIdle ( info.hProcess, 30000 ) == 0xFFFFFFFF)
 831             WARN("WaitForInputIdle failed: Error %ld\n", GetLastError() );
 832         hInstance = 33;
 833         /* Close off the handles */
 834         CloseHandle( info.hThread );
 835         CloseHandle( info.hProcess );
 836     }
 837     else if ((hInstance = GetLastError()) >= 32)
 838     {
 839         FIXME("Strange error set by CreateProcess: %d\n", hInstance );
 840         hInstance = 11;
 841     }
 842     HeapFree( GetProcessHeap(), 0, cmdline );
 843     return hInstance;
 844 }
 845
 846 /**********************************************************************
 847  *          LoadModule    (KERNEL32.499)
 848  */
 849 HINSTANCE WINAPI LoadModule( LPCSTR name, LPVOID paramBlock )
 850 {
 851     LOADPARAMS *params = (LOADPARAMS *)paramBlock;
 852     PROCESS_INFORMATION info;
 853     STARTUPINFOA startup;
 854     HINSTANCE hInstance;
 855     LPSTR cmdline, p;
 856     char filename[MAX_PATH];
 857     BYTE len;
 858
 859     if (!name) return ERROR_FILE_NOT_FOUND;
 860
 861     if (!SearchPathA( NULL, name, ".exe", sizeof(filename), filename, NULL ) &&
 862         !SearchPathA( NULL, name, NULL, sizeof(filename), filename, NULL ))
 863         return GetLastError();
 864
 865     len = (BYTE)params->lpCmdLine[0];
 866     if (!(cmdline = HeapAlloc( GetProcessHeap(), 0, strlen(filename) + len + 2 )))
 867         return ERROR_NOT_ENOUGH_MEMORY;
 868
 869     strcpy( cmdline, filename );
 870     p = cmdline + strlen(cmdline);
 871     *p++ = ' ';
 872     memcpy( p, params->lpCmdLine + 1, len );
 873     p[len] = 0;
 874
 875     memset( &startup, 0, sizeof(startup) );
 876     startup.cb = sizeof(startup);
 877     if (params->lpCmdShow)
 878     {
 879         startup.dwFlags = STARTF_USESHOWWINDOW;
 880         startup.wShowWindow = params->lpCmdShow[1];
 881     }
 882
 883     if (CreateProcessA( filename, cmdline, NULL, NULL, FALSE, 0,
 884                         params->lpEnvAddress, NULL, &startup, &info ))
 885     {
 886         /* Give 30 seconds to the app to come up */
 887         if ( Callout.WaitForInputIdle ( info.hProcess, 30000 ) ==  0xFFFFFFFF )
 888             WARN("WaitForInputIdle failed: Error %ld\n", GetLastError() );
 889         hInstance = 33;
 890         /* Close off the handles */
 891         CloseHandle( info.hThread );
 892         CloseHandle( info.hProcess );
 893     }
 894     else if ((hInstance = GetLastError()) >= 32)
 895     {
 896         FIXME("Strange error set by CreateProcess: %d\n", hInstance );
 897         hInstance = 11;
 898     }
 899
 900     HeapFree( GetProcessHeap(), 0, cmdline );
 901     return hInstance;
 902 }
 903
 904
 905 /*************************************************************************
 906  *               get_file_name
 907  *
 908  * Helper for CreateProcess: retrieve the file name to load from the
 909  * app name and command line. Store the file name in buffer, and
 910  * return a possibly modified command line.
 911  */
 912 static LPSTR get_file_name( LPCSTR appname, LPSTR cmdline, LPSTR buffer, int buflen )
 913 {
 914     char *name, *pos, *ret = NULL;
 915     const char *p;
 916
 917     /* if we have an app name, everything is easy */
 918
 919     if (appname)
 920     {
 921         /* use the unmodified app name as file name */
 922         lstrcpynA( buffer, appname, buflen );
 923         if (!(ret = cmdline))
 924         {
 925             /* no command-line, create one */
 926             if ((ret = HeapAlloc( GetProcessHeap(), 0, strlen(appname) + 3 )))
 927                 sprintf( ret, "\"%s\"", appname );
 928         }
 929         return ret;
 930     }
 931
 932     if (!cmdline)
 933     {
 934         SetLastError( ERROR_INVALID_PARAMETER );
 935         return NULL;
 936     }
 937
 938     /* first check for a quoted file name */
 939
 940     if ((cmdline[0] == '"') && ((p = strchr( cmdline + 1, '"' ))))
 941     {
 942         int len = p - cmdline - 1;
 943         /* extract the quoted portion as file name */
 944         if (!(name = HeapAlloc( GetProcessHeap(), 0, len + 1 ))) return NULL;
 945         memcpy( name, cmdline + 1, len );
 946         name[len] = 0;
 947
 948         if (SearchPathA( NULL, name, ".exe", buflen, buffer, NULL ) ||
 949             SearchPathA( NULL, name, NULL, buflen, buffer, NULL ))
 950             ret = cmdline;  /* no change necessary */
 951         goto done;
 952     }
 953
 954     /* now try the command-line word by word */
 955
 956     if (!(name = HeapAlloc( GetProcessHeap(), 0, strlen(cmdline) + 1 ))) return NULL;
 957     pos = name;
 958     p = cmdline;
 959
 960     while (*p)
 961     {
 962         do *pos++ = *p++; while (*p && *p != ' ');
 963         *pos = 0;
 964         TRACE("trying '%s'\n", name );
 965         if (SearchPathA( NULL, name, ".exe", buflen, buffer, NULL ) ||
 966             SearchPathA( NULL, name, NULL, buflen, buffer, NULL ))
 967         {
 968             ret = cmdline;
 969             break;
 970         }
 971     }
 972
 973     if (!ret || !strchr( name, ' ' )) goto done;  /* no change necessary */
 974
 975     /* now build a new command-line with quotes */
 976
 977     if (!(ret = HeapAlloc( GetProcessHeap(), 0, strlen(cmdline) + 3 ))) goto done;
 978     sprintf( ret, "\"%s\"%s", name, p );
 979
 980  done:
 981     HeapFree( GetProcessHeap(), 0, name );
 982     return ret;
 983 }
 984
 985
 986 /**********************************************************************
 987  *       CreateProcessA          (KERNEL32.171)
 988  */
 989 BOOL WINAPI CreateProcessA( LPCSTR lpApplicationName, LPSTR lpCommandLine,
 990                             LPSECURITY_ATTRIBUTES lpProcessAttributes,
 991                             LPSECURITY_ATTRIBUTES lpThreadAttributes,
 992                             BOOL bInheritHandles, DWORD dwCreationFlags,
 993                             LPVOID lpEnvironment, LPCSTR lpCurrentDirectory,
 994                             LPSTARTUPINFOA lpStartupInfo,
 995                             LPPROCESS_INFORMATION lpProcessInfo )
 996 {
 997     BOOL retv = FALSE;
 998     HANDLE hFile;
 999     DWORD type;
1000     char name[MAX_PATH];
1001     LPSTR tidy_cmdline;
1002
1003     /* Process the AppName and/or CmdLine to get module name and path */
1004
1005     TRACE("app '%s' cmdline '%s'\n", lpApplicationName, lpCommandLine );
1006
1007     if (!(tidy_cmdline = get_file_name( lpApplicationName, lpCommandLine, name, sizeof(name) )))
1008         return FALSE;
1009
1010     /* Warn if unsupported features are used */
1011
1012     if (dwCreationFlags & DETACHED_PROCESS)
1013         FIXME("(%s,...): DETACHED_PROCESS ignored\n", name);
1014     if (dwCreationFlags & CREATE_NEW_CONSOLE)
1015         FIXME("(%s,...): CREATE_NEW_CONSOLE ignored\n", name);
1016     if (dwCreationFlags & NORMAL_PRIORITY_CLASS)
1017         FIXME("(%s,...): NORMAL_PRIORITY_CLASS ignored\n", name);
1018     if (dwCreationFlags & IDLE_PRIORITY_CLASS)
1019         FIXME("(%s,...): IDLE_PRIORITY_CLASS ignored\n", name);
1020     if (dwCreationFlags & HIGH_PRIORITY_CLASS)
1021         FIXME("(%s,...): HIGH_PRIORITY_CLASS ignored\n", name);
1022     if (dwCreationFlags & REALTIME_PRIORITY_CLASS)
1023         FIXME("(%s,...): REALTIME_PRIORITY_CLASS ignored\n", name);
1024     if (dwCreationFlags & CREATE_NEW_PROCESS_GROUP)
1025         FIXME("(%s,...): CREATE_NEW_PROCESS_GROUP ignored\n", name);
1026     if (dwCreationFlags & CREATE_UNICODE_ENVIRONMENT)
1027         FIXME("(%s,...): CREATE_UNICODE_ENVIRONMENT ignored\n", name);
1028     if (dwCreationFlags & CREATE_SEPARATE_WOW_VDM)
1029         FIXME("(%s,...): CREATE_SEPARATE_WOW_VDM ignored\n", name);
1030     if (dwCreationFlags & CREATE_SHARED_WOW_VDM)
1031         FIXME("(%s,...): CREATE_SHARED_WOW_VDM ignored\n", name);
1032     if (dwCreationFlags & CREATE_DEFAULT_ERROR_MODE)
1033         FIXME("(%s,...): CREATE_DEFAULT_ERROR_MODE ignored\n", name);
1034     if (dwCreationFlags & CREATE_NO_WINDOW)
1035         FIXME("(%s,...): CREATE_NO_WINDOW ignored\n", name);
1036     if (dwCreationFlags & PROFILE_USER)
1037         FIXME("(%s,...): PROFILE_USER ignored\n", name);
1038     if (dwCreationFlags & PROFILE_KERNEL)
1039         FIXME("(%s,...): PROFILE_KERNEL ignored\n", name);
1040     if (dwCreationFlags & PROFILE_SERVER)
1041         FIXME("(%s,...): PROFILE_SERVER ignored\n", name);
1042     if (lpStartupInfo->lpDesktop)
1043         FIXME("(%s,...): lpStartupInfo->lpDesktop %s ignored\n",
1044                       name, lpStartupInfo->lpDesktop);
1045     if (lpStartupInfo->lpTitle)
1046         FIXME("(%s,...): lpStartupInfo->lpTitle %s ignored\n",
1047                       name, lpStartupInfo->lpTitle);
1048     if (lpStartupInfo->dwFlags & STARTF_USECOUNTCHARS)
1049         FIXME("(%s,...): STARTF_USECOUNTCHARS (%ld,%ld) ignored\n",
1050                       name, lpStartupInfo->dwXCountChars, lpStartupInfo->dwYCountChars);
1051     if (lpStartupInfo->dwFlags & STARTF_USEFILLATTRIBUTE)
1052         FIXME("(%s,...): STARTF_USEFILLATTRIBUTE %lx ignored\n",
1053                       name, lpStartupInfo->dwFillAttribute);
1054     if (lpStartupInfo->dwFlags & STARTF_RUNFULLSCREEN)
1055         FIXME("(%s,...): STARTF_RUNFULLSCREEN ignored\n", name);
1056     if (lpStartupInfo->dwFlags & STARTF_FORCEONFEEDBACK)
1057         FIXME("(%s,...): STARTF_FORCEONFEEDBACK ignored\n", name);
1058     if (lpStartupInfo->dwFlags & STARTF_FORCEOFFFEEDBACK)
1059         FIXME("(%s,...): STARTF_FORCEOFFFEEDBACK ignored\n", name);
1060     if (lpStartupInfo->dwFlags & STARTF_USEHOTKEY)
1061         FIXME("(%s,...): STARTF_USEHOTKEY ignored\n", name);
1062
1063     /* Open file and determine executable type */
1064
1065     hFile = CreateFileA( name, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, -1 );
1066     if (hFile == INVALID_HANDLE_VALUE) goto done;
1067
1068     if ( !MODULE_GetBinaryType( hFile, name, &type ) )
1069     {
1070         CloseHandle( hFile );
1071         retv = PROCESS_Create( -1, name, tidy_cmdline, lpEnvironment,
1072                                lpProcessAttributes, lpThreadAttributes,
1073                                bInheritHandles, dwCreationFlags,
1074                                lpStartupInfo, lpProcessInfo, lpCurrentDirectory );
1075         goto done;
1076     }
1077
1078     /* Create process */
1079
1080     switch ( type )
1081     {
1082     case SCS_32BIT_BINARY:
1083     case SCS_WOW_BINARY:
1084     case SCS_DOS_BINARY:
1085         retv = PROCESS_Create( hFile, name, tidy_cmdline, lpEnvironment,
1086                                lpProcessAttributes, lpThreadAttributes,
1087                                bInheritHandles, dwCreationFlags,
1088                                lpStartupInfo, lpProcessInfo, lpCurrentDirectory);
1089         break;
1090
1091     case SCS_PIF_BINARY:
1092     case SCS_POSIX_BINARY:
1093     case SCS_OS216_BINARY:
1094         FIXME("Unsupported executable type: %ld\n", type );
1095         /* fall through */
1096
1097     default:
1098         SetLastError( ERROR_BAD_FORMAT );
1099         break;
1100     }
1101     CloseHandle( hFile );
1102
1103  done:
1104     if (tidy_cmdline != lpCommandLine) HeapFree( GetProcessHeap(), 0, tidy_cmdline );
1105     return retv;
1106 }
1107
1108 /**********************************************************************
1109  *       CreateProcessW          (KERNEL32.172)
1110  * NOTES
1111  *  lpReserved is not converted
1112  */
1113 BOOL WINAPI CreateProcessW( LPCWSTR lpApplicationName, LPWSTR lpCommandLine,
1114                                 LPSECURITY_ATTRIBUTES lpProcessAttributes,
1115                                 LPSECURITY_ATTRIBUTES lpThreadAttributes,
1116                                 BOOL bInheritHandles, DWORD dwCreationFlags,
1117                                 LPVOID lpEnvironment, LPCWSTR lpCurrentDirectory,
1118                                 LPSTARTUPINFOW lpStartupInfo,
1119                                 LPPROCESS_INFORMATION lpProcessInfo )
1120 {   BOOL ret;
1121     STARTUPINFOA StartupInfoA;
1122
1123     LPSTR lpApplicationNameA = HEAP_strdupWtoA (GetProcessHeap(),0,lpApplicationName);
1124     LPSTR lpCommandLineA = HEAP_strdupWtoA (GetProcessHeap(),0,lpCommandLine);
1125     LPSTR lpCurrentDirectoryA = HEAP_strdupWtoA (GetProcessHeap(),0,lpCurrentDirectory);
1126
1127     memcpy (&StartupInfoA, lpStartupInfo, sizeof(STARTUPINFOA));
1128     StartupInfoA.lpDesktop = HEAP_strdupWtoA (GetProcessHeap(),0,lpStartupInfo->lpDesktop);
1129     StartupInfoA.lpTitle = HEAP_strdupWtoA (GetProcessHeap(),0,lpStartupInfo->lpTitle);
1130
1131     TRACE_(win32)("(%s,%s,...)\n", debugstr_w(lpApplicationName), debugstr_w(lpCommandLine));
1132
1133     if (lpStartupInfo->lpReserved)
1134       FIXME_(win32)("StartupInfo.lpReserved is used, please report (%s)\n", debugstr_w(lpStartupInfo->lpReserved));
1135
1136     ret = CreateProcessA(  lpApplicationNameA,  lpCommandLineA,
1137                              lpProcessAttributes, lpThreadAttributes,
1138                              bInheritHandles, dwCreationFlags,
1139                              lpEnvironment, lpCurrentDirectoryA,
1140                              &StartupInfoA, lpProcessInfo );
1141
1142     HeapFree( GetProcessHeap(), 0, lpCurrentDirectoryA );
1143     HeapFree( GetProcessHeap(), 0, lpCommandLineA );
1144     HeapFree( GetProcessHeap(), 0, StartupInfoA.lpDesktop );
1145     HeapFree( GetProcessHeap(), 0, StartupInfoA.lpTitle );
1146
1147     return ret;
1148 }
1149
1150 /***********************************************************************
1151  *              GetModuleHandleA         (KERNEL32.237)
1152  */
1153 HMODULE WINAPI GetModuleHandleA(LPCSTR module)
1154 {
1155     WINE_MODREF *wm;
1156
1157     if ( module == NULL )
1158         wm = PROCESS_Current()->exe_modref;
1159     else
1160         wm = MODULE_FindModule( module );
1161
1162     return wm? wm->module : 0;
1163 }
1164
1165 /***********************************************************************
1166  *              GetModuleHandleW
1167  */
1168 HMODULE WINAPI GetModuleHandleW(LPCWSTR module)
1169 {
1170     HMODULE hModule;
1171     LPSTR modulea = HEAP_strdupWtoA( GetProcessHeap(), 0, module );
1172     hModule = GetModuleHandleA( modulea );
1173     HeapFree( GetProcessHeap(), 0, modulea );
1174     return hModule;
1175 }
1176
1177
1178 /***********************************************************************
1179  *              GetModuleFileNameA      (KERNEL32.235)
1180  *
1181  * GetModuleFileNameA seems to *always* return the long path;
1182  * it's only GetModuleFileName16 that decides between short/long path
1183  * by checking if exe version >= 4.0.
1184  * (SDK docu doesn't mention this)
1185  */
1186 DWORD WINAPI GetModuleFileNameA(
1187         HMODULE hModule,        /* [in] module handle (32bit) */
1188         LPSTR lpFileName,       /* [out] filenamebuffer */
1189         DWORD size )            /* [in] size of filenamebuffer */
1190 {
1191     WINE_MODREF *wm;
1192
1193     EnterCriticalSection( &PROCESS_Current()->crit_section );
1194
1195     lpFileName[0] = 0;
1196     if ((wm = MODULE32_LookupHMODULE( hModule )))
1197         lstrcpynA( lpFileName, wm->filename, size );
1198
1199     LeaveCriticalSection( &PROCESS_Current()->crit_section );
1200     TRACE("%s\n", lpFileName );
1201     return strlen(lpFileName);
1202 }
1203
1204
1205 /***********************************************************************
1206  *              GetModuleFileNameW      (KERNEL32.236)
1207  */
1208 DWORD WINAPI GetModuleFileNameW( HMODULE hModule, LPWSTR lpFileName,
1209                                    DWORD size )
1210 {
1211     LPSTR fnA = (char*)HeapAlloc( GetProcessHeap(), 0, size );
1212     DWORD res = GetModuleFileNameA( hModule, fnA, size );
1213     lstrcpynAtoW( lpFileName, fnA, size );
1214     HeapFree( GetProcessHeap(), 0, fnA );
1215     return res;
1216 }
1217
1218
1219 /***********************************************************************
1220  *           LoadLibraryExA   (KERNEL32)
1221  */
1222 HMODULE WINAPI LoadLibraryExA(LPCSTR libname, HANDLE hfile, DWORD flags)
1223 {
1224         WINE_MODREF *wm;
1225
1226         if(!libname)
1227         {
1228                 SetLastError(ERROR_INVALID_PARAMETER);
1229                 return 0;
1230         }
1231
1232         if (flags & LOAD_LIBRARY_AS_DATAFILE)
1233         {
1234             char filename[256];
1235             HFILE hFile;
1236             HMODULE hmod = 0;
1237
1238             if (!SearchPathA( NULL, libname, ".dll", sizeof(filename), filename, NULL ))
1239                 return 0;
1240             /* FIXME: maybe we should use the hfile parameter instead */
1241             hFile = CreateFileA( filename, GENERIC_READ, FILE_SHARE_READ,
1242                                  NULL, OPEN_EXISTING, 0, -1 );
1243             if (hFile != INVALID_HANDLE_VALUE)
1244             {
1245                 hmod = PE_LoadImage( hFile, filename, flags );
1246                 CloseHandle( hFile );
1247             }
1248             return hmod;
1249         }
1250
1251         EnterCriticalSection(&PROCESS_Current()->crit_section);
1252
1253         wm = MODULE_LoadLibraryExA( libname, hfile, flags );
1254         if ( wm )
1255         {
1256                 if ( !MODULE_DllProcessAttach( wm, NULL ) )
1257                 {
1258                         WARN_(module)("Attach failed for module '%s', \n", libname);
1259                         MODULE_FreeLibrary(wm);
1260                         SetLastError(ERROR_DLL_INIT_FAILED);
1261                         wm = NULL;
1262                 }
1263         }
1264
1265         LeaveCriticalSection(&PROCESS_Current()->crit_section);
1266
1267         return wm ? wm->module : 0;
1268 }
1269
1270 /***********************************************************************
1271  *      MODULE_LoadLibraryExA   (internal)
1272  *
1273  * Load a PE style module according to the load order.
1274  *
1275  * The HFILE parameter is not used and marked reserved in the SDK. I can
1276  * only guess that it should force a file to be mapped, but I rather
1277  * ignore the parameter because it would be extremely difficult to
1278  * integrate this with different types of module represenations.
1279  *
1280  */
1281 WINE_MODREF *MODULE_LoadLibraryExA( LPCSTR libname, HFILE hfile, DWORD flags )
1282 {
1283         DWORD err = GetLastError();
1284         WINE_MODREF *pwm;
1285         int i;
1286         module_loadorder_t *plo;
1287         LPSTR filename, p;
1288
1289         if ( !libname ) return NULL;
1290
1291         filename = HeapAlloc ( GetProcessHeap(), 0, MAX_PATH + 1 );
1292         if ( !filename ) return NULL;
1293
1294         /* build the modules filename */
1295         if (!SearchPathA( NULL, libname, ".dll", MAX_PATH, filename, NULL ))
1296         {
1297             if ( ! GetSystemDirectoryA ( filename, MAX_PATH ) )
1298                 goto error;
1299
1300             /* if the library name contains a path and can not be found, return an error.
1301                exception: if the path is the system directory, proceed, so that modules,
1302                which are not PE-modules can be loaded
1303
1304                if the library name does not contain a path and can not be found, assume the
1305                system directory is meant */
1306
1307             if ( ! strncasecmp ( filename, libname, strlen ( filename ) ))
1308                 strcpy ( filename, libname );
1309             else
1310             {
1311                 if ( strchr ( libname, '\\' ) || strchr ( libname, ':') || strchr ( libname, '/' ) )
1312                     goto error;
1313                 else
1314                 {
1315                     strcat ( filename, "\\" );
1316                     strcat ( filename, libname );
1317                 }
1318             }
1319
1320             /* if the filename doesn't have an extension append .DLL */
1321             if (!(p = strrchr( filename, '.')) || strchr( p, '/' ) || strchr( p, '\\'))
1322                 strcat( filename, ".DLL" );
1323         }
1324
1325         EnterCriticalSection(&PROCESS_Current()->crit_section);
1326
1327         /* Check for already loaded module */
1328         if (!(pwm = MODULE_FindModule(filename)) &&
1329             /* no path in libpath */
1330             !strchr( libname, '\\' ) && !strchr( libname, ':') && !strchr( libname, '/' ))
1331         {
1332             LPSTR       fn = HeapAlloc ( GetProcessHeap(), 0, MAX_PATH + 1 );
1333             if (fn)
1334             {
1335                 /* since the default loading mechanism uses a more detailed algorithm
1336                  * than SearchPath (like using PATH, which can even be modified between
1337                  * two attempts of loading the same DLL), the look-up above (with
1338                  * SearchPath) can have put the file in system directory, whereas it
1339                  * has already been loaded but with a different path. So do a specific
1340                  * look-up with filename (without any path)
1341                  */
1342                 strcpy ( fn, libname );
1343                 /* if the filename doesn't have an extension append .DLL */
1344                 if (!strrchr( fn, '.')) strcat( fn, ".dll" );
1345                 if ((pwm = MODULE_FindModule( fn )) != NULL)
1346                    strcpy( filename, fn );
1347                 HeapFree( GetProcessHeap(), 0, fn );
1348             }
1349         }
1350         if (pwm)
1351         {
1352                 if(!(pwm->flags & WINE_MODREF_MARKER))
1353                         pwm->refCount++;
1354
1355                 if ((pwm->flags & WINE_MODREF_DONT_RESOLVE_REFS) &&
1356                     !(flags & DONT_RESOLVE_DLL_REFERENCES))
1357                 {
1358                     extern DWORD fixup_imports(WINE_MODREF *wm); /*FIXME*/
1359                     pwm->flags &= ~WINE_MODREF_DONT_RESOLVE_REFS;
1360                     fixup_imports( pwm );
1361                 }
1362                 TRACE("Already loaded module '%s' at 0x%08x, count=%d, \n", filename, pwm->module, pwm->refCount);
1363                 LeaveCriticalSection(&PROCESS_Current()->crit_section);
1364                 HeapFree ( GetProcessHeap(), 0, filename );
1365                 return pwm;
1366         }
1367
1368         plo = MODULE_GetLoadOrder(filename, TRUE);
1369
1370         for(i = 0; i < MODULE_LOADORDER_NTYPES; i++)
1371         {
1372                 SetLastError( ERROR_FILE_NOT_FOUND );
1373                 switch(plo->loadorder[i])
1374                 {
1375                 case MODULE_LOADORDER_DLL:
1376                         TRACE("Trying native dll '%s'\n", filename);
1377                         pwm = PE_LoadLibraryExA(filename, flags);
1378                         break;
1379
1380                 case MODULE_LOADORDER_ELFDLL:
1381                         TRACE("Trying elfdll '%s'\n", filename);
1382                         if (!(pwm = BUILTIN32_LoadLibraryExA(filename, flags)))
1383                             pwm = ELFDLL_LoadLibraryExA(filename, flags);
1384                         break;
1385
1386                 case MODULE_LOADORDER_SO:
1387                         TRACE("Trying so-library '%s'\n", filename);
1388                         if (!(pwm = BUILTIN32_LoadLibraryExA(filename, flags)))
1389                             pwm = ELF_LoadLibraryExA(filename, flags);
1390                         break;
1391
1392                 case MODULE_LOADORDER_BI:
1393                         TRACE("Trying built-in '%s'\n", filename);
1394                         pwm = BUILTIN32_LoadLibraryExA(filename, flags);
1395                         break;
1396
1397                 default:
1398                         ERR("Got invalid loadorder type %d (%s index %d)\n", plo->loadorder[i], plo->modulename, i);
1399                 /* Fall through */
1400
1401                 case MODULE_LOADORDER_INVALID:  /* We ignore this as it is an empty entry */
1402                         pwm = NULL;
1403                         break;
1404                 }
1405
1406                 if(pwm)
1407                 {
1408                         /* Initialize DLL just loaded */
1409                         TRACE("Loaded module '%s' at 0x%08x, \n", filename, pwm->module);
1410
1411                         /* Set the refCount here so that an attach failure will */
1412                         /* decrement the dependencies through the MODULE_FreeLibrary call. */
1413                         pwm->refCount++;
1414
1415                         LeaveCriticalSection(&PROCESS_Current()->crit_section);
1416                         SetLastError( err );  /* restore last error */
1417                         HeapFree ( GetProcessHeap(), 0, filename );
1418                         return pwm;
1419                 }
1420
1421                 if(GetLastError() != ERROR_FILE_NOT_FOUND)
1422                         break;
1423         }
1424
1425         LeaveCriticalSection(&PROCESS_Current()->crit_section);
1426  error:
1427         WARN("Failed to load module '%s'; error=0x%08lx, \n", filename, GetLastError());
1428         HeapFree ( GetProcessHeap(), 0, filename );
1429         return NULL;
1430 }
1431
1432 /***********************************************************************
1433  *           LoadLibraryA         (KERNEL32)
1434  */
1435 HMODULE WINAPI LoadLibraryA(LPCSTR libname) {
1436         return LoadLibraryExA(libname,0,0);
1437 }
1438
1439 /***********************************************************************
1440  *           LoadLibraryW         (KERNEL32)
1441  */
1442 HMODULE WINAPI LoadLibraryW(LPCWSTR libnameW)
1443 {
1444     return LoadLibraryExW(libnameW,0,0);
1445 }
1446
1447 /***********************************************************************
1448  *           LoadLibrary32_16   (KERNEL.452)
1449  */
1450 HMODULE WINAPI LoadLibrary32_16( LPCSTR libname )
1451 {
1452     HMODULE hModule;
1453
1454     SYSLEVEL_ReleaseWin16Lock();
1455     hModule = LoadLibraryA( libname );
1456     SYSLEVEL_RestoreWin16Lock();
1457
1458     return hModule;
1459 }
1460
1461 /***********************************************************************
1462  *           LoadLibraryExW       (KERNEL32)
1463  */
1464 HMODULE WINAPI LoadLibraryExW(LPCWSTR libnameW,HANDLE hfile,DWORD flags)
1465 {
1466     LPSTR libnameA = HEAP_strdupWtoA( GetProcessHeap(), 0, libnameW );
1467     HMODULE ret = LoadLibraryExA( libnameA , hfile, flags );
1468
1469     HeapFree( GetProcessHeap(), 0, libnameA );
1470     return ret;
1471 }
1472
1473 /***********************************************************************
1474  *           MODULE_FlushModrefs
1475  *
1476  * NOTE: Assumes that the process critical section is held!
1477  *
1478  * Remove all unused modrefs and call the internal unloading routines
1479  * for the library type.
1480  */
1481 static void MODULE_FlushModrefs(void)
1482 {
1483         WINE_MODREF *wm, *next;
1484
1485         for(wm = PROCESS_Current()->modref_list; wm; wm = next)
1486         {
1487                 next = wm->next;
1488
1489                 if(wm->refCount)
1490                         continue;
1491
1492                 /* Unlink this modref from the chain */
1493                 if(wm->next)
1494                         wm->next->prev = wm->prev;
1495                 if(wm->prev)
1496                         wm->prev->next = wm->next;
1497                 if(wm == PROCESS_Current()->modref_list)
1498                         PROCESS_Current()->modref_list = wm->next;
1499
1500                 TRACE(" unloading %s\n", wm->filename);
1501                 /* VirtualFree( (LPVOID)wm->module, 0, MEM_RELEASE ); */  /* FIXME */
1502                 /* if (wm->dlhandle) dlclose( wm->dlhandle ); */  /* FIXME */
1503                 FreeLibrary16(wm->hDummyMod);
1504                 HeapFree( GetProcessHeap(), 0, wm->deps );
1505                 HeapFree( GetProcessHeap(), 0, wm->filename );
1506                 HeapFree( GetProcessHeap(), 0, wm->short_filename );
1507                 HeapFree( GetProcessHeap(), 0, wm );
1508         }
1509 }
1510
1511 /***********************************************************************
1512  *           FreeLibrary
1513  */
1514 BOOL WINAPI FreeLibrary(HINSTANCE hLibModule)
1515 {
1516     BOOL retv = FALSE;
1517     WINE_MODREF *wm;
1518
1519     EnterCriticalSection( &PROCESS_Current()->crit_section );
1520     PROCESS_Current()->free_lib_count++;
1521
1522     wm = MODULE32_LookupHMODULE( hLibModule );
1523     if ( !wm || !hLibModule )
1524         SetLastError( ERROR_INVALID_HANDLE );
1525     else
1526         retv = MODULE_FreeLibrary( wm );
1527
1528     PROCESS_Current()->free_lib_count--;
1529     LeaveCriticalSection( &PROCESS_Current()->crit_section );
1530
1531     return retv;
1532 }
1533
1534 /***********************************************************************
1535  *           MODULE_DecRefCount
1536  *
1537  * NOTE: Assumes that the process critical section is held!
1538  */
1539 static void MODULE_DecRefCount( WINE_MODREF *wm )
1540 {
1541     int i;
1542
1543     if ( wm->flags & WINE_MODREF_MARKER )
1544         return;
1545
1546     if ( wm->refCount <= 0 )
1547         return;
1548
1549     --wm->refCount;
1550     TRACE("(%s) refCount: %d\n", wm->modname, wm->refCount );
1551
1552     if ( wm->refCount == 0 )
1553     {
1554         wm->flags |= WINE_MODREF_MARKER;
1555
1556         for ( i = 0; i < wm->nDeps; i++ )
1557             if ( wm->deps[i] )
1558                 MODULE_DecRefCount( wm->deps[i] );
1559
1560         wm->flags &= ~WINE_MODREF_MARKER;
1561     }
1562 }
1563
1564 /***********************************************************************
1565  *           MODULE_FreeLibrary
1566  *
1567  * NOTE: Assumes that the process critical section is held!
1568  */
1569 BOOL MODULE_FreeLibrary( WINE_MODREF *wm )
1570 {
1571     TRACE("(%s) - START\n", wm->modname );
1572
1573     /* Recursively decrement reference counts */
1574     MODULE_DecRefCount( wm );
1575
1576     /* Call process detach notifications */
1577     if ( PROCESS_Current()->free_lib_count <= 1 )
1578     {
1579         MODULE_DllProcessDetach( FALSE, NULL );
1580         SERVER_START_REQ
1581         {
1582             struct unload_dll_request *req = server_alloc_req( sizeof(*req), 0 );
1583             req->base = (void *)wm->module;
1584             server_call_noerr( REQ_UNLOAD_DLL );
1585         }
1586         SERVER_END_REQ;
1587         MODULE_FlushModrefs();
1588     }
1589
1590     TRACE("END\n");
1591
1592     return TRUE;
1593 }
1594
1595
1596 /***********************************************************************
1597  *           FreeLibraryAndExitThread
1598  */
1599 VOID WINAPI FreeLibraryAndExitThread(HINSTANCE hLibModule, DWORD dwExitCode)
1600 {
1601     FreeLibrary(hLibModule);
1602     ExitThread(dwExitCode);
1603 }
1604
1605 /***********************************************************************
1606  *           PrivateLoadLibrary       (KERNEL32)
1607  *
1608  * FIXME: rough guesswork, don't know what "Private" means
1609  */
1610 HINSTANCE WINAPI PrivateLoadLibrary(LPCSTR libname)
1611 {
1612         return (HINSTANCE)LoadLibrary16(libname);
1613 }
1614
1615
1616
1617 /***********************************************************************
1618  *           PrivateFreeLibrary       (KERNEL32)
1619  *
1620  * FIXME: rough guesswork, don't know what "Private" means
1621  */
1622 void WINAPI PrivateFreeLibrary(HINSTANCE handle)
1623 {
1624         FreeLibrary16((HINSTANCE16)handle);
1625 }
1626
1627
1628 /***********************************************************************
1629  *           WIN32_GetProcAddress16   (KERNEL32.36)
1630  * Get procaddress in 16bit module from win32... (kernel32 undoc. ordinal func)
1631  */
1632 FARPROC16 WINAPI WIN32_GetProcAddress16( HMODULE hModule, LPCSTR name )
1633 {
1634     WORD        ordinal;
1635     FARPROC16   ret;
1636
1637     if (!hModule) {
1638         WARN("hModule may not be 0!\n");
1639         return (FARPROC16)0;
1640     }
1641     if (HIWORD(hModule))
1642     {
1643         WARN("hModule is Win32 handle (%08x)\n", hModule );
1644         return (FARPROC16)0;
1645     }
1646     hModule = GetExePtr( hModule );
1647     if (HIWORD(name)) {
1648         ordinal = NE_GetOrdinal( hModule, name );
1649         TRACE("%04x '%s'\n", hModule, name );
1650     } else {
1651         ordinal = LOWORD(name);
1652         TRACE("%04x %04x\n", hModule, ordinal );
1653     }
1654     if (!ordinal) return (FARPROC16)0;
1655     ret = NE_GetEntryPoint( hModule, ordinal );
1656     TRACE("returning %08x\n",(UINT)ret);
1657     return ret;
1658 }
1659
1660 /***********************************************************************
1661  *           GetProcAddress16   (KERNEL.50)
1662  */
1663 FARPROC16 WINAPI GetProcAddress16( HMODULE16 hModule, SEGPTR name )
1664 {
1665     WORD ordinal;
1666     FARPROC16 ret;
1667
1668     if (!hModule) hModule = GetCurrentTask();
1669     hModule = GetExePtr( hModule );
1670
1671     if (HIWORD(name) != 0)
1672     {
1673         ordinal = NE_GetOrdinal( hModule, (LPSTR)PTR_SEG_TO_LIN(name) );
1674         TRACE("%04x '%s'\n", hModule, (LPSTR)PTR_SEG_TO_LIN(name) );
1675     }
1676     else
1677     {
1678         ordinal = LOWORD(name);
1679         TRACE("%04x %04x\n", hModule, ordinal );
1680     }
1681     if (!ordinal) return (FARPROC16)0;
1682
1683     ret = NE_GetEntryPoint( hModule, ordinal );
1684
1685     TRACE("returning %08x\n", (UINT)ret );
1686     return ret;
1687 }
1688
1689
1690 /***********************************************************************
1691  *           GetProcAddress             (KERNEL32.257)
1692  */
1693 FARPROC WINAPI GetProcAddress( HMODULE hModule, LPCSTR function )
1694 {
1695     return MODULE_GetProcAddress( hModule, function, TRUE );
1696 }
1697
1698 /***********************************************************************
1699  *           GetProcAddress32                   (KERNEL.453)
1700  */
1701 FARPROC WINAPI GetProcAddress32_16( HMODULE hModule, LPCSTR function )
1702 {
1703     return MODULE_GetProcAddress( hModule, function, FALSE );
1704 }
1705
1706 /***********************************************************************
1707  *           MODULE_GetProcAddress              (internal)
1708  */
1709 FARPROC MODULE_GetProcAddress(
1710         HMODULE hModule,        /* [in] current module handle */
1711         LPCSTR function,        /* [in] function to be looked up */
1712         BOOL snoop )
1713 {
1714     WINE_MODREF *wm;
1715     FARPROC     retproc = 0;
1716
1717     if (HIWORD(function))
1718         TRACE_(win32)("(%08lx,%s)\n",(DWORD)hModule,function);
1719     else
1720         TRACE_(win32)("(%08lx,%p)\n",(DWORD)hModule,function);
1721
1722     EnterCriticalSection( &PROCESS_Current()->crit_section );
1723     if ((wm = MODULE32_LookupHMODULE( hModule )))
1724     {
1725         retproc = wm->find_export( wm, function, snoop );
1726         if (!retproc) SetLastError(ERROR_PROC_NOT_FOUND);
1727     }
1728     LeaveCriticalSection( &PROCESS_Current()->crit_section );
1729     return retproc;
1730 }
1731
1732
1733 /***************************************************************************
1734  *              HasGPHandler                    (KERNEL.338)
1735  */
1736
1737 #include "pshpack1.h"
1738 typedef struct _GPHANDLERDEF
1739 {
1740     WORD selector;
1741     WORD rangeStart;
1742     WORD rangeEnd;
1743     WORD handler;
1744 } GPHANDLERDEF;
1745 #include "poppack.h"
1746
1747 SEGPTR WINAPI HasGPHandler16( SEGPTR address )
1748 {
1749     HMODULE16 hModule;
1750     int gpOrdinal;
1751     SEGPTR gpPtr;
1752     GPHANDLERDEF *gpHandler;
1753
1754     if (    (hModule = FarGetOwner16( SELECTOROF(address) )) != 0
1755          && (gpOrdinal = NE_GetOrdinal( hModule, "__GP" )) != 0
1756          && (gpPtr = (SEGPTR)NE_GetEntryPointEx( hModule, gpOrdinal, FALSE )) != 0
1757          && !IsBadReadPtr16( gpPtr, sizeof(GPHANDLERDEF) )
1758          && (gpHandler = PTR_SEG_TO_LIN( gpPtr )) != NULL )
1759     {
1760         while (gpHandler->selector)
1761         {
1762             if (    SELECTOROF(address) == gpHandler->selector
1763                  && OFFSETOF(address)   >= gpHandler->rangeStart
1764                  && OFFSETOF(address)   <  gpHandler->rangeEnd  )
1765                 return PTR_SEG_OFF_TO_SEGPTR( gpHandler->selector,
1766                                               gpHandler->handler );
1767             gpHandler++;
1768         }
1769     }
1770
1771     return 0;
1772 }
1773