[PATCH 4/6] advapi32/tests: Add some test for perflib keys.

[Zebediah Figura]

Signed-off-by: Zebediah Figura 
---
 dlls/advapi32/tests/registry.c | 114 +++++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)

diff --git a/dlls/advapi32/tests/registry.c b/dlls/advapi32/tests/registry.c
index 72423eae94c..3330179a1e2 100644
--- a/dlls/advapi32/tests/registry.c
+++ b/dlls/advapi32/tests/registry.c
@@ -99,6 +99,11 @@ static void InitFunctionPtrs(void)
     pNtUnloadKey = (void *)GetProcAddress( hntdll, "NtUnloadKey" );
 }

+static BOOL is_special_key(HKEY key)
+{
+    return !!((ULONG_PTR)key & 0x80000000);
+}
+
 /* delete key and all its subkeys */
 static DWORD delete_key( HKEY hkey )
 {
@@ -3809,6 +3814,114 @@ todo_wine
     ok(dwret == ERROR_SUCCESS, "got %u\n", dwret);
 }

+static void test_perflib_key(void)
+{
+    unsigned int primary_lang = PRIMARYLANGID(GetUserDefaultLangID());
+    unsigned int buffer_size = 1024 * 1024;
+    OBJECT_NAME_INFORMATION *name_info;
+    HKEY perflib_key, key, key2;
+    OBJECT_ATTRIBUTES attr;
+    UNICODE_STRING string;
+    char lang_name[4];
+    char *buffer;
+    DWORD size;
+    LONG ret;
+
+    buffer = malloc(buffer_size);
+
+    ret = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
+            "Software\Microsoft\Windows NT\CurrentVersion\Perflib", 0, KEY_READ, &perflib_key);
+    ok(!ret, "got %u\n", ret);
+
+    ret = RegOpenKeyExA(perflib_key, "009", 0, KEY_READ, &key);
+    todo_wine ok(!ret, "got %u\n", ret);
+    todo_wine ok(key == HKEY_PERFORMANCE_TEXT, "got key %p\n", key);
+
+    ret = RegCloseKey(key);
+    todo_wine ok(!ret, "got %u\n", ret);
+
+    RtlInitUnicodeString(&string, L"009");
+    InitializeObjectAttributes(&attr, &string, OBJ_CASE_INSENSITIVE, perflib_key, NULL);
+    ret = NtOpenKey((HANDLE *)&key, KEY_ALL_ACCESS, &attr);
+    todo_wine ok(ret == STATUS_PREDEFINED_HANDLE, "got %#x\n", ret);
+    ok(!is_special_key(key), "expected a normal handle, got %p\n", key);
+
+    ret = NtQueryKey(key, KeyFullInformation, buffer, buffer_size, &size);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    ret = NtEnumerateKey(key, 0, KeyFullInformation, buffer, buffer_size, &size);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    RtlInitUnicodeString(&string, L"counter");
+    ret = NtQueryValueKey(key, &string, KeyValuePartialInformation, buffer, buffer_size, &size);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    ret = NtEnumerateValueKey(key, 0, KeyValuePartialInformation, buffer, buffer_size, &size);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    ret = NtSetValueKey(key, &string, 0, REG_SZ, "test", 5);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    ret = NtDeleteValueKey(key, &string);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    ret = NtDeleteKey(key);
+    ok(ret == STATUS_INVALID_HANDLE, "got %#x\n", ret);
+
+    RtlInitUnicodeString(&string, L"subkey");
+    InitializeObjectAttributes(&attr, &string, OBJ_CASE_INSENSITIVE, key, NULL);
+    ret = NtOpenKey((HANDLE *)&key2, KEY_READ, &attr);
+    ok(ret == STATUS_OBJECT_NAME_NOT_FOUND, "got %#x\n", ret);
+
+    ret = NtCreateKey((HANDLE *)&key2, KEY_ALL_ACCESS, &attr, 0, NULL, 0, NULL);
+    ok(!ret || ret == STATUS_ACCESS_DENIED, "got %#x\n", ret);
+
+    if (!ret)
+    {
+        ret = NtDeleteKey(key2);
+        ok(!ret, "got %#x\n", ret);
+    }
+
+    /* it's a real handle, though */
+    ret = NtQueryObject(key, ObjectNameInformation, buffer, buffer_size, &size);
+    todo_wine ok(!ret, "got %#x\n", ret);
+    if (!ret)
+    {
+        name_info = (OBJECT_NAME_INFORMATION *)buffer;
+        ok(!wcsicmp(name_info->Name.Buffer, L"\REGISTRY\MACHINE\SOFTWARE\Microsoft\Windows NT"
+                "\CurrentVersion\Perflib\009"), "got name %s\n", debugstr_w(name_info->Name.Buffer));
+    }
+
+    ret = NtClose(key);
+    todo_wine ok(!ret, "got %#x\n", ret);
+
+    /* multilingual support was not really completely thought through */
+
+    sprintf(lang_name, "%03x", primary_lang);
+    if (primary_lang != LANG_ENGLISH)
+    {
+        ret = RegOpenKeyExA(perflib_key, lang_name, 0, KEY_READ, &key);
+        ok(!ret, "got %u\n", ret);
+        ok(!is_special_key(key), "expected a normal handle, got %p\n", key);
+
+        ret = RegQueryValueExA(key, "counter", NULL, NULL, (BYTE *)buffer, &size);
+        ok(ret == ERROR_FILE_NOT_FOUND, "got %u\n", ret);
+
+        ret = RegCloseKey(key);
+        ok(!ret, "got %u\n", ret);
+    }
+
+    ret = RegCloseKey(perflib_key);
+    ok(!ret, "got %u\n", ret);
+
+    RtlInitUnicodeString(&string, L"\Registry\PerfData");
+    InitializeObjectAttributes(&attr, &string, OBJ_CASE_INSENSITIVE, NULL, NULL);
+    ret = NtOpenKey((HANDLE *)&key, KEY_READ, &attr);
+    todo_wine ok(ret == STATUS_OBJECT_NAME_NOT_FOUND, "got %#x\n", ret);
+
+    free(buffer);
+}
+
 static void test_RegLoadMUIString(void)
 {
     HMODULE hUser32, hResDll, hFile;
@@ -4213,6 +4326,7 @@ START_TEST(registry)
     test_RegQueryValueExPerformanceData();
     test_RegLoadMUIString();
     test_EnumDynamicTimeZoneInformation();
+    test_perflib_key();

/* cleanup */
     delete_key( hkey_main );