ReactOS 0.4.15-dev-7788-g1ad9096
RtlRegistry.c File Reference
#include <kmt_test.h>
Include dependency graph for RtlRegistry.c:

Go to the source code of this file.

Classes

struct  EXPECTED_VALUE
 
struct  EXPECTED_VALUES
 

Macros

#define KMT_EMULATE_KERNEL
 
#define RTL_NUMBER_OF(x)   (sizeof(x) / sizeof(x[0]))
 
#define CONST_STR_AND_LEN(d)   { (d), sizeof(d) }
 
#define CSAL   CONST_STR_AND_LEN
 
#define NO_AUTO_LEN   1
 
#define NO_DEFAULT   2
 
#define AUTO_DIFFERS   4
 
#define DEFAULT_DIFFERS   8
 
#define DO_QUERY(ExpectedArray, ExpectedCount)
 

Typedefs

typedef struct EXPECTED_VALUEPEXPECTED_VALUE
 
typedef struct EXPECTED_VALUESPEXPECTED_VALUES
 

Functions

static NTSTATUS NTAPI QueryRoutine (_In_ PWSTR ValueName, _In_ ULONG ValueType, _In_ PVOID ValueData, _In_ ULONG ValueLength, _In_ PVOID Context, _In_ PVOID EntryContext)
 
static VOID TestRtlQueryRegistryValues (_In_ HANDLE KeyHandle)
 
 START_TEST (RtlRegistry)
 

Macro Definition Documentation

◆ AUTO_DIFFERS

#define AUTO_DIFFERS   4

◆ CONST_STR_AND_LEN

#define CONST_STR_AND_LEN (   d)    { (d), sizeof(d) }

◆ CSAL

◆ DEFAULT_DIFFERS

#define DEFAULT_DIFFERS   8

◆ DO_QUERY

#define DO_QUERY (   ExpectedArray,
  ExpectedCount 
)
Value:
do \
{ \
ULONG _i; \
ULONG _ExpectedCount = (ExpectedCount); \
for (_i = 0; _i < _ExpectedCount; _i++) \
{ \
Expected.Values[_i].ValueName = ValueName.Buffer; \
Expected.Values[_i].ValueType = REG_SZ; \
Expected.Values[_i].ValueData = (ExpectedArray)[_i].Str; \
Expected.Values[_i].ValueLength = (ExpectedArray)[_i].Len; \
} \
Expected.CurrentIndex = 0; \
Expected.Count = _ExpectedCount; \
if (_ExpectedCount == 1) \
QueryTable[0].EntryContext = &Expected.Values[0]; \
else \
QueryTable[0].EntryContext = NULL; \
NULL); \
ok_eq_hex(Status, STATUS_SUCCESS); \
ok_eq_ulong(Expected.CurrentIndex, Expected.Count); \
} while(0)
BOOLEAN Expected
#define Len
Definition: deflate.h:82
#define NULL
Definition: types.h:112
Status
Definition: gdiplustypes.h:25
NTSYSAPI NTSTATUS WINAPI RtlQueryRegistryValues(ULONG, PCWSTR, PRTL_QUERY_REGISTRY_TABLE, PVOID, PVOID)
#define REG_SZ
Definition: layer.c:22
_Must_inspect_result_ _Out_ PNDIS_STATUS _In_ NDIS_HANDLE _In_ ULONG _Out_ PNDIS_STRING _Out_ PNDIS_HANDLE KeyHandle
Definition: ndis.h:4715
_In_ PCWSTR _Inout_ _At_ QueryTable _Pre_unknown_ PRTL_QUERY_REGISTRY_TABLE QueryTable
Definition: rtlfuncs.h:4208
#define RTL_REGISTRY_HANDLE
Definition: nt_native.h:168
#define STATUS_SUCCESS
Definition: shellext.h:65
const uint16_t * PCWSTR
Definition: typedefs.h:57
uint32_t ULONG
Definition: typedefs.h:59
_Must_inspect_result_ _In_ WDFKEY _In_ PCUNICODE_STRING ValueName
Definition: wdfregistry.h:243

◆ KMT_EMULATE_KERNEL

