dlls/advapi32/registry.c

   1 /*
   2  * Registry management
   3  *
   4  * Copyright (C) 1999 Alexandre Julliard
   5  *
   6  * Based on misc/registry.c code
   7  * Copyright (C) 1996 Marcus Meissner
   8  * Copyright (C) 1998 Matthew Becker
   9  * Copyright (C) 1999 Sylvain St-Germain
  10  *
  11  * This file is concerned about handle management and interaction with the Wine server.
  12  * Registry file I/O is in misc/registry.c.
  13  *
  14  * This library is free software; you can redistribute it and/or
  15  * modify it under the terms of the GNU Lesser General Public
  16  * License as published by the Free Software Foundation; either
  17  * version 2.1 of the License, or (at your option) any later version.
  18  *
  19  * This library is distributed in the hope that it will be useful,
  20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  22  * Lesser General Public License for more details.
  23  *
  24  * You should have received a copy of the GNU Lesser General Public
  25  * License along with this library; if not, write to the Free Software
  26  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  27  */
  28
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <unistd.h>
  32
  33 #include "winbase.h"
  34 #include "winreg.h"
  35 #include "winerror.h"
  36 #include "wine/unicode.h"
  37 #include "heap.h"
  38 #include "wine/server.h"
  39 #include "wine/debug.h"
  40
  41 WINE_DEFAULT_DEBUG_CHANNEL(reg);
  42
  43
  44 /* check if value type needs string conversion (Ansi<->Unicode) */
  45 inline static int is_string( DWORD type )
  46 {
  47     return (type == REG_SZ) || (type == REG_EXPAND_SZ) || (type == REG_MULTI_SZ);
  48 }
  49
  50 /* check if current version is NT or Win95 */
  51 inline static int is_version_nt(void)
  52 {
  53     return !(GetVersion() & 0x80000000);
  54 }
  55
  56 /* allowed bits for access mask */
  57 #define KEY_ACCESS_MASK (KEY_ALL_ACCESS | MAXIMUM_ALLOWED)
  58
  59 /******************************************************************************
  60  *           RegCreateKeyExW   [ADVAPI32.@]
  61  *
  62  * PARAMS
  63  *    hkey       [I] Handle of an open key
  64  *    name       [I] Address of subkey name
  65  *    reserved   [I] Reserved - must be 0
  66  *    class      [I] Address of class string
  67  *    options    [I] Special options flag
  68  *    access     [I] Desired security access
  69  *    sa         [I] Address of key security structure
  70  *    retkey     [O] Address of buffer for opened handle
  71  *    dispos     [O] Receives REG_CREATED_NEW_KEY or REG_OPENED_EXISTING_KEY
  72  *
  73  * NOTES
  74  *  in case of failing retkey remains untouched
  75  *
  76  * FIXME MAXIMUM_ALLOWED in access mask not supported by server
  77  */
  78 DWORD WINAPI RegCreateKeyExW( HKEY hkey, LPCWSTR name, DWORD reserved, LPWSTR class,
  79                               DWORD options, REGSAM access, SECURITY_ATTRIBUTES *sa,
  80                               LPHKEY retkey, LPDWORD dispos )
  81 {
  82     OBJECT_ATTRIBUTES attr;
  83     UNICODE_STRING nameW, classW;
  84
  85     if (reserved) return ERROR_INVALID_PARAMETER;
  86     if (!(access & KEY_ACCESS_MASK) || (access & ~KEY_ACCESS_MASK)) return ERROR_ACCESS_DENIED;
  87
  88     attr.Length = sizeof(attr);
  89     attr.RootDirectory = hkey;
  90     attr.ObjectName = &nameW;
  91     attr.Attributes = 0;
  92     attr.SecurityDescriptor = NULL;
  93     attr.SecurityQualityOfService = NULL;
  94     RtlInitUnicodeString( &nameW, name );
  95     RtlInitUnicodeString( &classW, class );
  96
  97     return RtlNtStatusToDosError( NtCreateKey( retkey, access, &attr, 0,
  98                                                &classW, options, dispos ) );
  99 }
 100
 101
 102 /******************************************************************************
 103  *           RegCreateKeyExA   [ADVAPI32.@]
 104  *
 105  * FIXME MAXIMUM_ALLOWED in access mask not supported by server
 106  */
 107 DWORD WINAPI RegCreateKeyExA( HKEY hkey, LPCSTR name, DWORD reserved, LPSTR class,
 108                               DWORD options, REGSAM access, SECURITY_ATTRIBUTES *sa,
 109                               LPHKEY retkey, LPDWORD dispos )
 110 {
 111     OBJECT_ATTRIBUTES attr;
 112     UNICODE_STRING classW;
 113     ANSI_STRING nameA, classA;
 114     NTSTATUS status;
 115
 116     if (reserved) return ERROR_INVALID_PARAMETER;
 117     if (!is_version_nt()) access = KEY_ALL_ACCESS;  /* Win95 ignores the access mask */
 118     else if (!(access & KEY_ACCESS_MASK) || (access & ~KEY_ACCESS_MASK)) return ERROR_ACCESS_DENIED;
 119
 120     attr.Length = sizeof(attr);
 121     attr.RootDirectory = hkey;
 122     attr.ObjectName = &NtCurrentTeb()->StaticUnicodeString;
 123     attr.Attributes = 0;
 124     attr.SecurityDescriptor = NULL;
 125     attr.SecurityQualityOfService = NULL;
 126     RtlInitAnsiString( &nameA, name );
 127     RtlInitAnsiString( &classA, class );
 128
 129     if (!(status = RtlAnsiStringToUnicodeString( &NtCurrentTeb()->StaticUnicodeString,
 130                                                  &nameA, FALSE )))
 131     {
 132         if (!(status = RtlAnsiStringToUnicodeString( &classW, &classA, TRUE )))
 133         {
 134             status = NtCreateKey( retkey, access, &attr, 0, &classW, options, dispos );
 135             RtlFreeUnicodeString( &classW );
 136         }
 137     }
 138     return RtlNtStatusToDosError( status );
 139 }
 140
 141
 142 /******************************************************************************
 143  *           RegCreateKeyW   [ADVAPI32.@]
 144  */
 145 DWORD WINAPI RegCreateKeyW( HKEY hkey, LPCWSTR name, LPHKEY retkey )
 146 {
 147     /* FIXME: previous implementation converted ERROR_INVALID_HANDLE to ERROR_BADKEY, */
 148     /* but at least my version of NT (4.0 SP5) doesn't do this.  -- AJ */
 149     return RegCreateKeyExW( hkey, name, 0, NULL, REG_OPTION_NON_VOLATILE,
 150                             KEY_ALL_ACCESS, NULL, retkey, NULL );
 151 }
 152
 153
 154 /******************************************************************************
 155  *           RegCreateKeyA   [ADVAPI32.@]
 156  */
 157 DWORD WINAPI RegCreateKeyA( HKEY hkey, LPCSTR name, LPHKEY retkey )
 158 {
 159     return RegCreateKeyExA( hkey, name, 0, NULL, REG_OPTION_NON_VOLATILE,
 160                             KEY_ALL_ACCESS, NULL, retkey, NULL );
 161 }
 162
 163
 164
 165 /******************************************************************************
 166  *           RegOpenKeyExW   [ADVAPI32.@]
 167  *
 168  * Opens the specified key
 169  *
 170  * Unlike RegCreateKeyEx, this does not create the key if it does not exist.
 171  *
 172  * PARAMS
 173  *    hkey       [I] Handle of open key
 174  *    name       [I] Name of subkey to open
 175  *    reserved   [I] Reserved - must be zero
 176  *    access     [I] Security access mask
 177  *    retkey     [O] Handle to open key
 178  *
 179  * RETURNS
 180  *    Success: ERROR_SUCCESS
 181  *    Failure: Error code
 182  *
 183  * NOTES
 184  *  in case of failing is retkey = 0
 185  */
 186 DWORD WINAPI RegOpenKeyExW( HKEY hkey, LPCWSTR name, DWORD reserved, REGSAM access, LPHKEY retkey )
 187 {
 188     OBJECT_ATTRIBUTES attr;
 189     UNICODE_STRING nameW;
 190
 191     attr.Length = sizeof(attr);
 192     attr.RootDirectory = hkey;
 193     attr.ObjectName = &nameW;
 194     attr.Attributes = 0;
 195     attr.SecurityDescriptor = NULL;
 196     attr.SecurityQualityOfService = NULL;
 197     RtlInitUnicodeString( &nameW, name );
 198     return RtlNtStatusToDosError( NtOpenKey( retkey, access, &attr ) );
 199 }
 200
 201
 202 /******************************************************************************
 203  *           RegOpenKeyExA   [ADVAPI32.@]
 204  */
 205 DWORD WINAPI RegOpenKeyExA( HKEY hkey, LPCSTR name, DWORD reserved, REGSAM access, LPHKEY retkey )
 206 {
 207     OBJECT_ATTRIBUTES attr;
 208     STRING nameA;
 209     NTSTATUS status;
 210
 211     if (!is_version_nt()) access = KEY_ALL_ACCESS;  /* Win95 ignores the access mask */
 212
 213     attr.Length = sizeof(attr);
 214     attr.RootDirectory = hkey;
 215     attr.ObjectName = &NtCurrentTeb()->StaticUnicodeString;
 216     attr.Attributes = 0;
 217     attr.SecurityDescriptor = NULL;
 218     attr.SecurityQualityOfService = NULL;
 219
 220     RtlInitAnsiString( &nameA, name );
 221     if (!(status = RtlAnsiStringToUnicodeString( &NtCurrentTeb()->StaticUnicodeString,
 222                                                  &nameA, FALSE )))
 223     {
 224         status = NtOpenKey( retkey, access, &attr );
 225     }
 226     return RtlNtStatusToDosError( status );
 227 }
 228
 229
 230 /******************************************************************************
 231  *           RegOpenKeyW   [ADVAPI32.@]
 232  *
 233  * PARAMS
 234  *    hkey    [I] Handle of open key
 235  *    name    [I] Address of name of subkey to open
 236  *    retkey  [O] Handle to open key
 237  *
 238  * RETURNS
 239  *    Success: ERROR_SUCCESS
 240  *    Failure: Error code
 241  *
 242  * NOTES
 243  *  in case of failing is retkey = 0
 244  */
 245 DWORD WINAPI RegOpenKeyW( HKEY hkey, LPCWSTR name, LPHKEY retkey )
 246 {
 247     return RegOpenKeyExW( hkey, name, 0, KEY_ALL_ACCESS, retkey );
 248 }
 249
 250
 251 /******************************************************************************
 252  *           RegOpenKeyA   [ADVAPI32.@]
 253  */
 254 DWORD WINAPI RegOpenKeyA( HKEY hkey, LPCSTR name, LPHKEY retkey )
 255 {
 256     return RegOpenKeyExA( hkey, name, 0, KEY_ALL_ACCESS, retkey );
 257 }
 258
 259
 260 /******************************************************************************
 261  *           RegOpenCurrentUser   [ADVAPI32.@]
 262  * FIXME: This function is supposed to retrieve a handle to the
 263  * HKEY_CURRENT_USER for the user the current thread is impersonating.
 264  * Since Wine does not currently allow threads to impersonate other users,
 265  * this stub should work fine.
 266  */
 267 DWORD WINAPI RegOpenCurrentUser( REGSAM access, PHKEY retkey )
 268 {
 269     return RegOpenKeyExA( HKEY_CURRENT_USER, "", 0, access, retkey );
 270 }
 271
 272
 273
 274 /******************************************************************************
 275  *           RegEnumKeyExW   [ADVAPI32.@]
 276  *
 277  * PARAMS
 278  *    hkey         [I] Handle to key to enumerate
 279  *    index        [I] Index of subkey to enumerate
 280  *    name         [O] Buffer for subkey name
 281  *    name_len     [O] Size of subkey buffer
 282  *    reserved     [I] Reserved
 283  *    class        [O] Buffer for class string
 284  *    class_len    [O] Size of class buffer
 285  *    ft           [O] Time key last written to
 286  */
 287 DWORD WINAPI RegEnumKeyExW( HKEY hkey, DWORD index, LPWSTR name, LPDWORD name_len,
 288                             LPDWORD reserved, LPWSTR class, LPDWORD class_len, FILETIME *ft )
 289 {
 290     NTSTATUS status;
 291     char buffer[256], *buf_ptr = buffer;
 292     KEY_NODE_INFORMATION *info = (KEY_NODE_INFORMATION *)buffer;
 293     DWORD total_size;
 294
 295     TRACE( "(0x%x,%ld,%p,%p(%ld),%p,%p,%p,%p)\n", hkey, index, name, name_len,
 296            name_len ? *name_len : -1, reserved, class, class_len, ft );
 297
 298     if (reserved) return ERROR_INVALID_PARAMETER;
 299
 300     if (!hkey) return ERROR_INVALID_HANDLE;
 301
 302     status = NtEnumerateKey( hkey, index, KeyNodeInformation,
 303                              buffer, sizeof(buffer), &total_size );
 304
 305     while (status == STATUS_BUFFER_OVERFLOW)
 306     {
 307         /* retry with a dynamically allocated buffer */
 308         if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 309         if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 310             return ERROR_NOT_ENOUGH_MEMORY;
 311         info = (KEY_NODE_INFORMATION *)buf_ptr;
 312         status = NtEnumerateKey( hkey, index, KeyNodeInformation,
 313                                  buf_ptr, total_size, &total_size );
 314     }
 315
 316     if (!status)
 317     {
 318         DWORD len = info->NameLength / sizeof(WCHAR);
 319         DWORD cls_len = info->ClassLength / sizeof(WCHAR);
 320
 321         if (ft) *ft = *(FILETIME *)&info->LastWriteTime;
 322
 323         if (len >= *name_len || (class && class_len && (cls_len >= *class_len)))
 324             status = STATUS_BUFFER_OVERFLOW;
 325         else
 326         {
 327             *name_len = len;
 328             memcpy( name, info->Name, info->NameLength );
 329             name[len] = 0;
 330             if (class_len)
 331             {
 332                 *class_len = cls_len;
 333                 if (class)
 334                 {
 335                     memcpy( class, buf_ptr + info->ClassOffset, info->ClassLength );
 336                     class[cls_len] = 0;
 337                 }
 338             }
 339         }
 340     }
 341
 342     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 343     return RtlNtStatusToDosError( status );
 344 }
 345
 346
 347 /******************************************************************************
 348  *           RegEnumKeyExA   [ADVAPI32.@]
 349  */
 350 DWORD WINAPI RegEnumKeyExA( HKEY hkey, DWORD index, LPSTR name, LPDWORD name_len,
 351                             LPDWORD reserved, LPSTR class, LPDWORD class_len, FILETIME *ft )
 352 {
 353     NTSTATUS status;
 354     char buffer[256], *buf_ptr = buffer;
 355     KEY_NODE_INFORMATION *info = (KEY_NODE_INFORMATION *)buffer;
 356     DWORD total_size;
 357
 358     TRACE( "(0x%x,%ld,%p,%p(%ld),%p,%p,%p,%p)\n", hkey, index, name, name_len,
 359            name_len ? *name_len : -1, reserved, class, class_len, ft );
 360
 361     if (reserved) return ERROR_INVALID_PARAMETER;
 362
 363     if (!hkey) return ERROR_INVALID_HANDLE;
 364
 365     status = NtEnumerateKey( hkey, index, KeyNodeInformation,
 366                              buffer, sizeof(buffer), &total_size );
 367
 368     while (status == STATUS_BUFFER_OVERFLOW)
 369     {
 370         /* retry with a dynamically allocated buffer */
 371         if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 372         if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 373             return ERROR_NOT_ENOUGH_MEMORY;
 374         info = (KEY_NODE_INFORMATION *)buf_ptr;
 375         status = NtEnumerateKey( hkey, index, KeyNodeInformation,
 376                                  buf_ptr, total_size, &total_size );
 377     }
 378
 379     if (!status)
 380     {
 381         DWORD len, cls_len;
 382
 383         RtlUnicodeToMultiByteSize( &len, info->Name, info->NameLength );
 384         RtlUnicodeToMultiByteSize( &cls_len, (WCHAR *)(buf_ptr + info->ClassOffset),
 385                                    info->ClassLength );
 386         if (ft) *ft = *(FILETIME *)&info->LastWriteTime;
 387
 388         if (len >= *name_len || (class && class_len && (cls_len >= *class_len)))
 389             status = STATUS_BUFFER_OVERFLOW;
 390         else
 391         {
 392             *name_len = len;
 393             RtlUnicodeToMultiByteN( name, len, NULL, info->Name, info->NameLength );
 394             name[len] = 0;
 395             if (class_len)
 396             {
 397                 *class_len = cls_len;
 398                 if (class)
 399                 {
 400                     RtlUnicodeToMultiByteN( class, cls_len, NULL,
 401                                             (WCHAR *)(buf_ptr + info->ClassOffset),
 402                                             info->ClassLength );
 403                     class[cls_len] = 0;
 404                 }
 405             }
 406         }
 407     }
 408
 409     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 410     return RtlNtStatusToDosError( status );
 411 }
 412
 413
 414 /******************************************************************************
 415  *           RegEnumKeyW   [ADVAPI32.@]
 416  */
 417 DWORD WINAPI RegEnumKeyW( HKEY hkey, DWORD index, LPWSTR name, DWORD name_len )
 418 {
 419     return RegEnumKeyExW( hkey, index, name, &name_len, NULL, NULL, NULL, NULL );
 420 }
 421
 422
 423 /******************************************************************************
 424  *           RegEnumKeyA   [ADVAPI32.@]
 425  */
 426 DWORD WINAPI RegEnumKeyA( HKEY hkey, DWORD index, LPSTR name, DWORD name_len )
 427 {
 428     return RegEnumKeyExA( hkey, index, name, &name_len, NULL, NULL, NULL, NULL );
 429 }
 430
 431
 432 /******************************************************************************
 433  *           RegQueryInfoKeyW   [ADVAPI32.@]
 434  *
 435  * PARAMS
 436  *    hkey       [I] Handle to key to query
 437  *    class      [O] Buffer for class string
 438  *    class_len  [O] Size of class string buffer
 439  *    reserved   [I] Reserved
 440  *    subkeys    [O] Buffer for number of subkeys
 441  *    max_subkey [O] Buffer for longest subkey name length
 442  *    max_class  [O] Buffer for longest class string length
 443  *    values     [O] Buffer for number of value entries
 444  *    max_value  [O] Buffer for longest value name length
 445  *    max_data   [O] Buffer for longest value data length
 446  *    security   [O] Buffer for security descriptor length
 447  *    modif      [O] Modification time
 448  *
 449  * - win95 allows class to be valid and class_len to be NULL
 450  * - winnt returns ERROR_INVALID_PARAMETER if class is valid and class_len is NULL
 451  * - both allow class to be NULL and class_len to be NULL
 452  * (it's hard to test validity, so test !NULL instead)
 453  */
 454 DWORD WINAPI RegQueryInfoKeyW( HKEY hkey, LPWSTR class, LPDWORD class_len, LPDWORD reserved,
 455                                LPDWORD subkeys, LPDWORD max_subkey, LPDWORD max_class,
 456                                LPDWORD values, LPDWORD max_value, LPDWORD max_data,
 457                                LPDWORD security, FILETIME *modif )
 458 {
 459     NTSTATUS status;
 460     char buffer[256], *buf_ptr = buffer;
 461     KEY_FULL_INFORMATION *info = (KEY_FULL_INFORMATION *)buffer;
 462     DWORD total_size;
 463
 464     TRACE( "(0x%x,%p,%ld,%p,%p,%p,%p,%p,%p,%p,%p)\n", hkey, class, class_len ? *class_len : 0,
 465            reserved, subkeys, max_subkey, values, max_value, max_data, security, modif );
 466
 467     if (class && !class_len && is_version_nt()) return ERROR_INVALID_PARAMETER;
 468
 469     status = NtQueryKey( hkey, KeyFullInformation, buffer, sizeof(buffer), &total_size );
 470     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
 471
 472     if (class)
 473     {
 474         /* retry with a dynamically allocated buffer */
 475         while (status == STATUS_BUFFER_OVERFLOW)
 476         {
 477             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 478             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 479                 return ERROR_NOT_ENOUGH_MEMORY;
 480             info = (KEY_FULL_INFORMATION *)buf_ptr;
 481             status = NtQueryKey( hkey, KeyFullInformation, buf_ptr, total_size, &total_size );
 482         }
 483
 484         if (status) goto done;
 485
 486         if (class_len && (info->ClassLength/sizeof(WCHAR) + 1 > *class_len))
 487         {
 488             status = STATUS_BUFFER_OVERFLOW;
 489         }
 490         else
 491         {
 492             memcpy( class, buf_ptr + info->ClassOffset, info->ClassLength );
 493             class[info->ClassLength/sizeof(WCHAR)] = 0;
 494         }
 495     }
 496     else status = STATUS_SUCCESS;
 497
 498     if (class_len) *class_len = info->ClassLength / sizeof(WCHAR);
 499     if (subkeys) *subkeys = info->SubKeys;
 500     if (max_subkey) *max_subkey = info->MaxNameLen;
 501     if (max_class) *max_class = info->MaxClassLen;
 502     if (values) *values = info->Values;
 503     if (max_value) *max_value = info->MaxValueNameLen;
 504     if (max_data) *max_data = info->MaxValueDataLen;
 505     if (modif) *modif = *(FILETIME *)&info->LastWriteTime;
 506
 507  done:
 508     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 509     return RtlNtStatusToDosError( status );
 510 }
 511
 512
 513 /******************************************************************************
 514  *           RegQueryInfoKeyA   [ADVAPI32.@]
 515  */
 516 DWORD WINAPI RegQueryInfoKeyA( HKEY hkey, LPSTR class, LPDWORD class_len, LPDWORD reserved,
 517                                LPDWORD subkeys, LPDWORD max_subkey, LPDWORD max_class,
 518                                LPDWORD values, LPDWORD max_value, LPDWORD max_data,
 519                                LPDWORD security, FILETIME *modif )
 520 {
 521     NTSTATUS status;
 522     char buffer[256], *buf_ptr = buffer;
 523     KEY_FULL_INFORMATION *info = (KEY_FULL_INFORMATION *)buffer;
 524     DWORD total_size, len;
 525
 526     TRACE( "(0x%x,%p,%ld,%p,%p,%p,%p,%p,%p,%p,%p)\n", hkey, class, class_len ? *class_len : 0,
 527            reserved, subkeys, max_subkey, values, max_value, max_data, security, modif );
 528
 529     if (class && !class_len && is_version_nt()) return ERROR_INVALID_PARAMETER;
 530
 531     status = NtQueryKey( hkey, KeyFullInformation, buffer, sizeof(buffer), &total_size );
 532     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
 533
 534     if (class || class_len)
 535     {
 536         /* retry with a dynamically allocated buffer */
 537         while (status == STATUS_BUFFER_OVERFLOW)
 538         {
 539             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 540             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 541                 return ERROR_NOT_ENOUGH_MEMORY;
 542             info = (KEY_FULL_INFORMATION *)buf_ptr;
 543             status = NtQueryKey( hkey, KeyFullInformation, buf_ptr, total_size, &total_size );
 544         }
 545
 546         if (status) goto done;
 547
 548         RtlUnicodeToMultiByteSize( &len, (WCHAR *)(buf_ptr + info->ClassOffset), info->ClassLength);
 549         if (class_len)
 550         {
 551             if (len + 1 > *class_len) status = STATUS_BUFFER_OVERFLOW;
 552             *class_len = len;
 553         }
 554         if (class && !status)
 555         {
 556             RtlUnicodeToMultiByteN( class, len, NULL, (WCHAR *)(buf_ptr + info->ClassOffset),
 557                                     info->ClassLength );
 558             class[len] = 0;
 559         }
 560     }
 561     else status = STATUS_SUCCESS;
 562
 563     if (subkeys) *subkeys = info->SubKeys;
 564     if (max_subkey) *max_subkey = info->MaxNameLen;
 565     if (max_class) *max_class = info->MaxClassLen;
 566     if (values) *values = info->Values;
 567     if (max_value) *max_value = info->MaxValueNameLen;
 568     if (max_data) *max_data = info->MaxValueDataLen;
 569     if (modif) *modif = *(FILETIME *)&info->LastWriteTime;
 570
 571  done:
 572     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 573     return RtlNtStatusToDosError( status );
 574 }
 575
 576
 577 /******************************************************************************
 578  *           RegCloseKey   [ADVAPI32.@]
 579  *
 580  * Releases the handle of the specified key
 581  *
 582  * PARAMS
 583  *    hkey [I] Handle of key to close
 584  *
 585  * RETURNS
 586  *    Success: ERROR_SUCCESS
 587  *    Failure: Error code
 588  */
 589 DWORD WINAPI RegCloseKey( HKEY hkey )
 590 {
 591     if (!hkey || hkey >= 0x80000000) return ERROR_SUCCESS;
 592     return RtlNtStatusToDosError( NtClose( hkey ) );
 593 }
 594
 595
 596 /******************************************************************************
 597  *           RegDeleteKeyW   [ADVAPI32.@]
 598  *
 599  * PARAMS
 600  *    hkey   [I] Handle to open key
 601  *    name   [I] Name of subkey to delete
 602  *
 603  * RETURNS
 604  *    Success: ERROR_SUCCESS
 605  *    Failure: Error code
 606  */
 607 DWORD WINAPI RegDeleteKeyW( HKEY hkey, LPCWSTR name )
 608 {
 609     DWORD ret;
 610     HKEY tmp;
 611
 612     if (!name || !*name)
 613     {
 614         ret = RtlNtStatusToDosError( NtDeleteKey( hkey ) );
 615     }
 616     else if (!(ret = RegOpenKeyExW( hkey, name, 0, 0, &tmp )))
 617     {
 618         if (!is_version_nt()) /* win95 does recursive key deletes */
 619         {
 620             WCHAR name[MAX_PATH];
 621
 622             while(!RegEnumKeyW(tmp, 0, name, sizeof name))
 623             {
 624                 if(RegDeleteKeyW(tmp, name))  /* recurse */
 625                     break;
 626             }
 627         }
 628         ret = RtlNtStatusToDosError( NtDeleteKey( tmp ) );
 629         RegCloseKey( tmp );
 630     }
 631     TRACE("%s ret=%08lx\n", debugstr_w(name), ret);
 632     return ret;
 633 }
 634
 635
 636 /******************************************************************************
 637  *           RegDeleteKeyA   [ADVAPI32.@]
 638  */
 639 DWORD WINAPI RegDeleteKeyA( HKEY hkey, LPCSTR name )
 640 {
 641     DWORD ret;
 642     HKEY tmp;
 643
 644     if (!name || !*name)
 645     {
 646         ret = RtlNtStatusToDosError( NtDeleteKey( hkey ) );
 647     }
 648     else if (!(ret = RegOpenKeyExA( hkey, name, 0, KEY_ALL_ACCESS, &tmp )))
 649     {
 650         if (!is_version_nt()) /* win95 does recursive key deletes */
 651         {
 652             CHAR name[MAX_PATH];
 653
 654             while(!RegEnumKeyA(tmp, 0, name, sizeof name))
 655             {
 656                 if(RegDeleteKeyA(tmp, name))  /* recurse */
 657                     break;
 658             }
 659         }
 660         ret = RtlNtStatusToDosError( NtDeleteKey( tmp ) );
 661         RegCloseKey( tmp );
 662     }
 663     TRACE("%s ret=%08lx\n", debugstr_a(name), ret);
 664     return ret;
 665 }
 666
 667
 668
 669 /******************************************************************************
 670  *           RegSetValueExW   [ADVAPI32.@]
 671  *
 672  * Sets the data and type of a value under a register key
 673  *
 674  * PARAMS
 675  *    hkey       [I] Handle of key to set value for
 676  *    name       [I] Name of value to set
 677  *    reserved   [I] Reserved - must be zero
 678  *    type       [I] Flag for value type
 679  *    data       [I] Address of value data
 680  *    count      [I] Size of value data
 681  *
 682  * RETURNS
 683  *    Success: ERROR_SUCCESS
 684  *    Failure: Error code
 685  *
 686  * NOTES
 687  *   win95 does not care about count for REG_SZ and finds out the len by itself (js)
 688  *   NT does definitely care (aj)
 689  */
 690 DWORD WINAPI RegSetValueExW( HKEY hkey, LPCWSTR name, DWORD reserved,
 691                              DWORD type, CONST BYTE *data, DWORD count )
 692 {
 693     UNICODE_STRING nameW;
 694
 695     if (!is_version_nt())  /* win95 */
 696     {
 697         if (type == REG_SZ) count = (strlenW( (WCHAR *)data ) + 1) * sizeof(WCHAR);
 698     }
 699     else if (count && is_string(type))
 700     {
 701         LPCWSTR str = (LPCWSTR)data;
 702         /* if user forgot to count terminating null, add it (yes NT does this) */
 703         if (str[count / sizeof(WCHAR) - 1] && !str[count / sizeof(WCHAR)])
 704             count += sizeof(WCHAR);
 705     }
 706
 707     RtlInitUnicodeString( &nameW, name );
 708     return RtlNtStatusToDosError( NtSetValueKey( hkey, &nameW, 0, type, data, count ) );
 709 }
 710
 711
 712 /******************************************************************************
 713  *           RegSetValueExA   [ADVAPI32.@]
 714  */
 715 DWORD WINAPI RegSetValueExA( HKEY hkey, LPCSTR name, DWORD reserved, DWORD type,
 716                              CONST BYTE *data, DWORD count )
 717 {
 718     ANSI_STRING nameA;
 719     WCHAR *dataW = NULL;
 720     NTSTATUS status;
 721
 722     if (!is_version_nt())  /* win95 */
 723     {
 724         if (type == REG_SZ) count = strlen(data) + 1;
 725     }
 726     else if (count && is_string(type))
 727     {
 728         /* if user forgot to count terminating null, add it (yes NT does this) */
 729         if (data[count-1] && !data[count]) count++;
 730     }
 731
 732     if (is_string( type )) /* need to convert to Unicode */
 733     {
 734         DWORD lenW;
 735         RtlMultiByteToUnicodeSize( &lenW, data, count );
 736         if (!(dataW = HeapAlloc( GetProcessHeap(), 0, lenW ))) return ERROR_OUTOFMEMORY;
 737         RtlMultiByteToUnicodeN( dataW, lenW, NULL, data, count );
 738         count = lenW;
 739         data = (BYTE *)dataW;
 740     }
 741
 742     RtlInitAnsiString( &nameA, name );
 743     if (!(status = RtlAnsiStringToUnicodeString( &NtCurrentTeb()->StaticUnicodeString,
 744                                                  &nameA, FALSE )))
 745     {
 746         status = NtSetValueKey( hkey, &NtCurrentTeb()->StaticUnicodeString, 0, type, data, count );
 747     }
 748     if (dataW) HeapFree( GetProcessHeap(), 0, dataW );
 749     return RtlNtStatusToDosError( status );
 750 }
 751
 752
 753 /******************************************************************************
 754  *           RegSetValueW   [ADVAPI32.@]
 755  */
 756 DWORD WINAPI RegSetValueW( HKEY hkey, LPCWSTR name, DWORD type, LPCWSTR data, DWORD count )
 757 {
 758     HKEY subkey = hkey;
 759     DWORD ret;
 760
 761     TRACE("(0x%x,%s,%ld,%s,%ld)\n", hkey, debugstr_w(name), type, debugstr_w(data), count );
 762
 763     if (type != REG_SZ) return ERROR_INVALID_PARAMETER;
 764
 765     if (name && name[0])  /* need to create the subkey */
 766     {
 767         if ((ret = RegCreateKeyW( hkey, name, &subkey )) != ERROR_SUCCESS) return ret;
 768     }
 769
 770     ret = RegSetValueExW( subkey, NULL, 0, REG_SZ, (LPBYTE)data,
 771                           (strlenW( data ) + 1) * sizeof(WCHAR) );
 772     if (subkey != hkey) RegCloseKey( subkey );
 773     return ret;
 774 }
 775
 776
 777 /******************************************************************************
 778  *           RegSetValueA   [ADVAPI32.@]
 779  */
 780 DWORD WINAPI RegSetValueA( HKEY hkey, LPCSTR name, DWORD type, LPCSTR data, DWORD count )
 781 {
 782     HKEY subkey = hkey;
 783     DWORD ret;
 784
 785     TRACE("(0x%x,%s,%ld,%s,%ld)\n", hkey, debugstr_a(name), type, debugstr_a(data), count );
 786
 787     if (type != REG_SZ) return ERROR_INVALID_PARAMETER;
 788
 789     if (name && name[0])  /* need to create the subkey */
 790     {
 791         if ((ret = RegCreateKeyA( hkey, name, &subkey )) != ERROR_SUCCESS) return ret;
 792     }
 793     ret = RegSetValueExA( subkey, NULL, 0, REG_SZ, (LPBYTE)data, strlen(data)+1 );
 794     if (subkey != hkey) RegCloseKey( subkey );
 795     return ret;
 796 }
 797
 798
 799
 800 /******************************************************************************
 801  *           RegQueryValueExW   [ADVAPI32.@]
 802  *
 803  * Retrieves type and data for a specified name associated with an open key
 804  *
 805  * PARAMS
 806  *    hkey      [I]   Handle of key to query
 807  *    name      [I]   Name of value to query
 808  *    reserved  [I]   Reserved - must be NULL
 809  *    type      [O]   Address of buffer for value type.  If NULL, the type
 810  *                        is not required.
 811  *    data      [O]   Address of data buffer.  If NULL, the actual data is
 812  *                        not required.
 813  *    count     [I/O] Address of data buffer size
 814  *
 815  * RETURNS
 816  *    ERROR_SUCCESS:   Success
 817  *    ERROR_MORE_DATA: !!! if the specified buffer is not big enough to hold the data
 818  *                     buffer is left untouched. The MS-documentation is wrong (js) !!!
 819  */
 820 DWORD WINAPI RegQueryValueExW( HKEY hkey, LPCWSTR name, LPDWORD reserved, LPDWORD type,
 821                                LPBYTE data, LPDWORD count )
 822 {
 823     NTSTATUS status;
 824     UNICODE_STRING name_str;
 825     DWORD total_size;
 826     char buffer[256], *buf_ptr = buffer;
 827     KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
 828     static const int info_size = offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data );
 829
 830     TRACE("(0x%x,%s,%p,%p,%p,%p=%ld)\n",
 831           hkey, debugstr_w(name), reserved, type, data, count, count ? *count : 0 );
 832
 833     if ((data && !count) || reserved) return ERROR_INVALID_PARAMETER;
 834
 835     RtlInitUnicodeString( &name_str, name );
 836
 837     if (data) total_size = min( sizeof(buffer), *count + info_size );
 838     else total_size = info_size;
 839
 840     status = NtQueryValueKey( hkey, &name_str, KeyValuePartialInformation,
 841                               buffer, total_size, &total_size );
 842     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
 843
 844     if (data)
 845     {
 846         /* retry with a dynamically allocated buffer */
 847         while (status == STATUS_BUFFER_OVERFLOW && total_size - info_size <= *count)
 848         {
 849             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 850             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 851                 return ERROR_NOT_ENOUGH_MEMORY;
 852             info = (KEY_VALUE_PARTIAL_INFORMATION *)buf_ptr;
 853             status = NtQueryValueKey( hkey, &name_str, KeyValuePartialInformation,
 854                                       buf_ptr, total_size, &total_size );
 855         }
 856
 857         if (!status)
 858         {
 859             memcpy( data, buf_ptr + info_size, total_size - info_size );
 860             /* if the type is REG_SZ and data is not 0-terminated
 861              * and there is enough space in the buffer NT appends a \0 */
 862             if (total_size - info_size <= *count-sizeof(WCHAR) && is_string(info->Type))
 863             {
 864                 WCHAR *ptr = (WCHAR *)(data + total_size - info_size);
 865                 if (ptr > (WCHAR *)data && ptr[-1]) *ptr = 0;
 866             }
 867         }
 868         else if (status != STATUS_BUFFER_OVERFLOW) goto done;
 869     }
 870     else status = STATUS_SUCCESS;
 871
 872     if (type) *type = info->Type;
 873     if (count) *count = total_size - info_size;
 874
 875  done:
 876     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 877     return RtlNtStatusToDosError(status);
 878 }
 879
 880
 881 /******************************************************************************
 882  *           RegQueryValueExA   [ADVAPI32.@]
 883  *
 884  * NOTES:
 885  * the documentation is wrong: if the buffer is too small it remains untouched
 886  */
 887 DWORD WINAPI RegQueryValueExA( HKEY hkey, LPCSTR name, LPDWORD reserved, LPDWORD type,
 888                                LPBYTE data, LPDWORD count )
 889 {
 890     NTSTATUS status;
 891     ANSI_STRING nameA;
 892     DWORD total_size;
 893     char buffer[256], *buf_ptr = buffer;
 894     KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
 895     static const int info_size = offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data );
 896
 897     TRACE("(0x%x,%s,%p,%p,%p,%p=%ld)\n",
 898           hkey, debugstr_a(name), reserved, type, data, count, count ? *count : 0 );
 899
 900     if ((data && !count) || reserved) return ERROR_INVALID_PARAMETER;
 901
 902     RtlInitAnsiString( &nameA, name );
 903     if ((status = RtlAnsiStringToUnicodeString( &NtCurrentTeb()->StaticUnicodeString,
 904                                                 &nameA, FALSE )))
 905         return RtlNtStatusToDosError(status);
 906
 907     status = NtQueryValueKey( hkey, &NtCurrentTeb()->StaticUnicodeString,
 908                               KeyValuePartialInformation, buffer, sizeof(buffer), &total_size );
 909     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
 910
 911     /* we need to fetch the contents for a string type even if not requested,
 912      * because we need to compute the length of the ASCII string. */
 913     if (data || is_string(info->Type))
 914     {
 915         /* retry with a dynamically allocated buffer */
 916         while (status == STATUS_BUFFER_OVERFLOW)
 917         {
 918             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 919             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
 920             {
 921                 status = STATUS_NO_MEMORY;
 922                 goto done;
 923             }
 924             info = (KEY_VALUE_PARTIAL_INFORMATION *)buf_ptr;
 925             status = NtQueryValueKey( hkey, &NtCurrentTeb()->StaticUnicodeString,
 926                                     KeyValuePartialInformation, buf_ptr, total_size, &total_size );
 927         }
 928
 929         if (status) goto done;
 930
 931         if (is_string(info->Type))
 932         {
 933             DWORD len;
 934
 935             RtlUnicodeToMultiByteSize( &len, (WCHAR *)(buf_ptr + info_size),
 936                                        total_size - info_size );
 937             if (data && len)
 938             {
 939                 if (len > *count) status = STATUS_BUFFER_OVERFLOW;
 940                 else
 941                 {
 942                     RtlUnicodeToMultiByteN( data, len, NULL, (WCHAR *)(buf_ptr + info_size),
 943                                             total_size - info_size );
 944                     /* if the type is REG_SZ and data is not 0-terminated
 945                      * and there is enough space in the buffer NT appends a \0 */
 946                     if (len < *count && data[len-1]) data[len] = 0;
 947                 }
 948             }
 949             total_size = len + info_size;
 950         }
 951         else if (data)
 952         {
 953             if (total_size - info_size > *count) status = STATUS_BUFFER_OVERFLOW;
 954             else memcpy( data, buf_ptr + info_size, total_size - info_size );
 955         }
 956     }
 957     else status = STATUS_SUCCESS;
 958
 959     if (type) *type = info->Type;
 960     if (count) *count = total_size - info_size;
 961
 962  done:
 963     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
 964     return RtlNtStatusToDosError(status);
 965 }
 966
 967
 968 /******************************************************************************
 969  *           RegQueryValueW   [ADVAPI32.@]
 970  */
 971 DWORD WINAPI RegQueryValueW( HKEY hkey, LPCWSTR name, LPWSTR data, LPLONG count )
 972 {
 973     DWORD ret;
 974     HKEY subkey = hkey;
 975
 976     TRACE("(%x,%s,%p,%ld)\n", hkey, debugstr_w(name), data, count ? *count : 0 );
 977
 978     if (name && name[0])
 979     {
 980         if ((ret = RegOpenKeyW( hkey, name, &subkey )) != ERROR_SUCCESS) return ret;
 981     }
 982     ret = RegQueryValueExW( subkey, NULL, NULL, NULL, (LPBYTE)data, count );
 983     if (subkey != hkey) RegCloseKey( subkey );
 984     if (ret == ERROR_FILE_NOT_FOUND)
 985     {
 986         /* return empty string if default value not found */
 987         if (data) *data = 0;
 988         if (count) *count = 1;
 989         ret = ERROR_SUCCESS;
 990     }
 991     return ret;
 992 }
 993
 994
 995 /******************************************************************************
 996  *           RegQueryValueA   [ADVAPI32.@]
 997  */
 998 DWORD WINAPI RegQueryValueA( HKEY hkey, LPCSTR name, LPSTR data, LPLONG count )
 999 {
1000     DWORD ret;
1001     HKEY subkey = hkey;
1002
1003     TRACE("(%x,%s,%p,%ld)\n", hkey, debugstr_a(name), data, count ? *count : 0 );
1004
1005     if (name && name[0])
1006     {
1007         if ((ret = RegOpenKeyA( hkey, name, &subkey )) != ERROR_SUCCESS) return ret;
1008     }
1009     ret = RegQueryValueExA( subkey, NULL, NULL, NULL, (LPBYTE)data, count );
1010     if (subkey != hkey) RegCloseKey( subkey );
1011     if (ret == ERROR_FILE_NOT_FOUND)
1012     {
1013         /* return empty string if default value not found */
1014         if (data) *data = 0;
1015         if (count) *count = 1;
1016         ret = ERROR_SUCCESS;
1017     }
1018     return ret;
1019 }
1020
1021
1022 /******************************************************************************
1023  *           RegEnumValueW   [ADVAPI32.@]
1024  *
1025  * PARAMS
1026  *    hkey       [I] Handle to key to query
1027  *    index      [I] Index of value to query
1028  *    value      [O] Value string
1029  *    val_count  [I/O] Size of value buffer (in wchars)
1030  *    reserved   [I] Reserved
1031  *    type       [O] Type code
1032  *    data       [O] Value data
1033  *    count      [I/O] Size of data buffer (in bytes)
1034  */
1035
1036 DWORD WINAPI RegEnumValueW( HKEY hkey, DWORD index, LPWSTR value, LPDWORD val_count,
1037                             LPDWORD reserved, LPDWORD type, LPBYTE data, LPDWORD count )
1038 {
1039     NTSTATUS status;
1040     DWORD total_size;
1041     char buffer[256], *buf_ptr = buffer;
1042     KEY_VALUE_FULL_INFORMATION *info = (KEY_VALUE_FULL_INFORMATION *)buffer;
1043     static const int info_size = offsetof( KEY_VALUE_FULL_INFORMATION, Name );
1044
1045     TRACE("(%x,%ld,%p,%p,%p,%p,%p,%p)\n",
1046           hkey, index, value, val_count, reserved, type, data, count );
1047
1048     /* NT only checks count, not val_count */
1049     if ((data && !count) || reserved) return ERROR_INVALID_PARAMETER;
1050
1051     total_size = info_size + (MAX_PATH + 1) * sizeof(WCHAR);
1052     if (data) total_size += *count;
1053     total_size = min( sizeof(buffer), total_size );
1054
1055     status = NtEnumerateValueKey( hkey, index, KeyValueFullInformation,
1056                                   buffer, total_size, &total_size );
1057     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
1058
1059     if (value || data)
1060     {
1061         /* retry with a dynamically allocated buffer */
1062         while (status == STATUS_BUFFER_OVERFLOW)
1063         {
1064             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
1065             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
1066                 return ERROR_NOT_ENOUGH_MEMORY;
1067             info = (KEY_VALUE_FULL_INFORMATION *)buf_ptr;
1068             status = NtEnumerateValueKey( hkey, index, KeyValueFullInformation,
1069                                           buf_ptr, total_size, &total_size );
1070         }
1071
1072         if (status) goto done;
1073
1074         if (value)
1075         {
1076             if (info->NameLength/sizeof(WCHAR) >= *val_count)
1077             {
1078                 status = STATUS_BUFFER_OVERFLOW;
1079                 goto overflow;
1080             }
1081             memcpy( value, info->Name, info->NameLength );
1082             *val_count = info->NameLength / sizeof(WCHAR);
1083             value[*val_count] = 0;
1084         }
1085
1086         if (data)
1087         {
1088             if (total_size - info->DataOffset > *count)
1089             {
1090                 status = STATUS_BUFFER_OVERFLOW;
1091                 goto overflow;
1092             }
1093             memcpy( data, buf_ptr + info->DataOffset, total_size - info->DataOffset );
1094             if (total_size - info->DataOffset <= *count-sizeof(WCHAR) && is_string(info->Type))
1095             {
1096                 /* if the type is REG_SZ and data is not 0-terminated
1097                  * and there is enough space in the buffer NT appends a \0 */
1098                 WCHAR *ptr = (WCHAR *)(data + total_size - info->DataOffset);
1099                 if (ptr > (WCHAR *)data && ptr[-1]) *ptr = 0;
1100             }
1101         }
1102     }
1103     else status = STATUS_SUCCESS;
1104
1105  overflow:
1106     if (type) *type = info->Type;
1107     if (count) *count = info->DataLength;
1108
1109  done:
1110     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
1111     return RtlNtStatusToDosError(status);
1112 }
1113
1114
1115 /******************************************************************************
1116  *           RegEnumValueA   [ADVAPI32.@]
1117  */
1118 DWORD WINAPI RegEnumValueA( HKEY hkey, DWORD index, LPSTR value, LPDWORD val_count,
1119                             LPDWORD reserved, LPDWORD type, LPBYTE data, LPDWORD count )
1120 {
1121     NTSTATUS status;
1122     DWORD total_size;
1123     char buffer[256], *buf_ptr = buffer;
1124     KEY_VALUE_FULL_INFORMATION *info = (KEY_VALUE_FULL_INFORMATION *)buffer;
1125     static const int info_size = offsetof( KEY_VALUE_FULL_INFORMATION, Name );
1126
1127     TRACE("(%x,%ld,%p,%p,%p,%p,%p,%p)\n",
1128           hkey, index, value, val_count, reserved, type, data, count );
1129
1130     /* NT only checks count, not val_count */
1131     if ((data && !count) || reserved) return ERROR_INVALID_PARAMETER;
1132
1133     total_size = info_size + (MAX_PATH + 1) * sizeof(WCHAR);
1134     if (data) total_size += *count;
1135     total_size = min( sizeof(buffer), total_size );
1136
1137     status = NtEnumerateValueKey( hkey, index, KeyValueFullInformation,
1138                                   buffer, total_size, &total_size );
1139     if (status && status != STATUS_BUFFER_OVERFLOW) goto done;
1140
1141     /* we need to fetch the contents for a string type even if not requested,
1142      * because we need to compute the length of the ASCII string. */
1143     if (value || data || is_string(info->Type))
1144     {
1145         /* retry with a dynamically allocated buffer */
1146         while (status == STATUS_BUFFER_OVERFLOW)
1147         {
1148             if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
1149             if (!(buf_ptr = HeapAlloc( GetProcessHeap(), 0, total_size )))
1150                 return ERROR_NOT_ENOUGH_MEMORY;
1151             info = (KEY_VALUE_FULL_INFORMATION *)buf_ptr;
1152             status = NtEnumerateValueKey( hkey, index, KeyValueFullInformation,
1153                                           buf_ptr, total_size, &total_size );
1154         }
1155
1156         if (status) goto done;
1157
1158         if (is_string(info->Type))
1159         {
1160             DWORD len;
1161             RtlUnicodeToMultiByteSize( &len, (WCHAR *)(buf_ptr + info->DataOffset),
1162                                        total_size - info->DataOffset );
1163             if (data && len)
1164             {
1165                 if (len > *count) status = STATUS_BUFFER_OVERFLOW;
1166                 else
1167                 {
1168                     RtlUnicodeToMultiByteN( data, len, NULL, (WCHAR *)(buf_ptr + info->DataOffset),
1169                                             total_size - info->DataOffset );
1170                     /* if the type is REG_SZ and data is not 0-terminated
1171                      * and there is enough space in the buffer NT appends a \0 */
1172                     if (len < *count && data[len-1]) data[len] = 0;
1173                 }
1174             }
1175             info->DataLength = len;
1176         }
1177         else if (data)
1178         {
1179             if (total_size - info->DataOffset > *count) status = STATUS_BUFFER_OVERFLOW;
1180             else memcpy( data, buf_ptr + info->DataOffset, total_size - info->DataOffset );
1181         }
1182
1183         if (value && !status)
1184         {
1185             DWORD len;
1186
1187             RtlUnicodeToMultiByteSize( &len, info->Name, info->NameLength );
1188             if (len >= *val_count)
1189             {
1190                 status = STATUS_BUFFER_OVERFLOW;
1191                 if (*val_count)
1192                 {
1193                     len = *val_count - 1;
1194                     RtlUnicodeToMultiByteN( value, len, NULL, info->Name, info->NameLength );
1195                     value[len] = 0;
1196                 }
1197             }
1198             else
1199             {
1200                 RtlUnicodeToMultiByteN( value, len, NULL, info->Name, info->NameLength );
1201                 value[len] = 0;
1202                 *val_count = len;
1203             }
1204         }
1205     }
1206     else status = STATUS_SUCCESS;
1207
1208     if (type) *type = info->Type;
1209     if (count) *count = info->DataLength;
1210
1211  done:
1212     if (buf_ptr != buffer) HeapFree( GetProcessHeap(), 0, buf_ptr );
1213     return RtlNtStatusToDosError(status);
1214 }
1215
1216
1217
1218 /******************************************************************************
1219  *           RegDeleteValueW   [ADVAPI32.@]
1220  *
1221  * PARAMS
1222  *    hkey   [I] handle to key
1223  *    name   [I] name of value to delete
1224  *
1225  * RETURNS
1226  *    error status
1227  */
1228 DWORD WINAPI RegDeleteValueW( HKEY hkey, LPCWSTR name )
1229 {
1230     UNICODE_STRING nameW;
1231     RtlInitUnicodeString( &nameW, name );
1232     return RtlNtStatusToDosError( NtDeleteValueKey( hkey, &nameW ) );
1233 }
1234
1235
1236 /******************************************************************************
1237  *           RegDeleteValueA   [ADVAPI32.@]
1238  */
1239 DWORD WINAPI RegDeleteValueA( HKEY hkey, LPCSTR name )
1240 {
1241     STRING nameA;
1242     NTSTATUS status;
1243
1244     RtlInitAnsiString( &nameA, name );
1245     if (!(status = RtlAnsiStringToUnicodeString( &NtCurrentTeb()->StaticUnicodeString,
1246                                                  &nameA, FALSE )))
1247     {
1248         status = NtDeleteValueKey( hkey, &NtCurrentTeb()->StaticUnicodeString );
1249     }
1250     return RtlNtStatusToDosError( status );
1251 }
1252
1253
1254 /******************************************************************************
1255  *           RegLoadKeyW   [ADVAPI32.@]
1256  *
1257  * PARAMS
1258  *    hkey      [I] Handle of open key
1259  *    subkey    [I] Address of name of subkey
1260  *    filename  [I] Address of filename for registry information
1261  */
1262 LONG WINAPI RegLoadKeyW( HKEY hkey, LPCWSTR subkey, LPCWSTR filename )
1263 {
1264     HANDLE file;
1265     DWORD ret, len, err = GetLastError();
1266
1267     TRACE( "(%x,%s,%s)\n", hkey, debugstr_w(subkey), debugstr_w(filename) );
1268
1269     if (!filename || !*filename) return ERROR_INVALID_PARAMETER;
1270     if (!subkey || !*subkey) return ERROR_INVALID_PARAMETER;
1271
1272     len = strlenW( subkey ) * sizeof(WCHAR);
1273     if (len > MAX_PATH*sizeof(WCHAR)) return ERROR_INVALID_PARAMETER;
1274
1275     if ((file = CreateFileW( filename, GENERIC_READ, 0, NULL, OPEN_EXISTING,
1276                              FILE_ATTRIBUTE_NORMAL, 0 )) == INVALID_HANDLE_VALUE)
1277     {
1278         ret = GetLastError();
1279         goto done;
1280     }
1281
1282     SERVER_START_REQ( load_registry )
1283     {
1284         req->hkey  = hkey;
1285         req->file  = file;
1286         wine_server_add_data( req, subkey, len );
1287         ret = RtlNtStatusToDosError( wine_server_call(req) );
1288     }
1289     SERVER_END_REQ;
1290     CloseHandle( file );
1291
1292  done:
1293     SetLastError( err );  /* restore the last error code */
1294     return ret;
1295 }
1296
1297
1298 /******************************************************************************
1299  *           RegLoadKeyA   [ADVAPI32.@]
1300  */
1301 LONG WINAPI RegLoadKeyA( HKEY hkey, LPCSTR subkey, LPCSTR filename )
1302 {
1303     WCHAR buffer[MAX_PATH];
1304     HANDLE file;
1305     DWORD ret, len, err = GetLastError();
1306
1307     TRACE( "(%x,%s,%s)\n", hkey, debugstr_a(subkey), debugstr_a(filename) );
1308
1309     if (!filename || !*filename) return ERROR_INVALID_PARAMETER;
1310     if (!subkey || !*subkey) return ERROR_INVALID_PARAMETER;
1311
1312     if (!(len = MultiByteToWideChar( CP_ACP, 0, subkey, strlen(subkey), buffer, MAX_PATH )))
1313         return ERROR_INVALID_PARAMETER;
1314
1315     if ((file = CreateFileA( filename, GENERIC_READ, 0, NULL, OPEN_EXISTING,
1316                              FILE_ATTRIBUTE_NORMAL, 0 )) == INVALID_HANDLE_VALUE)
1317     {
1318         ret = GetLastError();
1319         goto done;
1320     }
1321
1322     SERVER_START_REQ( load_registry )
1323     {
1324         req->hkey  = hkey;
1325         req->file  = file;
1326         wine_server_add_data( req, buffer, len * sizeof(WCHAR) );
1327         ret = RtlNtStatusToDosError( wine_server_call(req) );
1328     }
1329     SERVER_END_REQ;
1330     CloseHandle( file );
1331
1332  done:
1333     SetLastError( err );  /* restore the last error code */
1334     return ret;
1335 }
1336
1337
1338 /******************************************************************************
1339  *           RegSaveKeyA   [ADVAPI32.@]
1340  *
1341  * PARAMS
1342  *    hkey   [I] Handle of key where save begins
1343  *    lpFile [I] Address of filename to save to
1344  *    sa     [I] Address of security structure
1345  */
1346 LONG WINAPI RegSaveKeyA( HKEY hkey, LPCSTR file, LPSECURITY_ATTRIBUTES sa )
1347 {
1348     char buffer[1024];
1349     int count = 0;
1350     LPSTR name;
1351     DWORD ret, err;
1352     HFILE handle;
1353
1354     TRACE( "(%x,%s,%p)\n", hkey, debugstr_a(file), sa );
1355
1356     if (!file || !*file) return ERROR_INVALID_PARAMETER;
1357
1358     err = GetLastError();
1359     GetFullPathNameA( file, sizeof(buffer), buffer, &name );
1360     for (;;)
1361     {
1362         sprintf( name, "reg%04x.tmp", count++ );
1363         handle = CreateFileA( buffer, GENERIC_WRITE, 0, NULL,
1364                             CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0 );
1365         if (handle != INVALID_HANDLE_VALUE) break;
1366         if ((ret = GetLastError()) != ERROR_ALREADY_EXISTS) goto done;
1367
1368         /* Something gone haywire ? Please report if this happens abnormally */
1369         if (count >= 100)
1370             MESSAGE("Wow, we are already fiddling with a temp file %s with an ordinal as high as %d !\nYou might want to delete all corresponding temp files in that directory.\n", buffer, count);
1371     }
1372
1373     SERVER_START_REQ( save_registry )
1374     {
1375         req->hkey = hkey;
1376         req->file = handle;
1377         ret = RtlNtStatusToDosError( wine_server_call( req ) );
1378     }
1379     SERVER_END_REQ;
1380
1381     CloseHandle( handle );
1382     if (!ret)
1383     {
1384         if (!MoveFileExA( buffer, file, MOVEFILE_REPLACE_EXISTING ))
1385         {
1386             ERR( "Failed to move %s to %s\n", buffer, file );
1387             ret = GetLastError();
1388         }
1389     }
1390     if (ret) DeleteFileA( buffer );
1391
1392 done:
1393     SetLastError( err );  /* restore last error code */
1394     return ret;
1395 }
1396
1397
1398 /******************************************************************************
1399  *           RegSaveKeyW   [ADVAPI32.@]
1400  */
1401 LONG WINAPI RegSaveKeyW( HKEY hkey, LPCWSTR file, LPSECURITY_ATTRIBUTES sa )
1402 {
1403     LPSTR fileA = HEAP_strdupWtoA( GetProcessHeap(), 0, file );
1404     DWORD ret = RegSaveKeyA( hkey, fileA, sa );
1405     if (fileA) HeapFree( GetProcessHeap(), 0, fileA );
1406     return ret;
1407 }
1408
1409
1410 /******************************************************************************
1411  * RegRestoreKeyW [ADVAPI32.@]
1412  *
1413  * PARAMS
1414  *    hkey    [I] Handle of key where restore begins
1415  *    lpFile  [I] Address of filename containing saved tree
1416  *    dwFlags [I] Optional flags
1417  */
1418 LONG WINAPI RegRestoreKeyW( HKEY hkey, LPCWSTR lpFile, DWORD dwFlags )
1419 {
1420     TRACE("(%x,%s,%ld)\n",hkey,debugstr_w(lpFile),dwFlags);
1421
1422     /* It seems to do this check before the hkey check */
1423     if (!lpFile || !*lpFile)
1424         return ERROR_INVALID_PARAMETER;
1425
1426     FIXME("(%x,%s,%ld): stub\n",hkey,debugstr_w(lpFile),dwFlags);
1427
1428     /* Check for file existence */
1429
1430     return ERROR_SUCCESS;
1431 }
1432
1433
1434 /******************************************************************************
1435  * RegRestoreKeyA [ADVAPI32.@]
1436  */
1437 LONG WINAPI RegRestoreKeyA( HKEY hkey, LPCSTR lpFile, DWORD dwFlags )
1438 {
1439     LPWSTR lpFileW = HEAP_strdupAtoW( GetProcessHeap(), 0, lpFile );
1440     LONG ret = RegRestoreKeyW( hkey, lpFileW, dwFlags );
1441     HeapFree( GetProcessHeap(), 0, lpFileW );
1442     return ret;
1443 }
1444
1445
1446 /******************************************************************************
1447  * RegUnLoadKeyW [ADVAPI32.@]
1448  *
1449  * PARAMS
1450  *    hkey     [I] Handle of open key
1451  *    lpSubKey [I] Address of name of subkey to unload
1452  */
1453 LONG WINAPI RegUnLoadKeyW( HKEY hkey, LPCWSTR lpSubKey )
1454 {
1455     FIXME("(%x,%s): stub\n",hkey, debugstr_w(lpSubKey));
1456     return ERROR_SUCCESS;
1457 }
1458
1459
1460 /******************************************************************************
1461  * RegUnLoadKeyA [ADVAPI32.@]
1462  */
1463 LONG WINAPI RegUnLoadKeyA( HKEY hkey, LPCSTR lpSubKey )
1464 {
1465     LPWSTR lpSubKeyW = HEAP_strdupAtoW( GetProcessHeap(), 0, lpSubKey );
1466     LONG ret = RegUnLoadKeyW( hkey, lpSubKeyW );
1467     if(lpSubKeyW) HeapFree( GetProcessHeap(), 0, lpSubKeyW);
1468     return ret;
1469 }
1470
1471
1472 /******************************************************************************
1473  * RegReplaceKeyW [ADVAPI32.@]
1474  *
1475  * PARAMS
1476  *    hkey      [I] Handle of open key
1477  *    lpSubKey  [I] Address of name of subkey
1478  *    lpNewFile [I] Address of filename for file with new data
1479  *    lpOldFile [I] Address of filename for backup file
1480  */
1481 LONG WINAPI RegReplaceKeyW( HKEY hkey, LPCWSTR lpSubKey, LPCWSTR lpNewFile,
1482                               LPCWSTR lpOldFile )
1483 {
1484     FIXME("(%x,%s,%s,%s): stub\n", hkey, debugstr_w(lpSubKey),
1485           debugstr_w(lpNewFile),debugstr_w(lpOldFile));
1486     return ERROR_SUCCESS;
1487 }
1488
1489
1490 /******************************************************************************
1491  * RegReplaceKeyA [ADVAPI32.@]
1492  */
1493 LONG WINAPI RegReplaceKeyA( HKEY hkey, LPCSTR lpSubKey, LPCSTR lpNewFile,
1494                               LPCSTR lpOldFile )
1495 {
1496     LPWSTR lpSubKeyW = HEAP_strdupAtoW( GetProcessHeap(), 0, lpSubKey );
1497     LPWSTR lpNewFileW = HEAP_strdupAtoW( GetProcessHeap(), 0, lpNewFile );
1498     LPWSTR lpOldFileW = HEAP_strdupAtoW( GetProcessHeap(), 0, lpOldFile );
1499     LONG ret = RegReplaceKeyW( hkey, lpSubKeyW, lpNewFileW, lpOldFileW );
1500     HeapFree( GetProcessHeap(), 0, lpOldFileW );
1501     HeapFree( GetProcessHeap(), 0, lpNewFileW );
1502     HeapFree( GetProcessHeap(), 0, lpSubKeyW );
1503     return ret;
1504 }
1505
1506
1507 /******************************************************************************
1508  * RegSetKeySecurity [ADVAPI32.@]
1509  *
1510  * PARAMS
1511  *    hkey          [I] Open handle of key to set
1512  *    SecurityInfo  [I] Descriptor contents
1513  *    pSecurityDesc [I] Address of descriptor for key
1514  */
1515 LONG WINAPI RegSetKeySecurity( HKEY hkey, SECURITY_INFORMATION SecurityInfo,
1516                                PSECURITY_DESCRIPTOR pSecurityDesc )
1517 {
1518     TRACE("(%x,%ld,%p)\n",hkey,SecurityInfo,pSecurityDesc);
1519
1520     /* It seems to perform this check before the hkey check */
1521     if ((SecurityInfo & OWNER_SECURITY_INFORMATION) ||
1522         (SecurityInfo & GROUP_SECURITY_INFORMATION) ||
1523         (SecurityInfo & DACL_SECURITY_INFORMATION) ||
1524         (SecurityInfo & SACL_SECURITY_INFORMATION)) {
1525         /* Param OK */
1526     } else
1527         return ERROR_INVALID_PARAMETER;
1528
1529     if (!pSecurityDesc)
1530         return ERROR_INVALID_PARAMETER;
1531
1532     FIXME(":(%x,%ld,%p): stub\n",hkey,SecurityInfo,pSecurityDesc);
1533
1534     return ERROR_SUCCESS;
1535 }
1536
1537
1538 /******************************************************************************
1539  * RegGetKeySecurity [ADVAPI32.@]
1540  * Retrieves a copy of security descriptor protecting the registry key
1541  *
1542  * PARAMS
1543  *    hkey                   [I]   Open handle of key to set
1544  *    SecurityInformation    [I]   Descriptor contents
1545  *    pSecurityDescriptor    [O]   Address of descriptor for key
1546  *    lpcbSecurityDescriptor [I/O] Address of size of buffer and description
1547  *
1548  * RETURNS
1549  *    Success: ERROR_SUCCESS
1550  *    Failure: Error code
1551  */
1552 LONG WINAPI RegGetKeySecurity( HKEY hkey, SECURITY_INFORMATION SecurityInformation,
1553                                PSECURITY_DESCRIPTOR pSecurityDescriptor,
1554                                LPDWORD lpcbSecurityDescriptor )
1555 {
1556     TRACE("(%x,%ld,%p,%ld)\n",hkey,SecurityInformation,pSecurityDescriptor,
1557           lpcbSecurityDescriptor?*lpcbSecurityDescriptor:0);
1558
1559     /* FIXME: Check for valid SecurityInformation values */
1560
1561     if (*lpcbSecurityDescriptor < sizeof(SECURITY_DESCRIPTOR))
1562         return ERROR_INSUFFICIENT_BUFFER;
1563
1564     FIXME("(%x,%ld,%p,%ld): stub\n",hkey,SecurityInformation,
1565           pSecurityDescriptor,lpcbSecurityDescriptor?*lpcbSecurityDescriptor:0);
1566
1567     /* Do not leave security descriptor filled with garbage */
1568     RtlCreateSecurityDescriptor(pSecurityDescriptor, SECURITY_DESCRIPTOR_REVISION);
1569
1570     return ERROR_SUCCESS;
1571 }
1572
1573
1574 /******************************************************************************
1575  * RegFlushKey [ADVAPI32.@]
1576  * Immediately writes key to registry.
1577  * Only returns after data has been written to disk.
1578  *
1579  * FIXME: does it really wait until data is written ?
1580  *
1581  * PARAMS
1582  *    hkey [I] Handle of key to write
1583  *
1584  * RETURNS
1585  *    Success: ERROR_SUCCESS
1586  *    Failure: Error code
1587  */
1588 DWORD WINAPI RegFlushKey( HKEY hkey )
1589 {
1590     FIXME( "(%x): stub\n", hkey );
1591     return ERROR_SUCCESS;
1592 }
1593
1594
1595 /******************************************************************************
1596  * RegConnectRegistryW [ADVAPI32.@]
1597  *
1598  * PARAMS
1599  *    lpMachineName [I] Address of name of remote computer
1600  *    hHey          [I] Predefined registry handle
1601  *    phkResult     [I] Address of buffer for remote registry handle
1602  */
1603 LONG WINAPI RegConnectRegistryW( LPCWSTR lpMachineName, HKEY hKey,
1604                                    LPHKEY phkResult )
1605 {
1606     TRACE("(%s,%x,%p): stub\n",debugstr_w(lpMachineName),hKey,phkResult);
1607
1608     if (!lpMachineName || !*lpMachineName) {
1609         /* Use the local machine name */
1610         return RegOpenKeyA( hKey, "", phkResult );
1611     }
1612
1613     FIXME("Cannot connect to %s\n",debugstr_w(lpMachineName));
1614     return ERROR_BAD_NETPATH;
1615 }
1616
1617
1618 /******************************************************************************
1619  * RegConnectRegistryA [ADVAPI32.@]
1620  */
1621 LONG WINAPI RegConnectRegistryA( LPCSTR machine, HKEY hkey, LPHKEY reskey )
1622 {
1623     LPWSTR machineW = HEAP_strdupAtoW( GetProcessHeap(), 0, machine );
1624     DWORD ret = RegConnectRegistryW( machineW, hkey, reskey );
1625     HeapFree( GetProcessHeap(), 0, machineW );
1626     return ret;
1627 }
1628
1629
1630 /******************************************************************************
1631  * RegNotifyChangeKeyValue [ADVAPI32.@]
1632  *
1633  * PARAMS
1634  *    hkey            [I] Handle of key to watch
1635  *    fWatchSubTree   [I] Flag for subkey notification
1636  *    fdwNotifyFilter [I] Changes to be reported
1637  *    hEvent          [I] Handle of signaled event
1638  *    fAsync          [I] Flag for asynchronous reporting
1639  */
1640 LONG WINAPI RegNotifyChangeKeyValue( HKEY hkey, BOOL fWatchSubTree,
1641                                      DWORD fdwNotifyFilter, HANDLE hEvent,
1642                                      BOOL fAsync )
1643 {
1644     FIXME("(%x,%i,%ld,%x,%i): stub\n",hkey,fWatchSubTree,fdwNotifyFilter,
1645           hEvent,fAsync);
1646     return ERROR_SUCCESS;
1647 }