ReactOS 0.4.15-dev-7918-g2a2556c
Functions | Variables
query.c File Reference
#include <stdio.h>
#include "reg.h"
Include dependency graph for query.c:

Go to the source code of this file.

Functions

static const WCHAR * reg_type_to_wchar (DWORD type)
 
static WCHAR * reg_data_to_wchar (DWORD type, const BYTE *src, DWORD size_bytes)
 
static void output_value (const WCHAR *value_name, DWORD type, BYTE *data, DWORD data_size)
 
static int query_value (HKEY hkey, WCHAR *value_name, WCHAR *path, BOOL recurse)
 
static int query_all (HKEY hkey, WCHAR *path, BOOL recurse, BOOL recursing)
 
static int run_query (HKEY root, WCHAR *path, WCHAR *key_name, WCHAR *value_name, BOOL value_empty, BOOL recurse)
 
int reg_query (int argc, WCHAR *argvW[])
 

Variables

static const WCHAR * newlineW = L"\n"
 
static unsigned int num_values_found = 0
 
static REGSAM sam = 0
 

Function Documentation

◆ output_value()

static void output_value ( const WCHAR *  value_name,
DWORD  type,
BYTE *  data,
DWORD  data_size 
)
static

Definition at line 113 of file query.c.

114{
115 static const WCHAR *fmt = L" %1";
116 WCHAR defval[32];
117 WCHAR *reg_data;
118
119 if (value_name && value_name[0])
120 output_string(fmt, value_name);
121 else
122 {
123 LoadStringW(GetModuleHandleW(NULL), STRING_DEFAULT_VALUE, defval, ARRAY_SIZE(defval));
124 output_string(fmt, defval);
125 }
126 output_string(fmt, reg_type_to_wchar(type));
127
128 if (data)
129 {
130 reg_data = reg_data_to_wchar(type, data, data_size);
131 output_string(fmt, reg_data);
132 free(reg_data);
133 }
134 else
135 {
136 LoadStringW(GetModuleHandleW(NULL), STRING_VALUE_NOT_SET, defval, ARRAY_SIZE(defval));
137 output_string(fmt, defval);
138 }
139 output_string(newlineW);
140}
newlineW
static const WCHAR * newlineW
Definition: query.c:111
reg_data_to_wchar
static WCHAR * reg_data_to_wchar(DWORD type, const BYTE *src, DWORD size_bytes)
Definition: query.c:35
reg_type_to_wchar
static const WCHAR * reg_type_to_wchar(DWORD type)
Definition: query.c:22
output_string
void WINAPIV output_string(const WCHAR *fmt,...)
Definition: reg.c:117
STRING_VALUE_NOT_SET
#define STRING_VALUE_NOT_SET
Definition: resource.h:85
STRING_DEFAULT_VALUE
#define STRING_DEFAULT_VALUE
Definition: resource.h:40
ARRAY_SIZE
#define ARRAY_SIZE(A)
Definition: main.h:33
free
#define free
Definition: debug_ros.c:5
NULL
#define NULL
Definition: types.h:112
GetModuleHandleW
HMODULE WINAPI GetModuleHandleW(LPCWSTR lpModuleName)
Definition: loader.c:838
type
GLuint GLuint GLsizei GLenum type
Definition: gl.h:1545
data
GLint GLenum GLsizei GLsizei GLsizei GLint GLsizei const GLvoid * data
Definition: gl.h:1950
L
#define L(x)
Definition: ntvdm.h:50
fmt
Definition: dsound.c:943
LoadStringW
int WINAPI LoadStringW(_In_opt_ HINSTANCE hInstance, _In_ UINT uID, _Out_writes_to_(cchBufferMax, return+1) LPWSTR lpBuffer, _In_ int cchBufferMax)
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180

Referenced by query_all(), and query_value().

◆ query_all()

static int query_all ( HKEY  hkey,
WCHAR *  path,
BOOL  recurse,
BOOL  recursing 
)
static

Definition at line 222 of file query.c.

