ReactOS 0.4.16-dev-117-g38f21f9
Functions
NtSetInformationThread.c File Reference
#include "precomp.h"
#include <internal/ps_i.h>
Include dependency graph for NtSetInformationThread.c:

Go to the source code of this file.

Functions

static void Test_ThreadPriorityClass (void)
 
static void Test_ThreadSetAlignmentProbe (void)
 
 START_TEST (NtSetInformationThread)
 

Function Documentation

◆ START_TEST()

START_TEST ( NtSetInformationThread  )

Definition at line 130 of file NtSetInformationThread.c.

131{
132 Test_ThreadPriorityClass();
133 Test_ThreadSetAlignmentProbe();
134}
Test_ThreadSetAlignmentProbe
static void Test_ThreadSetAlignmentProbe(void)
Definition: NtSetInformationThread.c:100
Test_ThreadPriorityClass
static void Test_ThreadPriorityClass(void)
Definition: NtSetInformationThread.c:13

◆ Test_ThreadPriorityClass()

static void Test_ThreadPriorityClass ( void  )
static

Definition at line 13 of file NtSetInformationThread.c.

14{
15 NTSTATUS Status;
16 KPRIORITY *Priority;
17
18 Priority = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(KPRIORITY));
19 if (Priority == NULL)
20 {
21 skip("Failed to allocate memory for thread priority variable!\n");
22 return;
23 }
24
25 /* Assign a priority */
26 *Priority = 11;
27
28 /* Everything is NULL */
29 Status = NtSetInformationThread(NULL,
30 ThreadPriority,
31 NULL,
32 0);
33 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
34
35 /* Give an invalid thread handle */
36 Status = NtSetInformationThread(NULL,
37 ThreadPriority,
38 Priority,
39 sizeof(KPRIORITY));
40 ok_hex(Status, STATUS_INVALID_HANDLE);
41
42 /* Don't set any priority to the thread */
43 Status = NtSetInformationThread(GetCurrentThread(),
44 ThreadPriority,
45 NULL,
46 sizeof(KPRIORITY));
47 ok_hex(Status, STATUS_ACCESS_VIOLATION);
48
49 /* The information length is incorrect */
50 Status = NtSetInformationThread(GetCurrentThread(),
51 ThreadPriority,
52 Priority,
53 0);
54 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
55
56 /* The buffer is misaligned and information length is incorrect */
57 Status = NtSetInformationThread(GetCurrentThread(),
58 ThreadPriority,
59 (PVOID)1,
60 0);
61 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
62
63 /* The buffer is misaligned */
64 Status = NtSetInformationThread(GetCurrentThread(),
65 ThreadPriority,
66 (PVOID)1,
67 sizeof(KPRIORITY));
68 ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
69
70 /* The buffer is misaligned -- try with an alignment size of 2 */
71 Status = NtSetInformationThread(GetCurrentThread(),
72 ThreadPriority,
73 (PVOID)2,
74 sizeof(KPRIORITY));
75 ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
76
77 /* Set the priority to the current thread */
78 Status = NtSetInformationThread(GetCurrentThread(),
79 ThreadPriority,
80 Priority,
81 sizeof(KPRIORITY));
82 ok_hex(Status, STATUS_SUCCESS);
83
84 /*
85 * Set the priority as LOW_REALTIME_PRIORITY,
86 * we don't have privileges to do so.
87 */
88 *Priority = LOW_REALTIME_PRIORITY;
89 Status = NtSetInformationThread(GetCurrentThread(),
90 ThreadPriority,
91 Priority,
92 sizeof(KPRIORITY));
93 ok_hex(Status, STATUS_PRIVILEGE_NOT_HELD);
94
95 HeapFree(GetProcessHeap(), 0, Priority);
96}
STATUS_PRIVILEGE_NOT_HELD
#define STATUS_PRIVILEGE_NOT_HELD
Definition: DriverTester.h:9
ok_hex
#define ok_hex(expression, result)
Definition: atltest.h:94
skip
#define skip(...)
Definition: atltest.h:64
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
NULL
#define NULL
Definition: types.h:112
GetProcessHeap
#define GetProcessHeap()
Definition: compat.h:736
HeapAlloc
#define HeapAlloc
Definition: compat.h:733
ThreadPriority
@ ThreadPriority
Definition: compat.h:937
HeapFree
#define HeapFree(x, y, z)
Definition: compat.h:735
KPRIORITY
LONG KPRIORITY
Definition: compat.h:803
HEAP_ZERO_MEMORY
#define HEAP_ZERO_MEMORY
Definition: compat.h:134
Status
Status
Definition: gdiplustypes.h:25
LOW_REALTIME_PRIORITY
#define LOW_REALTIME_PRIORITY
NtSetInformationThread
NTSTATUS NTAPI NtSetInformationThread(IN HANDLE ThreadHandle, IN THREADINFOCLASS ThreadInformationClass, IN PVOID ThreadInformation, IN ULONG ThreadInformationLength)
Definition: query.c:2018
STATUS_INVALID_HANDLE
#define STATUS_INVALID_HANDLE
Definition: ntstatus.h:245
STATUS_ACCESS_VIOLATION
#define STATUS_ACCESS_VIOLATION
Definition: ntstatus.h:242
STATUS_DATATYPE_MISALIGNMENT
#define STATUS_DATATYPE_MISALIGNMENT
Definition: ntstatus.h:183
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
STATUS_INFO_LENGTH_MISMATCH
#define STATUS_INFO_LENGTH_MISMATCH
Definition: udferr_usr.h:133
Priority
_In_ WDFINTERRUPT _In_ WDF_INTERRUPT_POLICY _In_ WDF_INTERRUPT_PRIORITY Priority
Definition: wdfinterrupt.h:655
GetCurrentThread
HANDLE WINAPI GetCurrentThread(void)
Definition: proc.c:1148

