Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-07 02:42:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Made ServiceCurrent key volatile

Browse source 
svn path=/trunk/; revision=30190
This commit is contained in:
Gregor Brunmar 2007-11-05 17:41:20 +00:00
parent 6f0df2f887
commit eaa8991ba5
1 changed files with 31 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

55
reactos/base/system/services/database.c
View file

@ -698,6 +698,8 @@ ScmStartUserModeService(PSERVICE Service,
NTSTATUS Status; NTSTATUS Status;
DWORD dwError = ERROR_SUCCESS; DWORD dwError = ERROR_SUCCESS;
WCHAR NtControlPipeName[MAX_PATH + 1]; WCHAR NtControlPipeName[MAX_PATH + 1];
HKEY hServiceCurrentKey = INVALID_HANDLE_VALUE;
DWORD KeyDisposition;
RtlInitUnicodeString(&ImagePath, NULL); RtlInitUnicodeString(&ImagePath, NULL);
@ -727,39 +729,44 @@ ScmStartUserModeService(PSERVICE Service,
DPRINT("Type: %lx\n", Type); DPRINT("Type: %lx\n", Type);
/* Get the service number */ /* 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 */ /* TODO: Create registry entry with correct write access */
} Status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
else if (!NT_SUCCESS(Status)) L"SYSTEM\\CurrentControlSet\\Control\\ServiceCurrent", 0, NULL,
REG_OPTION_VOLATILE,
KEY_WRITE | KEY_READ,
NULL,
&hServiceCurrentKey,
&KeyDisposition);
if (ERROR_SUCCESS != Status)
{ {
DPRINT1("RtlQueryRegistryValues() failed (Status %lx)\n", Status); DPRINT1("RegCreateKeyEx() failed with status %u\n", Status);
return RtlNtStatusToDosError(Status); return Status;
} }
else
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++; ServiceCurrent++;
} }
Status = RtlWriteRegistryValue(RTL_REGISTRY_CONTROL, L"ServiceCurrent", L"", REG_DWORD, &ServiceCurrent, sizeof(ServiceCurrent)); Status = RegSetValueExW(hServiceCurrentKey, L"", 0, REG_DWORD, (BYTE*)&ServiceCurrent, sizeof(ServiceCurrent));
if (!NT_SUCCESS(Status)) RegCloseKey(hServiceCurrentKey);
if (ERROR_SUCCESS != Status)
{ {
DPRINT1("RtlWriteRegistryValue() failed (Status %lx)\n", Status); DPRINT1("RegSetValueExW() failed (Status %lx)\n", Status);
return RtlNtStatusToDosError(Status); return Status;
} }
/* Create '\\.\pipe\net\NtControlPipeXXX' instance */ /* Create '\\.\pipe\net\NtControlPipeXXX' instance */