ReactOS 0.4.15-dev-7942-gd23573b
Macros | Functions | Variables
ObReference.c File Reference
#include <kmt_test.h>
#include <debug.h>
Include dependency graph for ObReference.c:

Go to the source code of this file.

Macros

#define NDEBUG
 
#define CheckObject(Handle, Pointers, Handles)
 

Functions

static VOID TestReference (IN HANDLE Handle, IN PUNICODE_STRING Name OPTIONAL, IN PUNICODE_STRING NameUpper OPTIONAL, IN BOOLEAN CaseSensitive, IN ULONG AdditionalReferences, IN BOOLEAN Permanent)
 
 START_TEST (ObReference)
 

Variables

static POBJECT_TYPE ObDirectoryObjectType
 

Macro Definition Documentation

◆ CheckObject

#define CheckObject (   Handle,
  Pointers,
  Handles 
)
Value:
do \
{ \
PUBLIC_OBJECT_BASIC_INFORMATION ObjectInfo; \
Status = ZwQueryObject(Handle, ObjectBasicInformation, \
&ObjectInfo, sizeof ObjectInfo, NULL); \
ok_eq_hex(Status, STATUS_SUCCESS); \
ok_eq_ulong(ObjectInfo.PointerCount, Pointers); \
ok_eq_ulong(ObjectInfo.HandleCount, Handles); \
} while (0)
ObjectBasicInformation
@ ObjectBasicInformation
Definition: DriverTester.h:54
NULL
#define NULL
Definition: types.h:112
Handle
ULONG Handle
Definition: gdb_input.c:15
Status
Status
Definition: gdiplustypes.h:25
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_PUBLIC_OBJECT_BASIC_INFORMATION
Definition: winternl.h:444
_PUBLIC_OBJECT_BASIC_INFORMATION::PointerCount
ULONG PointerCount
Definition: winternl.h:448
_PUBLIC_OBJECT_BASIC_INFORMATION::HandleCount
ULONG HandleCount
Definition: winternl.h:447

Definition at line 12 of file ObReference.c.

◆ NDEBUG

#define NDEBUG

Definition at line 9 of file ObReference.c.

Function Documentation

◆ START_TEST()

START_TEST ( ObReference  )

Definition at line 142 of file ObReference.c.