223{
224 LONG rc;
225 DWORD num_subkeys, num_values;
226 DWORD max_value_len = 256, value_len;
227 DWORD max_data_bytes = 2048, data_size;
228 DWORD subkey_len;
229 DWORD i, type, path_len;
230 WCHAR *value_name, *subkey_name, *subkey_path;
231 BYTE *data;
232 HKEY subkey;
233
234 rc = RegQueryInfoKeyW(hkey, NULL, NULL, NULL, &num_subkeys, NULL,
235 NULL, &num_values, NULL, NULL, NULL, NULL);
236 if (rc) return 1;
237
238 if (num_values || recursing)
239 output_string(L"%1\n", path);
240
241 value_name = malloc(max_value_len * sizeof(WCHAR));
242 data = malloc(max_data_bytes);
243
244 i = 0;
245 for (;;)
246 {
247 value_len = max_value_len;
248 data_size = max_data_bytes;
249 rc = RegEnumValueW(hkey, i, value_name, &value_len, NULL, &type, data, &data_size);
250 if (rc == ERROR_SUCCESS)
251 {
252 output_value(value_name, type, data, data_size);
253 i++;
254 }
255 else if (rc == ERROR_MORE_DATA)
256 {
257 if (data_size > max_data_bytes)
258 {
259 max_data_bytes = data_size;
260 data = realloc(data, max_data_bytes);
261 }
262 else
263 {
264 max_value_len *= 2;
265 value_name = realloc(value_name, max_value_len * sizeof(WCHAR));
266 }
267 }
268 else break;
269 }
270
271 free(data);
272 free(value_name);
273
274 if (i || recursing)
275 output_string(newlineW);
276
277 if (!num_subkeys)
278 return 0;
279
280 subkey_name = malloc(MAX_SUBKEY_LEN * sizeof(WCHAR));
281
282 path_len = lstrlenW(path);
283
284 i = 0;
285 for (;;)
286 {
287 subkey_len = MAX_SUBKEY_LEN;
288 rc = RegEnumKeyExW(hkey, i, subkey_name, &subkey_len, NULL, NULL, NULL, NULL);
289 if (rc == ERROR_SUCCESS)
290 {
291 if (recurse)
292 {
293 subkey_path = build_subkey_path(path, path_len, subkey_name, subkey_len);
294 if (!RegOpenKeyExW(hkey, subkey_name, 0, KEY_READ|sam, &subkey))
295 {
296 query_all(subkey, subkey_path, recurse, TRUE);
297 RegCloseKey(subkey);
298 }
299 free(subkey_path);
300 }
301 else output_string(L"%1\\%2\n", path, subkey_name);
302 i++;
303 }
304 else break;
305 }
306
307 free(subkey_name);
308 return 0;
309}
query_all
static int query_all(HKEY hkey, WCHAR *path, BOOL recurse, BOOL recursing)
Definition: query.c:222
output_value
static void output_value(const WCHAR *value_name, DWORD type, BYTE *data, DWORD data_size)
Definition: query.c:113
sam
static REGSAM sam
Definition: query.c:143
build_subkey_path
WCHAR * build_subkey_path(WCHAR *path, DWORD path_len, WCHAR *subkey_name, DWORD subkey_len)
Definition: reg.c:198
MAX_SUBKEY_LEN
#define MAX_SUBKEY_LEN
Definition: reg.h:26
RegCloseKey
#define RegCloseKey(hKey)
Definition: registry.h:49
ERROR_MORE_DATA
#define ERROR_MORE_DATA
Definition: dderror.h:13
realloc
#define realloc
Definition: debug_ros.c:6
malloc
#define malloc
Definition: debug_ros.c:4
ERROR_SUCCESS
#define ERROR_SUCCESS
Definition: deptool.c:10
TRUE
#define TRUE
Definition: types.h:120
RegOpenKeyExW
LONG WINAPI RegOpenKeyExW(HKEY hKey, LPCWSTR lpSubKey, DWORD ulOptions, REGSAM samDesired, PHKEY phkResult)
Definition: reg.c:3333
RegEnumKeyExW
LONG WINAPI RegEnumKeyExW(_In_ HKEY hKey, _In_ DWORD dwIndex, _Out_ LPWSTR lpName, _Inout_ LPDWORD lpcbName, _Reserved_ LPDWORD lpReserved, _Out_opt_ LPWSTR lpClass, _Inout_opt_ LPDWORD lpcbClass, _Out_opt_ PFILETIME lpftLastWriteTime)
Definition: reg.c:2504
RegEnumValueW
LONG WINAPI RegEnumValueW(_In_ HKEY hKey, _In_ DWORD index, _Out_ LPWSTR value, _Inout_ PDWORD val_count, _Reserved_ PDWORD reserved, _Out_opt_ PDWORD type, _Out_opt_ LPBYTE data, _Inout_opt_ PDWORD count)
Definition: reg.c:2830
RegQueryInfoKeyW
LONG WINAPI RegQueryInfoKeyW(HKEY hKey, LPWSTR lpClass, LPDWORD lpcClass, LPDWORD lpReserved, LPDWORD lpcSubKeys, LPDWORD lpcMaxSubKeyLen, LPDWORD lpcMaxClassLen, LPDWORD lpcValues, LPDWORD lpcMaxValueNameLen, LPDWORD lpcMaxValueLen, LPDWORD lpcbSecurityDescriptor, PFILETIME lpftLastWriteTime)
Definition: reg.c:3662
lstrlenW
#define lstrlenW
Definition: compat.h:750
DWORD
unsigned long DWORD
Definition: ntddk_ex.h:95
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
path_len
static DWORD path_len
Definition: batch.c:31
KEY_READ
#define KEY_READ
Definition: nt_native.h:1023
LONG
long LONG
Definition: pedump.c:60
BYTE
unsigned char BYTE
Definition: xxhash.c:193

