[reactos] 01/01: [UMPNPMGR] PNP_GetDeviceListSize: Implement the buffer size calculation for a list of all device instances.

[Eric Kohl]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=969de9062c6dc95b5e3c7a...

commit 969de9062c6dc95b5e3c7a66a6c48dc03e03d534
Author:     Eric Kohl [e-mail]
AuthorDate: Sun Dec 30 14:42:12 2018 +0100
Commit:     Eric Kohl [e-mail]
CommitDate: Sun Dec 30 14:43:10 2018 +0100

[UMPNPMGR] PNP_GetDeviceListSize: Implement the buffer size calculation for a list of all device instances.
---
 base/services/umpnpmgr/umpnpmgr.c | 41 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/base/services/umpnpmgr/umpnpmgr.c b/base/services/umpnpmgr/umpnpmgr.c
index 56faf3be3b..2583d135ef 100644
--- a/base/services/umpnpmgr/umpnpmgr.c
+++ b/base/services/umpnpmgr/umpnpmgr.c
@@ -886,6 +886,41 @@ GetEnumeratorInstanceListSize(
 }

+static
+CONFIGRET
+GetAllInstanceListSize(
+    _In_ LPCWSTR pszEnumerator,
+    _Out_ PULONG pulLength)
+{
+    WCHAR szEnumeratorBuffer[MAX_DEVICE_ID_LEN];
+    DWORD dwIndex, dwEnumeratorLength, dwBufferLength;
+    DWORD dwError;
+    CONFIGRET ret = CR_SUCCESS;
+
+    for (dwIndex = 0; ; dwIndex++)
+    {
+        dwEnumeratorLength = MAX_DEVICE_ID_LEN;
+        dwError = RegEnumKeyExW(hEnumKey,
+                                dwIndex,
+                                szEnumeratorBuffer,
+                                &dwEnumeratorLength,
+                                NULL, NULL, NULL, NULL);
+        if (dwError != ERROR_SUCCESS)
+            break;
+
+        /* Get the size of all device instances for the enumerator */
+        ret = GetEnumeratorInstanceListSize(szEnumeratorBuffer,
+                                            &dwBufferLength);
+        if (ret != CR_SUCCESS)
+            break;
+
+        *pulLength += dwBufferLength;
+    }
+
+    return ret;
+}
+
+
 /* Function 11 */
 DWORD
 WINAPI
@@ -902,7 +937,8 @@ PNP_GetDeviceListSize(
     CONFIGRET ret = CR_SUCCESS;
     NTSTATUS Status;

-    DPRINT("PNP_GetDeviceListSize() called\n");
+    DPRINT("PNP_GetDeviceListSize(%p %S %p 0x%lx)\n",
+           hBinding, pszFilter, pulLength, ulFlags);

if (ulFlags & ~CM_GETIDLIST_FILTER_BITS)
         return CR_INVALID_FLAG;
@@ -980,7 +1016,8 @@ PNP_GetDeviceListSize(
     }
     else /* CM_GETIDLIST_FILTER_NONE */
     {
-        ret = CR_CALL_NOT_IMPLEMENTED;
+        ret = GetAllInstanceListSize(pszFilter,
+                                     pulLength);
     }

/* Add one character for the terminating double UNICODE_NULL */