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 }
327
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.