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