ReactOS 0.4.16-dev-1044-ga6558c0
Functions | Variables
NtContinue.c File Reference
#include "precomp.h"
#include <setjmp.h>
#include <time.h>
Include dependency graph for NtContinue.c:

Go to the source code of this file.

Functions

void continuePoint (void)
 
static int initrand (void)
 
static void randbytes (void *p, size_t n)
 
static ULONG randULONG (void)
 
void check (CONTEXT *pContext)
 
 START_TEST (NtContinue)
 

Variables

static jmp_buf jmpbuf
 
static CONTEXT continueContext
 
static unsigned int nRandBytes
 

Function Documentation

◆ check()

void check ( CONTEXT *  pContext)

Definition at line 72 of file NtContinue.c.

73{
74#ifdef _M_IX86
75 ok(pContext->ContextFlags == CONTEXT_FULL,
76 "ContextFlags=0x%lx\n", pContext->ContextFlags);
77
78 /* Random data segments */
79 ok((pContext->SegGs & NTC_SEGMENT_BITS) ==
80 (continueContext.SegGs & NTC_SEGMENT_BITS),
81 "SegGs=0x%lx / 0x%lx\n", pContext->SegGs, continueContext.SegGs);
82
83 ok((pContext->SegFs & NTC_SEGMENT_BITS) ==
84 (continueContext.SegFs & NTC_SEGMENT_BITS),
85 "SegFs=0x%lx / 0x%lx\n", pContext->SegFs, continueContext.SegFs);
86
87 ok((pContext->SegEs & NTC_SEGMENT_BITS) ==
88 (continueContext.SegEs & NTC_SEGMENT_BITS),
89 "SegEs=0x%lx / 0x%lx\n", pContext->SegEs, continueContext.SegEs);
90
91 ok((pContext->SegDs & NTC_SEGMENT_BITS) ==
92 (continueContext.SegDs & NTC_SEGMENT_BITS),
93 "SegDs=0x%lx / 0x%lx\n", pContext->SegDs, continueContext.SegDs);
94
95 /* Integer registers */
96 ok(pContext->Edi == continueContext.Edi,
97 "Edi: 0x%lx != 0x%lx\n", pContext->Edi, continueContext.Edi);
98 ok(pContext->Esi == continueContext.Esi,
99 "Esi: 0x%lx != 0x%lx\n", pContext->Esi, continueContext.Esi);
100 ok(pContext->Ebx == continueContext.Ebx,
101 "Ebx: 0x%lx != 0x%lx\n", pContext->Ebx, continueContext.Ebx);
102 ok(pContext->Edx == continueContext.Edx,
103 "Edx: 0x%lx != 0x%lx\n", pContext->Edx, continueContext.Edx);
104 ok(pContext->Ecx == continueContext.Ecx,
105 "Ecx: 0x%lx != 0x%lx\n", pContext->Ecx, continueContext.Ecx);
106 ok(pContext->Eax == continueContext.Eax,
107 "Eax: 0x%lx != 0x%lx\n", pContext->Eax, continueContext.Eax);
108
109 /* Control registers and segments */
110 ok(pContext->Ebp == continueContext.Ebp,
111 "Ebp: 0x%lx != 0x%lx\n", pContext->Ebp, continueContext.Ebp);
112 ok(pContext->Eip == continueContext.Eip,
113 "Eip: 0x%lx != 0x%lx\n", pContext->Eip, continueContext.Eip);
114 ok(pContext->Esp == continueContext.Esp,
115 "Esp: 0x%lx != 0x%lx\n", pContext->Esp, continueContext.Esp);
116
117 ok((pContext->SegCs & NTC_SEGMENT_BITS) ==
118 (continueContext.SegCs & NTC_SEGMENT_BITS),
119 "SegCs: 0x%lx != 0x%lx\n", pContext->SegCs, continueContext.SegCs);
120
121 ok((pContext->EFlags & NTC_EFLAGS_BITS) ==
122 (continueContext.EFlags & NTC_EFLAGS_BITS),
123 "EFlags: 0x%lx != 0x%lx\n", pContext->EFlags, continueContext.EFlags);
124
125 ok((pContext->SegSs & NTC_SEGMENT_BITS) ==
126 (continueContext.SegSs & NTC_SEGMENT_BITS),
127 "SegSs: 0x%lx != 0x%lx\n", pContext->SegSs, continueContext.SegSs);
128#else
129 ok_eq_hex64(pContext->ContextFlags, CONTEXT_FULL | CONTEXT_SEGMENTS);
130 ok_eq_hex(pContext->MxCsr, continueContext.MxCsr);
131 ok_eq_hex(pContext->SegCs, continueContext.SegCs);
132 ok_eq_hex(pContext->SegDs, 0x2B);
133 ok_eq_hex(pContext->SegEs, 0x2B);
134 ok_eq_hex(pContext->SegFs, 0x53);
135 ok_eq_hex(pContext->SegGs, 0x2B);
136 ok_eq_hex(pContext->SegSs, continueContext.SegSs);
137 ok_eq_hex(pContext->EFlags, (continueContext.EFlags & ~0x1C0000) | 0x202);
138
139 ok_eq_hex64(pContext->Rax, continueContext.Rax);
140 ok_eq_hex64(pContext->Rdx, continueContext.Rdx);
141 ok_eq_hex64(pContext->Rbx, continueContext.Rbx);
142 ok_eq_hex64(pContext->Rsp, continueContext.Rsp);
143 ok_eq_hex64(pContext->Rbp, continueContext.Rbp);
144 ok_eq_hex64(pContext->Rsi, continueContext.Rsi);
145 ok_eq_hex64(pContext->Rdi, continueContext.Rdi);
146 ok_eq_hex64(pContext->R8, continueContext.R8);
147 ok_eq_hex64(pContext->R9, continueContext.R9);
148 ok_eq_hex64(pContext->R10, continueContext.R10);
149 ok_eq_hex64(pContext->R11, continueContext.R11);
150 ok_eq_hex64(pContext->R12, continueContext.R12);
151 ok_eq_hex64(pContext->R13, continueContext.R13);
152 ok_eq_hex64(pContext->R14, continueContext.R14);
153 ok_eq_hex64(pContext->R15, continueContext.R15);
154 ok_eq_xmm(pContext->Xmm0, continueContext.Xmm0);
155 ok_eq_xmm(pContext->Xmm1, continueContext.Xmm1);
156 ok_eq_xmm(pContext->Xmm2, continueContext.Xmm2);
157 ok_eq_xmm(pContext->Xmm3, continueContext.Xmm3);
158 ok_eq_xmm(pContext->Xmm4, continueContext.Xmm4);
159 ok_eq_xmm(pContext->Xmm5, continueContext.Xmm5);
160 ok_eq_xmm(pContext->Xmm6, continueContext.Xmm6);
161 ok_eq_xmm(pContext->Xmm7, continueContext.Xmm7);
162 ok_eq_xmm(pContext->Xmm8, continueContext.Xmm8);
163 ok_eq_xmm(pContext->Xmm9, continueContext.Xmm9);
164 ok_eq_xmm(pContext->Xmm10, continueContext.Xmm10);
165 ok_eq_xmm(pContext->Xmm11, continueContext.Xmm11);
166 ok_eq_xmm(pContext->Xmm12, continueContext.Xmm12);
167 ok_eq_xmm(pContext->Xmm13, continueContext.Xmm13);
168 ok_eq_xmm(pContext->Xmm14, continueContext.Xmm14);
169 ok_eq_xmm(pContext->Xmm15, continueContext.Xmm15);
170
171 // Clear the frame register to prevent unwinding, which is broken
172 ((_JUMP_BUFFER*)&jmpbuf)->Frame = 0;
173#endif
174
175 /* Return where we came from */
176 longjmp(jmpbuf, 1);
177}
jmpbuf
static jmp_buf jmpbuf
Definition: NtContinue.c:24
continueContext
static CONTEXT continueContext
Definition: NtContinue.c:25
ok_eq_hex
#define ok_eq_hex(value, expected)
Definition: apitest.h:58
ok_eq_hex64
#define ok_eq_hex64(value, expected)
Definition: apitest.h:70
ok_eq_xmm
#define ok_eq_xmm(value, expected)
Definition: apitest.h:71
ok
#define ok(value,...)
Definition: atltest.h:57
CONTEXT_SEGMENTS
#define CONTEXT_SEGMENTS
Definition: nt_native.h:1371
CONTEXT_FULL
#define CONTEXT_FULL
Definition: nt_native.h:1375
_CONTEXT::Esp
ULONG Esp
Definition: nt_native.h:1479
_CONTEXT::SegFs
ULONG SegFs
Definition: nt_native.h:1454
_CONTEXT::Edx
ULONG Edx
Definition: nt_native.h:1466
_CONTEXT::Esi
ULONG Esi
Definition: nt_native.h:1464
_CONTEXT::Ebp
ULONG Ebp
Definition: nt_native.h:1475
_CONTEXT::ContextFlags
ULONG ContextFlags
Definition: nt_native.h:1426
_CONTEXT::R8
ULONG R8
Definition: ke.h:263
_CONTEXT::SegSs
ULONG SegSs
Definition: nt_native.h:1480
_CONTEXT::Ecx
ULONG Ecx
Definition: nt_native.h:1467
_CONTEXT::Eip
ULONG Eip
Definition: nt_native.h:1476
_CONTEXT::R12
ULONG R12
Definition: ke.h:267
_CONTEXT::SegCs
ULONG SegCs
Definition: nt_native.h:1477
_CONTEXT::SegDs
ULONG SegDs
Definition: nt_native.h:1456
_CONTEXT::R9
ULONG R9
Definition: ke.h:264
_CONTEXT::EFlags
ULONG EFlags
Definition: nt_native.h:1478
_CONTEXT::SegGs
ULONG SegGs
Definition: nt_native.h:1453
_CONTEXT::R10
ULONG R10
Definition: ke.h:265
_CONTEXT::Eax
ULONG Eax
Definition: nt_native.h:1468
_CONTEXT::SegEs
ULONG SegEs
Definition: nt_native.h:1455
_CONTEXT::Ebx
ULONG Ebx
Definition: nt_native.h:1465
_CONTEXT::Edi
ULONG Edi
Definition: nt_native.h:1463
_CONTEXT::R11
ULONG R11
Definition: ke.h:266

