Fixed some forwards.
[wine] / 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 = (PLOADED_IMAGE) HeapAlloc(
197     IMAGEHLP_hHeap, 0, sizeof(LOADED_IMAGE)
198   );
199
200   pNtHeader = RtlImageNtHeader(hModule);
201
202   pLoadedImage->ModuleName = HeapAlloc(IMAGEHLP_hHeap, 0, strlen(pszDllPath)+1); /* FIXME: Correct? */
203   strcpy( pLoadedImage->ModuleName, pszDllPath );
204   pLoadedImage->hFile = hFile;
205   pLoadedImage->MappedAddress = (PUCHAR) hModule;
206   pLoadedImage->FileHeader = pNtHeader;
207   pLoadedImage->Sections = (PIMAGE_SECTION_HEADER)
208     ((LPBYTE) &pNtHeader->OptionalHeader +
209       pNtHeader->FileHeader.SizeOfOptionalHeader);
210   pLoadedImage->NumberOfSections =
211     pNtHeader->FileHeader.NumberOfSections;
212   pLoadedImage->SizeOfImage =
213     pNtHeader->OptionalHeader.SizeOfImage;
214   pLoadedImage->Characteristics =
215     pNtHeader->FileHeader.Characteristics;
216   pLoadedImage->LastRvaSection = pLoadedImage->Sections;
217
218   pLoadedImage->fSystemImage = FALSE; /* FIXME */
219   pLoadedImage->fDOSImage = FALSE;    /* FIXME */
220
221   /* FIXME: Make thread safe */
222   pLoadedImage->Links.Flink = NULL;
223   pLoadedImage->Links.Blink = &IMAGEHLP_pLastLoadedImage->Links;
224   if(IMAGEHLP_pLastLoadedImage)
225     IMAGEHLP_pLastLoadedImage->Links.Flink = &pLoadedImage->Links;
226   IMAGEHLP_pLastLoadedImage = pLoadedImage;
227   if(!IMAGEHLP_pFirstLoadedImage)
228     IMAGEHLP_pFirstLoadedImage = pLoadedImage;
229
230   return TRUE;
231
232 Error:
233   if(hModule)
234     UnmapViewOfFile((PVOID) hModule);
235   if(hFileMapping)
236     CloseHandle(hFileMapping);
237   if(hFile)
238     CloseHandle(hFile);
239   return FALSE;
240 }
241
242 /***********************************************************************
243  *              SetImageConfigInformation (IMAGEHLP.@)
244  */
245 BOOL WINAPI SetImageConfigInformation(
246   PLOADED_IMAGE LoadedImage,
247   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
248 {
249   FIXME("(%p, %p): stub\n",
250     LoadedImage, ImageConfigInformation
251   );
252   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
253   return FALSE;
254 }
255
256 /***********************************************************************
257  *              UnMapAndLoad (IMAGEHLP.@)
258  */
259 BOOL WINAPI UnMapAndLoad(PLOADED_IMAGE LoadedImage)
260 {
261   FIXME("(%p): stub\n", LoadedImage);
262   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
263   return FALSE;
264 }