[PATCH v5 01/11] kernelbase: implement `GetNumaHighestNodeNumber` to report accurate node count

[wasertech]

From: wasertech

---
 dlls/kernelbase/memory.c | 40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/dlls/kernelbase/memory.c b/dlls/kernelbase/memory.c
index c0a3d65d341..dfa4c434d2b 100644
--- a/dlls/kernelbase/memory.c
+++ b/dlls/kernelbase/memory.c
@@ -1582,8 +1582,46 @@ BOOL WINAPI SetProcessDefaultCpuSets(HANDLE process, const ULONG *cpu_set_ids, U
  */
 BOOL WINAPI DECLSPEC_HOTPATCH GetNumaHighestNodeNumber( ULONG *node )
 {
-    FIXME( "semi-stub: %p\n", node );
+    NTSTATUS nts;
+    DWORD rel = RelationNumaNode;
+    DWORD iex_len = 0;
+    SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX *iex = NULL;
+    unsigned int i;
+
+    if (!node) { SetLastError( ERROR_INVALID_PARAMETER ); return FALSE; }
     *node = 0;
+
+    set_ntstatus( NtQuerySystemInformationEx( SystemLogicalProcessorInformationEx, 
+        &rel, sizeof(rel), NULL, 0, &iex_len ) );
+    if (!iex_len)
+    { FIXME( "stub: iex_len %lu, node %lu\n", iex_len, *node ); return TRUE; }
+
+    iex = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY, iex_len );
+    if (!iex) { FIXME( "stub: iex %p, node %lu\n", iex, *node ); return TRUE; }
+    
+    nts = set_ntstatus( NtQuerySystemInformationEx( SystemLogicalProcessorInformationEx, 
+            &rel, sizeof(rel), iex, iex_len, &iex_len ) );
+    if (!nts)
+    {
+        FIXME( "stub: nts %lu, node %lu\n", nts, *node );
+        RtlFreeHeap( GetProcessHeap(), HEAP_ZERO_MEMORY, iex );
+        return TRUE;
+    }
+
+    for ( i = 0; i < iex_len; )
+    {
+        SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX *ex = ( SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX * )
+            ( (char *)iex + i );
+        if (!ex->Size) break;
+        if (ex->Relationship == RelationNumaNode)
+        {
+            if (ex->NumaNode.NodeNumber > *node)
+                *node = ex->NumaNode.NodeNumber;
+        }
+        i += ex->Size;
+    }
+
+    RtlFreeHeap( GetProcessHeap(), HEAP_ZERO_MEMORY, iex );
     return TRUE;
 }