2 * Unit tests for registry functions
4 * Copyright (c) 2002 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #include "wine/test.h"
31 static HKEY hkey_main
;
34 static const char * sTestpath1
= "%LONGSYSTEMVAR%\\subdir1";
35 static const char * sTestpath2
= "%FOO%\\subdir1";
37 static HMODULE hadvapi32
;
38 static DWORD (WINAPI
*pRegGetValueA
)(HKEY
,LPCSTR
,LPCSTR
,DWORD
,LPDWORD
,PVOID
,LPDWORD
);
39 static DWORD (WINAPI
*pRegDeleteTreeA
)(HKEY
,LPCSTR
);
43 /* Debugging functions from wine/libs/wine/debug.c */
45 /* allocate some tmp string space */
46 /* FIXME: this is not 100% thread-safe */
47 static char *get_temp_buffer( int size
)
49 static char *list
[32];
54 idx
= ++pos
% (sizeof(list
)/sizeof(list
[0]));
55 if ((ret
= realloc( list
[idx
], size
))) list
[idx
] = ret
;
59 static const char *wine_debugstr_an( const char *str
, int n
)
61 static const char hex
[16] = "0123456789abcdef";
65 if (!((ULONG_PTR
)str
>> 16))
67 if (!str
) return "(null)";
68 res
= get_temp_buffer( 6 );
69 sprintf( res
, "#%04x", LOWORD(str
) );
72 if (n
== -1) n
= strlen(str
);
74 size
= 10 + min( 300, n
* 4 );
75 dst
= res
= get_temp_buffer( size
);
77 while (n
-- > 0 && dst
<= res
+ size
- 9)
79 unsigned char c
= *str
++;
82 case '\n': *dst
++ = '\\'; *dst
++ = 'n'; break;
83 case '\r': *dst
++ = '\\'; *dst
++ = 'r'; break;
84 case '\t': *dst
++ = '\\'; *dst
++ = 't'; break;
85 case '"': *dst
++ = '\\'; *dst
++ = '"'; break;
86 case '\\': *dst
++ = '\\'; *dst
++ = '\\'; break;
88 if (c
>= ' ' && c
<= 126)
94 *dst
++ = hex
[(c
>> 4) & 0x0f];
95 *dst
++ = hex
[c
& 0x0f];
110 static const char *wine_debugstr_wn( const WCHAR
*str
, int n
)
117 if (!str
) return "(null)";
118 res
= get_temp_buffer( 6 );
119 sprintf( res
, "#%04x", LOWORD(str
) );
122 if (n
== -1) n
= lstrlenW(str
);
124 size
= 12 + min( 300, n
* 5);
125 dst
= res
= get_temp_buffer( n
* 5 + 7 );
128 while (n
-- > 0 && dst
<= res
+ size
- 10)
133 case '\n': *dst
++ = '\\'; *dst
++ = 'n'; break;
134 case '\r': *dst
++ = '\\'; *dst
++ = 'r'; break;
135 case '\t': *dst
++ = '\\'; *dst
++ = 't'; break;
136 case '"': *dst
++ = '\\'; *dst
++ = '"'; break;
137 case '\\': *dst
++ = '\\'; *dst
++ = '\\'; break;
139 if (c
>= ' ' && c
<= 126)
144 sprintf(dst
,"%04x",c
);
161 #define ADVAPI32_GET_PROC(func) \
162 p ## func = (void*)GetProcAddress(hadvapi32, #func); \
164 trace("GetProcAddress(%s) failed\n", #func);
166 static void InitFunctionPtrs(void)
168 hadvapi32
= GetModuleHandleA("advapi32.dll");
170 /* This function was introduced with Windows 2003 SP1 */
171 ADVAPI32_GET_PROC(RegGetValueA
)
172 ADVAPI32_GET_PROC(RegDeleteTreeA
)
175 /* delete key and all its subkeys */
176 static DWORD
delete_key( HKEY hkey
)
181 while (!(ret
= RegEnumKeyA(hkey
, 0, name
, sizeof(name
))))
184 if (!(ret
= RegOpenKeyExA( hkey
, name
, 0, KEY_ENUMERATE_SUB_KEYS
, &tmp
)))
186 ret
= delete_key( tmp
);
191 if (ret
!= ERROR_NO_MORE_ITEMS
) return ret
;
192 RegDeleteKeyA( hkey
, "" );
196 static void setup_main_key(void)
198 if (!RegOpenKeyA( HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkey_main
)) delete_key( hkey_main
);
200 assert (!RegCreateKeyA( HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkey_main
));
203 static void test_hkey_main_Value_A(LPCSTR name
, LPCSTR string
,
206 DWORD ret
, type
, cbData
;
209 static const char nA
[]={'N', 0};
213 /* When successful RegQueryValueExA() leaves GLE as is,
214 * so we must reset it to detect unimplemented functions.
216 SetLastError(0xdeadbeef);
217 ret
= RegQueryValueExA(hkey_main
, name
, NULL
, &type
, NULL
, &cbData
);
218 GLE
= GetLastError();
219 ok(ret
== ERROR_SUCCESS
, "RegQueryValueExA failed: %d, GLE=%d\n", ret
, GLE
);
220 /* It is wrong for the Ansi version to not be implemented */
221 ok(GLE
== 0xdeadbeef, "RegQueryValueExA set GLE = %u\n", GLE
);
222 if(GLE
== ERROR_CALL_NOT_IMPLEMENTED
) return;
224 str_byte_len
= (string
? lstrlenA(string
) : 0) + 1;
225 ok(type
== REG_SZ
, "RegQueryValueExA returned type %d\n", type
);
226 ok(cbData
== full_byte_len
|| cbData
== str_byte_len
/* Win9x */,
227 "cbData=%d instead of %d or %d\n", cbData
, full_byte_len
, str_byte_len
);
229 value
= HeapAlloc(GetProcessHeap(), 0, (cbData
+2)*sizeof(*value
));
232 ret
= RegQueryValueExA(hkey_main
, name
, NULL
, &type
, (BYTE
*)value
, &cbData
);
233 GLE
= GetLastError();
234 ok(ret
== ERROR_SUCCESS
, "RegQueryValueExA failed: %d, GLE=%d\n", ret
, GLE
);
237 /* When cbData == 0, RegQueryValueExA() should not modify the buffer */
238 ok(strcmp(value
, nA
) == 0 || (cbData
== 1 && *value
== '\0') /* Win9x */,
239 "RegQueryValueExA failed: '%s' != '%s'\n", value
, string
);
243 ok(memcmp(value
, string
, cbData
) == 0, "RegQueryValueExA failed: %s/%d != %s/%d\n",
244 wine_debugstr_an(value
, cbData
), cbData
,
245 wine_debugstr_an(string
, full_byte_len
), full_byte_len
);
247 HeapFree(GetProcessHeap(), 0, value
);
250 static void test_hkey_main_Value_W(LPCWSTR name
, LPCWSTR string
,
253 DWORD ret
, type
, cbData
;
255 static const WCHAR nW
[]={'N', 0};
259 /* When successful RegQueryValueExW() leaves GLE as is,
260 * so we must reset it to detect unimplemented functions.
262 SetLastError(0xdeadbeef);
263 ret
= RegQueryValueExW(hkey_main
, name
, NULL
, &type
, NULL
, &cbData
);
264 GLE
= GetLastError();
265 ok(ret
== ERROR_SUCCESS
, "RegQueryValueExW failed: %d, GLE=%d\n", ret
, GLE
);
266 if(GLE
== ERROR_CALL_NOT_IMPLEMENTED
) return;
268 ok(type
== REG_SZ
, "RegQueryValueExW returned type %d\n", type
);
269 ok(cbData
== full_byte_len
,
270 "cbData=%d instead of %d\n", cbData
, full_byte_len
);
272 value
= HeapAlloc(GetProcessHeap(), 0, (cbData
+2)*sizeof(*value
));
275 ret
= RegQueryValueExW(hkey_main
, name
, NULL
, &type
, (BYTE
*)value
, &cbData
);
276 GLE
= GetLastError();
277 ok(ret
== ERROR_SUCCESS
, "RegQueryValueExW failed: %d, GLE=%d\n", ret
, GLE
);
280 /* When cbData == 0, RegQueryValueExW() should not modify the buffer */
283 ok(memcmp(value
, string
, cbData
) == 0, "RegQueryValueExW failed: %s/%d != %s/%d\n",
284 wine_debugstr_wn(value
, cbData
/ sizeof(WCHAR
)), cbData
,
285 wine_debugstr_wn(string
, full_byte_len
/ sizeof(WCHAR
)), full_byte_len
/ sizeof(WCHAR
));
286 HeapFree(GetProcessHeap(), 0, value
);
289 static void test_set_value(void)
293 static const WCHAR name1W
[] = {'C','l','e','a','n','S','i','n','g','l','e','S','t','r','i','n','g', 0};
294 static const WCHAR name2W
[] = {'S','o','m','e','I','n','t','r','a','Z','e','r','o','e','d','S','t','r','i','n','g', 0};
295 static const WCHAR emptyW
[] = {0};
296 static const WCHAR string1W
[] = {'T','h','i','s','N','e','v','e','r','B','r','e','a','k','s', 0};
297 static const WCHAR string2W
[] = {'T','h','i','s', 0 ,'B','r','e','a','k','s', 0 , 0 ,'A', 0 , 0 , 0 , 'L','o','t', 0 , 0 , 0 , 0, 0};
299 static const char name1A
[] = "CleanSingleString";
300 static const char name2A
[] = "SomeIntraZeroedString";
301 static const char emptyA
[] = "";
302 static const char string1A
[] = "ThisNeverBreaks";
303 static const char string2A
[] = "This\0Breaks\0\0A\0\0\0Lot\0\0\0\0";
305 /* Test RegSetValueExA with a 'zero-byte' string (as Office 2003 does).
306 * Surprisingly enough we're supposed to get zero bytes out of it.
307 * FIXME: Wine's on-disk file format does not differentiate this with
308 * regular empty strings but there's no way to test as it requires
309 * stopping the wineserver.
311 ret
= RegSetValueExA(hkey_main
, name1A
, 0, REG_SZ
, (const BYTE
*)emptyA
, 0);
312 ok(ret
== ERROR_SUCCESS
, "RegSetValueExA failed: %d, GLE=%d\n", ret
, GetLastError());
313 test_hkey_main_Value_A(name1A
, NULL
, 0);
314 test_hkey_main_Value_W(name1W
, NULL
, 0);
316 /* test RegSetValueExA with an empty string */
317 ret
= RegSetValueExA(hkey_main
, name1A
, 0, REG_SZ
, (const BYTE
*)emptyA
, sizeof(emptyA
));
318 ok(ret
== ERROR_SUCCESS
, "RegSetValueExA failed: %d, GLE=%d\n", ret
, GetLastError());
319 test_hkey_main_Value_A(name1A
, emptyA
, sizeof(emptyA
));
320 test_hkey_main_Value_W(name1W
, emptyW
, sizeof(emptyW
));
322 /* test RegSetValueExA with normal string */
323 ret
= RegSetValueExA(hkey_main
, name1A
, 0, REG_SZ
, (const BYTE
*)string1A
, sizeof(string1A
));
324 ok(ret
== ERROR_SUCCESS
, "RegSetValueExA failed: %d, GLE=%d\n", ret
, GetLastError());
325 test_hkey_main_Value_A(name1A
, string1A
, sizeof(string1A
));
326 test_hkey_main_Value_W(name1W
, string1W
, sizeof(string1W
));
328 /* test RegSetValueExA with intrazeroed string */
329 ret
= RegSetValueExA(hkey_main
, name2A
, 0, REG_SZ
, (const BYTE
*)string2A
, sizeof(string2A
));
330 ok(ret
== ERROR_SUCCESS
, "RegSetValueExA failed: %d, GLE=%d\n", ret
, GetLastError());
331 test_hkey_main_Value_A(name2A
, string2A
, sizeof(string2A
));
332 test_hkey_main_Value_W(name2W
, string2W
, sizeof(string2W
));
334 /* 9x doesn't support W-calls, so don't test them then */
335 if(GLE
== ERROR_CALL_NOT_IMPLEMENTED
) return;
337 /* test RegSetValueExW with normal string */
338 ret
= RegSetValueExW(hkey_main
, name1W
, 0, REG_SZ
, (const BYTE
*)string1W
, sizeof(string1W
));
339 ok(ret
== ERROR_SUCCESS
, "RegSetValueExW failed: %d, GLE=%d\n", ret
, GetLastError());
340 test_hkey_main_Value_A(name1A
, string1A
, sizeof(string1A
));
341 test_hkey_main_Value_W(name1W
, string1W
, sizeof(string1W
));
343 /* test RegSetValueExW with intrazeroed string */
344 ret
= RegSetValueExW(hkey_main
, name2W
, 0, REG_SZ
, (const BYTE
*)string2W
, sizeof(string2W
));
345 ok(ret
== ERROR_SUCCESS
, "RegSetValueExW failed: %d, GLE=%d\n", ret
, GetLastError());
346 test_hkey_main_Value_A(name2A
, string2A
, sizeof(string2A
));
347 test_hkey_main_Value_W(name2W
, string2W
, sizeof(string2W
));
350 static void create_test_entries(void)
352 static const DWORD qw
[2] = { 0x12345678, 0x87654321 };
354 SetEnvironmentVariableA("LONGSYSTEMVAR", "bar");
355 SetEnvironmentVariableA("FOO", "ImARatherLongButIndeedNeededString");
357 ok(!RegSetValueExA(hkey_main
,"TP1_EXP_SZ",0,REG_EXPAND_SZ
, (const BYTE
*)sTestpath1
, strlen(sTestpath1
)+1),
358 "RegSetValueExA failed\n");
359 ok(!RegSetValueExA(hkey_main
,"TP1_SZ",0,REG_SZ
, (const BYTE
*)sTestpath1
, strlen(sTestpath1
)+1),
360 "RegSetValueExA failed\n");
361 ok(!RegSetValueExA(hkey_main
,"TP2_EXP_SZ",0,REG_EXPAND_SZ
, (const BYTE
*)sTestpath2
, strlen(sTestpath2
)+1),
362 "RegSetValueExA failed\n");
363 ok(!RegSetValueExA(hkey_main
,"DWORD",0,REG_DWORD
, (const BYTE
*)qw
, 4),
364 "RegSetValueExA failed\n");
365 ok(!RegSetValueExA(hkey_main
,"BIN32",0,REG_BINARY
, (const BYTE
*)qw
, 4),
366 "RegSetValueExA failed\n");
367 ok(!RegSetValueExA(hkey_main
,"BIN64",0,REG_BINARY
, (const BYTE
*)qw
, 8),
368 "RegSetValueExA failed\n");
371 static void test_enum_value(void)
375 char value
[20], data
[20];
376 WCHAR valueW
[20], dataW
[20];
377 DWORD val_count
, data_count
, type
;
378 static const WCHAR foobarW
[] = {'f','o','o','b','a','r',0};
379 static const WCHAR testW
[] = {'T','e','s','t',0};
380 static const WCHAR xxxW
[] = {'x','x','x','x','x','x','x','x',0};
382 /* create the working key for new 'Test' value */
383 res
= RegCreateKeyA( hkey_main
, "TestKey", &test_key
);
384 ok( res
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", res
);
386 /* check NULL data with zero length */
387 res
= RegSetValueExA( test_key
, "Test", 0, REG_SZ
, NULL
, 0 );
388 if (GetVersion() & 0x80000000)
389 ok( res
== ERROR_INVALID_PARAMETER
, "RegSetValueExA returned %d\n", res
);
391 ok( !res
, "RegSetValueExA returned %d\n", res
);
392 res
= RegSetValueExA( test_key
, "Test", 0, REG_EXPAND_SZ
, NULL
, 0 );
393 ok( ERROR_SUCCESS
== res
|| ERROR_INVALID_PARAMETER
== res
, "RegSetValueExA returned %d\n", res
);
394 res
= RegSetValueExA( test_key
, "Test", 0, REG_BINARY
, NULL
, 0 );
395 ok( ERROR_SUCCESS
== res
|| ERROR_INVALID_PARAMETER
== res
, "RegSetValueExA returned %d\n", res
);
397 res
= RegSetValueExA( test_key
, "Test", 0, REG_SZ
, (const BYTE
*)"foobar", 7 );
398 ok( res
== 0, "RegSetValueExA failed error %d\n", res
);
400 /* overflow both name and data */
404 strcpy( value
, "xxxxxxxxxx" );
405 strcpy( data
, "xxxxxxxxxx" );
406 res
= RegEnumValueA( test_key
, 0, value
, &val_count
, NULL
, &type
, (LPBYTE
)data
, &data_count
);
407 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
408 ok( val_count
== 2, "val_count set to %d\n", val_count
);
409 ok( data_count
== 7, "data_count set to %d instead of 7\n", data_count
);
410 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
411 ok( !strcmp( value
, "xxxxxxxxxx" ), "value set to '%s'\n", value
);
412 ok( !strcmp( data
, "xxxxxxxxxx" ), "data set to '%s'\n", data
);
418 strcpy( value
, "xxxxxxxxxx" );
419 strcpy( data
, "xxxxxxxxxx" );
420 res
= RegEnumValueA( test_key
, 0, value
, &val_count
, NULL
, &type
, (LPBYTE
)data
, &data_count
);
421 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
422 /* Win9x returns 2 as specified by MSDN but NT returns 3... */
423 ok( val_count
== 2 || val_count
== 3, "val_count set to %d\n", val_count
);
424 ok( data_count
== 7, "data_count set to %d instead of 7\n", data_count
);
425 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
426 /* v5.1.2600.0 (XP Home and Proffesional) does not touch value or data in this case */
427 ok( !strcmp( value
, "Te" ) || !strcmp( value
, "xxxxxxxxxx" ),
428 "value set to '%s' instead of 'Te' or 'xxxxxxxxxx'\n", value
);
429 ok( !strcmp( data
, "foobar" ) || !strcmp( data
, "xxxxxxx" ),
430 "data set to '%s' instead of 'foobar' or 'xxxxxxx'\n", data
);
432 /* overflow empty name */
436 strcpy( value
, "xxxxxxxxxx" );
437 strcpy( data
, "xxxxxxxxxx" );
438 res
= RegEnumValueA( test_key
, 0, value
, &val_count
, NULL
, &type
, (LPBYTE
)data
, &data_count
);
439 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
440 ok( val_count
== 0, "val_count set to %d\n", val_count
);
441 ok( data_count
== 7, "data_count set to %d instead of 7\n", data_count
);
442 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
443 ok( !strcmp( value
, "xxxxxxxxxx" ), "value set to '%s'\n", value
);
444 /* v5.1.2600.0 (XP Home and Professional) does not touch data in this case */
445 ok( !strcmp( data
, "foobar" ) || !strcmp( data
, "xxxxxxx" ),
446 "data set to '%s' instead of 'foobar' or 'xxxxxxx'\n", data
);
452 strcpy( value
, "xxxxxxxxxx" );
453 strcpy( data
, "xxxxxxxxxx" );
454 res
= RegEnumValueA( test_key
, 0, value
, &val_count
, NULL
, &type
, (LPBYTE
)data
, &data_count
);
455 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
456 ok( val_count
== 20, "val_count set to %d\n", val_count
);
457 ok( data_count
== 7, "data_count set to %d instead of 7\n", data_count
);
458 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
459 ok( !strcmp( value
, "xxxxxxxxxx" ), "value set to '%s'\n", value
);
460 ok( !strcmp( data
, "xxxxxxxxxx" ), "data set to '%s'\n", data
);
466 strcpy( value
, "xxxxxxxxxx" );
467 strcpy( data
, "xxxxxxxxxx" );
468 res
= RegEnumValueA( test_key
, 0, value
, &val_count
, NULL
, &type
, (LPBYTE
)data
, &data_count
);
469 ok( res
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", res
);
470 ok( val_count
== 4, "val_count set to %d instead of 4\n", val_count
);
471 ok( data_count
== 7, "data_count set to %d instead of 7\n", data_count
);
472 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
473 ok( !strcmp( value
, "Test" ), "value is '%s' instead of Test\n", value
);
474 ok( !strcmp( data
, "foobar" ), "data is '%s' instead of foobar\n", data
);
478 SetLastError(0xdeadbeef);
479 res
= RegSetValueExW( test_key
, testW
, 0, REG_SZ
, (const BYTE
*)foobarW
, 7*sizeof(WCHAR
) );
480 if (res
==0 && GetLastError()==ERROR_CALL_NOT_IMPLEMENTED
)
482 skip("RegSetValueExW is not implemented\n");
485 ok( res
== 0, "RegSetValueExW failed error %d\n", res
);
487 /* overflow both name and data */
491 memcpy( valueW
, xxxW
, sizeof(xxxW
) );
492 memcpy( dataW
, xxxW
, sizeof(xxxW
) );
493 res
= RegEnumValueW( test_key
, 0, valueW
, &val_count
, NULL
, &type
, (BYTE
*)dataW
, &data_count
);
494 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
495 ok( val_count
== 2, "val_count set to %d\n", val_count
);
496 ok( data_count
== 7*sizeof(WCHAR
), "data_count set to %d instead of 7*sizeof(WCHAR)\n", data_count
);
497 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
498 ok( !memcmp( valueW
, xxxW
, sizeof(xxxW
) ), "value modified\n" );
499 ok( !memcmp( dataW
, xxxW
, sizeof(xxxW
) ), "data modified\n" );
505 memcpy( valueW
, xxxW
, sizeof(xxxW
) );
506 memcpy( dataW
, xxxW
, sizeof(xxxW
) );
507 res
= RegEnumValueW( test_key
, 0, valueW
, &val_count
, NULL
, &type
, (BYTE
*)dataW
, &data_count
);
508 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
509 ok( val_count
== 3, "val_count set to %d\n", val_count
);
510 ok( data_count
== 7*sizeof(WCHAR
), "data_count set to %d instead of 7*sizeof(WCHAR)\n", data_count
);
511 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
512 ok( !memcmp( valueW
, xxxW
, sizeof(xxxW
) ), "value modified\n" );
513 ok( !memcmp( dataW
, xxxW
, sizeof(xxxW
) ), "data modified\n" );
519 memcpy( valueW
, xxxW
, sizeof(xxxW
) );
520 memcpy( dataW
, xxxW
, sizeof(xxxW
) );
521 res
= RegEnumValueW( test_key
, 0, valueW
, &val_count
, NULL
, &type
, (BYTE
*)dataW
, &data_count
);
522 ok( res
== ERROR_MORE_DATA
, "expected ERROR_MORE_DATA, got %d\n", res
);
523 ok( val_count
== 4, "val_count set to %d instead of 4\n", val_count
);
524 ok( data_count
== 7*sizeof(WCHAR
), "data_count set to %d instead of 7*sizeof(WCHAR)\n", data_count
);
525 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
526 ok( !memcmp( valueW
, testW
, sizeof(testW
) ), "value is not 'Test'\n" );
527 ok( !memcmp( dataW
, xxxW
, sizeof(xxxW
) ), "data modified\n" );
533 memcpy( valueW
, xxxW
, sizeof(xxxW
) );
534 memcpy( dataW
, xxxW
, sizeof(xxxW
) );
535 res
= RegEnumValueW( test_key
, 0, valueW
, &val_count
, NULL
, &type
, (BYTE
*)dataW
, &data_count
);
536 ok( res
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", res
);
537 ok( val_count
== 4, "val_count set to %d instead of 4\n", val_count
);
538 ok( data_count
== 7*sizeof(WCHAR
), "data_count set to %d instead of 7*sizeof(WCHAR)\n", data_count
);
539 ok( type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
540 ok( !memcmp( valueW
, testW
, sizeof(testW
) ), "value is not 'Test'\n" );
541 ok( !memcmp( dataW
, foobarW
, sizeof(foobarW
) ), "data is not 'foobar'\n" );
544 RegDeleteKeyA(test_key
, "");
545 RegCloseKey(test_key
);
548 static void test_query_value_ex(void)
555 ret
= RegQueryValueExA(hkey_main
, "TP1_SZ", NULL
, &type
, NULL
, &size
);
556 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
557 ok(size
== strlen(sTestpath1
) + 1, "(%d,%d)\n", (DWORD
)strlen(sTestpath1
) + 1, size
);
558 ok(type
== REG_SZ
, "type %d is not REG_SZ\n", type
);
562 ret
= RegQueryValueExA(HKEY_CLASSES_ROOT
, "Nonexistent Value", NULL
, &type
, NULL
, &size
);
563 ok(ret
== ERROR_FILE_NOT_FOUND
, "expected ERROR_FILE_NOT_FOUND, got %d\n", ret
);
564 /* the type parameter is cleared on Win9x, but is set to a random value on
565 * NT, so don't do that test there. The size parameter is left untouched on Win9x
566 * but cleared on NT+, this can be tested on all platforms.
568 if (GetVersion() & 0x80000000)
570 ok(type
== 0, "type should have been set to 0 instead of 0x%x\n", type
);
571 ok(size
== 0xdeadbeef, "size should have been left untouched (0xdeadbeef)\n");
575 trace("test_query_value_ex: type set to: 0x%08x\n", type
);
576 ok(size
== 0, "size should have been set to 0 instead of %d\n", size
);
579 size
= sizeof(buffer
);
580 ret
= RegQueryValueExA(HKEY_CLASSES_ROOT
, "Nonexistent Value", NULL
, &type
, buffer
, &size
);
581 ok(ret
== ERROR_FILE_NOT_FOUND
, "expected ERROR_FILE_NOT_FOUND, got %d\n", ret
);
582 ok(size
== sizeof(buffer
), "size shouldn't have been changed to %d\n", size
);
585 static void test_get_value(void)
592 CHAR expanded
[] = "bar\\subdir1";
596 skip("RegGetValue not available on this platform\n");
600 /* Invalid parameter */
601 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_REG_DWORD
, &type
, &dw
, NULL
);
602 ok(ret
== ERROR_INVALID_PARAMETER
, "ret=%d\n", ret
);
604 /* Query REG_DWORD using RRF_RT_REG_DWORD (ok) */
605 size
= type
= dw
= 0xdeadbeef;
606 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_REG_DWORD
, &type
, &dw
, &size
);
607 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
608 ok(size
== 4, "size=%d\n", size
);
609 ok(type
== REG_DWORD
, "type=%d\n", type
);
610 ok(dw
== 0x12345678, "dw=%d\n", dw
);
612 /* Query by subkey-name */
613 ret
= pRegGetValueA(HKEY_CURRENT_USER
, "Software\\Wine\\Test", "DWORD", RRF_RT_REG_DWORD
, NULL
, NULL
, NULL
);
614 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
616 /* Query REG_DWORD using RRF_RT_REG_BINARY (restricted) */
617 size
= type
= dw
= 0xdeadbeef;
618 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_REG_BINARY
, &type
, &dw
, &size
);
619 ok(ret
== ERROR_UNSUPPORTED_TYPE
, "ret=%d\n", ret
);
620 /* Although the function failed all values are retrieved */
621 ok(size
== 4, "size=%d\n", size
);
622 ok(type
== REG_DWORD
, "type=%d\n", type
);
623 ok(dw
== 0x12345678, "dw=%d\n", dw
);
625 /* Test RRF_ZEROONFAILURE */
626 type
= dw
= 0xdeadbeef; size
= 4;
627 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_REG_SZ
|RRF_ZEROONFAILURE
, &type
, &dw
, &size
);
628 ok(ret
== ERROR_UNSUPPORTED_TYPE
, "ret=%d\n", ret
);
629 /* Again all values are retrieved ... */
630 ok(size
== 4, "size=%d\n", size
);
631 ok(type
== REG_DWORD
, "type=%d\n", type
);
632 /* ... except the buffer, which is zeroed out */
633 ok(dw
== 0, "dw=%d\n", dw
);
635 /* Test RRF_ZEROONFAILURE with a NULL buffer... */
636 type
= size
= 0xbadbeef;
637 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_REG_SZ
|RRF_ZEROONFAILURE
, &type
, NULL
, &size
);
638 ok(ret
== ERROR_UNSUPPORTED_TYPE
, "ret=%d\n", ret
);
639 ok(size
== 4, "size=%d\n", size
);
640 ok(type
== REG_DWORD
, "type=%d\n", type
);
642 /* Query REG_DWORD using RRF_RT_DWORD (ok) */
643 size
= type
= dw
= 0xdeadbeef;
644 ret
= pRegGetValueA(hkey_main
, NULL
, "DWORD", RRF_RT_DWORD
, &type
, &dw
, &size
);
645 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
646 ok(size
== 4, "size=%d\n", size
);
647 ok(type
== REG_DWORD
, "type=%d\n", type
);
648 ok(dw
== 0x12345678, "dw=%d\n", dw
);
650 /* Query 32-bit REG_BINARY using RRF_RT_DWORD (ok) */
651 size
= type
= dw
= 0xdeadbeef;
652 ret
= pRegGetValueA(hkey_main
, NULL
, "BIN32", RRF_RT_DWORD
, &type
, &dw
, &size
);
653 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
654 ok(size
== 4, "size=%d\n", size
);
655 ok(type
== REG_BINARY
, "type=%d\n", type
);
656 ok(dw
== 0x12345678, "dw=%d\n", dw
);
658 /* Query 64-bit REG_BINARY using RRF_RT_DWORD (type mismatch) */
659 qw
[0] = qw
[1] = size
= type
= 0xdeadbeef;
660 ret
= pRegGetValueA(hkey_main
, NULL
, "BIN64", RRF_RT_DWORD
, &type
, qw
, &size
);
661 ok(ret
== ERROR_DATATYPE_MISMATCH
, "ret=%d\n", ret
);
662 ok(size
== 8, "size=%d\n", size
);
663 ok(type
== REG_BINARY
, "type=%d\n", type
);
664 ok(qw
[0] == 0x12345678 &&
665 qw
[1] == 0x87654321, "qw={%d,%d}\n", qw
[0], qw
[1]);
667 /* Query 64-bit REG_BINARY using 32-bit buffer (buffer too small) */
668 type
= dw
= 0xdeadbeef; size
= 4;
669 ret
= pRegGetValueA(hkey_main
, NULL
, "BIN64", RRF_RT_REG_BINARY
, &type
, &dw
, &size
);
670 ok(ret
== ERROR_MORE_DATA
, "ret=%d\n", ret
);
671 ok(dw
== 0xdeadbeef, "dw=%d\n", dw
);
672 ok(size
== 8, "size=%d\n", size
);
674 /* Query 64-bit REG_BINARY using RRF_RT_QWORD (ok) */
675 qw
[0] = qw
[1] = size
= type
= 0xdeadbeef;
676 ret
= pRegGetValueA(hkey_main
, NULL
, "BIN64", RRF_RT_QWORD
, &type
, qw
, &size
);
677 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
678 ok(size
== 8, "size=%d\n", size
);
679 ok(type
== REG_BINARY
, "type=%d\n", type
);
680 ok(qw
[0] == 0x12345678 &&
681 qw
[1] == 0x87654321, "qw={%d,%d}\n", qw
[0], qw
[1]);
683 /* Query REG_SZ using RRF_RT_REG_SZ (ok) */
684 buf
[0] = 0; type
= 0xdeadbeef; size
= sizeof(buf
);
685 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_SZ", RRF_RT_REG_SZ
, &type
, buf
, &size
);
686 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
687 ok(size
== strlen(sTestpath1
)+1, "strlen(sTestpath1)=%d size=%d\n", lstrlenA(sTestpath1
), size
);
688 ok(type
== REG_SZ
, "type=%d\n", type
);
689 ok(!strcmp(sTestpath1
, buf
), "sTestpath=\"%s\" buf=\"%s\"\n", sTestpath1
, buf
);
691 /* Query REG_SZ using RRF_RT_REG_SZ and no buffer (ok) */
692 type
= 0xdeadbeef; size
= 0;
693 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_SZ", RRF_RT_REG_SZ
, &type
, NULL
, &size
);
694 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
695 /* v5.2.3790.1830 (2003 SP1) returns sTestpath1 length + 2 here. */
696 ok(size
== strlen(sTestpath1
)+1 || size
== strlen(sTestpath1
)+2,
697 "strlen(sTestpath1)=%d size=%d\n", lstrlenA(sTestpath1
), size
);
698 ok(type
== REG_SZ
, "type=%d\n", type
);
700 /* Query REG_SZ using RRF_RT_REG_SZ|RRF_NOEXPAND (ok) */
701 buf
[0] = 0; type
= 0xdeadbeef; size
= sizeof(buf
);
702 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_SZ", RRF_RT_REG_SZ
|RRF_NOEXPAND
, &type
, buf
, &size
);
703 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
704 ok(size
== strlen(sTestpath1
)+1, "strlen(sTestpath1)=%d size=%d\n", lstrlenA(sTestpath1
), size
);
705 ok(type
== REG_SZ
, "type=%d\n", type
);
706 ok(!strcmp(sTestpath1
, buf
), "sTestpath=\"%s\" buf=\"%s\"\n", sTestpath1
, buf
);
708 /* Query REG_EXPAND_SZ using RRF_RT_REG_SZ and no buffer (ok, expands) */
710 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_SZ
, NULL
, NULL
, &size
);
711 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
712 /* At least v5.2.3790.1830 (2003 SP1) returns the unexpanded sTestpath1 length + 1 here. */
713 ok((size
== strlen(expanded
)+1) || (size
== strlen(sTestpath1
)+1),
714 "strlen(expanded)=%d, strlen(sTestpath1)=%d, size=%d\n", lstrlenA(expanded
), lstrlenA(sTestpath1
), size
);
716 /* Query REG_EXPAND_SZ using RRF_RT_REG_SZ (ok, expands) */
717 buf
[0] = 0; type
= 0xdeadbeef; size
= sizeof(buf
);
718 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_SZ
, &type
, buf
, &size
);
719 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
720 /* At least v5.2.3790.1830 (2003 SP1) returns the unexpanded sTestpath1 length + 1 here. */
721 ok((size
== strlen(expanded
)+1) || (size
== strlen(sTestpath1
)+1),
722 "strlen(expanded)=%d, strlen(sTestpath1)=%d, size=%d\n", lstrlenA(expanded
), lstrlenA(sTestpath1
), size
);
723 ok(type
== REG_SZ
, "type=%d\n", type
);
724 ok(!strcmp(expanded
, buf
), "expanded=\"%s\" buf=\"%s\"\n", expanded
, buf
);
726 /* Query REG_EXPAND_SZ using RRF_RT_REG_EXPAND_SZ|RRF_NOEXPAND (ok, doesn't expand) */
727 buf
[0] = 0; type
= 0xdeadbeef; size
= sizeof(buf
);
728 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_EXPAND_SZ
|RRF_NOEXPAND
, &type
, buf
, &size
);
729 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
730 ok(size
== strlen(sTestpath1
)+1, "strlen(sTestpath1)=%d size=%d\n", lstrlenA(sTestpath1
), size
);
731 ok(type
== REG_EXPAND_SZ
, "type=%d\n", type
);
732 ok(!strcmp(sTestpath1
, buf
), "sTestpath=\"%s\" buf=\"%s\"\n", sTestpath1
, buf
);
734 /* Query REG_EXPAND_SZ using RRF_RT_REG_EXPAND_SZ|RRF_NOEXPAND and no buffer (ok, doesn't expand) */
736 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_EXPAND_SZ
|RRF_NOEXPAND
, NULL
, NULL
, &size
);
737 ok(ret
== ERROR_SUCCESS
, "ret=%d\n", ret
);
738 /* v5.2.3790.1830 (2003 SP1) returns sTestpath1 length + 2 here. */
739 ok(size
== strlen(sTestpath1
)+1 || size
== strlen(sTestpath1
)+2,
740 "strlen(sTestpath1)=%d size=%d\n", lstrlenA(sTestpath1
), size
);
742 /* Query REG_EXPAND_SZ using RRF_RT_REG_SZ|RRF_NOEXPAND (type mismatch) */
743 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_SZ
|RRF_NOEXPAND
, NULL
, NULL
, NULL
);
744 ok(ret
== ERROR_UNSUPPORTED_TYPE
, "ret=%d\n", ret
);
746 /* Query REG_EXPAND_SZ using RRF_RT_REG_EXPAND_SZ (not allowed without RRF_NOEXPAND) */
747 ret
= pRegGetValueA(hkey_main
, NULL
, "TP1_EXP_SZ", RRF_RT_REG_EXPAND_SZ
, NULL
, NULL
, NULL
);
748 ok(ret
== ERROR_INVALID_PARAMETER
, "ret=%d\n", ret
);
751 static void test_reg_open_key(void)
754 HKEY hkResult
= NULL
;
755 HKEY hkPreserve
= NULL
;
757 /* successful open */
758 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkResult
);
759 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
760 ok(hkResult
!= NULL
, "expected hkResult != NULL\n");
761 hkPreserve
= hkResult
;
763 /* these tests fail on Win9x, but we want to be compatible with NT, so
764 * run them if we can */
765 if (!(GetVersion() & 0x80000000))
767 /* open same key twice */
768 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkResult
);
769 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
770 ok(hkResult
!= hkPreserve
, "epxected hkResult != hkPreserve\n");
771 ok(hkResult
!= NULL
, "hkResult != NULL\n");
772 RegCloseKey(hkResult
);
774 /* open nonexistent key
775 * check that hkResult is set to NULL
777 hkResult
= hkPreserve
;
778 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Nonexistent", &hkResult
);
779 ok(ret
== ERROR_FILE_NOT_FOUND
, "expected ERROR_FILE_NOT_FOUND, got %d\n", ret
);
780 ok(hkResult
== NULL
, "expected hkResult == NULL\n");
782 /* open the same nonexistent key again to make sure the key wasn't created */
783 hkResult
= hkPreserve
;
784 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Nonexistent", &hkResult
);
785 ok(ret
== ERROR_FILE_NOT_FOUND
, "expected ERROR_FILE_NOT_FOUND, got %d\n", ret
);
786 ok(hkResult
== NULL
, "expected hkResult == NULL\n");
788 /* send in NULL lpSubKey
789 * check that hkResult receives the value of hKey
791 hkResult
= hkPreserve
;
792 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, NULL
, &hkResult
);
793 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
794 ok(hkResult
== HKEY_CURRENT_USER
, "expected hkResult == HKEY_CURRENT_USER\n");
796 /* send empty-string in lpSubKey */
797 hkResult
= hkPreserve
;
798 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "", &hkResult
);
799 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
800 ok(hkResult
== HKEY_CURRENT_USER
, "expected hkResult == HKEY_CURRENT_USER\n");
802 /* send in NULL lpSubKey and NULL hKey
803 * hkResult is set to NULL
805 hkResult
= hkPreserve
;
806 ret
= RegOpenKeyA(NULL
, NULL
, &hkResult
);
807 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
808 ok(hkResult
== NULL
, "expected hkResult == NULL\n");
811 /* only send NULL hKey
812 * the value of hkResult remains unchanged
814 hkResult
= hkPreserve
;
815 ret
= RegOpenKeyA(NULL
, "Software\\Wine\\Test", &hkResult
);
816 ok(ret
== ERROR_INVALID_HANDLE
|| ret
== ERROR_BADKEY
, /* Windows 95 returns BADKEY */
817 "expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", ret
);
818 ok(hkResult
== hkPreserve
, "expected hkResult == hkPreserve\n");
819 RegCloseKey(hkResult
);
821 /* send in NULL hkResult */
822 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Test", NULL
);
823 ok(ret
== ERROR_INVALID_PARAMETER
, "expected ERROR_INVALID_PARAMETER, got %d\n", ret
);
825 /* beginning backslash character */
826 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "\\Software\\Wine\\Test", &hkResult
);
827 ok(ret
== ERROR_BAD_PATHNAME
|| /* NT/2k/XP */
828 ret
== ERROR_FILE_NOT_FOUND
/* Win9x,ME */
829 , "expected ERROR_BAD_PATHNAME or ERROR_FILE_NOT_FOUND, got %d\n", ret
);
832 static void test_reg_create_key(void)
836 ret
= RegCreateKeyExA(hkey_main
, "Subkey1", 0, NULL
, 0, KEY_NOTIFY
, NULL
, &hkey1
, NULL
);
837 ok(!ret
, "RegCreateKeyExA failed with error %d\n", ret
);
838 /* should succeed: all versions of Windows ignore the access rights
839 * to the parent handle */
840 ret
= RegCreateKeyExA(hkey1
, "Subkey2", 0, NULL
, 0, KEY_SET_VALUE
, NULL
, &hkey2
, NULL
);
841 ok(!ret
, "RegCreateKeyExA failed with error %d\n", ret
);
844 RegDeleteKey(hkey2
, "");
845 RegDeleteKey(hkey1
, "");
847 /* beginning backslash character */
848 ret
= RegCreateKeyExA(hkey_main
, "\\Subkey3", 0, NULL
, 0, KEY_NOTIFY
, NULL
, &hkey1
, NULL
);
849 if (!(GetVersion() & 0x80000000))
850 ok(ret
== ERROR_BAD_PATHNAME
, "expected ERROR_BAD_PATHNAME, got %d\n", ret
);
852 ok(!ret
, "RegCreateKeyExA failed with error %d\n", ret
);
853 RegDeleteKey(hkey1
, NULL
);
857 static void test_reg_close_key(void)
862 /* successfully close key
863 * hkHandle remains changed after call to RegCloseKey
865 ret
= RegOpenKeyA(HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkHandle
);
866 ret
= RegCloseKey(hkHandle
);
867 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
869 /* try to close the key twice */
870 ret
= RegCloseKey(hkHandle
); /* Windows 95 doesn't mind. */
871 ok(ret
== ERROR_INVALID_HANDLE
|| ret
== ERROR_SUCCESS
,
872 "expected ERROR_INVALID_HANDLE or ERROR_SUCCESS, got %d\n", ret
);
874 /* try to close a NULL handle */
875 ret
= RegCloseKey(NULL
);
876 ok(ret
== ERROR_INVALID_HANDLE
|| ret
== ERROR_BADKEY
, /* Windows 95 returns BADKEY */
877 "expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", ret
);
879 /* Check to see if we didn't potentially close our main handle, which could happen on win98 as
880 * win98 doesn't give a new handle when the same key is opened.
881 * Not re-opening will make some next tests fail.
883 if (hkey_main
== hkHandle
)
885 trace("The main handle is most likely closed, so re-opening\n");
886 RegOpenKeyA( HKEY_CURRENT_USER
, "Software\\Wine\\Test", &hkey_main
);
890 static void test_reg_delete_key(void)
894 ret
= RegDeleteKey(hkey_main
, NULL
);
896 /* There is a bug in NT4 and W2K that doesn't check if the subkey is NULL. If
897 * there are also no subkeys available it will delete the key pointed to by hkey_main.
898 * Not re-creating will make some next tests fail.
900 if (ret
== ERROR_SUCCESS
)
902 trace("We are probably running on NT4 or W2K as the main key is deleted,"
903 " re-creating the main key\n");
907 ok(ret
== ERROR_INVALID_PARAMETER
||
908 ret
== ERROR_ACCESS_DENIED
||
909 ret
== ERROR_BADKEY
, /* Win95 */
913 static void test_reg_save_key(void)
917 ret
= RegSaveKey(hkey_main
, "saved_key", NULL
);
918 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
921 static void test_reg_load_key(void)
926 ret
= RegLoadKey(HKEY_LOCAL_MACHINE
, "Test", "saved_key");
927 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
929 ret
= RegOpenKey(HKEY_LOCAL_MACHINE
, "Test", &hkHandle
);
930 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
932 RegCloseKey(hkHandle
);
935 static void test_reg_unload_key(void)
939 ret
= RegUnLoadKey(HKEY_LOCAL_MACHINE
, "Test");
940 ok(ret
== ERROR_SUCCESS
, "expected ERROR_SUCCESS, got %d\n", ret
);
942 DeleteFile("saved_key");
943 DeleteFile("saved_key.LOG");
946 static BOOL
set_privileges(LPCSTR privilege
, BOOL set
)
952 if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES
, &hToken
))
955 if(!LookupPrivilegeValue(NULL
, privilege
, &luid
))
961 tp
.PrivilegeCount
= 1;
962 tp
.Privileges
[0].Luid
= luid
;
965 tp
.Privileges
[0].Attributes
= SE_PRIVILEGE_ENABLED
;
967 tp
.Privileges
[0].Attributes
= 0;
969 AdjustTokenPrivileges(hToken
, FALSE
, &tp
, sizeof(TOKEN_PRIVILEGES
), NULL
, NULL
);
970 if (GetLastError() != ERROR_SUCCESS
)
980 /* tests that show that RegConnectRegistry and
981 OpenSCManager accept computer names without the
982 \\ prefix (what MSDN says). */
983 static void test_regconnectregistry( void)
985 CHAR compName
[MAX_COMPUTERNAME_LENGTH
+ 1];
986 CHAR netwName
[MAX_COMPUTERNAME_LENGTH
+ 3]; /* 2 chars for double backslash */
987 DWORD len
= sizeof(compName
) ;
993 SetLastError(0xdeadbeef);
994 ret
= GetComputerNameA(compName
, &len
);
995 ok( ret
, "GetComputerName failed err = %d\n", GetLastError());
998 lstrcpyA(netwName
, "\\\\");
999 lstrcpynA(netwName
+2, compName
, MAX_COMPUTERNAME_LENGTH
+ 1);
1001 retl
= RegConnectRegistryA( compName
, HKEY_LOCAL_MACHINE
, &hkey
);
1002 ok( !retl
|| retl
== ERROR_DLL_INIT_FAILED
, "RegConnectRegistryA failed err = %d\n", retl
);
1003 if( !retl
) RegCloseKey( hkey
);
1005 retl
= RegConnectRegistryA( netwName
, HKEY_LOCAL_MACHINE
, &hkey
);
1006 ok( !retl
|| retl
== ERROR_DLL_INIT_FAILED
, "RegConnectRegistryA failed err = %d\n", retl
);
1007 if( !retl
) RegCloseKey( hkey
);
1009 SetLastError(0xdeadbeef);
1010 schnd
= OpenSCManagerA( compName
, NULL
, GENERIC_READ
);
1011 if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED
)
1013 skip("OpenSCManagerA is not implemented\n");
1017 ok( schnd
!= NULL
, "OpenSCManagerA failed err = %d\n", GetLastError());
1018 CloseServiceHandle( schnd
);
1020 SetLastError(0xdeadbeef);
1021 schnd
= OpenSCManagerA( netwName
, NULL
, GENERIC_READ
);
1022 ok( schnd
!= NULL
, "OpenSCManagerA failed err = %d\n", GetLastError());
1023 CloseServiceHandle( schnd
);
1027 static void test_reg_query_value(void)
1034 static const WCHAR expected
[] = {'d','a','t','a',0};
1036 ret
= RegCreateKeyA(hkey_main
, "subkey", &subkey
);
1037 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1039 ret
= RegSetValueA(subkey
, NULL
, REG_SZ
, "data", 4);
1040 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1042 /* try an invalid hkey */
1043 SetLastError(0xdeadbeef);
1045 ret
= RegQueryValueA((HKEY
)0xcafebabe, "subkey", val
, &size
);
1046 ok(ret
== ERROR_INVALID_HANDLE
|| ret
== ERROR_BADKEY
, /* Windows 98 returns BADKEY */
1047 "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", ret
);
1048 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1050 /* try a NULL hkey */
1051 SetLastError(0xdeadbeef);
1053 ret
= RegQueryValueA(NULL
, "subkey", val
, &size
);
1054 ok(ret
== ERROR_INVALID_HANDLE
|| ret
== ERROR_BADKEY
, /* Windows 98 returns BADKEY */
1055 "Expected ERROR_INVALID_HANDLE or ERROR_BADKEY, got %d\n", ret
);
1056 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1058 /* try a NULL value */
1060 ret
= RegQueryValueA(hkey_main
, "subkey", NULL
, &size
);
1061 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1062 ok(size
== 5, "Expected 5, got %d\n", size
);
1064 /* try a NULL size */
1065 SetLastError(0xdeadbeef);
1067 ret
= RegQueryValueA(hkey_main
, "subkey", val
, NULL
);
1068 ok(ret
== ERROR_INVALID_PARAMETER
, "Expected ERROR_INVALID_PARAMETER, got %d\n", ret
);
1069 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1070 ok(lstrlenA(val
) == 0, "Expected val to be untouched, got %s\n", val
);
1072 /* try a NULL value and size */
1073 ret
= RegQueryValueA(hkey_main
, "subkey", NULL
, NULL
);
1074 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1076 /* try a size too small */
1077 SetLastError(0xdeadbeef);
1080 ret
= RegQueryValueA(hkey_main
, "subkey", val
, &size
);
1081 ok(ret
== ERROR_MORE_DATA
, "Expected ERROR_MORE_DATA, got %d\n", ret
);
1082 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1083 ok(lstrlenA(val
) == 0, "Expected val to be untouched, got %s\n", val
);
1084 ok(size
== 5, "Expected 5, got %d\n", size
);
1086 /* successfully read the value using 'subkey' */
1088 ret
= RegQueryValueA(hkey_main
, "subkey", val
, &size
);
1089 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1090 ok(!lstrcmpA(val
, "data"), "Expected 'data', got '%s'\n", val
);
1091 ok(size
== 5, "Expected 5, got %d\n", size
);
1093 /* successfully read the value using the subkey key */
1095 ret
= RegQueryValueA(subkey
, NULL
, val
, &size
);
1096 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1097 ok(!lstrcmpA(val
, "data"), "Expected 'data', got '%s'\n", val
);
1098 ok(size
== 5, "Expected 5, got %d\n", size
);
1100 /* unicode - try size too small */
1101 SetLastError(0xdeadbeef);
1104 ret
= RegQueryValueW(subkey
, NULL
, valW
, &size
);
1105 if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED
)
1107 skip("RegQueryValueW is not implemented\n");
1110 ok(ret
== ERROR_MORE_DATA
, "Expected ERROR_MORE_DATA, got %d\n", ret
);
1111 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1112 ok(lstrlenW(valW
) == 0, "Expected valW to be untouched\n");
1113 ok(size
== sizeof(expected
), "Got wrong size: %d\n", size
);
1115 /* unicode - try size in WCHARS */
1116 SetLastError(0xdeadbeef);
1117 size
= sizeof(valW
) / sizeof(WCHAR
);
1118 ret
= RegQueryValueW(subkey
, NULL
, valW
, &size
);
1119 ok(ret
== ERROR_MORE_DATA
, "Expected ERROR_MORE_DATA, got %d\n", ret
);
1120 ok(GetLastError() == 0xdeadbeef, "Expected 0xdeadbeef, got %d\n", GetLastError());
1121 ok(lstrlenW(valW
) == 0, "Expected valW to be untouched\n");
1122 ok(size
== sizeof(expected
), "Got wrong size: %d\n", size
);
1124 /* unicode - successfully read the value */
1125 size
= sizeof(valW
);
1126 ret
= RegQueryValueW(subkey
, NULL
, valW
, &size
);
1127 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1128 ok(!lstrcmpW(valW
, expected
), "Got wrong value\n");
1129 ok(size
== sizeof(expected
), "Got wrong size: %d\n", size
);
1131 /* unicode - set the value without a NULL terminator */
1132 ret
= RegSetValueW(subkey
, NULL
, REG_SZ
, expected
, sizeof(expected
)-sizeof(WCHAR
));
1133 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1135 /* unicode - read the unterminated value, value is terminated for us */
1136 memset(valW
, 'a', sizeof(valW
));
1137 size
= sizeof(valW
);
1138 ret
= RegQueryValueW(subkey
, NULL
, valW
, &size
);
1139 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1140 ok(!lstrcmpW(valW
, expected
), "Got wrong value\n");
1141 ok(size
== sizeof(expected
), "Got wrong size: %d\n", size
);
1144 RegDeleteKeyA(subkey
, "");
1145 RegCloseKey(subkey
);
1148 static void test_reg_delete_tree(void)
1150 CHAR buffer
[MAX_PATH
];
1151 HKEY subkey
, subkey2
;
1154 if(!pRegDeleteTreeA
) {
1155 skip("Skipping RegDeleteTreeA tests, function not present\n");
1159 ret
= RegCreateKeyA(hkey_main
, "subkey", &subkey
);
1160 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1161 ret
= RegCreateKeyA(subkey
, "subkey2", &subkey2
);
1162 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1163 ret
= RegSetValueA(subkey
, NULL
, REG_SZ
, "data", 4);
1164 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1165 ret
= RegSetValueA(subkey2
, NULL
, REG_SZ
, "data2", 5);
1166 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1167 ret
= RegCloseKey(subkey2
);
1168 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1170 ret
= pRegDeleteTreeA(subkey
, "subkey2");
1171 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1172 ok(RegOpenKeyA(subkey
, "subkey2", &subkey2
),
1173 "subkey2 was not deleted\n");
1175 ok(!RegQueryValueA(subkey
, NULL
, buffer
, &size
),
1176 "Default value of subkey not longer present\n");
1178 ret
= RegCreateKeyA(subkey
, "subkey2", &subkey2
);
1179 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1180 ret
= RegCloseKey(subkey2
);
1181 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1182 ret
= pRegDeleteTreeA(hkey_main
, "subkey\\subkey2");
1183 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1184 ok(RegOpenKeyA(subkey
, "subkey2", &subkey2
),
1185 "subkey2 was not deleted\n");
1186 ok(!RegQueryValueA(subkey
, NULL
, buffer
, &size
),
1187 "Default value of subkey not longer present\n");
1189 ret
= RegCreateKeyA(subkey
, "subkey2", &subkey2
);
1190 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1191 ret
= RegCloseKey(subkey2
);
1192 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1193 ret
= RegCreateKeyA(subkey
, "subkey3", &subkey2
);
1194 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1195 ret
= RegCloseKey(subkey2
);
1196 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1197 ret
= RegSetValueA(subkey
, "value", REG_SZ
, "data2", 5);
1198 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1199 ret
= pRegDeleteTreeA(subkey
, NULL
);
1200 ok(ret
== ERROR_SUCCESS
, "Expected ERROR_SUCCESS, got %d\n", ret
);
1201 ok(!RegOpenKeyA(hkey_main
, "subkey", &subkey
),
1202 "subkey was deleted\n");
1203 ok(RegOpenKeyA(subkey
, "subkey2", &subkey2
),
1204 "subkey2 was not deleted\n");
1205 ok(RegOpenKeyA(subkey
, "subkey3", &subkey2
),
1206 "subkey3 was not deleted\n");
1208 ret
= RegQueryValueA(subkey
, NULL
, buffer
, &size
);
1209 ok(ret
== ERROR_SUCCESS
,
1210 "Default value of subkey is not present\n");
1211 ok(!lstrlenA(buffer
),
1212 "Expected length 0 got length %u(%s)\n", lstrlenA(buffer
), buffer
);
1214 ok(RegQueryValueA(subkey
, "value", buffer
, &size
),
1215 "Value is still present\n");
1217 ret
= pRegDeleteTreeA(hkey_main
, "not-here");
1218 ok(ret
== ERROR_FILE_NOT_FOUND
,
1219 "Expected ERROR_FILE_NOT_FOUND, got %d\n", ret
);
1222 START_TEST(registry
)
1224 /* Load pointers for functions that are not available in all Windows versions */
1229 create_test_entries();
1231 test_query_value_ex();
1233 test_reg_open_key();
1234 test_reg_create_key();
1235 test_reg_close_key();
1236 test_reg_delete_key();
1237 test_reg_query_value();
1239 /* SaveKey/LoadKey require the SE_BACKUP_NAME privilege to be set */
1240 if (set_privileges(SE_BACKUP_NAME
, TRUE
) &&
1241 set_privileges(SE_RESTORE_NAME
, TRUE
))
1243 test_reg_save_key();
1244 test_reg_load_key();
1245 test_reg_unload_key();
1247 set_privileges(SE_BACKUP_NAME
, FALSE
);
1248 set_privileges(SE_RESTORE_NAME
, FALSE
);
1251 test_reg_delete_tree();
1254 delete_key( hkey_main
);
1256 test_regconnectregistry();