loader/ne/module.c

   1 /*
   2  * NE modules
   3  *
   4  * Copyright 1995 Alexandre Julliard
   5  */
   6
   7 #include <assert.h>
   8 #include <fcntl.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <unistd.h>
  12 #include <ctype.h>
  13 #include "wine/port.h"
  14 #include "wine/winbase16.h"
  15 #include "wine/library.h"
  16 #include "winerror.h"
  17 #include "module.h"
  18 #include "toolhelp.h"
  19 #include "file.h"
  20 #include "heap.h"
  21 #include "task.h"
  22 #include "snoop.h"
  23 #include "builtin16.h"
  24 #include "stackframe.h"
  25 #include "debugtools.h"
  26 #include "loadorder.h"
  27 #include "server.h"
  28
  29 DEFAULT_DEBUG_CHANNEL(module);
  30 DECLARE_DEBUG_CHANNEL(loaddll);
  31
  32 /*
  33  * Segment table entry
  34  */
  35 struct ne_segment_table_entry_s
  36 {
  37     WORD seg_data_offset;   /* Sector offset of segment data    */
  38     WORD seg_data_length;   /* Length of segment data           */
  39     WORD seg_flags;         /* Flags associated with this segment       */
  40     WORD min_alloc;         /* Minimum allocation size for this */
  41 };
  42
  43 #define hFirstModule (pThhook->hExeHead)
  44
  45 static NE_MODULE *pCachedModule = 0;  /* Module cached by NE_OpenFile */
  46
  47 static HINSTANCE16 NE_LoadModule( LPCSTR name, BOOL lib_only );
  48 static BOOL16 NE_FreeModule( HMODULE16 hModule, BOOL call_wep );
  49 static void NE_InitProcess(void) WINE_NORETURN;
  50
  51 static HINSTANCE16 MODULE_LoadModule16( LPCSTR libname, BOOL implicit, BOOL lib_only );
  52
  53 static HMODULE16 NE_GetModuleByFilename( LPCSTR name );
  54
  55 /* ### start build ### */
  56 extern WORD CALLBACK NE_CallTo16_word_w(FARPROC16,WORD);
  57 /* ### stop build ### */
  58
  59 /***********************************************************************
  60  *           NE_GetPtr
  61  */
  62 NE_MODULE *NE_GetPtr( HMODULE16 hModule )
  63 {
  64     return (NE_MODULE *)GlobalLock16( GetExePtr(hModule) );
  65 }
  66
  67
  68 /***********************************************************************
  69  *           NE_DumpModule
  70  */
  71 void NE_DumpModule( HMODULE16 hModule )
  72 {
  73     int i, ordinal;
  74     SEGTABLEENTRY *pSeg;
  75     BYTE *pstr;
  76     WORD *pword;
  77     NE_MODULE *pModule;
  78     ET_BUNDLE *bundle;
  79     ET_ENTRY *entry;
  80
  81     if (!(pModule = NE_GetPtr( hModule )))
  82     {
  83         MESSAGE( "**** %04x is not a module handle\n", hModule );
  84         return;
  85     }
  86
  87       /* Dump the module info */
  88     DPRINTF( "---\n" );
  89     DPRINTF( "Module %04x:\n", hModule );
  90     DPRINTF( "count=%d flags=%04x heap=%d stack=%d\n",
  91           pModule->count, pModule->flags,
  92           pModule->heap_size, pModule->stack_size );
  93     DPRINTF( "cs:ip=%04x:%04x ss:sp=%04x:%04x ds=%04x nb seg=%d modrefs=%d\n",
  94           pModule->cs, pModule->ip, pModule->ss, pModule->sp, pModule->dgroup,
  95           pModule->seg_count, pModule->modref_count );
  96     DPRINTF( "os_flags=%d swap_area=%d version=%04x\n",
  97           pModule->os_flags, pModule->min_swap_area,
  98           pModule->expected_version );
  99     if (pModule->flags & NE_FFLAGS_WIN32)
 100         DPRINTF( "PE module=%08x\n", pModule->module32 );
 101
 102       /* Dump the file info */
 103     DPRINTF( "---\n" );
 104     DPRINTF( "Filename: '%s'\n", NE_MODULE_NAME(pModule) );
 105
 106       /* Dump the segment table */
 107     DPRINTF( "---\n" );
 108     DPRINTF( "Segment table:\n" );
 109     pSeg = NE_SEG_TABLE( pModule );
 110     for (i = 0; i < pModule->seg_count; i++, pSeg++)
 111         DPRINTF( "%02x: pos=%d size=%d flags=%04x minsize=%d hSeg=%04x\n",
 112               i + 1, pSeg->filepos, pSeg->size, pSeg->flags,
 113               pSeg->minsize, pSeg->hSeg );
 114
 115       /* Dump the resource table */
 116     DPRINTF( "---\n" );
 117     DPRINTF( "Resource table:\n" );
 118     if (pModule->res_table)
 119     {
 120         pword = (WORD *)((BYTE *)pModule + pModule->res_table);
 121         DPRINTF( "Alignment: %d\n", *pword++ );
 122         while (*pword)
 123         {
 124             NE_TYPEINFO *ptr = (NE_TYPEINFO *)pword;
 125             NE_NAMEINFO *pname = (NE_NAMEINFO *)(ptr + 1);
 126             DPRINTF( "id=%04x count=%d\n", ptr->type_id, ptr->count );
 127             for (i = 0; i < ptr->count; i++, pname++)
 128                 DPRINTF( "offset=%d len=%d id=%04x\n",
 129                       pname->offset, pname->length, pname->id );
 130             pword = (WORD *)pname;
 131         }
 132     }
 133     else DPRINTF( "None\n" );
 134
 135       /* Dump the resident name table */
 136     DPRINTF( "---\n" );
 137     DPRINTF( "Resident-name table:\n" );
 138     pstr = (char *)pModule + pModule->name_table;
 139     while (*pstr)
 140     {
 141         DPRINTF( "%*.*s: %d\n", *pstr, *pstr, pstr + 1,
 142               *(WORD *)(pstr + *pstr + 1) );
 143         pstr += *pstr + 1 + sizeof(WORD);
 144     }
 145
 146       /* Dump the module reference table */
 147     DPRINTF( "---\n" );
 148     DPRINTF( "Module ref table:\n" );
 149     if (pModule->modref_table)
 150     {
 151         pword = (WORD *)((BYTE *)pModule + pModule->modref_table);
 152         for (i = 0; i < pModule->modref_count; i++, pword++)
 153         {
 154             char name[10];
 155             GetModuleName16( *pword, name, sizeof(name) );
 156             DPRINTF( "%d: %04x -> '%s'\n", i, *pword, name );
 157         }
 158     }
 159     else DPRINTF( "None\n" );
 160
 161       /* Dump the entry table */
 162     DPRINTF( "---\n" );
 163     DPRINTF( "Entry table:\n" );
 164     bundle = (ET_BUNDLE *)((BYTE *)pModule+pModule->entry_table);
 165     do {
 166         entry = (ET_ENTRY *)((BYTE *)bundle+6);
 167         DPRINTF( "Bundle %d-%d: %02x\n", bundle->first, bundle->last, entry->type);
 168         ordinal = bundle->first;
 169         while (ordinal < bundle->last)
 170         {
 171             if (entry->type == 0xff)
 172                 DPRINTF("%d: %02x:%04x (moveable)\n", ordinal++, entry->segnum, entry->offs);
 173             else
 174                 DPRINTF("%d: %02x:%04x (fixed)\n", ordinal++, entry->segnum, entry->offs);
 175             entry++;
 176     }
 177     } while ( (bundle->next)
 178            && (bundle = ((ET_BUNDLE *)((BYTE *)pModule + bundle->next))) );
 179
 180     /* Dump the non-resident names table */
 181     DPRINTF( "---\n" );
 182     DPRINTF( "Non-resident names table:\n" );
 183     if (pModule->nrname_handle)
 184     {
 185         pstr = (char *)GlobalLock16( pModule->nrname_handle );
 186         while (*pstr)
 187         {
 188             DPRINTF( "%*.*s: %d\n", *pstr, *pstr, pstr + 1,
 189                    *(WORD *)(pstr + *pstr + 1) );
 190             pstr += *pstr + 1 + sizeof(WORD);
 191         }
 192     }
 193     DPRINTF( "\n" );
 194 }
 195
 196
 197 /***********************************************************************
 198  *           NE_WalkModules
 199  *
 200  * Walk the module list and print the modules.
 201  */
 202 void NE_WalkModules(void)
 203 {
 204     HMODULE16 hModule = hFirstModule;
 205     MESSAGE( "Module Flags Name\n" );
 206     while (hModule)
 207     {
 208         NE_MODULE *pModule = NE_GetPtr( hModule );
 209         if (!pModule)
 210         {
 211             MESSAGE( "Bad module %04x in list\n", hModule );
 212             return;
 213         }
 214         MESSAGE( " %04x  %04x  %.*s\n", hModule, pModule->flags,
 215                  *((char *)pModule + pModule->name_table),
 216                  (char *)pModule + pModule->name_table + 1 );
 217         hModule = pModule->next;
 218     }
 219 }
 220
 221
 222 /**********************************************************************
 223  *           NE_RegisterModule
 224  */
 225 void NE_RegisterModule( NE_MODULE *pModule )
 226 {
 227     pModule->next = hFirstModule;
 228     hFirstModule = pModule->self;
 229 }
 230
 231
 232 /***********************************************************************
 233  *           NE_GetOrdinal
 234  *
 235  * Lookup the ordinal for a given name.
 236  */
 237 WORD NE_GetOrdinal( HMODULE16 hModule, const char *name )
 238 {
 239     unsigned char buffer[256], *cpnt;
 240     BYTE len;
 241     NE_MODULE *pModule;
 242
 243     if (!(pModule = NE_GetPtr( hModule ))) return 0;
 244     assert( !(pModule->flags & NE_FFLAGS_WIN32) );
 245
 246     TRACE("(%04x,'%s')\n", hModule, name );
 247
 248       /* First handle names of the form '#xxxx' */
 249
 250     if (name[0] == '#') return atoi( name + 1 );
 251
 252       /* Now copy and uppercase the string */
 253
 254     strcpy( buffer, name );
 255     for (cpnt = buffer; *cpnt; cpnt++) *cpnt = FILE_toupper(*cpnt);
 256     len = cpnt - buffer;
 257
 258       /* First search the resident names */
 259
 260     cpnt = (char *)pModule + pModule->name_table;
 261
 262       /* Skip the first entry (module name) */
 263     cpnt += *cpnt + 1 + sizeof(WORD);
 264     while (*cpnt)
 265     {
 266         if (((BYTE)*cpnt == len) && !memcmp( cpnt+1, buffer, len ))
 267         {
 268             WORD ordinal = GET_UA_WORD( cpnt + *cpnt + 1 );
 269             TRACE("  Found: ordinal=%d\n", ordinal );
 270             return ordinal;
 271         }
 272         cpnt += *cpnt + 1 + sizeof(WORD);
 273     }
 274
 275       /* Now search the non-resident names table */
 276
 277     if (!pModule->nrname_handle) return 0;  /* No non-resident table */
 278     cpnt = (char *)GlobalLock16( pModule->nrname_handle );
 279
 280       /* Skip the first entry (module description string) */
 281     cpnt += *cpnt + 1 + sizeof(WORD);
 282     while (*cpnt)
 283     {
 284         if (((BYTE)*cpnt == len) && !memcmp( cpnt+1, buffer, len ))
 285         {
 286             WORD ordinal = GET_UA_WORD( cpnt + *cpnt + 1 );
 287             TRACE("  Found: ordinal=%d\n", ordinal );
 288             return ordinal;
 289         }
 290         cpnt += *cpnt + 1 + sizeof(WORD);
 291     }
 292     return 0;
 293 }
 294
 295
 296 /***********************************************************************
 297  *           NE_GetEntryPoint / EntryAddrProc16   (KERNEL Wine-specific export)
 298  *
 299  * Return the entry point for a given ordinal.
 300  */
 301 FARPROC16 WINAPI EntryAddrProc16( HMODULE16 hModule, WORD ordinal )
 302 {
 303     FARPROC16 ret = NE_GetEntryPointEx( hModule, ordinal, TRUE );
 304     CURRENT_STACK16->ecx = hModule; /* FIXME: might be incorrect value */
 305     return ret;
 306 }
 307 FARPROC16 WINAPI NE_GetEntryPoint( HMODULE16 hModule, WORD ordinal )
 308 {
 309     return NE_GetEntryPointEx( hModule, ordinal, TRUE );
 310 }
 311 FARPROC16 NE_GetEntryPointEx( HMODULE16 hModule, WORD ordinal, BOOL16 snoop )
 312 {
 313     NE_MODULE *pModule;
 314     WORD sel, offset, i;
 315
 316     ET_ENTRY *entry;
 317     ET_BUNDLE *bundle;
 318
 319     if (!(pModule = NE_GetPtr( hModule ))) return 0;
 320     assert( !(pModule->flags & NE_FFLAGS_WIN32) );
 321
 322     bundle = (ET_BUNDLE *)((BYTE *)pModule + pModule->entry_table);
 323     while ((ordinal < bundle->first + 1) || (ordinal > bundle->last))
 324     {
 325         if (!(bundle->next))
 326             return 0;
 327         bundle = (ET_BUNDLE *)((BYTE *)pModule + bundle->next);
 328     }
 329
 330     entry = (ET_ENTRY *)((BYTE *)bundle+6);
 331     for (i=0; i < (ordinal - bundle->first - 1); i++)
 332         entry++;
 333
 334     sel = entry->segnum;
 335     offset = GET_UA_WORD( &entry->offs );
 336
 337     if (sel == 0xfe) sel = 0xffff;  /* constant entry */
 338     else sel = GlobalHandleToSel16(NE_SEG_TABLE(pModule)[sel-1].hSeg);
 339     if (sel==0xffff)
 340         return (FARPROC16)MAKESEGPTR( sel, offset );
 341     if (!snoop)
 342         return (FARPROC16)MAKESEGPTR( sel, offset );
 343     else
 344         return (FARPROC16)SNOOP16_GetProcAddress16(hModule,ordinal,(FARPROC16)MAKESEGPTR( sel, offset ));
 345 }
 346
 347
 348 /***********************************************************************
 349  *           NE_SetEntryPoint
 350  *
 351  * Change the value of an entry point. Use with caution!
 352  * It can only change the offset value, not the selector.
 353  */
 354 BOOL16 NE_SetEntryPoint( HMODULE16 hModule, WORD ordinal, WORD offset )
 355 {
 356     NE_MODULE *pModule;
 357     ET_ENTRY *entry;
 358     ET_BUNDLE *bundle;
 359     int i;
 360
 361     if (!(pModule = NE_GetPtr( hModule ))) return FALSE;
 362     assert( !(pModule->flags & NE_FFLAGS_WIN32) );
 363
 364     bundle = (ET_BUNDLE *)((BYTE *)pModule + pModule->entry_table);
 365     while ((ordinal < bundle->first + 1) || (ordinal > bundle->last))
 366         {
 367         bundle = (ET_BUNDLE *)((BYTE *)pModule + bundle->next);
 368         if (!(bundle->next))
 369             return 0;
 370     }
 371
 372     entry = (ET_ENTRY *)((BYTE *)bundle+6);
 373     for (i=0; i < (ordinal - bundle->first - 1); i++)
 374         entry++;
 375
 376     PUT_UA_WORD( &entry->offs, offset );
 377     return TRUE;
 378 }
 379
 380
 381 /***********************************************************************
 382  *           NE_OpenFile
 383  */
 384 HANDLE NE_OpenFile( NE_MODULE *pModule )
 385 {
 386     char *name;
 387
 388     static HANDLE cachedfd = INVALID_HANDLE_VALUE;
 389
 390     TRACE("(%p) cache: mod=%p fd=%d\n",
 391            pModule, pCachedModule, cachedfd );
 392     if (pCachedModule == pModule) return cachedfd;
 393     CloseHandle( cachedfd );
 394     pCachedModule = pModule;
 395     name = NE_MODULE_NAME( pModule );
 396     if ((cachedfd = CreateFileA( name, GENERIC_READ, FILE_SHARE_READ,
 397                                    NULL, OPEN_EXISTING, 0, 0 )) == INVALID_HANDLE_VALUE)
 398         MESSAGE( "Can't open file '%s' for module %04x\n", name, pModule->self );
 399     TRACE("opened '%s' -> %d\n",
 400                     name, cachedfd );
 401     return cachedfd;
 402 }
 403
 404
 405 /***********************************************************************
 406  *           NE_LoadExeHeader
 407  */
 408 static HMODULE16 NE_LoadExeHeader( HANDLE hFile, LPCSTR path )
 409 {
 410     IMAGE_DOS_HEADER mz_header;
 411     IMAGE_OS2_HEADER ne_header;
 412     int size;
 413     HMODULE16 hModule;
 414     NE_MODULE *pModule;
 415     BYTE *pData, *pTempEntryTable;
 416     char *buffer, *fastload = NULL;
 417     int fastload_offset = 0, fastload_length = 0;
 418     ET_ENTRY *entry;
 419     ET_BUNDLE *bundle, *oldbundle;
 420     OFSTRUCT *ofs;
 421
 422   /* Read a block from either the file or the fast-load area. */
 423 #define READ(offset,size,buffer) \
 424        ((fastload && ((offset) >= fastload_offset) && \
 425          ((offset)+(size) <= fastload_offset+fastload_length)) ? \
 426         (memcpy( buffer, fastload+(offset)-fastload_offset, (size) ), TRUE) : \
 427         (_llseek( hFile, (offset), SEEK_SET), \
 428          _lread( hFile, (buffer), (size) ) == (size)))
 429
 430     _llseek( hFile, 0, SEEK_SET );
 431     if ((_lread(hFile,&mz_header,sizeof(mz_header)) != sizeof(mz_header)) ||
 432         (mz_header.e_magic != IMAGE_DOS_SIGNATURE))
 433         return (HMODULE16)11;  /* invalid exe */
 434
 435     _llseek( hFile, mz_header.e_lfanew, SEEK_SET );
 436     if (_lread( hFile, &ne_header, sizeof(ne_header) ) != sizeof(ne_header))
 437         return (HMODULE16)11;  /* invalid exe */
 438
 439     if (ne_header.ne_magic == IMAGE_NT_SIGNATURE) return (HMODULE16)21;  /* win32 exe */
 440     if (ne_header.ne_magic == IMAGE_OS2_SIGNATURE_LX) {
 441         MESSAGE("Sorry, this is an OS/2 linear executable (LX) file !\n");
 442         return (HMODULE16)12;
 443     }
 444     if (ne_header.ne_magic != IMAGE_OS2_SIGNATURE) return (HMODULE16)11;  /* invalid exe */
 445
 446     /* We now have a valid NE header */
 447
 448     size = sizeof(NE_MODULE) +
 449              /* segment table */
 450            ne_header.ne_cseg * sizeof(SEGTABLEENTRY) +
 451              /* resource table */
 452            ne_header.ne_restab - ne_header.ne_rsrctab +
 453              /* resident names table */
 454            ne_header.ne_modtab - ne_header.ne_restab +
 455              /* module ref table */
 456            ne_header.ne_cmod * sizeof(WORD) +
 457              /* imported names table */
 458            ne_header.ne_enttab - ne_header.ne_imptab +
 459              /* entry table length */
 460            ne_header.ne_cbenttab +
 461              /* entry table extra conversion space */
 462            sizeof(ET_BUNDLE) +
 463            2 * (ne_header.ne_cbenttab - ne_header.ne_cmovent*6) +
 464              /* loaded file info */
 465            sizeof(OFSTRUCT) - sizeof(ofs->szPathName) + strlen(path) + 1;
 466
 467     hModule = GlobalAlloc16( GMEM_FIXED | GMEM_ZEROINIT, size );
 468     if (!hModule) return (HMODULE16)11;  /* invalid exe */
 469
 470     FarSetOwner16( hModule, hModule );
 471     pModule = (NE_MODULE *)GlobalLock16( hModule );
 472     memcpy( pModule, &ne_header, sizeof(ne_header) );
 473     pModule->count = 0;
 474     /* check *programs* for default minimal stack size */
 475     if ( (!(pModule->flags & NE_FFLAGS_LIBMODULE))
 476                 && (pModule->stack_size < 0x1400) )
 477                 pModule->stack_size = 0x1400;
 478     pModule->module32 = 0;
 479     pModule->self = hModule;
 480     pModule->self_loading_sel = 0;
 481     pData = (BYTE *)(pModule + 1);
 482
 483     /* Clear internal Wine flags in case they are set in the EXE file */
 484
 485     pModule->flags &= ~(NE_FFLAGS_BUILTIN | NE_FFLAGS_WIN32);
 486
 487     /* Read the fast-load area */
 488
 489     if (ne_header.ne_flagsothers & NE_AFLAGS_FASTLOAD)
 490     {
 491         fastload_offset=ne_header.fastload_offset << ne_header.ne_align;
 492         fastload_length=ne_header.fastload_length << ne_header.ne_align;
 493         TRACE("Using fast-load area offset=%x len=%d\n",
 494                         fastload_offset, fastload_length );
 495         if ((fastload = HeapAlloc( GetProcessHeap(), 0, fastload_length )) != NULL)
 496         {
 497             _llseek( hFile, fastload_offset, SEEK_SET);
 498             if (_lread(hFile, fastload, fastload_length) != fastload_length)
 499             {
 500                 HeapFree( GetProcessHeap(), 0, fastload );
 501                 WARN("Error reading fast-load area!\n");
 502                 fastload = NULL;
 503             }
 504         }
 505     }
 506
 507     /* Get the segment table */
 508
 509     pModule->seg_table = pData - (BYTE *)pModule;
 510     buffer = HeapAlloc( GetProcessHeap(), 0, ne_header.ne_cseg *
 511                                       sizeof(struct ne_segment_table_entry_s));
 512     if (buffer)
 513     {
 514         int i;
 515         struct ne_segment_table_entry_s *pSeg;
 516
 517         if (!READ( mz_header.e_lfanew + ne_header.ne_segtab,
 518              ne_header.ne_cseg * sizeof(struct ne_segment_table_entry_s),
 519              buffer ))
 520         {
 521             HeapFree( GetProcessHeap(), 0, buffer );
 522             if (fastload)
 523                 HeapFree( GetProcessHeap(), 0, fastload );
 524             GlobalFree16( hModule );
 525             return (HMODULE16)11;  /* invalid exe */
 526         }
 527         pSeg = (struct ne_segment_table_entry_s *)buffer;
 528         for (i = ne_header.ne_cseg; i > 0; i--, pSeg++)
 529         {
 530             memcpy( pData, pSeg, sizeof(*pSeg) );
 531             pData += sizeof(SEGTABLEENTRY);
 532         }
 533         HeapFree( GetProcessHeap(), 0, buffer );
 534     }
 535     else
 536     {
 537         if (fastload)
 538             HeapFree( GetProcessHeap(), 0, fastload );
 539         GlobalFree16( hModule );
 540         return (HMODULE16)11;  /* invalid exe */
 541     }
 542
 543     /* Get the resource table */
 544
 545     if (ne_header.ne_rsrctab < ne_header.ne_restab)
 546     {
 547         pModule->res_table = pData - (BYTE *)pModule;
 548         if (!READ(mz_header.e_lfanew + ne_header.ne_rsrctab,
 549                   ne_header.ne_restab - ne_header.ne_rsrctab,
 550                   pData ))
 551             return (HMODULE16)11;  /* invalid exe */
 552         pData += ne_header.ne_restab - ne_header.ne_rsrctab;
 553         NE_InitResourceHandler( hModule );
 554     }
 555     else pModule->res_table = 0;  /* No resource table */
 556
 557     /* Get the resident names table */
 558
 559     pModule->name_table = pData - (BYTE *)pModule;
 560     if (!READ( mz_header.e_lfanew + ne_header.ne_restab,
 561                ne_header.ne_modtab - ne_header.ne_restab,
 562                pData ))
 563     {
 564         if (fastload)
 565             HeapFree( GetProcessHeap(), 0, fastload );
 566         GlobalFree16( hModule );
 567         return (HMODULE16)11;  /* invalid exe */
 568     }
 569     pData += ne_header.ne_modtab - ne_header.ne_restab;
 570
 571     /* Get the module references table */
 572
 573     if (ne_header.ne_cmod > 0)
 574     {
 575         pModule->modref_table = pData - (BYTE *)pModule;
 576         if (!READ( mz_header.e_lfanew + ne_header.ne_modtab,
 577                   ne_header.ne_cmod * sizeof(WORD),
 578                   pData ))
 579         {
 580             if (fastload)
 581                 HeapFree( GetProcessHeap(), 0, fastload );
 582             GlobalFree16( hModule );
 583             return (HMODULE16)11;  /* invalid exe */
 584         }
 585         pData += ne_header.ne_cmod * sizeof(WORD);
 586     }
 587     else pModule->modref_table = 0;  /* No module references */
 588
 589     /* Get the imported names table */
 590
 591     pModule->import_table = pData - (BYTE *)pModule;
 592     if (!READ( mz_header.e_lfanew + ne_header.ne_imptab,
 593                ne_header.ne_enttab - ne_header.ne_imptab,
 594                pData ))
 595     {
 596         if (fastload)
 597             HeapFree( GetProcessHeap(), 0, fastload );
 598         GlobalFree16( hModule );
 599         return (HMODULE16)11;  /* invalid exe */
 600     }
 601     pData += ne_header.ne_enttab - ne_header.ne_imptab;
 602
 603     /* Load entry table, convert it to the optimized version used by Windows */
 604
 605     if ((pTempEntryTable = HeapAlloc( GetProcessHeap(), 0, ne_header.ne_cbenttab)) != NULL)
 606     {
 607         BYTE nr_entries, type, *s;
 608
 609         TRACE("Converting entry table.\n");
 610     pModule->entry_table = pData - (BYTE *)pModule;
 611     if (!READ( mz_header.e_lfanew + ne_header.ne_enttab,
 612                 ne_header.ne_cbenttab, pTempEntryTable ))
 613     {
 614             HeapFree( GetProcessHeap(), 0, pTempEntryTable );
 615             if (fastload)
 616                 HeapFree( GetProcessHeap(), 0, fastload );
 617         GlobalFree16( hModule );
 618         return (HMODULE16)11;  /* invalid exe */
 619     }
 620
 621         s = pTempEntryTable;
 622         TRACE("entry table: offs %04x, len %04x, entries %d\n", ne_header.ne_enttab, ne_header.ne_cbenttab, *s);
 623
 624         bundle = (ET_BUNDLE *)pData;
 625         TRACE("first bundle: %p\n", bundle);
 626         memset(bundle, 0, sizeof(ET_BUNDLE)); /* in case no entry table exists */
 627         entry = (ET_ENTRY *)((BYTE *)bundle+6);
 628
 629         while ((nr_entries = *s++))
 630         {
 631             if ((type = *s++))
 632             {
 633                 bundle->last += nr_entries;
 634                 if (type == 0xff)
 635                 while (nr_entries--)
 636                 {
 637                     entry->type   = type;
 638                     entry->flags  = *s++;
 639                     s += 2;
 640                     entry->segnum = *s++;
 641                     entry->offs   = *(WORD *)s; s += 2;
 642                     /*TRACE(module, "entry: %p, type: %d, flags: %d, segnum: %d, offs: %04x\n", entry, entry->type, entry->flags, entry->segnum, entry->offs);*/
 643                     entry++;
 644                 }
 645                 else
 646                 while (nr_entries--)
 647                 {
 648                     entry->type   = type;
 649                     entry->flags  = *s++;
 650                     entry->segnum = type;
 651                     entry->offs   = *(WORD *)s; s += 2;
 652                     /*TRACE(module, "entry: %p, type: %d, flags: %d, segnum: %d, offs: %04x\n", entry, entry->type, entry->flags, entry->segnum, entry->offs);*/
 653                     entry++;
 654                 }
 655             }
 656             else
 657             {
 658                 if (bundle->first == bundle->last)
 659                 {
 660                     bundle->first += nr_entries;
 661                     bundle->last += nr_entries;
 662                 }
 663                 else
 664                 {
 665                     oldbundle = bundle;
 666                     oldbundle->next = ((int)entry - (int)pModule);
 667                     bundle = (ET_BUNDLE *)entry;
 668                     TRACE("new bundle: %p\n", bundle);
 669                     bundle->first = bundle->last =
 670                         oldbundle->last + nr_entries;
 671                     bundle->next = 0;
 672                     (BYTE *)entry += sizeof(ET_BUNDLE);
 673                 }
 674             }
 675         }
 676         HeapFree( GetProcessHeap(), 0, pTempEntryTable );
 677     }
 678     else
 679     {
 680         if (fastload)
 681             HeapFree( GetProcessHeap(), 0, fastload );
 682         GlobalFree16( hModule );
 683         return (HMODULE16)11;  /* invalid exe */
 684     }
 685
 686     pData += ne_header.ne_cbenttab + sizeof(ET_BUNDLE) +
 687              2 * (ne_header.ne_cbenttab - ne_header.ne_cmovent*6);
 688
 689     if ((DWORD)entry > (DWORD)pData)
 690        ERR("converted entry table bigger than reserved space !!!\nentry: %p, pData: %p. Please report !\n", entry, pData);
 691
 692     /* Store the filename information */
 693
 694     pModule->fileinfo = pData - (BYTE *)pModule;
 695     size = sizeof(OFSTRUCT) - sizeof(ofs->szPathName) + strlen(path) + 1;
 696     ofs = (OFSTRUCT *)pData;
 697     ofs->cBytes = size - 1;
 698     ofs->fFixedDisk = 1;
 699     strcpy( ofs->szPathName, path );
 700     pData += size;
 701
 702     /* Free the fast-load area */
 703
 704 #undef READ
 705     if (fastload)
 706         HeapFree( GetProcessHeap(), 0, fastload );
 707
 708     /* Get the non-resident names table */
 709
 710     if (ne_header.ne_cbnrestab)
 711     {
 712         pModule->nrname_handle = GlobalAlloc16( 0, ne_header.ne_cbnrestab );
 713         if (!pModule->nrname_handle)
 714         {
 715             GlobalFree16( hModule );
 716             return (HMODULE16)11;  /* invalid exe */
 717         }
 718         FarSetOwner16( pModule->nrname_handle, hModule );
 719         buffer = GlobalLock16( pModule->nrname_handle );
 720         _llseek( hFile, ne_header.ne_nrestab, SEEK_SET );
 721         if (_lread( hFile, buffer, ne_header.ne_cbnrestab )
 722               != ne_header.ne_cbnrestab)
 723         {
 724             GlobalFree16( pModule->nrname_handle );
 725             GlobalFree16( hModule );
 726             return (HMODULE16)11;  /* invalid exe */
 727         }
 728     }
 729     else pModule->nrname_handle = 0;
 730
 731     /* Allocate a segment for the implicitly-loaded DLLs */
 732
 733     if (pModule->modref_count)
 734     {
 735         pModule->dlls_to_init = GlobalAlloc16( GMEM_ZEROINIT,
 736                                                (pModule->modref_count+1)*sizeof(HMODULE16) );
 737         if (!pModule->dlls_to_init)
 738         {
 739             if (pModule->nrname_handle) GlobalFree16( pModule->nrname_handle );
 740             GlobalFree16( hModule );
 741             return (HMODULE16)11;  /* invalid exe */
 742         }
 743         FarSetOwner16( pModule->dlls_to_init, hModule );
 744     }
 745     else pModule->dlls_to_init = 0;
 746
 747     NE_RegisterModule( pModule );
 748     SNOOP16_RegisterDLL(pModule,path);
 749     return hModule;
 750 }
 751
 752
 753 /***********************************************************************
 754  *           NE_LoadDLLs
 755  *
 756  * Load all DLLs implicitly linked to a module.
 757  */
 758 static BOOL NE_LoadDLLs( NE_MODULE *pModule )
 759 {
 760     int i;
 761     WORD *pModRef = (WORD *)((char *)pModule + pModule->modref_table);
 762     WORD *pDLLs = (WORD *)GlobalLock16( pModule->dlls_to_init );
 763
 764     for (i = 0; i < pModule->modref_count; i++, pModRef++)
 765     {
 766         char buffer[260], *p;
 767         BYTE *pstr = (BYTE *)pModule + pModule->import_table + *pModRef;
 768         memcpy( buffer, pstr + 1, *pstr );
 769         *(buffer + *pstr) = 0; /* terminate it */
 770
 771         TRACE("Loading '%s'\n", buffer );
 772         if (!(*pModRef = GetModuleHandle16( buffer )))
 773         {
 774             /* If the DLL is not loaded yet, load it and store */
 775             /* its handle in the list of DLLs to initialize.   */
 776             HMODULE16 hDLL;
 777
 778             /* Append .DLL to name if no extension present */
 779             if (!(p = strrchr( buffer, '.')) || strchr( p, '/' ) || strchr( p, '\\'))
 780                     strcat( buffer, ".DLL" );
 781
 782             if ((hDLL = MODULE_LoadModule16( buffer, TRUE, TRUE )) < 32)
 783             {
 784                 /* FIXME: cleanup what was done */
 785
 786                 MESSAGE( "Could not load '%s' required by '%.*s', error=%d\n",
 787                      buffer, *((BYTE*)pModule + pModule->name_table),
 788                      (char *)pModule + pModule->name_table + 1, hDLL );
 789                 return FALSE;
 790             }
 791             *pModRef = GetExePtr( hDLL );
 792             *pDLLs++ = *pModRef;
 793         }
 794         else  /* Increment the reference count of the DLL */
 795         {
 796             NE_MODULE *pOldDLL;
 797
 798             pOldDLL = NE_GetPtr( *pModRef );
 799             if (pOldDLL) pOldDLL->count++;
 800         }
 801     }
 802     return TRUE;
 803 }
 804
 805
 806 /**********************************************************************
 807  *          NE_DoLoadModule
 808  *
 809  * Load first instance of NE module from file.
 810  *
 811  * pModule must point to a module structure prepared by NE_LoadExeHeader.
 812  * This routine must never be called twice on a module.
 813  *
 814  */
 815 static HINSTANCE16 NE_DoLoadModule( NE_MODULE *pModule )
 816 {
 817     /* Allocate the segments for this module */
 818
 819     if (!NE_CreateAllSegments( pModule ))
 820         return ERROR_NOT_ENOUGH_MEMORY; /* 8 */
 821
 822     /* Load the referenced DLLs */
 823
 824     if (!NE_LoadDLLs( pModule ))
 825         return ERROR_FILE_NOT_FOUND; /* 2 */
 826
 827     /* Load the segments */
 828
 829     NE_LoadAllSegments( pModule );
 830
 831     /* Make sure the usage count is 1 on the first loading of  */
 832     /* the module, even if it contains circular DLL references */
 833
 834     pModule->count = 1;
 835
 836     return NE_GetInstance( pModule );
 837 }
 838
 839 /**********************************************************************
 840  *          NE_LoadModule
 841  *
 842  * Load first instance of NE module. (Note: caller is responsible for
 843  * ensuring the module isn't already loaded!)
 844  *
 845  * If the module turns out to be an executable module, only a
 846  * handle to a module stub is returned; this needs to be initialized
 847  * by calling NE_DoLoadModule later, in the context of the newly
 848  * created process.
 849  *
 850  * If lib_only is TRUE, however, the module is perforce treated
 851  * like a DLL module, even if it is an executable module.
 852  *
 853  */
 854 static HINSTANCE16 NE_LoadModule( LPCSTR name, BOOL lib_only )
 855 {
 856     NE_MODULE *pModule;
 857     HMODULE16 hModule;
 858     HINSTANCE16 hInstance;
 859     HFILE16 hFile;
 860     OFSTRUCT ofs;
 861
 862     /* Open file */
 863     if ((hFile = OpenFile16( name, &ofs, OF_READ )) == HFILE_ERROR16)
 864         return (HMODULE16)2;  /* File not found */
 865
 866     hModule = NE_LoadExeHeader( DosFileHandleToWin32Handle(hFile), ofs.szPathName );
 867     _lclose16( hFile );
 868     if (hModule < 32) return hModule;
 869
 870     pModule = NE_GetPtr( hModule );
 871     if ( !pModule ) return hModule;
 872
 873     if ( !lib_only && !( pModule->flags & NE_FFLAGS_LIBMODULE ) )
 874         return hModule;
 875
 876     hInstance = NE_DoLoadModule( pModule );
 877     if ( hInstance < 32 )
 878     {
 879         /* cleanup ... */
 880         NE_FreeModule( hModule, 0 );
 881     }
 882
 883     return hInstance;
 884 }
 885
 886
 887 /**********************************************************************
 888  *          MODULE_LoadModule16
 889  *
 890  * Load a NE module in the order of the loadorder specification.
 891  * The caller is responsible that the module is not loaded already.
 892  *
 893  */
 894 static HINSTANCE16 MODULE_LoadModule16( LPCSTR libname, BOOL implicit, BOOL lib_only )
 895 {
 896         HINSTANCE16 hinst;
 897         int i;
 898         module_loadorder_t *plo;
 899         const char *filetype = "";
 900
 901         plo = MODULE_GetLoadOrder(libname, FALSE);
 902
 903         for(i = 0; i < MODULE_LOADORDER_NTYPES; i++)
 904         {
 905                 switch(plo->loadorder[i])
 906                 {
 907                 case MODULE_LOADORDER_DLL:
 908                         TRACE("Trying native dll '%s'\n", libname);
 909                         hinst = NE_LoadModule(libname, lib_only);
 910                         filetype = "native";
 911                         break;
 912
 913                 case MODULE_LOADORDER_BI:
 914                         TRACE("Trying built-in '%s'\n", libname);
 915                         hinst = BUILTIN_LoadModule(libname);
 916                         filetype = "builtin";
 917                         break;
 918
 919                 default:
 920                         ERR("Got invalid loadorder type %d (%s index %d)\n", plo->loadorder[i], plo->modulename, i);
 921                 /* Fall through */
 922
 923                 case MODULE_LOADORDER_SO:       /* This is not supported for NE modules */
 924                 case MODULE_LOADORDER_INVALID:  /* We ignore this as it is an empty entry */
 925                         hinst = 2;
 926                         break;
 927                 }
 928
 929                 if(hinst >= 32)
 930                 {
 931                         if(!implicit)
 932                         {
 933                                 HMODULE16 hModule;
 934                                 NE_MODULE *pModule;
 935
 936                                 hModule = GetModuleHandle16(libname);
 937                                 if(!hModule)
 938                                 {
 939                                         ERR("Serious trouble. Just loaded module '%s' (hinst=0x%04x), but can't get module handle. Filename too long ?\n",
 940                                                 libname, hinst);
 941                                         return 6;       /* ERROR_INVALID_HANDLE seems most appropriate */
 942                                 }
 943
 944                                 pModule = NE_GetPtr(hModule);
 945                                 if(!pModule)
 946                                 {
 947                                         ERR("Serious trouble. Just loaded module '%s' (hinst=0x%04x), but can't get NE_MODULE pointer\n",
 948                                                 libname, hinst);
 949                                         return 6;       /* ERROR_INVALID_HANDLE seems most appropriate */
 950                                 }
 951
 952                                 TRACE("Loaded module '%s' at 0x%04x.\n", libname, hinst);
 953                                 if (!TRACE_ON(module))
 954                                     TRACE_(loaddll)("Loaded module '%s' : %s\n", libname, filetype);
 955
 956                                 /*
 957                                  * Call initialization routines for all loaded DLLs. Note that
 958                                  * when we load implicitly linked DLLs this will be done by InitTask().
 959                                  */
 960                                 if(pModule->flags & NE_FFLAGS_LIBMODULE)
 961                                 {
 962                                         NE_InitializeDLLs(hModule);
 963                                         NE_DllProcessAttach(hModule);
 964                                 }
 965                         }
 966                         return hinst;
 967                 }
 968
 969                 if(hinst != 2)
 970                 {
 971                         /* We quit searching when we get another error than 'File not found' */
 972                         break;
 973                 }
 974         }
 975         return hinst;   /* The last error that occured */
 976 }
 977
 978
 979 /**********************************************************************
 980  *          NE_CreateThread
 981  *
 982  * Create the thread for a 16-bit module.
 983  */
 984 static HINSTANCE16 NE_CreateThread( NE_MODULE *pModule, WORD cmdShow, LPCSTR cmdline )
 985 {
 986     TEB *teb = NULL;
 987     HANDLE hThread = 0;
 988     int socket = -1;
 989     HTASK hTask;
 990     TDB *pTask;
 991     HINSTANCE16 instance = 0;
 992
 993     SERVER_START_REQ
 994     {
 995         struct new_thread_request *req = server_alloc_req( sizeof(*req), 0 );
 996         req->suspend = 0;
 997         req->inherit = 0;
 998         if (!server_call( REQ_NEW_THREAD ))
 999         {
1000             hThread = req->handle;
1001             socket = wine_server_recv_fd( hThread, 0 );
1002         }
1003     }
1004     SERVER_END_REQ;
1005     if (!hThread) return 0;
1006
1007     if (!(teb = THREAD_Create( socket, 0, FALSE ))) goto error;
1008     teb->tibflags &= ~TEBF_WIN32;
1009     teb->startup = NE_InitProcess;
1010
1011     /* Create a task for this process */
1012
1013     if (!TASK_Create( pModule, cmdShow, teb, cmdline + 1, *cmdline )) goto error;
1014     hTask = teb->htask16;
1015     if (SYSDEPS_SpawnThread( teb ) == -1) goto error;
1016
1017     /* Post event to start the task */
1018     PostEvent16( hTask );
1019
1020     /* Wait until we get the instance handle */
1021     do
1022     {
1023         DirectedYield16( hTask );
1024         if (!IsTask16( hTask ))  /* thread has died */
1025         {
1026             DWORD exit_code;
1027             WaitForSingleObject( hThread, INFINITE );
1028             GetExitCodeThread( hThread, &exit_code );
1029             CloseHandle( hThread );
1030             return exit_code;
1031         }
1032         if (!(pTask = (TDB *)GlobalLock16( hTask ))) break;
1033         instance = pTask->hInstance;
1034         GlobalUnlock16( hTask );
1035     } while (!instance);
1036
1037     return instance;
1038
1039  error:
1040     /* FIXME: free TEB and task */
1041     close( socket );
1042     CloseHandle( hThread );
1043     return 0;  /* FIXME */
1044 }
1045
1046
1047 /**********************************************************************
1048  *          LoadModule16    (KERNEL.45)
1049  */
1050 HINSTANCE16 WINAPI LoadModule16( LPCSTR name, LPVOID paramBlock )
1051 {
1052     BOOL lib_only = !paramBlock || (paramBlock == (LPVOID)-1);
1053     LOADPARAMS16 *params;
1054     HMODULE16 hModule;
1055     NE_MODULE *pModule;
1056     LPSTR cmdline;
1057     WORD cmdShow;
1058
1059     /* Load module */
1060
1061     if ( (hModule = NE_GetModuleByFilename(name) ) != 0 )
1062     {
1063         /* Special case: second instance of an already loaded NE module */
1064
1065         if ( !( pModule = NE_GetPtr( hModule ) ) ) return (HINSTANCE16)11;
1066         if ( pModule->module32 ) return (HINSTANCE16)21;
1067
1068         /* Increment refcount */
1069
1070         pModule->count++;
1071     }
1072     else
1073     {
1074         /* Main case: load first instance of NE module */
1075
1076         if ( (hModule = MODULE_LoadModule16( name, FALSE, lib_only )) < 32 )
1077             return hModule;
1078
1079         if ( !(pModule = NE_GetPtr( hModule )) )
1080             return (HINSTANCE16)11;
1081     }
1082
1083     /* If library module, we just retrieve the instance handle */
1084
1085     if ( ( pModule->flags & NE_FFLAGS_LIBMODULE ) || lib_only )
1086         return NE_GetInstance( pModule );
1087
1088     /*
1089      *  At this point, we need to create a new process.
1090      *
1091      *  pModule points either to an already loaded module, whose refcount
1092      *  has already been incremented (to avoid having the module vanish
1093      *  in the meantime), or else to a stub module which contains only header
1094      *  information.
1095      */
1096     params = (LOADPARAMS16 *)paramBlock;
1097     cmdShow = ((WORD *)MapSL(params->showCmd))[1];
1098     cmdline = MapSL( params->cmdLine );
1099     return NE_CreateThread( pModule, cmdShow, cmdline );
1100 }
1101
1102
1103 /**********************************************************************
1104  *          NE_StartMain
1105  *
1106  * Start the main NE task.
1107  */
1108 HINSTANCE16 NE_StartMain( LPCSTR name, HANDLE file )
1109 {
1110     STARTUPINFOA info;
1111     HMODULE16 hModule;
1112     NE_MODULE *pModule;
1113     INT len;
1114     LPSTR pCmdLine, cmdline = GetCommandLineA();
1115
1116     if ((hModule = NE_LoadExeHeader( file, name )) < 32) return hModule;
1117
1118     if (!(pModule = NE_GetPtr( hModule ))) return (HINSTANCE16)11;
1119     if (pModule->flags & NE_FFLAGS_LIBMODULE)
1120     {
1121         MESSAGE( "%s is not a valid Win16 executable\n", name );
1122         ExitProcess( ERROR_BAD_EXE_FORMAT );
1123     }
1124
1125     while (*cmdline && *cmdline != ' ') cmdline++;
1126     if (*cmdline) cmdline++;
1127     len = strlen(cmdline);
1128     pCmdLine = HeapAlloc(GetProcessHeap(), 0, len+2);
1129     if (pCmdLine)
1130     {
1131         strcpy(pCmdLine+1, cmdline);
1132         *pCmdLine = len;
1133     }
1134     GetStartupInfoA( &info );
1135     if (!(info.dwFlags & STARTF_USESHOWWINDOW)) info.wShowWindow = 1;
1136
1137     return NE_CreateThread( pModule, info.wShowWindow, pCmdLine );
1138 }
1139
1140
1141 /**********************************************************************
1142  *          NE_InitProcess
1143  */
1144 static void NE_InitProcess(void)
1145 {
1146     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
1147     NE_MODULE *pModule = NE_GetPtr( pTask->hModule );
1148     HINSTANCE16 hInstance, hPrevInstance;
1149     SEGTABLEENTRY *pSegTable = NE_SEG_TABLE( pModule );
1150     WORD sp;
1151
1152     _EnterWin16Lock();
1153
1154     if ( pModule->count > 0 )
1155     {
1156         /* Second instance of an already loaded NE module */
1157         /* Note that the refcount was already incremented by the parent */
1158
1159         hPrevInstance = NE_GetInstance( pModule );
1160
1161         if ( pModule->dgroup )
1162             if ( NE_CreateSegment( pModule, pModule->dgroup ) )
1163                 NE_LoadSegment( pModule, pModule->dgroup );
1164
1165         hInstance = NE_GetInstance( pModule );
1166         TRACE("created second instance %04x[%d] of instance %04x.\n", hInstance, pModule->dgroup, hPrevInstance);
1167
1168     }
1169     else
1170     {
1171         /* Load first instance of NE module */
1172
1173         pModule->flags |= NE_FFLAGS_GUI;  /* FIXME: is this necessary? */
1174
1175         hInstance = NE_DoLoadModule( pModule );
1176         hPrevInstance = 0;
1177     }
1178
1179     if ( hInstance >= 32 )
1180     {
1181         CONTEXT86 context;
1182
1183         /* Enter instance handles into task struct */
1184
1185         pTask->hInstance = hInstance;
1186         pTask->hPrevInstance = hPrevInstance;
1187
1188         /* Use DGROUP for 16-bit stack */
1189
1190         if (!(sp = pModule->sp))
1191             sp = pSegTable[pModule->ss-1].minsize + pModule->stack_size;
1192         sp &= ~1;
1193         sp -= sizeof(STACK16FRAME);
1194         pTask->teb->cur_stack = MAKESEGPTR( GlobalHandleToSel16(hInstance), sp );
1195
1196         /* Registers at initialization must be:
1197          * ax   zero
1198          * bx   stack size in bytes
1199          * cx   heap size in bytes
1200          * si   previous app instance
1201          * di   current app instance
1202          * bp   zero
1203          * es   selector to the PSP
1204          * ds   dgroup of the application
1205          * ss   stack selector
1206          * sp   top of the stack
1207          */
1208         memset( &context, 0, sizeof(context) );
1209         context.SegCs  = GlobalHandleToSel16(pSegTable[pModule->cs - 1].hSeg);
1210         context.SegDs  = GlobalHandleToSel16(pTask->hInstance);
1211         context.SegEs  = pTask->hPDB;
1212         context.Eip    = pModule->ip;
1213         context.Ebx    = pModule->stack_size;
1214         context.Ecx    = pModule->heap_size;
1215         context.Edi    = pTask->hInstance;
1216         context.Esi    = pTask->hPrevInstance;
1217
1218         /* Now call 16-bit entry point */
1219
1220         TRACE("Starting main program: cs:ip=%04lx:%04lx ds=%04lx ss:sp=%04x:%04x\n",
1221               context.SegCs, context.Eip, context.SegDs,
1222               SELECTOROF(pTask->teb->cur_stack),
1223               OFFSETOF(pTask->teb->cur_stack) );
1224
1225         wine_call_to_16_regs_short( &context, 0 );
1226         ExitThread( LOWORD(context.Eax) );
1227     }
1228
1229     _LeaveWin16Lock();
1230     ExitThread( hInstance );
1231 }
1232
1233 /***********************************************************************
1234  *           LoadLibrary16   (KERNEL.95)
1235  */
1236 HINSTANCE16 WINAPI LoadLibrary16( LPCSTR libname )
1237 {
1238     return LoadModule16(libname, (LPVOID)-1 );
1239 }
1240
1241
1242 /**********************************************************************
1243  *          MODULE_CallWEP
1244  *
1245  * Call a DLL's WEP, allowing it to shut down.
1246  * FIXME: we always pass the WEP WEP_FREE_DLL, never WEP_SYSTEM_EXIT
1247  */
1248 static BOOL16 MODULE_CallWEP( HMODULE16 hModule )
1249 {
1250     FARPROC16 WEP = GetProcAddress16( hModule, "WEP" );
1251     if (!WEP) return FALSE;
1252     return NE_CallTo16_word_w( WEP, WEP_FREE_DLL );
1253 }
1254
1255
1256 /**********************************************************************
1257  *          NE_FreeModule
1258  *
1259  * Implementation of FreeModule16().
1260  */
1261 static BOOL16 NE_FreeModule( HMODULE16 hModule, BOOL call_wep )
1262 {
1263     HMODULE16 *hPrevModule;
1264     NE_MODULE *pModule;
1265     HMODULE16 *pModRef;
1266     int i;
1267
1268     if (!(pModule = NE_GetPtr( hModule ))) return FALSE;
1269     hModule = pModule->self;
1270
1271     TRACE("%04x count %d\n", hModule, pModule->count );
1272
1273     if (((INT16)(--pModule->count)) > 0 ) return TRUE;
1274     else pModule->count = 0;
1275
1276     if (pModule->flags & NE_FFLAGS_BUILTIN)
1277         return FALSE;  /* Can't free built-in module */
1278
1279     if (call_wep && !(pModule->flags & NE_FFLAGS_WIN32))
1280     {
1281         if (pModule->flags & NE_FFLAGS_LIBMODULE)
1282         {
1283             MODULE_CallWEP( hModule );
1284
1285             /* Free the objects owned by the DLL module */
1286             TASK_CallTaskSignalProc( USIG16_DLL_UNLOAD, hModule );
1287             PROCESS_CallUserSignalProc( USIG_DLL_UNLOAD_WIN16, hModule );
1288         }
1289         else
1290             call_wep = FALSE;  /* We are freeing a task -> no more WEPs */
1291     }
1292
1293
1294     /* Clear magic number just in case */
1295
1296     pModule->magic = pModule->self = 0;
1297
1298       /* Remove it from the linked list */
1299
1300     hPrevModule = &hFirstModule;
1301     while (*hPrevModule && (*hPrevModule != hModule))
1302     {
1303         hPrevModule = &(NE_GetPtr( *hPrevModule ))->next;
1304     }
1305     if (*hPrevModule) *hPrevModule = pModule->next;
1306
1307     /* Free the referenced modules */
1308
1309     pModRef = (HMODULE16*)NE_MODULE_TABLE( pModule );
1310     for (i = 0; i < pModule->modref_count; i++, pModRef++)
1311     {
1312         NE_FreeModule( *pModRef, call_wep );
1313     }
1314
1315     /* Free the module storage */
1316
1317     GlobalFreeAll16( hModule );
1318
1319     /* Remove module from cache */
1320
1321     if (pCachedModule == pModule) pCachedModule = NULL;
1322     return TRUE;
1323 }
1324
1325
1326 /**********************************************************************
1327  *          FreeModule16    (KERNEL.46)
1328  */
1329 BOOL16 WINAPI FreeModule16( HMODULE16 hModule )
1330 {
1331     return NE_FreeModule( hModule, TRUE );
1332 }
1333
1334
1335 /***********************************************************************
1336  *           FreeLibrary16   (KERNEL.96)
1337  */
1338 void WINAPI FreeLibrary16( HINSTANCE16 handle )
1339 {
1340     TRACE("%04x\n", handle );
1341     FreeModule16( handle );
1342 }
1343
1344
1345 /**********************************************************************
1346  *          GetModuleName    (KERNEL.27)
1347  */
1348 BOOL16 WINAPI GetModuleName16( HINSTANCE16 hinst, LPSTR buf, INT16 count )
1349 {
1350     NE_MODULE *pModule;
1351     BYTE *p;
1352
1353     if (!(pModule = NE_GetPtr( hinst ))) return FALSE;
1354     p = (BYTE *)pModule + pModule->name_table;
1355     if (count > *p) count = *p + 1;
1356     if (count > 0)
1357     {
1358         memcpy( buf, p + 1, count - 1 );
1359         buf[count-1] = '\0';
1360     }
1361     return TRUE;
1362 }
1363
1364
1365 /**********************************************************************
1366  *          GetModuleUsage    (KERNEL.48)
1367  */
1368 INT16 WINAPI GetModuleUsage16( HINSTANCE16 hModule )
1369 {
1370     NE_MODULE *pModule = NE_GetPtr( hModule );
1371     return pModule ? pModule->count : 0;
1372 }
1373
1374
1375 /**********************************************************************
1376  *          GetExpWinVer    (KERNEL.167)
1377  */
1378 WORD WINAPI GetExpWinVer16( HMODULE16 hModule )
1379 {
1380     NE_MODULE *pModule = NE_GetPtr( hModule );
1381     if ( !pModule ) return 0;
1382
1383     /*
1384      * For built-in modules, fake the expected version the module should
1385      * have according to the Windows version emulated by Wine
1386      */
1387     if ( !pModule->expected_version )
1388     {
1389         OSVERSIONINFOA versionInfo;
1390         versionInfo.dwOSVersionInfoSize = sizeof(versionInfo);
1391
1392         if ( GetVersionExA( &versionInfo ) )
1393             pModule->expected_version =
1394                      (versionInfo.dwMajorVersion & 0xff) << 8
1395                    | (versionInfo.dwMinorVersion & 0xff);
1396     }
1397
1398     return pModule->expected_version;
1399 }
1400
1401
1402 /**********************************************************************
1403  *          GetModuleFileName16    (KERNEL.49)
1404  *
1405  * Comment: see GetModuleFileNameA
1406  *
1407  * Even if invoked by second instance of a program,
1408  * it still returns path of first one.
1409  */
1410 INT16 WINAPI GetModuleFileName16( HINSTANCE16 hModule, LPSTR lpFileName,
1411                                   INT16 nSize )
1412 {
1413     NE_MODULE *pModule;
1414
1415     /* Win95 does not query hModule if set to 0 !
1416      * Is this wrong or maybe Win3.1 only ? */
1417     if (!hModule) hModule = GetCurrentTask();
1418
1419     if (!(pModule = NE_GetPtr( hModule ))) return 0;
1420     lstrcpynA( lpFileName, NE_MODULE_NAME(pModule), nSize );
1421     if (pModule->expected_version >= 0x400)
1422         GetLongPathNameA(NE_MODULE_NAME(pModule), lpFileName, nSize);
1423     TRACE("%04x -> '%s'\n", hModule, lpFileName );
1424     return strlen(lpFileName);
1425 }
1426
1427
1428 /**********************************************************************
1429  *          GetModuleHandle    (KERNEL.47)
1430  *
1431  * Find a module from a module name.
1432  *
1433  * NOTE: The current implementation works the same way the Windows 95 one
1434  *       does. Do not try to 'fix' it, fix the callers.
1435  *       + It does not do ANY extension handling (except that strange .EXE bit)!
1436  *       + It does not care about paths, just about basenames. (same as Windows)
1437  *
1438  * RETURNS
1439  *   LOWORD:
1440  *      the win16 module handle if found
1441  *      0 if not
1442  *   HIWORD (undocumented, see "Undocumented Windows", chapter 5):
1443  *      Always hFirstModule
1444  */
1445 DWORD WINAPI WIN16_GetModuleHandle( SEGPTR name )
1446 {
1447     if (HIWORD(name) == 0)
1448         return MAKELONG(GetExePtr( (HINSTANCE16)name), hFirstModule );
1449     return MAKELONG(GetModuleHandle16( MapSL(name)), hFirstModule );
1450 }
1451
1452 /***********************************************************************
1453  *              GetModuleHandle16   (KERNEL but also exported from KERNEL32 in Wine)
1454  */
1455 HMODULE16 WINAPI GetModuleHandle16( LPCSTR name )
1456 {
1457     HMODULE16   hModule = hFirstModule;
1458     LPSTR       s;
1459     BYTE        len, *name_table;
1460     char        tmpstr[MAX_PATH];
1461     NE_MODULE *pModule;
1462
1463     TRACE("(%s)\n", name);
1464
1465     if (!HIWORD(name))
1466         return GetExePtr(LOWORD(name));
1467
1468     len = strlen(name);
1469     if (!len)
1470         return 0;
1471
1472     lstrcpynA(tmpstr, name, sizeof(tmpstr));
1473
1474     /* If 'name' matches exactly the module name of a module:
1475      * Return its handle.
1476      */
1477     for (hModule = hFirstModule; hModule ; hModule = pModule->next)
1478     {
1479         pModule = NE_GetPtr( hModule );
1480         if (!pModule) break;
1481         if (pModule->flags & NE_FFLAGS_WIN32) continue;
1482
1483         name_table = (BYTE *)pModule + pModule->name_table;
1484         if ((*name_table == len) && !strncmp(name, name_table+1, len))
1485             return hModule;
1486     }
1487
1488     /* If uppercased 'name' matches exactly the module name of a module:
1489      * Return its handle
1490      */
1491     for (s = tmpstr; *s; s++) *s = FILE_toupper(*s);
1492
1493     for (hModule = hFirstModule; hModule ; hModule = pModule->next)
1494     {
1495         pModule = NE_GetPtr( hModule );
1496         if (!pModule) break;
1497         if (pModule->flags & NE_FFLAGS_WIN32) continue;
1498
1499         name_table = (BYTE *)pModule + pModule->name_table;
1500         /* FIXME: the strncasecmp is WRONG. It should not be case insensitive,
1501          * but case sensitive! (Unfortunately Winword 6 and subdlls have
1502          * lowercased module names, but try to load uppercase DLLs, so this
1503          * 'i' compare is just a quickfix until the loader handles that
1504          * correctly. -MM 990705
1505          */
1506         if ((*name_table == len) && !FILE_strncasecmp(tmpstr, name_table+1, len))
1507             return hModule;
1508     }
1509
1510     /* If the base filename of 'name' matches the base filename of the module
1511      * filename of some module (case-insensitive compare):
1512      * Return its handle.
1513      */
1514
1515     /* basename: search backwards in passed name to \ / or : */
1516     s = tmpstr + strlen(tmpstr);
1517     while (s > tmpstr)
1518     {
1519         if (s[-1]=='/' || s[-1]=='\\' || s[-1]==':')
1520                 break;
1521         s--;
1522     }
1523
1524     /* search this in loaded filename list */
1525     for (hModule = hFirstModule; hModule ; hModule = pModule->next)
1526     {
1527         char            *loadedfn;
1528         OFSTRUCT        *ofs;
1529
1530         pModule = NE_GetPtr( hModule );
1531         if (!pModule) break;
1532         if (!pModule->fileinfo) continue;
1533         if (pModule->flags & NE_FFLAGS_WIN32) continue;
1534
1535         ofs = (OFSTRUCT*)((BYTE *)pModule + pModule->fileinfo);
1536         loadedfn = ((char*)ofs->szPathName) + strlen(ofs->szPathName);
1537         /* basename: search backwards in pathname to \ / or : */
1538         while (loadedfn > (char*)ofs->szPathName)
1539         {
1540             if (loadedfn[-1]=='/' || loadedfn[-1]=='\\' || loadedfn[-1]==':')
1541                     break;
1542             loadedfn--;
1543         }
1544         /* case insensitive compare ... */
1545         if (!FILE_strcasecmp(loadedfn, s))
1546             return hModule;
1547     }
1548
1549     /* If the extension of 'name' is '.EXE' and the base filename of 'name'
1550      * matches the base filename of the module filename of some 32-bit module:
1551      * Return the corresponding 16-bit dummy module handle.
1552      */
1553     if (len >= 4 && !FILE_strcasecmp(name+len-4, ".EXE"))
1554     {
1555         HMODULE hModule = GetModuleHandleA( name );
1556         if ( hModule )
1557             return MapHModuleLS( hModule );
1558     }
1559
1560     if (!strcmp(tmpstr,"MSDOS"))
1561         return 1;
1562
1563     if (!strcmp(tmpstr,"TIMER"))
1564     {
1565         FIXME("Eh... Should return caller's code segment, expect crash\n");
1566         return 0;
1567     }
1568
1569     return 0;
1570 }
1571
1572 /**********************************************************************
1573  *          NE_GetModuleByFilename
1574  */
1575 static HMODULE16 NE_GetModuleByFilename( LPCSTR name )
1576 {
1577     HMODULE16   hModule;
1578     LPSTR       s, p;
1579     BYTE        len, *name_table;
1580     char        tmpstr[MAX_PATH];
1581     NE_MODULE *pModule;
1582
1583     lstrcpynA(tmpstr, name, sizeof(tmpstr));
1584
1585     /* If the base filename of 'name' matches the base filename of the module
1586      * filename of some module (case-insensitive compare):
1587      * Return its handle.
1588      */
1589
1590     /* basename: search backwards in passed name to \ / or : */
1591     s = tmpstr + strlen(tmpstr);
1592     while (s > tmpstr)
1593     {
1594         if (s[-1]=='/' || s[-1]=='\\' || s[-1]==':')
1595                 break;
1596         s--;
1597     }
1598
1599     /* search this in loaded filename list */
1600     for (hModule = hFirstModule; hModule ; hModule = pModule->next)
1601     {
1602         char            *loadedfn;
1603         OFSTRUCT        *ofs;
1604
1605         pModule = NE_GetPtr( hModule );
1606         if (!pModule) break;
1607         if (!pModule->fileinfo) continue;
1608         if (pModule->flags & NE_FFLAGS_WIN32) continue;
1609
1610         ofs = (OFSTRUCT*)((BYTE *)pModule + pModule->fileinfo);
1611         loadedfn = ((char*)ofs->szPathName) + strlen(ofs->szPathName);
1612         /* basename: search backwards in pathname to \ / or : */
1613         while (loadedfn > (char*)ofs->szPathName)
1614         {
1615             if (loadedfn[-1]=='/' || loadedfn[-1]=='\\' || loadedfn[-1]==':')
1616                     break;
1617             loadedfn--;
1618         }
1619         /* case insensitive compare ... */
1620         if (!FILE_strcasecmp(loadedfn, s))
1621             return hModule;
1622     }
1623     /* If basename (without ext) matches the module name of a module:
1624      * Return its handle.
1625      */
1626
1627     if ( (p = strrchr( s, '.' )) != NULL ) *p = '\0';
1628     len = strlen(s);
1629
1630     for (hModule = hFirstModule; hModule ; hModule = pModule->next)
1631     {
1632         pModule = NE_GetPtr( hModule );
1633         if (!pModule) break;
1634         if (pModule->flags & NE_FFLAGS_WIN32) continue;
1635
1636         name_table = (BYTE *)pModule + pModule->name_table;
1637         if ((*name_table == len) && !FILE_strncasecmp(s, name_table+1, len))
1638             return hModule;
1639     }
1640
1641     return 0;
1642 }
1643
1644 /**********************************************************************
1645  *          ModuleFirst    (TOOLHELP.59)
1646  */
1647 BOOL16 WINAPI ModuleFirst16( MODULEENTRY *lpme )
1648 {
1649     lpme->wNext = hFirstModule;
1650     return ModuleNext16( lpme );
1651 }
1652
1653
1654 /**********************************************************************
1655  *          ModuleNext    (TOOLHELP.60)
1656  */
1657 BOOL16 WINAPI ModuleNext16( MODULEENTRY *lpme )
1658 {
1659     NE_MODULE *pModule;
1660     char *name;
1661
1662     if (!lpme->wNext) return FALSE;
1663     if (!(pModule = NE_GetPtr( lpme->wNext ))) return FALSE;
1664     name = (char *)pModule + pModule->name_table;
1665     memcpy( lpme->szModule, name + 1, min(*name, MAX_MODULE_NAME) );
1666     lpme->szModule[min(*name, MAX_MODULE_NAME)] = '\0';
1667     lpme->hModule = lpme->wNext;
1668     lpme->wcUsage = pModule->count;
1669     lstrcpynA( lpme->szExePath, NE_MODULE_NAME(pModule), sizeof(lpme->szExePath) );
1670     lpme->wNext = pModule->next;
1671     return TRUE;
1672 }
1673
1674
1675 /**********************************************************************
1676  *          ModuleFindName    (TOOLHELP.61)
1677  */
1678 BOOL16 WINAPI ModuleFindName16( MODULEENTRY *lpme, LPCSTR name )
1679 {
1680     lpme->wNext = GetModuleHandle16( name );
1681     return ModuleNext16( lpme );
1682 }
1683
1684
1685 /**********************************************************************
1686  *          ModuleFindHandle    (TOOLHELP.62)
1687  */
1688 BOOL16 WINAPI ModuleFindHandle16( MODULEENTRY *lpme, HMODULE16 hModule )
1689 {
1690     hModule = GetExePtr( hModule );
1691     lpme->wNext = hModule;
1692     return ModuleNext16( lpme );
1693 }
1694
1695
1696 /***************************************************************************
1697  *          IsRomModule16    (KERNEL.323)
1698  */
1699 BOOL16 WINAPI IsRomModule16( HMODULE16 unused )
1700 {
1701     return FALSE;
1702 }
1703
1704 /***************************************************************************
1705  *          IsRomFile16    (KERNEL.326)
1706  */
1707 BOOL16 WINAPI IsRomFile16( HFILE16 unused )
1708 {
1709     return FALSE;
1710 }
1711
1712 /***************************************************************************
1713  *              MapHModuleLS                    (KERNEL32.520)
1714  */
1715 HMODULE16 WINAPI MapHModuleLS(HMODULE hmod) {
1716         NE_MODULE       *pModule;
1717
1718         if (!hmod)
1719                 return ((TDB*)GlobalLock16(GetCurrentTask()))->hInstance;
1720         if (!HIWORD(hmod))
1721                 return hmod; /* we already have a 16 bit module handle */
1722         pModule = (NE_MODULE*)GlobalLock16(hFirstModule);
1723         while (pModule)  {
1724                 if (pModule->module32 == hmod)
1725                         return pModule->self;
1726                 pModule = (NE_MODULE*)GlobalLock16(pModule->next);
1727         }
1728         return 0;
1729 }
1730
1731 /***************************************************************************
1732  *              MapHModuleSL                    (KERNEL32.521)
1733  */
1734 HMODULE WINAPI MapHModuleSL(HMODULE16 hmod) {
1735         NE_MODULE       *pModule;
1736
1737         if (!hmod) {
1738                 TDB *pTask = (TDB*)GlobalLock16(GetCurrentTask());
1739
1740                 hmod = pTask->hModule;
1741         }
1742         pModule = (NE_MODULE*)GlobalLock16(hmod);
1743         if (    (pModule->magic!=IMAGE_OS2_SIGNATURE)   ||
1744                 !(pModule->flags & NE_FFLAGS_WIN32)
1745         )
1746                 return 0;
1747         return pModule->module32;
1748 }
1749
1750 /***************************************************************************
1751  *              MapHInstLS                      (KERNEL32.516)(KERNEL.472)
1752  */
1753 void WINAPI MapHInstLS( CONTEXT86 *context )
1754 {
1755     context->Eax = MapHModuleLS(context->Eax);
1756 }
1757
1758 /***************************************************************************
1759  *              MapHInstSL                      (KERNEL32.518)(KERNEL.473)
1760  */
1761 void WINAPI MapHInstSL( CONTEXT86 *context )
1762 {
1763     context->Eax = MapHModuleSL(context->Eax);
1764 }
1765
1766 /***************************************************************************
1767  *              MapHInstLS_PN                   (KERNEL32.517)
1768  */
1769 void WINAPI MapHInstLS_PN( CONTEXT86 *context )
1770 {
1771     if (context->Eax) context->Eax = MapHModuleLS(context->Eax);
1772 }
1773
1774 /***************************************************************************
1775  *              MapHInstSL_PN                   (KERNEL32.519)
1776  */
1777 void WINAPI MapHInstSL_PN( CONTEXT86 *context )
1778 {
1779     if (context->Eax) context->Eax = MapHModuleSL(context->Eax);
1780 }
1781