◆ continuePoint()

void continuePoint ( void  )

Referenced by START_TEST().

◆ initrand()

static int initrand ( void  )
static

Definition at line 28 of file NtContinue.c.

29{
30 unsigned int nRandMax;
31 unsigned int nRandMaxBits;
32 time_t tLoc;
33
34 nRandMax = RAND_MAX;
35 for(nRandMaxBits = 0; nRandMax != 0; nRandMax >>= 1, ++ nRandMaxBits);
36 nRandBytes = nRandMaxBits / CHAR_BIT;
37 //assert(nRandBytes != 0);
38 srand((unsigned)(time(&tLoc) & UINT_MAX));
39 return 1;
40}
nRandBytes
static unsigned int nRandBytes
Definition: NtContinue.c:26
CHAR_BIT
#define CHAR_BIT
Definition: urlcache.c:62
time_t
__kernel_time_t time_t
Definition: linux.h:252
srand
void __cdecl srand(_In_ unsigned int _Seed)
RAND_MAX
#define RAND_MAX
Definition: stdlib.h:87
UINT_MAX
#define UINT_MAX
Definition: intsafe.h:152
time
__u16 time
Definition: mkdosfs.c:8

Referenced by START_TEST().

◆ randbytes()

static void randbytes ( void *  p,
size_t  n 
)
static