Referenced by query_all(), and run_query().

◆ query_value()

static int query_value ( HKEY  hkey,
WCHAR *  value_name,
WCHAR *  path,
BOOL  recurse 
)
static

Definition at line 145 of file query.c.

146{
147 LONG rc;
148 DWORD max_data_bytes = 2048, data_size;
149 DWORD subkey_len;
150 DWORD type, path_len, i;
151 BYTE *data;
152 static const WCHAR *fmt = L"%1\n";
153 WCHAR *subkey_name, *subkey_path;
154 HKEY subkey;
155
156 data = malloc(max_data_bytes);
157
158 for (;;)
159 {
160 data_size = max_data_bytes;
161 rc = RegQueryValueExW(hkey, value_name, NULL, &type, data, &data_size);
162 if (rc == ERROR_MORE_DATA)
163 {
164 max_data_bytes = data_size;
165 data = realloc(data, max_data_bytes);
166 }
167 else break;
168 }
169
170 if (rc == ERROR_SUCCESS)
171 {
172 output_string(fmt, path);
173 output_value(value_name, type, data, data_size);
174 output_string(newlineW);
175 num_values_found++;
176 }
177
178 free(data);
179
180 if (!recurse)
181 {
182 if (rc == ERROR_FILE_NOT_FOUND)
183 {
184 if (value_name && *value_name)
185 {
186 output_message(STRING_VALUE_NONEXIST);
187 return 1;
188 }
189 output_string(fmt, path);
190 output_value(NULL, REG_SZ, NULL, 0);
191 }
192 return 0;
193 }
194
195 subkey_name = malloc(MAX_SUBKEY_LEN * sizeof(WCHAR));
196
197 path_len = lstrlenW(path);
198
199 i = 0;
200 for (;;)
201 {
202 subkey_len = MAX_SUBKEY_LEN;
203 rc = RegEnumKeyExW(hkey, i, subkey_name, &subkey_len, NULL, NULL, NULL, NULL);
204 if (rc == ERROR_SUCCESS)
205 {
206 subkey_path = build_subkey_path(path, path_len, subkey_name, subkey_len);
207 if (!RegOpenKeyExW(hkey, subkey_name, 0, KEY_READ|sam, &subkey))
208 {
209 query_value(subkey, value_name, subkey_path, recurse);
210 RegCloseKey(subkey);
211 }
212 free(subkey_path);
213 i++;
214 }
215 else break;
216 }
217
218 free(subkey_name);
219 return 0;
220}
query_value
static int query_value(HKEY hkey, WCHAR *value_name, WCHAR *path, BOOL recurse)
Definition: query.c:145
num_values_found
static unsigned int num_values_found
Definition: query.c:142
output_message
void WINAPIV output_message(unsigned int id,...)
Definition: reg.c:92
STRING_VALUE_NONEXIST
#define STRING_VALUE_NONEXIST
Definition: resource.h:39
RegQueryValueExW
LONG WINAPI RegQueryValueExW(_In_ HKEY hkeyorg, _In_ LPCWSTR name, _In_ LPDWORD reserved, _In_ LPDWORD type, _In_ LPBYTE data, _In_ LPDWORD count)
Definition: reg.c:4103
REG_SZ
#define REG_SZ
Definition: layer.c:22
ERROR_FILE_NOT_FOUND
#define ERROR_FILE_NOT_FOUND
Definition: disk.h:79

Referenced by query_value(), and run_query().

◆ reg_data_to_wchar()

static WCHAR * reg_data_to_wchar ( DWORD  type,
const BYTE *  src,
DWORD  size_bytes 
)
static

Definition at line 35 of file query.c.

