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