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

Go to the source code of this file.

Functions

static void Test_ThreadBasicInformationClass (void)
 
static void Test_ThreadQueryAlignmentProbe (void)
 
 START_TEST (NtQueryInformationThread)
 

Function Documentation

◆ START_TEST()

START_TEST ( NtQueryInformationThread  )

Definition at line 136 of file NtQueryInformationThread.c.

137{
138 Test_ThreadBasicInformationClass();
139 Test_ThreadQueryAlignmentProbe();
140}
Test_ThreadBasicInformationClass
static void Test_ThreadBasicInformationClass(void)
Definition: NtQueryInformationThread.c:13
Test_ThreadQueryAlignmentProbe
static void Test_ThreadQueryAlignmentProbe(void)
Definition: NtQueryInformationThread.c:106

◆ Test_ThreadBasicInformationClass()

static void Test_ThreadBasicInformationClass ( void  )
static

Definition at line 13 of file NtQueryInformationThread.c.

14{
15 NTSTATUS Status;
16 PTHREAD_BASIC_INFORMATION ThreadInfoBasic;
17 ULONG ReturnedLength;
18
19 ThreadInfoBasic = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(THREAD_BASIC_INFORMATION));
20 if (!ThreadInfoBasic)
21 {
22 skip("Failed to allocate memory for THREAD_BASIC_INFORMATION!\n");
23 return;
24 }
25
26 /* Everything is NULL */
27 Status = NtQueryInformationThread(NULL,
28 ThreadBasicInformation,
29 NULL,
30 0,
31 NULL);
32 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
33
34 /* Don't give a valid thread handle */
35 Status = NtQueryInformationThread(NULL,
36 ThreadBasicInformation,
37 ThreadInfoBasic,
38 sizeof(THREAD_BASIC_INFORMATION),
39 NULL);
40 ok_hex(Status, STATUS_INVALID_HANDLE);
41
42 /* The information length is incorrect */
43 Status = NtQueryInformationThread(GetCurrentThread(),
44 ThreadBasicInformation,
45 ThreadInfoBasic,
46 0,
47 NULL);
48 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
49
50 /* Don't query anything from the function */
51 Status = NtQueryInformationThread(GetCurrentThread(),
52 ThreadBasicInformation,
53 NULL,
54 sizeof(THREAD_BASIC_INFORMATION),
55 NULL);
56 ok_hex(Status, STATUS_ACCESS_VIOLATION);
57
58 /* The buffer is misaligned and length information is wrong */
59 Status = NtQueryInformationThread(GetCurrentThread(),
60 ThreadBasicInformation,
61 (PVOID)1,
62 0,
63 NULL);
64 ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH);
65
66 /* The buffer is misaligned */
67 Status = NtQueryInformationThread(GetCurrentThread(),
68 ThreadBasicInformation,
69 (PVOID)1,
70 sizeof(THREAD_BASIC_INFORMATION),
71 NULL);
72 ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
73
74 /* The buffer is misaligned, try with an alignment size of 2 */
75 Status = NtQueryInformationThread(GetCurrentThread(),
76 ThreadBasicInformation,
77 (PVOID)2,
78 sizeof(THREAD_BASIC_INFORMATION),
79 NULL);
80 ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT);
81
82 /* Query the basic information we need from the thread */
83 Status = NtQueryInformationThread(GetCurrentThread(),
84 ThreadBasicInformation,
85 ThreadInfoBasic,
86 sizeof(THREAD_BASIC_INFORMATION),
87 &ReturnedLength);
88 ok_hex(Status, STATUS_SUCCESS);
89 ok(ReturnedLength != 0, "The size of the buffer pointed by ThreadInformation shouldn't be 0!\n");
90
91 /* Output the thread basic information details */
92 trace("ReturnedLength = %lu\n", ReturnedLength);
93 trace("ThreadInfoBasic->ExitStatus = 0x%08lx\n", ThreadInfoBasic->ExitStatus);
94 trace("ThreadInfoBasic->TebBaseAddress = %p\n", ThreadInfoBasic->TebBaseAddress);
95 trace("ThreadInfoBasic->ClientId.UniqueProcess = %p\n", ThreadInfoBasic->ClientId.UniqueProcess);
96 trace("ThreadInfoBasic->ClientId.UniqueThread = %p\n", ThreadInfoBasic->ClientId.UniqueThread);
97 trace("ThreadInfoBasic->AffinityMask = %lu\n", ThreadInfoBasic->AffinityMask);
98 trace("ThreadInfoBasic->Priority = %li\n", ThreadInfoBasic->Priority);
99 trace("ThreadInfoBasic->BasePriority = %li\n", ThreadInfoBasic->BasePriority);
100
101 HeapFree(GetProcessHeap(), 0, ThreadInfoBasic);
102}
ok_hex
#define ok_hex(expression, result)
Definition: atltest.h:94
trace
#define trace
Definition: atltest.h:70
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
ReturnedLength
_In_ ULONG _In_ BATTERY_QUERY_INFORMATION_LEVEL _In_ LONG _In_ ULONG _Out_ PULONG ReturnedLength
Definition: batclass.h:188
STATUS_INVALID_HANDLE
#define STATUS_INVALID_HANDLE
Definition: d3dkmdt.h:40
NULL
#define NULL
Definition: types.h:112
GetProcessHeap
#define GetProcessHeap()
Definition: compat.h:736
HeapAlloc
#define HeapAlloc
Definition: compat.h:733
ThreadBasicInformation
@ ThreadBasicInformation
Definition: compat.h:935
HeapFree
#define HeapFree(x, y, z)
Definition: compat.h:735
HEAP_ZERO_MEMORY
#define HEAP_ZERO_MEMORY
Definition: compat.h:134
Status
Status
Definition: gdiplustypes.h:25
NtQueryInformationThread
NTSTATUS NTAPI NtQueryInformationThread(IN HANDLE ThreadHandle, IN THREADINFOCLASS ThreadInformationClass, OUT PVOID ThreadInformation, IN ULONG ThreadInformationLength, OUT PULONG ReturnLength OPTIONAL)
Definition: query.c:2673
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
_CLIENT_ID::UniqueThread
HANDLE UniqueThread
Definition: compat.h:826
_CLIENT_ID::UniqueProcess
HANDLE UniqueProcess
Definition: compat.h:825
_THREAD_BASIC_INFORMATION
Definition: compat.h:926
_THREAD_BASIC_INFORMATION::ClientId
CLIENT_ID ClientId
Definition: compat.h:929
_THREAD_BASIC_INFORMATION::Priority
KPRIORITY Priority
Definition: compat.h:931
_THREAD_BASIC_INFORMATION::BasePriority
KPRIORITY BasePriority
Definition: compat.h:932
_THREAD_BASIC_INFORMATION::ExitStatus
NTSTATUS ExitStatus
Definition: compat.h:927
_THREAD_BASIC_INFORMATION::TebBaseAddress
PVOID TebBaseAddress
Definition: compat.h:928
_THREAD_BASIC_INFORMATION::AffinityMask
KAFFINITY AffinityMask
Definition: compat.h:930
ULONG
uint32_t ULONG
Definition: typedefs.h:59
STATUS_INFO_LENGTH_MISMATCH
#define STATUS_INFO_LENGTH_MISMATCH
Definition: udferr_usr.h:133
GetCurrentThread
HANDLE WINAPI GetCurrentThread(void)
Definition: proc.c:1148

