From d4acd8cc91ae6ed5ea21906af886f00ef32be328 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?George=20Bi=C8=99oc?= <fraizeraust99@gmail.com>
Date: Sat, 6 Jun 2020 17:48:39 +0200
Subject: [PATCH] [NTDLL_APITEST] Add NtSetInformationThread testcase (#2793)

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

diff --git a/modules/rostests/apitests/ntdll/CMakeLists.txt b/modules/rostests/apitests/ntdll/CMakeLists.txt
index 6e25db84d12..a8a314ae62f 100644
--- a/modules/rostests/apitests/ntdll/CMakeLists.txt
+++ b/modules/rostests/apitests/ntdll/CMakeLists.txt
@@ -35,6 +35,7 @@ list(APPEND SOURCE
     NtSaveKey.c
     NtSetInformationFile.c
     NtSetInformationProcess.c
+    NtSetInformationThread.c
     NtSetValueKey.c
     NtSetVolumeInformationFile.c
     NtUnloadDriver.c
diff --git a/modules/rostests/apitests/ntdll/NtSetInformationThread.c b/modules/rostests/apitests/ntdll/NtSetInformationThread.c
new file mode 100644
index 00000000000..f1835ed5198
--- /dev/null
+++ b/modules/rostests/apitests/ntdll/NtSetInformationThread.c
@@ -0,0 +1,100 @@
+/*
+ * PROJECT:         ReactOS API tests
+ * LICENSE:         GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
+ * PURPOSE:         Tests for the NtSetInformationThread API
+ * COPYRIGHT:       Copyright 2020 George Bișoc <george.bisoc@reactos.org>
+ */
+
+#include "precomp.h"
+
+static
+void
+Test_ThreadPriorityClass(void)
+{
+    NTSTATUS Status;
+    KPRIORITY *Priority;
+
+    Priority = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(KPRIORITY));
+    if (Priority == NULL)
+    {
+        skip("Failed to allocate memory for thread priority variable!\n");
+        return;
+    }
+
+    /* Assign a priority */
+    *Priority = 11;
+
+    /* Everything is NULL */
+    Status = NtSetInformationThread(NULL,
+                                    ThreadPriority,
+                                    NULL,
+                                    0);
+    ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
+
+    /* Give an invalid thread handle */
+    Status = NtSetInformationThread(NULL,
+                                    ThreadPriority,
+                                    Priority,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_INVALID_HANDLE);
+
+    /* Don't set any priority to the thread */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    NULL,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_ACCESS_VIOLATION);
+
+    /* The information length is incorrect */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    Priority,
+                                    0);
+    ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
+
+    /* The buffer is misaligned and information length is incorrect */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    (PVOID)1,
+                                    0);
+    ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
+
+    /* The buffer is misaligned */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    (PVOID)1,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
+
+    /* The buffer is misaligned -- try with an alignment size of 2 */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    (PVOID)2,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
+
+    /* Set the priority to the current thread */
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    Priority,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_SUCCESS);
+
+    /*
+     * Set the priority as LOW_REALTIME_PRIORITY,
+     * we don't have privileges to do so.
+    */
+    *Priority = LOW_REALTIME_PRIORITY;
+    Status = NtSetInformationThread(GetCurrentThread(),
+                                    ThreadPriority,
+                                    Priority,
+                                    sizeof(KPRIORITY));
+    ok_hex(Status, STATUS_PRIVILEGE_NOT_HELD);
+
+    HeapFree(GetProcessHeap(), 0, Priority);
+}
+
+START_TEST(NtSetInformationThread)
+{
+    Test_ThreadPriorityClass();
+}
diff --git a/modules/rostests/apitests/ntdll/testlist.c b/modules/rostests/apitests/ntdll/testlist.c
index 293a7cc45c5..40a0d686c8b 100644
--- a/modules/rostests/apitests/ntdll/testlist.c
+++ b/modules/rostests/apitests/ntdll/testlist.c
@@ -33,6 +33,7 @@ extern void func_NtReadFile(void);
 extern void func_NtSaveKey(void);
 extern void func_NtSetInformationFile(void);
 extern void func_NtSetInformationProcess(void);
+extern void func_NtSetInformationThread(void);
 extern void func_NtSetValueKey(void);
 extern void func_NtSetVolumeInformationFile(void);
 extern void func_NtSystemInformation(void);
@@ -106,6 +107,7 @@ const struct test winetest_testlist[] =
     { "NtSaveKey",                      func_NtSaveKey},
     { "NtSetInformationFile",           func_NtSetInformationFile },
     { "NtSetInformationProcess",        func_NtSetInformationProcess },
+    { "NtSetInformationThread",         func_NtSetInformationThread },
     { "NtSetValueKey",                  func_NtSetValueKey},
     { "NtSetVolumeInformationFile",     func_NtSetVolumeInformationFile },
     { "NtSystemInformation",            func_NtSystemInformation },