ntdll: Allocate a new virtual region for large blocks, and ensure 16-byte alignment.
[wine] / dlls / kernel32 / tests / resource.c
1 /*
2  * Unit test suite for environment functions.
3  *
4  * Copyright 2006 Mike McCormack
5  *
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.
10  *
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.
15  *
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
19  */
20
21 #include <windows.h>
22 #include <stdio.h>
23
24 #include "wine/test.h"
25
26 static const char filename[] = "test_.exe";
27 static DWORD GLE;
28
29 static int build_exe( void )
30 {
31     IMAGE_DOS_HEADER *dos;
32     IMAGE_NT_HEADERS *nt;
33     IMAGE_SECTION_HEADER *sec;
34     IMAGE_OPTIONAL_HEADER *opt;
35     HANDLE file;
36     DWORD written;
37     BYTE page[0x1000];
38     const int page_size = 0x1000;
39
40     memset( page, 0, sizeof page );
41
42     dos = (void*) page;
43     dos->e_magic = IMAGE_DOS_SIGNATURE;
44     dos->e_lfanew = sizeof *dos;
45
46     nt = (void*) &dos[1];
47
48     nt->Signature = IMAGE_NT_SIGNATURE;
49     nt->FileHeader.Machine = IMAGE_FILE_MACHINE_I386;
50     nt->FileHeader.NumberOfSections = 2;
51     nt->FileHeader.SizeOfOptionalHeader = sizeof nt->OptionalHeader;
52     nt->FileHeader.Characteristics = IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL;
53
54     opt = &nt->OptionalHeader;
55
56     opt->Magic = IMAGE_NT_OPTIONAL_HDR_MAGIC;
57     opt->MajorLinkerVersion = 1;
58     opt->BaseOfCode = 0x10;
59     opt->ImageBase = 0x10000000;
60     opt->MajorOperatingSystemVersion = 4;
61     opt->MajorImageVersion = 1;
62     opt->MajorSubsystemVersion = 4;
63     opt->SizeOfHeaders = sizeof *dos + sizeof *nt + sizeof *sec * 2;
64     opt->SizeOfImage = page_size*3;
65     opt->Subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
66
67     /* if SectionAlignment and File alignment are not specified */
68     /* UpdateResource fails trying to create a huge temporary file */
69     opt->SectionAlignment = page_size;
70     opt->FileAlignment = page_size;
71
72     sec = (void*) &nt[1];
73
74     memcpy( sec[0].Name, ".rodata", 8 );
75     sec[0].Misc.VirtualSize = page_size;
76     sec[0].PointerToRawData = page_size;
77     sec[0].SizeOfRawData = page_size;
78     sec[0].VirtualAddress = page_size;
79     sec[0].Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ;
80
81     memcpy( sec[1].Name, ".rsrc", 6 );
82     sec[1].Misc.VirtualSize = page_size;
83     sec[1].SizeOfRawData = page_size;
84     sec[1].PointerToRawData = page_size*2;
85     sec[1].VirtualAddress = page_size*2;
86     sec[1].Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ;
87
88     file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
89     ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
90
91     /* write out the header */
92     WriteFile( file, page, sizeof page, &written, NULL );
93
94     /* write out an empty page for rodata */
95     memset( page, 0, sizeof page );
96     WriteFile( file, page, sizeof page, &written, NULL );
97
98     /* write out an empty page for the resources */
99     memset( page, 0, sizeof page );
100     WriteFile( file, page, sizeof page, &written, NULL );
101
102     CloseHandle( file );
103
104     return 0;
105 }
106
107 static void update_missing_exe( void )
108 {
109     HANDLE res;
110
111     SetLastError(0xdeadbeef);
112     res = BeginUpdateResource( filename, TRUE );
113     GLE = GetLastError();
114     ok( res == NULL, "BeginUpdateResource should fail\n");
115 }
116
117 static void update_empty_exe( void )
118 {
119     HANDLE file, res, test;
120     BOOL r;
121
122     file = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
123     ok (file != INVALID_HANDLE_VALUE, "failed to create file\n");
124
125     CloseHandle( file );
126
127     res = BeginUpdateResource( filename, TRUE );
128     ok( res != NULL, "BeginUpdateResource failed\n");
129
130     /* check if it's possible to open the file now */
131     test = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
132     ok (test != INVALID_HANDLE_VALUE, "failed to create file\n");
133
134     CloseHandle( test );
135
136     r = EndUpdateResource( res, FALSE );
137     ok( r == FALSE, "EndUpdateResource failed\n");
138
139     res = BeginUpdateResource( filename, FALSE );
140     ok( res == NULL, "BeginUpdateResource failed\n");
141 }
142
143 static void update_resources_none( void )
144 {
145     HMODULE res;
146     BOOL r;
147
148     res = BeginUpdateResource( filename, FALSE );
149     ok( res != NULL, "BeginUpdateResource failed\n");
150
151     r = EndUpdateResource( res, FALSE );
152     ok( r, "EndUpdateResource failed\n");
153 }
154
155 static void update_resources_delete( void )
156 {
157     HMODULE res;
158     BOOL r;
159
160     res = BeginUpdateResource( filename, TRUE );
161     ok( res != NULL, "BeginUpdateResource failed\n");
162
163     r = EndUpdateResource( res, FALSE );
164     ok( r, "EndUpdateResource failed\n");
165 }
166
167 static void update_resources_version(void)
168 {
169     HANDLE res = NULL;
170     BOOL r;
171     char foo[] = "red and white";
172
173     res = BeginUpdateResource( filename, TRUE );
174     ok( res != NULL, "BeginUpdateResource failed\n");
175
176     if (0)  /* this causes subsequent tests to fail on Vista */
177     {
178         r = UpdateResource( res,
179                             MAKEINTRESOURCE(0x1230),
180                             MAKEINTRESOURCE(0x4567),
181                             0xabcd,
182                             NULL, 0 );
183         ok( r == FALSE, "UpdateResource failed\n");
184     }
185
186     r = UpdateResource( res,
187                         MAKEINTRESOURCE(0x1230),
188                         MAKEINTRESOURCE(0x4567),
189                         0xabcd,
190                         foo, sizeof foo );
191     ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
192
193     r = EndUpdateResource( res, FALSE );
194     ok( r, "EndUpdateResource failed: %d\n", GetLastError());
195 }
196
197
198 typedef void (*res_check_func)( IMAGE_RESOURCE_DIRECTORY* );
199
200 static void check_empty( IMAGE_RESOURCE_DIRECTORY *dir )
201 {
202     char *pad;
203
204     ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries);
205     ok( dir->NumberOfIdEntries == 0, "NumberOfIdEntries should be 0 instead of %d\n", dir->NumberOfIdEntries);
206
207     pad = (char*) &dir[1];
208
209     ok( !memcmp( pad, "PADDINGXXPADDING", 16), "padding wrong\n");
210 }
211
212 static void check_not_empty( IMAGE_RESOURCE_DIRECTORY *dir )
213 {
214     ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries);
215     ok( dir->NumberOfIdEntries == 1, "NumberOfIdEntries should be 1 instead of %d\n", dir->NumberOfIdEntries);
216 }
217
218 static void check_exe( res_check_func fn )
219 {
220     IMAGE_DOS_HEADER *dos;
221     IMAGE_NT_HEADERS *nt;
222     IMAGE_SECTION_HEADER *sec;
223     IMAGE_RESOURCE_DIRECTORY *dir;
224     HANDLE file, mapping;
225     DWORD length;
226
227     file = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
228     ok (file != INVALID_HANDLE_VALUE, "failed to create file (%d)\n", GetLastError());
229
230     length = GetFileSize( file, NULL );
231     ok( length == 0x3000, "file size wrong\n");
232
233     mapping = CreateFileMapping( file, NULL, PAGE_READONLY, 0, 0, NULL );
234     ok (mapping != NULL, "failed to create file\n");
235
236     dos = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, length );
237     ok( dos != NULL, "failed to map file\n");
238
239     if (!dos)
240         goto end;
241
242     nt = (void*) ((BYTE*) dos + dos->e_lfanew);
243     ok( nt->FileHeader.NumberOfSections == 2, "number of sections wrong\n" );
244
245     if (nt->FileHeader.NumberOfSections < 2)
246         goto end;
247
248     sec = (void*) &nt[1];
249
250     ok( !memcmp(sec[1].Name, ".rsrc", 6), "resource section name wrong\n");
251
252     dir = (void*) ((BYTE*) dos + sec[1].VirtualAddress);
253
254     ok( dir->Characteristics == 0, "Characteristics wrong\n");
255     ok( dir->TimeDateStamp == 0, "TimeDateStamp wrong\n");
256     ok( dir->MajorVersion == 4, "MajorVersion wrong\n");
257     ok( dir->MinorVersion == 0, "MinorVersion wrong\n");
258
259     fn( dir );
260
261 end:
262     UnmapViewOfFile( dos );
263
264     CloseHandle( mapping );
265
266     CloseHandle( file );
267 }
268
269 static void test_find_resource(void)
270 {
271     HRSRC rsrc;
272
273     rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_MENU );
274     ok( rsrc != 0, "resource not found\n" );
275     rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
276                             MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ));
277     ok( rsrc != 0, "resource not found\n" );
278     rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
279                             MAKELANGID( LANG_GERMAN, SUBLANG_DEFAULT ));
280     ok( rsrc != 0, "resource not found\n" );
281
282     SetLastError( 0xdeadbeef );
283     rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_DIALOG );
284     ok( !rsrc, "resource found\n" );
285     ok( GetLastError() == ERROR_RESOURCE_TYPE_NOT_FOUND, "wrong error %u\n", GetLastError() );
286
287     SetLastError( 0xdeadbeef );
288     rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(2), (LPCWSTR)RT_MENU );
289     ok( !rsrc, "resource found\n" );
290     ok( GetLastError() == ERROR_RESOURCE_NAME_NOT_FOUND, "wrong error %u\n", GetLastError() );
291
292     SetLastError( 0xdeadbeef );
293     rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
294                             MAKELANGID( LANG_ENGLISH, SUBLANG_DEFAULT ) );
295     ok( !rsrc, "resource found\n" );
296     ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
297
298     SetLastError( 0xdeadbeef );
299     rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
300                             MAKELANGID( LANG_FRENCH, SUBLANG_DEFAULT ) );
301     ok( !rsrc, "resource found\n" );
302     ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
303 }
304
305 START_TEST(resource)
306 {
307     DeleteFile( filename );
308     update_missing_exe();
309
310     if (GLE == ERROR_CALL_NOT_IMPLEMENTED)
311     {
312         skip("Resource calls are not implemented\n");
313         return;
314     }
315
316     update_empty_exe();
317     build_exe();
318     update_resources_none();
319     check_exe( check_empty );
320     update_resources_delete();
321     check_exe( check_empty );
322     update_resources_version();
323     check_exe( check_not_empty );
324     DeleteFile( filename );
325     test_find_resource();
326 }