Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 18:15:11 +00:00
Code Issues Projects Releases Packages Wiki Activity

[UMPNPMGR] PNP_GetDeviceListSize: Implement the buffer size calculation for the enumeration of services instances.

Browse source
This commit is contained in:
Eric Kohl 2018-12-31 16:14:04 +01:00
parent a179012563
commit 5da008c338
1 changed files with 115 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

116
base/services/umpnpmgr/umpnpmgr.c
View file

@ -580,6 +580,17 @@ GetRelationsInstanceList(
}
static
CONFIGRET
GetServiceInstanceList(
_In_ PWSTR pszService,
_Inout_ PWSTR pszBuffer,
_Inout_ PDWORD pulLength)
{
return CR_CALL_NOT_IMPLEMENTED;
}
static
CONFIGRET
GetDeviceInstanceList(
@ -810,6 +821,9 @@ PNP_GetDeviceList(
}
else if (ulFlags & CM_GETIDLIST_FILTER_SERVICE)
{
ret = GetServiceInstanceList(pszFilter,
Buffer,
pulLength);
ret = CR_CALL_NOT_IMPLEMENTED;
}
else if (ulFlags & CM_GETIDLIST_FILTER_ENUMERATOR)
@ -892,6 +906,105 @@ GetRelationsInstanceListSize(
}
static
CONFIGRET
GetServiceInstanceListSize(
_In_ PWSTR pszService,
_Out_ PDWORD pulLength)
{
HKEY hServicesKey = NULL, hServiceKey = NULL, hEnumKey = NULL;
DWORD dwValues, dwMaxValueLength, dwSize;
DWORD dwError;
CONFIGRET ret = CR_SUCCESS;
/* Open the device key */
dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
L"System\\CurrentControlSet\\Services",
0,
KEY_READ,
&hServicesKey);
if (dwError != ERROR_SUCCESS)
{
DPRINT("Failed to open the services key (Error %lu)\n", dwError);
return CR_REGISTRY_ERROR;
}
dwError = RegOpenKeyExW(hServicesKey,
pszService,
0,
KEY_READ,
&hServiceKey);
if (dwError != ERROR_SUCCESS)
{
DPRINT("Failed to open the service key (Error %lu)\n", dwError);
ret = CR_REGISTRY_ERROR;
goto Done;
}
dwError = RegOpenKeyExW(hServiceKey,
L"Enum",
0,
KEY_READ,
&hEnumKey);
if (dwError != ERROR_SUCCESS)
{
DPRINT("Failed to open the service enum key (Error %lu)\n", dwError);
ret = CR_REGISTRY_ERROR;
goto Done;
}
/* Retrieve the number of device instances */
dwSize = sizeof(DWORD);
dwError = RegQueryValueExW(hEnumKey,
L"Count",
NULL,
NULL,
(LPBYTE)&dwValues,
&dwSize);
if (dwError != ERROR_SUCCESS)
{
DPRINT("RegQueryValueExW failed (Error %lu)\n", dwError);
dwValues = 1;
}
/* Retrieve the maximum instance name length */
dwError = RegQueryInfoKeyW(hEnumKey,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
&dwMaxValueLength,
NULL,
NULL);
if (dwError != ERROR_SUCCESS)
{
DPRINT("RegQueryInfoKeyW failed (Error %lu)\n", dwError);
dwMaxValueLength = MAX_DEVICE_ID_LEN;
}
DPRINT("dwValues %lu dwMaxValueLength %lu\n", dwValues, dwMaxValueLength / sizeof(WCHAR));
/* Return the largest possible buffer size */
*pulLength = dwValues * dwMaxValueLength / sizeof(WCHAR) + 2;
Done:
if (hEnumKey != NULL)
RegCloseKey(hEnumKey);
if (hServiceKey != NULL)
RegCloseKey(hServiceKey);
if (hServicesKey != NULL)
RegCloseKey(hServicesKey);
return ret;
}
static
CONFIGRET
GetDeviceInstanceListSize(
@ -1080,7 +1193,8 @@ PNP_GetDeviceListSize(
}
else if (ulFlags & CM_GETIDLIST_FILTER_SERVICE)
{
ret = CR_CALL_NOT_IMPLEMENTED;
ret = GetServiceInstanceListSize(pszFilter,
pulLength);
}
else if (ulFlags & CM_GETIDLIST_FILTER_ENUMERATOR)
{