git.oblomov.eu Git - wine/blob - dlls/kernel32/tests/loader.c

   1 /*
   2  * Unit test suite for the PE loader.
   3  *
   4  * Copyright 2006 Dmitry Timoshkov
   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 <stdarg.h>
  22 #include <assert.h>
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "wine/test.h"
  27
  28 #define ALIGN_SIZE(size, alignment) (((size) + (alignment - 1)) & ~((alignment - 1)))
  29
  30 static const struct
  31 {
  32     WORD e_magic;      /* 00: MZ Header signature */
  33     WORD unused[29];
  34     DWORD e_lfanew;    /* 3c: Offset to extended header */
  35 } dos_header =
  36 {
  37     IMAGE_DOS_SIGNATURE, { 0 }, sizeof(dos_header)
  38 };
  39
  40 static IMAGE_NT_HEADERS nt_header =
  41 {
  42     IMAGE_NT_SIGNATURE, /* Signature */
  43 #ifdef __i386__
  44     { IMAGE_FILE_MACHINE_I386, /* Machine */
  45 #else
  46 # error You must specify the machine type
  47 #endif
  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 */
  54     },
  55     { IMAGE_NT_OPTIONAL_HDR_MAGIC, /* Magic */
  56       1, /* MajorLinkerVersion */
  57       0, /* MinorLinkerVersion */
  58       0, /* SizeOfCode */
  59       0, /* SizeOfInitializedData */
  60       0, /* SizeOfUninitializedData */
  61       0, /* AddressOfEntryPoint */
  62       0, /* BaseOfCode */
  63       0, /* BaseOfData */
  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 */
  76       0, /* CheckSum */
  77       IMAGE_SUBSYSTEM_WINDOWS_CUI, /* Subsystem */
  78       0, /* DllCharacteristics */
  79       0, /* SizeOfStackReserve */
  80       0, /* SizeOfStackCommit */
  81       0, /* SizeOfHeapReserve */
  82       0, /* SizeOfHeapCommit */
  83       0, /* LoaderFlags */
  84       0, /* NumberOfRvaAndSizes */
  85       { { 0 } } /* DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES] */
  86     }
  87 };
  88
  89 static IMAGE_SECTION_HEADER section =
  90 {
  91     ".rodata", /* Name */
  92     { 0x10 }, /* Misc */
  93     0, /* VirtualAddress */
  94     0x10, /* 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 */
 101 };
 102
 103 START_TEST(loader)
 104 {
 105     static const struct test_data
 106     {
 107         WORD size_of_optional_header;
 108         DWORD section_alignment, file_alignment;
 109         DWORD size_of_image, size_of_headers;
 110         DWORD error; /* 0 means LoadLibrary should succeed */
 111     } td[] =
 112     {
 113         { 0, 0, 0, 0, 0,
 114           ERROR_BAD_EXE_FORMAT
 115         },
 116         { sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
 117           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0xe00,
 118           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
 119           ERROR_BAD_EXE_FORMAT /* XP doesn't like too small image size */
 120         },
 121         { sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
 122           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
 123           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
 124           ERROR_SUCCESS
 125         },
 126         { sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x1000,
 127           0x1f00,
 128           0x1000,
 129           ERROR_SUCCESS
 130         },
 131         { sizeof(IMAGE_OPTIONAL_HEADER), 0x200, 0x200,
 132           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
 133           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
 134           ERROR_SUCCESS
 135         },
 136         { sizeof(IMAGE_OPTIONAL_HEADER), 0x200, 0x1000,
 137           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
 138           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
 139           ERROR_BAD_EXE_FORMAT /* XP doesn't like aligments */
 140         },
 141         { sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x200,
 142           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
 143           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER),
 144           ERROR_SUCCESS
 145         },
 146         { sizeof(IMAGE_OPTIONAL_HEADER), 0x1000, 0x200,
 147           sizeof(dos_header) + sizeof(nt_header) + sizeof(IMAGE_SECTION_HEADER) + 0x1000,
 148           0x200,
 149           ERROR_SUCCESS
 150         },
 151         /* Mandatory are all fields up to SizeOfHeaders, everything else
 152          * is really optional (at least that's true for XP).
 153          */
 154         { FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum), 0x200, 0x200,
 155           sizeof(dos_header) + sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER) + FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum) + sizeof(IMAGE_SECTION_HEADER) + 0x10,
 156           sizeof(dos_header) + sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER) + FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, CheckSum) + sizeof(IMAGE_SECTION_HEADER),
 157           ERROR_SUCCESS
 158         }
 159     };
 160     static const char filler[0x1000];
 161     int i;
 162     DWORD dummy, file_size, file_align;
 163     HANDLE hfile, hlib;
 164     SYSTEM_INFO si;
 165     char temp_path[MAX_PATH];
 166     char dll_name[MAX_PATH];
 167
 168     GetSystemInfo(&si);
 169     trace("system page size 0x%04x\n", si.dwPageSize);
 170
 171     /* prevent displaying of the "Unable to load this DLL" message box */
 172     SetErrorMode(SEM_FAILCRITICALERRORS);
 173
 174     GetTempPath(MAX_PATH, temp_path);
 175     GetTempFileName(temp_path, "ldr", 0, dll_name);
 176
 177     for (i = 0; i < sizeof(td)/sizeof(td[0]); i++)
 178     {
 179         /*trace("creating %s\n", dll_name);*/
 180         hfile = CreateFileA(dll_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0);
 181         if (hfile == INVALID_HANDLE_VALUE)
 182         {
 183             ok(0, "could not create %s\n", dll_name);
 184             break;
 185         }
 186
 187         SetLastError(0xdeadbeef);
 188         ok(WriteFile(hfile, &dos_header, sizeof(dos_header), &dummy, NULL),
 189            "WriteFile error %d\n", GetLastError());
 190
 191         nt_header.FileHeader.SizeOfOptionalHeader = td[i].size_of_optional_header;
 192
 193         nt_header.OptionalHeader.SectionAlignment = td[i].section_alignment;
 194         nt_header.OptionalHeader.FileAlignment = td[i].file_alignment;
 195         nt_header.OptionalHeader.SizeOfImage = td[i].size_of_image;
 196         nt_header.OptionalHeader.SizeOfHeaders = td[i].size_of_headers;
 197         SetLastError(0xdeadbeef);
 198         ok(WriteFile(hfile, &nt_header, sizeof(DWORD) + sizeof(IMAGE_FILE_HEADER), &dummy, NULL),
 199            "WriteFile error %d\n", GetLastError());
 200
 201         if (nt_header.FileHeader.SizeOfOptionalHeader)
 202         {
 203             assert(nt_header.FileHeader.SizeOfOptionalHeader <= sizeof(IMAGE_OPTIONAL_HEADER));
 204
 205             SetLastError(0xdeadbeef);
 206             ok(WriteFile(hfile, &nt_header.OptionalHeader, nt_header.FileHeader.SizeOfOptionalHeader, &dummy, NULL),
 207                "WriteFile error %d\n", GetLastError());
 208         }
 209
 210         if (nt_header.OptionalHeader.SectionAlignment == si.dwPageSize)
 211         {
 212             section.VirtualAddress = nt_header.OptionalHeader.SectionAlignment;
 213             section.PointerToRawData = nt_header.OptionalHeader.FileAlignment;
 214             SetLastError(0xdeadbeef);
 215             ok(WriteFile(hfile, &section, sizeof(section), &dummy, NULL),
 216                "WriteFile error %d\n", GetLastError());
 217
 218             file_size = GetFileSize(hfile, NULL);
 219
 220             file_align = ALIGN_SIZE(file_size, nt_header.OptionalHeader.FileAlignment) - file_size;
 221             SetLastError(0xdeadbeef);
 222             ok(WriteFile(hfile, filler, file_align, &dummy, NULL),
 223                "WriteFile error %d\n", GetLastError());
 224         }
 225         else
 226         {
 227             section.VirtualAddress = nt_header.OptionalHeader.SizeOfHeaders;
 228             section.PointerToRawData = nt_header.OptionalHeader.SizeOfHeaders;
 229             SetLastError(0xdeadbeef);
 230             ok(WriteFile(hfile, &section, sizeof(section), &dummy, NULL),
 231                "WriteFile error %d\n", GetLastError());
 232         }
 233
 234         /* section data */
 235         SetLastError(0xdeadbeef);
 236         ok(WriteFile(hfile, filler, 0x10, &dummy, NULL),
 237            "WriteFile error %d\n", GetLastError());
 238
 239         CloseHandle(hfile);
 240
 241         SetLastError(0xdeadbeef);
 242         hlib = LoadLibrary(dll_name);
 243         if (td[i].error == ERROR_SUCCESS)
 244         {
 245             MEMORY_BASIC_INFORMATION info;
 246
 247             ok(hlib != 0, "%d: LoadLibrary error %d\n", i, GetLastError());
 248
 249             SetLastError(0xdeadbeef);
 250             ok(VirtualQuery(hlib, &info, sizeof(info)) == sizeof(info),
 251                 "%d: VirtualQuery error %d\n", i, GetLastError());
 252             ok(info.BaseAddress == hlib, "%d: %p != %p\n", i, info.BaseAddress, hlib);
 253             ok(info.AllocationBase == hlib, "%d: %p != %p\n", i, info.AllocationBase, hlib);
 254             ok(info.AllocationProtect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.AllocationProtect);
 255             ok(info.RegionSize == ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize), "%d: got %lx != expected %x\n",
 256                i, info.RegionSize, ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize));
 257             ok(info.State == MEM_COMMIT, "%d: %x != MEM_COMMIT\n", i, info.State);
 258             if (nt_header.OptionalHeader.SectionAlignment != si.dwPageSize)
 259                 ok(info.Protect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.Protect);
 260             else
 261                 ok(info.Protect == PAGE_READONLY, "%d: %x != PAGE_READONLY\n", i, info.Protect);
 262             ok(info.Type == SEC_IMAGE, "%d: %x != SEC_IMAGE\n", i, info.Type);
 263
 264             SetLastError(0xdeadbeef);
 265             ok(VirtualQuery((char *)hlib + info.RegionSize, &info, sizeof(info)) == sizeof(info),
 266                 "%d: VirtualQuery error %d\n", i, GetLastError());
 267             if (nt_header.OptionalHeader.SectionAlignment == si.dwPageSize ||
 268                 nt_header.OptionalHeader.SectionAlignment == nt_header.OptionalHeader.FileAlignment)
 269             {
 270                 ok(info.BaseAddress == (char *)hlib + ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize), "%d: %p != %p\n",
 271                    i, info.BaseAddress, (char *)hlib + ALIGN_SIZE(nt_header.OptionalHeader.SizeOfImage, si.dwPageSize));
 272 todo_wine {
 273                 ok(info.AllocationBase == 0, "%d: %p != 0\n", i, info.AllocationBase);
 274 }
 275                 ok(info.AllocationProtect == 0, "%d: %x != 0\n", i, info.AllocationProtect);
 276                 /*ok(info.RegionSize == not_practical_value, "%d: %lx != not_practical_value\n", i, info.RegionSize);*/
 277                 ok(info.State == MEM_FREE, "%d: %x != MEM_FREE\n", i, info.State);
 278                 ok(info.Type == 0, "%d: %x != 0\n", i, info.Type);
 279 todo_wine {
 280                 ok(info.Protect == PAGE_NOACCESS, "%d: %x != PAGE_NOACCESS\n", i, info.Protect);
 281 }
 282             }
 283             else
 284             {
 285                 ok(info.Protect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.Protect);
 286                 ok(info.BaseAddress == hlib, "%d: %p != %p\n", i, info.BaseAddress, hlib);
 287                 ok(info.AllocationBase == hlib, "%d: %p != %p\n", i, info.AllocationBase, hlib);
 288                 ok(info.AllocationProtect == PAGE_EXECUTE_WRITECOPY, "%d: %x != PAGE_EXECUTE_WRITECOPY\n", i, info.AllocationProtect);
 289                 ok(info.RegionSize == 0x2000, "%d: %lx != 0x2000\n", i, info.RegionSize);
 290                 ok(info.State == MEM_COMMIT, "%d: %x != MEM_COMMIT\n", i, info.State);
 291                 ok(info.Protect == PAGE_READONLY, "%d: %x != PAGE_READONLY\n", i, info.Protect);
 292                 ok(info.Type == SEC_IMAGE, "%d: %x != SEC_IMAGE\n", i, info.Type);
 293             }
 294
 295             SetLastError(0xdeadbeef);
 296             ok(FreeLibrary(hlib), "FreeLibrary error %d\n", GetLastError());
 297         }
 298         else
 299         {
 300             if (hlib) /* remove completely once Wine is fixed */
 301             {
 302                 todo_wine ok(!hlib, "%d: LoadLibrary should fail\n", i);
 303                 FreeLibrary(hlib);
 304                 DeleteFile(dll_name);
 305                 continue;
 306             }
 307
 308             ok(!hlib, "%d: LoadLibrary should fail\n", i);
 309
 310             ok(td[i].error == GetLastError(), "%d: expected error %d, got %d\n",
 311                i, td[i].error, GetLastError());
 312         }
 313
 314         SetLastError(0xdeadbeef);
 315         ok(DeleteFile(dll_name), "DeleteFile error %d\n", GetLastError());
 316     }
 317 }