#define KMT_EMULATE_KERNEL

Definition at line 8 of file RtlRegistry.c.

◆ NO_AUTO_LEN

#define NO_AUTO_LEN   1

◆ NO_DEFAULT

#define NO_DEFAULT   2

◆ RTL_NUMBER_OF

#define RTL_NUMBER_OF (   x)    (sizeof(x) / sizeof(x[0]))

Definition at line 12 of file RtlRegistry.c.

Typedef Documentation

◆ PEXPECTED_VALUE

◆ PEXPECTED_VALUES

Function Documentation

◆ QueryRoutine()

static NTSTATUS NTAPI QueryRoutine ( _In_ PWSTR  ValueName,
_In_ ULONG  ValueType,
_In_ PVOID  ValueData,
_In_ ULONG  ValueLength,
_In_ PVOID  Context,
_In_ PVOID  EntryContext 
)
static

Definition at line 34 of file RtlRegistry.c.

41{
42 PEXPECTED_VALUES ExpectedValues = Context;
44 SIZE_T EqualBytes;
45
46 ok(ExpectedValues->CurrentIndex < ExpectedValues->Count,
47 "Call number %lu, expected only %lu\n",
48 ExpectedValues->CurrentIndex, ExpectedValues->Count);
49 if (!skip(ExpectedValues->CurrentIndex < ExpectedValues->Count, "Out of bounds\n"))
50 {
51 Expected = &ExpectedValues->Values[ExpectedValues->CurrentIndex];
52 if (EntryContext)
54 ok_eq_wstr(ValueName, Expected->ValueName);
55 ok_eq_ulong(ValueType, Expected->ValueType);
56 ok_eq_ulong(ValueLength, Expected->ValueLength);
57 EqualBytes = RtlCompareMemory(ValueData,
58 Expected->ValueData,
59 min(ValueLength, Expected->ValueLength));
60 ok_eq_size(EqualBytes, Expected->ValueLength);
61 }
62
63 ExpectedValues->CurrentIndex++;
64 return STATUS_SUCCESS;
65}
#define ok_eq_pointer(value, expected)
Definition: apitest.h:58
#define ok_eq_ulong(value, expected)
Definition: apitest.h:62
#define ok_eq_wstr(value, expected)
Definition: apitest.h:83
#define ok_eq_size(value, expected)
Definition: apitest.h:68
#define ok(value,...)
Definition: atltest.h:57
#define skip(...)
Definition: atltest.h:64
#define RtlCompareMemory(s1, s2, l)
Definition: env_spec_w32.h:465
_In_ GUID _In_ PVOID ValueData
Definition: hubbusif.h:312
#define min(a, b)
Definition: monoChain.cc:55
_In_ PCWSTR _Inout_ _At_ QueryTable EntryContext
Definition: rtlfuncs.h:4207
EXPECTED_VALUE Values[20]
Definition: RtlRegistry.c:27
ULONG CurrentIndex
Definition: RtlRegistry.c:26
ULONG_PTR SIZE_T
Definition: typedefs.h:80
_Must_inspect_result_ _In_ WDFKEY _In_ PCUNICODE_STRING _In_ ULONG _Out_opt_ PULONG _Out_opt_ PULONG ValueType
Definition: wdfregistry.h:282
_Must_inspect_result_ _In_ WDFKEY _In_ PCUNICODE_STRING _In_ ULONG ValueLength
Definition: wdfregistry.h:275

Referenced by TestRtlQueryRegistryValues().

◆ START_TEST()

START_TEST ( RtlRegistry  )

Definition at line 198 of file RtlRegistry.c.

