[PEFILE_APITEST] Introduce tests that load ntoskrnl.exe and validate various section...

svn path=/trunk/; revision=69154

diff --git a/rostests/apitests/CMakeLists.txt b/rostests/apitests/CMakeLists.txt
index 7490b33..5423e69 100644 (file)
--- a/rostests/apitests/CMakeLists.txt
+++ b/rostests/apitests/CMakeLists.txt
@@ -15,6 +15,7 @@ endif()
 add_subdirectory(msvcrt)
 add_subdirectory(ntdll)
 add_subdirectory(ole32)
+add_subdirectory(pefile)
 add_subdirectory(powrprof)
 add_subdirectory(setupapi)
 add_subdirectory(shell32)

diff --git a/rostests/apitests/pefile/CMakeLists.txt b/rostests/apitests/pefile/CMakeLists.txt
new file mode 100644 (file)
index 0000000..44297fa
--- /dev/null
+++ b/rostests/apitests/pefile/CMakeLists.txt
@@ -0,0 +1,5 @@
+
+add_executable(pefile_apitest ntoskrnl.c testlist.c)
+set_module_type(pefile_apitest win32cui)
+add_importlibs(pefile_apitest msvcrt kernel32)
+add_cd_file(TARGET pefile_apitest DESTINATION reactos/bin FOR all)

