2 * Unit test suite for resource functions.
4 * Copyright 2006 Mike McCormack
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #include "wine/test.h"
26 static const char filename[] = "test_.exe";
31 rva_rsrc_start = page_size * 3,
35 /* rodata @ [0x1000-0x3000) */
36 static const IMAGE_SECTION_HEADER sh_rodata_1 =
38 ".rodata", {2*page_size}, page_size, 2*page_size, page_size, 0, 0, 0, 0,
39 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
42 /* rodata @ [0x1000-0x4000) */
43 static const IMAGE_SECTION_HEADER sh_rodata_2 =
45 ".rodata", {3*page_size}, page_size, 3*page_size, page_size, 0, 0, 0, 0,
46 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
49 /* rodata @ [0x1000-0x2000) */
50 static const IMAGE_SECTION_HEADER sh_rodata_3 =
52 ".rodata", {page_size}, page_size, page_size, page_size, 0, 0, 0, 0,
53 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
56 /* rsrc @ [0x3000-0x4000) */
57 static const IMAGE_SECTION_HEADER sh_rsrc_1 =
59 ".rsrc\0\0", {page_size}, rva_rsrc_start, page_size, rva_rsrc_start, 0, 0, 0, 0,
60 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
63 /* rsrc @ [0x4000-0x5000) */
64 static const IMAGE_SECTION_HEADER sh_rsrc_2 =
66 ".rsrc\0\0", {page_size}, 4*page_size, page_size, 4*page_size, 0, 0, 0, 0,
67 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
70 /* rsrc @ [0x2000-0x4000) */
71 static const IMAGE_SECTION_HEADER sh_rsrc_3 =
73 ".rsrc\0\0", {2*page_size}, rva_rsrc_start-page_size, 2*page_size, rva_rsrc_start-page_size, 0, 0, 0, 0,
74 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
77 /* rsrc @ [0x2000-0x3000) */
78 static const IMAGE_SECTION_HEADER sh_rsrc_4 =
80 ".rsrc\0\0", {page_size}, rva_rsrc_start-page_size, page_size, rva_rsrc_start-page_size, 0, 0, 0, 0,
81 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
84 /* rsrc @ [0x3000-0x6000) */
85 static const IMAGE_SECTION_HEADER sh_rsrc_5 =
87 ".rsrc\0\0", {3*page_size}, rva_rsrc_start, 3*page_size, rva_rsrc_start, 0, 0, 0, 0,
88 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
91 /* rsrc @ [0x4000-0x7000) */
92 static const IMAGE_SECTION_HEADER sh_rsrc_6 =
94 ".rsrc\0\0", {3*page_size}, 4*page_size, 3*page_size, 4*page_size, 0, 0, 0, 0,
95 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
98 /* rsrc @ [0x2000-0x5000) */
99 static const IMAGE_SECTION_HEADER sh_rsrc_7 =
101 ".rsrc\0\0", {3*page_size}, 2*page_size, 3*page_size, 2*page_size, 0, 0, 0, 0,
102 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
105 /* rsrc @ [0x3000-0x4000), small SizeOfRawData */
106 static const IMAGE_SECTION_HEADER sh_rsrc_8 =
108 ".rsrc\0\0", {page_size}, rva_rsrc_start, 8, rva_rsrc_start, 0, 0, 0, 0,
109 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
112 /* reloc @ [0x4000-0x5000) */
113 static const IMAGE_SECTION_HEADER sh_junk =
115 ".reloc\0", {page_size}, 4*page_size, page_size, 4*page_size, 0, 0, 0, 0,
116 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
119 /* reloc @ [0x6000-0x7000) */
120 static const IMAGE_SECTION_HEADER sh_junk_2 =
122 ".reloc\0", {page_size}, 6*page_size, page_size, 6*page_size, 0, 0, 0, 0,
123 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ
126 typedef struct _sec_build
128 const IMAGE_SECTION_HEADER *sect_in[max_sections];
131 typedef struct _sec_verify
133 const IMAGE_SECTION_HEADER *sect_out[max_sections];
136 DWORD NumberOfNamedEntries, NumberOfIdEntries;
139 static const struct _sec_variants
142 sec_verify chk_none, chk_delete, chk_version, chk_bigdata;
145 /* .rsrc is the last section, data directory entry points to whole section */
147 {{&sh_rodata_1, &sh_rsrc_1, NULL}},
148 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 0},
149 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 0},
150 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 1},
151 {{&sh_rodata_1, &sh_rsrc_5, NULL}, 6*page_size, 1, 0, 1}
153 /* single .rodata section with compatible characteristics, data directory entry points to section end */
154 /* Vista+ - existing section isn't used, new section is created at the end of file */
155 /* NT4/2000/2003 - image is broken */
158 {{&sh_rodata_2, NULL, NULL}},
159 {{&sh_rodata_2, &sh_rsrc_2, NULL}, 5*page_size, 1, 0, 0},
160 {{&sh_rodata_2, &sh_rsrc_2, NULL}, 5*page_size, 1, 0, 0},
161 {{&sh_rodata_2, &sh_rsrc_2, NULL}, 5*page_size, 1, 0, 1},
162 {{&sh_rodata_2, &sh_rsrc_6, NULL}, 7*page_size, 1, 0, 1}
165 /* .rsrc is the last section, data directory entry points to section end */
166 /* Vista+ - resources are moved to section start (trashing data that could be there), and section is trimmed */
167 /* NT4/2000/2003 - resources are moved to section start (trashing data that could be there); section isn't trimmed */
169 {{&sh_rodata_3, &sh_rsrc_3, NULL}},
170 {{&sh_rodata_3, &sh_rsrc_4, NULL}, 3*page_size, 1, 0, 0},
171 {{&sh_rodata_3, &sh_rsrc_4, NULL}, 3*page_size, 1, 0, 0},
172 {{&sh_rodata_3, &sh_rsrc_4, NULL}, 3*page_size, 1, 0, 1},
173 {{&sh_rodata_3, &sh_rsrc_7, NULL}, 5*page_size, 1, 0, 1}
175 /* .rsrc is not the last section */
176 /* section is reused; sections after .rsrc are shifted to give space to rsrc (in-image offset and RVA!) */
178 {{&sh_rodata_1, &sh_rsrc_1, &sh_junk}},
179 {{&sh_rodata_1, &sh_rsrc_1, &sh_junk}, 5*page_size, 1, 0, 0},
180 {{&sh_rodata_1, &sh_rsrc_1, &sh_junk}, 5*page_size, 1, 0, 0},
181 {{&sh_rodata_1, &sh_rsrc_1, &sh_junk}, 5*page_size, 1, 0, 1},
182 {{&sh_rodata_1, &sh_rsrc_5, &sh_junk_2}, 7*page_size, 1, 0, 1}
184 /* .rsrc is the last section, data directory entry points to whole section, file size is not aligned on FileAlign */
186 {{&sh_rodata_1, &sh_rsrc_8, NULL}},
187 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 0},
188 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 0},
189 {{&sh_rodata_1, &sh_rsrc_1, NULL}, 4*page_size, 1, 0, 1},
190 {{&sh_rodata_1, &sh_rsrc_5, NULL}, 6*page_size, 1, 0, 1}
194 static int build_exe( const sec_build* sec_descr )
196 IMAGE_DOS_HEADER *dos;
197 IMAGE_NT_HEADERS *nt;
198 IMAGE_SECTION_HEADER *sec;
199 IMAGE_OPTIONAL_HEADER *opt;
201 DWORD written, i, file_size;
202 BYTE page[page_size];
204 memset( page, 0, sizeof page );
207 dos->e_magic = IMAGE_DOS_SIGNATURE;
208 dos->e_lfanew = sizeof *dos;
210 nt = (void*) &dos[1];
212 nt->Signature = IMAGE_NT_SIGNATURE;
213 nt->FileHeader.Machine = IMAGE_FILE_MACHINE_I386;
214 nt->FileHeader.NumberOfSections = 0;
215 nt->FileHeader.SizeOfOptionalHeader = sizeof nt->OptionalHeader;
216 nt->FileHeader.Characteristics = IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL;
218 opt = &nt->OptionalHeader;
220 opt->Magic = IMAGE_NT_OPTIONAL_HDR_MAGIC;
221 opt->MajorLinkerVersion = 1;
222 opt->BaseOfCode = 0x10;
223 opt->ImageBase = 0x10000000;
224 opt->MajorOperatingSystemVersion = 4;
225 opt->MajorImageVersion = 1;
226 opt->MajorSubsystemVersion = 4;
227 opt->SizeOfHeaders = sizeof *dos + sizeof *nt + sizeof *sec * 2;
228 opt->SizeOfImage = page_size;
229 opt->Subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
231 /* if SectionAlignment and File alignment are not specified */
232 /* UpdateResource fails trying to create a huge temporary file */
233 opt->SectionAlignment = page_size;
234 opt->FileAlignment = page_size;
236 opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].VirtualAddress = rva_rsrc_start;
237 opt->DataDirectory[IMAGE_FILE_RESOURCE_DIRECTORY].Size = page_size;
239 sec = (void*) &nt[1];
242 for ( i = 0; i < max_sections; i++ )
243 if ( sec_descr->sect_in[i] )
245 DWORD virt_end_of_section = sec_descr->sect_in[i]->Misc.VirtualSize +
246 sec_descr->sect_in[i]->VirtualAddress;
247 DWORD phys_end_of_section = sec_descr->sect_in[i]->SizeOfRawData +
248 sec_descr->sect_in[i]->PointerToRawData;
249 memcpy( sec + nt->FileHeader.NumberOfSections, sec_descr->sect_in[i],
251 nt->FileHeader.NumberOfSections++;
252 if ( opt->SizeOfImage < virt_end_of_section )
253 opt->SizeOfImage = virt_end_of_section;
254 if ( file_size < phys_end_of_section )
255 file_size = phys_end_of_section;
258 file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
259 ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
261 /* write out the header */
262 WriteFile( file, page, sizeof page, &written, NULL );
264 /* write out zeroed pages for sections */
265 memset( page, 0, sizeof page );
266 for ( i = page_size; i < file_size; i += page_size )
268 DWORD size = min(page_size, file_size - i);
269 WriteFile( file, page, size, &written, NULL );
277 static void update_missing_exe( void )
281 SetLastError(0xdeadbeef);
282 res = BeginUpdateResource( filename, TRUE );
283 GLE = GetLastError();
284 ok( res == NULL, "BeginUpdateResource should fail\n");
287 static void update_empty_exe( void )
289 HANDLE file, res, test;
292 file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
293 ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
297 res = BeginUpdateResource( filename, TRUE );
298 if ( res != NULL || GetLastError() != ERROR_FILE_INVALID )
300 ok( res != NULL, "BeginUpdateResource failed\n");
302 /* check if it's possible to open the file now */
303 test = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
304 ok (test != INVALID_HANDLE_VALUE, "failed to create file\n");
308 r = EndUpdateResource( res, FALSE );
309 ok( r == FALSE, "EndUpdateResource failed\n");
312 skip( "Can't update resource in empty file\n" );
314 res = BeginUpdateResource( filename, FALSE );
315 ok( res == NULL, "BeginUpdateResource failed\n");
318 static void update_resources_none( void )
323 res = BeginUpdateResource( filename, FALSE );
324 ok( res != NULL, "BeginUpdateResource failed\n");
326 r = EndUpdateResource( res, FALSE );
327 ok( r, "EndUpdateResource failed\n");
330 static void update_resources_delete( void )
335 res = BeginUpdateResource( filename, TRUE );
336 ok( res != NULL, "BeginUpdateResource failed\n");
338 r = EndUpdateResource( res, FALSE );
339 ok( r, "EndUpdateResource failed\n");
342 static void update_resources_version( void )
346 char foo[] = "red and white";
348 res = BeginUpdateResource( filename, TRUE );
349 ok( res != NULL, "BeginUpdateResource failed\n");
351 if (0) /* this causes subsequent tests to fail on Vista */
353 r = UpdateResource( res,
354 MAKEINTRESOURCE(0x1230),
355 MAKEINTRESOURCE(0x4567),
358 ok( r == FALSE, "UpdateResource failed\n");
361 r = UpdateResource( res,
362 MAKEINTRESOURCE(0x1230),
363 MAKEINTRESOURCE(0x4567),
366 ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
368 r = EndUpdateResource( res, FALSE );
369 ok( r, "EndUpdateResource failed: %d\n", GetLastError());
372 static void update_resources_bigdata( void )
376 char foo[2*page_size] = "foobar";
378 res = BeginUpdateResource( filename, TRUE );
379 ok( res != NULL, "BeginUpdateResource succeeded\n");
381 r = UpdateResource( res,
382 MAKEINTRESOURCE(0x3012),
383 MAKEINTRESOURCE(0x5647),
386 ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
388 r = EndUpdateResource( res, FALSE );
389 ok( r, "EndUpdateResource failed\n");
392 static void check_exe( const sec_verify *verify )
395 IMAGE_DOS_HEADER *dos;
396 IMAGE_NT_HEADERS *nt;
397 IMAGE_SECTION_HEADER *sec;
398 IMAGE_RESOURCE_DIRECTORY *dir;
399 HANDLE file, mapping;
400 DWORD length, sec_count = 0;
402 file = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
403 ok (file != INVALID_HANDLE_VALUE, "failed to create file (%d)\n", GetLastError());
405 length = GetFileSize( file, NULL );
406 ok( length >= verify->length, "file size wrong\n");
408 mapping = CreateFileMapping( file, NULL, PAGE_READONLY, 0, 0, NULL );
409 ok (mapping != NULL, "failed to create file\n");
411 dos = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, length );
412 ok( dos != NULL, "failed to map file\n");
417 nt = (void*) ((BYTE*) dos + dos->e_lfanew);
418 sec = (void*) &nt[1];
420 for(i = 0; i < max_sections; i++)
421 if (verify->sect_out[i])
423 ok( !memcmp(&verify->sect_out[i]->Name, &sec[sec_count].Name, 8), "section %d name wrong\n", sec_count);
424 ok( verify->sect_out[i]->VirtualAddress == sec[sec_count].VirtualAddress, "section %d vaddr wrong\n", sec_count);
425 ok( verify->sect_out[i]->SizeOfRawData <= sec[sec_count].SizeOfRawData, "section %d SizeOfRawData wrong (%d vs %d)\n", sec_count, verify->sect_out[i]->SizeOfRawData ,sec[sec_count].SizeOfRawData);
426 ok( verify->sect_out[i]->PointerToRawData == sec[sec_count].PointerToRawData, "section %d PointerToRawData wrong\n", sec_count);
427 ok( verify->sect_out[i]->Characteristics == sec[sec_count].Characteristics , "section %d characteristics wrong\n", sec_count);
431 ok( nt->FileHeader.NumberOfSections == sec_count, "number of sections wrong\n" );
433 if (verify->rsrc_section >= 0 && verify->rsrc_section < nt->FileHeader.NumberOfSections)
435 dir = (void*) ((BYTE*) dos + sec[verify->rsrc_section].VirtualAddress);
437 ok( dir->Characteristics == 0, "Characteristics wrong\n");
438 ok( dir->TimeDateStamp == 0 || abs( dir->TimeDateStamp - GetTickCount() ) < 1000 /* nt4 */,
439 "TimeDateStamp wrong %u\n", dir->TimeDateStamp);
440 ok( dir->MajorVersion == 4, "MajorVersion wrong\n");
441 ok( dir->MinorVersion == 0, "MinorVersion wrong\n");
443 ok( dir->NumberOfNamedEntries == verify->NumberOfNamedEntries, "NumberOfNamedEntries should be %d instead of %d\n",
444 verify->NumberOfNamedEntries, dir->NumberOfNamedEntries);
445 ok( dir->NumberOfIdEntries == verify->NumberOfIdEntries, "NumberOfIdEntries should be %d instead of %d\n",
446 verify->NumberOfIdEntries, dir->NumberOfIdEntries);
450 UnmapViewOfFile( dos );
452 CloseHandle( mapping );
457 static void test_find_resource(void)
461 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_MENU );
462 ok( rsrc != 0, "resource not found\n" );
463 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
464 MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ));
465 ok( rsrc != 0, "resource not found\n" );
466 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
467 MAKELANGID( LANG_GERMAN, SUBLANG_DEFAULT ));
468 ok( rsrc != 0, "resource not found\n" );
470 SetLastError( 0xdeadbeef );
471 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_DIALOG );
472 ok( !rsrc, "resource found\n" );
473 ok( GetLastError() == ERROR_RESOURCE_TYPE_NOT_FOUND, "wrong error %u\n", GetLastError() );
475 SetLastError( 0xdeadbeef );
476 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(2), (LPCWSTR)RT_MENU );
477 ok( !rsrc, "resource found\n" );
478 ok( GetLastError() == ERROR_RESOURCE_NAME_NOT_FOUND, "wrong error %u\n", GetLastError() );
480 SetLastError( 0xdeadbeef );
481 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
482 MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) );
483 ok( !rsrc, "resource found\n" );
484 ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
486 SetLastError( 0xdeadbeef );
487 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
488 MAKELANGID( LANG_FRENCH, SUBLANG_DEFAULT ) );
489 ok( !rsrc, "resource found\n" );
490 ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
497 DeleteFile( filename );
498 update_missing_exe();
500 if (GLE == ERROR_CALL_NOT_IMPLEMENTED)
502 win_skip("Resource calls are not implemented\n");
508 for(i=0; i < sizeof( sec_variants ) / sizeof( sec_variants[0] ); i++)
510 const struct _sec_variants *sec = &sec_variants[i];
511 build_exe( &sec->build );
512 update_resources_none();
513 check_exe( &sec->chk_none );
514 update_resources_delete();
515 check_exe( &sec->chk_delete );
516 update_resources_version();
517 check_exe( &sec->chk_version );
518 update_resources_bigdata();
519 check_exe( &sec->chk_bigdata );
520 DeleteFile( filename );
522 test_find_resource();