dlls/imagehlp/access.c

   1 /*
   2  *      IMAGEHLP library
   3  *
   4  *      Copyright 1998  Patrik Stridvall
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdarg.h>
  22 #include <string.h>
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winnt.h"
  26 #include "winreg.h"
  27 #include "winternl.h"
  28 #include "winerror.h"
  29 #include "wine/debug.h"
  30 #include "imagehlp.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(imagehlp);
  33
  34 /***********************************************************************
  35  *           Data
  36  */
  37
  38 static PLOADED_IMAGE IMAGEHLP_pFirstLoadedImage=NULL;
  39 static PLOADED_IMAGE IMAGEHLP_pLastLoadedImage=NULL;
  40
  41 static LOADED_IMAGE IMAGEHLP_EmptyLoadedImage = {
  42   NULL,       /* ModuleName */
  43   0,          /* hFile */
  44   NULL,       /* MappedAddress */
  45   NULL,       /* FileHeader */
  46   NULL,       /* LastRvaSection */
  47   0,          /* NumberOfSections */
  48   NULL,       /* Sections */
  49   1,          /* Characteristics */
  50   FALSE,      /* fSystemImage */
  51   FALSE,      /* fDOSImage */
  52   { &IMAGEHLP_EmptyLoadedImage.Links, &IMAGEHLP_EmptyLoadedImage.Links }, /* Links */
  53   148,        /* SizeOfImage; */
  54 };
  55
  56 extern HANDLE IMAGEHLP_hHeap;
  57
  58 /***********************************************************************
  59  *              GetImageConfigInformation (IMAGEHLP.@)
  60  */
  61 BOOL WINAPI GetImageConfigInformation(
  62   PLOADED_IMAGE LoadedImage,
  63   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
  64 {
  65   FIXME("(%p, %p): stub\n",
  66     LoadedImage, ImageConfigInformation
  67   );
  68   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  69   return FALSE;
  70 }
  71
  72 /***********************************************************************
  73  *              GetImageUnusedHeaderBytes (IMAGEHLP.@)
  74  */
  75 DWORD WINAPI GetImageUnusedHeaderBytes(
  76   PLOADED_IMAGE LoadedImage,
  77   LPDWORD SizeUnusedHeaderBytes)
  78 {
  79   FIXME("(%p, %p): stub\n",
  80     LoadedImage, SizeUnusedHeaderBytes
  81   );
  82   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  83   return 0;
  84 }
  85
  86 /***********************************************************************
  87  *              ImageLoad (IMAGEHLP.@)
  88  */
  89 PLOADED_IMAGE WINAPI ImageLoad(LPSTR DllName, LPSTR DllPath)
  90 {
  91   PLOADED_IMAGE pLoadedImage;
  92
  93   FIXME("(%s, %s): stub\n", DllName, DllPath);
  94
  95   pLoadedImage = HeapAlloc(IMAGEHLP_hHeap, 0, sizeof(LOADED_IMAGE));
  96   if (pLoadedImage)
  97     pLoadedImage->FileHeader = HeapAlloc(IMAGEHLP_hHeap, 0, sizeof(IMAGE_NT_HEADERS));
  98
  99   return pLoadedImage;
 100 }
 101
 102 /***********************************************************************
 103  *              ImageUnload (IMAGEHLP.@)
 104  */
 105 BOOL WINAPI ImageUnload(PLOADED_IMAGE pLoadedImage)
 106 {
 107   LIST_ENTRY *pCurrent, *pFind;
 108
 109   TRACE("(%p)\n", pLoadedImage);
 110
 111   if(!IMAGEHLP_pFirstLoadedImage || !pLoadedImage)
 112     {
 113       /* No image loaded or null pointer */
 114       SetLastError(ERROR_INVALID_PARAMETER);
 115       return FALSE;
 116     }
 117
 118   pFind=&pLoadedImage->Links;
 119   pCurrent=&IMAGEHLP_pFirstLoadedImage->Links;
 120   while((pCurrent != pFind) &&
 121     (pCurrent != NULL))
 122       pCurrent = pCurrent->Flink;
 123   if(!pCurrent)
 124     {
 125       /* Not found */
 126       SetLastError(ERROR_INVALID_PARAMETER);
 127       return FALSE;
 128     }
 129
 130   if(pCurrent->Blink)
 131     pCurrent->Blink->Flink = pCurrent->Flink;
 132   else
 133     IMAGEHLP_pFirstLoadedImage = pCurrent->Flink?CONTAINING_RECORD(
 134       pCurrent->Flink, LOADED_IMAGE, Links):NULL;
 135
 136   if(pCurrent->Flink)
 137     pCurrent->Flink->Blink = pCurrent->Blink;
 138   else
 139     IMAGEHLP_pLastLoadedImage = pCurrent->Blink?CONTAINING_RECORD(
 140       pCurrent->Blink, LOADED_IMAGE, Links):NULL;
 141
 142   return FALSE;
 143 }
 144
 145 /***********************************************************************
 146  *              MapAndLoad (IMAGEHLP.@)
 147  */
 148 BOOL WINAPI MapAndLoad(
 149   LPSTR pszImageName, LPSTR pszDllPath, PLOADED_IMAGE pLoadedImage,
 150   BOOL bDotDll, BOOL bReadOnly)
 151 {
 152   CHAR szFileName[MAX_PATH];
 153   HANDLE hFile = NULL;
 154   HANDLE hFileMapping = NULL;
 155   HMODULE hModule = NULL;
 156   PIMAGE_NT_HEADERS pNtHeader = NULL;
 157
 158   TRACE("(%s, %s, %p, %d, %d)\n", pszImageName, pszDllPath, pLoadedImage,
 159                                     bDotDll, bReadOnly);
 160
 161   /* PathCombine(&szFileName, pszDllPath, pszImageName); */
 162   /* PathRenameExtension(&szFileName, bDotDll?:"dll":"exe"); */
 163
 164   /* FIXME: Check if the file already loaded (use IMAGEHLP_pFirstLoadedImage) */
 165   if(!(hFile = CreateFileA(
 166     szFileName, GENERIC_READ, 1, /* FIXME: FILE_SHARE_READ not defined */
 167     NULL, OPEN_EXISTING, 0, NULL)))
 168     {
 169       SetLastError(ERROR_FILE_NOT_FOUND);
 170       goto Error;
 171     }
 172
 173   if(!(hFileMapping = CreateFileMappingA(
 174     hFile, NULL, PAGE_READONLY | SEC_COMMIT, 0, 0, NULL)))
 175     {
 176       DWORD dwLastError = GetLastError();
 177       WARN("CreateFileMapping: Error = %ld\n", dwLastError);
 178       SetLastError(dwLastError);
 179       goto Error;
 180     }
 181   CloseHandle(hFile);
 182   hFile = NULL;
 183
 184   if(!(hModule = (HMODULE) MapViewOfFile(
 185     hFileMapping, FILE_MAP_READ, 0, 0, 0)))
 186     {
 187       DWORD dwLastError = GetLastError();
 188       WARN("MapViewOfFile: Error = %ld\n", dwLastError);
 189       SetLastError(dwLastError);
 190       goto Error;
 191     }
 192
 193   CloseHandle(hFileMapping);
 194   hFileMapping=NULL;
 195
 196   pLoadedImage = HeapAlloc(IMAGEHLP_hHeap, 0, sizeof(LOADED_IMAGE));
 197
 198   pNtHeader = RtlImageNtHeader(hModule);
 199
 200   pLoadedImage->ModuleName = HeapAlloc(IMAGEHLP_hHeap, 0, strlen(pszDllPath)+1); /* FIXME: Correct? */
 201   strcpy( pLoadedImage->ModuleName, pszDllPath );
 202   pLoadedImage->hFile = hFile;
 203   pLoadedImage->MappedAddress = (PUCHAR) hModule;
 204   pLoadedImage->FileHeader = pNtHeader;
 205   pLoadedImage->Sections = (PIMAGE_SECTION_HEADER)
 206     ((LPBYTE) &pNtHeader->OptionalHeader +
 207       pNtHeader->FileHeader.SizeOfOptionalHeader);
 208   pLoadedImage->NumberOfSections =
 209     pNtHeader->FileHeader.NumberOfSections;
 210   pLoadedImage->SizeOfImage =
 211     pNtHeader->OptionalHeader.SizeOfImage;
 212   pLoadedImage->Characteristics =
 213     pNtHeader->FileHeader.Characteristics;
 214   pLoadedImage->LastRvaSection = pLoadedImage->Sections;
 215
 216   pLoadedImage->fSystemImage = FALSE; /* FIXME */
 217   pLoadedImage->fDOSImage = FALSE;    /* FIXME */
 218
 219   /* FIXME: Make thread safe */
 220   pLoadedImage->Links.Flink = NULL;
 221   pLoadedImage->Links.Blink = &IMAGEHLP_pLastLoadedImage->Links;
 222   if(IMAGEHLP_pLastLoadedImage)
 223     IMAGEHLP_pLastLoadedImage->Links.Flink = &pLoadedImage->Links;
 224   IMAGEHLP_pLastLoadedImage = pLoadedImage;
 225   if(!IMAGEHLP_pFirstLoadedImage)
 226     IMAGEHLP_pFirstLoadedImage = pLoadedImage;
 227
 228   return TRUE;
 229
 230 Error:
 231   if(hModule)
 232     UnmapViewOfFile((PVOID) hModule);
 233   if(hFileMapping)
 234     CloseHandle(hFileMapping);
 235   if(hFile)
 236     CloseHandle(hFile);
 237   return FALSE;
 238 }
 239
 240 /***********************************************************************
 241  *              SetImageConfigInformation (IMAGEHLP.@)
 242  */
 243 BOOL WINAPI SetImageConfigInformation(
 244   PLOADED_IMAGE LoadedImage,
 245   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
 246 {
 247   FIXME("(%p, %p): stub\n",
 248     LoadedImage, ImageConfigInformation
 249   );
 250   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 251   return FALSE;
 252 }
 253
 254 /***********************************************************************
 255  *              UnMapAndLoad (IMAGEHLP.@)
 256  */
 257 BOOL WINAPI UnMapAndLoad(PLOADED_IMAGE LoadedImage)
 258 {
 259   FIXME("(%p): stub\n", LoadedImage);
 260   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 261   return FALSE;
 262 }