[PATCH v5 4/4] win32u: Implement NtGdiDdDDIEnumAdapters2.

[Krzysztof Bogacki]

From: Krzysztof Bogacki

Signed-off-by: Krzysztof Bogacki 
---
 dlls/gdi32/tests/driver.c |  2 +-
 dlls/win32u/sysparams.c   | 86 ++++++++++++++++++++++++++++++++++++++-
 2 files changed, 85 insertions(+), 3 deletions(-)

diff --git a/dlls/gdi32/tests/driver.c b/dlls/gdi32/tests/driver.c
index 03abdb541b5..bf32e109361 100644
--- a/dlls/gdi32/tests/driver.c
+++ b/dlls/gdi32/tests/driver.c
@@ -199,7 +199,7 @@ static void test_D3DKMTEnumAdapters2(void)

if (!pD3DKMTEnumAdapters2 || pD3DKMTEnumAdapters2(&enum_adapters_2_desc) == STATUS_PROCEDURE_NOT_FOUND)
     {
-        skip("D3DKMTEnumAdapters2() is unavailable.\n");
+        win_skip("D3DKMTEnumAdapters2() is unavailable.\n");
         return;
     }

diff --git a/dlls/win32u/sysparams.c b/dlls/win32u/sysparams.c
index b3f8092a60b..cf38f63a8ef 100644
--- a/dlls/win32u/sysparams.c
+++ b/dlls/win32u/sysparams.c
@@ -6657,6 +6657,88 @@ NTSTATUS WINAPI NtUserDisplayConfigGetDeviceInfo( DISPLAYCONFIG_DEVICE_INFO_HEAD
  */
 NTSTATUS WINAPI NtGdiDdDDIEnumAdapters2( D3DKMT_ENUMADAPTERS2 *desc )
 {
-    FIXME(" (%p): stub\n", desc );
-    return STATUS_PROCEDURE_NOT_FOUND;
+    D3DKMT_OPENADAPTERFROMLUID open_adapter_from_luid;
+    static const ULONG max_adapters = 34;
+    D3DKMT_CLOSEADAPTER close_adapter;
+    NTSTATUS status = STATUS_SUCCESS;
+    struct adapter *adapter;
+    ULONG idx = 0, count;
+    struct gpu *gpu;
+
+    TRACE( "(%p)\n", desc );
+
+    if (!desc) return STATUS_INVALID_PARAMETER;
+
+    if (!desc->pAdapters)
+    {
+        desc->NumAdapters = max_adapters;
+        return STATUS_SUCCESS;
+    }
+
+    if (!lock_display_devices()) return STATUS_UNSUCCESSFUL;
+
+    if ((count = list_count( &gpus )) > max_adapters)
+    {
+        WARN( "Too many adapters (%u), only up to %u can be enumerated.\n",
+              (unsigned int)count, (unsigned int)max_adapters );
+        count = max_adapters;
+    }
+
+    if (count > desc->NumAdapters)
+    {
+        status = STATUS_BUFFER_TOO_SMALL;
+        goto done;
+    }
+
+    LIST_FOR_EACH_ENTRY( gpu, &gpus, struct gpu, entry )
+    {
+        if (gpu->luid.LowPart || gpu->luid.HighPart)
+        {
+            open_adapter_from_luid.AdapterLuid = gpu->luid;
+
+            /* give the GDI driver a chance to be notified about new adapter handle */
+            if (NT_SUCCESS( NtGdiDdDDIOpenAdapterFromLuid( &open_adapter_from_luid ) ))
+            {
+                desc->pAdapters[idx].AdapterLuid = gpu->luid;
+                desc->pAdapters[idx].hAdapter = open_adapter_from_luid.hAdapter;
+                desc->pAdapters[idx].bPrecisePresentRegionsPreferred = FALSE;
+                desc->pAdapters[idx].NumOfSources = 0;
+
+                LIST_FOR_EACH_ENTRY( adapter, &adapters, struct adapter, entry )
+                {
+                    if (!memcmp( &gpu->luid, &adapter->gpu_luid, sizeof(LUID) ) )
+                        desc->pAdapters[idx].NumOfSources++;
+                }
+
+                idx += 1;
+                continue;
+            }
+            else
+            {
+                ERR( "Failed to open adapter %u from LUID.\n", (unsigned int)idx );
+            }
+        }
+        else
+        {
+            ERR( "Adapter %u does not have a LUID.\n", (unsigned int)idx );
+        }
+
+        while (idx--)
+        {
+            close_adapter.hAdapter = desc->pAdapters[idx].hAdapter;
+            NtGdiDdDDICloseAdapter( &close_adapter );
+        }
+
+        memset( desc->pAdapters, 0, desc->NumAdapters * sizeof(D3DKMT_ADAPTERINFO) );
+        status = STATUS_UNSUCCESSFUL;
+        break;
+    }
+
+done:
+    if (status == STATUS_SUCCESS)
+        desc->NumAdapters = idx;
+
+    unlock_display_devices();
+
+    return status;
 }