?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
push 97f44e0adb27fff75ba63d8fb97c65db9edfbe82
[wine/hacks.git] / dlls / kernel32 / resource.c
blobff2ec956b9d67fe0bf40d8d9c78dc5a9365e8701
1 /*
2 * Resources
3 *
4 * Copyright 1993 Robert J. Amstadt
5 * Copyright 1995, 2003 Alexandre Julliard
6 * Copyright 2006 Mike McCormack
7 *
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <stdarg.h>
27
28 #define NONAMELESSUNION
29 #define NONAMELESSSTRUCT
30 #include "ntstatus.h"
31 #define WIN32_NO_STATUS
32 #include "windef.h"
33 #include "winbase.h"
34 #include "winternl.h"
35 #include "wine/winbase16.h"
36 #include "wine/debug.h"
37 #include "wine/exception.h"
38 #include "wine/unicode.h"
39 #include "wine/list.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(resource);
42
43 /* handle conversions */
44 #define HRSRC_32(h16) ((HRSRC)(ULONG_PTR)(h16))
45 #define HRSRC_16(h32) (LOWORD(h32))
46 #define HGLOBAL_32(h16) ((HGLOBAL)(ULONG_PTR)(h16))
47 #define HGLOBAL_16(h32) (LOWORD(h32))
48 #define HMODULE_16(h32) (LOWORD(h32))
49
50 /* retrieve the resource name to pass to the ntdll functions */
51 static NTSTATUS get_res_nameA( LPCSTR name, UNICODE_STRING *str )
52 {
53 if (!HIWORD(name))
54 {
55 str->Buffer = ULongToPtr(LOWORD(name));
56 return STATUS_SUCCESS;
57 }
58 if (name[0] == '#')
59 {
60 ULONG value;
61 if (RtlCharToInteger( name + 1, 10, &value ) != STATUS_SUCCESS || HIWORD(value))
62 return STATUS_INVALID_PARAMETER;
63 str->Buffer = ULongToPtr(value);
64 return STATUS_SUCCESS;
65 }
66 RtlCreateUnicodeStringFromAsciiz( str, name );
67 RtlUpcaseUnicodeString( str, str, FALSE );
68 return STATUS_SUCCESS;
69 }
70
71 /* retrieve the resource name to pass to the ntdll functions */
72 static NTSTATUS get_res_nameW( LPCWSTR name, UNICODE_STRING *str )
73 {
74 if (!HIWORD(name))
75 {
76 str->Buffer = ULongToPtr(LOWORD(name));
77 return STATUS_SUCCESS;
78 }
79 if (name[0] == '#')
80 {
81 ULONG value;
82 RtlInitUnicodeString( str, name + 1 );
83 if (RtlUnicodeStringToInteger( str, 10, &value ) != STATUS_SUCCESS || HIWORD(value))
84 return STATUS_INVALID_PARAMETER;
85 str->Buffer = ULongToPtr(value);
86 return STATUS_SUCCESS;
87 }
88 RtlCreateUnicodeString( str, name );
89 RtlUpcaseUnicodeString( str, str, FALSE );
90 return STATUS_SUCCESS;
91 }
92
93 /* retrieve the resource names for the 16-bit FindResource function */
94 static BOOL get_res_name_type_WtoA( LPCWSTR name, LPCWSTR type, LPSTR *nameA, LPSTR *typeA )
95 {
96 *nameA = *typeA = NULL;
97
98 __TRY
99 {
100 if (HIWORD(name))
101 {
102 DWORD len = WideCharToMultiByte( CP_ACP, 0, name, -1, NULL, 0, NULL, NULL );
103 *nameA = HeapAlloc( GetProcessHeap(), 0, len );
104 if (*nameA) WideCharToMultiByte( CP_ACP, 0, name, -1, *nameA, len, NULL, NULL );
105 }
106 else *nameA = ULongToPtr(LOWORD(name));
107
108 if (HIWORD(type))
109 {
110 DWORD len = WideCharToMultiByte( CP_ACP, 0, type, -1, NULL, 0, NULL, NULL );
111 *typeA = HeapAlloc( GetProcessHeap(), 0, len );
112 if (*typeA) WideCharToMultiByte( CP_ACP, 0, type, -1, *typeA, len, NULL, NULL );
113 }
114 else *typeA = ULongToPtr(LOWORD(type));
115 }
116 __EXCEPT_PAGE_FAULT
117 {
118 if (HIWORD(*nameA)) HeapFree( GetProcessHeap(), 0, *nameA );
119 if (HIWORD(*typeA)) HeapFree( GetProcessHeap(), 0, *typeA );
120 SetLastError( ERROR_INVALID_PARAMETER );
121 return FALSE;
122 }
123 __ENDTRY
124 return TRUE;
125 }
126
127 /* implementation of FindResourceExA */
128 static HRSRC find_resourceA( HMODULE hModule, LPCSTR type, LPCSTR name, WORD lang )
129 {
130 NTSTATUS status;
131 UNICODE_STRING nameW, typeW;
132 LDR_RESOURCE_INFO info;
133 const IMAGE_RESOURCE_DATA_ENTRY *entry = NULL;
134
135 __TRY
136 {
137 if ((status = get_res_nameA( name, &nameW )) != STATUS_SUCCESS) goto done;
138 if ((status = get_res_nameA( type, &typeW )) != STATUS_SUCCESS) goto done;
139 info.Type = (ULONG_PTR)typeW.Buffer;
140 info.Name = (ULONG_PTR)nameW.Buffer;
141 info.Language = lang;
142 status = LdrFindResource_U( hModule, &info, 3, &entry );
143 done:
144 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
145 }
146 __EXCEPT_PAGE_FAULT
147 {
148 SetLastError( ERROR_INVALID_PARAMETER );
149 }
150 __ENDTRY
151
152 if (HIWORD(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
153 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
154 return (HRSRC)entry;
155 }
156
157
158 /* implementation of FindResourceExW */
159 static HRSRC find_resourceW( HMODULE hModule, LPCWSTR type, LPCWSTR name, WORD lang )
160 {
161 NTSTATUS status;
162 UNICODE_STRING nameW, typeW;
163 LDR_RESOURCE_INFO info;
164 const IMAGE_RESOURCE_DATA_ENTRY *entry = NULL;
165
166 nameW.Buffer = typeW.Buffer = NULL;
167
168 __TRY
169 {
170 if ((status = get_res_nameW( name, &nameW )) != STATUS_SUCCESS) goto done;
171 if ((status = get_res_nameW( type, &typeW )) != STATUS_SUCCESS) goto done;
172 info.Type = (ULONG_PTR)typeW.Buffer;
173 info.Name = (ULONG_PTR)nameW.Buffer;
174 info.Language = lang;
175 status = LdrFindResource_U( hModule, &info, 3, &entry );
176 done:
177 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
178 }
179 __EXCEPT_PAGE_FAULT
180 {
181 SetLastError( ERROR_INVALID_PARAMETER );
182 }
183 __ENDTRY
184
185 if (HIWORD(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
186 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
187 return (HRSRC)entry;
188 }
189
190 /**********************************************************************
191 * FindResourceExA (KERNEL32.@)
192 */
193 HRSRC WINAPI FindResourceExA( HMODULE hModule, LPCSTR type, LPCSTR name, WORD lang )
194 {
195 TRACE( "%p %s %s %04x\n", hModule, debugstr_a(type), debugstr_a(name), lang );
196
197 if (!hModule) hModule = GetModuleHandleW(0);
198 else if (!HIWORD(hModule))
199 {
200 return HRSRC_32( FindResource16( HMODULE_16(hModule), name, type ) );
201 }
202 return find_resourceA( hModule, type, name, lang );
203 }
204
205
206 /**********************************************************************
207 * FindResourceA (KERNEL32.@)
208 */
209 HRSRC WINAPI FindResourceA( HMODULE hModule, LPCSTR name, LPCSTR type )
210 {
211 return FindResourceExA( hModule, type, name, MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ) );
212 }
213
214
215 /**********************************************************************
216 * FindResourceExW (KERNEL32.@)
217 */
218 HRSRC WINAPI FindResourceExW( HMODULE hModule, LPCWSTR type, LPCWSTR name, WORD lang )
219 {
220 TRACE( "%p %s %s %04x\n", hModule, debugstr_w(type), debugstr_w(name), lang );
221
222 if (!hModule) hModule = GetModuleHandleW(0);
223 else if (!HIWORD(hModule))
224 {
225 LPSTR nameA, typeA;
226 HRSRC16 ret;
227
228 if (!get_res_name_type_WtoA( name, type, &nameA, &typeA )) return NULL;
229
230 ret = FindResource16( HMODULE_16(hModule), nameA, typeA );
231 if (HIWORD(nameA)) HeapFree( GetProcessHeap(), 0, nameA );
232 if (HIWORD(typeA)) HeapFree( GetProcessHeap(), 0, typeA );
233 return HRSRC_32(ret);
234 }
235
236 return find_resourceW( hModule, type, name, lang );
237 }
238
239
240 /**********************************************************************
241 * FindResourceW (KERNEL32.@)
242 */
243 HRSRC WINAPI FindResourceW( HINSTANCE hModule, LPCWSTR name, LPCWSTR type )
244 {
245 return FindResourceExW( hModule, type, name, MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ) );
246 }
247
248
249 /**********************************************************************
250 * EnumResourceTypesA (KERNEL32.@)
251 */
252 BOOL WINAPI EnumResourceTypesA( HMODULE hmod, ENUMRESTYPEPROCA lpfun, LONG_PTR lparam )
253 {
254 int i;
255 BOOL ret = FALSE;
256 LPSTR type = NULL;
257 DWORD len = 0, newlen;
258 NTSTATUS status;
259 const IMAGE_RESOURCE_DIRECTORY *resdir;
260 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
261 const IMAGE_RESOURCE_DIR_STRING_U *str;
262
263 TRACE( "%p %p %lx\n", hmod, lpfun, lparam );
264
265 if (!hmod) hmod = GetModuleHandleA( NULL );
266
267 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &resdir )) != STATUS_SUCCESS)
268 {
269 SetLastError( RtlNtStatusToDosError(status) );
270 return FALSE;
271 }
272 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
273 for (i = 0; i < resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries; i++)
274 {
275 if (et[i].u1.s1.NameIsString)
276 {
277 str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const BYTE *)resdir + et[i].u1.s1.NameOffset);
278 newlen = WideCharToMultiByte( CP_ACP, 0, str->NameString, str->Length, NULL, 0, NULL, NULL);
279 if (newlen + 1 > len)
280 {
281 len = newlen + 1;
282 HeapFree( GetProcessHeap(), 0, type );
283 if (!(type = HeapAlloc( GetProcessHeap(), 0, len ))) return FALSE;
284 }
285 WideCharToMultiByte( CP_ACP, 0, str->NameString, str->Length, type, len, NULL, NULL);
286 type[newlen] = 0;
287 ret = lpfun(hmod,type,lparam);
288 }
289 else
290 {
291 ret = lpfun( hmod, UIntToPtr(et[i].u1.s2.Id), lparam );
292 }
293 if (!ret) break;
294 }
295 HeapFree( GetProcessHeap(), 0, type );
296 return ret;
297 }
298
299
300 /**********************************************************************
301 * EnumResourceTypesW (KERNEL32.@)
302 */
303 BOOL WINAPI EnumResourceTypesW( HMODULE hmod, ENUMRESTYPEPROCW lpfun, LONG_PTR lparam )
304 {
305 int i, len = 0;
306 BOOL ret = FALSE;
307 LPWSTR type = NULL;
308 NTSTATUS status;
309 const IMAGE_RESOURCE_DIRECTORY *resdir;
310 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
311 const IMAGE_RESOURCE_DIR_STRING_U *str;
312
313 TRACE( "%p %p %lx\n", hmod, lpfun, lparam );
314
315 if (!hmod) hmod = GetModuleHandleW( NULL );
316
317 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &resdir )) != STATUS_SUCCESS)
318 {
319 SetLastError( RtlNtStatusToDosError(status) );
320 return FALSE;
321 }
322 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
323 for (i = 0; i < resdir->NumberOfNamedEntries + resdir->NumberOfIdEntries; i++)
324 {
325 if (et[i].u1.s1.NameIsString)
326 {
327 str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const BYTE *)resdir + et[i].u1.s1.NameOffset);
328 if (str->Length + 1 > len)
329 {
330 len = str->Length + 1;
331 HeapFree( GetProcessHeap(), 0, type );
332 if (!(type = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) return FALSE;
333 }
334 memcpy(type, str->NameString, str->Length * sizeof (WCHAR));
335 type[str->Length] = 0;
336 ret = lpfun(hmod,type,lparam);
337 }
338 else
339 {
340 ret = lpfun( hmod, UIntToPtr(et[i].u1.s2.Id), lparam );
341 }
342 if (!ret) break;
343 }
344 HeapFree( GetProcessHeap(), 0, type );
345 return ret;
346 }
347
348
349 /**********************************************************************
350 * EnumResourceNamesA (KERNEL32.@)
351 */
352 BOOL WINAPI EnumResourceNamesA( HMODULE hmod, LPCSTR type, ENUMRESNAMEPROCA lpfun, LONG_PTR lparam )
353 {
354 int i;
355 BOOL ret = FALSE;
356 DWORD len = 0, newlen;
357 LPSTR name = NULL;
358 NTSTATUS status;
359 UNICODE_STRING typeW;
360 LDR_RESOURCE_INFO info;
361 const IMAGE_RESOURCE_DIRECTORY *basedir, *resdir;
362 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
363 const IMAGE_RESOURCE_DIR_STRING_U *str;
364
365 TRACE( "%p %s %p %lx\n", hmod, debugstr_a(type), lpfun, lparam );
366
367 if (!hmod) hmod = GetModuleHandleA( NULL );
368 typeW.Buffer = NULL;
369 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &basedir )) != STATUS_SUCCESS)
370 goto done;
371 if ((status = get_res_nameA( type, &typeW )) != STATUS_SUCCESS)
372 goto done;
373 info.Type = (ULONG_PTR)typeW.Buffer;
374 if ((status = LdrFindResourceDirectory_U( hmod, &info, 1, &resdir )) != STATUS_SUCCESS)
375 goto done;
376
377 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
378 for (i = 0; i < resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries; i++)
379 {
380 if (et[i].u1.s1.NameIsString)
381 {
382 str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const BYTE *)basedir + et[i].u1.s1.NameOffset);
383 newlen = WideCharToMultiByte(CP_ACP, 0, str->NameString, str->Length, NULL, 0, NULL, NULL);
384 if (newlen + 1 > len)
385 {
386 len = newlen + 1;
387 HeapFree( GetProcessHeap(), 0, name );
388 if (!(name = HeapAlloc(GetProcessHeap(), 0, len + 1 )))
389 {
390 ret = FALSE;
391 break;
392 }
393 }
394 WideCharToMultiByte( CP_ACP, 0, str->NameString, str->Length, name, len, NULL, NULL );
395 name[newlen] = 0;
396 ret = lpfun(hmod,type,name,lparam);
397 }
398 else
399 {
400 ret = lpfun( hmod, type, UIntToPtr(et[i].u1.s2.Id), lparam );
401 }
402 if (!ret) break;
403 }
404 done:
405 HeapFree( GetProcessHeap(), 0, name );
406 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
407 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
408 return ret;
409 }
410
411
412 /**********************************************************************
413 * EnumResourceNamesW (KERNEL32.@)
414 */
415 BOOL WINAPI EnumResourceNamesW( HMODULE hmod, LPCWSTR type, ENUMRESNAMEPROCW lpfun, LONG_PTR lparam )
416 {
417 int i, len = 0;
418 BOOL ret = FALSE;
419 LPWSTR name = NULL;
420 NTSTATUS status;
421 UNICODE_STRING typeW;
422 LDR_RESOURCE_INFO info;
423 const IMAGE_RESOURCE_DIRECTORY *basedir, *resdir;
424 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
425 const IMAGE_RESOURCE_DIR_STRING_U *str;
426
427 TRACE( "%p %s %p %lx\n", hmod, debugstr_w(type), lpfun, lparam );
428
429 if (!hmod) hmod = GetModuleHandleW( NULL );
430 typeW.Buffer = NULL;
431 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &basedir )) != STATUS_SUCCESS)
432 goto done;
433 if ((status = get_res_nameW( type, &typeW )) != STATUS_SUCCESS)
434 goto done;
435 info.Type = (ULONG_PTR)typeW.Buffer;
436 if ((status = LdrFindResourceDirectory_U( hmod, &info, 1, &resdir )) != STATUS_SUCCESS)
437 goto done;
438
439 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
440 for (i = 0; i < resdir->NumberOfNamedEntries+resdir->NumberOfIdEntries; i++)
441 {
442 if (et[i].u1.s1.NameIsString)
443 {
444 str = (const IMAGE_RESOURCE_DIR_STRING_U *)((const BYTE *)basedir + et[i].u1.s1.NameOffset);
445 if (str->Length + 1 > len)
446 {
447 len = str->Length + 1;
448 HeapFree( GetProcessHeap(), 0, name );
449 if (!(name = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) )))
450 {
451 ret = FALSE;
452 break;
453 }
454 }
455 memcpy(name, str->NameString, str->Length * sizeof (WCHAR));
456 name[str->Length] = 0;
457 ret = lpfun(hmod,type,name,lparam);
458 }
459 else
460 {
461 ret = lpfun( hmod, type, UIntToPtr(et[i].u1.s2.Id), lparam );
462 }
463 if (!ret) break;
464 }
465 done:
466 HeapFree( GetProcessHeap(), 0, name );
467 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
468 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
469 return ret;
470 }
471
472
473 /**********************************************************************
474 * EnumResourceLanguagesA (KERNEL32.@)
475 */
476 BOOL WINAPI EnumResourceLanguagesA( HMODULE hmod, LPCSTR type, LPCSTR name,
477 ENUMRESLANGPROCA lpfun, LONG_PTR lparam )
478 {
479 int i;
480 BOOL ret = FALSE;
481 NTSTATUS status;
482 UNICODE_STRING typeW, nameW;
483 LDR_RESOURCE_INFO info;
484 const IMAGE_RESOURCE_DIRECTORY *basedir, *resdir;
485 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
486
487 TRACE( "%p %s %s %p %lx\n", hmod, debugstr_a(type), debugstr_a(name), lpfun, lparam );
488
489 if (!hmod) hmod = GetModuleHandleA( NULL );
490 typeW.Buffer = nameW.Buffer = NULL;
491 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &basedir )) != STATUS_SUCCESS)
492 goto done;
493 if ((status = get_res_nameA( type, &typeW )) != STATUS_SUCCESS)
494 goto done;
495 if ((status = get_res_nameA( name, &nameW )) != STATUS_SUCCESS)
496 goto done;
497 info.Type = (ULONG_PTR)typeW.Buffer;
498 info.Name = (ULONG_PTR)nameW.Buffer;
499 if ((status = LdrFindResourceDirectory_U( hmod, &info, 2, &resdir )) != STATUS_SUCCESS)
500 goto done;
501
502 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
503 for (i = 0; i < resdir->NumberOfNamedEntries + resdir->NumberOfIdEntries; i++)
504 {
505 ret = lpfun( hmod, type, name, et[i].u1.s2.Id, lparam );
506 if (!ret) break;
507 }
508 done:
509 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
510 if (HIWORD(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
511 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
512 return ret;
513 }
514
515
516 /**********************************************************************
517 * EnumResourceLanguagesW (KERNEL32.@)
518 */
519 BOOL WINAPI EnumResourceLanguagesW( HMODULE hmod, LPCWSTR type, LPCWSTR name,
520 ENUMRESLANGPROCW lpfun, LONG_PTR lparam )
521 {
522 int i;
523 BOOL ret = FALSE;
524 NTSTATUS status;
525 UNICODE_STRING typeW, nameW;
526 LDR_RESOURCE_INFO info;
527 const IMAGE_RESOURCE_DIRECTORY *basedir, *resdir;
528 const IMAGE_RESOURCE_DIRECTORY_ENTRY *et;
529
530 TRACE( "%p %s %s %p %lx\n", hmod, debugstr_w(type), debugstr_w(name), lpfun, lparam );
531
532 if (!hmod) hmod = GetModuleHandleW( NULL );
533 typeW.Buffer = nameW.Buffer = NULL;
534 if ((status = LdrFindResourceDirectory_U( hmod, NULL, 0, &basedir )) != STATUS_SUCCESS)
535 goto done;
536 if ((status = get_res_nameW( type, &typeW )) != STATUS_SUCCESS)
537 goto done;
538 if ((status = get_res_nameW( name, &nameW )) != STATUS_SUCCESS)
539 goto done;
540 info.Type = (ULONG_PTR)typeW.Buffer;
541 info.Name = (ULONG_PTR)nameW.Buffer;
542 if ((status = LdrFindResourceDirectory_U( hmod, &info, 2, &resdir )) != STATUS_SUCCESS)
543 goto done;
544
545 et = (const IMAGE_RESOURCE_DIRECTORY_ENTRY *)(resdir + 1);
546 for (i = 0; i < resdir->NumberOfNamedEntries + resdir->NumberOfIdEntries; i++)
547 {
548 ret = lpfun( hmod, type, name, et[i].u1.s2.Id, lparam );
549 if (!ret) break;
550 }
551 done:
552 if (HIWORD(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
553 if (HIWORD(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
554 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
555 return ret;
556 }
557
558
559 /**********************************************************************
560 * LoadResource (KERNEL32.@)
561 */
562 HGLOBAL WINAPI LoadResource( HINSTANCE hModule, HRSRC hRsrc )
563 {
564 NTSTATUS status;
565 void *ret = NULL;
566
567 TRACE( "%p %p\n", hModule, hRsrc );
568
569 if (hModule && !HIWORD(hModule))
570 /* FIXME: should convert return to 32-bit resource */
571 return HGLOBAL_32( LoadResource16( HMODULE_16(hModule), HRSRC_16(hRsrc) ) );
572
573 if (!hRsrc) return 0;
574 if (!hModule) hModule = GetModuleHandleA( NULL );
575 status = LdrAccessResource( hModule, (IMAGE_RESOURCE_DATA_ENTRY *)hRsrc, &ret, NULL );
576 if (status != STATUS_SUCCESS) SetLastError( RtlNtStatusToDosError(status) );
577 return ret;
578 }
579
580
581 /**********************************************************************
582 * LockResource (KERNEL32.@)
583 */
584 LPVOID WINAPI LockResource( HGLOBAL handle )
585 {
586 TRACE("(%p)\n", handle );
587
588 if (HIWORD( handle )) /* 32-bit memory handle */
589 return (LPVOID)handle;
590
591 /* 16-bit memory handle */
592 return LockResource16( HGLOBAL_16(handle) );
593 }
594
595
596 /**********************************************************************
597 * FreeResource (KERNEL32.@)
598 */
599 BOOL WINAPI FreeResource( HGLOBAL handle )
600 {
601 if (HIWORD(handle)) return 0; /* 32-bit memory handle: nothing to do */
602 return FreeResource16( HGLOBAL_16(handle) );
603 }
604
605
606 /**********************************************************************
607 * SizeofResource (KERNEL32.@)
608 */
609 DWORD WINAPI SizeofResource( HINSTANCE hModule, HRSRC hRsrc )
610 {
611 if (hModule && !HIWORD(hModule))
612 return SizeofResource16( HMODULE_16(hModule), HRSRC_16(hRsrc) );
613
614 if (!hRsrc) return 0;
615 return ((PIMAGE_RESOURCE_DATA_ENTRY)hRsrc)->Size;
616 }
617
618 /*
619 * Data structure for updating resources.
620 * Type/Name/Language is a keyset for accessing resource data.
621 *
622 * QUEUEDUPDATES (root) ->
623 * list of struct resource_dir_entry (Type) ->
624 * list of struct resource_dir_entry (Name) ->
625 * list of struct resource_data Language + Data
626 */
627
628 typedef struct
629 {
630 LPWSTR pFileName;
631 BOOL bDeleteExistingResources;
632 struct list root;
633 } QUEUEDUPDATES;
634
635 /* this structure is shared for types and names */
636 struct resource_dir_entry {
637 struct list entry;
638 LPWSTR id;
639 struct list children;
640 };
641
642 /* this structure is the leaf */
643 struct resource_data {
644 struct list entry;
645 LANGID lang;
646 DWORD codepage;
647 DWORD cbData;
648 void *lpData;
649 };
650
651 static int resource_strcmp( LPCWSTR a, LPCWSTR b )
652 {
653 if ( a == b )
654 return 0;
655 if (HIWORD( a ) && HIWORD( b ) )
656 return lstrcmpW( a, b );
657 /* strings come before ids */
658 if (HIWORD( a ) && !HIWORD( b ))
659 return -1;
660 if (HIWORD( b ) && !HIWORD( a ))
661 return 1;
662 return ( a < b ) ? -1 : 1;
663 }
664
665 static struct resource_dir_entry *find_resource_dir_entry( struct list *dir, LPCWSTR id )
666 {
667 struct resource_dir_entry *ent;
668
669 /* match either IDs or strings */
670 LIST_FOR_EACH_ENTRY( ent, dir, struct resource_dir_entry, entry )
671 if (!resource_strcmp( id, ent->id ))
672 return ent;
673
674 return NULL;
675 }
676
677 static struct resource_data *find_resource_data( struct list *dir, LANGID lang )
678 {
679 struct resource_data *res_data;
680
681 /* match only languages here */
682 LIST_FOR_EACH_ENTRY( res_data, dir, struct resource_data, entry )
683 if ( lang == res_data->lang )
684 return res_data;
685
686 return NULL;
687 }
688
689 static void add_resource_dir_entry( struct list *dir, struct resource_dir_entry *resdir )
690 {
691 struct resource_dir_entry *ent;
692
693 LIST_FOR_EACH_ENTRY( ent, dir, struct resource_dir_entry, entry )
694 {
695 if (0>resource_strcmp( ent->id, resdir->id ))
696 continue;
697
698 list_add_before( &ent->entry, &resdir->entry );
699 return;
700 }
701 list_add_tail( dir, &resdir->entry );
702 }
703
704 static void add_resource_data_entry( struct list *dir, struct resource_data *resdata )
705 {
706 struct resource_data *ent;
707
708 LIST_FOR_EACH_ENTRY( ent, dir, struct resource_data, entry )
709 {
710 if (ent->lang < resdata->lang)
711 continue;
712
713 list_add_before( &ent->entry, &resdata->entry );
714 return;
715 }
716 list_add_tail( dir, &resdata->entry );
717 }
718
719 static LPWSTR res_strdupW( LPCWSTR str )
720 {
721 LPWSTR ret;
722 UINT len;
723
724 if (HIWORD(str) == 0)
725 return (LPWSTR) (UINT_PTR) LOWORD(str);
726 len = (lstrlenW( str ) + 1) * sizeof (WCHAR);
727 ret = HeapAlloc( GetProcessHeap(), 0, len );
728 memcpy( ret, str, len );
729 return ret;
730 }
731
732 static void res_free_str( LPWSTR str )
733 {
734 if (HIWORD(str))
735 HeapFree( GetProcessHeap(), 0, str );
736 }
737
738 static BOOL update_add_resource( QUEUEDUPDATES *updates, LPCWSTR Type, LPCWSTR Name,
739 struct resource_data *resdata, BOOL overwrite_existing )
740 {
741 struct resource_dir_entry *restype, *resname;
742 struct resource_data *existing;
743
744 TRACE("%p %s %s %p %d\n", updates,
745 debugstr_w(Type), debugstr_w(Name), resdata, overwrite_existing );
746
747 restype = find_resource_dir_entry( &updates->root, Type );
748 if (!restype)
749 {
750 restype = HeapAlloc( GetProcessHeap(), 0, sizeof *restype );
751 restype->id = res_strdupW( Type );
752 list_init( &restype->children );
753 add_resource_dir_entry( &updates->root, restype );
754 }
755
756 resname = find_resource_dir_entry( &restype->children, Name );
757 if (!resname)
758 {
759 resname = HeapAlloc( GetProcessHeap(), 0, sizeof *resname );
760 resname->id = res_strdupW( Name );
761 list_init( &resname->children );
762 add_resource_dir_entry( &restype->children, resname );
763 }
764
765 /*
766 * If there's an existing resource entry with matching (Type,Name,Language)
767 * it needs to be removed before adding the new data.
768 */
769 existing = find_resource_data( &resname->children, resdata->lang );
770 if (existing)
771 {
772 if (!overwrite_existing)
773 return TRUE;
774 list_remove( &existing->entry );
775 HeapFree( GetProcessHeap(), 0, existing );
776 }
777
778 add_resource_data_entry( &resname->children, resdata );
779
780 return TRUE;
781 }
782
783 static struct resource_data *allocate_resource_data( WORD Language, DWORD codepage,
784 LPVOID lpData, DWORD cbData, BOOL copy_data )
785 {
786 struct resource_data *resdata;
787
788 if (!lpData || !cbData)
789 return NULL;
790
791 resdata = HeapAlloc( GetProcessHeap(), 0, sizeof *resdata + (copy_data ? cbData : 0) );
792 if (resdata)
793 {
794 resdata->lang = Language;
795 resdata->codepage = codepage;
796 resdata->cbData = cbData;
797 if (copy_data)
798 {
799 resdata->lpData = &resdata[1];
800 memcpy( resdata->lpData, lpData, cbData );
801 }
802 else
803 resdata->lpData = lpData;
804 }
805
806 return resdata;
807 }
808
809 static void free_resource_directory( struct list *head, int level )
810 {
811 struct list *ptr = NULL;
812
813 while ((ptr = list_head( head )))
814 {
815 list_remove( ptr );
816 if (level)
817 {
818 struct resource_dir_entry *ent;
819
820 ent = LIST_ENTRY( ptr, struct resource_dir_entry, entry );
821 res_free_str( ent->id );
822 free_resource_directory( &ent->children, level - 1 );
823 HeapFree(GetProcessHeap(), 0, ent);
824 }
825 else
826 {
827 struct resource_data *data;
828
829 data = LIST_ENTRY( ptr, struct resource_data, entry );
830 HeapFree( GetProcessHeap(), 0, data );
831 }
832 }
833 }
834
835 static IMAGE_NT_HEADERS *get_nt_header( void *base, DWORD mapping_size )
836 {
837 IMAGE_NT_HEADERS *nt;
838 IMAGE_DOS_HEADER *dos;
839
840 if (mapping_size<sizeof (*dos))
841 return NULL;
842
843 dos = base;
844 if (dos->e_magic != IMAGE_DOS_SIGNATURE)
845 return NULL;
846
847 if ((dos->e_lfanew + sizeof (*nt)) > mapping_size)
848 return NULL;
849
850 nt = (void*) ((BYTE*)base + dos->e_lfanew);
851
852 if (nt->Signature != IMAGE_NT_SIGNATURE)
853 return NULL;
854
855 return nt;
856 }
857
858 static IMAGE_SECTION_HEADER *get_section_header( void *base, DWORD mapping_size, DWORD *num_sections )
859 {
860 IMAGE_NT_HEADERS *nt;
861 IMAGE_SECTION_HEADER *sec;
862 DWORD section_ofs;
863
864 nt = get_nt_header( base, mapping_size );
865 if (!nt)
866 return NULL;
867
868 /* check that we don't go over the end of the file accessing the sections */
869 section_ofs = FIELD_OFFSET(IMAGE_NT_HEADERS, OptionalHeader) + nt->FileHeader.SizeOfOptionalHeader;
870 if ((nt->FileHeader.NumberOfSections * sizeof (*sec) + section_ofs) > mapping_size)
871 return NULL;
872
873 if (num_sections)
874 *num_sections = nt->FileHeader.NumberOfSections;
875
876 /* from here we have a valid PE exe to update */
877 return (void*) ((BYTE*)nt + section_ofs);
878 }
879
880 static BOOL check_pe_exe( HANDLE file, QUEUEDUPDATES *updates )
881 {
882 const IMAGE_NT_HEADERS *nt;
883 const IMAGE_SECTION_HEADER *sec;
884 BOOL ret = FALSE;
885 HANDLE mapping;
886 DWORD mapping_size, num_sections = 0;
887 void *base = NULL;
888
889 mapping_size = GetFileSize( file, NULL );
890
891 mapping = CreateFileMappingW( file, NULL, PAGE_READONLY, 0, 0, NULL );
892 if (!mapping)
893 goto done;
894
895 base = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, mapping_size );
896 if (!base)
897 goto done;
898
899 nt = get_nt_header( base, mapping_size );
900 if (!nt)
901 goto done;
902
903 TRACE("resources: %08x %08x\n",
904 nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress,
905 nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size);
906
907 sec = get_section_header( base, mapping_size, &num_sections );
908 if (!sec)
909 goto done;
910
911 ret = TRUE;
912
913 done:
914 if (base)
915 UnmapViewOfFile( base );
916 if (mapping)
917 CloseHandle( mapping );
918
919 return ret;
920 }
921
922 struct resource_size_info {
923 DWORD types_ofs;
924 DWORD names_ofs;
925 DWORD langs_ofs;
926 DWORD data_entry_ofs;
927 DWORD strings_ofs;
928 DWORD data_ofs;
929 DWORD total_size;
930 };
931
932 struct mapping_info {
933 HANDLE file;
934 HANDLE mapping;
935 void *base;
936 DWORD size;
937 BOOL read_write;
938 };
939
940 static const IMAGE_SECTION_HEADER *section_from_rva( void *base, DWORD mapping_size, DWORD rva )
941 {
942 const IMAGE_SECTION_HEADER *sec;
943 DWORD num_sections = 0;
944 int i;
945
946 sec = get_section_header( base, mapping_size, &num_sections );
947 if (!sec)
948 return NULL;
949
950 for (i=num_sections-1; i>=0; i--)
951 {
952 if (sec[i].VirtualAddress <= rva &&
953 rva <= (DWORD)sec[i].VirtualAddress + sec[i].SizeOfRawData)
954 {
955 return &sec[i];
956 }
957 }
958
959 return NULL;
960 }
961
962 static void *address_from_rva( void *base, DWORD mapping_size, DWORD rva, DWORD len )
963 {
964 const IMAGE_SECTION_HEADER *sec;
965
966 sec = section_from_rva( base, mapping_size, rva );
967 if (!sec)
968 return NULL;
969
970 if (rva + len <= (DWORD)sec->VirtualAddress + sec->SizeOfRawData)
971 return (void*)((LPBYTE) base + (sec->PointerToRawData + rva - sec->VirtualAddress));
972
973 return NULL;
974 }
975
976 static LPWSTR resource_dup_string( const IMAGE_RESOURCE_DIRECTORY *root, const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry )
977 {
978 const IMAGE_RESOURCE_DIR_STRING_U* string;
979 LPWSTR s;
980
981 if (!entry->u1.s1.NameIsString)
982 return UIntToPtr(entry->u1.s2.Id);
983
984 string = (const IMAGE_RESOURCE_DIR_STRING_U*) (((const char *)root) + entry->u1.s1.NameOffset);
985 s = HeapAlloc(GetProcessHeap(), 0, (string->Length + 1)*sizeof (WCHAR) );
986 memcpy( s, string->NameString, (string->Length + 1)*sizeof (WCHAR) );
987 s[string->Length] = 0;
988
989 return s;
990 }
991
992 /* this function is based on the code in winedump's pe.c */
993 static BOOL enumerate_mapped_resources( QUEUEDUPDATES *updates,
994 void *base, DWORD mapping_size,
995 const IMAGE_RESOURCE_DIRECTORY *root )
996 {
997 const IMAGE_RESOURCE_DIRECTORY *namedir, *langdir;
998 const IMAGE_RESOURCE_DIRECTORY_ENTRY *e1, *e2, *e3;
999 const IMAGE_RESOURCE_DATA_ENTRY *data;
1000 DWORD i, j, k;
1001
1002 TRACE("version (%d.%d) %d named %d id entries\n",
1003 root->MajorVersion, root->MinorVersion, root->NumberOfNamedEntries, root->NumberOfIdEntries);
1004
1005 for (i = 0; i< root->NumberOfNamedEntries + root->NumberOfIdEntries; i++)
1006 {
1007 LPWSTR Type;
1008
1009 e1 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(root + 1) + i;
1010
1011 Type = resource_dup_string( root, e1 );
1012
1013 namedir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e1->u2.s3.OffsetToDirectory);
1014 for (j = 0; j < namedir->NumberOfNamedEntries + namedir->NumberOfIdEntries; j++)
1015 {
1016 LPWSTR Name;
1017
1018 e2 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(namedir + 1) + j;
1019
1020 Name = resource_dup_string( root, e2 );
1021
1022 langdir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e2->u2.s3.OffsetToDirectory);
1023 for (k = 0; k < langdir->NumberOfNamedEntries + langdir->NumberOfIdEntries; k++)
1024 {
1025 LANGID Lang;
1026 void *p;
1027 struct resource_data *resdata;
1028
1029 e3 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(langdir + 1) + k;
1030
1031 Lang = e3->u1.s2.Id;
1032
1033 data = (const IMAGE_RESOURCE_DATA_ENTRY *)((const char *)root + e3->u2.OffsetToData);
1034
1035 p = address_from_rva( base, mapping_size, data->OffsetToData, data->Size );
1036
1037 resdata = allocate_resource_data( Lang, data->CodePage, p, data->Size, FALSE );
1038 if (resdata)
1039 update_add_resource( updates, Type, Name, resdata, FALSE );
1040 }
1041 res_free_str( Name );
1042 }
1043 res_free_str( Type );
1044 }
1045
1046 return TRUE;
1047 }
1048
1049 static BOOL read_mapped_resources( QUEUEDUPDATES *updates, void *base, DWORD mapping_size )
1050 {
1051 const IMAGE_RESOURCE_DIRECTORY *root;
1052 const IMAGE_NT_HEADERS *nt;
1053 const IMAGE_SECTION_HEADER *sec;
1054 DWORD num_sections = 0, i;
1055
1056 nt = get_nt_header( base, mapping_size );
1057 if (!nt)
1058 return FALSE;
1059
1060 sec = get_section_header( base, mapping_size, &num_sections );
1061 if (!sec)
1062 return FALSE;
1063
1064 for (i=0; i<num_sections; i++)
1065 if (!memcmp(sec[i].Name, ".rsrc", 6))
1066 break;
1067
1068 if (i == num_sections)
1069 return TRUE;
1070
1071 /* check the resource data is inside the mapping */
1072 if (sec[i].PointerToRawData > mapping_size ||
1073 (sec[i].PointerToRawData + sec[i].SizeOfRawData) > mapping_size)
1074 return TRUE;
1075
1076 TRACE("found .rsrc at %08x, size %08x\n", sec[i].PointerToRawData, sec[i].SizeOfRawData);
1077
1078 root = (void*) ((BYTE*)base + sec[i].PointerToRawData);
1079 enumerate_mapped_resources( updates, base, mapping_size, root );
1080
1081 return TRUE;
1082 }
1083
1084 static BOOL map_file_into_memory( struct mapping_info *mi )
1085 {
1086 DWORD page_attr, perm;
1087
1088 if (mi->read_write)
1089 {
1090 page_attr = PAGE_READWRITE;
1091 perm = FILE_MAP_WRITE | FILE_MAP_READ;
1092 }
1093 else
1094 {
1095 page_attr = PAGE_READONLY;
1096 perm = FILE_MAP_READ;
1097 }
1098
1099 mi->mapping = CreateFileMappingW( mi->file, NULL, page_attr, 0, 0, NULL );
1100 if (!mi->mapping)
1101 return FALSE;
1102
1103 mi->base = MapViewOfFile( mi->mapping, perm, 0, 0, mi->size );
1104 if (!mi->base)
1105 return FALSE;
1106
1107 return TRUE;
1108 }
1109
1110 static BOOL unmap_file_from_memory( struct mapping_info *mi )
1111 {
1112 if (mi->base)
1113 UnmapViewOfFile( mi->base );
1114 mi->base = NULL;
1115 if (mi->mapping)
1116 CloseHandle( mi->mapping );
1117 mi->mapping = NULL;
1118 return TRUE;
1119 }
1120
1121 static void destroy_mapping( struct mapping_info *mi )
1122 {
1123 if (!mi)
1124 return;
1125 unmap_file_from_memory( mi );
1126 if (mi->file)
1127 CloseHandle( mi->file );
1128 HeapFree( GetProcessHeap(), 0, mi );
1129 }
1130
1131 static struct mapping_info *create_mapping( LPCWSTR name, BOOL rw )
1132 {
1133 struct mapping_info *mi;
1134
1135 mi = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *mi );
1136 if (!mi)
1137 return NULL;
1138
1139 mi->read_write = rw;
1140
1141 mi->file = CreateFileW( name, GENERIC_READ | (rw ? GENERIC_WRITE : 0),
1142 0, NULL, OPEN_EXISTING, 0, 0 );
1143
1144 if (mi->file != INVALID_HANDLE_VALUE)
1145 {
1146 mi->size = GetFileSize( mi->file, NULL );
1147
1148 if (map_file_into_memory( mi ))
1149 return mi;
1150 }
1151
1152 unmap_file_from_memory( mi );
1153 HeapFree( GetProcessHeap(), 0, mi );
1154
1155 return NULL;
1156 }
1157
1158 static BOOL resize_mapping( struct mapping_info *mi, DWORD new_size )
1159 {
1160 if (!unmap_file_from_memory( mi ))
1161 return FALSE;
1162
1163 /* change the file size */
1164 SetFilePointer( mi->file, new_size, NULL, FILE_BEGIN );
1165 if (!SetEndOfFile( mi->file ))
1166 {
1167 ERR("failed to set file size to %08x\n", new_size );
1168 return FALSE;
1169 }
1170
1171 mi->size = new_size;
1172
1173 return map_file_into_memory( mi );
1174 }
1175
1176 static void get_resource_sizes( QUEUEDUPDATES *updates, struct resource_size_info *si )
1177 {
1178 struct resource_dir_entry *types, *names;
1179 struct resource_data *data;
1180 DWORD num_types = 0, num_names = 0, num_langs = 0, strings_size = 0, data_size = 0;
1181
1182 memset( si, 0, sizeof *si );
1183
1184 LIST_FOR_EACH_ENTRY( types, &updates->root, struct resource_dir_entry, entry )
1185 {
1186 num_types++;
1187 if (HIWORD( types->id ))
1188 strings_size += sizeof (WORD) + lstrlenW( types->id )*sizeof (WCHAR);
1189
1190 LIST_FOR_EACH_ENTRY( names, &types->children, struct resource_dir_entry, entry )
1191 {
1192 num_names++;
1193
1194 if (HIWORD( names->id ))
1195 strings_size += sizeof (WORD) + lstrlenW( names->id )*sizeof (WCHAR);
1196
1197 LIST_FOR_EACH_ENTRY( data, &names->children, struct resource_data, entry )
1198 {
1199 num_langs++;
1200 data_size += (data->cbData + 3) & ~3;
1201 }
1202 }
1203 }
1204
1205 /* names are at the end of the types */
1206 si->names_ofs = sizeof (IMAGE_RESOURCE_DIRECTORY) +
1207 num_types * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1208
1209 /* language directories are at the end of the names */
1210 si->langs_ofs = si->names_ofs +
1211 num_types * sizeof (IMAGE_RESOURCE_DIRECTORY) +
1212 num_names * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1213
1214 si->data_entry_ofs = si->langs_ofs +
1215 num_names * sizeof (IMAGE_RESOURCE_DIRECTORY) +
1216 num_langs * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1217
1218 si->strings_ofs = si->data_entry_ofs +
1219 num_langs * sizeof (IMAGE_RESOURCE_DATA_ENTRY);
1220
1221 si->data_ofs = si->strings_ofs + ((strings_size + 3) & ~3);
1222
1223 si->total_size = si->data_ofs + data_size;
1224
1225 TRACE("names %08x langs %08x data entries %08x strings %08x data %08x total %08x\n",
1226 si->names_ofs, si->langs_ofs, si->data_entry_ofs,
1227 si->strings_ofs, si->data_ofs, si->total_size);
1228 }
1229
1230 static void res_write_padding( BYTE *res_base, DWORD size )
1231 {
1232 static const BYTE pad[] = {
1233 'P','A','D','D','I','N','G','X','X','P','A','D','D','I','N','G' };
1234 DWORD i;
1235
1236 for ( i = 0; i < size / sizeof pad; i++ )
1237 memcpy( &res_base[i*sizeof pad], pad, sizeof pad );
1238 memcpy( &res_base[i*sizeof pad], pad, size%sizeof pad );
1239 }
1240
1241 static BOOL write_resources( QUEUEDUPDATES *updates, LPBYTE base, struct resource_size_info *si, DWORD rva )
1242 {
1243 struct resource_dir_entry *types, *names;
1244 struct resource_data *data;
1245 IMAGE_RESOURCE_DIRECTORY *root;
1246
1247 TRACE("%p %p %p %08x\n", updates, base, si, rva );
1248
1249 memset( base, 0, si->total_size );
1250
1251 /* the root entry always exists */
1252 root = (IMAGE_RESOURCE_DIRECTORY*) base;
1253 memset( root, 0, sizeof *root );
1254 root->MajorVersion = 4;
1255 si->types_ofs = sizeof *root;
1256 LIST_FOR_EACH_ENTRY( types, &updates->root, struct resource_dir_entry, entry )
1257 {
1258 IMAGE_RESOURCE_DIRECTORY_ENTRY *e1;
1259 IMAGE_RESOURCE_DIRECTORY *namedir;
1260
1261 e1 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->types_ofs];
1262 memset( e1, 0, sizeof *e1 );
1263 if (HIWORD( types->id ))
1264 {
1265 WCHAR *strings;
1266 DWORD len;
1267
1268 root->NumberOfNamedEntries++;
1269 e1->u1.s1.NameIsString = 1;
1270 e1->u1.s1.NameOffset = si->strings_ofs;
1271
1272 strings = (WCHAR*) &base[si->strings_ofs];
1273 len = lstrlenW( types->id );
1274 strings[0] = len;
1275 memcpy( &strings[1], types->id, len * sizeof (WCHAR) );
1276 si->strings_ofs += (len + 1) * sizeof (WCHAR);
1277 }
1278 else
1279 {
1280 root->NumberOfIdEntries++;
1281 e1->u1.s2.Id = LOWORD( types->id );
1282 }
1283 e1->u2.s3.OffsetToDirectory = si->names_ofs;
1284 e1->u2.s3.DataIsDirectory = TRUE;
1285 si->types_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1286
1287 namedir = (IMAGE_RESOURCE_DIRECTORY*) &base[si->names_ofs];
1288 memset( namedir, 0, sizeof *namedir );
1289 namedir->MajorVersion = 4;
1290 si->names_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY);
1291
1292 LIST_FOR_EACH_ENTRY( names, &types->children, struct resource_dir_entry, entry )
1293 {
1294 IMAGE_RESOURCE_DIRECTORY_ENTRY *e2;
1295 IMAGE_RESOURCE_DIRECTORY *langdir;
1296
1297 e2 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->names_ofs];
1298 memset( e2, 0, sizeof *e2 );
1299 if (HIWORD( names->id ))
1300 {
1301 WCHAR *strings;
1302 DWORD len;
1303
1304 namedir->NumberOfNamedEntries++;
1305 e2->u1.s1.NameIsString = 1;
1306 e2->u1.s1.NameOffset = si->strings_ofs;
1307
1308 strings = (WCHAR*) &base[si->strings_ofs];
1309 len = lstrlenW( names->id );
1310 strings[0] = len;
1311 memcpy( &strings[1], names->id, len * sizeof (WCHAR) );
1312 si->strings_ofs += (len + 1) * sizeof (WCHAR);
1313 }
1314 else
1315 {
1316 namedir->NumberOfIdEntries++;
1317 e2->u1.s2.Id = LOWORD( names->id );
1318 }
1319 e2->u2.s3.OffsetToDirectory = si->langs_ofs;
1320 e2->u2.s3.DataIsDirectory = TRUE;
1321 si->names_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1322
1323 langdir = (IMAGE_RESOURCE_DIRECTORY*) &base[si->langs_ofs];
1324 memset( langdir, 0, sizeof *langdir );
1325 langdir->MajorVersion = 4;
1326 si->langs_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY);
1327
1328 LIST_FOR_EACH_ENTRY( data, &names->children, struct resource_data, entry )
1329 {
1330 IMAGE_RESOURCE_DIRECTORY_ENTRY *e3;
1331 IMAGE_RESOURCE_DATA_ENTRY *de;
1332 int pad_size;
1333
1334 e3 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->langs_ofs];
1335 memset( e3, 0, sizeof *e3 );
1336 langdir->NumberOfIdEntries++;
1337 e3->u1.s2.Id = LOWORD( data->lang );
1338 e3->u2.OffsetToData = si->data_entry_ofs;
1339
1340 si->langs_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
1341
1342 /* write out all the data entries */
1343 de = (IMAGE_RESOURCE_DATA_ENTRY*) &base[si->data_entry_ofs];
1344 memset( de, 0, sizeof *de );
1345 de->OffsetToData = si->data_ofs + rva;
1346 de->Size = data->cbData;
1347 de->CodePage = data->codepage;
1348 si->data_entry_ofs += sizeof (IMAGE_RESOURCE_DATA_ENTRY);
1349
1350 /* write out the resource data */
1351 memcpy( &base[si->data_ofs], data->lpData, data->cbData );
1352 si->data_ofs += data->cbData;
1353
1354 pad_size = (-si->data_ofs)&3;
1355 res_write_padding( &base[si->data_ofs], pad_size );
1356 si->data_ofs += pad_size;
1357 }
1358 }
1359 }
1360
1361 return TRUE;
1362 }
1363
1364 /*
1365 * FIXME:
1366 * Assumes that the resources are in .rsrc
1367 * and .rsrc is the last section in the file.
1368 * Not sure whether updating resources will other cases on Windows.
1369 * If the resources lie in a section containing other data,
1370 * resizing that section could possibly cause trouble.
1371 * If the section with the resources isn't last, the remaining
1372 * sections need to be moved down in the file, and the section header
1373 * would need to be adjusted.
1374 * If we needed to add a section, what would we name it?
1375 * If we needed to add a section and there wasn't space in the file
1376 * header, how would that work?
1377 * Seems that at least some of these cases can't be handled properly.
1378 */
1379 static IMAGE_SECTION_HEADER *get_resource_section( void *base, DWORD mapping_size )
1380 {
1381 IMAGE_SECTION_HEADER *sec;
1382 IMAGE_NT_HEADERS *nt;
1383 DWORD i, num_sections = 0;
1384
1385 nt = get_nt_header( base, mapping_size );
1386 if (!nt)
1387 return NULL;
1388
1389 sec = get_section_header( base, mapping_size, &num_sections );
1390 if (!sec)
1391 return NULL;
1392
1393 /* find the resources section */
1394 for (i=0; i<num_sections; i++)
1395 if (!memcmp(sec[i].Name, ".rsrc", 6))
1396 break;
1397
1398 if (i == num_sections)
1399 {
1400 FIXME(".rsrc doesn't exist\n");
1401 return NULL;
1402 }
1403
1404 /* check that the resources section is last */
1405 if (i != num_sections - 1)
1406 {
1407 FIXME(".rsrc isn't the last section\n");
1408 return NULL;
1409 }
1410
1411 return &sec[i];
1412 }
1413
1414 static DWORD get_init_data_size( void *base, DWORD mapping_size )
1415 {
1416 DWORD i, sz = 0, num_sections = 0;
1417 IMAGE_SECTION_HEADER *s;
1418
1419 s = get_section_header( base, mapping_size, &num_sections );
1420
1421 for (i=0; i<num_sections; i++)
1422 if (s[i].Characteristics & IMAGE_SCN_CNT_INITIALIZED_DATA)
1423 sz += s[i].SizeOfRawData;
1424
1425 TRACE("size = %08x\n", sz);
1426
1427 return sz;
1428 }
1429
1430 static BOOL write_raw_resources( QUEUEDUPDATES *updates )
1431 {
1432 static const WCHAR prefix[] = { 'r','e','s','u',0 };
1433 WCHAR tempdir[MAX_PATH], tempfile[MAX_PATH];
1434 DWORD mapping_size, section_size, old_size;
1435 BOOL ret = FALSE;
1436 IMAGE_SECTION_HEADER *sec;
1437 IMAGE_NT_HEADERS *nt;
1438 struct resource_size_info res_size;
1439 BYTE *res_base;
1440 struct mapping_info *read_map = NULL, *write_map = NULL;
1441
1442 /* copy the exe to a temp file then update the temp file... */
1443 tempdir[0] = 0;
1444 if (!GetTempPathW( MAX_PATH, tempdir ))
1445 return ret;
1446
1447 if (!GetTempFileNameW( tempdir, prefix, 0, tempfile ))
1448 return ret;
1449
1450 if (!CopyFileW( updates->pFileName, tempfile, FALSE ))
1451 goto done;
1452
1453 TRACE("tempfile %s\n", debugstr_w(tempfile));
1454
1455 if (!updates->bDeleteExistingResources)
1456 {
1457 read_map = create_mapping( updates->pFileName, FALSE );
1458 if (!read_map)
1459 goto done;
1460
1461 ret = read_mapped_resources( updates, read_map->base, read_map->size );
1462 if (!ret)
1463 {
1464 ERR("failed to read existing resources\n");
1465 goto done;
1466 }
1467 }
1468
1469 write_map = create_mapping( tempfile, TRUE );
1470 if (!write_map)
1471 goto done;
1472
1473 nt = get_nt_header( write_map->base, write_map->size );
1474 if (!nt)
1475 goto done;
1476
1477 if (nt->OptionalHeader.SectionAlignment <= 0)
1478 {
1479 ERR("invalid section alignment %04x\n", nt->OptionalHeader.SectionAlignment);
1480 goto done;
1481 }
1482
1483 sec = get_resource_section( write_map->base, write_map->size );
1484 if (!sec)
1485 goto done;
1486
1487 if ((sec->SizeOfRawData + sec->PointerToRawData) != write_map->size)
1488 {
1489 FIXME(".rsrc isn't at the end of the image %08x + %08x != %08x\n",
1490 sec->SizeOfRawData, sec->PointerToRawData, write_map->size);
1491 goto done;
1492 }
1493
1494 TRACE("before .rsrc at %08x, size %08x\n", sec->PointerToRawData, sec->SizeOfRawData);
1495
1496 get_resource_sizes( updates, &res_size );
1497
1498 /* round up the section size */
1499 section_size = res_size.total_size;
1500 section_size += (-section_size) % nt->OptionalHeader.SectionAlignment;
1501
1502 mapping_size = sec->PointerToRawData + section_size;
1503
1504 TRACE("requires %08x (%08x) bytes\n", res_size.total_size, section_size );
1505
1506 /* check if the file size needs to be changed */
1507 if (section_size != sec->SizeOfRawData)
1508 {
1509 old_size = write_map->size;
1510
1511 TRACE("file size %08x -> %08x\n", old_size, mapping_size);
1512
1513 /* unmap the file before changing the file size */
1514 ret = resize_mapping( write_map, mapping_size );
1515
1516 /* get the pointers again - they might be different after remapping */
1517 nt = get_nt_header( write_map->base, mapping_size );
1518 if (!nt)
1519 {
1520 ERR("couldn't get NT header\n");
1521 goto done;
1522 }
1523
1524 sec = get_resource_section( write_map->base, mapping_size );
1525 if (!sec)
1526 goto done;
1527
1528 /* adjust the PE header information */
1529 nt->OptionalHeader.SizeOfImage += (mapping_size - old_size);
1530 sec->SizeOfRawData = section_size;
1531 sec->Misc.VirtualSize = section_size;
1532 nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size;
1533 nt->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size );
1534 }
1535
1536 res_base = (LPBYTE) write_map->base + sec->PointerToRawData;
1537
1538 TRACE("base = %p offset = %08x\n", write_map->base, sec->PointerToRawData);
1539
1540 ret = write_resources( updates, res_base, &res_size, sec->VirtualAddress );
1541
1542 res_write_padding( res_base + res_size.total_size, section_size - res_size.total_size );
1543
1544 TRACE("after .rsrc at %08x, size %08x\n", sec->PointerToRawData, sec->SizeOfRawData);
1545
1546 done:
1547 destroy_mapping( read_map );
1548 destroy_mapping( write_map );
1549
1550 if (ret)
1551 ret = CopyFileW( tempfile, updates->pFileName, FALSE );
1552
1553 DeleteFileW( tempfile );
1554
1555 return ret;
1556 }
1557
1558 /***********************************************************************
1559 * BeginUpdateResourceW (KERNEL32.@)
1560 */
1561 HANDLE WINAPI BeginUpdateResourceW( LPCWSTR pFileName, BOOL bDeleteExistingResources )
1562 {
1563 QUEUEDUPDATES *updates = NULL;
1564 HANDLE hUpdate, file, ret = NULL;
1565
1566 TRACE("%s, %d\n", debugstr_w(pFileName), bDeleteExistingResources);
1567
1568 hUpdate = GlobalAlloc(GHND, sizeof(QUEUEDUPDATES));
1569 if (!hUpdate)
1570 return ret;
1571
1572 updates = GlobalLock(hUpdate);
1573 if (updates)
1574 {
1575 list_init( &updates->root );
1576 updates->bDeleteExistingResources = bDeleteExistingResources;
1577 updates->pFileName = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(pFileName)+1)*sizeof(WCHAR));
1578 if (updates->pFileName)
1579 {
1580 lstrcpyW(updates->pFileName, pFileName);
1581
1582 file = CreateFileW( pFileName, GENERIC_READ | GENERIC_WRITE,
1583 0, NULL, OPEN_EXISTING, 0, 0 );
1584
1585 /* if resources are deleted, only the file's presence is checked */
1586 if (file != INVALID_HANDLE_VALUE &&
1587 (bDeleteExistingResources || check_pe_exe( file, updates )))
1588 ret = hUpdate;
1589 else
1590 HeapFree( GetProcessHeap(), 0, updates->pFileName );
1591
1592 CloseHandle( file );
1593 }
1594 GlobalUnlock(hUpdate);
1595 }
1596
1597 if (!ret)
1598 GlobalFree(hUpdate);
1599
1600 return ret;
1601 }
1602
1603
1604 /***********************************************************************
1605 * BeginUpdateResourceA (KERNEL32.@)
1606 */
1607 HANDLE WINAPI BeginUpdateResourceA( LPCSTR pFileName, BOOL bDeleteExistingResources )
1608 {
1609 UNICODE_STRING FileNameW;
1610 HANDLE ret;
1611 RtlCreateUnicodeStringFromAsciiz(&FileNameW, pFileName);
1612 ret = BeginUpdateResourceW(FileNameW.Buffer, bDeleteExistingResources);
1613 RtlFreeUnicodeString(&FileNameW);
1614 return ret;
1615 }
1616
1617
1618 /***********************************************************************
1619 * EndUpdateResourceW (KERNEL32.@)
1620 */
1621 BOOL WINAPI EndUpdateResourceW( HANDLE hUpdate, BOOL fDiscard )
1622 {
1623 QUEUEDUPDATES *updates;
1624 BOOL ret;
1625
1626 TRACE("%p %d\n", hUpdate, fDiscard);
1627
1628 updates = GlobalLock(hUpdate);
1629 if (!updates)
1630 return FALSE;
1631
1632 ret = fDiscard || write_raw_resources( updates );
1633
1634 free_resource_directory( &updates->root, 2 );
1635
1636 HeapFree( GetProcessHeap(), 0, updates->pFileName );
1637 GlobalUnlock( hUpdate );
1638 GlobalFree( hUpdate );
1639
1640 return ret;
1641 }
1642
1643
1644 /***********************************************************************
1645 * EndUpdateResourceA (KERNEL32.@)
1646 */
1647 BOOL WINAPI EndUpdateResourceA( HANDLE hUpdate, BOOL fDiscard )
1648 {
1649 return EndUpdateResourceW(hUpdate, fDiscard);
1650 }
1651
1652
1653 /***********************************************************************
1654 * UpdateResourceW (KERNEL32.@)
1655 */
1656 BOOL WINAPI UpdateResourceW( HANDLE hUpdate, LPCWSTR lpType, LPCWSTR lpName,
1657 WORD wLanguage, LPVOID lpData, DWORD cbData)
1658 {
1659 QUEUEDUPDATES *updates;
1660 BOOL ret = FALSE;
1661
1662 TRACE("%p %s %s %08x %p %d\n", hUpdate,
1663 debugstr_w(lpType), debugstr_w(lpName), wLanguage, lpData, cbData);
1664
1665 updates = GlobalLock(hUpdate);
1666 if (updates)
1667 {
1668 struct resource_data *data;
1669 data = allocate_resource_data( wLanguage, 0, lpData, cbData, TRUE );
1670 if (data)
1671 ret = update_add_resource( updates, lpType, lpName, data, TRUE );
1672 GlobalUnlock(hUpdate);
1673 }
1674 return ret;
1675 }
1676
1677
1678 /***********************************************************************
1679 * UpdateResourceA (KERNEL32.@)
1680 */
1681 BOOL WINAPI UpdateResourceA( HANDLE hUpdate, LPCSTR lpType, LPCSTR lpName,
1682 WORD wLanguage, LPVOID lpData, DWORD cbData)
1683 {
1684 BOOL ret;
1685 UNICODE_STRING TypeW;
1686 UNICODE_STRING NameW;
1687 if(!HIWORD(lpType))
1688 TypeW.Buffer = ULongToPtr(LOWORD(lpType));
1689 else
1690 RtlCreateUnicodeStringFromAsciiz(&TypeW, lpType);
1691 if(!HIWORD(lpName))
1692 NameW.Buffer = ULongToPtr(LOWORD(lpName));
1693 else
1694 RtlCreateUnicodeStringFromAsciiz(&NameW, lpName);
1695 ret = UpdateResourceW(hUpdate, TypeW.Buffer, NameW.Buffer, wLanguage, lpData, cbData);
1696 if(HIWORD(lpType)) RtlFreeUnicodeString(&TypeW);
1697 if(HIWORD(lpName)) RtlFreeUnicodeString(&NameW);
1698 return ret;
1699 }
useful hacks and other patches not (yet) suitable for mainline