Referenced by START_TEST().

◆ Test_ThreadQueryAlignmentProbe()

static void Test_ThreadQueryAlignmentProbe ( void  )
static

Definition at line 106 of file NtQueryInformationThread.c.

107{
108 ULONG InfoClass;
109
110 /* Iterate over the process info classes and begin the tests */
111 for (InfoClass = 0; InfoClass < _countof(PsThreadInfoClass); InfoClass++)
112 {
113 /* The buffer is misaligned */
114 QuerySetThreadValidator(QUERY,
115 InfoClass,
116 (PVOID)(ULONG_PTR)1,
117 PsThreadInfoClass[InfoClass].RequiredSizeQUERY,
118 STATUS_DATATYPE_MISALIGNMENT);
119
120 /* We query an invalid buffer address */
121 QuerySetThreadValidator(QUERY,
122 InfoClass,
123 (PVOID)(ULONG_PTR)PsThreadInfoClass[InfoClass].AlignmentQUERY,
124 PsThreadInfoClass[InfoClass].RequiredSizeQUERY,
125 STATUS_ACCESS_VIOLATION);
126
127 /* The information length is wrong */
128 QuerySetThreadValidator(QUERY,
129 InfoClass,
130 (PVOID)(ULONG_PTR)PsThreadInfoClass[InfoClass].AlignmentQUERY,
131 PsThreadInfoClass[InfoClass].RequiredSizeQUERY - 1,
132 STATUS_INFO_LENGTH_MISMATCH);
133 }
134}
QUERY
@ QUERY
Definition: precomp.h:18
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:362
_countof
#define _countof(array)
Definition: sndvol32.h:70
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65

Referenced by START_TEST().