2 * Unit test suite for the PE loader.
4 * Copyright 2006 Dmitry Timoshkov
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
26 #include "wine/test.h"
28 #define ALIGN_SIZE(size, alignment) (((size) + (alignment - 1)) & ~((alignment - 1)))
32 WORD e_magic; /* 00: MZ Header signature */
34 DWORD e_lfanew; /* 3c: Offset to extended header */
37 IMAGE_DOS_SIGNATURE, { 0 }, sizeof(dos_header)
40 static IMAGE_NT_HEADERS nt_header =
42 IMAGE_NT_SIGNATURE, /* Signature */
44 { IMAGE_FILE_MACHINE_I386, /* Machine */
46 # error You must specify the machine type
48 1, /* NumberOfSections */
49 0, /* TimeDateStamp */
50 0, /* PointerToSymbolTable */
51 0, /* NumberOfSymbols */
52 sizeof(IMAGE_OPTIONAL_HEADER), /* SizeOfOptionalHeader */
53 IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL /* Characteristics */
55 { IMAGE_NT_OPTIONAL_HDR_MAGIC, /* Magic */
56 1, /* MajorLinkerVersion */
57 0, /* MinorLinkerVersion */
59 0, /* SizeOfInitializedData */
60 0, /* SizeOfUninitializedData */
61 0, /* AddressOfEntryPoint */
62 0x10, /* BaseOfCode, also serves as e_lfanew in the truncated MZ header */
64 0x10000000, /* ImageBase */
65 0, /* SectionAlignment */
66 0, /* FileAlignment */
67 4, /* MajorOperatingSystemVersion */
68 0, /* MinorOperatingSystemVersion */
69 1, /* MajorImageVersion */
70 0, /* MinorImageVersion */
71 4, /* MajorSubsystemVersion */
72 0, /* MinorSubsystemVersion */
73 0, /* Win32VersionValue */
74 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000, /* SizeOfImage */
75 sizeof(dos_header) + sizeof(nt_header), /* SizeOfHeaders */
77 IMAGE_SUBSYSTEM_WINDOWS_CUI, /* Subsystem */
78 0, /* DllCharacteristics */
79 0, /* SizeOfStackReserve */
80 0, /* SizeOfStackCommit */
81 0, /* SizeOfHeapReserve */
82 0, /* SizeOfHeapCommit */
84 0, /* NumberOfRvaAndSizes */
85 { { 0 } } /* DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES] */
89 static IMAGE_SECTION_HEADER section =
93 0, /* VirtualAddress */
94 0x0a, /* SizeOfRawData */
95 0, /* PointerToRawData */
96 0, /* PointerToRelocations */
97 0, /* PointerToLinenumbers */
98 0, /* NumberOfRelocations */
99 0, /* NumberOfLinenumbers */
100 IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ, /* Characteristics */
105 static const struct test_data
107 const void *dos_header;
108 DWORD size_of_dos_header;
109 WORD number_of_sections, size_of_optional_header;
110 DWORD section_alignment, file_alignment;
111 DWORD size_of_image, size_of_headers;
112 DWORD error; /* 0 means LoadLibrary should succeed */
115 { &dos_header, sizeof(dos_header),
119 { &dos_header, sizeof(dos_header),
120 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
121 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0xe00,
122 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
123 ERROR_BAD_EXE_FORMAT /* XP doesn't like too small image size */
125 { &dos_header, sizeof(dos_header),
126 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
127 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
128 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
131 { &dos_header, sizeof(dos_header),
132 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
137 { &dos_header, sizeof(dos_header),
138 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x200, 0x200,
139 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
140 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
143 { &dos_header, sizeof(dos_header),
144 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x200, 0x1000,
145 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
146 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
147 ERROR_BAD_EXE_FORMAT /* XP doesn't like aligments */
149 { &dos_header, sizeof(dos_header),
150 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x200,
151 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
152 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
155 { &dos_header, sizeof(dos_header),
156 1, sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x200,
157 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
161 /* Mandatory are all fields up to SizeOfHeaders, everything else
162 * is really optional (at least that's true for XP).
164 { &dos_header, sizeof(dos_header),
165 1, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
166 sizeof(dos_header) + sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER) + FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum) + sizeof(IMAGE_SECTION_HEADER) + 0x10,
167 sizeof(dos_header) + sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER) + FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum) + sizeof(IMAGE_SECTION_HEADER),
170 { &dos_header, sizeof(dos_header),
171 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
172 0xd0, /* beyond of the end of file */
173 0xc0, /* beyond of the end of file */
176 { &dos_header, sizeof(dos_header),
177 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
182 { &dos_header, sizeof(dos_header),
183 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
188 { &dos_header, sizeof(dos_header),
189 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 4, 4,
194 { &dos_header, sizeof(dos_header),
195 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 1, 1,
200 { &dos_header, sizeof(dos_header),
201 0, FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
204 ERROR_BAD_EXE_FORMAT /* image size == 0 -> failure */
206 /* the following data mimics the PE image which upack creates */
208 1, 0x148, 0x1000, 0x200,
209 sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
214 static const char filler[0x1000];
215 static const char section_data[0x10] = "section data";
217 DWORD dummy, file_size, file_align;
220 char temp_path[MAX_PATH];
221 char dll_name[MAX_PATH];
224 trace("system page size 0x%04x\n", si.dwPageSize);
226 /* prevent displaying of the "Unable to load this DLL" message box */
227 SetErrorMode(SEM_FAILCRITICALERRORS);
229 GetTempPath(MAX_PATH, temp_path);
231 for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
233 GetTempFileName(temp_path, "ldr", 0, dll_name);
235 /*trace("creating %s\n", dll_name);*/
236 hfile = CreateFileA(dll_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
237 if (hfile == INVALID_HANDLE_VALUE)
239 ok(0, "could not create %s\n", dll_name);
243 SetLastError(0xdeadbeef);
244 ok(WriteFile(hfile, td[i].dos_header, td[i].size_of_dos_header, &dummy, NULL),
245 "WriteFile error %d\n", GetLastError());
247 nt_header.FileHeader.NumberOfSections = td[i].number_of_sections;
248 nt_header.FileHeader.SizeOfOptionalHeader = td[i].size_of_optional_header;
250 nt_header.OptionalHeader.SectionAlignment = td[i].section_alignment;
251 nt_header.OptionalHeader.FileAlignment = td[i].file_alignment;
252 nt_header.OptionalHeader.SizeOfImage = td[i].size_of_image;
253 nt_header.OptionalHeader.SizeOfHeaders = td[i].size_of_headers;
254 SetLastError(0xdeadbeef);
255 ok(WriteFile(hfile, &nt_header, sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER), &dummy, NULL),
256 "WriteFile error %d\n", GetLastError());
258 if (nt_header.FileHeader.SizeOfOptionalHeader)
260 SetLastError(0xdeadbeef);
261 ok(WriteFile(hfile, &nt_header.OptionalHeader,
262 min(nt_header.FileHeader.SizeOfOptionalHeader, sizeof(IMAGE_OPTIONAL_HEADER)),
264 "WriteFile error %d\n", GetLastError());
265 if (nt_header.FileHeader.SizeOfOptionalHeader > sizeof(IMAGE_OPTIONAL_HEADER))
267 file_align = nt_header.FileHeader.SizeOfOptionalHeader - sizeof(IMAGE_OPTIONAL_HEADER);
268 assert(file_align < sizeof(filler));
269 SetLastError(0xdeadbeef);
270 ok(WriteFile(hfile, filler, file_align, &dummy, NULL),
271 "WriteFile error %d\n", GetLastError());
275 assert(nt_header.FileHeader.NumberOfSections <= 1);
276 if (nt_header.FileHeader.NumberOfSections)
278 if (nt_header.OptionalHeader.SectionAlignment >= si.dwPageSize)
280 section.PointerToRawData = td[i].size_of_dos_header;
281 section.VirtualAddress = nt_header.OptionalHeader.SectionAlignment;
282 section.Misc.VirtualSize = section.SizeOfRawData * 10;
286 section.PointerToRawData = nt_header.OptionalHeader.SizeOfHeaders;
287 section.VirtualAddress = nt_header.OptionalHeader.SizeOfHeaders;
288 section.Misc.VirtualSize = 5;
291 SetLastError(0xdeadbeef);
292 ok(WriteFile(hfile, §ion, sizeof(section), &dummy, NULL),
293 "WriteFile error %d\n", GetLastError());
296 SetLastError(0xdeadbeef);
297 ok(WriteFile(hfile, section_data, sizeof(section_data), &dummy, NULL),
298 "WriteFile error %d\n", GetLastError());
301 file_size = GetFileSize(hfile, NULL);
304 SetLastError(0xdeadbeef);
305 hlib = LoadLibrary(dll_name);
306 if (td[i].error == ERROR_SUCCESS)
308 MEMORY_BASIC_INFORMATION info;
310 ok(hlib != 0, "%d: LoadLibrary error %d\n", i, GetLastError());
312 SetLastError(0xdeadbeef);
313 ok(VirtualQuery(hlib, &info, sizeof(info)) == sizeof(info),
314 "%d: VirtualQuery error %d\n", i, GetLastError());
315 ok(info.BaseAddress == hlib, "%d: %p != %p\n", i, info.BaseAddress, hlib);
316 ok(info.AllocationBase == hlib, "%d: %p != %p\n", i, info.AllocationBase, hlib);
317 ok(info.AllocationProtect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.AllocationProtect);
318 ok(info.RegionSize == ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize), "%d: got %lx != expected %x\n",
319 i, info.RegionSize, ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize));
320 ok(info.State == MEM_COMMIT, "%d: %x != MEM_COMMIT\n", i, info.State);
321 if (nt_header.OptionalHeader.SectionAlignment < si.dwPageSize)
322 ok(info.Protect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.Protect);
324 ok(info.Protect == PAGE_READONLY, "%d: %x != PAGE_READONLY\n", i, info.Protect);
325 ok(info.Type == SEC_IMAGE, "%d: %x != SEC_IMAGE\n", i, info.Type);
327 SetLastError(0xdeadbeef);
328 ok(VirtualQuery((char *)hlib + info.RegionSize, &info, sizeof(info)) == sizeof(info),
329 "%d: VirtualQuery error %d\n", i, GetLastError());
330 if (nt_header.OptionalHeader.SectionAlignment == si.dwPageSize ||
331 nt_header.OptionalHeader.SectionAlignment == nt_header.OptionalHeader.FileAlignment)
333 ok(info.BaseAddress == (char *)hlib + ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize), "%d: got %p != expected %p\n",
334 i, info.BaseAddress, (char *)hlib + ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize));
335 ok(info.AllocationBase == 0, "%d: %p != 0\n", i, info.AllocationBase);
336 ok(info.AllocationProtect == 0, "%d: %x != 0\n", i, info.AllocationProtect);
337 /*ok(info.RegionSize == not_practical_value, "%d: %lx != not_practical_value\n", i, info.RegionSize);*/
338 ok(info.State == MEM_FREE, "%d: %x != MEM_FREE\n", i, info.State);
339 ok(info.Type == 0, "%d: %x != 0\n", i, info.Type);
340 ok(info.Protect == PAGE_NOACCESS, "%d: %x != PAGE_NOACCESS\n", i, info.Protect);
344 ok(info.Protect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.Protect);
345 ok(info.BaseAddress == hlib, "%d: got %p != expected %p\n", i, info.BaseAddress, hlib);
346 ok(info.AllocationBase == hlib, "%d: %p != %p\n", i, info.AllocationBase, hlib);
347 ok(info.AllocationProtect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.AllocationProtect);
348 ok(info.RegionSize == ALIGN_SIZE(file_size, si.dwPageSize), "%d: got %lx != expected %x\n",
349 i, info.RegionSize, ALIGN_SIZE(file_size, si.dwPageSize));
350 ok(info.State == MEM_COMMIT, "%d: %x != MEM_COMMIT\n", i, info.State);
351 ok(info.Protect == PAGE_READONLY, "%d: %x != PAGE_READONLY\n", i, info.Protect);
352 ok(info.Type == SEC_IMAGE, "%d: %x != SEC_IMAGE\n", i, info.Type);
355 /* header: check the zeroing of alignment */
356 if (nt_header.OptionalHeader.SectionAlignment >= si.dwPageSize)
361 start = (const char *)hlib + nt_header.OptionalHeader.SizeOfHeaders;
362 size = ALIGN_SIZE((ULONG_PTR)start, si.dwPageSize) - (ULONG_PTR)start;
363 /* remove next 'if' and todo_wine once Wine is fixed */
364 if (nt_header.OptionalHeader.SizeOfHeaders < nt_header.OptionalHeader.FileAlignment)
366 todo_wine ok(!memcmp(start, filler, size), "%d: header alignment is not cleared\n", i);
369 ok(!memcmp(start, filler, size), "%d: header alignment is not cleared\n", i);
372 if (nt_header.FileHeader.NumberOfSections)
374 SetLastError(0xdeadbeef);
375 ok(VirtualQuery((char *)hlib + section.VirtualAddress, &info, sizeof(info)) == sizeof(info),
376 "%d: VirtualQuery error %d\n", i, GetLastError());
377 if (nt_header.OptionalHeader.SectionAlignment < si.dwPageSize)
379 ok(info.BaseAddress == (char *)hlib, "%d: got %p != expected %p\n", i, info.BaseAddress, (char *)hlib);
380 ok(info.RegionSize == ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize), "%d: got %lx != expected %x\n",
381 i, info.RegionSize, ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize));
382 ok(info.Protect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.Protect);
386 ok(info.BaseAddress == (char *)hlib + section.VirtualAddress, "%d: got %p != expected %p\n", i, info.BaseAddress, (char *)hlib + section.VirtualAddress);
387 ok(info.RegionSize == ALIGN_SIZE(section.Misc.VirtualSize, si.dwPageSize), "%d: got %lx != expected %x\n",
388 i, info.RegionSize, ALIGN_SIZE(section.Misc.VirtualSize, si.dwPageSize));
389 ok(info.Protect == PAGE_READONLY, "%d: %x != PAGE_READONLY\n", i, info.Protect);
391 ok(info.AllocationBase == hlib, "%d: %p != %p\n", i, info.AllocationBase, hlib);
392 ok(info.AllocationProtect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.AllocationProtect);
393 ok(info.State == MEM_COMMIT, "%d: %x != MEM_COMMIT\n", i, info.State);
394 ok(info.Type == SEC_IMAGE, "%d: %x != SEC_IMAGE\n", i, info.Type);
396 if (nt_header.OptionalHeader.SectionAlignment >= si.dwPageSize)
398 ok(!memcmp((const char *)hlib + section.VirtualAddress + section.PointerToRawData, &nt_header, section.SizeOfRawData), "wrong section data\n");
401 ok(!memcmp((const char *)hlib + section.PointerToRawData, section_data, section.SizeOfRawData), "wrong section data\n");
403 /* check the zeroing of alignment */
404 if (nt_header.OptionalHeader.SectionAlignment >= si.dwPageSize)
409 start = (const char *)hlib + section.VirtualAddress + section.PointerToRawData + section.SizeOfRawData;
410 size = ALIGN_SIZE((ULONG_PTR)start, si.dwPageSize) - (ULONG_PTR)start;
412 ok(memcmp(start, filler, size), "%d: alignment should not be not cleared\n", i);
417 SetLastError(0xdeadbeef);
418 ok(FreeLibrary(hlib), "FreeLibrary error %d\n", GetLastError());
421 { /* LoadLibrary has failed */
422 ok(!hlib, "%d: LoadLibrary should fail\n", i);
424 if (GetLastError() == ERROR_GEN_FAILURE) /* Win9x, broken behaviour */
426 trace("skipping the loader test on Win9x\n");
427 DeleteFile(dll_name);
431 ok(td[i].error == GetLastError(), "%d: expected error %d, got %d\n",
432 i, td[i].error, GetLastError());
435 SetLastError(0xdeadbeef);
436 ok(DeleteFile(dll_name), "DeleteFile error %d\n", GetLastError());