36{
37 WCHAR *buffer = NULL;
38 int i;
39
40 switch (type)
41 {
42 case REG_SZ:
43 case REG_EXPAND_SZ:
44 buffer = malloc(size_bytes);
45 lstrcpyW(buffer, (WCHAR *)src);
46 break;
47 case REG_NONE:
48 case REG_BINARY:
49 {
50 WCHAR *ptr;
51
52 buffer = malloc((size_bytes * 2 + 1) * sizeof(WCHAR));
53
54 if (!size_bytes)
55 {
56 *buffer = 0;
57 break;
58 }
59
60 ptr = buffer;
61
62 for (i = 0; i < size_bytes; i++)
63 ptr += swprintf(ptr, L"%02X", src[i]);
64 break;
65 }
66 case REG_DWORD:
67 /* case REG_DWORD_LITTLE_ENDIAN: */
68 case REG_DWORD_BIG_ENDIAN:
69 {
70 const int zero_x_dword = 10;
71
72 buffer = malloc((zero_x_dword + 1) * sizeof(WCHAR));
73 swprintf(buffer, L"0x%x", *(DWORD *)src);
74 break;
75 }
76 case REG_MULTI_SZ:
77 {
78 const int two_wchars = 2 * sizeof(WCHAR);
79 DWORD tmp_size;
80 const WCHAR *tmp = (const WCHAR *)src;
81 int len, destindex;
82
83 if (size_bytes <= two_wchars)
84 {
85 buffer = malloc(sizeof(WCHAR));
86 *buffer = 0;
87 return buffer;
88 }
89
90 tmp_size = size_bytes - two_wchars; /* exclude both null terminators */
91 buffer = malloc(tmp_size * 2 + sizeof(WCHAR));
92 len = tmp_size / sizeof(WCHAR);
93
94 for (i = 0, destindex = 0; i < len; i++, destindex++)
95 {
96 if (tmp[i])
97 buffer[destindex] = tmp[i];
98 else
99 {
100 buffer[destindex++] = '\\';
101 buffer[destindex] = '0';
102 }
103 }
104 buffer[destindex] = 0;
105 break;
106 }
107 }
108 return buffer;
109}
lstrcpyW
#define lstrcpyW
Definition: compat.h:749
swprintf
#define swprintf
Definition: precomp.h:40
src
GLenum src
Definition: glext.h:6340
buffer
GLuint buffer
Definition: glext.h:5915
len
GLenum GLsizei len
Definition: glext.h:6722
ptr
static PVOID ptr
Definition: dispmode.c:27
REG_BINARY
#define REG_BINARY
Definition: nt_native.h:1496
REG_DWORD_BIG_ENDIAN
#define REG_DWORD_BIG_ENDIAN
Definition: nt_native.h:1499
REG_MULTI_SZ
#define REG_MULTI_SZ
Definition: nt_native.h:1501
REG_NONE
#define REG_NONE
Definition: nt_native.h:1492
REG_EXPAND_SZ
#define REG_EXPAND_SZ
Definition: nt_native.h:1494
REG_DWORD
#define REG_DWORD
Definition: sdbapi.c:596

Referenced by output_value().

◆ reg_query()

int reg_query ( int  argc,
WCHAR *  argvW[] 
)

Definition at line 339 of file query.c.

