dlls/ntdll/reg.c

   1 /*
   2  * Registry functions
   3  *
   4  * Copyright (C) 1999 Juergen Schmied
   5  * Copyright (C) 2000 Alexandre Julliard
   6  * Copyright 2005 Ivan Leo Puoti, Laurent Pinchart
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  *
  22  * NOTES:
  23  *      HKEY_LOCAL_MACHINE      \\REGISTRY\\MACHINE
  24  *      HKEY_USERS              \\REGISTRY\\USER
  25  *      HKEY_CURRENT_CONFIG     \\REGISTRY\\MACHINE\\SYSTEM\\CURRENTCONTROLSET\\HARDWARE PROFILES\\CURRENT
  26   *     HKEY_CLASSES            \\REGISTRY\\MACHINE\\SOFTWARE\\CLASSES
  27  */
  28
  29 #include <stdarg.h>
  30 #include <stdio.h>
  31 #include <string.h>
  32
  33 #include "ntstatus.h"
  34 #define WIN32_NO_STATUS
  35 #include "ntdll_misc.h"
  36 #include "wine/debug.h"
  37
  38 WINE_DEFAULT_DEBUG_CHANNEL(reg);
  39
  40
  41 /******************************************************************************
  42  *  RtlpNtCreateKey [NTDLL.@]
  43  *
  44  *  See NtCreateKey.
  45  */
  46 NTSTATUS WINAPI RtlpNtCreateKey( PHANDLE retkey, ACCESS_MASK access, const OBJECT_ATTRIBUTES *attr,
  47                                  ULONG TitleIndex, const UNICODE_STRING *class, ULONG options,
  48                                  PULONG dispos )
  49 {
  50     OBJECT_ATTRIBUTES oa;
  51
  52     if (attr)
  53     {
  54         oa = *attr;
  55         oa.Attributes &= ~(OBJ_PERMANENT|OBJ_EXCLUSIVE);
  56         attr = &oa;
  57     }
  58
  59     return NtCreateKey(retkey, access, attr, 0, NULL, 0, dispos);
  60 }
  61
  62 /******************************************************************************
  63  * RtlpNtOpenKey [NTDLL.@]
  64  *
  65  * See NtOpenKey.
  66  */
  67 NTSTATUS WINAPI RtlpNtOpenKey( PHANDLE retkey, ACCESS_MASK access, OBJECT_ATTRIBUTES *attr )
  68 {
  69     if (attr)
  70         attr->Attributes &= ~(OBJ_PERMANENT|OBJ_EXCLUSIVE);
  71     return NtOpenKey(retkey, access, attr);
  72 }
  73
  74 /******************************************************************************
  75  * RtlpNtMakeTemporaryKey [NTDLL.@]
  76  *
  77  *  See NtDeleteKey.
  78  */
  79 NTSTATUS WINAPI RtlpNtMakeTemporaryKey( HANDLE hkey )
  80 {
  81     return NtDeleteKey(hkey);
  82 }
  83
  84 /******************************************************************************
  85  * RtlpNtEnumerateSubKey [NTDLL.@]
  86  *
  87  */
  88 NTSTATUS WINAPI RtlpNtEnumerateSubKey( HANDLE handle, UNICODE_STRING *out, ULONG index )
  89 {
  90   KEY_BASIC_INFORMATION *info;
  91   DWORD dwLen, dwResultLen;
  92   NTSTATUS ret;
  93
  94   if (out->MaximumLength)
  95   {
  96     dwLen = out->MaximumLength + offsetof(KEY_BASIC_INFORMATION, Name);
  97     info = RtlAllocateHeap( GetProcessHeap(), 0, dwLen );
  98     if (!info)
  99       return STATUS_NO_MEMORY;
 100   }
 101   else
 102   {
 103     dwLen = 0;
 104     info = NULL;
 105   }
 106
 107   ret = NtEnumerateKey( handle, index, KeyBasicInformation, info, dwLen, &dwResultLen );
 108   dwResultLen -= offsetof(KEY_BASIC_INFORMATION, Name);
 109
 110   if (ret == STATUS_BUFFER_OVERFLOW)
 111     out->Length = dwResultLen;
 112   else if (!ret)
 113   {
 114     if (out->MaximumLength < info->NameLength)
 115     {
 116       out->Length = dwResultLen;
 117       ret = STATUS_BUFFER_OVERFLOW;
 118     }
 119     else
 120     {
 121       out->Length = info->NameLength;
 122       memcpy(out->Buffer, info->Name, info->NameLength);
 123     }
 124   }
 125
 126   RtlFreeHeap( GetProcessHeap(), 0, info );
 127   return ret;
 128 }
 129
 130 /******************************************************************************
 131  * RtlpNtQueryValueKey [NTDLL.@]
 132  *
 133  */
 134 NTSTATUS WINAPI RtlpNtQueryValueKey( HANDLE handle, ULONG *result_type, PBYTE dest,
 135                                      DWORD *result_len, void *unknown )
 136 {
 137     KEY_VALUE_PARTIAL_INFORMATION *info;
 138     UNICODE_STRING name;
 139     NTSTATUS ret;
 140     DWORD dwResultLen;
 141     DWORD dwLen = offsetof(KEY_VALUE_PARTIAL_INFORMATION, Data[result_len ? *result_len : 0]);
 142
 143     info = RtlAllocateHeap( GetProcessHeap(), 0, dwLen );
 144     if (!info)
 145       return STATUS_NO_MEMORY;
 146
 147     name.Length = 0;
 148     ret = NtQueryValueKey( handle, &name, KeyValuePartialInformation, info, dwLen, &dwResultLen );
 149
 150     if (!ret || ret == STATUS_BUFFER_OVERFLOW)
 151     {
 152         if (result_len)
 153             *result_len = info->DataLength;
 154
 155         if (result_type)
 156             *result_type = info->Type;
 157
 158         if (ret != STATUS_BUFFER_OVERFLOW)
 159             memcpy( dest, info->Data, info->DataLength );
 160     }
 161
 162     RtlFreeHeap( GetProcessHeap(), 0, info );
 163     return ret;
 164 }
 165
 166 /******************************************************************************
 167  * RtlpNtSetValueKey [NTDLL.@]
 168  *
 169  */
 170 NTSTATUS WINAPI RtlpNtSetValueKey( HANDLE hkey, ULONG type, const void *data,
 171                                    ULONG count )
 172 {
 173     UNICODE_STRING name;
 174
 175     name.Length = 0;
 176     return NtSetValueKey( hkey, &name, 0, type, data, count );
 177 }
 178
 179 /******************************************************************************
 180  *  RtlFormatCurrentUserKeyPath         [NTDLL.@]
 181  *
 182  */
 183 NTSTATUS WINAPI RtlFormatCurrentUserKeyPath( IN OUT PUNICODE_STRING KeyPath)
 184 {
 185     static const WCHAR pathW[] = {'\\','R','e','g','i','s','t','r','y','\\','U','s','e','r','\\'};
 186     char buffer[sizeof(TOKEN_USER) + sizeof(SID) + sizeof(DWORD)*SID_MAX_SUB_AUTHORITIES];
 187     DWORD len = sizeof(buffer);
 188     NTSTATUS status;
 189
 190     status = NtQueryInformationToken(GetCurrentThreadEffectiveToken(), TokenUser, buffer, len, &len);
 191     if (status == STATUS_SUCCESS)
 192     {
 193         KeyPath->MaximumLength = 0;
 194         status = RtlConvertSidToUnicodeString(KeyPath, ((TOKEN_USER *)buffer)->User.Sid, FALSE);
 195         if (status == STATUS_BUFFER_OVERFLOW)
 196         {
 197             PWCHAR buf = RtlAllocateHeap(GetProcessHeap(), 0,
 198                                          sizeof(pathW) + KeyPath->Length + sizeof(WCHAR));
 199             if (buf)
 200             {
 201                 memcpy(buf, pathW, sizeof(pathW));
 202                 KeyPath->MaximumLength = KeyPath->Length + sizeof(WCHAR);
 203                 KeyPath->Buffer = (PWCHAR)((LPBYTE)buf + sizeof(pathW));
 204                 status = RtlConvertSidToUnicodeString(KeyPath,
 205                                                       ((TOKEN_USER *)buffer)->User.Sid, FALSE);
 206                 KeyPath->Buffer = buf;
 207                 KeyPath->Length += sizeof(pathW);
 208                 KeyPath->MaximumLength += sizeof(pathW);
 209             }
 210             else
 211                 status = STATUS_NO_MEMORY;
 212         }
 213     }
 214     return status;
 215 }
 216
 217 /******************************************************************************
 218  *  RtlOpenCurrentUser          [NTDLL.@]
 219  *
 220  * NOTES
 221  *  If we return just HKEY_CURRENT_USER the advapi tries to find a remote
 222  *  registry (odd handle) and fails.
 223  */
 224 NTSTATUS WINAPI RtlOpenCurrentUser(
 225         IN ACCESS_MASK DesiredAccess, /* [in] */
 226         OUT PHANDLE KeyHandle)        /* [out] handle of HKEY_CURRENT_USER */
 227 {
 228         OBJECT_ATTRIBUTES ObjectAttributes;
 229         UNICODE_STRING ObjectName;
 230         NTSTATUS ret;
 231
 232         TRACE("(0x%08x, %p)\n",DesiredAccess, KeyHandle);
 233
 234         if ((ret = RtlFormatCurrentUserKeyPath(&ObjectName))) return ret;
 235         InitializeObjectAttributes(&ObjectAttributes,&ObjectName,OBJ_CASE_INSENSITIVE,0, NULL);
 236         ret = NtCreateKey(KeyHandle, DesiredAccess, &ObjectAttributes, 0, NULL, 0, NULL);
 237         RtlFreeUnicodeString(&ObjectName);
 238         return ret;
 239 }
 240
 241
 242 static NTSTATUS RTL_ReportRegistryValue(PKEY_VALUE_FULL_INFORMATION pInfo,
 243                                         PRTL_QUERY_REGISTRY_TABLE pQuery, PVOID pContext, PVOID pEnvironment)
 244 {
 245     PUNICODE_STRING str;
 246     UNICODE_STRING src, dst;
 247     LONG *bin;
 248     ULONG offset;
 249     PWSTR wstr;
 250     DWORD res;
 251     NTSTATUS status = STATUS_SUCCESS;
 252     ULONG len;
 253     LPWSTR String;
 254     ULONG count = 0;
 255
 256     if (pInfo == NULL)
 257     {
 258         if (pQuery->Flags & RTL_QUERY_REGISTRY_DIRECT)
 259             return STATUS_INVALID_PARAMETER;
 260         else
 261         {
 262             status = pQuery->QueryRoutine(pQuery->Name, pQuery->DefaultType, pQuery->DefaultData,
 263                                           pQuery->DefaultLength, pContext, pQuery->EntryContext);
 264         }
 265         return status;
 266     }
 267     len = pInfo->DataLength;
 268
 269     if (pQuery->Flags & RTL_QUERY_REGISTRY_DIRECT)
 270     {
 271         str = pQuery->EntryContext;
 272
 273         switch(pInfo->Type)
 274         {
 275         case REG_EXPAND_SZ:
 276             if (!(pQuery->Flags & RTL_QUERY_REGISTRY_NOEXPAND))
 277             {
 278                 RtlInitUnicodeString(&src, (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset));
 279                 res = 0;
 280                 dst.MaximumLength = 0;
 281                 RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 282                 dst.Length = 0;
 283                 dst.MaximumLength = res;
 284                 dst.Buffer = RtlAllocateHeap(GetProcessHeap(), 0, res * sizeof(WCHAR));
 285                 RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 286                 status = pQuery->QueryRoutine(pQuery->Name, pInfo->Type, dst.Buffer,
 287                                      dst.Length, pContext, pQuery->EntryContext);
 288                 RtlFreeHeap(GetProcessHeap(), 0, dst.Buffer);
 289             }
 290
 291         case REG_SZ:
 292         case REG_LINK:
 293             if (str->Buffer == NULL)
 294                 RtlCreateUnicodeString(str, (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset));
 295             else
 296                 RtlAppendUnicodeToString(str, (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset));
 297             break;
 298
 299         case REG_MULTI_SZ:
 300             if (!(pQuery->Flags & RTL_QUERY_REGISTRY_NOEXPAND))
 301                 return STATUS_INVALID_PARAMETER;
 302
 303             if (str->Buffer == NULL)
 304             {
 305                 str->Buffer = RtlAllocateHeap(GetProcessHeap(), 0, len);
 306                 str->MaximumLength = len;
 307             }
 308             len = min(len, str->MaximumLength);
 309             memcpy(str->Buffer, ((CHAR*)pInfo) + pInfo->DataOffset, len);
 310             str->Length = len;
 311             break;
 312
 313         default:
 314             bin = pQuery->EntryContext;
 315             if (pInfo->DataLength <= sizeof(ULONG))
 316                 memcpy(bin, ((CHAR*)pInfo) + pInfo->DataOffset,
 317                     pInfo->DataLength);
 318             else
 319             {
 320                 if (bin[0] <= sizeof(ULONG))
 321                 {
 322                     memcpy(&bin[1], ((CHAR*)pInfo) + pInfo->DataOffset,
 323                     min(-bin[0], pInfo->DataLength));
 324                 }
 325                 else
 326                 {
 327                    len = min(bin[0], pInfo->DataLength);
 328                     bin[1] = len;
 329                     bin[2] = pInfo->Type;
 330                     memcpy(&bin[3], ((CHAR*)pInfo) + pInfo->DataOffset, len);
 331                 }
 332            }
 333            break;
 334         }
 335     }
 336     else
 337     {
 338         if((pQuery->Flags & RTL_QUERY_REGISTRY_NOEXPAND) ||
 339            (pInfo->Type != REG_EXPAND_SZ && pInfo->Type != REG_MULTI_SZ))
 340         {
 341             status = pQuery->QueryRoutine(pQuery->Name, pInfo->Type,
 342                 ((CHAR*)pInfo) + pInfo->DataOffset, pInfo->DataLength,
 343                 pContext, pQuery->EntryContext);
 344         }
 345         else if (pInfo->Type == REG_EXPAND_SZ)
 346         {
 347             RtlInitUnicodeString(&src, (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset));
 348             res = 0;
 349             dst.MaximumLength = 0;
 350             RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 351             dst.Length = 0;
 352             dst.MaximumLength = res;
 353             dst.Buffer = RtlAllocateHeap(GetProcessHeap(), 0, res * sizeof(WCHAR));
 354             RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 355             status = pQuery->QueryRoutine(pQuery->Name, pInfo->Type, dst.Buffer,
 356                                           dst.Length, pContext, pQuery->EntryContext);
 357             RtlFreeHeap(GetProcessHeap(), 0, dst.Buffer);
 358         }
 359         else /* REG_MULTI_SZ */
 360         {
 361             if(pQuery->Flags & RTL_QUERY_REGISTRY_NOEXPAND)
 362             {
 363                 for (offset = 0; offset <= pInfo->DataLength; offset += len + sizeof(WCHAR))
 364                     {
 365                     wstr = (WCHAR*)(((CHAR*)pInfo) + offset);
 366                     len = wcslen(wstr) * sizeof(WCHAR);
 367                     status = pQuery->QueryRoutine(pQuery->Name, pInfo->Type, wstr, len,
 368                         pContext, pQuery->EntryContext);
 369                     if(status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 370                         return status;
 371                     }
 372             }
 373             else
 374             {
 375                 while(count<=pInfo->DataLength)
 376                 {
 377                     String = (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset)+count;
 378                     count+=wcslen(String)+1;
 379                     RtlInitUnicodeString(&src, (WCHAR*)(((CHAR*)pInfo) + pInfo->DataOffset));
 380                     res = 0;
 381                     dst.MaximumLength = 0;
 382                     RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 383                     dst.Length = 0;
 384                     dst.MaximumLength = res;
 385                     dst.Buffer = RtlAllocateHeap(GetProcessHeap(), 0, res * sizeof(WCHAR));
 386                     RtlExpandEnvironmentStrings_U(pEnvironment, &src, &dst, &res);
 387                     status = pQuery->QueryRoutine(pQuery->Name, pInfo->Type, dst.Buffer,
 388                                                   dst.Length, pContext, pQuery->EntryContext);
 389                     RtlFreeHeap(GetProcessHeap(), 0, dst.Buffer);
 390                     if(status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 391                         return status;
 392                 }
 393             }
 394         }
 395     }
 396     return status;
 397 }
 398
 399
 400 static NTSTATUS RTL_KeyHandleCreateObject(ULONG RelativeTo, PCWSTR Path, POBJECT_ATTRIBUTES regkey, PUNICODE_STRING str)
 401 {
 402     PCWSTR base;
 403     INT len;
 404
 405     switch (RelativeTo & 0xff)
 406     {
 407     case RTL_REGISTRY_ABSOLUTE:
 408         base = L"";
 409         break;
 410
 411     case RTL_REGISTRY_CONTROL:
 412         base = L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\";
 413         break;
 414
 415     case RTL_REGISTRY_DEVICEMAP:
 416         base = L"\\Registry\\Machine\\Hardware\\DeviceMap\\";
 417         break;
 418
 419     case RTL_REGISTRY_SERVICES:
 420         base = L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\";
 421         break;
 422
 423     case RTL_REGISTRY_USER:
 424         base = L"\\Registry\\User\\CurrentUser\\";
 425         break;
 426
 427     case RTL_REGISTRY_WINDOWS_NT:
 428         base = L"\\Registry\\Machine\\Software\\Microsoft\\Windows NT\\CurrentVersion\\";
 429         break;
 430
 431     default:
 432         return STATUS_INVALID_PARAMETER;
 433     }
 434
 435     len = (wcslen(base) + wcslen(Path) + 1) * sizeof(WCHAR);
 436     str->Buffer = RtlAllocateHeap(GetProcessHeap(), 0, len);
 437     if (str->Buffer == NULL)
 438         return STATUS_NO_MEMORY;
 439
 440     wcscpy(str->Buffer, base);
 441     wcscat(str->Buffer, Path);
 442     str->Length = len - sizeof(WCHAR);
 443     str->MaximumLength = len;
 444     InitializeObjectAttributes(regkey, str, OBJ_CASE_INSENSITIVE, NULL, NULL);
 445     return STATUS_SUCCESS;
 446 }
 447
 448 static NTSTATUS RTL_GetKeyHandle(ULONG RelativeTo, PCWSTR Path, PHANDLE handle)
 449 {
 450     OBJECT_ATTRIBUTES regkey;
 451     UNICODE_STRING string;
 452     NTSTATUS status;
 453
 454     status = RTL_KeyHandleCreateObject(RelativeTo, Path, &regkey, &string);
 455     if(status != STATUS_SUCCESS)
 456         return status;
 457
 458     status = NtOpenKey(handle, KEY_ALL_ACCESS, &regkey);
 459     RtlFreeUnicodeString( &string );
 460     return status;
 461 }
 462
 463 /******************************************************************************
 464  *              RtlQueryRegistryValues  (NTDLL.@)
 465  *              RtlQueryRegistryValuesEx  (NTDLL.@)
 466  */
 467 NTSTATUS WINAPI RtlQueryRegistryValues(IN ULONG RelativeTo, IN PCWSTR Path,
 468                                        IN PRTL_QUERY_REGISTRY_TABLE QueryTable, IN PVOID Context,
 469                                        IN PVOID Environment OPTIONAL)
 470 {
 471     UNICODE_STRING Value;
 472     HANDLE handle, topkey;
 473     PKEY_VALUE_FULL_INFORMATION pInfo = NULL;
 474     ULONG len, buflen = 0;
 475     NTSTATUS status=STATUS_SUCCESS, ret = STATUS_SUCCESS;
 476     INT i;
 477
 478     TRACE("(%d, %s, %p, %p, %p)\n", RelativeTo, debugstr_w(Path), QueryTable, Context, Environment);
 479
 480     if(Path == NULL)
 481         return STATUS_INVALID_PARAMETER;
 482
 483     /* get a valid handle */
 484     if (RelativeTo & RTL_REGISTRY_HANDLE)
 485         topkey = handle = (HANDLE)Path;
 486     else
 487     {
 488         status = RTL_GetKeyHandle(RelativeTo, Path, &topkey);
 489         handle = topkey;
 490     }
 491     if(status != STATUS_SUCCESS)
 492         return status;
 493
 494     /* Process query table entries */
 495     for (; QueryTable->QueryRoutine != NULL || QueryTable->Name != NULL; ++QueryTable)
 496     {
 497         if (QueryTable->Flags &
 498             (RTL_QUERY_REGISTRY_SUBKEY | RTL_QUERY_REGISTRY_TOPKEY))
 499         {
 500             /* topkey must be kept open just in case we will reuse it later */
 501             if (handle != topkey)
 502                 NtClose(handle);
 503
 504             if (QueryTable->Flags & RTL_QUERY_REGISTRY_SUBKEY)
 505             {
 506                 handle = 0;
 507                 status = RTL_GetKeyHandle(PtrToUlong(QueryTable->Name), Path, &handle);
 508                 if(status != STATUS_SUCCESS)
 509                 {
 510                     ret = status;
 511                     goto out;
 512                 }
 513             }
 514             else
 515                 handle = topkey;
 516         }
 517
 518         if (QueryTable->Flags & RTL_QUERY_REGISTRY_NOVALUE)
 519         {
 520             QueryTable->QueryRoutine(QueryTable->Name, REG_NONE, NULL, 0,
 521                 Context, QueryTable->EntryContext);
 522             continue;
 523         }
 524
 525         if (!handle)
 526         {
 527             if (QueryTable->Flags & RTL_QUERY_REGISTRY_REQUIRED)
 528             {
 529                 ret = STATUS_OBJECT_NAME_NOT_FOUND;
 530                 goto out;
 531             }
 532             continue;
 533         }
 534
 535         if (QueryTable->Name == NULL)
 536         {
 537             if (QueryTable->Flags & RTL_QUERY_REGISTRY_DIRECT)
 538             {
 539                 ret = STATUS_INVALID_PARAMETER;
 540                 goto out;
 541             }
 542
 543             /* Report all subkeys */
 544             for (i = 0;; ++i)
 545             {
 546                 status = NtEnumerateValueKey(handle, i,
 547                     KeyValueFullInformation, pInfo, buflen, &len);
 548                 if (status == STATUS_NO_MORE_ENTRIES)
 549                     break;
 550                 if (status == STATUS_BUFFER_OVERFLOW ||
 551                     status == STATUS_BUFFER_TOO_SMALL)
 552                 {
 553                     buflen = len;
 554                     RtlFreeHeap(GetProcessHeap(), 0, pInfo);
 555                     pInfo = RtlAllocateHeap(GetProcessHeap(), 0, buflen);
 556                     NtEnumerateValueKey(handle, i, KeyValueFullInformation,
 557                         pInfo, buflen, &len);
 558                 }
 559
 560                 status = RTL_ReportRegistryValue(pInfo, QueryTable, Context, Environment);
 561                 if(status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 562                 {
 563                     ret = status;
 564                     goto out;
 565                 }
 566                 if (QueryTable->Flags & RTL_QUERY_REGISTRY_DELETE)
 567                 {
 568                     RtlInitUnicodeString(&Value, pInfo->Name);
 569                     NtDeleteValueKey(handle, &Value);
 570                 }
 571             }
 572
 573             if (i == 0  && (QueryTable->Flags & RTL_QUERY_REGISTRY_REQUIRED))
 574             {
 575                 ret = STATUS_OBJECT_NAME_NOT_FOUND;
 576                 goto out;
 577             }
 578         }
 579         else
 580         {
 581             RtlInitUnicodeString(&Value, QueryTable->Name);
 582             status = NtQueryValueKey(handle, &Value, KeyValueFullInformation,
 583                 pInfo, buflen, &len);
 584             if (status == STATUS_BUFFER_OVERFLOW ||
 585                 status == STATUS_BUFFER_TOO_SMALL)
 586             {
 587                 buflen = len;
 588                 RtlFreeHeap(GetProcessHeap(), 0, pInfo);
 589                 pInfo = RtlAllocateHeap(GetProcessHeap(), 0, buflen);
 590                 status = NtQueryValueKey(handle, &Value,
 591                     KeyValueFullInformation, pInfo, buflen, &len);
 592             }
 593             if (status != STATUS_SUCCESS)
 594             {
 595                 if (QueryTable->Flags & RTL_QUERY_REGISTRY_REQUIRED)
 596                 {
 597                     ret = STATUS_OBJECT_NAME_NOT_FOUND;
 598                     goto out;
 599                 }
 600                 status = RTL_ReportRegistryValue(NULL, QueryTable, Context, Environment);
 601                 if(status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 602                 {
 603                     ret = status;
 604                     goto out;
 605                 }
 606             }
 607             else
 608             {
 609                 status = RTL_ReportRegistryValue(pInfo, QueryTable, Context, Environment);
 610                 if(status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 611                 {
 612                     ret = status;
 613                     goto out;
 614                 }
 615                 if (QueryTable->Flags & RTL_QUERY_REGISTRY_DELETE)
 616                     NtDeleteValueKey(handle, &Value);
 617             }
 618         }
 619     }
 620
 621 out:
 622     RtlFreeHeap(GetProcessHeap(), 0, pInfo);
 623     if (handle != topkey)
 624         NtClose(handle);
 625     NtClose(topkey);
 626     return ret;
 627 }
 628
 629 /*************************************************************************
 630  * RtlCheckRegistryKey   [NTDLL.@]
 631  *
 632  * Query multiple registry values with a single call.
 633  *
 634  * PARAMS
 635  *  RelativeTo [I] Registry path that Path refers to
 636  *  Path       [I] Path to key
 637  *
 638  * RETURNS
 639  *  STATUS_SUCCESS if the specified key exists, or an NTSTATUS error code.
 640  */
 641 NTSTATUS WINAPI RtlCheckRegistryKey(IN ULONG RelativeTo, IN PWSTR Path)
 642 {
 643     HANDLE handle;
 644     NTSTATUS status;
 645
 646     TRACE("(%d, %s)\n", RelativeTo, debugstr_w(Path));
 647
 648     if(!RelativeTo && (Path == NULL || Path[0] == 0))
 649         return STATUS_OBJECT_PATH_SYNTAX_BAD;
 650     if(RelativeTo & RTL_REGISTRY_HANDLE)
 651         return STATUS_SUCCESS;
 652     if((RelativeTo <= RTL_REGISTRY_USER) && (Path == NULL || Path[0] == 0))
 653         return STATUS_SUCCESS;
 654
 655     status = RTL_GetKeyHandle(RelativeTo, Path, &handle);
 656     if (handle) NtClose(handle);
 657     if (status == STATUS_INVALID_HANDLE) status = STATUS_OBJECT_NAME_NOT_FOUND;
 658     return status;
 659 }
 660
 661 /*************************************************************************
 662  * RtlCreateRegistryKey   [NTDLL.@]
 663  *
 664  * Add a key to the registry given by absolute or relative path
 665  *
 666  * PARAMS
 667  *  RelativeTo  [I] Registry path that Path refers to
 668  *  path        [I] Path to key
 669  *
 670  * RETURNS
 671  *  STATUS_SUCCESS or an appropriate NTSTATUS error code.
 672  */
 673 NTSTATUS WINAPI RtlCreateRegistryKey(ULONG RelativeTo, PWSTR path)
 674 {
 675     OBJECT_ATTRIBUTES regkey;
 676     UNICODE_STRING string;
 677     HANDLE handle;
 678     NTSTATUS status;
 679
 680     RelativeTo &= ~RTL_REGISTRY_OPTIONAL;
 681
 682     if (!RelativeTo && (path == NULL || path[0] == 0))
 683         return STATUS_OBJECT_PATH_SYNTAX_BAD;
 684     if (RelativeTo <= RTL_REGISTRY_USER && (path == NULL || path[0] == 0))
 685         return STATUS_SUCCESS;
 686     status = RTL_KeyHandleCreateObject(RelativeTo, path, &regkey, &string);
 687     if(status != STATUS_SUCCESS)
 688         return status;
 689
 690     status = NtCreateKey(&handle, KEY_ALL_ACCESS, &regkey, 0, NULL, REG_OPTION_NON_VOLATILE, NULL);
 691     if (handle) NtClose(handle);
 692     RtlFreeUnicodeString( &string );
 693     return status;
 694 }
 695
 696 /*************************************************************************
 697  * RtlDeleteRegistryValue   [NTDLL.@]
 698  *
 699  * Query multiple registry values with a single call.
 700  *
 701  * PARAMS
 702  *  RelativeTo [I] Registry path that Path refers to
 703  *  Path       [I] Path to key
 704  *  ValueName  [I] Name of the value to delete
 705  *
 706  * RETURNS
 707  *  STATUS_SUCCESS if the specified key is successfully deleted, or an NTSTATUS error code.
 708  */
 709 NTSTATUS WINAPI RtlDeleteRegistryValue(IN ULONG RelativeTo, IN PCWSTR Path, IN PCWSTR ValueName)
 710 {
 711     NTSTATUS status;
 712     HANDLE handle;
 713     UNICODE_STRING Value;
 714
 715     TRACE("(%d, %s, %s)\n", RelativeTo, debugstr_w(Path), debugstr_w(ValueName));
 716
 717     RtlInitUnicodeString(&Value, ValueName);
 718     if(RelativeTo == RTL_REGISTRY_HANDLE)
 719     {
 720         return NtDeleteValueKey((HANDLE)Path, &Value);
 721     }
 722     status = RTL_GetKeyHandle(RelativeTo, Path, &handle);
 723     if (status) return status;
 724     status = NtDeleteValueKey(handle, &Value);
 725     NtClose(handle);
 726     return status;
 727 }
 728
 729 /*************************************************************************
 730  * RtlWriteRegistryValue   [NTDLL.@]
 731  *
 732  * Sets the registry value with provided data.
 733  *
 734  * PARAMS
 735  *  RelativeTo [I] Registry path that path parameter refers to
 736  *  path       [I] Path to the key (or handle - see RTL_GetKeyHandle)
 737  *  name       [I] Name of the registry value to set
 738  *  type       [I] Type of the registry key to set
 739  *  data       [I] Pointer to the user data to be set
 740  *  length     [I] Length of the user data pointed by data
 741  *
 742  * RETURNS
 743  *  STATUS_SUCCESS if the specified key is successfully set,
 744  *  or an NTSTATUS error code.
 745  */
 746 NTSTATUS WINAPI RtlWriteRegistryValue( ULONG RelativeTo, PCWSTR path, PCWSTR name,
 747                                        ULONG type, PVOID data, ULONG length )
 748 {
 749     HANDLE hkey;
 750     NTSTATUS status;
 751     UNICODE_STRING str;
 752
 753     TRACE( "(%d, %s, %s) -> %d: %p [%d]\n", RelativeTo, debugstr_w(path), debugstr_w(name),
 754            type, data, length );
 755
 756     RtlInitUnicodeString( &str, name );
 757
 758     if (RelativeTo == RTL_REGISTRY_HANDLE)
 759         return NtSetValueKey( (HANDLE)path, &str, 0, type, data, length );
 760
 761     status = RTL_GetKeyHandle( RelativeTo, path, &hkey );
 762     if (status != STATUS_SUCCESS) return status;
 763
 764     status = NtSetValueKey( hkey, &str, 0, type, data, length );
 765     NtClose( hkey );
 766
 767     return status;
 768 }