143{
144 NTSTATUS Status;
145 HANDLE DirectoryHandle = NULL;
146 OBJECT_ATTRIBUTES ObjectAttributes;
147 UNICODE_STRING Name, *pName;
148 UNICODE_STRING NameUpper, *pNameUpper;
149 ULONG i;
150 struct
151 {
152 PWSTR Name;
153 ULONG Flags;
154 ULONG AdditionalReferences;
155 } Tests[] =
156 {
157 { NULL, 0, 0 },
158 { NULL, OBJ_CASE_INSENSITIVE, 0 },
159 { NULL, OBJ_KERNEL_HANDLE, 0 },
160 { NULL, OBJ_PERMANENT, 0 },
161 { NULL, OBJ_CASE_INSENSITIVE | OBJ_PERMANENT, 0 },
162 { NULL, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, 0 },
163 { NULL, OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 0 },
164 { NULL, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 0 },
165 { L"\\YayDirectory0", 0, 1 },
166 { L"\\YayDirectory1", OBJ_CASE_INSENSITIVE, 1 },
167 { L"\\YayDirectory2", OBJ_KERNEL_HANDLE, 1 },
168 { L"\\YayDirectory3", OBJ_PERMANENT, 1 },
169 { L"\\YayDirectory4", OBJ_CASE_INSENSITIVE | OBJ_PERMANENT, 1 },
170 { L"\\YayDirectory5", OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, 1 },
171 { L"\\YayDirectory6", OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 1 },
172 { L"\\YayDirectory7", OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 1 },
173 };
174 HANDLE ObjectTypeHandle;
175
176 /* ObReferenceObjectByName needs the object type... so get it... */
177 RtlInitUnicodeString(&Name, L"\\ObjectTypes\\Directory");
178 InitializeObjectAttributes(&ObjectAttributes, &Name, OBJ_KERNEL_HANDLE, NULL, NULL);
179 Status = ObOpenObjectByName(&ObjectAttributes, NULL, KernelMode, NULL, 0, NULL, &ObjectTypeHandle);
180 ok_eq_hex(Status, STATUS_SUCCESS);
181 ok(ObjectTypeHandle != NULL, "ObjectTypeHandle = NULL\n");
182 if (!skip(Status == STATUS_SUCCESS && ObjectTypeHandle, "No handle\n"))
183 {
184 Status = ObReferenceObjectByHandle(ObjectTypeHandle, 0, NULL, KernelMode, (PVOID)&ObDirectoryObjectType, NULL);
185 ok_eq_hex(Status, STATUS_SUCCESS);
186 ok(ObDirectoryObjectType != NULL, "ObDirectoryObjectType = NULL\n");
187 Status = ZwClose(ObjectTypeHandle);
188 ok_eq_hex(Status, STATUS_SUCCESS);
189 }
190
191 for (i = 0; i < sizeof Tests / sizeof Tests[0]; ++i)
192 {
193 DPRINT("Run %d\n", i);
194 if (Tests[i].Name)
195 {
196 RtlInitUnicodeString(&Name, Tests[i].Name);
197 pName = &Name;
198 Status = RtlUpcaseUnicodeString(&NameUpper, &Name, TRUE);
199 ok_eq_hex(Status, STATUS_SUCCESS);
200 if (skip(Status == STATUS_SUCCESS, "No upper case name\n"))
201 pNameUpper = NULL;
202 else
203 pNameUpper = &NameUpper;
204 }
205 else
206 {
207 pName = NULL;
208 pNameUpper = NULL;
209 }
210 InitializeObjectAttributes(&ObjectAttributes, pName, Tests[i].Flags, NULL, NULL);
211 Status = ZwCreateDirectoryObject(&DirectoryHandle, DIRECTORY_ALL_ACCESS, &ObjectAttributes);
212 ok_eq_hex(Status, STATUS_SUCCESS);
213 ok(DirectoryHandle != NULL, "DirectoryHandle = NULL\n");
214
215 if (!skip(Status == STATUS_SUCCESS && DirectoryHandle, "Cannot proceed without an object\n"))
216 {
217 TestReference(DirectoryHandle, pName, pNameUpper, FALSE, Tests[i].AdditionalReferences, (Tests[i].Flags & OBJ_PERMANENT) != 0);
218 /* try again for good measure */
219 TestReference(DirectoryHandle, pName, pNameUpper, FALSE, Tests[i].AdditionalReferences, FALSE);
220
221 Status = ZwClose(DirectoryHandle);
222 ok_eq_hex(Status, STATUS_SUCCESS);
223 DirectoryHandle = NULL;
224 }
225
226 if (pNameUpper)
227 RtlFreeUnicodeString(pNameUpper);
228 }
229
230 if (DirectoryHandle)
231 {
232 Status = ZwClose(DirectoryHandle);
233 ok_eq_hex(Status, STATUS_SUCCESS);
234 }
235
236 /* parameter tests */
237 /* bugcheck at APC_LEVEL
238 Status = ObReferenceObject(NULL);
239 Status = ObReferenceObjectByPointer(NULL, 0, NULL, UserMode);
240 Status = ObReferenceObjectByPointer(NULL, 0, NULL, KernelMode);*/
241
242 if (ObDirectoryObjectType)
243 {
244 ObDereferenceObject(ObDirectoryObjectType);
245 ObDirectoryObjectType = NULL;
246 }
247}
TestReference
static VOID TestReference(IN HANDLE Handle, IN PUNICODE_STRING Name OPTIONAL, IN PUNICODE_STRING NameUpper OPTIONAL, IN BOOLEAN CaseSensitive, IN ULONG AdditionalReferences, IN BOOLEAN Permanent)
Definition: ObReference.c:26
ObDirectoryObjectType
static POBJECT_TYPE ObDirectoryObjectType
Definition: ObReference.c:22
DirectoryHandle
static HANDLE DirectoryHandle
Definition: ObType.c:48
Tests
struct test_data Tests[]
Definition: RtlDosApplyFileIsolationRedirection_Ustr.c:22
Name
struct NameRec_ * Name
Definition: cdprocs.h:460
ok_eq_hex
#define ok_eq_hex(value, expected)
Definition: apitest.h:77
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
ObjectAttributes
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
RtlUpcaseUnicodeString
NTSTATUS RtlUpcaseUnicodeString(PUNICODE_STRING dst, PUNICODE_STRING src, BOOLEAN Alloc)
Definition: string_lib.cpp:46
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
OBJ_KERNEL_HANDLE
#define OBJ_KERNEL_HANDLE
Definition: winternl.h:231
OBJ_CASE_INSENSITIVE
#define OBJ_CASE_INSENSITIVE
Definition: winternl.h:228
OBJ_PERMANENT
#define OBJ_PERMANENT
Definition: winternl.h:226
pName
static LPSTR pName
Definition: security.c:75
InitializeObjectAttributes
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
KernelMode
#define KernelMode
Definition: asm.h:34
ZwClose
NTSYSAPI NTSTATUS NTAPI ZwClose(_In_ HANDLE Handle)
ZwCreateDirectoryObject
NTSYSAPI NTSTATUS NTAPI ZwCreateDirectoryObject(_Out_ PHANDLE DirectoryHandle, _In_ ACCESS_MASK DesiredAccess, _In_ POBJECT_ATTRIBUTES ObjectAttributes)
RtlInitUnicodeString
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
RtlFreeUnicodeString
NTSYSAPI VOID NTAPI RtlFreeUnicodeString(PUNICODE_STRING UnicodeString)
DIRECTORY_ALL_ACCESS
#define DIRECTORY_ALL_ACCESS
Definition: nt_native.h:1259
L
#define L(x)
Definition: ntvdm.h:50
ObOpenObjectByName
NTSTATUS NTAPI ObOpenObjectByName(IN POBJECT_ATTRIBUTES ObjectAttributes, IN POBJECT_TYPE ObjectType, IN KPROCESSOR_MODE AccessMode, IN PACCESS_STATE PassedAccessState, IN ACCESS_MASK DesiredAccess, IN OUT PVOID ParseContext, OUT PHANDLE Handle)
Definition: obhandle.c:2532
ObReferenceObjectByHandle
NTSTATUS NTAPI ObReferenceObjectByHandle(IN HANDLE Handle, IN ACCESS_MASK DesiredAccess, IN POBJECT_TYPE ObjectType, IN KPROCESSOR_MODE AccessMode, OUT PVOID *Object, OUT POBJECT_HANDLE_INFORMATION HandleInformation OPTIONAL)
Definition: obref.c:494
DPRINT
#define DPRINT
Definition: sndvol32.h:71
NameRec_
Definition: apinames.c:49
_OBJECT_ATTRIBUTES
Definition: umtypes.h:182
_UNICODE_STRING
Definition: env_spec_w32.h:368
PWSTR
uint16_t * PWSTR
Definition: typedefs.h:56
ULONG
uint32_t ULONG
Definition: typedefs.h:59
Flags
_Must_inspect_result_ _In_ ULONG Flags
Definition: wsk.h:170
ObDereferenceObject
#define ObDereferenceObject
Definition: obfuncs.h:203

