From 32db51fcffb8096fe1ce2c8cd509f2db1e1d53fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?George=20Bi=C8=99oc?= <george.bisoc@reactos.org>
Date: Sat, 18 Sep 2021 17:18:04 +0200
Subject: [PATCH] [NTDLL_APITEST] Implement NtAdjustGroupsToken API tests

---
 .../rostests/apitests/ntdll/CMakeLists.txt    |  1 +
 .../apitests/ntdll/NtAdjustGroupsToken.c      | 77 +++++++++++++++++++
 modules/rostests/apitests/ntdll/testlist.c    |  2 +
 3 files changed, 80 insertions(+)
 create mode 100644 modules/rostests/apitests/ntdll/NtAdjustGroupsToken.c

diff --git a/modules/rostests/apitests/ntdll/CMakeLists.txt b/modules/rostests/apitests/ntdll/CMakeLists.txt
index c90cc46c567..403641119fb 100644
--- a/modules/rostests/apitests/ntdll/CMakeLists.txt
+++ b/modules/rostests/apitests/ntdll/CMakeLists.txt
@@ -10,6 +10,7 @@ list(APPEND SOURCE
     load_notifications.c
     locale.c
     NtAcceptConnectPort.c
+    NtAdjustGroupsToken.c
     NtAdjustPrivilegesToken.c
     NtAllocateVirtualMemory.c
     NtApphelpCacheControl.c
diff --git a/modules/rostests/apitests/ntdll/NtAdjustGroupsToken.c b/modules/rostests/apitests/ntdll/NtAdjustGroupsToken.c
new file mode 100644
index 00000000000..0529243270b
--- /dev/null
+++ b/modules/rostests/apitests/ntdll/NtAdjustGroupsToken.c
@@ -0,0 +1,77 @@
+/*
+ * PROJECT:         ReactOS API tests
+ * LICENSE:         GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
+ * PURPOSE:         Tests for the NtAdjustGroupsToken API
+ * COPYRIGHT:       Copyright 2021 George Bișoc <george.bisoc@reactos.org>
+ */
+
+#include "precomp.h"
+
+static
+HANDLE
+GetProcessToken(
+    _In_ DWORD Access)
+{
+    BOOL Success;
+    HANDLE Token;
+
+    Success = OpenProcessToken(GetCurrentProcess(), Access, &Token);
+    if (!Success)
+    {
+        skip("Failed to open the process' token (error code: %lu)!\n", GetLastError());
+        return NULL;
+    }
+
+    return Token;
+}
+
+START_TEST(NtAdjustGroupsToken)
+{
+    HANDLE TokenHandle;
+    NTSTATUS Status;
+
+    /* Get the token from current process but with incorrect rights */
+    TokenHandle = GetProcessToken(TOKEN_DUPLICATE);
+
+    /* We give an invalid handle */
+    Status = NtAdjustGroupsToken(NULL,
+                                 TRUE,
+                                 NULL,
+                                 0,
+                                 NULL,
+                                 NULL);
+    ok_hex(Status, STATUS_INVALID_HANDLE);
+
+    /* We're trying to adjust the token's groups with wrong rights */
+    Status = NtAdjustGroupsToken(TokenHandle,
+                                 TRUE,
+                                 NULL,
+                                 0,
+                                 NULL,
+                                 NULL);
+    ok_hex(Status, STATUS_ACCESS_DENIED);
+
+    /* Close our handle and open a new one with right access right */
+    CloseHandle(TokenHandle);
+    TokenHandle = GetProcessToken(TOKEN_ADJUST_GROUPS);
+
+    /* We don't give a list of groups to be adjusted in token */
+    Status = NtAdjustGroupsToken(TokenHandle,
+                                 FALSE,
+                                 NULL,
+                                 0,
+                                 NULL,
+                                 NULL);
+    ok_hex(Status, STATUS_INVALID_PARAMETER);
+
+    /* Reset the groups of an access token to default */
+    Status = NtAdjustGroupsToken(TokenHandle,
+                                 TRUE,
+                                 NULL,
+                                 0,
+                                 NULL,
+                                 NULL);
+    ok_hex(Status, STATUS_SUCCESS);
+
+    CloseHandle(TokenHandle);
+}
diff --git a/modules/rostests/apitests/ntdll/testlist.c b/modules/rostests/apitests/ntdll/testlist.c
index 1a921a2f2b1..9642602fd77 100644
--- a/modules/rostests/apitests/ntdll/testlist.c
+++ b/modules/rostests/apitests/ntdll/testlist.c
@@ -6,6 +6,7 @@
 extern void func_LdrEnumResources(void);
 extern void func_load_notifications(void);
 extern void func_NtAcceptConnectPort(void);
+extern void func_NtAdjustGroupsToken(void);
 extern void func_NtAdjustPrivilegesToken(void);
 extern void func_NtAllocateVirtualMemory(void);
 extern void func_NtApphelpCacheControl(void);
@@ -92,6 +93,7 @@ const struct test winetest_testlist[] =
     { "LdrEnumResources",               func_LdrEnumResources },
     { "load_notifications",             func_load_notifications },
     { "NtAcceptConnectPort",            func_NtAcceptConnectPort },
+    { "NtAdjustGroupsToken",            func_NtAdjustGroupsToken },
     { "NtAdjustPrivilegesToken",        func_NtAdjustPrivilegesToken },
     { "NtAllocateVirtualMemory",        func_NtAllocateVirtualMemory },
     { "NtApphelpCacheControl",          func_NtApphelpCacheControl },