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 if ( res != NULL || GetLastError() != ERROR_FILE_INVALID )
129 {
130 ok( res != NULL, "BeginUpdateResource failed\n");
131
132 /* check if it's possible to open the file now */
133 test = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
134 ok (test != INVALID_HANDLE_VALUE, "failed to create file\n");
135
136 CloseHandle( test );
137
138 r = EndUpdateResource( res, FALSE );
139 ok( r == FALSE, "EndUpdateResource failed\n");
140 }
141 else
142 skip( "Can't update resource in empty file\n" );
143
144 res = BeginUpdateResource( filename, FALSE );
145 ok( res == NULL, "BeginUpdateResource failed\n");
146 }
147
148 static void update_resources_none( void )
149 {
150 HMODULE res;
151 BOOL r;
152
153 res = BeginUpdateResource( filename, FALSE );
154 ok( res != NULL, "BeginUpdateResource failed\n");
155
156 r = EndUpdateResource( res, FALSE );
157 ok( r, "EndUpdateResource failed\n");
158 }
159
160 static void update_resources_delete( void )
161 {
162 HMODULE res;
163 BOOL r;
164
165 res = BeginUpdateResource( filename, TRUE );
166 ok( res != NULL, "BeginUpdateResource failed\n");
167
168 r = EndUpdateResource( res, FALSE );
169 ok( r, "EndUpdateResource failed\n");
170 }
171
172 static void update_resources_version(void)
173 {
174 HANDLE res = NULL;
175 BOOL r;
176 char foo[] = "red and white";
177
178 res = BeginUpdateResource( filename, TRUE );
179 ok( res != NULL, "BeginUpdateResource failed\n");
180
181 if (0) /* this causes subsequent tests to fail on Vista */
182 {
183 r = UpdateResource( res,
184 MAKEINTRESOURCE(0x1230),
185 MAKEINTRESOURCE(0x4567),
186 0xabcd,
187 NULL, 0 );
188 ok( r == FALSE, "UpdateResource failed\n");
189 }
190
191 r = UpdateResource( res,
192 MAKEINTRESOURCE(0x1230),
193 MAKEINTRESOURCE(0x4567),
194 0xabcd,
195 foo, sizeof foo );
196 ok( r == TRUE, "UpdateResource failed: %d\n", GetLastError());
197
198 r = EndUpdateResource( res, FALSE );
199 ok( r, "EndUpdateResource failed: %d\n", GetLastError());
200 }
201
202
203 typedef void (*res_check_func)( IMAGE_RESOURCE_DIRECTORY* );
204
205 static void check_empty( IMAGE_RESOURCE_DIRECTORY *dir )
206 {
207 char *pad;
208
209 ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries);
210 ok( dir->NumberOfIdEntries == 0, "NumberOfIdEntries should be 0 instead of %d\n", dir->NumberOfIdEntries);
211
212 pad = (char*) &dir[1];
213
214 ok( !memcmp( pad, "PADDINGXXPADDING", 16), "padding wrong\n");
215 }
216
217 static void check_not_empty( IMAGE_RESOURCE_DIRECTORY *dir )
218 {
219 ok( dir->NumberOfNamedEntries == 0, "NumberOfNamedEntries should be 0 instead of %d\n", dir->NumberOfNamedEntries);
220 ok( dir->NumberOfIdEntries == 1, "NumberOfIdEntries should be 1 instead of %d\n", dir->NumberOfIdEntries);
221 }
222
223 static void check_exe( res_check_func fn )
224 {
225 IMAGE_DOS_HEADER *dos;
226 IMAGE_NT_HEADERS *nt;
227 IMAGE_SECTION_HEADER *sec;
228 IMAGE_RESOURCE_DIRECTORY *dir;
229 HANDLE file, mapping;
230 DWORD length;
231
232 file = CreateFile(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
233 ok (file != INVALID_HANDLE_VALUE, "failed to create file (%d)\n", GetLastError());
234
235 length = GetFileSize( file, NULL );
236 ok( length == 0x3000, "file size wrong\n");
237
238 mapping = CreateFileMapping( file, NULL, PAGE_READONLY, 0, 0, NULL );
239 ok (mapping != NULL, "failed to create file\n");
240
241 dos = MapViewOfFile( mapping, FILE_MAP_READ, 0, 0, length );
242 ok( dos != NULL, "failed to map file\n");
243
244 if (!dos)
245 goto end;
246
247 nt = (void*) ((BYTE*) dos + dos->e_lfanew);
248 ok( nt->FileHeader.NumberOfSections == 2, "number of sections wrong\n" );
249
250 if (nt->FileHeader.NumberOfSections < 2)
251 goto end;
252
253 sec = (void*) &nt[1];
254
255 ok( !memcmp(sec[1].Name, ".rsrc", 6), "resource section name wrong\n");
256
257 dir = (void*) ((BYTE*) dos + sec[1].VirtualAddress);
258
259 ok( dir->Characteristics == 0, "Characteristics wrong\n");
260 ok( dir->TimeDateStamp == 0 || abs( dir->TimeDateStamp - GetTickCount() ) < 1000 /* nt4 */,
261 "TimeDateStamp wrong %u\n", dir->TimeDateStamp);
262 ok( dir->MajorVersion == 4, "MajorVersion wrong\n");
263 ok( dir->MinorVersion == 0, "MinorVersion wrong\n");
264
265 fn( dir );
266
267 end:
268 UnmapViewOfFile( dos );
269
270 CloseHandle( mapping );
271
272 CloseHandle( file );
273 }
274
275 static void test_find_resource(void)
276 {
277 HRSRC rsrc;
278
279 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_MENU );
280 ok( rsrc != 0, "resource not found\n" );
281 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
282 MAKELANGID( LANG_NEUTRAL, SUBLANG_NEUTRAL ));
283 ok( rsrc != 0, "resource not found\n" );
284 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
285 MAKELANGID( LANG_GERMAN, SUBLANG_DEFAULT ));
286 ok( rsrc != 0, "resource not found\n" );
287
288 SetLastError( 0xdeadbeef );
289 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(1), (LPCWSTR)RT_DIALOG );
290 ok( !rsrc, "resource found\n" );
291 ok( GetLastError() == ERROR_RESOURCE_TYPE_NOT_FOUND, "wrong error %u\n", GetLastError() );
292
293 SetLastError( 0xdeadbeef );
294 rsrc = FindResourceW( GetModuleHandle(0), (LPCWSTR)MAKEINTRESOURCE(2), (LPCWSTR)RT_MENU );
295 ok( !rsrc, "resource found\n" );
296 ok( GetLastError() == ERROR_RESOURCE_NAME_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_ENGLISH, SUBLANG_DEFAULT ) );
301 ok( !rsrc, "resource found\n" );
302 ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
303
304 SetLastError( 0xdeadbeef );
305 rsrc = FindResourceExW( GetModuleHandle(0), (LPCWSTR)RT_MENU, (LPCWSTR)MAKEINTRESOURCE(1),
306 MAKELANGID( LANG_FRENCH, SUBLANG_DEFAULT ) );
307 ok( !rsrc, "resource found\n" );
308 ok( GetLastError() == ERROR_RESOURCE_LANG_NOT_FOUND, "wrong error %u\n", GetLastError() );
309 }
310
311 START_TEST(resource)
312 {
313 DeleteFile( filename );
314 update_missing_exe();
315
316 if (GLE == ERROR_CALL_NOT_IMPLEMENTED)
317 {
318 win_skip("Resource calls are not implemented\n");
319 return;
320 }
321
322 update_empty_exe();
323 build_exe();
324 update_resources_none();
325 check_exe( check_empty );
326 update_resources_delete();
327 check_exe( check_empty );
328 update_resources_version();
329 check_exe( check_not_empty );
330 DeleteFile( filename );
331 test_find_resource();
332 }
333
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.