mirror of
https://github.com/reactos/reactos.git
synced 2024-08-21 10:53:15 +00:00
174 lines
5 KiB
C
174 lines
5 KiB
C
/*
|
|
* 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_NOT_PAGED,
|
|
IMAGE_SCN_MEM_DISCARDABLE
|
|
},
|
|
{
|
|
".data",
|
|
IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE | IMAGE_SCN_MEM_NOT_PAGED,
|
|
IMAGE_SCN_MEM_DISCARDABLE
|
|
},
|
|
{
|
|
".rsrc",
|
|
IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ,
|
|
IMAGE_SCN_MEM_DISCARDABLE
|
|
},
|
|
{
|
|
".rdata",
|
|
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
|
|
},
|
|
{ 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");
|
|
}
|
|
}
|
|
|