| /* |
| * Unit test suite for resource functions. |
| * |
| * Copyright 2006 Mike McCormack |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include <windows.h> |
| #include <stdio.h> |
| |
| #include "wine/test.h" |
| |
| static const char filename[] = "test_.exe"; |
| static DWORD GLE; |
| |
| static int build_exe( void ) |
| { |
| IMAGE_DOS_HEADER *dos; |
| IMAGE_NT_HEADERS *nt; |
| IMAGE_SECTION_HEADER *sec; |
| IMAGE_OPTIONAL_HEADER *opt; |
| HANDLE file; |
| DWORD written; |
| BYTE page[0x1000]; |
| const int page_size = 0x1000; |
| |
| memset( page, 0, sizeof page ); |
| |
| dos = (void*) page; |
| dos->e_magic = IMAGE_DOS_SIGNATURE; |
| dos->e_lfanew = sizeof *dos; |
| |
| nt = (void*) &dos[1]; |
| |
| nt->Signature = IMAGE_NT_SIGNATURE; |
| nt->FileHeader.Machine = IMAGE_FILE_MACHINE_I386; |
| nt->FileHeader.NumberOfSections = 2; |
| nt->FileHeader.SizeOfOptionalHeader = sizeof nt->OptionalHeader; |
| nt->FileHeader.Characteristics = IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL; |
| |
| opt = &nt->OptionalHeader; |
| |
| opt->Magic = IMAGE_NT_OPTIONAL_HDR_MAGIC; |
| opt->MajorLinkerVersion = 1; |
| opt->BaseOfCode = 0x10; |
| opt->ImageBase = 0x10000000; |
| opt->MajorOperatingSystemVersion = 4; |
| opt->MajorImageVersion = 1; |
| opt->MajorSubsystemVersion = 4; |
| opt->SizeOfHeaders = sizeof *dos + sizeof *nt + sizeof *sec * 2; |
| opt->SizeOfImage = page_size*3; |
| opt->Subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI; |
| |
| /* if SectionAlignment and File alignment are not specified */ |
| /* UpdateResource fails trying to create a huge temporary file */ |
| opt->SectionAlignment = page_size; |
| opt->FileAlignment = page_size; |
| |
| sec = (void*) &nt[1]; |
| |
| memcpy( sec[0].Name, ".rodata", 8 ); |
| sec[0].Misc.VirtualSize = page_size; |
| sec[0].PointerToRawData = page_size; |
| sec[0].SizeOfRawData = page_size; |
| sec[0].VirtualAddress = page_size; |
| sec[0].Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ; |
| |
| memcpy( sec[1].Name, ".rsrc", 6 ); |
| sec[1].Misc.VirtualSize = page_size; |
| sec[1].SizeOfRawData = page_size; |
| sec[1].PointerToRawData = page_size*2; |
| sec[1].VirtualAddress = page_size*2; |
| sec[1].Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ; |
| |
| file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0); |
| ok (file != INVALID_HANDLE_VALUE, "failed to create file\n"); |
| |
| /* write out the header */ |
| WriteFile( file, page, sizeof page, &written, NULL ); |
| |
| /* write out an empty page for rodata */ |
| memset( page, 0, sizeof page ); |
| WriteFile( file, page, sizeof page, &written, NULL ); |
| |
| /* write out an empty page for the resources */ |
| memset( page, 0, sizeof page ); |
| WriteFile( file, page, sizeof page, &written, NULL ); |
| |
| CloseHandle( file ); |
| |
| return 0; |
| } |
| |
| static void update_missing_exe( void ) |
| { |
| HANDLE res; |
| |
| SetLastError(0xdeadbeef); |
| res = BeginUpdateResource( filename, TRUE ); |
| GLE = GetLastError(); |
| ok( res == NULL, "BeginUpdateResource should fail\n"); |
| } |
| |
| static void update_empty_exe( void ) |
| { |
| HANDLE file, res, test; |
| BOOL r; |
| |
| file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0); |
| ok (file != INVALID_HANDLE_VALUE, "failed to create file\n"); |
| |
| CloseHandle( file ); |
| |
| res = BeginUpdateResource( filename, TRUE ); |
| if ( res != NULL || GetLastError() != ERROR_FILE_INVALID ) |
| { |
| ok( res != NULL, "BeginUpdateResource failed\n"); |
| |
| /* check if it's possible to open the file now */ |
| test = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0); |
| ok (test != INVALID_HANDLE_VALUE, "failed to create file\n"); |
| |
| CloseHandle( test ); |
| |
| r = EndUpdateResource( res, FALSE ); |
| ok( r == FALSE, "EndUpdateResource failed\n"); |
| } |
| else |
| skip( "Can't update resource in empty file\n" ); |
| |
| res = BeginUpdateResource( filename, FALSE ); |
| ok( res == NULL, "BeginUpdateResource failed\n"); |
| } |
| |
| static void update_resources_none( void ) |
| { |
| HMODULE res; |
| BOOL r; |
| |
| res = BeginUpdateResource( filename, FALSE ); |
| ok( res != NULL, "BeginUpdateResource failed\n"); |
| |
| r = EndUpdateResource( res, FALSE ); |
| ok( r, "EndUpdateResource failed\n"); |
| } |
| |
| static void update_resources_delete( void ) |
| { |
| HMODULE res; |
| BOOL r; |
| |
| res = BeginUpdateResource( filename, TRUE ); |
| ok( res != NULL, "BeginUpdateResource failed\n"); |
| |
| r = EndUpdateResource( res, FALSE ); |
| ok( r, "EndUpdateResource failed\n"); |
| } |
| |
| static void update_resources_version(void) |
| { |
| HANDLE res = NULL; |
| BOOL r; |
| char foo[] = "red and white"; |
| |
| res = BeginUpdateResource( filename, TRUE ); |
| ok( res != NULL, "BeginUpdateResource failed\n"); |
| |
| if (0) /* this causes subsequent tests to fail on Vista */ |
| { |
| r = UpdateResource( res, |
| MAKEINTRESOURCE(0x1230), |
| MAKEINTRESOURCE(0x4567), |
| 0xabcd, |
| NULL, 0 ); |
| ok( r == FALSE, "UpdateResource failed\n"); |
| } |
| |
| r = UpdateResource( res, |
| MAKEINTRESOURCE(0x1230), |
| MAKEINTRESOURCE(0x4567), |
| 0xabcd, |
| foo, sizeof foo ); |
| ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError()); |
| |
| r = EndUpdateResource( res, FALSE ); |
| ok( r, "EndUpdateResource failed: %d\n", GetLastError()); |
| } |
| |
| |
| typedef void (*res_check_func)( IMAGE_RESOURCE_DIRECTORY* ); |
| |
| static void check_empty( IMAGE_RESOURCE_DIRECTORY *dir ) |
| { |
| char *pad; |
| |
| ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries); |
| ok( dir->NumberOfIdEntries == 0, "NumberOfIdEntries should be 0 instead of %d\n", dir->NumberOfIdEntries); |
| |
| pad = (char*) &dir[1]; |
| |
| ok( !memcmp( pad, "PADDINGXXPADDING", 16), "padding wrong\n"); |
| } |
| |
| static void check_not_empty( IMAGE_RESOURCE_DIRECTORY *dir ) |
| { |
| ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries); |
| ok( dir->NumberOfIdEntries == 1, "NumberOfIdEntries should be 1 instead of %d\n", dir->NumberOfIdEntries); |
| } |
| |
| static void check_exe( res_check_func fn ) |
| { |
| IMAGE_DOS_HEADER *dos; |
| IMAGE_NT_HEADERS *nt; |
| IMAGE_SECTION_HEADER *sec; |
| IMAGE_RESOURCE_DIRECTORY *dir; |
| HANDLE file, mapping; |
| DWORD length; |
| |
| file = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0); |
| ok (file != INVALID_HANDLE_VALUE, "failed to create file (%d)\n", GetLastError()); |
| |
| length = GetFileSize( file, NULL ); |
| ok( length == 0x3000, "file size wrong\n"); |
| |
| mapping = CreateFileMapping( file, NULL, PAGE_READONLY, 0, 0, NULL ); |
| ok (mapping != NULL, "failed to create file\n"); |
| |
| dos = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, length ); |
| ok( dos != NULL, "failed to map file\n"); |
| |
| if (!dos) |
| goto end; |
| |
| nt = (void*) ((BYTE*) dos + dos->e_lfanew); |
| ok( nt->FileHeader.NumberOfSections == 2, "number of sections wrong\n" ); |
| |
| if (nt->FileHeader.NumberOfSections < 2) |
| goto end; |
| |
| sec = (void*) &nt[1]; |
| |
| ok( !memcmp(sec[1].Name, ".rsrc", 6), "resource section name wrong\n"); |
| |
| dir = (void*) ((BYTE*) dos + sec[1].VirtualAddress); |
| |
| ok( dir->Characteristics == 0, "Characteristics wrong\n"); |
| ok( dir->TimeDateStamp == 0 || abs( dir->TimeDateStamp - GetTickCount() ) < 1000 /* nt4 */, |
| "TimeDateStamp wrong %u\n", dir->TimeDateStamp); |
| ok( dir->MajorVersion == 4, "MajorVersion wrong\n"); |
| ok( dir->MinorVersion == 0, "MinorVersion wrong\n"); |
| |
| fn( dir ); |
| |
| end: |
| UnmapViewOfFile( dos ); |
| |
| CloseHandle( mapping ); |
| |
| CloseHandle( file ); |
| } |
| |
| static void test_find_resource(void) |
| { |
| HRSRC rsrc; |
| |
| rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_MENU ); |
| ok( rsrc != 0, "resource not found\n" ); |
| rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1), |
| MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL )); |
| ok( rsrc != 0, "resource not found\n" ); |
| rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1), |
| MAKELANGID( LANG_GERMAN, SUBLANG_DEFAULT )); |
| ok( rsrc != 0, "resource not found\n" ); |
| |
| SetLastError( 0xdeadbeef ); |
| rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_DIALOG ); |
| ok( !rsrc, "resource found\n" ); |
| ok( GetLastError() == ERROR_RESOURCE_TYPE_NOT_FOUND, "wrong error %u\n", GetLastError() ); |
| |
| SetLastError( 0xdeadbeef ); |
| rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(2), (LPCWSTR)RT_MENU ); |
| ok( !rsrc, "resource found\n" ); |
| ok( GetLastError() == ERROR_RESOURCE_NAME_NOT_FOUND, "wrong error %u\n", GetLastError() ); |
| |
| SetLastError( 0xdeadbeef ); |
| rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1), |
| MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) ); |
| ok( !rsrc, "resource found\n" ); |
| ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() ); |
| |
| SetLastError( 0xdeadbeef ); |
| rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1), |
| MAKELANGID( LANG_FRENCH, SUBLANG_DEFAULT ) ); |
| ok( !rsrc, "resource found\n" ); |
| ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() ); |
| } |
| |
| START_TEST(resource) |
| { |
| DeleteFile( filename ); |
| update_missing_exe(); |
| |
| if (GLE == ERROR_CALL_NOT_IMPLEMENTED) |
| { |
| win_skip("Resource calls are not implemented\n"); |
| return; |
| } |
| |
| update_empty_exe(); |
| build_exe(); |
| update_resources_none(); |
| check_exe( check_empty ); |
| update_resources_delete(); |
| check_exe( check_empty ); |
| update_resources_version(); |
| check_exe( check_not_empty ); |
| DeleteFile( filename ); |
| test_find_resource(); |
| } |