ReactOS 0.4.15-dev-7918-g2a2556c
PsNotify.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite Process Notification Routines test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8#include <kmt_test.h>
9
10//#define NDEBUG
11#include <debug.h>
12
13static
14VOID
15NTAPI
16CreateProcessNotifyRoutine(
17 IN HANDLE ParentId,
18 IN HANDLE ProcessId,
19 IN BOOLEAN Create)
20{
21 ok_irql(PASSIVE_LEVEL);
22 if (!Create)
23 ok_eq_pointer(ProcessId, PsGetCurrentProcessId());
24 else
25 ok(ProcessId != PsGetCurrentProcessId(),
26 "ProcessId %p equals current\n", ProcessId);
27 DPRINT("%s(%p, %p, %d)\n", __FUNCTION__, ParentId, ProcessId, Create);
28}
29
30static
31VOID
32TestCreateProcessNotify(VOID)
33{
34 NTSTATUS Status;
35
36 Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
37 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
38
39 Status = PsSetCreateProcessNotifyRoutine(NULL, FALSE);
40 ok_eq_hex(Status, STATUS_SUCCESS);
41
42 Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
43 ok_eq_hex(Status, STATUS_SUCCESS);
44
45 Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
46 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
47
48 Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
49 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
50
51 Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, FALSE);
52 ok_eq_hex(Status, STATUS_SUCCESS);
53
54 /* TODO: test whether the callback is notified on process creation */
55
56 Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
57 ok_eq_hex(Status, STATUS_SUCCESS);
58
59 Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
60 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
61
62 /* TODO: register the same routine twice */
63 /* TODO: register more than the maximum number of notifications */
64}
65
66static
67VOID
68NTAPI
69CreateThreadNotifyRoutine(
70 IN HANDLE ProcessId,
71 IN HANDLE ThreadId,
72 IN BOOLEAN Create)
73{
74 ok_irql(PASSIVE_LEVEL);
75 if (!Create)
76 {
77 ok_eq_pointer(ProcessId, PsGetCurrentProcessId());
78 ok_eq_pointer(ThreadId, PsGetCurrentThreadId());
79 }
80 else
81 {
82 ok(ThreadId != PsGetCurrentThreadId(),
83 "ThreadId %p equals current\n", ThreadId);
84 }
85 DPRINT("%s(%p, %p, %d)\n", __FUNCTION__, ProcessId, ThreadId, Create);
86}
87
88static
89VOID
90TestCreateThreadNotify(VOID)
91{
92 NTSTATUS Status;
93
94 Status = PsRemoveCreateThreadNotifyRoutine(NULL);
95 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
96
97 Status = PsSetCreateThreadNotifyRoutine(NULL);
98 ok_eq_hex(Status, STATUS_SUCCESS);
99
100 Status = PsRemoveCreateThreadNotifyRoutine(NULL);
101 ok_eq_hex(Status, STATUS_SUCCESS);
102
103 Status = PsRemoveCreateThreadNotifyRoutine(NULL);
104 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
105
106 Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
107 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
108
109 Status = PsSetCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
110 ok_eq_hex(Status, STATUS_SUCCESS);
111
112 /* TODO: test whether the callback is notified on thread creation */
113
114 Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
115 ok_eq_hex(Status, STATUS_SUCCESS);
116
117 Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
118 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
119
120 /* TODO: register the same routine twice */
121 /* TODO: register more than the maximum number of notifications */
122}
123
124static
125VOID
126NTAPI
127LoadImageNotifyRoutine(
128 IN PUNICODE_STRING FullImageName OPTIONAL,
129 IN HANDLE ProcessId,
130 IN PIMAGE_INFO ImageInfo)
131{
132 ok_irql(PASSIVE_LEVEL);
133 DPRINT("%s('%wZ', %p, %p)\n", __FUNCTION__, FullImageName, ProcessId, ImageInfo);
134}
135
136static
137VOID
138TestLoadImageNotify(VOID)
139{
140 NTSTATUS Status;
141
142 Status = PsRemoveLoadImageNotifyRoutine(NULL);
143 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
144
145 Status = PsSetLoadImageNotifyRoutine(NULL);
146 ok_eq_hex(Status, STATUS_SUCCESS);
147
148 Status = PsRemoveLoadImageNotifyRoutine(NULL);
149 ok_eq_hex(Status, STATUS_SUCCESS);
150
151 Status = PsRemoveLoadImageNotifyRoutine(NULL);
152 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
153
154 Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
155 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
156
157 Status = PsSetLoadImageNotifyRoutine(LoadImageNotifyRoutine);
158 ok_eq_hex(Status, STATUS_SUCCESS);
159
160 /* TODO: test whether the callback is notified on image load */
161
162 Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
163 ok_eq_hex(Status, STATUS_SUCCESS);
164
165 Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
166 ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
167
168 /* TODO: register the same routine twice */
169 /* TODO: register more than the maximum number of notifications */
170}
171
172START_TEST(PsNotify)
173{
174 TestCreateProcessNotify();
175 TestCreateThreadNotify();
176 TestLoadImageNotify();
177}
BOOLEAN
unsigned char BOOLEAN
Definition: ProcessorBind.h:185
TestLoadImageNotify
static VOID TestLoadImageNotify(VOID)
Definition: PsNotify.c:138
TestCreateProcessNotify
static VOID TestCreateProcessNotify(VOID)
Definition: PsNotify.c:32
TestCreateThreadNotify
static VOID TestCreateThreadNotify(VOID)
Definition: PsNotify.c:90
CreateProcessNotifyRoutine
static VOID NTAPI CreateProcessNotifyRoutine(IN HANDLE ParentId, IN HANDLE ProcessId, IN BOOLEAN Create)
Definition: PsNotify.c:16
CreateThreadNotifyRoutine
static VOID NTAPI CreateThreadNotifyRoutine(IN HANDLE ProcessId, IN HANDLE ThreadId, IN BOOLEAN Create)
Definition: PsNotify.c:69
LoadImageNotifyRoutine
static VOID NTAPI LoadImageNotifyRoutine(IN PUNICODE_STRING FullImageName OPTIONAL, IN HANDLE ProcessId, IN PIMAGE_INFO ImageInfo)
Definition: PsNotify.c:127
ok_eq_pointer
#define ok_eq_pointer(value, expected)
Definition: apitest.h:59
ok_eq_hex
#define ok_eq_hex(value, expected)
Definition: apitest.h:77
ok
#define ok(value,...)
Definition: atltest.h:57
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
Create
@ Create
Definition: registry.c:563
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
__FUNCTION__
#define __FUNCTION__
Definition: types.h:116
PASSIVE_LEVEL
#define PASSIVE_LEVEL
Definition: env_spec_w32.h:693
ProcessId
IN PLARGE_INTEGER IN PLARGE_INTEGER PEPROCESS ProcessId
Definition: fatprocs.h:2711
PsGetCurrentThreadId
PsGetCurrentThreadId
Definition: CrNtStubs.h:8
Status
Status
Definition: gdiplustypes.h:25
kmt_test.h
PsGetCurrentProcessId
HANDLE NTAPI PsGetCurrentProcessId(VOID)
Definition: process.c:1123
STATUS_PROCEDURE_NOT_FOUND
#define STATUS_PROCEDURE_NOT_FOUND
Definition: ntstatus.h:358
PsSetLoadImageNotifyRoutine
NTSTATUS NTAPI PsSetLoadImageNotifyRoutine(IN PLOAD_IMAGE_NOTIFY_ROUTINE NotifyRoutine)
Definition: psnotify.c:176
PsSetCreateThreadNotifyRoutine
NTSTATUS NTAPI PsSetCreateThreadNotifyRoutine(IN PCREATE_THREAD_NOTIFY_ROUTINE NotifyRoutine)
Definition: psnotify.c:260
PsSetCreateProcessNotifyRoutine
NTSTATUS NTAPI PsSetCreateProcessNotifyRoutine(IN PCREATE_PROCESS_NOTIFY_ROUTINE NotifyRoutine, IN BOOLEAN Remove)
Definition: psnotify.c:33
PsRemoveLoadImageNotifyRoutine
NTSTATUS NTAPI PsRemoveLoadImageNotifyRoutine(IN PLOAD_IMAGE_NOTIFY_ROUTINE NotifyRoutine)
Definition: psnotify.c:127
PsRemoveCreateThreadNotifyRoutine
NTSTATUS NTAPI PsRemoveCreateThreadNotifyRoutine(IN PCREATE_THREAD_NOTIFY_ROUTINE NotifyRoutine)
Definition: psnotify.c:211
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
DPRINT
#define DPRINT
Definition: sndvol32.h:71
OPTIONAL
PULONG MinorVersion OPTIONAL
Definition: CrossNt.h:68
_IMAGE_INFO
Definition: pstypes.h:196
_UNICODE_STRING
Definition: env_spec_w32.h:368
NTAPI
#define NTAPI
Definition: typedefs.h:36
IN
#define IN
Definition: typedefs.h:39