Referenced by START_TEST().

◆ Test_ThreadSetAlignmentProbe()

static void Test_ThreadSetAlignmentProbe ( void  )
static

Definition at line 100 of file NtSetInformationThread.c.

101{
102 ULONG InfoClass;
103
104 /* Iterate over the process info classes and begin the tests */
105 for (InfoClass = 0; InfoClass < _countof(PsThreadInfoClass); InfoClass++)
106 {
107 /* The buffer is misaligned */
108 QuerySetThreadValidator(SET,
109 InfoClass,
110 (PVOID)(ULONG_PTR)1,
111 PsThreadInfoClass[InfoClass].RequiredSizeSET,
112 STATUS_DATATYPE_MISALIGNMENT);
113
114 /* We set an invalid buffer address */
115 QuerySetThreadValidator(SET,
116 InfoClass,
117 (PVOID)(ULONG_PTR)PsThreadInfoClass[InfoClass].AlignmentSET,
118 PsThreadInfoClass[InfoClass].RequiredSizeSET,
119 STATUS_ACCESS_VIOLATION);
120
121 /* The information length is wrong */
122 QuerySetThreadValidator(SET,
123 InfoClass,
124 (PVOID)(ULONG_PTR)PsThreadInfoClass[InfoClass].AlignmentSET,
125 PsThreadInfoClass[InfoClass].RequiredSizeSET - 1,
126 STATUS_INFO_LENGTH_MISMATCH);
127 }
128}
SET
#define SET(field, seg, reg)
QuerySetThreadValidator
VOID QuerySetThreadValidator(_In_ ALIGNMENT_PROBE_MODE ValidationMode, _In_ ULONG InfoClassIndex, _In_ PVOID InfoPointer, _In_ ULONG InfoLength, _In_ NTSTATUS ExpectedStatus)
Definition: probelib.c:210
PsThreadInfoClass
static const INFORMATION_CLASS_INFO PsThreadInfoClass[]
Definition: ps_i.h:360
_countof
#define _countof(array)
Definition: sndvol32.h:70
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65
ULONG
uint32_t ULONG
Definition: typedefs.h:59

Referenced by START_TEST().