[SHELL32_APITEST] -Add tests for Control_RunDLLW.
[reactos.git] / rostests / kmtests / rtl / RtlStack.c
1 /*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite Runtime library stack trace test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8 #define KMT_EMULATE_KERNEL
9 #include <kmt_test.h>
10
11 static PVOID ReturnAddresses[4];
12
13 static
14 VOID
15 TestStackWalk3(VOID);
16
17 DECLSPEC_NOINLINE
18 static
19 VOID
20 TestStackWalk4(VOID)
21 {
22 PVOID Frames[5];
23 ULONG Ret;
24 ULONG Hash;
25 ULONG ExpectedHash;
26 ULONG i;
27 const ULONG FunctionSizeGuess = 0x1000;
28
29 ReturnAddresses[3] = _ReturnAddress();
30
31 Ret = RtlWalkFrameChain(NULL, 5, 0);
32 ok_eq_ulong(Ret, 0);
33
34 RtlFillMemory(Frames, sizeof(Frames), 0x55);
35 Ret = RtlWalkFrameChain(Frames, 0, 0);
36 ok_eq_ulong(Ret, 0);
37 ok_eq_pointer(Frames[0], (PVOID)(ULONG_PTR)0x5555555555555555);
38
39 RtlFillMemory(Frames, sizeof(Frames), 0x55);
40 Ret = RtlWalkFrameChain(Frames, 5, 0);
41 ok_eq_ulong(Ret, 5);
42 ok((ULONG_PTR)Frames[0] > (ULONG_PTR)TestStackWalk4, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
43 ok((ULONG_PTR)Frames[0] < (ULONG_PTR)TestStackWalk4 + FunctionSizeGuess, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
44 ok_eq_pointer(Frames[1], ReturnAddresses[3]);
45 ok_eq_pointer(Frames[2], ReturnAddresses[2]);
46 ok_eq_pointer(Frames[3], ReturnAddresses[1]);
47 ok_eq_pointer(Frames[4], ReturnAddresses[0]);
48
49 RtlFillMemory(Frames, sizeof(Frames), 0x55);
50 Ret = RtlWalkFrameChain(Frames, 4, 0);
51 ok_eq_ulong(Ret, 4);
52 ok((ULONG_PTR)Frames[0] > (ULONG_PTR)TestStackWalk4, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
53 ok((ULONG_PTR)Frames[0] < (ULONG_PTR)TestStackWalk4 + FunctionSizeGuess, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
54 ok_eq_pointer(Frames[1], ReturnAddresses[3]);
55 ok_eq_pointer(Frames[2], ReturnAddresses[2]);
56 ok_eq_pointer(Frames[3], ReturnAddresses[1]);
57 ok_eq_pointer(Frames[4], (PVOID)(ULONG_PTR)0x5555555555555555);
58
59 KmtStartSeh()
60 RtlCaptureStackBackTrace(0, 5, NULL, NULL);
61 KmtEndSeh(STATUS_ACCESS_VIOLATION);
62
63 RtlFillMemory(Frames, sizeof(Frames), 0x55);
64 Hash = 0x55555555;
65 Ret = RtlCaptureStackBackTrace(0, 0, Frames, &Hash);
66 ok_eq_ulong(Ret, 0);
67 ok_eq_hex(Hash, 0x55555555);
68 ok_eq_pointer(Frames[0], (PVOID)(ULONG_PTR)0x5555555555555555);
69
70 RtlFillMemory(Frames, sizeof(Frames), 0x55);
71 Hash = 0x55555555;
72 Ret = RtlCaptureStackBackTrace(0, 1, Frames, NULL);
73 ok_eq_ulong(Ret, 1);
74 ok((ULONG_PTR)Frames[0] > (ULONG_PTR)TestStackWalk4, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
75 ok((ULONG_PTR)Frames[0] < (ULONG_PTR)TestStackWalk4 + FunctionSizeGuess, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
76 ok_eq_pointer(Frames[1], (PVOID)(ULONG_PTR)0x5555555555555555);
77
78 RtlFillMemory(Frames, sizeof(Frames), 0x55);
79 Ret = RtlCaptureStackBackTrace(0, 5, Frames, &Hash);
80 ok_eq_ulong(Ret, 5);
81 ExpectedHash = 0;
82 for (i = 0; i < 5; i++)
83 ExpectedHash += (ULONG)(ULONG_PTR)Frames[i];
84 ok_eq_hex(Hash, ExpectedHash);
85 ok((ULONG_PTR)Frames[0] > (ULONG_PTR)TestStackWalk4, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
86 ok((ULONG_PTR)Frames[0] < (ULONG_PTR)TestStackWalk4 + FunctionSizeGuess, "Frame is %p, function is %p\n", Frames[0], TestStackWalk4);
87 ok_eq_pointer(Frames[1], ReturnAddresses[3]);
88 ok_eq_pointer(Frames[2], ReturnAddresses[2]);
89 ok_eq_pointer(Frames[3], ReturnAddresses[1]);
90 ok_eq_pointer(Frames[4], ReturnAddresses[0]);
91
92 RtlFillMemory(Frames, sizeof(Frames), 0x55);
93 Ret = RtlCaptureStackBackTrace(1, 4, Frames, &Hash);
94 ok_eq_ulong(Ret, 4);
95 ExpectedHash = 0;
96 for (i = 0; i < 4; i++)
97 ExpectedHash += (ULONG)(ULONG_PTR)Frames[i];
98 ok_eq_hex(Hash, ExpectedHash);
99 ok_eq_pointer(Frames[0], ReturnAddresses[3]);
100 ok_eq_pointer(Frames[1], ReturnAddresses[2]);
101 ok_eq_pointer(Frames[2], ReturnAddresses[1]);
102 ok_eq_pointer(Frames[3], ReturnAddresses[0]);
103 ok_eq_pointer(Frames[4], (PVOID)(ULONG_PTR)0x5555555555555555);
104 }
105
106 DECLSPEC_NOINLINE
107 static
108 VOID
109 TestStackWalk3(VOID)
110 {
111 ReturnAddresses[2] = _ReturnAddress();
112 TestStackWalk4();
113 }
114
115 DECLSPEC_NOINLINE
116 static
117 VOID
118 TestStackWalk2(VOID)
119 {
120 ReturnAddresses[1] = _ReturnAddress();
121 TestStackWalk3();
122 }
123
124 DECLSPEC_NOINLINE
125 static
126 VOID
127 TestStackWalk1(VOID)
128 {
129 ReturnAddresses[0] = _ReturnAddress();
130 TestStackWalk2();
131 }
132
133 START_TEST(RtlStack)
134 {
135 TestStackWalk1();
136 }