[NTDLL_APITEST] Disable NtContinue test when runtime checks are enabled
[reactos.git] / rostests / apitests / ntdll / NtContinue.c
1 /*
2 * PROJECT: ReactOS api tests
3 * LICENSE: GPL - See COPYING in the top level directory
4 * PURPOSE: Test for NtContinue
5 * PROGRAMMER:
6 */
7
8 #include <apitest.h>
9 #include <ndk/kefuncs.h>
10
11 #include <setjmp.h>
12 #include <time.h>
13
14 #ifdef _M_IX86
15 #define NTC_SEGMENT_BITS (0xFFFF)
16 #define NTC_EFLAGS_BITS (0x3C0CD5)
17 #endif
18
19 void continuePoint(void);
20
21 static jmp_buf jmpbuf;
22 static CONTEXT continueContext;
23 static unsigned int nRandBytes;
24
25 static int initrand(void)
26 {
27 unsigned int nRandMax;
28 unsigned int nRandMaxBits;
29 time_t tLoc;
30
31 nRandMax = RAND_MAX;
32 for(nRandMaxBits = 0; nRandMax != 0; nRandMax >>= 1, ++ nRandMaxBits);
33 nRandBytes = nRandMaxBits / CHAR_BIT;
34 //assert(nRandBytes != 0);
35 srand((unsigned)(time(&tLoc) & UINT_MAX));
36 return 1;
37 }
38
39 static void randbytes(void * p, size_t n)
40 {
41 unsigned char * b;
42 size_t i;
43 int r = rand();
44
45 b = (unsigned char *)p;
46 for(i = 0; i < n; ++ i)
47 {
48 if(i % nRandBytes == 0)
49 r = rand();
50 b[i] = (unsigned char)(r & UCHAR_MAX);
51 r >>= CHAR_BIT;
52 }
53 }
54
55 static ULONG randULONG(void)
56 {
57 ULONG n;
58 randbytes(&n, sizeof(n));
59 return n;
60 }
61
62 void check(CONTEXT * pContext)
63 {
64 #ifdef _M_IX86
65 ok(pContext->ContextFlags == CONTEXT_FULL,
66 "ContextFlags=0x%lx\n", pContext->ContextFlags);
67
68 /* Random data segments */
69 ok((pContext->SegGs & NTC_SEGMENT_BITS) ==
70 (continueContext.SegGs & NTC_SEGMENT_BITS),
71 "SegGs=0x%lx / 0x%lx\n", pContext->SegGs, continueContext.SegGs);
72
73 ok((pContext->SegFs & NTC_SEGMENT_BITS) ==
74 (continueContext.SegFs & NTC_SEGMENT_BITS),
75 "SegFs=0x%lx / 0x%lx\n", pContext->SegFs, continueContext.SegFs);
76
77 ok((pContext->SegEs & NTC_SEGMENT_BITS) ==
78 (continueContext.SegEs & NTC_SEGMENT_BITS),
79 "SegEs=0x%lx / 0x%lx\n", pContext->SegEs, continueContext.SegEs);
80
81 ok((pContext->SegDs & NTC_SEGMENT_BITS) ==
82 (continueContext.SegDs & NTC_SEGMENT_BITS),
83 "SegDs=0x%lx / 0x%lx\n", pContext->SegDs, continueContext.SegDs);
84
85 /* Integer registers */
86 ok(pContext->Edi == continueContext.Edi,
87 "Edi: 0x%lx != 0x%lx\n", pContext->Edi, continueContext.Edi);
88 ok(pContext->Esi == continueContext.Esi,
89 "Esi: 0x%lx != 0x%lx\n", pContext->Esi, continueContext.Esi);
90 ok(pContext->Ebx == continueContext.Ebx,
91 "Ebx: 0x%lx != 0x%lx\n", pContext->Ebx, continueContext.Ebx);
92 ok(pContext->Edx == continueContext.Edx,
93 "Edx: 0x%lx != 0x%lx\n", pContext->Edx, continueContext.Edx);
94 ok(pContext->Ecx == continueContext.Ecx,
95 "Ecx: 0x%lx != 0x%lx\n", pContext->Ecx, continueContext.Ecx);
96 ok(pContext->Eax == continueContext.Eax,
97 "Eax: 0x%lx != 0x%lx\n", pContext->Eax, continueContext.Eax);
98
99 /* Control registers and segments */
100 ok(pContext->Ebp == continueContext.Ebp,
101 "Ebp: 0x%lx != 0x%lx\n", pContext->Ebp, continueContext.Ebp);
102 ok(pContext->Eip == continueContext.Eip,
103 "Eip: 0x%lx != 0x%lx\n", pContext->Eip, continueContext.Eip);
104 ok(pContext->Esp == continueContext.Esp,
105 "Esp: 0x%lx != 0x%lx\n", pContext->Esp, continueContext.Esp);
106
107 ok((pContext->SegCs & NTC_SEGMENT_BITS) ==
108 (continueContext.SegCs & NTC_SEGMENT_BITS),
109 "SegCs: 0x%lx != 0x%lx\n", pContext->SegCs, continueContext.SegCs);
110
111 ok((pContext->EFlags & NTC_EFLAGS_BITS) ==
112 (continueContext.EFlags & NTC_EFLAGS_BITS),
113 "EFlags: 0x%lx != 0x%lx\n", pContext->EFlags, continueContext.EFlags);
114
115 ok((pContext->SegSs & NTC_SEGMENT_BITS) ==
116 (continueContext.SegSs & NTC_SEGMENT_BITS),
117 "SegSs: 0x%lx != 0x%lx\n", pContext->SegSs, continueContext.SegSs);
118 #endif
119
120 /* Return where we came from */
121 longjmp(jmpbuf, 1);
122 }
123
124 START_TEST(NtContinue)
125 {
126 #ifdef __RUNTIME_CHECKS__
127 skip("This test breaks MSVC runtime checks!");
128 return;
129 #endif /* __RUNTIME_CHECKS__ */
130 initrand();
131
132 /* First time */
133 if(setjmp(jmpbuf) == 0)
134 {
135 CONTEXT bogus;
136
137 continueContext.ContextFlags = CONTEXT_FULL;
138 GetThreadContext(GetCurrentThread(), &continueContext);
139
140 #ifdef _M_IX86
141 continueContext.ContextFlags = CONTEXT_FULL;
142
143 /* Fill the integer registers with random values */
144 continueContext.Edi = randULONG();
145 continueContext.Esi = randULONG();
146 continueContext.Ebx = randULONG();
147 continueContext.Edx = randULONG();
148 continueContext.Ecx = randULONG();
149 continueContext.Eax = randULONG();
150 continueContext.Ebp = randULONG();
151
152 /* Randomize all the allowed flags (determined experimentally with WinDbg) */
153 continueContext.EFlags = randULONG() & 0x3C0CD5;
154
155 /* Randomize the stack pointer as much as possible */
156 continueContext.Esp = (ULONG)(((ULONG_PTR)&bogus) & 0xFFFFFFFF) +
157 sizeof(bogus) - (randULONG() & 0xF) * 4;
158
159 /* continuePoint() is implemented in assembler */
160 continueContext.Eip = (ULONG)((ULONG_PTR)continuePoint & 0xFFFFFFF);
161
162 /* Can't do a lot about segments */
163 #endif
164
165 NtContinue(&continueContext, FALSE);
166 ok(0, "should never get here\n");
167 }
168
169 /* Second time */
170 return;
171 }