mirror of
https://github.com/reactos/reactos.git
synced 2024-11-18 13:01:40 +00:00
167 lines
5 KiB
C
167 lines
5 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* PURPOSE: Test for NtContinue
|
|
* PROGRAMMER:
|
|
*/
|
|
|
|
#include <apitest.h>
|
|
#include <ndk/kefuncs.h>
|
|
|
|
#include <setjmp.h>
|
|
#include <time.h>
|
|
|
|
#ifdef _M_IX86
|
|
#define NTC_SEGMENT_BITS (0xFFFF)
|
|
#define NTC_EFLAGS_BITS (0x3C0CD5)
|
|
#endif
|
|
|
|
void continuePoint(void);
|
|
|
|
static jmp_buf jmpbuf;
|
|
static CONTEXT continueContext;
|
|
static unsigned int nRandBytes;
|
|
|
|
static int initrand(void)
|
|
{
|
|
unsigned int nRandMax;
|
|
unsigned int nRandMaxBits;
|
|
time_t tLoc;
|
|
|
|
nRandMax = RAND_MAX;
|
|
for(nRandMaxBits = 0; nRandMax != 0; nRandMax >>= 1, ++ nRandMaxBits);
|
|
nRandBytes = nRandMaxBits / CHAR_BIT;
|
|
//assert(nRandBytes != 0);
|
|
srand((unsigned)(time(&tLoc) & UINT_MAX));
|
|
return 1;
|
|
}
|
|
|
|
static void randbytes(void * p, size_t n)
|
|
{
|
|
unsigned char * b;
|
|
size_t i;
|
|
int r = rand();
|
|
|
|
b = (unsigned char *)p;
|
|
for(i = 0; i < n; ++ i)
|
|
{
|
|
if(i % nRandBytes == 0)
|
|
r = rand();
|
|
b[i] = (unsigned char)(r & UCHAR_MAX);
|
|
r >>= CHAR_BIT;
|
|
}
|
|
}
|
|
|
|
static ULONG randULONG(void)
|
|
{
|
|
ULONG n;
|
|
randbytes(&n, sizeof(n));
|
|
return n;
|
|
}
|
|
|
|
void check(CONTEXT * pContext)
|
|
{
|
|
#ifdef _M_IX86
|
|
ok(pContext->ContextFlags == CONTEXT_FULL,
|
|
"ContextFlags=0x%lx\n", pContext->ContextFlags);
|
|
|
|
/* Random data segments */
|
|
ok((pContext->SegGs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegGs & NTC_SEGMENT_BITS),
|
|
"SegGs=0x%lx / 0x%lx\n", pContext->SegGs, continueContext.SegGs);
|
|
|
|
ok((pContext->SegFs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegFs & NTC_SEGMENT_BITS),
|
|
"SegFs=0x%lx / 0x%lx\n", pContext->SegFs, continueContext.SegFs);
|
|
|
|
ok((pContext->SegEs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegEs & NTC_SEGMENT_BITS),
|
|
"SegEs=0x%lx / 0x%lx\n", pContext->SegEs, continueContext.SegEs);
|
|
|
|
ok((pContext->SegDs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegDs & NTC_SEGMENT_BITS),
|
|
"SegDs=0x%lx / 0x%lx\n", pContext->SegDs, continueContext.SegDs);
|
|
|
|
/* Integer registers */
|
|
ok(pContext->Edi == continueContext.Edi,
|
|
"Edi: 0x%lx != 0x%lx\n", pContext->Edi, continueContext.Edi);
|
|
ok(pContext->Esi == continueContext.Esi,
|
|
"Esi: 0x%lx != 0x%lx\n", pContext->Esi, continueContext.Esi);
|
|
ok(pContext->Ebx == continueContext.Ebx,
|
|
"Ebx: 0x%lx != 0x%lx\n", pContext->Ebx, continueContext.Ebx);
|
|
ok(pContext->Edx == continueContext.Edx,
|
|
"Edx: 0x%lx != 0x%lx\n", pContext->Edx, continueContext.Edx);
|
|
ok(pContext->Ecx == continueContext.Ecx,
|
|
"Ecx: 0x%lx != 0x%lx\n", pContext->Ecx, continueContext.Ecx);
|
|
ok(pContext->Eax == continueContext.Eax,
|
|
"Eax: 0x%lx != 0x%lx\n", pContext->Eax, continueContext.Eax);
|
|
|
|
/* Control registers and segments */
|
|
ok(pContext->Ebp == continueContext.Ebp,
|
|
"Ebp: 0x%lx != 0x%lx\n", pContext->Ebp, continueContext.Ebp);
|
|
ok(pContext->Eip == continueContext.Eip,
|
|
"Eip: 0x%lx != 0x%lx\n", pContext->Eip, continueContext.Eip);
|
|
ok(pContext->Esp == continueContext.Esp,
|
|
"Esp: 0x%lx != 0x%lx\n", pContext->Esp, continueContext.Esp);
|
|
|
|
ok((pContext->SegCs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegCs & NTC_SEGMENT_BITS),
|
|
"SegCs: 0x%lx != 0x%lx\n", pContext->SegCs, continueContext.SegCs);
|
|
|
|
ok((pContext->EFlags & NTC_EFLAGS_BITS) ==
|
|
(continueContext.EFlags & NTC_EFLAGS_BITS),
|
|
"EFlags: 0x%lx != 0x%lx\n", pContext->EFlags, continueContext.EFlags);
|
|
|
|
ok((pContext->SegSs & NTC_SEGMENT_BITS) ==
|
|
(continueContext.SegSs & NTC_SEGMENT_BITS),
|
|
"SegSs: 0x%lx != 0x%lx\n", pContext->SegSs, continueContext.SegSs);
|
|
#endif
|
|
|
|
/* Return where we came from */
|
|
longjmp(jmpbuf, 1);
|
|
}
|
|
|
|
START_TEST(NtContinue)
|
|
{
|
|
initrand();
|
|
|
|
/* First time */
|
|
if(setjmp(jmpbuf) == 0)
|
|
{
|
|
CONTEXT bogus;
|
|
|
|
continueContext.ContextFlags = CONTEXT_FULL;
|
|
GetThreadContext(GetCurrentThread(), &continueContext);
|
|
|
|
#ifdef _M_IX86
|
|
continueContext.ContextFlags = CONTEXT_FULL;
|
|
|
|
/* Fill the integer registers with random values */
|
|
continueContext.Edi = randULONG();
|
|
continueContext.Esi = randULONG();
|
|
continueContext.Ebx = randULONG();
|
|
continueContext.Edx = randULONG();
|
|
continueContext.Ecx = randULONG();
|
|
continueContext.Eax = randULONG();
|
|
continueContext.Ebp = randULONG();
|
|
|
|
/* Randomize all the allowed flags (determined experimentally with WinDbg) */
|
|
continueContext.EFlags = randULONG() & 0x3C0CD5;
|
|
|
|
/* Randomize the stack pointer as much as possible */
|
|
continueContext.Esp = (ULONG)(((ULONG_PTR)&bogus) & 0xFFFFFFFF) +
|
|
sizeof(bogus) - (randULONG() & 0xF) * 4;
|
|
|
|
/* continuePoint() is implemented in assembler */
|
|
continueContext.Eip = (ULONG)((ULONG_PTR)continuePoint & 0xFFFFFFF);
|
|
|
|
/* Can't do a lot about segments */
|
|
#endif
|
|
|
|
NtContinue(&continueContext, FALSE);
|
|
ok(0, "should never get here\n");
|
|
}
|
|
|
|
/* Second time */
|
|
return;
|
|
}
|