Signed-off-by: Derek Lesho <dereklesho52@Gmail.com>
---
��dlls/ntoskrnl.exe/ntoskrnl.c�� �� �� �� | 18 ++++++++++++++++++
��dlls/ntoskrnl.exe/ntoskrnl.exe.spec |�� 2 +-
��dlls/ntoskrnl.exe/tests/driver.c�� �� | 18 ++++++++++++++++++
��include/ddk/ntifs.h�� �� �� �� �� �� �� �� ��|�� 1 +
��4 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/dlls/ntoskrnl.exe/ntoskrnl.c b/dlls/ntoskrnl.exe/ntoskrnl.c
index 3789abda35..0ab150ea65 100644
--- a/dlls/ntoskrnl.exe/ntoskrnl.c
+++ b/dlls/ntoskrnl.exe/ntoskrnl.c
@@ -3233,6 +3233,24 @@ NTSTATUS WINAPI PsLookupProcessByProcessId(HANDLE processid, PEPROCESS *process)
��}


+/*****************************************************
+ *�� �� �� �� �� ��PsLookupThreadByThreadId�� ��(NTOSKRNL.EXE.@)
+ */
+NTSTATUS WINAPI PsLookupThreadByThreadId(HANDLE threadid, PETHREAD *thread)
+{
+�� �� NTSTATUS status;
+�� �� HANDLE hThread = OpenThread( THREAD_ALL_ACCESS, FALSE, HandleToUlong(threadid) );
+
+�� �� if (!hThread)
+�� �� �� �� return STATUS_INVALID_PARAMETER;
+
+�� �� status = ObReferenceObjectByHandle( hThread, THREAD_ALL_ACCESS, PsThreadType, KernelMode, (void**)thread, NULL );
+
+�� �� NtClose( hThread );
+�� �� return status;
+}
+
+
��/*****************************************************
�� *�� �� �� �� �� ��IoSetThreadHardErrorMode�� (NTOSKRNL.EXE.@)
�� */
diff --git a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
index 141c8aa639..f965ff55ec 100644
--- a/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
+++ b/dlls/ntoskrnl.exe/ntoskrnl.exe.spec
@@ -914,7 +914,7 @@
��@ stub PsJobType
��@ stdcall PsLookupProcessByProcessId(ptr ptr)
��@ stub PsLookupProcessThreadByCid
-@ stub PsLookupThreadByThreadId
+@ stdcall PsLookupThreadByThreadId(ptr ptr)
��@ extern PsProcessType
��@ stub PsReferenceImpersonationToken
��@ stub PsReferencePrimaryToken
diff --git a/dlls/ntoskrnl.exe/tests/driver.c b/dlls/ntoskrnl.exe/tests/driver.c
index c3839da3bf..4b846869fc 100644
--- a/dlls/ntoskrnl.exe/tests/driver.c
+++ b/dlls/ntoskrnl.exe/tests/driver.c
@@ -29,6 +29,7 @@
��#include "winternl.h"
��#include "winioctl.h"
��#include "ddk/ntddk.h"
+#include "ddk/ntifs.h"
��#include "ddk/wdm.h"

��#include "driver.h"
@@ -1167,6 +1168,22 @@ static void test_resource(void)
�� �� ��ok(status == STATUS_SUCCESS, "got status %#x\n", status);
��}

+static void test_lookup_thread(void)
+{
+�� �� NTSTATUS status;
+�� �� PETHREAD thread = NULL;
+
+�� �� status = PsLookupThreadByThreadId(PsGetCurrentThreadId(), &thread);
+�� �� ok(!status, "PsLookupThreadByThreadId failed: %#x\n", status);
+�� �� ok((PKTHREAD)thread == KeGetCurrentThread(), "thread != KeGetCurrentThread\n");
+
+�� �� if (thread)
+�� �� �� �� ObDereferenceObject(thread);
+
+�� �� status = PsLookupThreadByThreadId(NULL, &thread);
+�� �� ok(status == STATUS_INVALID_PARAMETER, "PsLookupThreadByThreadId returned %#x\n", status);
+}
+
��static NTSTATUS main_test(DEVICE_OBJECT *device, IRP *irp, IO_STACK_LOCATION *stack, ULONG_PTR *info)
��{
�� �� ��ULONG length = stack->Parameters.DeviceIoControl.OutputBufferLength;
@@ -1210,6 +1227,7 @@ static NTSTATUS main_test(DEVICE_OBJECT *device, IRP *irp, IO_STACK_LOCATION *st
�� �� ��test_lookaside_list();
�� �� ��test_ob_reference(test_input->path);
�� �� ��test_resource();
+�� �� test_lookup_thread();

�� �� ��/* print process report */
�� �� ��if (winetest_debug)
diff --git a/include/ddk/ntifs.h b/include/ddk/ntifs.h
index abe357fbc9..9b57ae7ad7 100644
--- a/include/ddk/ntifs.h
+++ b/include/ddk/ntifs.h
@@ -131,6 +131,7 @@ typedef struct _FS_FILTER_CALLBACKS

��BOOLEAN WINAPI FsRtlIsNameInExpression(PUNICODE_STRING, PUNICODE_STRING, BOOLEAN, PWCH);
��NTSTATUS WINAPI ObQueryNameString(PVOID,POBJECT_NAME_INFORMATION,ULONG,PULONG);
+NTSTATUS WINAPI PsLookupThreadByThreadId(HANDLE,PETHREAD*);
��void WINAPI PsRevertToSelf(void);

��#endif
--
2.20.1