Alexandre Julliard : ntoskrnl/tests: Fix affinity test when there are too many CPUs.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 6ca6bb542e1335b416533074e166ca6b95d16746
URL:    https://gitlab.winehq.org/wine/wine/-/commit/6ca6bb542e1335b416533074e166ca6...

Author: Alexandre Julliard 
Date:   Mon Sep 12 11:54:41 2022 +0200

ntoskrnl/tests: Fix affinity test when there are too many CPUs.

---

dlls/ntoskrnl.exe/tests/driver.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/dlls/ntoskrnl.exe/tests/driver.c b/dlls/ntoskrnl.exe/tests/driver.c
index dc4f4ad993d..d293889c823 100644
--- a/dlls/ntoskrnl.exe/tests/driver.c
+++ b/dlls/ntoskrnl.exe/tests/driver.c
@@ -2031,7 +2031,10 @@ static void test_affinity(void)
     count = pKeQueryActiveProcessorCountEx(ALL_PROCESSOR_GROUPS);
     ok(count == cpu_count, "Got unexpected count %lu.\n", count);

-    mask_all_cpus = ~((~0u) << cpu_count);
+    if (cpu_count >= 8 * sizeof(KAFFINITY))
+        mask_all_cpus = ~(KAFFINITY)0;
+    else
+        mask_all_cpus = ((KAFFINITY)1 << cpu_count) - 1;

mask = pKeQueryActiveProcessors();
     ok(mask == mask_all_cpus, "Got unexpected mask %#Ix.\n", mask);