Definition at line 42 of file NtContinue.c.

43{
44 unsigned char * b;
45 size_t i;
46 int r = rand();
47
48 b = (unsigned char *)p;
49 for(i = 0; i < n; ++ i)
50 {
51 if(i % nRandBytes == 0)
52 r = rand();
53 b[i] = (unsigned char)(r & UCHAR_MAX);
54 r >>= CHAR_BIT;
55 }
56}
char
unsigned char
Definition: typeof.h:29
r
GLdouble GLdouble GLdouble r
Definition: gl.h:2055
n
GLdouble n
Definition: glext.h:7729
b
GLboolean GLboolean GLboolean b
Definition: glext.h:6204
p
GLfloat GLfloat p
Definition: glext.h:8902
i
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble const GLfloat const GLdouble const GLfloat GLint i
Definition: glfuncs.h:248
rand
_Check_return_ int __cdecl rand(void)
Definition: rand.c:10
UCHAR_MAX
#define UCHAR_MAX
Definition: limits.h:25
b
#define b
Definition: ke_i.h:79

Referenced by randULONG().

◆ randULONG()

static ULONG randULONG ( void  )
static

Definition at line 58 of file NtContinue.c.

59{
60 ULONG n;
61 randbytes(&n, sizeof(n));
62 return n;
63}
randbytes
static void randbytes(void *p, size_t n)
Definition: NtContinue.c:42
ULONG
uint32_t ULONG
Definition: typedefs.h:59