199{
203 HANDLE SoftwareHandle;
205
206 RtlInitUnicodeString(&KeyName, L"\\Registry\\MACHINE\\Software");
208 &KeyName,
210 NULL,
211 NULL);
212 Status = ZwOpenKey(&SoftwareHandle,
216 if (skip(NT_SUCCESS(Status) && SoftwareHandle != NULL, "No software key\n"))
217 return;
218
219 RtlInitUnicodeString(&KeyName, L"RtlRegistryKmtestKey");
221 &KeyName,
223 SoftwareHandle,
224 NULL);
225 Status = ZwCreateKey(&KeyHandle,
228 0,
229 NULL,
231 NULL);
233
234 if (!skip(NT_SUCCESS(Status) && KeyHandle != NULL, "No test key\n"))
235 {
237
238 Status = ZwDeleteKey(KeyHandle);
242 }
243}
static VOID TestRtlQueryRegistryValues(_In_ HANDLE KeyHandle)
Definition: RtlRegistry.c:69
#define ok_eq_hex(value, expected)
Definition: apitest.h:76
LONG NTSTATUS
Definition: precomp.h:26
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
#define OBJ_KERNEL_HANDLE
Definition: winternl.h:231
#define OBJ_CASE_INSENSITIVE
Definition: winternl.h:228
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
NTSYSAPI NTSTATUS NTAPI ZwClose(_In_ HANDLE Handle)
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
#define KEY_CREATE_SUB_KEY
Definition: nt_native.h:1018
#define KEY_QUERY_VALUE
Definition: nt_native.h:1016
#define DELETE
Definition: nt_native.h:57
#define REG_OPTION_VOLATILE
Definition: nt_native.h:1060
#define KEY_SET_VALUE
Definition: nt_native.h:1017
#define L(x)
Definition: ntvdm.h:50
_Must_inspect_result_ _In_ WDFDEVICE _In_ PCUNICODE_STRING KeyName
Definition: wdfdevice.h:2699

◆ TestRtlQueryRegistryValues()

static VOID TestRtlQueryRegistryValues ( _In_ HANDLE  KeyHandle)
static

Definition at line 69 of file RtlRegistry.c.

