ntoskrnl.exe: Reimplement PsGetCurrentProcessId and PsGetCurrentThreadId on top of KeGetCurrentThread.

Signed-off-by: Jacek Caban <jacek@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Jacek Caban 2019-04-08 14:04:29 +02:00 committed by Alexandre Julliard
parent f680eda832
commit d9d55bb1d0
2 changed files with 9 additions and 9 deletions

View File

@ -93,9 +93,8 @@ static const WCHAR servicesW[] = {'\\','R','e','g','i','s','t','r','y',
/* tid of the thread running client request */ /* tid of the thread running client request */
static DWORD request_thread; static DWORD request_thread;
/* pid/tid of the client thread */ /* tid of the client thread */
static DWORD client_tid; static DWORD client_tid;
static DWORD client_pid;
struct wine_driver struct wine_driver
{ {
@ -980,7 +979,6 @@ NTSTATUS CDECL wine_ntoskrnl_main_loop( HANDLE stop_event )
irp = wine_server_ptr_handle( reply->next ); irp = wine_server_ptr_handle( reply->next );
irp_params = reply->params; irp_params = reply->params;
client_tid = reply->client_tid; client_tid = reply->client_tid;
client_pid = reply->client_pid;
in_size = reply->in_size; in_size = reply->in_size;
out_size = reply->out_size; out_size = reply->out_size;
} }
@ -2482,12 +2480,17 @@ PEPROCESS WINAPI IoGetCurrentProcess(void)
static void *create_thread_object( HANDLE handle ) static void *create_thread_object( HANDLE handle )
{ {
THREAD_BASIC_INFORMATION info;
struct _KTHREAD *thread; struct _KTHREAD *thread;
if (!(thread = alloc_kernel_object( PsThreadType, handle, sizeof(*thread), 0 ))) return NULL; if (!(thread = alloc_kernel_object( PsThreadType, handle, sizeof(*thread), 0 ))) return NULL;
thread->header.Type = 6; thread->header.Type = 6;
thread->header.WaitListHead.Blink = INVALID_HANDLE_VALUE; /* mark as kernel object */ thread->header.WaitListHead.Blink = INVALID_HANDLE_VALUE; /* mark as kernel object */
if (!NtQueryInformationThread( handle, ThreadBasicInformation, &info, sizeof(info), NULL ))
thread->id = info.ClientId;
return thread; return thread;
} }
@ -2971,9 +2974,7 @@ NTSTATUS WINAPI PsCreateSystemThread(PHANDLE ThreadHandle, ULONG DesiredAccess,
*/ */
HANDLE WINAPI PsGetCurrentProcessId(void) HANDLE WINAPI PsGetCurrentProcessId(void)
{ {
if (GetCurrentThreadId() == request_thread) return KeGetCurrentThread()->id.UniqueProcess;
return UlongToHandle(client_pid);
return UlongToHandle(GetCurrentProcessId());
} }
@ -2982,9 +2983,7 @@ HANDLE WINAPI PsGetCurrentProcessId(void)
*/ */
HANDLE WINAPI PsGetCurrentThreadId(void) HANDLE WINAPI PsGetCurrentThreadId(void)
{ {
if (GetCurrentThreadId() == request_thread) return KeGetCurrentThread()->id.UniqueThread;
return UlongToHandle(client_tid);
return UlongToHandle(GetCurrentThreadId());
} }

View File

@ -31,6 +31,7 @@ struct _OBJECT_TYPE
struct _KTHREAD struct _KTHREAD
{ {
DISPATCHER_HEADER header; DISPATCHER_HEADER header;
CLIENT_ID id;
}; };
void *alloc_kernel_object( POBJECT_TYPE type, HANDLE handle, SIZE_T size, LONG ref ) DECLSPEC_HIDDEN; void *alloc_kernel_object( POBJECT_TYPE type, HANDLE handle, SIZE_T size, LONG ref ) DECLSPEC_HIDDEN;