Referenced by START_TEST().

◆ START_TEST()

START_TEST ( NtContinue  )

Definition at line 179 of file NtContinue.c.

180{
181 initrand();
182
183 RtlFillMemory(&continueContext, sizeof(continueContext), 0xBBBBBBBB);
184
185 /* First time */
186 if(setjmp(jmpbuf) == 0)
187 {
188 CONTEXT bogus[2];
189
190 RtlFillMemory(&bogus, sizeof(bogus), 0xCCCCCCCC);
191
192 continueContext.ContextFlags = CONTEXT_FULL;
193 GetThreadContext(GetCurrentThread(), &continueContext);
194
195#ifdef _M_IX86
196 continueContext.ContextFlags = CONTEXT_FULL;
197
198 /* Fill the integer registers with random values */
199 continueContext.Edi = randULONG();
200 continueContext.Esi = randULONG();
201 continueContext.Ebx = randULONG();
202 continueContext.Edx = randULONG();
203 continueContext.Ecx = randULONG();
204 continueContext.Eax = randULONG();
205 continueContext.Ebp = randULONG();
206
207 /* Randomize all the allowed flags (determined experimentally with WinDbg) */
208 continueContext.EFlags = randULONG() & 0x3C0CD5;
209
210 /* Randomize the stack pointer as much as possible */
211 continueContext.Esp = (ULONG)(((ULONG_PTR)&bogus) & 0xFFFFFFFF) +
212 sizeof(bogus) - (randULONG() & 0xF) * 4;
213
214 /* continuePoint() is implemented in assembler */
215 continueContext.Eip = (ULONG)((ULONG_PTR)continuePoint & 0xFFFFFFF);
216
217 /* Can't do a lot about segments */
218#elif defined(_M_AMD64)
219 continueContext.ContextFlags = CONTEXT_FULL | CONTEXT_DEBUG_REGISTERS;
220
221 /* Fill the integer registers with random values */
222 PULONG64 Registers = &continueContext.Rax;
223 for (ULONG i = 0; i < 16; i++)
224 {
225 Registers[i] = randULONG64();
226 }
227
228 /* Fill the XMM registers with random values */
229 Registers = (PULONG64)&continueContext.Xmm0;
230 for (ULONG i = 0; i < 32; i++)
231 {
232 Registers[i] = randULONG64();
233 }
234
235 continueContext.Dr0 = randULONG64() & 0xFFFF;
236 continueContext.Dr1 = randULONG64() & 0xFFFF;
237 continueContext.Dr2 = randULONG64() & 0xFFFF;
238 continueContext.Dr3 = randULONG64() & 0xFFFF;
239 continueContext.Dr6 = randULONG64() & 0xFFFF;
240 continueContext.Dr7 = randULONG64() & 0xFFFF;
241
242 /* Randomize all the allowed flags (determined experimentally with WinDbg) */
243 continueContext.EFlags = randULONG64() & 0x3C0CD5;
244
245 /* Randomize the stack pointer as much as possible */
246 continueContext.Rsp = (((ULONG_PTR)&bogus)) + (randULONG() & 0xF) * 16;
247 continueContext.Rsp = ALIGN_DOWN_BY(continueContext.Rsp, 16);
248
249 /* continuePoint() is implemented in assembler */
250 continueContext.Rip = ((ULONG_PTR)continuePoint);
251#endif
252
253 NtContinue(&continueContext, FALSE);
254 ok(0, "should never get here\n");
255 }
256
257 /* Second time */
258 return;
259}
ALIGN_DOWN_BY
#define ALIGN_DOWN_BY(size, align)
Definition: GetWindowPlacement.c:10
continuePoint
void continuePoint(void)
randULONG
static ULONG randULONG(void)
Definition: NtContinue.c:58
initrand
static int initrand(void)
Definition: NtContinue.c:28
FALSE
#define FALSE
Definition: types.h:117
GetThreadContext
BOOL WINAPI GetThreadContext(IN HANDLE hThread, OUT LPCONTEXT lpContext)
Definition: thread.c:501
ULONG_PTR
#define ULONG_PTR
Definition: config.h:101
RtlFillMemory
#define RtlFillMemory(Dest, Length, Fill)
Definition: winternl.h:599
for
#define for
Definition: utility.h:88
PULONG64
unsigned __int64 * PULONG64
Definition: imports.h:198
CONTEXT_DEBUG_REGISTERS
#define CONTEXT_DEBUG_REGISTERS
Definition: nt_native.h:1373
NtContinue
NTSTATUS NTAPI NtContinue(_In_ PCONTEXT Context, _In_ BOOLEAN TestAlert)
Definition: except.c:216
_CONTEXT
Definition: nt_native.h:1406
_CONTEXT::Dr3
ULONG Dr3
Definition: nt_native.h:1437
_CONTEXT::Dr1
ULONG Dr1
Definition: nt_native.h:1435
_CONTEXT::Dr6
ULONG Dr6
Definition: nt_native.h:1438
_CONTEXT::Dr0
ULONG Dr0
Definition: nt_native.h:1434
_CONTEXT::Dr2
ULONG Dr2
Definition: nt_native.h:1436
_CONTEXT::Dr7
ULONG Dr7
Definition: nt_native.h:1439
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65
setjmp
#define setjmp
Definition: setjmp.h:209
GetCurrentThread
HANDLE WINAPI GetCurrentThread(void)
Definition: proc.c:1148

Variable Documentation

◆ continueContext

CONTEXT continueContext
static

Definition at line 25 of file NtContinue.c.

Referenced by check(), and START_TEST().

◆ jmpbuf

jmp_buf jmpbuf
static

Definition at line 24 of file NtContinue.c.

Referenced by check(), jpeg_decoder_initialize(), jpeg_encoder_commit_frame(), jpeg_encoder_create_frame(), jpeg_encoder_initialize(), jpeg_encoder_write_lines(), and START_TEST().

◆ nRandBytes

unsigned int nRandBytes
static

Definition at line 26 of file NtContinue.c.

Referenced by initrand(), and randbytes().