Allow passing task handles to GetThreadQueue() and SetFastQueue().

901ca36b · Ulrich Weigand · Alexandre Julliard · 0be24af7 · 901ca36b
Commit 901ca36b authored Jan 20, 1999 by Ulrich Weigand Committed by Alexandre Julliard Jan 20, 1999
Hide whitespace changes
Inline Side-by-side

Showing with 16 additions and 2 deletions

task.c loader/task.c +16 -2

No files found.
--- a/loader/task.c
+++ b/loader/task.c
@@ -1244,7 +1244,14 @@ HQUEUE16 WINAPI SetThreadQueue( DWORD thread, HQUEUE16 hQueue )
 */
 HQUEUE16 WINAPI GetThreadQueue( DWORD thread )
 {
-    THDB *thdb = thread? THREAD_IdToTHDB( thread ) : THREAD_Current();
+    THDB *thdb = NULL;
+    if ( !thread )
+        thdb = THREAD_Current();
+    else if ( HIWORD(thread) )
+        thdb = THREAD_IdToTHDB( thread );
+    else if ( IsTask( (HTASK16)thread ) )
+        thdb = ((TDB *)GlobalLock16( (HANDLE16)thread ))->thdb;
+
    return (HQUEUE16)(thdb? thdb->teb.queue : 0);
 }

@@ -1253,7 +1260,14 @@ HQUEUE16 WINAPI GetThreadQueue( DWORD thread )
 */
 VOID WINAPI SetFastQueue( DWORD thread, HANDLE32 hQueue )
 {
-    THDB *thdb = thread? THREAD_IdToTHDB( thread ) : THREAD_Current();
+    THDB *thdb = NULL;
+    if ( !thread )
+        thdb = THREAD_Current();
+    else if ( HIWORD(thread) )
+        thdb = THREAD_IdToTHDB( thread );
+    else if ( IsTask( (HTASK16)thread ) )
+        thdb = ((TDB *)GlobalLock16( (HANDLE16)thread ))->thdb;
+
    if ( thdb ) thdb->teb.queue = (HQUEUE16) hQueue;
 }