ntoskrnl.exe: Implement KeRevertToUserAffinityThread() function.

Signed-off-by: Paul Gofman <pgofman@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

ntoskrnl.exe: Implement KeRevertToUserAffinityThread() function.
cda4fa29 · Paul Gofman · Alexandre Julliard · 7d4f4783 · cda4fa29 · cda4fa29
Commit cda4fa29 authored May 26, 2020 by Paul Gofman Committed by Alexandre Julliard May 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 2 deletions

ntoskrnl.c dlls/ntoskrnl.exe/ntoskrnl.c +1 -1

driver.c dlls/ntoskrnl.exe/tests/driver.c +11 -1

No files found.
--- a/dlls/ntoskrnl.exe/ntoskrnl.c
+++ b/dlls/ntoskrnl.exe/ntoskrnl.c
@@ -2485,7 +2485,7 @@ KAFFINITY WINAPI KeSetSystemAffinityThreadEx(KAFFINITY affinity)
 */
 void WINAPI KeRevertToUserAffinityThread(void)
 {
-    FIXME("() stub\n");
+    KeRevertToUserAffinityThreadEx(0);
 }

 void WINAPI KeRevertToUserAffinityThreadEx(KAFFINITY affinity)

--- a/dlls/ntoskrnl.exe/tests/driver.c
+++ b/dlls/ntoskrnl.exe/tests/driver.c
@@ -1766,7 +1766,17 @@ static void test_affinity(void)
    mask = pKeSetSystemAffinityThreadEx(0x1);
    ok(mask == mask_all_cpus, "Got unexpected mask %#lx.\n", mask);

-    pKeRevertToUserAffinityThreadEx(mask_all_cpus);
+    pKeRevertToUserAffinityThreadEx(0);
+
+    mask = pKeSetSystemAffinityThreadEx(0x1);
+    ok(!mask, "Got unexpected mask %#lx.\n", mask);
+
+    KeRevertToUserAffinityThread();
+
+    mask = pKeSetSystemAffinityThreadEx(0x1);
+    ok(!mask, "Got unexpected mask %#lx.\n", mask);
+
+    KeRevertToUserAffinityThread();
 }

 static NTSTATUS main_test(DEVICE_OBJECT *device, IRP *irp, IO_STACK_LOCATION *stack)