71{
75 {
76 { QueryRoutine, 0, L"TestValue", NULL, REG_NONE, NULL, 0 },
77 { NULL }
78 };
80 typedef struct
81 {
82 PWSTR Str;
83 ULONG Len;
84 } STR_AND_LEN;
85#define CONST_STR_AND_LEN(d) { (d), sizeof(d) }
86#define CSAL CONST_STR_AND_LEN
87
88#define NO_AUTO_LEN 1
89#define NO_DEFAULT 2
90#define AUTO_DIFFERS 4
91#define DEFAULT_DIFFERS 8
92 struct
93 {
94 STR_AND_LEN Value;
95 ULONG ExpectedCount;
96 STR_AND_LEN Expected[20];
98 ULONG DefaultExpectedCount;
99 STR_AND_LEN DefaultExpected[20];
100
101 } Tests[] =
102 {
103 { { NULL, 0 }, 0, { { NULL, 0 } }, NO_AUTO_LEN | NO_DEFAULT },
104 { CSAL(L""), 0, { { NULL, 0 } }, NO_AUTO_LEN },
105 { CSAL(L"\0"), 1, { CSAL(L"") },
106 AUTO_DIFFERS | DEFAULT_DIFFERS, 0, { { NULL, 0 } } },
107 { CSAL(L"String"), 1, { CSAL(L"String") }, NO_AUTO_LEN },
108 { CSAL(L"String\0"), 1, { CSAL(L"String") } },
109 { CSAL(L"String1\0String2"), 2, { CSAL(L"String1"), CSAL(L"String2") }, NO_AUTO_LEN },
110 { CSAL(L"String1\0String2\0"), 2, { CSAL(L"String1"), CSAL(L"String2") } },
111 { CSAL(L"String1\0\0String3"), 3, { CSAL(L"String1"), CSAL(L""), CSAL(L"String3") }, NO_AUTO_LEN },
112 { CSAL(L"String1\0\0String3\0"), 3, { CSAL(L"String1"), CSAL(L""), CSAL(L"String3") },
113 AUTO_DIFFERS, 1, { CSAL(L"String1") } },
114 };
115
116#define DO_QUERY(ExpectedArray, ExpectedCount) do \
117 { \
118 ULONG _i; \
119 ULONG _ExpectedCount = (ExpectedCount); \
120 for (_i = 0; _i < _ExpectedCount; _i++) \
121 { \
122 Expected.Values[_i].ValueName = ValueName.Buffer; \
123 Expected.Values[_i].ValueType = REG_SZ; \
124 Expected.Values[_i].ValueData = (ExpectedArray)[_i].Str; \
125 Expected.Values[_i].ValueLength = (ExpectedArray)[_i].Len; \
126 } \
127 Expected.CurrentIndex = 0; \
128 Expected.Count = _ExpectedCount; \
129 if (_ExpectedCount == 1) \
130 QueryTable[0].EntryContext = &Expected.Values[0]; \
131 else \
132 QueryTable[0].EntryContext = NULL; \
133 Status = RtlQueryRegistryValues(RTL_REGISTRY_HANDLE, \
134 (PCWSTR)KeyHandle, \
135 QueryTable, \
136 &Expected, \
137 NULL); \
138 ok_eq_hex(Status, STATUS_SUCCESS); \
139 ok_eq_ulong(Expected.CurrentIndex, Expected.Count); \
140 } while(0)
141
142 ULONG TestCount = RTL_NUMBER_OF(Tests);
143 ULONG i;
144
145 for (i = 0; i < TestCount; i++)
146 {
147 trace("Set: %lu\n", i);
148 Status = ZwSetValueKey(KeyHandle,
149 &ValueName,
150 0,
152 Tests[i].Value.Str,
153 Tests[i].Value.Len);
155
156 DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
157 }
158
159 /* Delete value to test default values */
162 "ZwDeleteValueKey returned %lx\n", Status);
163
164 /* Default: REG_NONE */
165 DO_QUERY((STR_AND_LEN *)NULL, 0);
166
167 for (i = 0; i < TestCount; i++)
168 {
169 if (Tests[i].Flags & NO_DEFAULT)
170 continue;
171 trace("Default: %lu\n", i);
173 QueryTable[0].DefaultData = Tests[i].Value.Str;
174 QueryTable[0].DefaultLength = Tests[i].Value.Len;
175
177 DO_QUERY(Tests[i].DefaultExpected, Tests[i].DefaultExpectedCount);
178 else
179 DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
180 }
181
182 for (i = 0; i < TestCount; i++)
183 {
184 if (Tests[i].Flags & NO_AUTO_LEN)
185 continue;
186 trace("Auto: %lu\n", i);
188 QueryTable[0].DefaultData = Tests[i].Value.Str;
190
191 if (Tests[i].Flags & AUTO_DIFFERS)
192 DO_QUERY(Tests[i].DefaultExpected, Tests[i].DefaultExpectedCount);
193 else
194 DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
195 }
196}
struct test_data Tests[]
static NTSTATUS NTAPI QueryRoutine(_In_ PWSTR ValueName, _In_ ULONG ValueType, _In_ PVOID ValueData, _In_ ULONG ValueLength, _In_ PVOID Context, _In_ PVOID EntryContext)
Definition: RtlRegistry.c:34
#define RTL_NUMBER_OF(x)
Definition: RtlRegistry.c:12
#define NO_DEFAULT
#define CSAL
#define AUTO_DIFFERS
#define NO_AUTO_LEN
#define DO_QUERY(ExpectedArray, ExpectedCount)
#define DEFAULT_DIFFERS
#define trace
Definition: atltest.h:70
NTSYSAPI NTSTATUS NTAPI ZwDeleteValueKey(__in IN HANDLE Key, __in IN PUNICODE_STRING ValueName)
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
#define REG_MULTI_SZ
Definition: nt_native.h:1501
#define REG_NONE
Definition: nt_native.h:1492
#define RTL_CONSTANT_STRING(s)
Definition: tunneltest.c:14
uint16_t * PWSTR
Definition: typedefs.h:56
#define STATUS_OBJECT_NAME_NOT_FOUND
Definition: udferr_usr.h:149
_Must_inspect_result_ _In_ WDFKEY _In_ PCUNICODE_STRING _Out_opt_ PUSHORT _Inout_opt_ PUNICODE_STRING Value
Definition: wdfregistry.h:413
_Must_inspect_result_ _In_ ULONG Flags
Definition: wsk.h:170

Referenced by START_TEST().