pe_image.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:32k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*
  2.  * $Id: e6cb36bb78e268c6795f68deddf94be469b564ef $
  3.  *
  4.  *  Copyright 1994 Eric Youndale & Erik Bos
  5.  *  Copyright 1995 Martin von Löwis
  6.  *  Copyright   1996-98 Marcus Meissner
  7.  *
  8.  * based on Eric Youndale's pe-test and:
  9.  *
  10.  * ftp.microsoft.com:/pub/developer/MSDN/CD8/PEFILE.ZIP
  11.  * make that:
  12.  * ftp.microsoft.com:/developr/MSDN/OctCD/PEFILE.ZIP
  13.  *
  14.  * Modified for use with MPlayer, detailed CVS changelog at
  15.  * http://www.mplayerhq.hu/cgi-bin/cvsweb.cgi/main/
  16.  *
  17.  * File now distributed as part of VLC media player with no modifications.
  18.  *
  19.  * This program is free software; you can redistribute it and/or modify
  20.  * it under the terms of the GNU General Public License as published by
  21.  * the Free Software Foundation; either version 2 of the License, or
  22.  * (at your option) any later version.
  23.  *
  24.  * This program is distributed in the hope that it will be useful,
  25.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  26.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27.  * GNU General Public License for more details.
  28.  *
  29.  * You should have received a copy of the GNU General Public License
  30.  * along with this program; if not, write to the Free Software
  31.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  32.  */
  33. /* Notes:
  34.  * Before you start changing something in this file be aware of the following:
  35.  *
  36.  * - There are several functions called recursively. In a very subtle and 
  37.  *   obscure way. DLLs can reference each other recursively etc.
  38.  * - If you want to enhance, speed up or clean up something in here, think
  39.  *   twice WHY it is implemented in that strange way. There is usually a reason.
  40.  *   Though sometimes it might just be lazyness ;)
  41.  * - In PE_MapImage, right before fixup_imports() all external and internal 
  42.  *   state MUST be correct since this function can be called with the SAME image
  43.  *   AGAIN. (Thats recursion for you.) That means MODREF.module and
  44.  *   NE_MODULE.module32.
  45.  * - Sometimes, we can't use Linux mmap() to mmap() the images directly.
  46.  *
  47.  *   The problem is, that there is not direct 1:1 mapping from a diskimage and
  48.  *   a memoryimage. The headers at the start are mapped linear, but the sections
  49.  *   are not. Older x86 pe binaries are 512 byte aligned in file and 4096 byte
  50.  *   aligned in memory. Linux likes them 4096 byte aligned in memory (due to
  51.  *   x86 pagesize, this cannot be fixed without a rather large kernel rewrite)
  52.  *   and 'blocksize' file-aligned (offsets). Since we have 512/1024/2048 (CDROM)
  53.  *   and other byte blocksizes, we can't always do this.  We *can* do this for
  54.  *   newer pe binaries produced by MSVC 5 and later, since they are also aligned
  55.  *   to 4096 byte boundaries on disk.
  56.  */
  57. #include "config.h"
  58. #include <errno.h>
  59. #include <assert.h>
  60. #include <stdio.h>
  61. #include <stdlib.h>
  62. #include <string.h>
  63. #include <unistd.h>
  64. #include <sys/types.h>
  65. #include <sys/stat.h>
  66. #include <fcntl.h>
  67. #ifdef HAVE_SYS_MMAN_H
  68. #include <sys/mman.h>
  69. #endif
  70. #include "wine/windef.h"
  71. #include "wine/winbase.h"
  72. #include "wine/winerror.h"
  73. #include "wine/heap.h"
  74. #include "wine/pe_image.h"
  75. #include "wine/module.h"
  76. #include "wine/debugtools.h"
  77. #include "ext.h"
  78. #include "win32.h"
  79. #define RVA(x) ((void *)((char *)load_addr+(unsigned int)(x)))
  80. #define AdjustPtr(ptr,delta) ((char *)(ptr) + (delta))
  81. extern void* LookupExternal(const char* library, int ordinal);
  82. extern void* LookupExternalByName(const char* library, const char* name);
  83. static void dump_exports( HMODULE hModule )
  84.   char *Module;
  85.   unsigned int i, j;
  86.   u_short *ordinal;
  87.   u_long *function,*functions;
  88.   u_char **name;
  89.   unsigned int load_addr = hModule;
  90.   DWORD rva_start = PE_HEADER(hModule)->OptionalHeader
  91.                    .DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
  92.   DWORD rva_end = rva_start + PE_HEADER(hModule)->OptionalHeader
  93.                    .DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].Size;
  94.   IMAGE_EXPORT_DIRECTORY *pe_exports = (IMAGE_EXPORT_DIRECTORY*)RVA(rva_start);
  95.   Module = (char*)RVA(pe_exports->Name);
  96.   TRACE("*******EXPORT DATA*******n");
  97.   TRACE("Module name is %s, %ld functions, %ld namesn", 
  98.         Module, pe_exports->NumberOfFunctions, pe_exports->NumberOfNames);
  99.   ordinal=(u_short*) RVA(pe_exports->AddressOfNameOrdinals);
  100.   functions=function=(u_long*) RVA(pe_exports->AddressOfFunctions);
  101.   name=(u_char**) RVA(pe_exports->AddressOfNames);
  102.   TRACE(" Ord    RVA     Addr   Namen" );
  103.   for (i=0;i<pe_exports->NumberOfFunctions;i++, function++)
  104.   {
  105.       if (!*function) continue;  
  106.       if (TRACE_ON(win32))
  107.       {
  108. DPRINTF( "%4ld %08lx %p", i + pe_exports->Base, *function, RVA(*function) );
  109. for (j = 0; j < pe_exports->NumberOfNames; j++)
  110.           if (ordinal[j] == i)
  111.           {
  112.               DPRINTF( "  %s", (char*)RVA(name[j]) );
  113.               break;
  114.           }
  115. if ((*function >= rva_start) && (*function <= rva_end))
  116.   DPRINTF(" (forwarded -> %s)", (char *)RVA(*function));
  117. DPRINTF("n");
  118.       }
  119.   }
  120. }
  121. /* Look up the specified function or ordinal in the exportlist:
  122.  * If it is a string:
  123.  *  - look up the name in the Name list. 
  124.  * - look up the ordinal with that index.
  125.  * - use the ordinal as offset into the functionlist
  126.  * If it is a ordinal:
  127.  * - use ordinal-pe_export->Base as offset into the functionlist
  128.  */
  129. FARPROC PE_FindExportedFunction( 
  130. WINE_MODREF *wm,
  131. LPCSTR funcName,
  132.         WIN_BOOL snoop )
  133. {
  134. u_short * ordinals;
  135. u_long * function;
  136. u_char ** name;
  137. const char *ename = NULL;
  138. int i, ordinal;
  139. PE_MODREF *pem = &(wm->binfmt.pe);
  140. IMAGE_EXPORT_DIRECTORY  *exports = pem->pe_export;
  141. unsigned int load_addr = wm->module;
  142. u_long rva_start, rva_end, addr;
  143. char * forward;
  144. if (HIWORD(funcName))
  145. TRACE("(%s)n",funcName);
  146. else
  147. TRACE("(%d)n",(int)funcName);
  148. if (!exports) {
  149. /* Not a fatal problem, some apps do
  150.  * GetProcAddress(0,"RegisterPenApp") which triggers this
  151.  * case.
  152.  */
  153. WARN("Module %08x(%s)/MODREF %p doesn't have a exports table.n",wm->module,wm->modname,pem);
  154. return NULL;
  155. }
  156. ordinals= (u_short*)  RVA(exports->AddressOfNameOrdinals);
  157. function= (u_long*)   RVA(exports->AddressOfFunctions);
  158. name = (u_char **) RVA(exports->AddressOfNames);
  159. forward = NULL;
  160. rva_start = PE_HEADER(wm->module)->OptionalHeader
  161. .DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
  162. rva_end = rva_start + PE_HEADER(wm->module)->OptionalHeader
  163. .DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].Size;
  164. if (HIWORD(funcName))
  165.         {
  166.             
  167.             int min = 0, max = exports->NumberOfNames - 1;
  168.             while (min <= max)
  169.             {
  170.                 int res, pos = (min + max) / 2;
  171. ename = (const char*) RVA(name[pos]);
  172. if (!(res = strcmp( ename, funcName )))
  173.                 {
  174.                     ordinal = ordinals[pos];
  175.                     goto found;
  176.                 }
  177.                 if (res > 0) max = pos - 1;
  178.                 else min = pos + 1;
  179.             }
  180.             
  181.     for (i = 0; i < exports->NumberOfNames; i++)
  182.             {
  183. ename = (const char*) RVA(name[i]);
  184.                 if (!strcmp( ename, funcName ))
  185.                 {
  186.     ERR( "%s.%s required a linear searchn", wm->modname, funcName );
  187.                     ordinal = ordinals[i];
  188.                     goto found;
  189.                 }
  190.             }
  191.             return NULL;
  192. }
  193.         else  
  194.         {
  195.             ordinal = LOWORD(funcName) - exports->Base;
  196.             if (snoop && name)  
  197.             {
  198.                 for (i = 0; i < exports->NumberOfNames; i++)
  199.                     if (ordinals[i] == ordinal)
  200.                     {
  201.                         ename = RVA(name[i]);
  202.                         break;
  203.                     }
  204.             }
  205. }
  206.  found:
  207.         if (ordinal >= exports->NumberOfFunctions)
  208.         {
  209.             TRACE(" ordinal %ld out of range!n", ordinal + exports->Base );
  210.             return NULL;
  211.         }
  212.         addr = function[ordinal];
  213.         if (!addr) return NULL;
  214.         if ((addr < rva_start) || (addr >= rva_end))
  215.         {
  216.             FARPROC proc = RVA(addr);
  217.             if (snoop)
  218.             {
  219.                 if (!ename) ename = "@";
  220. //                proc = SNOOP_GetProcAddress(wm->module,ename,ordinal,proc);
  221. TRACE("SNOOP_GetProcAddress n/an");
  222.             }
  223.             return proc;
  224.         }
  225.         else  
  226.         {
  227.                 WINE_MODREF *wm;
  228.                 char *forward = RVA(addr);
  229. char module[256];
  230. char *end = strchr(forward, '.');
  231. if (!end) return NULL;
  232.                 if (end - forward >= sizeof(module)) return NULL;
  233.                 memcpy( module, forward, end - forward );
  234. module[end-forward] = 0;
  235.                 if (!(wm = MODULE_FindModule( module )))
  236.                 {
  237.                     ERR("module not found for forward '%s'n", forward );
  238.                     return NULL;
  239.                 }
  240. return MODULE_GetProcAddress( wm->module, end + 1, snoop );
  241. }
  242. }
  243. static DWORD fixup_imports( WINE_MODREF *wm )
  244. {
  245.     IMAGE_IMPORT_DESCRIPTOR *pe_imp;
  246.     PE_MODREF *pem;
  247.     unsigned int load_addr = wm->module;
  248.     int i,characteristics_detection=1;
  249.     char *modname;
  250.     
  251.     assert(wm->type==MODULE32_PE);
  252.     pem = &(wm->binfmt.pe);
  253.     if (pem->pe_export)
  254.      modname = (char*) RVA(pem->pe_export->Name);
  255.     else
  256.         modname = "<unknown>";
  257.     
  258.     TRACE("Dumping imports listn");
  259.     
  260.     pe_imp = pem->pe_import;
  261.     if (!pe_imp) return 0;
  262.     /* We assume that we have at least one import with !0 characteristics and
  263.      * detect broken imports with all characteristsics 0 (notably Borland) and
  264.      * switch the detection off for them.
  265.      */
  266.     for (i = 0; pe_imp->Name ; pe_imp++) {
  267. if (!i && !pe_imp->u.Characteristics)
  268. characteristics_detection = 0;
  269. if (characteristics_detection && !pe_imp->u.Characteristics)
  270. break;
  271. i++;
  272.     }
  273.     if (!i) return 0;  
  274.     
  275.     wm->nDeps = i;
  276.     wm->deps  = HeapAlloc( GetProcessHeap(), 0, i*sizeof(WINE_MODREF *) );
  277.     /* load the imported modules. They are automatically 
  278.      * added to the modref list of the process.
  279.      */
  280.  
  281.     for (i = 0, pe_imp = pem->pe_import; pe_imp->Name ; pe_imp++) {
  282.      WINE_MODREF *wmImp;
  283. IMAGE_IMPORT_BY_NAME *pe_name;
  284. PIMAGE_THUNK_DATA import_list,thunk_list;
  285.   char *name = (char *) RVA(pe_imp->Name);
  286. if (characteristics_detection && !pe_imp->u.Characteristics)
  287. break;
  288. //#warning FIXME: here we should fill imports
  289.         TRACE("Loading imports for %s.dlln", name);
  290.     
  291. if (pe_imp->u.OriginalFirstThunk != 0) { 
  292.     TRACE("Microsoft style imports usedn");
  293.     import_list =(PIMAGE_THUNK_DATA) RVA(pe_imp->u.OriginalFirstThunk);
  294.     thunk_list = (PIMAGE_THUNK_DATA) RVA(pe_imp->FirstThunk);
  295.     while (import_list->u1.Ordinal) {
  296. if (IMAGE_SNAP_BY_ORDINAL(import_list->u1.Ordinal)) {
  297.     int ordinal = IMAGE_ORDINAL(import_list->u1.Ordinal);
  298. //     TRACE("--- Ordinal %s,%dn", name, ordinal);
  299.     
  300.     thunk_list->u1.Function=LookupExternal(name, ordinal);
  301. } else {
  302.     pe_name = (PIMAGE_IMPORT_BY_NAME)RVA(import_list->u1.AddressOfData);
  303. //     TRACE("--- %s %s.%dn", pe_name->Name, name, pe_name->Hint);
  304.     thunk_list->u1.Function=LookupExternalByName(name, pe_name->Name);
  305. }
  306. import_list++;
  307. thunk_list++;
  308.     }
  309. } else {
  310.     TRACE("Borland style imports usedn");
  311.     thunk_list = (PIMAGE_THUNK_DATA) RVA(pe_imp->FirstThunk);
  312.     while (thunk_list->u1.Ordinal) {
  313. if (IMAGE_SNAP_BY_ORDINAL(thunk_list->u1.Ordinal)) {
  314.     
  315.     int ordinal = IMAGE_ORDINAL(thunk_list->u1.Ordinal);
  316.     TRACE("--- Ordinal %s.%dn",name,ordinal);
  317.     thunk_list->u1.Function=LookupExternal(
  318.       name, ordinal);
  319. } else {
  320.     pe_name=(PIMAGE_IMPORT_BY_NAME) RVA(thunk_list->u1.AddressOfData);
  321.     TRACE("--- %s %s.%dn",
  322.       pe_name->Name,name,pe_name->Hint);
  323.     thunk_list->u1.Function=LookupExternalByName(
  324.       name, pe_name->Name);
  325. }
  326. thunk_list++;
  327.     }
  328. }
  329.     }
  330.     return 0;
  331. }
  332. static int calc_vma_size( HMODULE hModule )
  333. {
  334.     int i,vma_size = 0;
  335.     IMAGE_SECTION_HEADER *pe_seg = PE_SECTIONS(hModule);
  336.     TRACE("Dump of segment tablen");
  337.     TRACE("   Name    VSz  Vaddr     SzRaw   Fileadr  *Reloc *Lineum #Reloc #Linum Charn");
  338.     for (i = 0; i< PE_HEADER(hModule)->FileHeader.NumberOfSections; i++)
  339.     {
  340.         TRACE("%8s: %4.4lx %8.8lx %8.8lx %8.8lx %8.8lx %8.8lx %4.4x %4.4x %8.8lxn", 
  341.                       pe_seg->Name, 
  342.                       pe_seg->Misc.VirtualSize,
  343.                       pe_seg->VirtualAddress,
  344.                       pe_seg->SizeOfRawData,
  345.                       pe_seg->PointerToRawData,
  346.                       pe_seg->PointerToRelocations,
  347.                       pe_seg->PointerToLinenumbers,
  348.                       pe_seg->NumberOfRelocations,
  349.                       pe_seg->NumberOfLinenumbers,
  350.                       pe_seg->Characteristics);
  351.         vma_size=max(vma_size, pe_seg->VirtualAddress+pe_seg->SizeOfRawData);
  352.         vma_size=max(vma_size, pe_seg->VirtualAddress+pe_seg->Misc.VirtualSize);
  353.         pe_seg++;
  354.     }
  355.     return vma_size;
  356. }
  357. static void do_relocations( unsigned int load_addr, IMAGE_BASE_RELOCATION *r )
  358. {
  359.     int delta = load_addr - PE_HEADER(load_addr)->OptionalHeader.ImageBase;
  360.     int hdelta = (delta >> 16) & 0xFFFF;
  361.     int ldelta = delta & 0xFFFF;
  362. if(delta == 0)
  363. return;
  364. while(r->VirtualAddress)
  365. {
  366. char *page = (char*) RVA(r->VirtualAddress);
  367. int count = (r->SizeOfBlock - 8)/2;
  368. int i;
  369. TRACE_(fixup)("%x relocations for page %lxn",
  370. count, r->VirtualAddress);
  371. for(i=0;i<count;i++)
  372. {
  373. int offset = r->TypeOffset[i] & 0xFFF;
  374. int type = r->TypeOffset[i] >> 12;
  375. // TRACE_(fixup)("patching %x type %xn", offset, type);
  376. switch(type)
  377. {
  378. case IMAGE_REL_BASED_ABSOLUTE: break;
  379. case IMAGE_REL_BASED_HIGH:
  380. *(short*)(page+offset) += hdelta;
  381. break;
  382. case IMAGE_REL_BASED_LOW:
  383. *(short*)(page+offset) += ldelta;
  384. break;
  385. case IMAGE_REL_BASED_HIGHLOW:
  386. *(int*)(page+offset) += delta;
  387. break;
  388. case IMAGE_REL_BASED_HIGHADJ:
  389. FIXME("Don't know what to do with IMAGE_REL_BASED_HIGHADJn");
  390. break;
  391. case IMAGE_REL_BASED_MIPS_JMPADDR:
  392. FIXME("Is this a MIPS machine ???n");
  393. break;
  394. default:
  395. FIXME("Unknown fixup typen");
  396. break;
  397. }
  398. }
  399. r = (IMAGE_BASE_RELOCATION*)((char*)r + r->SizeOfBlock);
  400. }
  401. }
  402. /**********************************************************************
  403.  * PE_LoadImage
  404.  * Load one PE format DLL/EXE into memory
  405.  * 
  406.  * Unluckily we can't just mmap the sections where we want them, for 
  407.  * (at least) Linux does only support offsets which are page-aligned.
  408.  *
  409.  * BUT we have to map the whole image anyway, for Win32 programs sometimes
  410.  * want to access them. (HMODULE32 point to the start of it)
  411.  */
  412. HMODULE PE_LoadImage( int handle, LPCSTR filename, WORD *version )
  413. {
  414.     HMODULE hModule;
  415.     HANDLE mapping;
  416.     IMAGE_NT_HEADERS *nt;
  417.     IMAGE_SECTION_HEADER *pe_sec;
  418.     IMAGE_DATA_DIRECTORY *dir;
  419.     BY_HANDLE_FILE_INFORMATION bhfi;
  420.     int i, rawsize, lowest_va, vma_size, file_size = 0;
  421.     DWORD load_addr = 0, aoep, reloc = 0;
  422. //    struct get_read_fd_request *req = get_req_buffer();
  423.     int unix_handle = handle;
  424.     int page_size = getpagesize();
  425.     
  426. //    if ( GetFileInformationByHandle( hFile, &bhfi ) ) 
  427. //     file_size = bhfi.nFileSizeLow; 
  428.     file_size=lseek(handle, 0, SEEK_END);
  429.     lseek(handle, 0, SEEK_SET);
  430. //#warning fix CreateFileMappingA
  431.     mapping = CreateFileMappingA( handle, NULL, PAGE_READONLY | SEC_COMMIT,
  432.                                     0, 0, NULL );
  433.     if (!mapping)
  434.     {
  435.         WARN("CreateFileMapping error %ldn", GetLastError() );
  436.         return 0;
  437.     }
  438. //    hModule = (HMODULE)MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, 0 );
  439.     hModule=(HMODULE)mapping;
  440. //    CloseHandle( mapping );
  441.     if (!hModule)
  442.     {
  443.         WARN("MapViewOfFile error %ldn", GetLastError() );
  444.         return 0;
  445.     }
  446.     if ( *(WORD*)hModule !=IMAGE_DOS_SIGNATURE)
  447.     {
  448.         WARN("%s image doesn't have DOS signature, but 0x%04xn", filename,*(WORD*)hModule);
  449.         goto error;
  450.     }
  451.     nt = PE_HEADER( hModule );
  452.     
  453.     if ( nt->Signature != IMAGE_NT_SIGNATURE )
  454.     {
  455.         WARN("%s image doesn't have PE signature, but 0x%08lxn", filename, nt->Signature );
  456.         goto error;
  457.     }
  458.     
  459.     if ( nt->FileHeader.Machine != IMAGE_FILE_MACHINE_I386 )
  460.     {
  461.         MESSAGE("Trying to load PE image for unsupported architecture (");
  462.         switch (nt->FileHeader.Machine)
  463.         {
  464.         case IMAGE_FILE_MACHINE_UNKNOWN: MESSAGE("Unknown"); break;
  465.         case IMAGE_FILE_MACHINE_I860:    MESSAGE("I860"); break;
  466.         case IMAGE_FILE_MACHINE_R3000:   MESSAGE("R3000"); break;
  467.         case IMAGE_FILE_MACHINE_R4000:   MESSAGE("R4000"); break;
  468.         case IMAGE_FILE_MACHINE_R10000:  MESSAGE("R10000"); break;
  469.         case IMAGE_FILE_MACHINE_ALPHA:   MESSAGE("Alpha"); break;
  470.         case IMAGE_FILE_MACHINE_POWERPC: MESSAGE("PowerPC"); break;
  471.         default: MESSAGE("Unknown-%04x", nt->FileHeader.Machine); break;
  472.         }
  473.         MESSAGE(")n");
  474.         goto error;
  475.     }
  476.     
  477.     pe_sec = PE_SECTIONS( hModule );
  478.     rawsize = 0; lowest_va = 0x10000;
  479.     for (i = 0; i < nt->FileHeader.NumberOfSections; i++) 
  480.     {
  481.         if (lowest_va > pe_sec[i].VirtualAddress)
  482.            lowest_va = pe_sec[i].VirtualAddress;
  483.      if (pe_sec[i].Characteristics & IMAGE_SCN_CNT_UNINITIALIZED_DATA)
  484.     continue;
  485. if (pe_sec[i].PointerToRawData+pe_sec[i].SizeOfRawData > rawsize)
  486.     rawsize = pe_sec[i].PointerToRawData+pe_sec[i].SizeOfRawData;
  487.     }
  488.  
  489.     
  490.     if ( file_size && file_size < rawsize )
  491.     {
  492.         ERR("PE module is too small (header: %d, filesize: %d), "
  493.                     "probably truncated download?n", 
  494.                     rawsize, file_size );
  495.         goto error;
  496.     }
  497.     
  498.     aoep = nt->OptionalHeader.AddressOfEntryPoint;
  499.     if (aoep && (aoep < lowest_va))
  500.         FIXME("VIRUS WARNING: '%s' has an invalid entrypoint (0x%08lx) "
  501.                       "below the first virtual address (0x%08x) "
  502.                       "(possibly infected by Tchernobyl/SpaceFiller virus)!n",
  503.                        filename, aoep, lowest_va );
  504.     /* FIXME:  Hack!  While we don't really support shared sections yet,
  505.      *         this checks for those special cases where the whole DLL
  506.      *         consists only of shared sections and is mapped into the
  507.      *         shared address space > 2GB.  In this case, we assume that
  508.      *         the module got mapped at its base address. Thus we simply
  509.      *         check whether the module has actually been mapped there
  510.      *         and use it, if so.  This is needed to get Win95 USER32.DLL
  511.      *         to work (until we support shared sections properly).
  512.      */
  513.     if ( nt->OptionalHeader.ImageBase & 0x80000000 )
  514.     {
  515.         HMODULE sharedMod = (HMODULE)nt->OptionalHeader.ImageBase; 
  516.         IMAGE_NT_HEADERS *sharedNt = (PIMAGE_NT_HEADERS)
  517.                ( (LPBYTE)sharedMod + ((LPBYTE)nt - (LPBYTE)hModule) );
  518.         /* Well, this check is not really comprehensive, 
  519.            but should be good enough for now ... */
  520.         if (    !IsBadReadPtr( (LPBYTE)sharedMod, sizeof(IMAGE_DOS_HEADER) )
  521.              && memcmp( (LPBYTE)sharedMod, (LPBYTE)hModule, sizeof(IMAGE_DOS_HEADER) ) == 0
  522.              && !IsBadReadPtr( sharedNt, sizeof(IMAGE_NT_HEADERS) )
  523.              && memcmp( sharedNt, nt, sizeof(IMAGE_NT_HEADERS) ) == 0 )
  524.         {
  525.             UnmapViewOfFile( (LPVOID)hModule );
  526.             return sharedMod;
  527.         }
  528.     }
  529.     
  530.     load_addr = nt->OptionalHeader.ImageBase;
  531.     vma_size = calc_vma_size( hModule );
  532.     load_addr = (DWORD)VirtualAlloc( (void*)load_addr, vma_size,
  533.                                      MEM_RESERVE | MEM_COMMIT,
  534.                                      PAGE_EXECUTE_READWRITE );
  535.     if (load_addr == 0) 
  536.     {
  537.         
  538.         FIXME("We need to perform base relocations for %sn", filename);
  539. dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_BASERELOC;
  540.         if (dir->Size)
  541.             reloc = dir->VirtualAddress;
  542.         else 
  543.         {
  544.             FIXME( "FATAL: Need to relocate %s, but no relocation records present (%s). Try to run that file directly !n",
  545.                    filename,
  546.                    (nt->FileHeader.Characteristics&IMAGE_FILE_RELOCS_STRIPPED)?
  547.                    "stripped during link" : "unknown reason" );
  548.             goto error;
  549.         }
  550.         /* FIXME: If we need to relocate a system DLL (base > 2GB) we should
  551.          *        really make sure that the *new* base address is also > 2GB.
  552.          *        Some DLLs really check the MSB of the module handle :-/
  553.          */
  554.         if ( nt->OptionalHeader.ImageBase & 0x80000000 )
  555.             ERR( "Forced to relocate system DLL (base > 2GB). This is not good.n" );
  556.         load_addr = (DWORD)VirtualAlloc( NULL, vma_size,
  557.  MEM_RESERVE | MEM_COMMIT,
  558.  PAGE_EXECUTE_READWRITE );
  559. if (!load_addr) {
  560.             FIXME_(win32)(
  561.                    "FATAL: Couldn't load module %s (out of memory, %d needed)!n", filename, vma_size);
  562.             goto error;
  563. }
  564.     }
  565.     TRACE("Load addr is %lx (base %lx), range %xn",
  566.           load_addr, nt->OptionalHeader.ImageBase, vma_size );
  567.     TRACE_(segment)("Loading %s at %lx, range %xn",
  568.                     filename, load_addr, vma_size );
  569. #if 0
  570.     
  571.     *(PIMAGE_DOS_HEADER)load_addr = *(PIMAGE_DOS_HEADER)hModule;
  572.     *PE_HEADER( load_addr ) = *nt;
  573.     memcpy( PE_SECTIONS(load_addr), PE_SECTIONS(hModule),
  574.             sizeof(IMAGE_SECTION_HEADER) * nt->FileHeader.NumberOfSections );
  575.     
  576.     memcpy( load_addr, hModule, lowest_fa );
  577. #endif
  578.     if ((void*)FILE_dommap( handle, (void *)load_addr, 0, nt->OptionalHeader.SizeOfHeaders,
  579.                      0, 0, PROT_EXEC | PROT_WRITE | PROT_READ,
  580.                      MAP_PRIVATE | MAP_FIXED ) != (void*)load_addr)
  581.     {
  582.         ERR_(win32)( "Critical Error: failed to map PE header to necessary address.n");
  583.         goto error;
  584.     }
  585.     
  586.     pe_sec = PE_SECTIONS( hModule );
  587.     for (i = 0; i < nt->FileHeader.NumberOfSections; i++, pe_sec++)
  588.     {
  589.         if (!pe_sec->SizeOfRawData || !pe_sec->PointerToRawData) continue;
  590.         TRACE("%s: mmaping section %s at %p off %lx size %lx/%lxn",
  591.               filename, pe_sec->Name, (void*)RVA(pe_sec->VirtualAddress),
  592.               pe_sec->PointerToRawData, pe_sec->SizeOfRawData, pe_sec->Misc.VirtualSize );
  593.         if ((void*)FILE_dommap( unix_handle, (void*)RVA(pe_sec->VirtualAddress),
  594.                          0, pe_sec->SizeOfRawData, 0, pe_sec->PointerToRawData,
  595.                          PROT_EXEC | PROT_WRITE | PROT_READ,
  596.                          MAP_PRIVATE | MAP_FIXED ) != (void*)RVA(pe_sec->VirtualAddress))
  597.         {
  598.             
  599.             ERR_(win32)( "Critical Error: failed to map PE section to necessary address.n");
  600.             goto error;
  601.         }
  602.         if ((pe_sec->SizeOfRawData < pe_sec->Misc.VirtualSize) &&
  603.             (pe_sec->SizeOfRawData & (page_size-1)))
  604.         {
  605.             DWORD end = (pe_sec->SizeOfRawData & ~(page_size-1)) + page_size;
  606.             if (end > pe_sec->Misc.VirtualSize) end = pe_sec->Misc.VirtualSize;
  607.             TRACE("clearing %p - %pn",
  608.                   RVA(pe_sec->VirtualAddress) + pe_sec->SizeOfRawData,
  609.                   RVA(pe_sec->VirtualAddress) + end );
  610.             memset( (char*)RVA(pe_sec->VirtualAddress) + pe_sec->SizeOfRawData, 0,
  611.                     end - pe_sec->SizeOfRawData );
  612.         }
  613.     }
  614.     
  615.     if ( reloc )
  616.         do_relocations( load_addr, (IMAGE_BASE_RELOCATION *)RVA(reloc) );
  617.     
  618.     *version =   ( (nt->OptionalHeader.MajorSubsystemVersion & 0xff) << 8 )
  619.                |   (nt->OptionalHeader.MinorSubsystemVersion & 0xff);
  620.     
  621.     UnmapViewOfFile( (LPVOID)hModule );
  622.     return (HMODULE)load_addr;
  623. error:
  624.     if (unix_handle != -1) close( unix_handle );
  625.     if (load_addr) 
  626.     VirtualFree( (LPVOID)load_addr, 0, MEM_RELEASE );
  627.     UnmapViewOfFile( (LPVOID)hModule );
  628.     return 0;
  629. }
  630. /**********************************************************************
  631.  *                 PE_CreateModule
  632.  *
  633.  * Create WINE_MODREF structure for loaded HMODULE32, link it into
  634.  * process modref_list, and fixup all imports.
  635.  *
  636.  * Note: hModule must point to a correctly allocated PE image,
  637.  *       with base relocations applied; the 16-bit dummy module
  638.  *       associated to hModule must already exist.
  639.  *
  640.  * Note: This routine must always be called in the context of the
  641.  *       process that is to own the module to be created.
  642.  */
  643. WINE_MODREF *PE_CreateModule( HMODULE hModule, 
  644.                               LPCSTR filename, DWORD flags, WIN_BOOL builtin )
  645. {
  646.     DWORD load_addr = (DWORD)hModule;  
  647.     IMAGE_NT_HEADERS *nt = PE_HEADER(hModule);
  648.     IMAGE_DATA_DIRECTORY *dir;
  649.     IMAGE_IMPORT_DESCRIPTOR *pe_import = NULL;
  650.     IMAGE_EXPORT_DIRECTORY *pe_export = NULL;
  651.     IMAGE_RESOURCE_DIRECTORY *pe_resource = NULL;
  652.     WINE_MODREF *wm;
  653.     int result;
  654.     
  655.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_EXPORT;
  656.     if (dir->Size)
  657.         pe_export = (PIMAGE_EXPORT_DIRECTORY)RVA(dir->VirtualAddress);
  658.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_IMPORT;
  659.     if (dir->Size)
  660.         pe_import = (PIMAGE_IMPORT_DESCRIPTOR)RVA(dir->VirtualAddress);
  661.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_RESOURCE;
  662.     if (dir->Size)
  663.         pe_resource = (PIMAGE_RESOURCE_DIRECTORY)RVA(dir->VirtualAddress);
  664.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_EXCEPTION;
  665.     if (dir->Size) FIXME("Exception directory ignoredn" );
  666.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_SECURITY;
  667.     if (dir->Size) FIXME("Security directory ignoredn" );
  668.     
  669.     
  670.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_DEBUG;
  671.     if (dir->Size) TRACE("Debug directory ignoredn" );
  672.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_COPYRIGHT;
  673.     if (dir->Size) FIXME("Copyright string ignoredn" );
  674.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_GLOBALPTR;
  675.     if (dir->Size) FIXME("Global Pointer (MIPS) ignoredn" );
  676.     
  677.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG;
  678.     if (dir->Size) FIXME("Load Configuration directory ignoredn" );
  679.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT;
  680.     if (dir->Size) TRACE("Bound Import directory ignoredn" );
  681.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_IAT;
  682.     if (dir->Size) TRACE("Import Address Table directory ignoredn" );
  683.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT;
  684.     if (dir->Size)
  685.     {
  686. TRACE("Delayed import, stub calls LoadLibraryn" );
  687. /*
  688.  * Nothing to do here.
  689.  */
  690. #ifdef ImgDelayDescr
  691. /*
  692.  * This code is useful to observe what the heck is going on.
  693.  */
  694. {
  695. ImgDelayDescr *pe_delay = NULL;
  696.         pe_delay = (PImgDelayDescr)RVA(dir->VirtualAddress);
  697.         TRACE_(delayhlp)("pe_delay->grAttrs = %08xn", pe_delay->grAttrs);
  698.         TRACE_(delayhlp)("pe_delay->szName = %sn", pe_delay->szName);
  699.         TRACE_(delayhlp)("pe_delay->phmod = %08xn", pe_delay->phmod);
  700.         TRACE_(delayhlp)("pe_delay->pIAT = %08xn", pe_delay->pIAT);
  701.         TRACE_(delayhlp)("pe_delay->pINT = %08xn", pe_delay->pINT);
  702.         TRACE_(delayhlp)("pe_delay->pBoundIAT = %08xn", pe_delay->pBoundIAT);
  703.         TRACE_(delayhlp)("pe_delay->pUnloadIAT = %08xn", pe_delay->pUnloadIAT);
  704.         TRACE_(delayhlp)("pe_delay->dwTimeStamp = %08xn", pe_delay->dwTimeStamp);
  705.         }
  706. #endif 
  707. }
  708.     dir = nt->OptionalHeader.DataDirectory+IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR;
  709.     if (dir->Size) FIXME("Unknown directory 14 ignoredn" );
  710.     dir = nt->OptionalHeader.DataDirectory+15;
  711.     if (dir->Size) FIXME("Unknown directory 15 ignoredn" );
  712.     
  713.     wm = (WINE_MODREF *)HeapAlloc( GetProcessHeap(), 
  714.                                    HEAP_ZERO_MEMORY, sizeof(*wm) );
  715.     wm->module = hModule;
  716.     if ( builtin ) 
  717.         wm->flags |= WINE_MODREF_INTERNAL;
  718.     if ( flags & DONT_RESOLVE_DLL_REFERENCES )
  719.         wm->flags |= WINE_MODREF_DONT_RESOLVE_REFS;
  720.     if ( flags & LOAD_LIBRARY_AS_DATAFILE )
  721.         wm->flags |= WINE_MODREF_LOAD_AS_DATAFILE;
  722.     wm->type = MODULE32_PE;
  723.     wm->binfmt.pe.pe_export = pe_export;
  724.     wm->binfmt.pe.pe_import = pe_import;
  725.     wm->binfmt.pe.pe_resource = pe_resource;
  726.     wm->binfmt.pe.tlsindex = -1;
  727.     wm->filename = malloc(strlen(filename)+1);
  728.     strcpy(wm->filename, filename );
  729.     wm->modname = strrchr( wm->filename, '\' );
  730.     if (!wm->modname) wm->modname = wm->filename;
  731.     else wm->modname++;
  732.     if ( pe_export )
  733.         dump_exports( hModule );
  734.     /* Fixup Imports */
  735.     if (    pe_import
  736.          && !( wm->flags & WINE_MODREF_LOAD_AS_DATAFILE )
  737.          && !( wm->flags & WINE_MODREF_DONT_RESOLVE_REFS ) 
  738.          && fixup_imports( wm ) ) 
  739.     {
  740.         /* remove entry from modref chain */
  741.          return NULL;
  742.     }
  743.     return wm;
  744.     return wm;
  745. }
  746. /******************************************************************************
  747.  * The PE Library Loader frontend. 
  748.  * FIXME: handle the flags.
  749.  */
  750. WINE_MODREF *PE_LoadLibraryExA (LPCSTR name, DWORD flags)
  751. {
  752. HMODULE hModule32;
  753. WINE_MODREF *wm;
  754. char         filename[256];
  755. int hFile;
  756. WORD version = 0;
  757. strncpy(filename, name, sizeof(filename));      
  758. hFile=open(filename, O_RDONLY);
  759. if(hFile==-1)
  760.     return NULL;
  761. hModule32 = PE_LoadImage( hFile, filename, &version );
  762. if (!hModule32)
  763. {
  764. SetLastError( ERROR_OUTOFMEMORY );
  765. return NULL;
  766. }
  767. if ( !(wm = PE_CreateModule( hModule32, filename, flags, FALSE )) )
  768. {
  769. ERR( "can't load %sn", filename );
  770. SetLastError( ERROR_OUTOFMEMORY );
  771. return NULL;
  772. }
  773. close(hFile);
  774. //printf("^^^^^^^^^^^^^^^^Alloc VM1  %pn", wm);
  775. return wm;
  776. }
  777. /*****************************************************************************
  778.  * PE_UnloadLibrary
  779.  *
  780.  * Unload the library unmapping the image and freeing the modref structure.
  781.  */
  782. void PE_UnloadLibrary(WINE_MODREF *wm)
  783. {
  784.     TRACE(" unloading %sn", wm->filename);
  785.     free(wm->filename);
  786.     free(wm->short_filename);
  787.     HeapFree( GetProcessHeap(), 0, wm->deps );
  788.     VirtualFree( (LPVOID)wm->module, 0, MEM_RELEASE );
  789.     HeapFree( GetProcessHeap(), 0, wm );
  790.     //printf("^^^^^^^^^^^^^^^^Free VM1  %pn", wm);
  791. }
  792. /*****************************************************************************
  793.  * Load the PE main .EXE. All other loading is done by PE_LoadLibraryExA
  794.  * FIXME: this function should use PE_LoadLibraryExA, but currently can't
  795.  * due to the PROCESS_Create stuff.
  796.  */
  797. /*
  798.  * This is a dirty hack.
  799.  * The win32 DLLs contain an alloca routine, that first probes the soon
  800.  * to be allocated new memory *below* the current stack pointer in 4KByte
  801.  * increments.  After the mem probing below the current %esp,  the stack
  802.  * pointer is finally decremented to make room for the "alloca"ed memory.
  803.  * Maybe the probing code is intended to extend the stack on a windows box.
  804.  * Anyway, the linux kernel does *not* extend the stack by simply accessing
  805.  * memory below %esp;  it segfaults.
  806.  * The extend_stack_for_dll_alloca() routine just preallocates a big chunk
  807.  * of memory on the stack, for use by the DLLs alloca routine.
  808.  * Added the noinline attribute as e.g. gcc 3.2.2 inlines this function
  809.  * in a way that breaks it.
  810.  */
  811. static void __attribute__((noinline)) extend_stack_for_dll_alloca(void)
  812. {
  813. #if !defined(__FreeBSD__) && !defined(__DragonFly__)
  814.     volatile int* mem=alloca(0x20000);
  815.     *mem=0x1234;
  816. #endif
  817. }
  818. /* Called if the library is loaded or freed.
  819.  * NOTE: if a thread attaches a DLL, the current thread will only do
  820.  * DLL_PROCESS_ATTACH. Only new created threads do DLL_THREAD_ATTACH
  821.  * (SDK)
  822.  */
  823. WIN_BOOL PE_InitDLL( WINE_MODREF *wm, DWORD type, LPVOID lpReserved )
  824. {
  825.     WIN_BOOL retv = TRUE;
  826.     assert( wm->type == MODULE32_PE );
  827.     
  828.     if ((PE_HEADER(wm->module)->FileHeader.Characteristics & IMAGE_FILE_DLL) &&
  829.         (PE_HEADER(wm->module)->OptionalHeader.AddressOfEntryPoint)
  830.     ) {
  831. DLLENTRYPROC entry ;
  832. entry = (void*)PE_FindExportedFunction(wm, "DllMain", 0);
  833. if(entry==NULL)
  834.     entry = (void*)RVA_PTR( wm->module,OptionalHeader.AddressOfEntryPoint );
  835.         
  836. TRACE_(relay)("CallTo32(entryproc=%p,module=%08x,type=%ld,res=%p)n",
  837.                        entry, wm->module, type, lpReserved );
  838. TRACE("Entering DllMain(");
  839. switch(type)
  840. {
  841.     case DLL_PROCESS_DETACH:
  842.         TRACE("DLL_PROCESS_DETACH) ");
  843. break;
  844.     case DLL_PROCESS_ATTACH:
  845.         TRACE("DLL_PROCESS_ATTACH) ");
  846. break;
  847.     case DLL_THREAD_DETACH:
  848.         TRACE("DLL_THREAD_DETACH) ");
  849. break;
  850.     case DLL_THREAD_ATTACH:
  851.         TRACE("DLL_THREAD_ATTACH) ");
  852. break;
  853. }
  854. TRACE("for %sn", wm->filename);
  855. extend_stack_for_dll_alloca();
  856.         retv = entry( wm->module, type, lpReserved );
  857.     }
  858.     return retv;
  859. }
  860. static LPVOID
  861. _fixup_address(PIMAGE_OPTIONAL_HEADER opt,int delta,LPVOID addr) {
  862. if ( ((DWORD)addr>opt->ImageBase) &&
  863. ((DWORD)addr<opt->ImageBase+opt->SizeOfImage)
  864. )
  865. return (LPVOID)(((DWORD)addr)+delta);
  866. else
  867. return addr;
  868. }