340{
341 HKEY root;
342 WCHAR *path, *key_name, *value_name = NULL;
343 BOOL value_empty = FALSE, recurse = FALSE;
344 int i;
345
346 if (!parse_registry_key(argvW[2], &root, &path))
347 return 1;
348
349 for (i = 3; i < argc; i++)
350 {
351 WCHAR *str;
352
353 if (argvW[i][0] != '/' && argvW[i][0] != '-')
354 goto invalid;
355
356 str = &argvW[i][1];
357
358 if (!lstrcmpiW(str, L"ve"))
359 {
360 if (value_empty) goto invalid;
361 value_empty = TRUE;
362 continue;
363 }
364 else if (!lstrcmpiW(str, L"reg:32"))
365 {
366 if (sam & KEY_WOW64_32KEY) goto invalid;
367 sam |= KEY_WOW64_32KEY;
368 continue;
369 }
370 else if (!lstrcmpiW(str, L"reg:64"))
371 {
372 if (sam & KEY_WOW64_64KEY) goto invalid;
373 sam |= KEY_WOW64_64KEY;
374 continue;
375 }
376 else if (!str[0] || str[1])
377 goto invalid;
378
379 switch (towlower(*str))
380 {
381 case 'v':
382 if (value_name || !(value_name = argvW[++i]))
383 goto invalid;
384 break;
385 case 's':
386 if (recurse) goto invalid;
387 recurse = TRUE;
388 break;
389 default:
390 goto invalid;
391 }
392 }
393
394 if (value_name && value_empty)
395 goto invalid;
396
397 if (sam == (KEY_WOW64_32KEY|KEY_WOW64_64KEY))
398 goto invalid;
399
400 key_name = get_long_key(root, path);
401
402 return run_query(root, path, key_name, value_name, value_empty, recurse);
403
404invalid:
405 output_message(STRING_INVALID_SYNTAX);
406 output_message(STRING_FUNC_HELP, _wcsupr(argvW[1]));
407 return 1;
408}
argc
static int argc
Definition: ServiceArgs.c:12
run_query
static int run_query(HKEY root, WCHAR *path, WCHAR *key_name, WCHAR *value_name, BOOL value_empty, BOOL recurse)
Definition: query.c:311
get_long_key
WCHAR * get_long_key(HKEY root, WCHAR *path)
Definition: reg.c:208
parse_registry_key
BOOL parse_registry_key(const WCHAR *key, HKEY *root, WCHAR **path)
Definition: reg.c:234
STRING_INVALID_SYNTAX
#define STRING_INVALID_SYNTAX
Definition: resource.h:33
STRING_FUNC_HELP
#define STRING_FUNC_HELP
Definition: resource.h:34
root
struct _root root
FALSE
#define FALSE
Definition: types.h:117
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
lstrcmpiW
int WINAPI lstrcmpiW(LPCWSTR lpString1, LPCWSTR lpString2)
Definition: lstring.c:194
invalid
static const WCHAR invalid[]
Definition: assoc.c:39
str
const WCHAR * str
Definition: rpc_transport.c:2724
_wcsupr
_CRTIMP wchar_t *__cdecl _wcsupr(_Inout_z_ wchar_t *_String)
_root
Definition: btrfs_drv.h:450
key_name
Definition: registry.c:449
towlower
#define towlower(c)
Definition: wctype.h:97
KEY_WOW64_32KEY
#define KEY_WOW64_32KEY
Definition: cmtypes.h:45
KEY_WOW64_64KEY
#define KEY_WOW64_64KEY
Definition: cmtypes.h:46

Referenced by wmain().

◆ reg_type_to_wchar()

static const WCHAR * reg_type_to_wchar ( DWORD  type)
static

Definition at line 22 of file query.c.

23{
24 int i, array_size = ARRAY_SIZE(type_rels);
25
26 for (i = 0; i < array_size; i++)
27 {
28 if (type == type_rels[i].type)
29 return type_rels[i].name;
30 }
31
32 return NULL;
33}
type_rels
const struct reg_type_rels type_rels[]
Definition: reg.c:41
array_size
static UINT array_size
Definition: msctf.c:67

Referenced by output_value().

◆ run_query()

static int run_query ( HKEY  root,
WCHAR *  path,
WCHAR *  key_name,
WCHAR *  value_name,
BOOL  value_empty,
BOOL  recurse 
)
static

Definition at line 311 of file query.c.

313{
314 HKEY hkey;
315 int ret;
316
317 if (RegOpenKeyExW(root, path, 0, KEY_READ|sam, &hkey))
318 {
319 output_message(STRING_KEY_NONEXIST);
320 return 1;
321 }
322
323 output_string(newlineW);
324
325 if (value_name || value_empty)
326 {
327 ret = query_value(hkey, value_name, key_name, recurse);
328 if (recurse)
329 output_message(STRING_MATCHES_FOUND, num_values_found);
330 }
331 else
332 ret = query_all(hkey, key_name, recurse, FALSE);
333
334 RegCloseKey(hkey);
335
336 return ret;
337}
STRING_KEY_NONEXIST
#define STRING_KEY_NONEXIST
Definition: resource.h:38
STRING_MATCHES_FOUND
#define STRING_MATCHES_FOUND
Definition: resource.h:86
ret
int ret
Definition: wcstombs-tests.c:31

Referenced by create_package_db(), and reg_query().

Variable Documentation

◆ newlineW

const WCHAR* newlineW = L"\n"
static

Definition at line 111 of file query.c.

Referenced by comment_line(), output_value(), query_all(), query_value(), and run_query().

◆ num_values_found

unsigned int num_values_found = 0
static

Definition at line 142 of file query.c.

Referenced by query_value(), and run_query().

◆ sam

REGSAM sam = 0
static

Definition at line 143 of file query.c.

Referenced by add_key_(), create_container_key(), CRYPT_SysRegOpenStoreW(), delete_key_(), delete_tree_(), export_registry_data(), open_key_(), query_all(), query_value(), reg_add(), reg_copy(), reg_delete(), reg_export(), reg_query(), run_add(), run_copy(), run_delete(), run_query(), verify_key_(), and verify_key_nonexist_().