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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, 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 "winternl.h"
  27 #include "winerror.h"
  28 #include "winver.h"
  29 #include "wine/debug.h"
  30 #include "imagehlp.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(imagehlp);
  33
  34 /***********************************************************************
  35  *           Data
  36  */
  37 static LIST_ENTRY image_list = { &image_list, &image_list };
  38
  39
  40 /***********************************************************************
  41  *              GetImageConfigInformation (IMAGEHLP.@)
  42  */
  43 BOOL WINAPI GetImageConfigInformation(
  44   PLOADED_IMAGE LoadedImage,
  45   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
  46 {
  47   FIXME("(%p, %p): stub\n",
  48     LoadedImage, ImageConfigInformation
  49   );
  50   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  51   return FALSE;
  52 }
  53
  54 /***********************************************************************
  55  *              GetImageUnusedHeaderBytes (IMAGEHLP.@)
  56  */
  57 DWORD WINAPI GetImageUnusedHeaderBytes(
  58   PLOADED_IMAGE LoadedImage,
  59   LPDWORD SizeUnusedHeaderBytes)
  60 {
  61   FIXME("(%p, %p): stub\n",
  62     LoadedImage, SizeUnusedHeaderBytes
  63   );
  64   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
  65   return 0;
  66 }
  67
  68 /***********************************************************************
  69  *              ImageLoad (IMAGEHLP.@)
  70  */
  71 PLOADED_IMAGE WINAPI ImageLoad(PCSTR dll_name, PCSTR dll_path)
  72 {
  73     LOADED_IMAGE *image;
  74
  75     TRACE("(%s, %s)\n", dll_name, dll_path);
  76
  77     image = HeapAlloc(GetProcessHeap(), 0, sizeof(*image));
  78     if (!image) return NULL;
  79
  80     if (!MapAndLoad(dll_name, dll_path, image, TRUE, TRUE))
  81     {
  82         HeapFree(GetProcessHeap(), 0, image);
  83         return NULL;
  84     }
  85
  86     image->Links.Flink = image_list.Flink;
  87     image->Links.Blink = &image_list;
  88     image_list.Flink->Blink = &image->Links;
  89     image_list.Flink = &image->Links;
  90
  91     return image;
  92 }
  93
  94 /***********************************************************************
  95  *              ImageUnload (IMAGEHLP.@)
  96  */
  97 BOOL WINAPI ImageUnload(PLOADED_IMAGE loaded_image)
  98 {
  99     LIST_ENTRY *entry, *mark;
 100     PLOADED_IMAGE image;
 101
 102     TRACE("(%p)\n", loaded_image);
 103
 104     /* FIXME: do we really need to check this? */
 105     mark = &image_list;
 106     for (entry = mark->Flink; entry != mark; entry = entry->Flink)
 107     {
 108         image = CONTAINING_RECORD(entry, LOADED_IMAGE, Links);
 109         if (image == loaded_image)
 110             break;
 111     }
 112
 113     if (entry == mark)
 114     {
 115         /* Not found */
 116         SetLastError(ERROR_INVALID_PARAMETER);
 117         return FALSE;
 118     }
 119
 120     entry->Blink->Flink = entry->Flink;
 121     entry->Flink->Blink = entry->Blink;
 122
 123     UnMapAndLoad(loaded_image);
 124     HeapFree(GetProcessHeap(), 0, loaded_image);
 125
 126     return TRUE;
 127 }
 128
 129 /***********************************************************************
 130  *              MapAndLoad (IMAGEHLP.@)
 131  */
 132 BOOL WINAPI MapAndLoad(PCSTR pszImageName, PCSTR pszDllPath, PLOADED_IMAGE pLoadedImage,
 133                        BOOL bDotDll, BOOL bReadOnly)
 134 {
 135     CHAR szFileName[MAX_PATH];
 136     HANDLE hFile = INVALID_HANDLE_VALUE;
 137     HANDLE hFileMapping = NULL;
 138     PVOID mapping = NULL;
 139     PIMAGE_NT_HEADERS pNtHeader = NULL;
 140
 141     TRACE("(%s, %s, %p, %d, %d)\n",
 142           pszImageName, pszDllPath, pLoadedImage, bDotDll, bReadOnly);
 143
 144     if (!SearchPathA(pszDllPath, pszImageName, bDotDll ? ".DLL" : ".EXE",
 145                      sizeof(szFileName), szFileName, NULL))
 146     {
 147         SetLastError(ERROR_FILE_NOT_FOUND);
 148         goto Error;
 149     }
 150
 151     hFile = CreateFileA(szFileName,
 152                         GENERIC_READ | (bReadOnly ? 0 : GENERIC_WRITE),
 153                         FILE_SHARE_READ,
 154                         NULL, OPEN_EXISTING, 0, NULL);
 155     if (hFile == INVALID_HANDLE_VALUE)
 156     {
 157         WARN("CreateFile: Error = %ld\n", GetLastError());
 158         goto Error;
 159     }
 160
 161     hFileMapping = CreateFileMappingA(hFile, NULL,
 162                                       (bReadOnly ? PAGE_READONLY : PAGE_READWRITE) | SEC_COMMIT,
 163                                       0, 0, NULL);
 164     if (!hFileMapping)
 165     {
 166         WARN("CreateFileMapping: Error = %ld\n", GetLastError());
 167         goto Error;
 168     }
 169
 170     mapping = MapViewOfFile(hFileMapping, bReadOnly ? FILE_MAP_READ : FILE_MAP_WRITE, 0, 0, 0);
 171     CloseHandle(hFileMapping);
 172     if (!mapping)
 173     {
 174         WARN("MapViewOfFile: Error = %ld\n", GetLastError());
 175         goto Error;
 176     }
 177
 178     if (!(pNtHeader = RtlImageNtHeader(mapping)))
 179     {
 180         WARN("Not an NT header\n");
 181         UnmapViewOfFile(mapping);
 182         goto Error;
 183     }
 184
 185     pLoadedImage->ModuleName       = HeapAlloc(GetProcessHeap(), 0,
 186                                                strlen(szFileName) + 1);
 187     if (pLoadedImage->ModuleName) strcpy(pLoadedImage->ModuleName, szFileName);
 188     pLoadedImage->hFile            = hFile;
 189     pLoadedImage->MappedAddress    = mapping;
 190     pLoadedImage->FileHeader       = pNtHeader;
 191     pLoadedImage->Sections         = IMAGE_FIRST_SECTION(pNtHeader);
 192     pLoadedImage->NumberOfSections = pNtHeader->FileHeader.NumberOfSections;
 193     pLoadedImage->SizeOfImage      = GetFileSize(hFile, NULL);
 194     pLoadedImage->Characteristics  = pNtHeader->FileHeader.Characteristics;
 195     pLoadedImage->LastRvaSection   = pLoadedImage->Sections;
 196
 197     pLoadedImage->fSystemImage     = FALSE; /* FIXME */
 198     pLoadedImage->fDOSImage        = FALSE; /* FIXME */
 199
 200     pLoadedImage->Links.Flink      = &pLoadedImage->Links;
 201     pLoadedImage->Links.Blink      = &pLoadedImage->Links;
 202
 203     return TRUE;
 204
 205 Error:
 206     if (hFile != INVALID_HANDLE_VALUE) CloseHandle(hFile);
 207     return FALSE;
 208 }
 209
 210 /***********************************************************************
 211  *              SetImageConfigInformation (IMAGEHLP.@)
 212  */
 213 BOOL WINAPI SetImageConfigInformation(
 214   PLOADED_IMAGE LoadedImage,
 215   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
 216 {
 217   FIXME("(%p, %p): stub\n",
 218     LoadedImage, ImageConfigInformation
 219   );
 220   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 221   return FALSE;
 222 }
 223
 224 /***********************************************************************
 225  *              UnMapAndLoad (IMAGEHLP.@)
 226  */
 227 BOOL WINAPI UnMapAndLoad(PLOADED_IMAGE pLoadedImage)
 228 {
 229     HeapFree(GetProcessHeap(), 0, pLoadedImage->ModuleName);
 230     /* FIXME: MSDN states that a new checksum is computed and stored into the file */
 231     if (pLoadedImage->MappedAddress) UnmapViewOfFile(pLoadedImage->MappedAddress);
 232     if (pLoadedImage->hFile != INVALID_HANDLE_VALUE) CloseHandle(pLoadedImage->hFile);
 233     return TRUE;
 234 }