Fix subclassing to support nested messages.
[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  *              EnumerateLoadedModules (IMAGEHLP.@)
60  */
61 BOOL WINAPI EnumerateLoadedModules(
62   HANDLE hProcess,
63   PENUMLOADED_MODULES_CALLBACK EnumLoadedModulesCallback,
64   PVOID UserContext)
65 {
66   FIXME("(%p, %p, %p): stub\n",
67     hProcess, EnumLoadedModulesCallback, UserContext
68   );
69   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
70   return FALSE;
71 }
72
73 /***********************************************************************
74  *              GetTimestampForLoadedLibrary (IMAGEHLP.@)
75  */
76 DWORD WINAPI GetTimestampForLoadedLibrary(HMODULE Module)
77 {
78   FIXME("(%p): stub\n", Module);
79   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
80   return 0;
81 }
82
83 /***********************************************************************
84  *              GetImageConfigInformation (IMAGEHLP.@)
85  */
86 BOOL WINAPI GetImageConfigInformation(
87   PLOADED_IMAGE LoadedImage,
88   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
89 {
90   FIXME("(%p, %p): stub\n",
91     LoadedImage, ImageConfigInformation
92   );
93   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
94   return FALSE;
95 }
96
97 /***********************************************************************
98  *              GetImageUnusedHeaderBytes (IMAGEHLP.@)
99  */
100 DWORD WINAPI GetImageUnusedHeaderBytes(
101   PLOADED_IMAGE LoadedImage,
102   LPDWORD SizeUnusedHeaderBytes)
103 {
104   FIXME("(%p, %p): stub\n",
105     LoadedImage, SizeUnusedHeaderBytes
106   );
107   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
108   return 0;
109 }
110
111 /***********************************************************************
112  *              ImageLoad (IMAGEHLP.@)
113  */
114 PLOADED_IMAGE WINAPI ImageLoad(LPSTR DllName, LPSTR DllPath)
115 {
116   PLOADED_IMAGE pLoadedImage;
117
118   FIXME("(%s, %s): stub\n", DllName, DllPath);
119           
120   pLoadedImage = HeapAlloc(IMAGEHLP_hHeap, 0, sizeof(LOADED_IMAGE));
121   if (pLoadedImage)
122     pLoadedImage->FileHeader = HeapAlloc(IMAGEHLP_hHeap, 0, sizeof(IMAGE_NT_HEADERS));
123   
124   return pLoadedImage;
125 }
126
127 /***********************************************************************
128  *              ImageUnload (IMAGEHLP.@)
129  */
130 BOOL WINAPI ImageUnload(PLOADED_IMAGE pLoadedImage)
131 {
132   LIST_ENTRY *pCurrent, *pFind;
133
134   TRACE("(%p)\n", pLoadedImage);
135   
136   if(!IMAGEHLP_pFirstLoadedImage || !pLoadedImage)
137     {
138       /* No image loaded or null pointer */
139       SetLastError(ERROR_INVALID_PARAMETER);
140       return FALSE;
141     }
142
143   pFind=&pLoadedImage->Links;
144   pCurrent=&IMAGEHLP_pFirstLoadedImage->Links;
145   while((pCurrent != pFind) &&
146     (pCurrent != NULL))
147       pCurrent = pCurrent->Flink;
148   if(!pCurrent)
149     {
150       /* Not found */
151       SetLastError(ERROR_INVALID_PARAMETER);
152       return FALSE;
153     }
154
155   if(pCurrent->Blink)
156     pCurrent->Blink->Flink = pCurrent->Flink;
157   else
158     IMAGEHLP_pFirstLoadedImage = pCurrent->Flink?CONTAINING_RECORD(
159       pCurrent->Flink, LOADED_IMAGE, Links):NULL;
160
161   if(pCurrent->Flink)
162     pCurrent->Flink->Blink = pCurrent->Blink;
163   else
164     IMAGEHLP_pLastLoadedImage = pCurrent->Blink?CONTAINING_RECORD(
165       pCurrent->Blink, LOADED_IMAGE, Links):NULL;
166
167   return FALSE;
168 }
169
170 /***********************************************************************
171  *              MapAndLoad (IMAGEHLP.@)
172  */
173 BOOL WINAPI MapAndLoad(
174   LPSTR pszImageName, LPSTR pszDllPath, PLOADED_IMAGE pLoadedImage,
175   BOOL bDotDll, BOOL bReadOnly)
176 {
177   CHAR szFileName[MAX_PATH];
178   HANDLE hFile = NULL;
179   HANDLE hFileMapping = NULL;
180   HMODULE hModule = NULL;
181   PIMAGE_NT_HEADERS pNtHeader = NULL;
182
183   TRACE("(%s, %s, %p, %d, %d)\n", pszImageName, pszDllPath, pLoadedImage,
184                                     bDotDll, bReadOnly);
185   
186   /* PathCombine(&szFileName, pszDllPath, pszImageName); */
187   /* PathRenameExtension(&szFileName, bDotDll?:"dll":"exe"); */
188
189   /* FIXME: Check if the file already loaded (use IMAGEHLP_pFirstLoadedImage) */
190   if(!(hFile = CreateFileA(
191     szFileName, GENERIC_READ, 1, /* FIXME: FILE_SHARE_READ not defined */
192     NULL, OPEN_EXISTING, 0, NULL)))
193     {
194       SetLastError(ERROR_FILE_NOT_FOUND);
195       goto Error;
196     }
197
198   if(!(hFileMapping = CreateFileMappingA(
199     hFile, NULL, PAGE_READONLY | SEC_COMMIT, 0, 0, NULL)))
200     {
201       DWORD dwLastError = GetLastError();
202       WARN("CreateFileMapping: Error = %ld\n", dwLastError);
203       SetLastError(dwLastError);
204       goto Error;
205     }
206   CloseHandle(hFile);
207   hFile = NULL;
208
209   if(!(hModule = (HMODULE) MapViewOfFile(
210     hFileMapping, FILE_MAP_READ, 0, 0, 0)))
211     {
212       DWORD dwLastError = GetLastError();
213       WARN("MapViewOfFile: Error = %ld\n", dwLastError);
214       SetLastError(dwLastError);
215       goto Error;
216     }
217
218   CloseHandle(hFileMapping);
219   hFileMapping=NULL;
220
221   pLoadedImage = (PLOADED_IMAGE) HeapAlloc(
222     IMAGEHLP_hHeap, 0, sizeof(LOADED_IMAGE)
223   );
224
225   pNtHeader = RtlImageNtHeader(hModule);
226
227   pLoadedImage->ModuleName = HeapAlloc(IMAGEHLP_hHeap, 0, strlen(pszDllPath)+1); /* FIXME: Correct? */
228   strcpy( pLoadedImage->ModuleName, pszDllPath );
229   pLoadedImage->hFile = hFile;
230   pLoadedImage->MappedAddress = (PUCHAR) hModule;
231   pLoadedImage->FileHeader = pNtHeader;
232   pLoadedImage->Sections = (PIMAGE_SECTION_HEADER)
233     ((LPBYTE) &pNtHeader->OptionalHeader +
234       pNtHeader->FileHeader.SizeOfOptionalHeader);
235   pLoadedImage->NumberOfSections =
236     pNtHeader->FileHeader.NumberOfSections;
237   pLoadedImage->SizeOfImage =
238     pNtHeader->OptionalHeader.SizeOfImage;
239   pLoadedImage->Characteristics =
240     pNtHeader->FileHeader.Characteristics;
241   pLoadedImage->LastRvaSection = pLoadedImage->Sections;
242
243   pLoadedImage->fSystemImage = FALSE; /* FIXME */
244   pLoadedImage->fDOSImage = FALSE;    /* FIXME */
245
246   /* FIXME: Make thread safe */
247   pLoadedImage->Links.Flink = NULL;
248   pLoadedImage->Links.Blink = &IMAGEHLP_pLastLoadedImage->Links;
249   if(IMAGEHLP_pLastLoadedImage)
250     IMAGEHLP_pLastLoadedImage->Links.Flink = &pLoadedImage->Links;
251   IMAGEHLP_pLastLoadedImage = pLoadedImage;
252   if(!IMAGEHLP_pFirstLoadedImage)
253     IMAGEHLP_pFirstLoadedImage = pLoadedImage;
254
255   return TRUE;
256
257 Error:
258   if(hModule)
259     UnmapViewOfFile((PVOID) hModule);
260   if(hFileMapping)
261     CloseHandle(hFileMapping);
262   if(hFile)
263     CloseHandle(hFile);
264   return FALSE;
265 }
266
267 /***********************************************************************
268  *              SetImageConfigInformation (IMAGEHLP.@)
269  */
270 BOOL WINAPI SetImageConfigInformation(
271   PLOADED_IMAGE LoadedImage,
272   PIMAGE_LOAD_CONFIG_DIRECTORY ImageConfigInformation)
273 {
274   FIXME("(%p, %p): stub\n",
275     LoadedImage, ImageConfigInformation
276   );
277   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
278   return FALSE;
279 }
280
281 /***********************************************************************
282  *              UnMapAndLoad (IMAGEHLP.@)
283  */
284 BOOL WINAPI UnMapAndLoad(PLOADED_IMAGE LoadedImage)
285 {
286   FIXME("(%p): stub\n", LoadedImage);
287   SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
288   return FALSE;
289 }