[USP10_WINETEST] Sync with Wine Staging 2.9. CORE-13362
[reactos.git] / rostests / kmtests / ntos_ke / KeIrql.c
1 /*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite Interrupt Request Level test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8 #ifndef _M_AMD64
9 __declspec(dllimport) void __stdcall KeRaiseIrql(unsigned char, unsigned char *);
10 __declspec(dllimport) void __stdcall KeLowerIrql(unsigned char);
11 #else
12 #define CLOCK1_LEVEL CLOCK_LEVEL
13 #define CLOCK2_LEVEL CLOCK_LEVEL
14 #endif
15
16 #include <kmt_test.h>
17
18 #define NDEBUG
19 #include <debug.h>
20
21 START_TEST(KeIrql)
22 {
23 KIRQL Irql, Irql2, PrevIrql, SynchIrql;
24
25 /* we should be called at PASSIVE_LEVEL */
26 ok_irql(PASSIVE_LEVEL);
27
28 PrevIrql = KeGetCurrentIrql();
29
30 /* SYNCH_LEVEL is different for UP/MP */
31 if (KmtIsMultiProcessorBuild)
32 SynchIrql = IPI_LEVEL - 2;
33 else
34 SynchIrql = DISPATCH_LEVEL;
35
36 /* some Irqls MUST work */
37 {
38 const KIRQL Irqls[] = { LOW_LEVEL, PASSIVE_LEVEL, APC_LEVEL, DISPATCH_LEVEL,
39 CMCI_LEVEL, CLOCK1_LEVEL, CLOCK2_LEVEL, CLOCK_LEVEL,
40 PROFILE_LEVEL, IPI_LEVEL, /*POWER_LEVEL,*/ SynchIrql, HIGH_LEVEL };
41 int i;
42 for (i = 0; i < sizeof Irqls / sizeof Irqls[0]; ++i)
43 {
44 KeRaiseIrql(Irqls[i], &Irql2);
45 ok_eq_uint(Irql2, PrevIrql);
46 ok_irql(Irqls[i]);
47 KeLowerIrql(Irql2);
48 ok_irql(PrevIrql);
49 }
50 }
51
52 /* raising/lowering to the current level should have no effect */
53 ok_irql(PASSIVE_LEVEL);
54 KeRaiseIrql(PASSIVE_LEVEL, &Irql);
55 ok_eq_uint(Irql, PASSIVE_LEVEL);
56 KeLowerIrql(PASSIVE_LEVEL);
57 ok_irql(PASSIVE_LEVEL);
58
59 /* try to raise to each Irql and back */
60 for (Irql = PASSIVE_LEVEL; Irql <= HIGH_LEVEL; ++Irql)
61 {
62 DPRINT("Raising to %u\n", Irql);
63 KeRaiseIrql(Irql, &Irql2);
64 ok_eq_uint(Irql2, PrevIrql);
65 KeLowerIrql(Irql2);
66 ok_irql(PrevIrql);
67 }
68
69 /* go through all Irqls in order, skip the ones that the system doesn't accept */
70 for (Irql = PASSIVE_LEVEL; Irql <= HIGH_LEVEL; ++Irql)
71 {
72 DPRINT("Raising to %u\n", Irql);
73 KeRaiseIrql(Irql, &Irql2);
74 ok_eq_uint(Irql2, PrevIrql);
75 Irql2 = KeGetCurrentIrql();
76 ok(Irql2 <= Irql, "New Irql is %u, expected <= requested value of %u\n", Irql2, Irql);
77 PrevIrql = Irql2;
78 }
79
80 ok_irql(HIGH_LEVEL);
81
82 /* now go back again, skipping the ones that don't work */
83 for (Irql = HIGH_LEVEL; Irql > PASSIVE_LEVEL;)
84 {
85 DPRINT("Lowering to %u\n", Irql - 1);
86 KeLowerIrql(Irql - 1);
87 Irql2 = KeGetCurrentIrql();
88 ok(Irql2 < Irql, "New Irql is %u, expected <= requested value of %u\n", Irql2, Irql - 1);
89 if (Irql2 < Irql)
90 Irql = Irql2;
91 else
92 --Irql;
93 }
94
95 /* test KeRaiseIrqlToDpcLevel */
96 ok_irql(PASSIVE_LEVEL);
97 Irql = KeRaiseIrqlToDpcLevel();
98 ok_irql(DISPATCH_LEVEL);
99 ok_eq_uint(Irql, PASSIVE_LEVEL);
100 Irql = KeRaiseIrqlToDpcLevel();
101 ok_irql(DISPATCH_LEVEL);
102 ok_eq_uint(Irql, DISPATCH_LEVEL);
103 KeLowerIrql(PASSIVE_LEVEL);
104
105 /* test KeRaiseIrqlToSynchLevel */
106 ok_irql(PASSIVE_LEVEL);
107 Irql = KeRaiseIrqlToSynchLevel();
108 ok_irql(SynchIrql);
109 ok_eq_uint(Irql, PASSIVE_LEVEL);
110 Irql = KeRaiseIrqlToSynchLevel();
111 ok_irql(SynchIrql);
112 ok_eq_uint(Irql, SynchIrql);
113 KeLowerIrql(PASSIVE_LEVEL);
114
115 /* these bugcheck on a checked build but run fine on free! */
116 if (!KmtIsCheckedBuild)
117 {
118 KeRaiseIrql(HIGH_LEVEL, &Irql);
119 KeRaiseIrql(APC_LEVEL, &Irql);
120 ok_irql(APC_LEVEL);
121 KeLowerIrql(HIGH_LEVEL);
122 ok_irql(HIGH_LEVEL);
123 KeLowerIrql(PASSIVE_LEVEL);
124 }
125
126 #ifndef _M_AMD64
127 /* try the actual exports, not only the fastcall versions */
128 ok_irql(PASSIVE_LEVEL);
129 (KeRaiseIrql)(HIGH_LEVEL, &Irql);
130 ok_irql(HIGH_LEVEL);
131 ok_eq_uint(Irql, PASSIVE_LEVEL);
132 (KeLowerIrql)(Irql);
133 ok_irql(PASSIVE_LEVEL);
134 #endif
135
136 /* make sure we exit gracefully */
137 ok_irql(PASSIVE_LEVEL);
138 KeLowerIrql(PASSIVE_LEVEL);
139 }