dlls/kernel32/resource.c

   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 <stdarg.h>
  24
  25 #define NONAMELESSUNION
  26 #define NONAMELESSSTRUCT
  27 #include "ntstatus.h"
  28 #define WIN32_NO_STATUS
  29 #include "windef.h"
  30 #include "winbase.h"
  31 #include "winternl.h"
  32 #include "wine/debug.h"
  33 #include "wine/exception.h"
  34 #include "wine/list.h"
  35 #include "kernel_private.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(resource);
  38
  39 /* we don't want to include winuser.h just for this */
  40 #define IS_INTRESOURCE(x)   (((ULONG_PTR)(x) >> 16) == 0)
  41
  42 /* retrieve the resource name to pass to the ntdll functions */
  43 static NTSTATUS get_res_nameA( LPCSTR name, UNICODE_STRING *str )
  44 {
  45     if (IS_INTRESOURCE(name))
  46     {
  47         str->Buffer = ULongToPtr(LOWORD(name));
  48         return STATUS_SUCCESS;
  49     }
  50     if (name[0] == '#')
  51     {
  52         ULONG value;
  53         if (RtlCharToInteger( name + 1, 10, &value ) != STATUS_SUCCESS || HIWORD(value))
  54             return STATUS_INVALID_PARAMETER;
  55         str->Buffer = ULongToPtr(value);
  56         return STATUS_SUCCESS;
  57     }
  58     RtlCreateUnicodeStringFromAsciiz( str, name );
  59     return STATUS_SUCCESS;
  60 }
  61
  62 static NTSTATUS get_res_nameW( LPCWSTR name, UNICODE_STRING *str )
  63 {
  64     if (IS_INTRESOURCE(name))
  65     {
  66         str->Buffer = ULongToPtr( LOWORD(name) );
  67         return STATUS_SUCCESS;
  68     }
  69     if (name[0] == '#')
  70     {
  71         ULONG value;
  72         RtlInitUnicodeString( str, name + 1 );
  73         if (RtlUnicodeStringToInteger( str, 10, &value ) != STATUS_SUCCESS || HIWORD(value))
  74             return STATUS_INVALID_PARAMETER;
  75         str->Buffer = ULongToPtr(value);
  76         return STATUS_SUCCESS;
  77     }
  78     RtlCreateUnicodeString( str, name );
  79     RtlUpcaseUnicodeString( str, str, FALSE );
  80     return STATUS_SUCCESS;
  81 }
  82
  83 /**********************************************************************
  84  *          FindResourceExA  (KERNEL32.@)
  85  */
  86 HRSRC WINAPI FindResourceExA( HMODULE module, LPCSTR type, LPCSTR name, WORD lang )
  87 {
  88     NTSTATUS status;
  89     UNICODE_STRING nameW, typeW;
  90     HRSRC ret = NULL;
  91
  92     TRACE( "%p %s %s %04x\n", module, debugstr_a(type), debugstr_a(name), lang );
  93
  94     if (!module) module = GetModuleHandleW(0);
  95     nameW.Buffer = NULL;
  96     typeW.Buffer = NULL;
  97
  98     __TRY
  99     {
 100         if (!(status = get_res_nameA( name, &nameW )) && !(status = get_res_nameA( type, &typeW )))
 101             ret = FindResourceExW( module, typeW.Buffer, nameW.Buffer, lang );
 102         else
 103             SetLastError( RtlNtStatusToDosError(status) );
 104     }
 105     __EXCEPT_PAGE_FAULT
 106     {
 107         SetLastError( ERROR_INVALID_PARAMETER );
 108     }
 109     __ENDTRY
 110
 111     if (!IS_INTRESOURCE(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
 112     if (!IS_INTRESOURCE(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
 113     return ret;
 114 }
 115
 116
 117 /**********************************************************************
 118  *          FindResourceA    (KERNEL32.@)
 119  */
 120 HRSRC WINAPI FindResourceA( HMODULE hModule, LPCSTR name, LPCSTR type )
 121 {
 122     return FindResourceExA( hModule, type, name, MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ) );
 123 }
 124
 125
 126 /**********************************************************************
 127  *      EnumResourceTypesA      (KERNEL32.@)
 128  */
 129 BOOL WINAPI EnumResourceTypesA( HMODULE hmod, ENUMRESTYPEPROCA lpfun, LONG_PTR lparam )
 130 {
 131     return EnumResourceTypesExA( hmod, lpfun, lparam, 0, 0 );
 132 }
 133
 134
 135 /**********************************************************************
 136  *      EnumResourceTypesW      (KERNEL32.@)
 137  */
 138 BOOL WINAPI EnumResourceTypesW( HMODULE hmod, ENUMRESTYPEPROCW lpfun, LONG_PTR lparam )
 139 {
 140     return EnumResourceTypesExW( hmod, lpfun, lparam, 0, 0 );
 141 }
 142
 143
 144 /**********************************************************************
 145  *      EnumResourceNamesA      (KERNEL32.@)
 146  */
 147 BOOL WINAPI EnumResourceNamesA( HMODULE hmod, LPCSTR type, ENUMRESNAMEPROCA lpfun, LONG_PTR lparam )
 148 {
 149     return EnumResourceNamesExA( hmod, type, lpfun, lparam, 0, 0 );
 150 }
 151
 152
 153 /**********************************************************************
 154  *      EnumResourceLanguagesA  (KERNEL32.@)
 155  */
 156 BOOL WINAPI EnumResourceLanguagesA( HMODULE hmod, LPCSTR type, LPCSTR name,
 157                                     ENUMRESLANGPROCA lpfun, LONG_PTR lparam )
 158 {
 159     return EnumResourceLanguagesExA( hmod, type, name, lpfun, lparam, 0, 0 );
 160 }
 161
 162
 163 /**********************************************************************
 164  *      EnumResourceLanguagesW  (KERNEL32.@)
 165  */
 166 BOOL WINAPI EnumResourceLanguagesW( HMODULE hmod, LPCWSTR type, LPCWSTR name,
 167                                     ENUMRESLANGPROCW lpfun, LONG_PTR lparam )
 168 {
 169     return EnumResourceLanguagesExW( hmod, type, name, lpfun, lparam, 0, 0 );
 170 }
 171
 172
 173 /*
 174  *  Data structure for updating resources.
 175  *  Type/Name/Language is a keyset for accessing resource data.
 176  *
 177  *  QUEUEDUPDATES (root) ->
 178  *    list of struct resource_dir_entry    (Type) ->
 179  *      list of struct resource_dir_entry  (Name)   ->
 180  *         list of struct resource_data    Language + Data
 181  */
 182
 183 typedef struct
 184 {
 185     void *unknown[6];
 186     LPWSTR pFileName;
 187     BOOL bDeleteExistingResources;
 188     struct list root;
 189 } QUEUEDUPDATES;
 190
 191 /* this structure is shared for types and names */
 192 struct resource_dir_entry {
 193     struct list entry;
 194     LPWSTR id;
 195     struct list children;
 196 };
 197
 198 /* this structure is the leaf */
 199 struct resource_data {
 200     struct list entry;
 201     LANGID lang;
 202     DWORD codepage;
 203     DWORD cbData;
 204     void *lpData;
 205 };
 206
 207 static int resource_strcmp( LPCWSTR a, LPCWSTR b )
 208 {
 209     if ( a == b )
 210         return 0;
 211     if (!IS_INTRESOURCE( a ) && !IS_INTRESOURCE( b ) )
 212         return wcscmp( a, b );
 213     /* strings come before ids */
 214     if (!IS_INTRESOURCE( a ) && IS_INTRESOURCE( b ))
 215         return -1;
 216     if (!IS_INTRESOURCE( b ) && IS_INTRESOURCE( a ))
 217         return 1;
 218     return ( a < b ) ? -1 : 1;
 219 }
 220
 221 static struct resource_dir_entry *find_resource_dir_entry( struct list *dir, LPCWSTR id )
 222 {
 223     struct resource_dir_entry *ent;
 224
 225     /* match either IDs or strings */
 226     LIST_FOR_EACH_ENTRY( ent, dir, struct resource_dir_entry, entry )
 227         if (!resource_strcmp( id, ent->id ))
 228             return ent;
 229
 230     return NULL;
 231 }
 232
 233 static struct resource_data *find_resource_data( struct list *dir, LANGID lang )
 234 {
 235     struct resource_data *res_data;
 236
 237     /* match only languages here */
 238     LIST_FOR_EACH_ENTRY( res_data, dir, struct resource_data, entry )
 239         if ( lang == res_data->lang )
 240              return res_data;
 241
 242     return NULL;
 243 }
 244
 245 static void add_resource_dir_entry( struct list *dir, struct resource_dir_entry *resdir )
 246 {
 247     struct resource_dir_entry *ent;
 248
 249     LIST_FOR_EACH_ENTRY( ent, dir, struct resource_dir_entry, entry )
 250     {
 251         if (0>resource_strcmp( ent->id, resdir->id ))
 252             continue;
 253
 254         list_add_before( &ent->entry, &resdir->entry );
 255         return;
 256     }
 257     list_add_tail( dir, &resdir->entry );
 258 }
 259
 260 static void add_resource_data_entry( struct list *dir, struct resource_data *resdata )
 261 {
 262     struct resource_data *ent;
 263
 264     LIST_FOR_EACH_ENTRY( ent, dir, struct resource_data, entry )
 265     {
 266         if (ent->lang < resdata->lang)
 267             continue;
 268
 269         list_add_before( &ent->entry, &resdata->entry );
 270         return;
 271     }
 272     list_add_tail( dir, &resdata->entry );
 273 }
 274
 275 static LPWSTR res_strdupW( LPCWSTR str )
 276 {
 277     LPWSTR ret;
 278     UINT len;
 279
 280     if (IS_INTRESOURCE(str))
 281         return (LPWSTR) (UINT_PTR) LOWORD(str);
 282     len = (lstrlenW( str ) + 1) * sizeof (WCHAR);
 283     ret = HeapAlloc( GetProcessHeap(), 0, len );
 284     memcpy( ret, str, len );
 285     return ret;
 286 }
 287
 288 static void res_free_str( LPWSTR str )
 289 {
 290     if (!IS_INTRESOURCE(str))
 291         HeapFree( GetProcessHeap(), 0, str );
 292 }
 293
 294 static BOOL update_add_resource( QUEUEDUPDATES *updates, LPCWSTR Type, LPCWSTR Name,
 295                                  LANGID Lang, struct resource_data *resdata,
 296                                  BOOL overwrite_existing )
 297 {
 298     struct resource_dir_entry *restype, *resname;
 299     struct resource_data *existing;
 300
 301     TRACE("%p %s %s %p %d\n", updates,
 302           debugstr_w(Type), debugstr_w(Name), resdata, overwrite_existing );
 303
 304     restype = find_resource_dir_entry( &updates->root, Type );
 305     if (!restype)
 306     {
 307         restype = HeapAlloc( GetProcessHeap(), 0, sizeof *restype );
 308         restype->id = res_strdupW( Type );
 309         list_init( &restype->children );
 310         add_resource_dir_entry( &updates->root, restype );
 311     }
 312
 313     resname = find_resource_dir_entry( &restype->children, Name );
 314     if (!resname)
 315     {
 316         resname = HeapAlloc( GetProcessHeap(), 0, sizeof *resname );
 317         resname->id = res_strdupW( Name );
 318         list_init( &resname->children );
 319         add_resource_dir_entry( &restype->children, resname );
 320     }
 321
 322     /*
 323      * If there's an existing resource entry with matching (Type,Name,Language)
 324      *  it needs to be removed before adding the new data.
 325      */
 326     existing = find_resource_data( &resname->children, Lang );
 327     if (existing)
 328     {
 329         if (!overwrite_existing)
 330             return FALSE;
 331         list_remove( &existing->entry );
 332         HeapFree( GetProcessHeap(), 0, existing );
 333     }
 334
 335     if (resdata)
 336         add_resource_data_entry( &resname->children, resdata );
 337
 338     return TRUE;
 339 }
 340
 341 static struct resource_data *allocate_resource_data( WORD Language, DWORD codepage,
 342                                                      LPVOID lpData, DWORD cbData, BOOL copy_data )
 343 {
 344     struct resource_data *resdata;
 345
 346     if (!lpData || !cbData)
 347         return NULL;
 348
 349     resdata = HeapAlloc( GetProcessHeap(), 0, sizeof *resdata + (copy_data ? cbData : 0) );
 350     if (resdata)
 351     {
 352         resdata->lang = Language;
 353         resdata->codepage = codepage;
 354         resdata->cbData = cbData;
 355         if (copy_data)
 356         {
 357             resdata->lpData = &resdata[1];
 358             memcpy( resdata->lpData, lpData, cbData );
 359         }
 360         else
 361             resdata->lpData = lpData;
 362     }
 363
 364     return resdata;
 365 }
 366
 367 static void free_resource_directory( struct list *head, int level )
 368 {
 369     struct list *ptr = NULL;
 370
 371     while ((ptr = list_head( head )))
 372     {
 373         list_remove( ptr );
 374         if (level)
 375         {
 376             struct resource_dir_entry *ent;
 377
 378             ent = LIST_ENTRY( ptr, struct resource_dir_entry, entry );
 379             res_free_str( ent->id );
 380             free_resource_directory( &ent->children, level - 1 );
 381             HeapFree(GetProcessHeap(), 0, ent);
 382         }
 383         else
 384         {
 385             struct resource_data *data;
 386
 387             data = LIST_ENTRY( ptr, struct resource_data, entry );
 388             HeapFree( GetProcessHeap(), 0, data );
 389         }
 390     }
 391 }
 392
 393 static IMAGE_NT_HEADERS *get_nt_header( void *base, DWORD mapping_size )
 394 {
 395     IMAGE_NT_HEADERS *nt;
 396     IMAGE_DOS_HEADER *dos;
 397
 398     if (mapping_size<sizeof (*dos))
 399         return NULL;
 400
 401     dos = base;
 402     if (dos->e_magic != IMAGE_DOS_SIGNATURE)
 403         return NULL;
 404
 405     if ((dos->e_lfanew + sizeof (*nt)) > mapping_size)
 406         return NULL;
 407
 408     nt = (void*) ((BYTE*)base + dos->e_lfanew);
 409
 410     if (nt->Signature != IMAGE_NT_SIGNATURE)
 411         return NULL;
 412
 413     return nt;
 414 }
 415
 416 static IMAGE_SECTION_HEADER *get_section_header( void *base, DWORD mapping_size, DWORD *num_sections )
 417 {
 418     IMAGE_NT_HEADERS *nt;
 419     DWORD section_ofs;
 420
 421     nt = get_nt_header( base, mapping_size );
 422     if (!nt)
 423         return NULL;
 424
 425     /* check that we don't go over the end of the file accessing the sections */
 426     section_ofs = FIELD_OFFSET(IMAGE_NT_HEADERS, OptionalHeader) + nt->FileHeader.SizeOfOptionalHeader;
 427     if ((nt->FileHeader.NumberOfSections * sizeof (IMAGE_SECTION_HEADER) + section_ofs) > mapping_size)
 428         return NULL;
 429
 430     if (num_sections)
 431         *num_sections = nt->FileHeader.NumberOfSections;
 432
 433     /* from here we have a valid PE exe to update */
 434     return (void*) ((BYTE*)nt + section_ofs);
 435 }
 436
 437 static BOOL check_pe_exe( HANDLE file, QUEUEDUPDATES *updates )
 438 {
 439     const IMAGE_NT_HEADERS32 *nt;
 440     const IMAGE_NT_HEADERS64 *nt64;
 441     const IMAGE_SECTION_HEADER *sec;
 442     const IMAGE_DATA_DIRECTORY *dd;
 443     BOOL ret = FALSE;
 444     HANDLE mapping;
 445     DWORD mapping_size, num_sections = 0;
 446     void *base = NULL;
 447
 448     mapping_size = GetFileSize( file, NULL );
 449
 450     mapping = CreateFileMappingW( file, NULL, PAGE_READONLY, 0, 0, NULL );
 451     if (!mapping)
 452         goto done;
 453
 454     base = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, mapping_size );
 455     if (!base)
 456         goto done;
 457
 458     nt = (IMAGE_NT_HEADERS32 *)get_nt_header( base, mapping_size );
 459     if (!nt)
 460         goto done;
 461
 462     nt64 = (IMAGE_NT_HEADERS64*)nt;
 463     dd = &nt->OptionalHeader.DataDirectory[0];
 464     if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
 465         dd = &nt64->OptionalHeader.DataDirectory[0];
 466
 467     TRACE("resources: %08x %08x\n",
 468           dd[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress,
 469           dd[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size);
 470
 471     sec = get_section_header( base, mapping_size, &num_sections );
 472     if (!sec)
 473         goto done;
 474
 475     ret = TRUE;
 476
 477 done:
 478     if (base)
 479         UnmapViewOfFile( base );
 480     if (mapping)
 481         CloseHandle( mapping );
 482
 483     return ret;
 484 }
 485
 486 struct resource_size_info {
 487     DWORD types_ofs;
 488     DWORD names_ofs;
 489     DWORD langs_ofs;
 490     DWORD data_entry_ofs;
 491     DWORD strings_ofs;
 492     DWORD data_ofs;
 493     DWORD total_size;
 494 };
 495
 496 struct mapping_info {
 497     HANDLE file;
 498     void *base;
 499     DWORD size;
 500     BOOL read_write;
 501 };
 502
 503 static const IMAGE_SECTION_HEADER *section_from_rva( void *base, DWORD mapping_size, DWORD rva )
 504 {
 505     const IMAGE_SECTION_HEADER *sec;
 506     DWORD num_sections = 0;
 507     int i;
 508
 509     sec = get_section_header( base, mapping_size, &num_sections );
 510     if (!sec)
 511         return NULL;
 512
 513     for (i=num_sections-1; i>=0; i--)
 514     {
 515         if (sec[i].VirtualAddress <= rva &&
 516             rva <= (DWORD)sec[i].VirtualAddress + sec[i].SizeOfRawData)
 517         {
 518             return &sec[i];
 519         }
 520     }
 521
 522     return NULL;
 523 }
 524
 525 static void *address_from_rva( void *base, DWORD mapping_size, DWORD rva, DWORD len )
 526 {
 527     const IMAGE_SECTION_HEADER *sec;
 528
 529     sec = section_from_rva( base, mapping_size, rva );
 530     if (!sec)
 531         return NULL;
 532
 533     if (rva + len <= (DWORD)sec->VirtualAddress + sec->SizeOfRawData)
 534         return (void*)((LPBYTE) base + (sec->PointerToRawData + rva - sec->VirtualAddress));
 535
 536     return NULL;
 537 }
 538
 539 static LPWSTR resource_dup_string( const IMAGE_RESOURCE_DIRECTORY *root, const IMAGE_RESOURCE_DIRECTORY_ENTRY *entry )
 540 {
 541     const IMAGE_RESOURCE_DIR_STRING_U* string;
 542     LPWSTR s;
 543
 544     if (!entry->u.s.NameIsString)
 545         return UIntToPtr(entry->u.Id);
 546
 547     string = (const IMAGE_RESOURCE_DIR_STRING_U*) (((const char *)root) + entry->u.s.NameOffset);
 548     s = HeapAlloc(GetProcessHeap(), 0, (string->Length + 1)*sizeof (WCHAR) );
 549     memcpy( s, string->NameString, (string->Length + 1)*sizeof (WCHAR) );
 550     s[string->Length] = 0;
 551
 552     return s;
 553 }
 554
 555 /* this function is based on the code in winedump's pe.c */
 556 static BOOL enumerate_mapped_resources( QUEUEDUPDATES *updates,
 557                              void *base, DWORD mapping_size,
 558                              const IMAGE_RESOURCE_DIRECTORY *root )
 559 {
 560     const IMAGE_RESOURCE_DIRECTORY *namedir, *langdir;
 561     const IMAGE_RESOURCE_DIRECTORY_ENTRY *e1, *e2, *e3;
 562     const IMAGE_RESOURCE_DATA_ENTRY *data;
 563     DWORD i, j, k;
 564
 565     TRACE("version (%d.%d) %d named %d id entries\n",
 566           root->MajorVersion, root->MinorVersion, root->NumberOfNamedEntries, root->NumberOfIdEntries);
 567
 568     for (i = 0; i< root->NumberOfNamedEntries + root->NumberOfIdEntries; i++)
 569     {
 570         LPWSTR Type;
 571
 572         e1 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(root + 1) + i;
 573
 574         Type = resource_dup_string( root, e1 );
 575
 576         namedir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e1->u2.s2.OffsetToDirectory);
 577         for (j = 0; j < namedir->NumberOfNamedEntries + namedir->NumberOfIdEntries; j++)
 578         {
 579             LPWSTR Name;
 580
 581             e2 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(namedir + 1) + j;
 582
 583             Name = resource_dup_string( root, e2 );
 584
 585             langdir = (const IMAGE_RESOURCE_DIRECTORY *)((const char *)root + e2->u2.s2.OffsetToDirectory);
 586             for (k = 0; k < langdir->NumberOfNamedEntries + langdir->NumberOfIdEntries; k++)
 587             {
 588                 LANGID Lang;
 589                 void *p;
 590                 struct resource_data *resdata;
 591
 592                 e3 = (const IMAGE_RESOURCE_DIRECTORY_ENTRY*)(langdir + 1) + k;
 593
 594                 Lang = e3->u.Id;
 595
 596                 data = (const IMAGE_RESOURCE_DATA_ENTRY *)((const char *)root + e3->u2.OffsetToData);
 597
 598                 p = address_from_rva( base, mapping_size, data->OffsetToData, data->Size );
 599
 600                 resdata = allocate_resource_data( Lang, data->CodePage, p, data->Size, FALSE );
 601                 if (resdata)
 602                 {
 603                     if (!update_add_resource( updates, Type, Name, Lang, resdata, FALSE ))
 604                         HeapFree( GetProcessHeap(), 0, resdata );
 605                 }
 606             }
 607             res_free_str( Name );
 608         }
 609         res_free_str( Type );
 610     }
 611
 612     return TRUE;
 613 }
 614
 615 static BOOL read_mapped_resources( QUEUEDUPDATES *updates, void *base, DWORD mapping_size )
 616 {
 617     const IMAGE_RESOURCE_DIRECTORY *root;
 618     const IMAGE_NT_HEADERS *nt;
 619     const IMAGE_SECTION_HEADER *sec;
 620     DWORD num_sections = 0, i;
 621
 622     nt = get_nt_header( base, mapping_size );
 623     if (!nt)
 624         return FALSE;
 625
 626     sec = get_section_header( base, mapping_size, &num_sections );
 627     if (!sec)
 628         return FALSE;
 629
 630     for (i=0; i<num_sections; i++)
 631         if (!memcmp(sec[i].Name, ".rsrc", 6))
 632             break;
 633
 634     if (i == num_sections)
 635         return TRUE;
 636
 637     /* check the resource data is inside the mapping */
 638     if (sec[i].PointerToRawData > mapping_size ||
 639         (sec[i].PointerToRawData + sec[i].SizeOfRawData) > mapping_size)
 640         return TRUE;
 641
 642     TRACE("found .rsrc at %08x, size %08x\n", sec[i].PointerToRawData, sec[i].SizeOfRawData);
 643
 644     if (!sec[i].PointerToRawData || sec[i].SizeOfRawData < sizeof(IMAGE_RESOURCE_DIRECTORY))
 645         return TRUE;
 646
 647     root = (void*) ((BYTE*)base + sec[i].PointerToRawData);
 648     enumerate_mapped_resources( updates, base, mapping_size, root );
 649
 650     return TRUE;
 651 }
 652
 653 static BOOL map_file_into_memory( struct mapping_info *mi )
 654 {
 655     DWORD page_attr, perm;
 656     HANDLE mapping;
 657
 658     if (mi->read_write)
 659     {
 660         page_attr = PAGE_READWRITE;
 661         perm = FILE_MAP_WRITE | FILE_MAP_READ;
 662     }
 663     else
 664     {
 665         page_attr = PAGE_READONLY;
 666         perm = FILE_MAP_READ;
 667     }
 668
 669     mapping = CreateFileMappingW( mi->file, NULL, page_attr, 0, 0, NULL );
 670     if (!mapping) return FALSE;
 671
 672     mi->base = MapViewOfFile( mapping, perm, 0, 0, mi->size );
 673     CloseHandle( mapping );
 674
 675     return mi->base != NULL;
 676 }
 677
 678 static BOOL unmap_file_from_memory( struct mapping_info *mi )
 679 {
 680     if (mi->base)
 681         UnmapViewOfFile( mi->base );
 682     mi->base = NULL;
 683     return TRUE;
 684 }
 685
 686 static void destroy_mapping( struct mapping_info *mi )
 687 {
 688     if (!mi)
 689         return;
 690     unmap_file_from_memory( mi );
 691     if (mi->file)
 692         CloseHandle( mi->file );
 693     HeapFree( GetProcessHeap(), 0, mi );
 694 }
 695
 696 static struct mapping_info *create_mapping( LPCWSTR name, BOOL rw )
 697 {
 698     struct mapping_info *mi;
 699
 700     mi = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *mi );
 701     if (!mi)
 702         return NULL;
 703
 704     mi->read_write = rw;
 705
 706     mi->file = CreateFileW( name, GENERIC_READ | (rw ? GENERIC_WRITE : 0),
 707                             0, NULL, OPEN_EXISTING, 0, 0 );
 708
 709     if (mi->file != INVALID_HANDLE_VALUE)
 710     {
 711         mi->size = GetFileSize( mi->file, NULL );
 712
 713         if (map_file_into_memory( mi ))
 714             return mi;
 715     }
 716     destroy_mapping( mi );
 717     return NULL;
 718 }
 719
 720 static BOOL resize_mapping( struct mapping_info *mi, DWORD new_size )
 721 {
 722     if (!unmap_file_from_memory( mi ))
 723         return FALSE;
 724
 725     /* change the file size */
 726     SetFilePointer( mi->file, new_size, NULL, FILE_BEGIN );
 727     if (!SetEndOfFile( mi->file ))
 728     {
 729         ERR("failed to set file size to %08x\n", new_size );
 730         return FALSE;
 731     }
 732
 733     mi->size = new_size;
 734
 735     return map_file_into_memory( mi );
 736 }
 737
 738 static void get_resource_sizes( QUEUEDUPDATES *updates, struct resource_size_info *si )
 739 {
 740     struct resource_dir_entry *types, *names;
 741     struct resource_data *data;
 742     DWORD num_types = 0, num_names = 0, num_langs = 0, strings_size = 0, data_size = 0;
 743
 744     memset( si, 0, sizeof *si );
 745
 746     LIST_FOR_EACH_ENTRY( types, &updates->root, struct resource_dir_entry, entry )
 747     {
 748         num_types++;
 749         if (!IS_INTRESOURCE( types->id ))
 750             strings_size += sizeof (WORD) + lstrlenW( types->id )*sizeof (WCHAR);
 751
 752         LIST_FOR_EACH_ENTRY( names, &types->children, struct resource_dir_entry, entry )
 753         {
 754             num_names++;
 755
 756             if (!IS_INTRESOURCE( names->id ))
 757                 strings_size += sizeof (WORD) + lstrlenW( names->id )*sizeof (WCHAR);
 758
 759             LIST_FOR_EACH_ENTRY( data, &names->children, struct resource_data, entry )
 760             {
 761                 num_langs++;
 762                 data_size += (data->cbData + 3) & ~3;
 763             }
 764         }
 765     }
 766
 767     /* names are at the end of the types */
 768     si->names_ofs = sizeof (IMAGE_RESOURCE_DIRECTORY) +
 769             num_types * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 770
 771     /* language directories are at the end of the names */
 772     si->langs_ofs = si->names_ofs +
 773             num_types * sizeof (IMAGE_RESOURCE_DIRECTORY) +
 774             num_names * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 775
 776     si->data_entry_ofs = si->langs_ofs +
 777             num_names * sizeof (IMAGE_RESOURCE_DIRECTORY) +
 778             num_langs * sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 779
 780     si->strings_ofs = si->data_entry_ofs +
 781             num_langs * sizeof (IMAGE_RESOURCE_DATA_ENTRY);
 782
 783     si->data_ofs = si->strings_ofs + ((strings_size + 3) & ~3);
 784
 785     si->total_size = si->data_ofs + data_size;
 786
 787     TRACE("names %08x langs %08x data entries %08x strings %08x data %08x total %08x\n",
 788           si->names_ofs, si->langs_ofs, si->data_entry_ofs,
 789           si->strings_ofs, si->data_ofs, si->total_size);
 790 }
 791
 792 static void res_write_padding( BYTE *res_base, DWORD size )
 793 {
 794     static const BYTE pad[] = {
 795         'P','A','D','D','I','N','G','X','X','P','A','D','D','I','N','G' };
 796     DWORD i;
 797
 798     for ( i = 0; i < size / sizeof pad; i++ )
 799         memcpy( &res_base[i*sizeof pad], pad, sizeof pad );
 800     memcpy( &res_base[i*sizeof pad], pad, size%sizeof pad );
 801 }
 802
 803 static BOOL write_resources( QUEUEDUPDATES *updates, LPBYTE base, struct resource_size_info *si, DWORD rva )
 804 {
 805     struct resource_dir_entry *types, *names;
 806     struct resource_data *data;
 807     IMAGE_RESOURCE_DIRECTORY *root;
 808
 809     TRACE("%p %p %p %08x\n", updates, base, si, rva );
 810
 811     memset( base, 0, si->total_size );
 812
 813     /* the root entry always exists */
 814     root = (IMAGE_RESOURCE_DIRECTORY*) base;
 815     memset( root, 0, sizeof *root );
 816     root->MajorVersion = 4;
 817     si->types_ofs = sizeof *root;
 818     LIST_FOR_EACH_ENTRY( types, &updates->root, struct resource_dir_entry, entry )
 819     {
 820         IMAGE_RESOURCE_DIRECTORY_ENTRY *e1;
 821         IMAGE_RESOURCE_DIRECTORY *namedir;
 822
 823         e1 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->types_ofs];
 824         memset( e1, 0, sizeof *e1 );
 825         if (!IS_INTRESOURCE( types->id ))
 826         {
 827             WCHAR *strings;
 828             DWORD len;
 829
 830             root->NumberOfNamedEntries++;
 831             e1->u.s.NameIsString = 1;
 832             e1->u.s.NameOffset = si->strings_ofs;
 833
 834             strings = (WCHAR*) &base[si->strings_ofs];
 835             len = lstrlenW( types->id );
 836             strings[0] = len;
 837             memcpy( &strings[1], types->id, len * sizeof (WCHAR) );
 838             si->strings_ofs += (len + 1) * sizeof (WCHAR);
 839         }
 840         else
 841         {
 842             root->NumberOfIdEntries++;
 843             e1->u.Id = LOWORD( types->id );
 844         }
 845         e1->u2.s2.OffsetToDirectory = si->names_ofs;
 846         e1->u2.s2.DataIsDirectory = TRUE;
 847         si->types_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 848
 849         namedir = (IMAGE_RESOURCE_DIRECTORY*) &base[si->names_ofs];
 850         memset( namedir, 0, sizeof *namedir );
 851         namedir->MajorVersion = 4;
 852         si->names_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY);
 853
 854         LIST_FOR_EACH_ENTRY( names, &types->children, struct resource_dir_entry, entry )
 855         {
 856             IMAGE_RESOURCE_DIRECTORY_ENTRY *e2;
 857             IMAGE_RESOURCE_DIRECTORY *langdir;
 858
 859             e2 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->names_ofs];
 860             memset( e2, 0, sizeof *e2 );
 861             if (!IS_INTRESOURCE( names->id ))
 862             {
 863                 WCHAR *strings;
 864                 DWORD len;
 865
 866                 namedir->NumberOfNamedEntries++;
 867                 e2->u.s.NameIsString = 1;
 868                 e2->u.s.NameOffset = si->strings_ofs;
 869
 870                 strings = (WCHAR*) &base[si->strings_ofs];
 871                 len = lstrlenW( names->id );
 872                 strings[0] = len;
 873                 memcpy( &strings[1], names->id, len * sizeof (WCHAR) );
 874                 si->strings_ofs += (len + 1) * sizeof (WCHAR);
 875             }
 876             else
 877             {
 878                 namedir->NumberOfIdEntries++;
 879                 e2->u.Id = LOWORD( names->id );
 880             }
 881             e2->u2.s2.OffsetToDirectory = si->langs_ofs;
 882             e2->u2.s2.DataIsDirectory = TRUE;
 883             si->names_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 884
 885             langdir = (IMAGE_RESOURCE_DIRECTORY*) &base[si->langs_ofs];
 886             memset( langdir, 0, sizeof *langdir );
 887             langdir->MajorVersion = 4;
 888             si->langs_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY);
 889
 890             LIST_FOR_EACH_ENTRY( data, &names->children, struct resource_data, entry )
 891             {
 892                 IMAGE_RESOURCE_DIRECTORY_ENTRY *e3;
 893                 IMAGE_RESOURCE_DATA_ENTRY *de;
 894                 int pad_size;
 895
 896                 e3 = (IMAGE_RESOURCE_DIRECTORY_ENTRY*) &base[si->langs_ofs];
 897                 memset( e3, 0, sizeof *e3 );
 898                 langdir->NumberOfIdEntries++;
 899                 e3->u.Id = LOWORD( data->lang );
 900                 e3->u2.OffsetToData = si->data_entry_ofs;
 901
 902                 si->langs_ofs += sizeof (IMAGE_RESOURCE_DIRECTORY_ENTRY);
 903
 904                 /* write out all the data entries */
 905                 de = (IMAGE_RESOURCE_DATA_ENTRY*) &base[si->data_entry_ofs];
 906                 memset( de, 0, sizeof *de );
 907                 de->OffsetToData = si->data_ofs + rva;
 908                 de->Size = data->cbData;
 909                 de->CodePage = data->codepage;
 910                 si->data_entry_ofs += sizeof (IMAGE_RESOURCE_DATA_ENTRY);
 911
 912                 /* write out the resource data */
 913                 memcpy( &base[si->data_ofs], data->lpData, data->cbData );
 914                 si->data_ofs += data->cbData;
 915
 916                 pad_size = (-si->data_ofs)&3;
 917                 res_write_padding( &base[si->data_ofs], pad_size );
 918                 si->data_ofs += pad_size;
 919             }
 920         }
 921     }
 922
 923     return TRUE;
 924 }
 925
 926 /*
 927  *  FIXME:
 928  *  Assumes that the resources are in .rsrc
 929  *   and .rsrc is the last section in the file.
 930  *  Not sure whether updating resources will other cases on Windows.
 931  *  If the resources lie in a section containing other data,
 932  *   resizing that section could possibly cause trouble.
 933  *  If the section with the resources isn't last, the remaining
 934  *   sections need to be moved down in the file, and the section header
 935  *   would need to be adjusted.
 936  *  If we needed to add a section, what would we name it?
 937  *  If we needed to add a section and there wasn't space in the file
 938  *   header, how would that work?
 939  *  Seems that at least some of these cases can't be handled properly.
 940  */
 941 static IMAGE_SECTION_HEADER *get_resource_section( void *base, DWORD mapping_size )
 942 {
 943     IMAGE_SECTION_HEADER *sec;
 944     IMAGE_NT_HEADERS *nt;
 945     DWORD i, num_sections = 0;
 946
 947     nt = get_nt_header( base, mapping_size );
 948     if (!nt)
 949         return NULL;
 950
 951     sec = get_section_header( base, mapping_size, &num_sections );
 952     if (!sec)
 953         return NULL;
 954
 955     /* find the resources section */
 956     for (i=0; i<num_sections; i++)
 957         if (!memcmp(sec[i].Name, ".rsrc", 6))
 958             break;
 959
 960     if (i == num_sections)
 961         return NULL;
 962
 963     return &sec[i];
 964 }
 965
 966 static DWORD get_init_data_size( void *base, DWORD mapping_size )
 967 {
 968     DWORD i, sz = 0, num_sections = 0;
 969     IMAGE_SECTION_HEADER *s;
 970
 971     s = get_section_header( base, mapping_size, &num_sections );
 972
 973     for (i=0; i<num_sections; i++)
 974         if (s[i].Characteristics & IMAGE_SCN_CNT_INITIALIZED_DATA)
 975             sz += s[i].SizeOfRawData;
 976
 977     TRACE("size = %08x\n", sz);
 978
 979     return sz;
 980 }
 981
 982 static BOOL write_raw_resources( QUEUEDUPDATES *updates )
 983 {
 984     WCHAR tempdir[MAX_PATH], tempfile[MAX_PATH];
 985     DWORD i, section_size;
 986     BOOL ret = FALSE;
 987     IMAGE_SECTION_HEADER *sec;
 988     IMAGE_NT_HEADERS32 *nt;
 989     IMAGE_NT_HEADERS64 *nt64;
 990     struct resource_size_info res_size;
 991     BYTE *res_base;
 992     struct mapping_info *read_map = NULL, *write_map = NULL;
 993     DWORD PeSectionAlignment, PeFileAlignment, PeSizeOfImage;
 994
 995     /* copy the exe to a temp file then update the temp file... */
 996     tempdir[0] = 0;
 997     if (!GetTempPathW( MAX_PATH, tempdir ))
 998         return ret;
 999
1000     if (!GetTempFileNameW( tempdir, L"resu", 0, tempfile ))
1001         return ret;
1002
1003     if (!CopyFileW( updates->pFileName, tempfile, FALSE ))
1004         goto done;
1005
1006     TRACE("tempfile %s\n", debugstr_w(tempfile));
1007
1008     if (!updates->bDeleteExistingResources)
1009     {
1010         read_map = create_mapping( updates->pFileName, FALSE );
1011         if (!read_map)
1012             goto done;
1013
1014         ret = read_mapped_resources( updates, read_map->base, read_map->size );
1015         if (!ret)
1016         {
1017             ERR("failed to read existing resources\n");
1018             goto done;
1019         }
1020     }
1021
1022     write_map = create_mapping( tempfile, TRUE );
1023     if (!write_map)
1024         goto done;
1025
1026     nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, write_map->size );
1027     if (!nt)
1028         goto done;
1029
1030     nt64 = (IMAGE_NT_HEADERS64*)nt;
1031     if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) {
1032         PeSectionAlignment = nt64->OptionalHeader.SectionAlignment;
1033         PeFileAlignment = nt64->OptionalHeader.FileAlignment;
1034         PeSizeOfImage = nt64->OptionalHeader.SizeOfImage;
1035     } else {
1036         PeSectionAlignment = nt->OptionalHeader.SectionAlignment;
1037         PeFileAlignment = nt->OptionalHeader.FileAlignment;
1038         PeSizeOfImage = nt->OptionalHeader.SizeOfImage;
1039     }
1040
1041     if ((LONG)PeSectionAlignment <= 0)
1042     {
1043         ERR("invalid section alignment %08x\n", PeSectionAlignment);
1044         goto done;
1045     }
1046
1047     if ((LONG)PeFileAlignment <= 0)
1048     {
1049         ERR("invalid file alignment %08x\n", PeFileAlignment);
1050         goto done;
1051     }
1052
1053     sec = get_resource_section( write_map->base, write_map->size );
1054     if (!sec) /* no section, add one */
1055     {
1056         DWORD num_sections;
1057
1058         sec = get_section_header( write_map->base, write_map->size, &num_sections );
1059         if (!sec)
1060             goto done;
1061
1062         sec += num_sections;
1063         nt->FileHeader.NumberOfSections++;
1064
1065         memset( sec, 0, sizeof *sec );
1066         memcpy( sec->Name, ".rsrc", 5 );
1067         sec->Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ;
1068         sec->VirtualAddress = PeSizeOfImage;
1069     }
1070
1071     if (!sec->PointerToRawData)  /* empty section */
1072     {
1073         sec->PointerToRawData = write_map->size + (-write_map->size) % PeFileAlignment;
1074         sec->SizeOfRawData = 0;
1075     }
1076
1077     TRACE("before .rsrc at %08x, size %08x\n", sec->PointerToRawData, sec->SizeOfRawData);
1078
1079     get_resource_sizes( updates, &res_size );
1080
1081     /* round up the section size */
1082     section_size = res_size.total_size;
1083     section_size += (-section_size) % PeFileAlignment;
1084
1085     TRACE("requires %08x (%08x) bytes\n", res_size.total_size, section_size );
1086
1087     /* check if the file size needs to be changed */
1088     if (section_size != sec->SizeOfRawData)
1089     {
1090         DWORD old_size = write_map->size;
1091         DWORD virtual_section_size = res_size.total_size + (-res_size.total_size) % PeSectionAlignment;
1092         int delta = section_size - (sec->SizeOfRawData + (-sec->SizeOfRawData) % PeFileAlignment);
1093         int rva_delta = virtual_section_size -
1094             (sec->Misc.VirtualSize + (-sec->Misc.VirtualSize) % PeSectionAlignment);
1095         /* when new section is added it could end past current mapping size */
1096         BOOL rsrc_is_last = sec->PointerToRawData + sec->SizeOfRawData >= old_size;
1097         /* align .rsrc size when possible */
1098         DWORD mapping_size = rsrc_is_last ? sec->PointerToRawData + section_size : old_size + delta;
1099
1100         /* postpone file truncation if there are some data to be moved down from file end */
1101         BOOL resize_after = mapping_size < old_size && !rsrc_is_last;
1102
1103         TRACE("file size %08x -> %08x\n", old_size, mapping_size);
1104
1105         if (!resize_after)
1106         {
1107             /* unmap the file before changing the file size */
1108             ret = resize_mapping( write_map, mapping_size );
1109
1110             /* get the pointers again - they might be different after remapping */
1111             nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, mapping_size );
1112             if (!nt)
1113             {
1114                 ERR("couldn't get NT header\n");
1115                 goto done;
1116             }
1117             nt64 = (IMAGE_NT_HEADERS64*)nt;
1118
1119             sec = get_resource_section( write_map->base, mapping_size );
1120             if (!sec)
1121                  goto done;
1122         }
1123
1124         if (!rsrc_is_last) /* not last section, relocate trailing sections */
1125         {
1126             IMAGE_SECTION_HEADER *s;
1127             DWORD tail_start = sec->PointerToRawData + sec->SizeOfRawData;
1128             DWORD i, num_sections = 0;
1129
1130             memmove( (char*)write_map->base + tail_start + delta, (char*)write_map->base + tail_start, old_size - tail_start );
1131
1132             s = get_section_header( write_map->base, mapping_size, &num_sections );
1133
1134             for (i=0; i<num_sections; i++)
1135             {
1136                 if (s[i].PointerToRawData > sec->PointerToRawData)
1137                 {
1138                     s[i].PointerToRawData += delta;
1139                     s[i].VirtualAddress += rva_delta;
1140                 }
1141             }
1142         }
1143
1144         if (resize_after)
1145         {
1146             ret = resize_mapping( write_map, mapping_size );
1147
1148             nt = (IMAGE_NT_HEADERS32*)get_nt_header( write_map->base, mapping_size );
1149             if (!nt)
1150             {
1151                 ERR("couldn't get NT header\n");
1152                 goto done;
1153             }
1154             nt64 = (IMAGE_NT_HEADERS64*)nt;
1155
1156             sec = get_resource_section( write_map->base, mapping_size );
1157             if (!sec)
1158                  goto done;
1159         }
1160
1161         /* adjust the PE header information */
1162         sec->SizeOfRawData = section_size;
1163         sec->Misc.VirtualSize = virtual_section_size;
1164         if (nt->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC) {
1165             nt64->OptionalHeader.SizeOfImage += rva_delta;
1166             nt64->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = sec->VirtualAddress;
1167             nt64->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size;
1168             nt64->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size );
1169
1170             for (i=0; i<nt64->OptionalHeader.NumberOfRvaAndSizes; i++)
1171                 if (nt64->OptionalHeader.DataDirectory[i].VirtualAddress > sec->VirtualAddress)
1172                     nt64->OptionalHeader.DataDirectory[i].VirtualAddress += rva_delta;
1173         } else {
1174             nt->OptionalHeader.SizeOfImage += rva_delta;
1175             nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress = sec->VirtualAddress;
1176             nt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size = res_size.total_size;
1177             nt->OptionalHeader.SizeOfInitializedData = get_init_data_size( write_map->base, mapping_size );
1178
1179             for (i=0; i<nt->OptionalHeader.NumberOfRvaAndSizes; i++)
1180                 if (nt->OptionalHeader.DataDirectory[i].VirtualAddress > sec->VirtualAddress)
1181                     nt->OptionalHeader.DataDirectory[i].VirtualAddress += rva_delta;
1182         }
1183     }
1184
1185     res_base = (LPBYTE) write_map->base + sec->PointerToRawData;
1186
1187     TRACE("base = %p offset = %08x\n", write_map->base, sec->PointerToRawData);
1188
1189     ret = write_resources( updates, res_base, &res_size, sec->VirtualAddress );
1190
1191     res_write_padding( res_base + res_size.total_size, section_size - res_size.total_size );
1192
1193     TRACE("after  .rsrc at %08x, size %08x\n", sec->PointerToRawData, sec->SizeOfRawData);
1194
1195 done:
1196     destroy_mapping( read_map );
1197     destroy_mapping( write_map );
1198
1199     if (ret)
1200         ret = CopyFileW( tempfile, updates->pFileName, FALSE );
1201
1202     DeleteFileW( tempfile );
1203
1204     return ret;
1205 }
1206
1207 /***********************************************************************
1208  *          BeginUpdateResourceW                 (KERNEL32.@)
1209  */
1210 HANDLE WINAPI BeginUpdateResourceW( LPCWSTR pFileName, BOOL bDeleteExistingResources )
1211 {
1212     QUEUEDUPDATES *updates = NULL;
1213     HANDLE hUpdate, file, ret = NULL;
1214
1215     TRACE("%s, %d\n", debugstr_w(pFileName), bDeleteExistingResources);
1216
1217     hUpdate = GlobalAlloc(GHND, sizeof(QUEUEDUPDATES));
1218     if (!hUpdate)
1219         return ret;
1220
1221     updates = GlobalLock(hUpdate);
1222     if (updates)
1223     {
1224         list_init( &updates->root );
1225         updates->bDeleteExistingResources = bDeleteExistingResources;
1226         updates->pFileName = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(pFileName)+1)*sizeof(WCHAR));
1227         if (updates->pFileName)
1228         {
1229             lstrcpyW(updates->pFileName, pFileName);
1230
1231             file = CreateFileW( pFileName, GENERIC_READ | GENERIC_WRITE,
1232                                 0, NULL, OPEN_EXISTING, 0, 0 );
1233
1234             /* if resources are deleted, only the file's presence is checked */
1235             if (file != INVALID_HANDLE_VALUE &&
1236                 (bDeleteExistingResources || check_pe_exe( file, updates )))
1237                 ret = hUpdate;
1238             else
1239                 HeapFree( GetProcessHeap(), 0, updates->pFileName );
1240
1241             CloseHandle( file );
1242         }
1243         GlobalUnlock(hUpdate);
1244     }
1245
1246     if (!ret)
1247         GlobalFree(hUpdate);
1248
1249     return ret;
1250 }
1251
1252
1253 /***********************************************************************
1254  *          BeginUpdateResourceA                 (KERNEL32.@)
1255  */
1256 HANDLE WINAPI BeginUpdateResourceA( LPCSTR pFileName, BOOL bDeleteExistingResources )
1257 {
1258     UNICODE_STRING FileNameW;
1259     HANDLE ret;
1260     RtlCreateUnicodeStringFromAsciiz(&FileNameW, pFileName);
1261     ret = BeginUpdateResourceW(FileNameW.Buffer, bDeleteExistingResources);
1262     RtlFreeUnicodeString(&FileNameW);
1263     return ret;
1264 }
1265
1266
1267 /***********************************************************************
1268  *          EndUpdateResourceW                 (KERNEL32.@)
1269  */
1270 BOOL WINAPI EndUpdateResourceW( HANDLE hUpdate, BOOL fDiscard )
1271 {
1272     QUEUEDUPDATES *updates;
1273     BOOL ret;
1274
1275     TRACE("%p %d\n", hUpdate, fDiscard);
1276
1277     updates = GlobalLock(hUpdate);
1278     if (!updates)
1279         return FALSE;
1280
1281     ret = fDiscard || write_raw_resources( updates );
1282
1283     free_resource_directory( &updates->root, 2 );
1284
1285     HeapFree( GetProcessHeap(), 0, updates->pFileName );
1286     GlobalUnlock( hUpdate );
1287     GlobalFree( hUpdate );
1288
1289     return ret;
1290 }
1291
1292
1293 /***********************************************************************
1294  *          EndUpdateResourceA                 (KERNEL32.@)
1295  */
1296 BOOL WINAPI EndUpdateResourceA( HANDLE hUpdate, BOOL fDiscard )
1297 {
1298     return EndUpdateResourceW(hUpdate, fDiscard);
1299 }
1300
1301
1302 /***********************************************************************
1303  *           UpdateResourceW                 (KERNEL32.@)
1304  */
1305 BOOL WINAPI UpdateResourceW( HANDLE hUpdate, LPCWSTR lpType, LPCWSTR lpName,
1306                              WORD wLanguage, LPVOID lpData, DWORD cbData)
1307 {
1308     QUEUEDUPDATES *updates;
1309     UNICODE_STRING nameW, typeW;
1310     BOOL ret = FALSE;
1311
1312     TRACE("%p %s %s %08x %p %d\n", hUpdate,
1313           debugstr_w(lpType), debugstr_w(lpName), wLanguage, lpData, cbData);
1314
1315     nameW.Buffer = typeW.Buffer = NULL;
1316     updates = GlobalLock(hUpdate);
1317     if (updates)
1318     {
1319         if (!set_ntstatus( get_res_nameW( lpName, &nameW ))) goto done;
1320         if (!set_ntstatus( get_res_nameW( lpType, &typeW ))) goto done;
1321
1322         if (lpData == NULL && cbData == 0)  /* remove resource */
1323         {
1324             ret = update_add_resource( updates, typeW.Buffer, nameW.Buffer, wLanguage, NULL, TRUE );
1325         }
1326         else
1327         {
1328             struct resource_data *data;
1329             data = allocate_resource_data( wLanguage, 0, lpData, cbData, TRUE );
1330             if (data)
1331                 ret = update_add_resource( updates, typeW.Buffer, nameW.Buffer, wLanguage, data, TRUE );
1332         }
1333
1334     done:
1335         GlobalUnlock(hUpdate);
1336     }
1337
1338     if (!IS_INTRESOURCE(nameW.Buffer)) HeapFree( GetProcessHeap(), 0, nameW.Buffer );
1339     if (!IS_INTRESOURCE(typeW.Buffer)) HeapFree( GetProcessHeap(), 0, typeW.Buffer );
1340     return ret;
1341 }
1342
1343
1344 /***********************************************************************
1345  *           UpdateResourceA                 (KERNEL32.@)
1346  */
1347 BOOL WINAPI UpdateResourceA( HANDLE hUpdate, LPCSTR lpType, LPCSTR lpName,
1348                              WORD wLanguage, LPVOID lpData, DWORD cbData)
1349 {
1350     BOOL ret;
1351     UNICODE_STRING TypeW;
1352     UNICODE_STRING NameW;
1353     if(IS_INTRESOURCE(lpType))
1354         TypeW.Buffer = ULongToPtr(LOWORD(lpType));
1355     else
1356         RtlCreateUnicodeStringFromAsciiz(&TypeW, lpType);
1357     if(IS_INTRESOURCE(lpName))
1358         NameW.Buffer = ULongToPtr(LOWORD(lpName));
1359     else
1360         RtlCreateUnicodeStringFromAsciiz(&NameW, lpName);
1361     ret = UpdateResourceW(hUpdate, TypeW.Buffer, NameW.Buffer, wLanguage, lpData, cbData);
1362     if(!IS_INTRESOURCE(lpType)) RtlFreeUnicodeString(&TypeW);
1363     if(!IS_INTRESOURCE(lpName)) RtlFreeUnicodeString(&NameW);
1364     return ret;
1365 }