◆ TestReference()

static VOID TestReference ( IN HANDLE  Handle,
IN PUNICODE_STRING Name  OPTIONAL,
IN PUNICODE_STRING NameUpper  OPTIONAL,
IN BOOLEAN  CaseSensitive,
IN ULONG  AdditionalReferences,
IN BOOLEAN  Permanent 
)
static

Definition at line 26 of file ObReference.c.

33{
34 NTSTATUS Status;
35 LONG_PTR Ret;
36 PVOID Object = NULL;
37 PVOID Object2 = NULL;
38 PVOID Object3 = NULL;
39 PVOID Object4 = NULL;
40
41 CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
42
43 Status = ObReferenceObjectByHandle(Handle, DIRECTORY_ALL_ACCESS, NULL, KernelMode, &Object, NULL);
44 ok_eq_hex(Status, STATUS_SUCCESS);
45 ok(Object != NULL, "ObReferenceObjectByHandle returned NULL object\n");
46 CheckObject(Handle, 3LU + AdditionalReferences, 1LU);
47
48 Status = ObReferenceObjectByHandle(Handle, DIRECTORY_ALL_ACCESS, NULL, KernelMode, &Object2, NULL);
49 ok_eq_hex(Status, STATUS_SUCCESS);
50 ok(Object != NULL, "ObReferenceObjectByHandle returned NULL object\n");
51 ok_eq_pointer(Object, Object2);
52 CheckObject(Handle, 4LU + AdditionalReferences, 1LU);
53
54 if (!skip(Object != NULL, "No object to reference!\n"))
55 {
56 Ret = ObReferenceObject(Object);
57 ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences);
58 CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
59
60 Ret = ObReferenceObject(Object);
61 ok_eq_longptr(Ret, (LONG_PTR)5 + AdditionalReferences);
62 CheckObject(Handle, 6LU + AdditionalReferences, 1LU);
63
64 Status = ObReferenceObjectByPointer(Object, DIRECTORY_ALL_ACCESS, NULL, KernelMode);
65 ok_eq_hex(Status, STATUS_SUCCESS);
66 CheckObject(Handle, 7LU + AdditionalReferences, 1LU);
67
68 Status = ObReferenceObjectByPointer(Object, DIRECTORY_ALL_ACCESS, NULL, KernelMode);
69 ok_eq_hex(Status, STATUS_SUCCESS);
70 CheckObject(Handle, 8LU + AdditionalReferences, 1LU);
71
72 Ret = ObDereferenceObject(Object);
73 ok_eq_longptr(Ret, (LONG_PTR)6 + AdditionalReferences);
74 CheckObject(Handle, 7LU + AdditionalReferences, 1LU);
75
76 Ret = ObDereferenceObject(Object);
77 ok_eq_longptr(Ret, (LONG_PTR)5 + AdditionalReferences);
78 CheckObject(Handle, 6LU + AdditionalReferences, 1LU);
79
80 Ret = ObDereferenceObject(Object);
81 ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences);
82 CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
83
84 Ret = ObDereferenceObject(Object);
85 ok_eq_longptr(Ret, (LONG_PTR)3 + AdditionalReferences);
86 CheckObject(Handle, 4LU + AdditionalReferences, 1LU);
87 }
88
89 if (Name && !skip(ObDirectoryObjectType != NULL, "No directory object type\n"))
90 {
91 Status = ObReferenceObjectByName(Name, 0, NULL, DIRECTORY_ALL_ACCESS, ObDirectoryObjectType, KernelMode, NULL, &Object3);
92 ok_eq_hex(Status, STATUS_SUCCESS);
93 CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
94 }
95
96 if (NameUpper && !skip(ObDirectoryObjectType != NULL, "No directory object type\n"))
97 {
98 Status = ObReferenceObjectByName(NameUpper, 0, NULL, DIRECTORY_ALL_ACCESS, ObDirectoryObjectType, KernelMode, NULL, &Object4);
99 ok_eq_hex(Status, CaseSensitive ? STATUS_OBJECT_NAME_NOT_FOUND : STATUS_SUCCESS);
100 CheckObject(Handle, 5LU + AdditionalReferences + !CaseSensitive, 1LU);
101 }
102
103 if (NameUpper && !skip(Object4 != NULL, "No object to dereference\n"))
104 {
105 Ret = ObDereferenceObject(Object4);
106 ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences);
107 CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
108 }
109 if (Name && !skip(Object3 != NULL, "No object to dereference\n"))
110 {
111 Ret = ObDereferenceObject(Object3);
112 ok_eq_longptr(Ret, (LONG_PTR)3 + AdditionalReferences);
113 CheckObject(Handle, 4LU + AdditionalReferences, 1LU);
114 }
115 if (!skip(Object2 != NULL, "No object to dereference\n"))
116 {
117 Ret = ObDereferenceObject(Object2);
118 ok_eq_longptr(Ret, (LONG_PTR)2 + AdditionalReferences);
119 CheckObject(Handle, 3LU + AdditionalReferences, 1LU);
120 }
121 if (!skip(Object != NULL, "No object to dereference\n"))
122 {
123 Ret = ObDereferenceObject(Object);
124 ok_eq_longptr(Ret, (LONG_PTR)1 + AdditionalReferences);
125 CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
126 }
127
128 CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
129
130 if (Permanent)
131 {
132 Status = ZwMakeTemporaryObject(Handle);
133 ok_eq_hex(Status, STATUS_SUCCESS);
134 CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
135
136 Status = ZwMakeTemporaryObject(Handle);
137 ok_eq_hex(Status, STATUS_SUCCESS);
138 CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
139 }
140}
CheckObject
#define CheckObject(Handle, Pointers, Handles)
Definition: ObReference.c:12
ok_eq_pointer
#define ok_eq_pointer(value, expected)
Definition: apitest.h:59
ok_eq_longptr
#define ok_eq_longptr(value, expected)
Definition: apitest.h:70
U
#define U(x)
Definition: wordpad.c:45
LONG_PTR
__int3264 LONG_PTR
Definition: mstsclib_h.h:276
ZwMakeTemporaryObject
NTSYSAPI NTSTATUS NTAPI ZwMakeTemporaryObject(_In_ HANDLE Handle)
ObReferenceObjectByName
NTSTATUS NTAPI ObReferenceObjectByName(IN PUNICODE_STRING ObjectPath, IN ULONG Attributes, IN PACCESS_STATE PassedAccessState, IN ACCESS_MASK DesiredAccess, IN POBJECT_TYPE ObjectType, IN KPROCESSOR_MODE AccessMode, IN OUT PVOID ParseContext, OUT PVOID *ObjectPtr)
Definition: obref.c:409
ObReferenceObjectByPointer
NTSTATUS NTAPI ObReferenceObjectByPointer(IN PVOID Object, IN ACCESS_MASK DesiredAccess, IN POBJECT_TYPE ObjectType, IN KPROCESSOR_MODE AccessMode)
Definition: obref.c:381
STATUS_OBJECT_NAME_NOT_FOUND
#define STATUS_OBJECT_NAME_NOT_FOUND
Definition: udferr_usr.h:149
Object
_Must_inspect_result_ _In_ WDFCOLLECTION _In_ WDFOBJECT Object
Definition: wdfcollection.h:123
ObReferenceObject
#define ObReferenceObject
Definition: obfuncs.h:204

Referenced by START_TEST().

Variable Documentation

◆ ObDirectoryObjectType

POBJECT_TYPE ObDirectoryObjectType
static

Definition at line 22 of file ObReference.c.

Referenced by START_TEST(), and TestReference().