diff --git a/rostests/apitests/pefile/ntoskrnl.c b/rostests/apitests/pefile/ntoskrnl.c
new file mode 100644 (file)
index 0000000..41186dc
--- /dev/null
+++ b/rostests/apitests/pefile/ntoskrnl.c
@@ -0,0 +1,171 @@
+/*
+ * PROJECT:         ReactOS api tests
+ * LICENSE:         GPLv2+ - See COPYING in the top level directory
+ * PURPOSE:         Test to validate section flags in ntoskrnl
+ * PROGRAMMER:      Mark Jansen
+ */
+
+#include <apitest.h>
+#include <strsafe.h>
+
+typedef struct KnownSections
+{
+    const char* Name;
+    DWORD Required;
+    DWORD Disallowed;
+} KnownSections;
+
+static struct KnownSections g_Sections[] = {
+    {
+        ".text",
+        IMAGE_SCN_CNT_CODE | IMAGE_SCN_MEM_EXECUTE | IMAGE_SCN_MEM_READ,
+        IMAGE_SCN_MEM_DISCARDABLE
+        // optional: IMAGE_SCN_MEM_NOT_PAGED
+    },
+    {
+        ".data",
+        IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE,
+        IMAGE_SCN_MEM_DISCARDABLE
+        // optional: IMAGE_SCN_MEM_NOT_PAGED
+    },
+    {
+        ".rsrc",
+        IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ,
+        IMAGE_SCN_MEM_DISCARDABLE
+    },
+    {
+        ".reloc",
+        IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_DISCARDABLE | IMAGE_SCN_MEM_READ,
+        0
+    },
+    {
+        "INIT",
+        IMAGE_SCN_CNT_CODE | IMAGE_SCN_MEM_DISCARDABLE | IMAGE_SCN_MEM_EXECUTE | IMAGE_SCN_MEM_READ,
+        0
+        // optional?? : IMAGE_SCN_MEM_WRITE
+    },
+    { NULL, 0 },
+};
+
+static char* Chr2Str(DWORD value)
+{
+    static char buf[512];
+    buf[0] = '\0';
+#define IFX(x)      if( value & IMAGE_SCN_##x )\
+                    {\
+                        if(buf[0]) { StringCchCatA(buf, _countof(buf), "|" ); }\
+                        StringCchCatA(buf, _countof(buf), #x );\
+                        value &= ~(IMAGE_SCN_##x);\
+                    }
+    IFX(TYPE_NO_PAD);
+    IFX(CNT_CODE);
+    IFX(CNT_INITIALIZED_DATA);
+    IFX(CNT_UNINITIALIZED_DATA);
+    IFX(LNK_OTHER);
+    IFX(LNK_INFO);
+    IFX(LNK_REMOVE);
+    IFX(LNK_COMDAT);
+    //IFX(NO_DEFER_SPEC_EXC);
+    //IFX(GPREL);
+    IFX(MEM_FARDATA);
+    IFX(MEM_PURGEABLE);
+    IFX(MEM_16BIT);
+    IFX(MEM_LOCKED);
+    IFX(MEM_PRELOAD);
+    IFX(ALIGN_1BYTES);
+    IFX(ALIGN_2BYTES);
+    IFX(ALIGN_4BYTES);
+    IFX(ALIGN_8BYTES);
+    IFX(ALIGN_16BYTES);
+    IFX(ALIGN_32BYTES);
+    IFX(ALIGN_64BYTES);
+    //IFX(ALIGN_128BYTES);
+    //IFX(ALIGN_256BYTES);
+    //IFX(ALIGN_512BYTES);
+    //IFX(ALIGN_1024BYTES);
+    //IFX(ALIGN_2048BYTES);
+    //IFX(ALIGN_4096BYTES);
+    //IFX(ALIGN_8192BYTES);
+    IFX(LNK_NRELOC_OVFL);
+    IFX(MEM_DISCARDABLE);
+    IFX(MEM_NOT_CACHED);
+    IFX(MEM_NOT_PAGED);
+    IFX(MEM_SHARED);
+    IFX(MEM_EXECUTE);
+    IFX(MEM_READ);
+    IFX(MEM_WRITE);
+    if( value )
+    {
+        StringCchPrintfA(buf + strlen(buf), _countof(buf) - strlen(buf), "|0x%x", value);
+    }
+    return buf;
+}
+
+
+START_TEST(ntoskrnl_SectionFlags)
+{
+    char buf[MAX_PATH];
+    HMODULE mod;
+    GetSystemDirectoryA(buf, _countof(buf));
+    StringCchCatA(buf, _countof(buf), "\\ntoskrnl.exe");
+
+    mod = LoadLibraryExA( buf, NULL, LOAD_LIBRARY_AS_DATAFILE );
+    if( mod != NULL )
+    {
+        // we have to take into account that a datafile is not returned at the exact address it's loaded at.
+        PIMAGE_DOS_HEADER dos = (PIMAGE_DOS_HEADER)(((size_t)mod) & (~0xf));
+        PIMAGE_NT_HEADERS nt;
+        PIMAGE_SECTION_HEADER firstSection;
+        WORD numSections, n;
+
+        if( dos->e_magic != IMAGE_DOS_SIGNATURE )
+        {
+            skip("Couldn't find ntoskrnl.exe dos header\n");
+            FreeLibrary(mod);
+            return;
+        }
+        nt = (PIMAGE_NT_HEADERS)( ((PBYTE)dos) + dos->e_lfanew );
+        if( nt->Signature != IMAGE_NT_SIGNATURE )
+        {
+            skip("Couldn't find ntoskrnl.exe nt header\n");
+            FreeLibrary(mod);
+            return;
+        }
+        firstSection = IMAGE_FIRST_SECTION(nt);
+        numSections = nt->FileHeader.NumberOfSections;
+        for( n = 0; n < numSections; ++n )
+        {
+            PIMAGE_SECTION_HEADER section = firstSection + n;
+            char name[9] = {0};
+            size_t i;
+            StringCchCopyNA(name, _countof(name), (PCSTR)section->Name, 8);
+
+            for( i = 0; g_Sections[i].Name; ++i )
+            {
+                if( !_strnicmp(name, g_Sections[i].Name, 8) )
+                {
+                    if( g_Sections[i].Required )
+                    {
+                        DWORD Flags = g_Sections[i].Required & section->Characteristics;
+                        ok(Flags == g_Sections[i].Required,
+                            "Missing required Characteristics on %s: %s\n",
+                            name, Chr2Str(Flags ^ g_Sections[i].Required));
+                    }
+                    if( g_Sections[i].Disallowed )
+                    {
+                        DWORD Flags = g_Sections[i].Disallowed & section->Characteristics;
+                        ok(!Flags, "Disallowed section Characteristics on %s: %s\n",
+                            name, Chr2Str(section->Characteristics));
+                    }
+                    break;
+                }
+            }
+        }
+        FreeLibrary(mod);
+    }
+    else
+    {
+        skip("Couldn't load ntoskrnl.exe as datafile\n");
+    }
+}
+

diff --git a/rostests/apitests/pefile/testlist.c b/rostests/apitests/pefile/testlist.c
new file mode 100644 (file)
index 0000000..e71bafe
--- /dev/null
+++ b/rostests/apitests/pefile/testlist.c
@@ -0,0 +1,14 @@
+#define __ROS_LONG64__
+
+#define STANDALONE
+#include <wine/test.h>
+
+extern void func_ntoskrnl_SectionFlags(void);
+
+const struct test winetest_testlist[] =
+{
+    { "ntoskrnl_SectionFlags",                  func_ntoskrnl_SectionFlags },
+
+    { 0, 0 }
+};
+