[NTDLL_APITEST] Add alignment probing tests for Query/Set information process related routines

modules/rostests/apitests/ntdll/NtQueryInformationProcess.c

···
  * LICENSE:         LGPLv2.1+ - See COPYING.LIB in the top level directory
  * PURPOSE:         Tests for the NtQueryInformationProcess API
  * PROGRAMMER:      Thomas Faber <[e-mail]>
+ *                  George Bișoc <[e-mail]>
  */
 
 #include "precomp.h"
+#include <internal/ps_i.h>
 
 static LARGE_INTEGER TestStartTime;
 
···
     trace("VdmPower = %lu\n", VdmPower);
 }
 
+static
+void
+Test_ProcQueryAlignmentProbe(void)
+{
+    ULONG InfoClass;
+
+    /* Iterate over the process info classes and begin the tests */
+    for (InfoClass = 0; InfoClass < _countof(PsProcessInfoClass); InfoClass++)
+    {
+        /* The buffer is misaligned */
+        QuerySetProcessValidator(QUERY,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)1,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeQUERY,
+                                 STATUS_DATATYPE_MISALIGNMENT);
+
+        /* We query an invalid buffer address */
+        QuerySetProcessValidator(QUERY,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)PsProcessInfoClass[InfoClass].AlignmentQUERY,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeQUERY,
+                                 STATUS_ACCESS_VIOLATION);
+
+        /* The information length is wrong */
+        QuerySetProcessValidator(QUERY,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)PsProcessInfoClass[InfoClass].AlignmentQUERY,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeQUERY - 1,
+                                 STATUS_INFO_LENGTH_MISMATCH);
+    }
+}
+
 START_TEST(NtQueryInformationProcess)
 {
     NTSTATUS Status;
···
     Test_ProcessTimes();
     Test_ProcessPriorityClassAlignment();
     Test_ProcessWx86Information();
+    Test_ProcQueryAlignmentProbe();
 }

modules/rostests/apitests/ntdll/NtSetInformationProcess.c

···
  */
 
 #include "precomp.h"
+#include <internal/ps_i.h>
 
 static
 void
···
     ok_hex(Status, STATUS_PRIVILEGE_NOT_HELD);
 }
 
+static
+void
+Test_ProcSetAlignmentProbe(void)
+{
+    ULONG InfoClass;
+
+    /* Iterate over the process info classes and begin the tests */
+    for (InfoClass = 0; InfoClass < _countof(PsProcessInfoClass); InfoClass++)
+    {
+        /* The buffer is misaligned */
+        QuerySetProcessValidator(SET,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)1,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeSET,
+                                 STATUS_DATATYPE_MISALIGNMENT);
+
+        /* We set an invalid buffer address */
+        QuerySetProcessValidator(SET,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)PsProcessInfoClass[InfoClass].AlignmentSET,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeSET,
+                                 STATUS_ACCESS_VIOLATION);
+
+        /* The information length is wrong */
+        QuerySetProcessValidator(SET,
+                                 InfoClass,
+                                 (PVOID)(ULONG_PTR)PsProcessInfoClass[InfoClass].AlignmentSET,
+                                 PsProcessInfoClass[InfoClass].RequiredSizeSET - 1,
+                                 STATUS_INFO_LENGTH_MISMATCH);
+    }
+}
+
 START_TEST(NtSetInformationProcess)
 {
     Test_ProcForegroundBackgroundClass();
     Test_ProcBasePriorityClass();
     Test_ProcRaisePriorityClass();
     Test_ProcessWx86InformationClass();
+    Test_ProcSetAlignmentProbe();
 }