[ros-diffs] [gbrunmar] 30190: Made ServiceCurrent key volatile

[gbrunmar at svn.reactos.org]

Author: gbrunmar
Date: Mon Nov  5 20:41:20 2007
New Revision: 30190

URL: http://svn.reactos.org/svn/reactos?rev=30190&view=rev
Log:
Made ServiceCurrent key volatile

Modified:
    trunk/reactos/base/system/services/database.c

Modified: trunk/reactos/base/system/services/database.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/base/system/services/database.c?rev=30190&r1=30189&r2=30190&view=diff
==============================================================================
--- trunk/reactos/base/system/services/database.c (original)
+++ trunk/reactos/base/system/services/database.c Mon Nov  5 20:41:20 2007
@@ -698,6 +698,8 @@
     NTSTATUS Status;
     DWORD dwError = ERROR_SUCCESS;
     WCHAR NtControlPipeName[MAX_PATH + 1];
+    HKEY hServiceCurrentKey = INVALID_HANDLE_VALUE;
+    DWORD KeyDisposition;
 
     RtlInitUnicodeString(&ImagePath, NULL);
 
@@ -727,39 +729,44 @@
     DPRINT("Type: %lx\n", Type);
 
     /* Get the service number */
-    RtlZeroMemory(&QueryTable,
-                  sizeof(QueryTable));
-
-    QueryTable[0].Name = L"";
-    QueryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_QUERY_REGISTRY_REQUIRED;
-    QueryTable[0].EntryContext = &ServiceCurrent;
-
-    Status = RtlQueryRegistryValues(RTL_REGISTRY_CONTROL,
-                                    L"ServiceCurrent",
-                                    QueryTable,
-                                    NULL,
-                                    NULL);
-
-    if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
-    {
-        /* TODO: Create registry entry with correct write access */
-    }
-    else if (!NT_SUCCESS(Status))
-    {
-        DPRINT1("RtlQueryRegistryValues() failed (Status %lx)\n", Status);
-        return RtlNtStatusToDosError(Status);
-    }
-    else
-    {
+    /* TODO: Create registry entry with correct write access */
+    Status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
+                            L"SYSTEM\\CurrentControlSet\\Control\\ServiceCurrent", 0, NULL,
+                            REG_OPTION_VOLATILE,
+                            KEY_WRITE | KEY_READ,
+                            NULL,
+                            &hServiceCurrentKey,
+                            &KeyDisposition);
+
+    if (ERROR_SUCCESS != Status)
+    {
+        DPRINT1("RegCreateKeyEx() failed with status %u\n", Status);
+        return Status;
+    }
+
+    if (REG_OPENED_EXISTING_KEY == KeyDisposition)
+    {
+        DWORD KeySize = sizeof(ServiceCurrent);
+        Status = RegQueryValueExW(hServiceCurrentKey, L"", 0, NULL, (BYTE*)&ServiceCurrent, &KeySize);
+
+        if (ERROR_SUCCESS != Status)
+        {
+            RegCloseKey(hServiceCurrentKey);
+            DPRINT1("RegQueryValueEx() failed with status %u\n", Status);
+            return Status;
+        }
+
         ServiceCurrent++;
     }
 
-    Status = RtlWriteRegistryValue(RTL_REGISTRY_CONTROL, L"ServiceCurrent", L"", REG_DWORD, &ServiceCurrent, sizeof(ServiceCurrent));
-
-    if (!NT_SUCCESS(Status))
-    {
-        DPRINT1("RtlWriteRegistryValue() failed (Status %lx)\n", Status);
-        return RtlNtStatusToDosError(Status);
+    Status = RegSetValueExW(hServiceCurrentKey, L"", 0, REG_DWORD, (BYTE*)&ServiceCurrent, sizeof(ServiceCurrent));
+
+    RegCloseKey(hServiceCurrentKey);
+
+    if (ERROR_SUCCESS != Status)
+    {
+        DPRINT1("RegSetValueExW() failed (Status %lx)\n", Status);
+        return Status;
     }
 
     /* Create '\\.\pipe\net\NtControlPipeXXX' instance */