git.oblomov.eu Git - wine/blob - dlls/ntdll/thread.c

   1 /*
   2  * NT threads support
   3  *
   4  * Copyright 1996, 2003 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <assert.h>
  25 #include <stdarg.h>
  26 #include <sys/types.h>
  27 #ifdef HAVE_SYS_MMAN_H
  28 #include <sys/mman.h>
  29 #endif
  30 #ifdef HAVE_SYS_TIMES_H
  31 #include <sys/times.h>
  32 #endif
  33
  34 #define NONAMELESSUNION
  35 #include "ntstatus.h"
  36 #define WIN32_NO_STATUS
  37 #include "winternl.h"
  38 #include "wine/library.h"
  39 #include "wine/server.h"
  40 #include "wine/debug.h"
  41 #include "ntdll_misc.h"
  42 #include "ddk/wdm.h"
  43 #include "wine/exception.h"
  44
  45 WINE_DEFAULT_DEBUG_CHANNEL(thread);
  46 WINE_DECLARE_DEBUG_CHANNEL(relay);
  47
  48 struct _KUSER_SHARED_DATA *user_shared_data = NULL;
  49
  50 PUNHANDLED_EXCEPTION_FILTER unhandled_exception_filter = NULL;
  51
  52 /* info passed to a starting thread */
  53 struct startup_info
  54 {
  55     TEB                            *teb;
  56     PRTL_THREAD_START_ROUTINE       entry_point;
  57     void                           *entry_arg;
  58 };
  59
  60 static PEB *peb;
  61 static PEB_LDR_DATA ldr;
  62 static RTL_USER_PROCESS_PARAMETERS params;  /* default parameters if no parent */
  63 static WCHAR current_dir[MAX_NT_PATH_LENGTH];
  64 static RTL_BITMAP tls_bitmap;
  65 static RTL_BITMAP tls_expansion_bitmap;
  66 static RTL_BITMAP fls_bitmap;
  67 static LIST_ENTRY tls_links;
  68 static int nb_threads = 1;
  69
  70 /***********************************************************************
  71  *           get_unicode_string
  72  *
  73  * Copy a unicode string from the startup info.
  74  */
  75 static inline void get_unicode_string( UNICODE_STRING *str, WCHAR **src, WCHAR **dst, UINT len )
  76 {
  77     str->Buffer = *dst;
  78     str->Length = len;
  79     str->MaximumLength = len + sizeof(WCHAR);
  80     memcpy( str->Buffer, *src, len );
  81     str->Buffer[len / sizeof(WCHAR)] = 0;
  82     *src += len / sizeof(WCHAR);
  83     *dst += len / sizeof(WCHAR) + 1;
  84 }
  85
  86 /***********************************************************************
  87  *           init_user_process_params
  88  *
  89  * Fill the RTL_USER_PROCESS_PARAMETERS structure from the server.
  90  */
  91 static NTSTATUS init_user_process_params( SIZE_T data_size, HANDLE *exe_file )
  92 {
  93     void *ptr;
  94     WCHAR *src, *dst;
  95     SIZE_T info_size, env_size, size, alloc_size;
  96     NTSTATUS status;
  97     startup_info_t *info;
  98     RTL_USER_PROCESS_PARAMETERS *params = NULL;
  99
 100     if (!(info = RtlAllocateHeap( GetProcessHeap(), 0, data_size )))
 101         return STATUS_NO_MEMORY;
 102
 103     SERVER_START_REQ( get_startup_info )
 104     {
 105         wine_server_set_reply( req, info, data_size );
 106         if (!(status = wine_server_call( req )))
 107         {
 108             data_size = wine_server_reply_size( reply );
 109             info_size = reply->info_size;
 110             env_size  = data_size - info_size;
 111             *exe_file = wine_server_ptr_handle( reply->exe_file );
 112         }
 113     }
 114     SERVER_END_REQ;
 115     if (status != STATUS_SUCCESS) goto done;
 116
 117     size = sizeof(*params);
 118     size += MAX_NT_PATH_LENGTH * sizeof(WCHAR);
 119     size += info->dllpath_len + sizeof(WCHAR);
 120     size += info->imagepath_len + sizeof(WCHAR);
 121     size += info->cmdline_len + sizeof(WCHAR);
 122     size += info->title_len + sizeof(WCHAR);
 123     size += info->desktop_len + sizeof(WCHAR);
 124     size += info->shellinfo_len + sizeof(WCHAR);
 125     size += info->runtime_len + sizeof(WCHAR);
 126
 127     alloc_size = size;
 128     status = NtAllocateVirtualMemory( NtCurrentProcess(), (void **)&params, 0, &alloc_size,
 129                                       MEM_COMMIT, PAGE_READWRITE );
 130     if (status != STATUS_SUCCESS) goto done;
 131
 132     NtCurrentTeb()->Peb->ProcessParameters = params;
 133     params->AllocationSize  = alloc_size;
 134     params->Size            = size;
 135     params->Flags           = PROCESS_PARAMS_FLAG_NORMALIZED;
 136     params->DebugFlags      = info->debug_flags;
 137     params->ConsoleHandle   = wine_server_ptr_handle( info->console );
 138     params->ConsoleFlags    = info->console_flags;
 139     params->hStdInput       = wine_server_ptr_handle( info->hstdin );
 140     params->hStdOutput      = wine_server_ptr_handle( info->hstdout );
 141     params->hStdError       = wine_server_ptr_handle( info->hstderr );
 142     params->dwX             = info->x;
 143     params->dwY             = info->y;
 144     params->dwXSize         = info->xsize;
 145     params->dwYSize         = info->ysize;
 146     params->dwXCountChars   = info->xchars;
 147     params->dwYCountChars   = info->ychars;
 148     params->dwFillAttribute = info->attribute;
 149     params->dwFlags         = info->flags;
 150     params->wShowWindow     = info->show;
 151
 152     src = (WCHAR *)(info + 1);
 153     dst = (WCHAR *)(params + 1);
 154
 155     /* current directory needs more space */
 156     get_unicode_string( &params->CurrentDirectory.DosPath, &src, &dst, info->curdir_len );
 157     params->CurrentDirectory.DosPath.MaximumLength = MAX_NT_PATH_LENGTH * sizeof(WCHAR);
 158     dst = (WCHAR *)(params + 1) + MAX_NT_PATH_LENGTH;
 159
 160     get_unicode_string( &params->DllPath, &src, &dst, info->dllpath_len );
 161     get_unicode_string( &params->ImagePathName, &src, &dst, info->imagepath_len );
 162     get_unicode_string( &params->CommandLine, &src, &dst, info->cmdline_len );
 163     get_unicode_string( &params->WindowTitle, &src, &dst, info->title_len );
 164     get_unicode_string( &params->Desktop, &src, &dst, info->desktop_len );
 165     get_unicode_string( &params->ShellInfo, &src, &dst, info->shellinfo_len );
 166
 167     /* runtime info isn't a real string */
 168     params->RuntimeInfo.Buffer = dst;
 169     params->RuntimeInfo.Length = params->RuntimeInfo.MaximumLength = info->runtime_len;
 170     memcpy( dst, src, info->runtime_len );
 171
 172     /* environment needs to be a separate memory block */
 173     ptr = NULL;
 174     alloc_size = max( 1, env_size );
 175     status = NtAllocateVirtualMemory( NtCurrentProcess(), &ptr, 0, &alloc_size,
 176                                       MEM_COMMIT, PAGE_READWRITE );
 177     if (status != STATUS_SUCCESS) goto done;
 178     memcpy( ptr, (char *)info + info_size, env_size );
 179     params->Environment = ptr;
 180
 181 done:
 182     RtlFreeHeap( GetProcessHeap(), 0, info );
 183     return status;
 184 }
 185
 186 /***********************************************************************
 187  *           thread_init
 188  *
 189  * Setup the initial thread.
 190  *
 191  * NOTES: The first allocated TEB on NT is at 0x7ffde000.
 192  */
 193 HANDLE thread_init(void)
 194 {
 195     TEB *teb;
 196     void *addr;
 197     SIZE_T size, info_size;
 198     HANDLE exe_file = 0;
 199     LARGE_INTEGER now;
 200     struct ntdll_thread_data *thread_data;
 201     static struct debug_info debug_info;  /* debug info for initial thread */
 202
 203     virtual_init();
 204
 205     /* reserve space for shared user data */
 206
 207     addr = (void *)0x7ffe0000;
 208     size = 0x10000;
 209     NtAllocateVirtualMemory( NtCurrentProcess(), &addr, 0, &size, MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE );
 210     user_shared_data = addr;
 211
 212     /* allocate and initialize the PEB */
 213
 214     addr = NULL;
 215     size = sizeof(*peb);
 216     NtAllocateVirtualMemory( NtCurrentProcess(), &addr, 1, &size,
 217                              MEM_COMMIT | MEM_TOP_DOWN, PAGE_READWRITE );
 218     peb = addr;
 219
 220     peb->ProcessParameters  = &params;
 221     peb->TlsBitmap          = &tls_bitmap;
 222     peb->TlsExpansionBitmap = &tls_expansion_bitmap;
 223     peb->FlsBitmap          = &fls_bitmap;
 224     peb->LdrData            = &ldr;
 225     params.CurrentDirectory.DosPath.Buffer = current_dir;
 226     params.CurrentDirectory.DosPath.MaximumLength = sizeof(current_dir);
 227     params.wShowWindow = 1; /* SW_SHOWNORMAL */
 228     RtlInitializeBitMap( &tls_bitmap, peb->TlsBitmapBits, sizeof(peb->TlsBitmapBits) * 8 );
 229     RtlInitializeBitMap( &tls_expansion_bitmap, peb->TlsExpansionBitmapBits,
 230                          sizeof(peb->TlsExpansionBitmapBits) * 8 );
 231     RtlInitializeBitMap( &fls_bitmap, peb->FlsBitmapBits, sizeof(peb->FlsBitmapBits) * 8 );
 232     InitializeListHead( &peb->FlsListHead );
 233     InitializeListHead( &ldr.InLoadOrderModuleList );
 234     InitializeListHead( &ldr.InMemoryOrderModuleList );
 235     InitializeListHead( &ldr.InInitializationOrderModuleList );
 236     InitializeListHead( &tls_links );
 237
 238     /* allocate and initialize the initial TEB */
 239
 240     signal_alloc_thread( &teb );
 241     teb->Peb = peb;
 242     teb->Tib.StackBase = (void *)~0UL;
 243     teb->StaticUnicodeString.Buffer = teb->StaticUnicodeBuffer;
 244     teb->StaticUnicodeString.MaximumLength = sizeof(teb->StaticUnicodeBuffer);
 245
 246     thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
 247     thread_data->request_fd = -1;
 248     thread_data->reply_fd   = -1;
 249     thread_data->wait_fd[0] = -1;
 250     thread_data->wait_fd[1] = -1;
 251     thread_data->debug_info = &debug_info;
 252     InsertHeadList( &tls_links, &teb->TlsLinks );
 253
 254     signal_init_thread( teb );
 255     virtual_init_threading();
 256
 257     debug_info.str_pos = debug_info.strings;
 258     debug_info.out_pos = debug_info.output;
 259     debug_init();
 260
 261     /* setup the server connection */
 262     server_init_process();
 263     info_size = server_init_thread( peb );
 264
 265     /* create the process heap */
 266     if (!(peb->ProcessHeap = RtlCreateHeap( HEAP_GROWABLE, NULL, 0, 0, NULL, NULL )))
 267     {
 268         MESSAGE( "wine: failed to create the process heap\n" );
 269         exit(1);
 270     }
 271
 272     /* allocate user parameters */
 273     if (info_size)
 274     {
 275         init_user_process_params( info_size, &exe_file );
 276     }
 277     else
 278     {
 279         if (isatty(0) || isatty(1) || isatty(2))
 280             params.ConsoleHandle = (HANDLE)2; /* see kernel32/kernel_private.h */
 281         if (!isatty(0))
 282             wine_server_fd_to_handle( 0, GENERIC_READ|SYNCHRONIZE,  OBJ_INHERIT, &params.hStdInput );
 283         if (!isatty(1))
 284             wine_server_fd_to_handle( 1, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, &params.hStdOutput );
 285         if (!isatty(2))
 286             wine_server_fd_to_handle( 2, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, &params.hStdError );
 287     }
 288
 289     /* initialize time values in user_shared_data */
 290     NtQuerySystemTime( &now );
 291     user_shared_data->SystemTime.LowPart = now.u.LowPart;
 292     user_shared_data->SystemTime.High1Time = user_shared_data->SystemTime.High2Time = now.u.HighPart;
 293     user_shared_data->u.TickCountQuad = (now.QuadPart - server_start_time) / 10000;
 294     user_shared_data->u.TickCount.High2Time = user_shared_data->u.TickCount.High1Time;
 295     user_shared_data->TickCountLowDeprecated = user_shared_data->u.TickCount.LowPart;
 296     user_shared_data->TickCountMultiplier = 1 << 24;
 297
 298     fill_cpu_info();
 299     exceptions_init();
 300
 301     return exe_file;
 302 }
 303
 304
 305 /***********************************************************************
 306  *           terminate_thread
 307  */
 308 void terminate_thread( int status )
 309 {
 310     pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
 311     if (interlocked_xchg_add( &nb_threads, -1 ) <= 1) _exit( status );
 312
 313     close( ntdll_get_thread_data()->wait_fd[0] );
 314     close( ntdll_get_thread_data()->wait_fd[1] );
 315     close( ntdll_get_thread_data()->reply_fd );
 316     close( ntdll_get_thread_data()->request_fd );
 317     pthread_exit( UIntToPtr(status) );
 318 }
 319
 320
 321 /***********************************************************************
 322  *           exit_thread
 323  */
 324 void exit_thread( int status )
 325 {
 326     static void *prev_teb;
 327     TEB *teb;
 328
 329     if (status)  /* send the exit code to the server (0 is already the default) */
 330     {
 331         SERVER_START_REQ( terminate_thread )
 332         {
 333             req->handle    = wine_server_obj_handle( GetCurrentThread() );
 334             req->exit_code = status;
 335             wine_server_call( req );
 336         }
 337         SERVER_END_REQ;
 338     }
 339
 340     if (interlocked_xchg_add( &nb_threads, -1 ) <= 1)
 341     {
 342         LdrShutdownProcess();
 343         exit( status );
 344     }
 345
 346     LdrShutdownThread();
 347     RtlAcquirePebLock();
 348     RemoveEntryList( &NtCurrentTeb()->TlsLinks );
 349     RtlReleasePebLock();
 350     RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->FlsSlots );
 351     RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->TlsExpansionSlots );
 352
 353     pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
 354
 355     if ((teb = interlocked_xchg_ptr( &prev_teb, NtCurrentTeb() )))
 356     {
 357         struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
 358
 359         if (thread_data->pthread_id)
 360         {
 361             pthread_join( thread_data->pthread_id, NULL );
 362             signal_free_thread( teb );
 363         }
 364     }
 365
 366     close( ntdll_get_thread_data()->wait_fd[0] );
 367     close( ntdll_get_thread_data()->wait_fd[1] );
 368     close( ntdll_get_thread_data()->reply_fd );
 369     close( ntdll_get_thread_data()->request_fd );
 370     pthread_exit( UIntToPtr(status) );
 371 }
 372
 373
 374 /***********************************************************************
 375  *           start_thread
 376  *
 377  * Startup routine for a newly created thread.
 378  */
 379 static void start_thread( struct startup_info *info )
 380 {
 381     TEB *teb = info->teb;
 382     struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
 383     PRTL_THREAD_START_ROUTINE func = info->entry_point;
 384     void *arg = info->entry_arg;
 385     struct debug_info debug_info;
 386
 387     debug_info.str_pos = debug_info.strings;
 388     debug_info.out_pos = debug_info.output;
 389     thread_data->debug_info = &debug_info;
 390     thread_data->pthread_id = pthread_self();
 391
 392     signal_init_thread( teb );
 393     server_init_thread( func );
 394     pthread_sigmask( SIG_UNBLOCK, &server_block_set, NULL );
 395
 396     RtlAcquirePebLock();
 397     InsertHeadList( &tls_links, &teb->TlsLinks );
 398     RtlReleasePebLock();
 399
 400     MODULE_DllThreadAttach( NULL );
 401
 402     if (TRACE_ON(relay))
 403         DPRINTF( "%04x:Starting thread proc %p (arg=%p)\n", GetCurrentThreadId(), func, arg );
 404
 405     call_thread_entry_point( (LPTHREAD_START_ROUTINE)func, arg );
 406 }
 407
 408
 409 /***********************************************************************
 410  *              RtlCreateUserThread   (NTDLL.@)
 411  */
 412 NTSTATUS WINAPI RtlCreateUserThread( HANDLE process, const SECURITY_DESCRIPTOR *descr,
 413                                      BOOLEAN suspended, PVOID stack_addr,
 414                                      SIZE_T stack_reserve, SIZE_T stack_commit,
 415                                      PRTL_THREAD_START_ROUTINE start, void *param,
 416                                      HANDLE *handle_ptr, CLIENT_ID *id )
 417 {
 418     sigset_t sigset;
 419     pthread_t pthread_id;
 420     pthread_attr_t attr;
 421     struct ntdll_thread_data *thread_data;
 422     struct startup_info *info = NULL;
 423     HANDLE handle = 0;
 424     TEB *teb = NULL;
 425     DWORD tid = 0;
 426     int request_pipe[2];
 427     NTSTATUS status;
 428
 429     if (process != NtCurrentProcess())
 430     {
 431         apc_call_t call;
 432         apc_result_t result;
 433
 434         memset( &call, 0, sizeof(call) );
 435
 436         call.create_thread.type    = APC_CREATE_THREAD;
 437         call.create_thread.func    = wine_server_client_ptr( start );
 438         call.create_thread.arg     = wine_server_client_ptr( param );
 439         call.create_thread.reserve = stack_reserve;
 440         call.create_thread.commit  = stack_commit;
 441         call.create_thread.suspend = suspended;
 442         status = NTDLL_queue_process_apc( process, &call, &result );
 443         if (status != STATUS_SUCCESS) return status;
 444
 445         if (result.create_thread.status == STATUS_SUCCESS)
 446         {
 447             if (id) id->UniqueThread = ULongToHandle(result.create_thread.tid);
 448             if (handle_ptr) *handle_ptr = wine_server_ptr_handle( result.create_thread.handle );
 449             else NtClose( wine_server_ptr_handle( result.create_thread.handle ));
 450         }
 451         return result.create_thread.status;
 452     }
 453
 454     if (server_pipe( request_pipe ) == -1) return STATUS_TOO_MANY_OPENED_FILES;
 455     wine_server_send_fd( request_pipe[0] );
 456
 457     SERVER_START_REQ( new_thread )
 458     {
 459         req->access     = THREAD_ALL_ACCESS;
 460         req->attributes = 0;  /* FIXME */
 461         req->suspend    = suspended;
 462         req->request_fd = request_pipe[0];
 463         if (!(status = wine_server_call( req )))
 464         {
 465             handle = wine_server_ptr_handle( reply->handle );
 466             tid = reply->tid;
 467         }
 468         close( request_pipe[0] );
 469     }
 470     SERVER_END_REQ;
 471
 472     if (status)
 473     {
 474         close( request_pipe[1] );
 475         return status;
 476     }
 477
 478     pthread_sigmask( SIG_BLOCK, &server_block_set, &sigset );
 479
 480     if ((status = signal_alloc_thread( &teb ))) goto error;
 481
 482     teb->Peb = NtCurrentTeb()->Peb;
 483     teb->ClientId.UniqueProcess = ULongToHandle(GetCurrentProcessId());
 484     teb->ClientId.UniqueThread  = ULongToHandle(tid);
 485     teb->StaticUnicodeString.Buffer        = teb->StaticUnicodeBuffer;
 486     teb->StaticUnicodeString.MaximumLength = sizeof(teb->StaticUnicodeBuffer);
 487
 488     info = (struct startup_info *)(teb + 1);
 489     info->teb         = teb;
 490     info->entry_point = start;
 491     info->entry_arg   = param;
 492
 493     thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
 494     thread_data->request_fd  = request_pipe[1];
 495     thread_data->reply_fd    = -1;
 496     thread_data->wait_fd[0]  = -1;
 497     thread_data->wait_fd[1]  = -1;
 498
 499     if ((status = virtual_alloc_thread_stack( teb, stack_reserve, stack_commit ))) goto error;
 500
 501     pthread_attr_init( &attr );
 502     pthread_attr_setstack( &attr, teb->DeallocationStack,
 503                            (char *)teb->Tib.StackBase - (char *)teb->DeallocationStack );
 504     pthread_attr_setscope( &attr, PTHREAD_SCOPE_SYSTEM ); /* force creating a kernel thread */
 505     interlocked_xchg_add( &nb_threads, 1 );
 506     if (pthread_create( &pthread_id, &attr, (void * (*)(void *))start_thread, info ))
 507     {
 508         interlocked_xchg_add( &nb_threads, -1 );
 509         pthread_attr_destroy( &attr );
 510         status = STATUS_NO_MEMORY;
 511         goto error;
 512     }
 513     pthread_attr_destroy( &attr );
 514     pthread_sigmask( SIG_SETMASK, &sigset, NULL );
 515
 516     if (id) id->UniqueThread = ULongToHandle(tid);
 517     if (handle_ptr) *handle_ptr = handle;
 518     else NtClose( handle );
 519
 520     return STATUS_SUCCESS;
 521
 522 error:
 523     if (teb) signal_free_thread( teb );
 524     if (handle) NtClose( handle );
 525     pthread_sigmask( SIG_SETMASK, &sigset, NULL );
 526     close( request_pipe[1] );
 527     return status;
 528 }
 529
 530
 531 /******************************************************************************
 532  *              RtlGetNtGlobalFlags   (NTDLL.@)
 533  */
 534 ULONG WINAPI RtlGetNtGlobalFlags(void)
 535 {
 536     if (!peb) return 0;  /* init not done yet */
 537     return peb->NtGlobalFlag;
 538 }
 539
 540
 541 /***********************************************************************
 542  *              NtOpenThread   (NTDLL.@)
 543  *              ZwOpenThread   (NTDLL.@)
 544  */
 545 NTSTATUS WINAPI NtOpenThread( HANDLE *handle, ACCESS_MASK access,
 546                               const OBJECT_ATTRIBUTES *attr, const CLIENT_ID *id )
 547 {
 548     NTSTATUS ret;
 549
 550     SERVER_START_REQ( open_thread )
 551     {
 552         req->tid        = HandleToULong(id->UniqueThread);
 553         req->access     = access;
 554         req->attributes = attr ? attr->Attributes : 0;
 555         ret = wine_server_call( req );
 556         *handle = wine_server_ptr_handle( reply->handle );
 557     }
 558     SERVER_END_REQ;
 559     return ret;
 560 }
 561
 562
 563 /******************************************************************************
 564  *              NtSuspendThread   (NTDLL.@)
 565  *              ZwSuspendThread   (NTDLL.@)
 566  */
 567 NTSTATUS WINAPI NtSuspendThread( HANDLE handle, PULONG count )
 568 {
 569     NTSTATUS ret;
 570
 571     SERVER_START_REQ( suspend_thread )
 572     {
 573         req->handle = wine_server_obj_handle( handle );
 574         if (!(ret = wine_server_call( req ))) *count = reply->count;
 575     }
 576     SERVER_END_REQ;
 577     return ret;
 578 }
 579
 580
 581 /******************************************************************************
 582  *              NtResumeThread   (NTDLL.@)
 583  *              ZwResumeThread   (NTDLL.@)
 584  */
 585 NTSTATUS WINAPI NtResumeThread( HANDLE handle, PULONG count )
 586 {
 587     NTSTATUS ret;
 588
 589     SERVER_START_REQ( resume_thread )
 590     {
 591         req->handle = wine_server_obj_handle( handle );
 592         if (!(ret = wine_server_call( req ))) *count = reply->count;
 593     }
 594     SERVER_END_REQ;
 595     return ret;
 596 }
 597
 598
 599 /******************************************************************************
 600  *              NtAlertResumeThread   (NTDLL.@)
 601  *              ZwAlertResumeThread   (NTDLL.@)
 602  */
 603 NTSTATUS WINAPI NtAlertResumeThread( HANDLE handle, PULONG count )
 604 {
 605     FIXME( "stub: should alert thread %p\n", handle );
 606     return NtResumeThread( handle, count );
 607 }
 608
 609
 610 /******************************************************************************
 611  *              NtAlertThread   (NTDLL.@)
 612  *              ZwAlertThread   (NTDLL.@)
 613  */
 614 NTSTATUS WINAPI NtAlertThread( HANDLE handle )
 615 {
 616     FIXME( "stub: %p\n", handle );
 617     return STATUS_NOT_IMPLEMENTED;
 618 }
 619
 620
 621 /******************************************************************************
 622  *              NtTerminateThread  (NTDLL.@)
 623  *              ZwTerminateThread  (NTDLL.@)
 624  */
 625 NTSTATUS WINAPI NtTerminateThread( HANDLE handle, LONG exit_code )
 626 {
 627     NTSTATUS ret;
 628     BOOL self;
 629
 630     SERVER_START_REQ( terminate_thread )
 631     {
 632         req->handle    = wine_server_obj_handle( handle );
 633         req->exit_code = exit_code;
 634         ret = wine_server_call( req );
 635         self = !ret && reply->self;
 636     }
 637     SERVER_END_REQ;
 638
 639     if (self) abort_thread( exit_code );
 640     return ret;
 641 }
 642
 643
 644 /******************************************************************************
 645  *              NtQueueApcThread  (NTDLL.@)
 646  */
 647 NTSTATUS WINAPI NtQueueApcThread( HANDLE handle, PNTAPCFUNC func, ULONG_PTR arg1,
 648                                   ULONG_PTR arg2, ULONG_PTR arg3 )
 649 {
 650     NTSTATUS ret;
 651     SERVER_START_REQ( queue_apc )
 652     {
 653         req->handle = wine_server_obj_handle( handle );
 654         if (func)
 655         {
 656             req->call.type         = APC_USER;
 657             req->call.user.func    = wine_server_client_ptr( func );
 658             req->call.user.args[0] = arg1;
 659             req->call.user.args[1] = arg2;
 660             req->call.user.args[2] = arg3;
 661         }
 662         else req->call.type = APC_NONE;  /* wake up only */
 663         ret = wine_server_call( req );
 664     }
 665     SERVER_END_REQ;
 666     return ret;
 667 }
 668
 669
 670 /***********************************************************************
 671  *              NtSetContextThread  (NTDLL.@)
 672  *              ZwSetContextThread  (NTDLL.@)
 673  */
 674 NTSTATUS WINAPI NtSetContextThread( HANDLE handle, const CONTEXT *context )
 675 {
 676     NTSTATUS ret;
 677     DWORD dummy, i;
 678     BOOL self = FALSE;
 679
 680 #ifdef __i386__
 681     /* on i386 debug registers always require a server call */
 682     self = (handle == GetCurrentThread());
 683     if (self && (context->ContextFlags & (CONTEXT_DEBUG_REGISTERS & ~CONTEXT_i386)))
 684     {
 685         self = (ntdll_get_thread_data()->dr0 == context->Dr0 &&
 686                 ntdll_get_thread_data()->dr1 == context->Dr1 &&
 687                 ntdll_get_thread_data()->dr2 == context->Dr2 &&
 688                 ntdll_get_thread_data()->dr3 == context->Dr3 &&
 689                 ntdll_get_thread_data()->dr6 == context->Dr6 &&
 690                 ntdll_get_thread_data()->dr7 == context->Dr7);
 691     }
 692 #endif
 693
 694     if (!self)
 695     {
 696         context_t server_context;
 697
 698         context_to_server( &server_context, context );
 699
 700         SERVER_START_REQ( set_thread_context )
 701         {
 702             req->handle  = wine_server_obj_handle( handle );
 703             req->suspend = 0;
 704             wine_server_add_data( req, &server_context, sizeof(server_context) );
 705             ret = wine_server_call( req );
 706             self = reply->self;
 707         }
 708         SERVER_END_REQ;
 709
 710         if (ret == STATUS_PENDING)
 711         {
 712             if (NtSuspendThread( handle, &dummy ) == STATUS_SUCCESS)
 713             {
 714                 for (i = 0; i < 100; i++)
 715                 {
 716                     SERVER_START_REQ( set_thread_context )
 717                     {
 718                         req->handle  = wine_server_obj_handle( handle );
 719                         req->suspend = 0;
 720                         wine_server_add_data( req, &server_context, sizeof(server_context) );
 721                         ret = wine_server_call( req );
 722                     }
 723                     SERVER_END_REQ;
 724                     if (ret == STATUS_PENDING)
 725                     {
 726                         LARGE_INTEGER timeout;
 727                         timeout.QuadPart = -10000;
 728                         NtDelayExecution( FALSE, &timeout );
 729                     }
 730                     else break;
 731                 }
 732                 NtResumeThread( handle, &dummy );
 733             }
 734             if (ret == STATUS_PENDING) ret = STATUS_ACCESS_DENIED;
 735         }
 736
 737         if (ret) return ret;
 738     }
 739
 740     if (self) set_cpu_context( context );
 741     return STATUS_SUCCESS;
 742 }
 743
 744
 745 /* convert CPU-specific flags to generic server flags */
 746 static inline unsigned int get_server_context_flags( DWORD flags )
 747 {
 748     unsigned int ret = 0;
 749
 750     flags &= 0x3f;  /* mask CPU id flags */
 751     if (flags & CONTEXT_CONTROL) ret |= SERVER_CTX_CONTROL;
 752     if (flags & CONTEXT_INTEGER) ret |= SERVER_CTX_INTEGER;
 753 #ifdef CONTEXT_SEGMENTS
 754     if (flags & CONTEXT_SEGMENTS) ret |= SERVER_CTX_SEGMENTS;
 755 #endif
 756 #ifdef CONTEXT_FLOATING_POINT
 757     if (flags & CONTEXT_FLOATING_POINT) ret |= SERVER_CTX_FLOATING_POINT;
 758 #endif
 759 #ifdef CONTEXT_DEBUG_REGISTERS
 760     if (flags & CONTEXT_DEBUG_REGISTERS) ret |= SERVER_CTX_DEBUG_REGISTERS;
 761 #endif
 762 #ifdef CONTEXT_EXTENDED_REGISTERS
 763     if (flags & CONTEXT_EXTENDED_REGISTERS) ret |= SERVER_CTX_EXTENDED_REGISTERS;
 764 #endif
 765     return ret;
 766 }
 767
 768 /***********************************************************************
 769  *              NtGetContextThread  (NTDLL.@)
 770  *              ZwGetContextThread  (NTDLL.@)
 771  */
 772 NTSTATUS WINAPI NtGetContextThread( HANDLE handle, CONTEXT *context )
 773 {
 774     NTSTATUS ret;
 775     DWORD dummy, i;
 776     DWORD needed_flags = context->ContextFlags;
 777     BOOL self = (handle == GetCurrentThread());
 778
 779 #ifdef __i386__
 780     /* on i386 debug registers always require a server call */
 781     if (context->ContextFlags & (CONTEXT_DEBUG_REGISTERS & ~CONTEXT_i386)) self = FALSE;
 782 #endif
 783
 784     if (!self)
 785     {
 786         unsigned int server_flags = get_server_context_flags( context->ContextFlags );
 787         context_t server_context;
 788
 789         SERVER_START_REQ( get_thread_context )
 790         {
 791             req->handle  = wine_server_obj_handle( handle );
 792             req->flags   = server_flags;
 793             req->suspend = 0;
 794             wine_server_set_reply( req, &server_context, sizeof(server_context) );
 795             ret = wine_server_call( req );
 796             self = reply->self;
 797         }
 798         SERVER_END_REQ;
 799
 800         if (ret == STATUS_PENDING)
 801         {
 802             if (NtSuspendThread( handle, &dummy ) == STATUS_SUCCESS)
 803             {
 804                 for (i = 0; i < 100; i++)
 805                 {
 806                     SERVER_START_REQ( get_thread_context )
 807                     {
 808                         req->handle  = wine_server_obj_handle( handle );
 809                         req->flags   = server_flags;
 810                         req->suspend = 0;
 811                         wine_server_set_reply( req, &server_context, sizeof(server_context) );
 812                         ret = wine_server_call( req );
 813                     }
 814                     SERVER_END_REQ;
 815                     if (ret == STATUS_PENDING)
 816                     {
 817                         LARGE_INTEGER timeout;
 818                         timeout.QuadPart = -10000;
 819                         NtDelayExecution( FALSE, &timeout );
 820                     }
 821                     else break;
 822                 }
 823                 NtResumeThread( handle, &dummy );
 824             }
 825             if (ret == STATUS_PENDING) ret = STATUS_ACCESS_DENIED;
 826         }
 827         if (!ret) ret = context_from_server( context, &server_context );
 828         if (ret) return ret;
 829         needed_flags &= ~context->ContextFlags;
 830     }
 831
 832     if (self)
 833     {
 834         if (needed_flags)
 835         {
 836             CONTEXT ctx;
 837             RtlCaptureContext( &ctx );
 838             copy_context( context, &ctx, ctx.ContextFlags & needed_flags );
 839             context->ContextFlags |= ctx.ContextFlags & needed_flags;
 840         }
 841 #ifdef __i386__
 842         /* update the cached version of the debug registers */
 843         if (context->ContextFlags & (CONTEXT_DEBUG_REGISTERS & ~CONTEXT_i386))
 844         {
 845             ntdll_get_thread_data()->dr0 = context->Dr0;
 846             ntdll_get_thread_data()->dr1 = context->Dr1;
 847             ntdll_get_thread_data()->dr2 = context->Dr2;
 848             ntdll_get_thread_data()->dr3 = context->Dr3;
 849             ntdll_get_thread_data()->dr6 = context->Dr6;
 850             ntdll_get_thread_data()->dr7 = context->Dr7;
 851         }
 852 #endif
 853     }
 854     return STATUS_SUCCESS;
 855 }
 856
 857
 858 /******************************************************************************
 859  *              NtQueryInformationThread  (NTDLL.@)
 860  *              ZwQueryInformationThread  (NTDLL.@)
 861  */
 862 NTSTATUS WINAPI NtQueryInformationThread( HANDLE handle, THREADINFOCLASS class,
 863                                           void *data, ULONG length, ULONG *ret_len )
 864 {
 865     NTSTATUS status;
 866
 867     switch(class)
 868     {
 869     case ThreadBasicInformation:
 870         {
 871             THREAD_BASIC_INFORMATION info;
 872             const ULONG_PTR affinity_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
 873
 874             SERVER_START_REQ( get_thread_info )
 875             {
 876                 req->handle = wine_server_obj_handle( handle );
 877                 req->tid_in = 0;
 878                 if (!(status = wine_server_call( req )))
 879                 {
 880                     info.ExitStatus             = reply->exit_code;
 881                     info.TebBaseAddress         = wine_server_get_ptr( reply->teb );
 882                     info.ClientId.UniqueProcess = ULongToHandle(reply->pid);
 883                     info.ClientId.UniqueThread  = ULongToHandle(reply->tid);
 884                     info.AffinityMask           = reply->affinity & affinity_mask;
 885                     info.Priority               = reply->priority;
 886                     info.BasePriority           = reply->priority;  /* FIXME */
 887                 }
 888             }
 889             SERVER_END_REQ;
 890             if (status == STATUS_SUCCESS)
 891             {
 892                 if (data) memcpy( data, &info, min( length, sizeof(info) ));
 893                 if (ret_len) *ret_len = min( length, sizeof(info) );
 894             }
 895         }
 896         return status;
 897     case ThreadAffinityMask:
 898         {
 899             const ULONG_PTR affinity_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
 900             ULONG_PTR affinity = 0;
 901
 902             SERVER_START_REQ( get_thread_info )
 903             {
 904                 req->handle = wine_server_obj_handle( handle );
 905                 req->tid_in = 0;
 906                 if (!(status = wine_server_call( req )))
 907                     affinity = reply->affinity & affinity_mask;
 908             }
 909             SERVER_END_REQ;
 910             if (status == STATUS_SUCCESS)
 911             {
 912                 if (data) memcpy( data, &affinity, min( length, sizeof(affinity) ));
 913                 if (ret_len) *ret_len = min( length, sizeof(affinity) );
 914             }
 915         }
 916         return status;
 917     case ThreadTimes:
 918         {
 919             KERNEL_USER_TIMES   kusrt;
 920             /* We need to do a server call to get the creation time or exit time */
 921             /* This works on any thread */
 922             SERVER_START_REQ( get_thread_info )
 923             {
 924                 req->handle = wine_server_obj_handle( handle );
 925                 req->tid_in = 0;
 926                 status = wine_server_call( req );
 927                 if (status == STATUS_SUCCESS)
 928                 {
 929                     kusrt.CreateTime.QuadPart = reply->creation_time;
 930                     kusrt.ExitTime.QuadPart = reply->exit_time;
 931                 }
 932             }
 933             SERVER_END_REQ;
 934             if (status == STATUS_SUCCESS)
 935             {
 936                 /* We call times(2) for kernel time or user time */
 937                 /* We can only (portably) do this for the current thread */
 938                 if (handle == GetCurrentThread())
 939                 {
 940                     struct tms time_buf;
 941                     long clocks_per_sec = sysconf(_SC_CLK_TCK);
 942
 943                     times(&time_buf);
 944                     kusrt.KernelTime.QuadPart = (ULONGLONG)time_buf.tms_stime * 10000000 / clocks_per_sec;
 945                     kusrt.UserTime.QuadPart = (ULONGLONG)time_buf.tms_utime * 10000000 / clocks_per_sec;
 946                 }
 947                 else
 948                 {
 949                     static BOOL reported = FALSE;
 950
 951                     kusrt.KernelTime.QuadPart = 0;
 952                     kusrt.UserTime.QuadPart = 0;
 953                     if (reported)
 954                         TRACE("Cannot get kerneltime or usertime of other threads\n");
 955                     else
 956                     {
 957                         FIXME("Cannot get kerneltime or usertime of other threads\n");
 958                         reported = TRUE;
 959                     }
 960                 }
 961                 if (data) memcpy( data, &kusrt, min( length, sizeof(kusrt) ));
 962                 if (ret_len) *ret_len = min( length, sizeof(kusrt) );
 963             }
 964         }
 965         return status;
 966     case ThreadDescriptorTableEntry:
 967         {
 968 #ifdef __i386__
 969             THREAD_DESCRIPTOR_INFORMATION*      tdi = data;
 970             if (length < sizeof(*tdi))
 971                 status = STATUS_INFO_LENGTH_MISMATCH;
 972             else if (!(tdi->Selector & 4))  /* GDT selector */
 973             {
 974                 unsigned sel = tdi->Selector & ~3;  /* ignore RPL */
 975                 status = STATUS_SUCCESS;
 976                 if (!sel)  /* null selector */
 977                     memset( &tdi->Entry, 0, sizeof(tdi->Entry) );
 978                 else
 979                 {
 980                     tdi->Entry.BaseLow                   = 0;
 981                     tdi->Entry.HighWord.Bits.BaseMid     = 0;
 982                     tdi->Entry.HighWord.Bits.BaseHi      = 0;
 983                     tdi->Entry.LimitLow                  = 0xffff;
 984                     tdi->Entry.HighWord.Bits.LimitHi     = 0xf;
 985                     tdi->Entry.HighWord.Bits.Dpl         = 3;
 986                     tdi->Entry.HighWord.Bits.Sys         = 0;
 987                     tdi->Entry.HighWord.Bits.Pres        = 1;
 988                     tdi->Entry.HighWord.Bits.Granularity = 1;
 989                     tdi->Entry.HighWord.Bits.Default_Big = 1;
 990                     tdi->Entry.HighWord.Bits.Type        = 0x12;
 991                     /* it has to be one of the system GDT selectors */
 992                     if (sel != (wine_get_ds() & ~3) && sel != (wine_get_ss() & ~3))
 993                     {
 994                         if (sel == (wine_get_cs() & ~3))
 995                             tdi->Entry.HighWord.Bits.Type |= 8;  /* code segment */
 996                         else status = STATUS_ACCESS_DENIED;
 997                     }
 998                 }
 999             }
1000             else
1001             {
1002                 SERVER_START_REQ( get_selector_entry )
1003                 {
1004                     req->handle = wine_server_obj_handle( handle );
1005                     req->entry = tdi->Selector >> 3;
1006                     status = wine_server_call( req );
1007                     if (!status)
1008                     {
1009                         if (!(reply->flags & WINE_LDT_FLAGS_ALLOCATED))
1010                             status = STATUS_ACCESS_VIOLATION;
1011                         else
1012                         {
1013                             wine_ldt_set_base ( &tdi->Entry, (void *)reply->base );
1014                             wine_ldt_set_limit( &tdi->Entry, reply->limit );
1015                             wine_ldt_set_flags( &tdi->Entry, reply->flags );
1016                         }
1017                     }
1018                 }
1019                 SERVER_END_REQ;
1020             }
1021             if (status == STATUS_SUCCESS && ret_len)
1022                 /* yes, that's a bit strange, but it's the way it is */
1023                 *ret_len = sizeof(LDT_ENTRY);
1024 #else
1025             status = STATUS_NOT_IMPLEMENTED;
1026 #endif
1027             return status;
1028         }
1029     case ThreadAmILastThread:
1030         {
1031             SERVER_START_REQ(get_thread_info)
1032             {
1033                 req->handle = wine_server_obj_handle( handle );
1034                 req->tid_in = 0;
1035                 status = wine_server_call( req );
1036                 if (status == STATUS_SUCCESS)
1037                 {
1038                     BOOLEAN last = reply->last;
1039                     if (data) memcpy( data, &last, min( length, sizeof(last) ));
1040                     if (ret_len) *ret_len = min( length, sizeof(last) );
1041                 }
1042             }
1043             SERVER_END_REQ;
1044             return status;
1045         }
1046     case ThreadPriority:
1047     case ThreadBasePriority:
1048     case ThreadImpersonationToken:
1049     case ThreadEnableAlignmentFaultFixup:
1050     case ThreadEventPair_Reusable:
1051     case ThreadQuerySetWin32StartAddress:
1052     case ThreadZeroTlsCell:
1053     case ThreadPerformanceCount:
1054     case ThreadIdealProcessor:
1055     case ThreadPriorityBoost:
1056     case ThreadSetTlsArrayAddress:
1057     case ThreadIsIoPending:
1058     default:
1059         FIXME( "info class %d not supported yet\n", class );
1060         return STATUS_NOT_IMPLEMENTED;
1061     }
1062 }
1063
1064
1065 /******************************************************************************
1066  *              NtSetInformationThread  (NTDLL.@)
1067  *              ZwSetInformationThread  (NTDLL.@)
1068  */
1069 NTSTATUS WINAPI NtSetInformationThread( HANDLE handle, THREADINFOCLASS class,
1070                                         LPCVOID data, ULONG length )
1071 {
1072     NTSTATUS status;
1073     switch(class)
1074     {
1075     case ThreadZeroTlsCell:
1076         if (handle == GetCurrentThread())
1077         {
1078             LIST_ENTRY *entry;
1079             DWORD index;
1080
1081             if (length != sizeof(DWORD)) return STATUS_INVALID_PARAMETER;
1082             index = *(const DWORD *)data;
1083             if (index < TLS_MINIMUM_AVAILABLE)
1084             {
1085                 RtlAcquirePebLock();
1086                 for (entry = tls_links.Flink; entry != &tls_links; entry = entry->Flink)
1087                 {
1088                     TEB *teb = CONTAINING_RECORD(entry, TEB, TlsLinks);
1089                     teb->TlsSlots[index] = 0;
1090                 }
1091                 RtlReleasePebLock();
1092             }
1093             else
1094             {
1095                 index -= TLS_MINIMUM_AVAILABLE;
1096                 if (index >= 8 * sizeof(NtCurrentTeb()->Peb->TlsExpansionBitmapBits))
1097                     return STATUS_INVALID_PARAMETER;
1098                 RtlAcquirePebLock();
1099                 for (entry = tls_links.Flink; entry != &tls_links; entry = entry->Flink)
1100                 {
1101                     TEB *teb = CONTAINING_RECORD(entry, TEB, TlsLinks);
1102                     if (teb->TlsExpansionSlots) teb->TlsExpansionSlots[index] = 0;
1103                 }
1104                 RtlReleasePebLock();
1105             }
1106             return STATUS_SUCCESS;
1107         }
1108         FIXME( "ZeroTlsCell not supported on other threads\n" );
1109         return STATUS_NOT_IMPLEMENTED;
1110
1111     case ThreadImpersonationToken:
1112         {
1113             const HANDLE *phToken = data;
1114             if (length != sizeof(HANDLE)) return STATUS_INVALID_PARAMETER;
1115             TRACE("Setting ThreadImpersonationToken handle to %p\n", *phToken );
1116             SERVER_START_REQ( set_thread_info )
1117             {
1118                 req->handle   = wine_server_obj_handle( handle );
1119                 req->token    = wine_server_obj_handle( *phToken );
1120                 req->mask     = SET_THREAD_INFO_TOKEN;
1121                 status = wine_server_call( req );
1122             }
1123             SERVER_END_REQ;
1124         }
1125         return status;
1126     case ThreadBasePriority:
1127         {
1128             const DWORD *pprio = data;
1129             if (length != sizeof(DWORD)) return STATUS_INVALID_PARAMETER;
1130             SERVER_START_REQ( set_thread_info )
1131             {
1132                 req->handle   = wine_server_obj_handle( handle );
1133                 req->priority = *pprio;
1134                 req->mask     = SET_THREAD_INFO_PRIORITY;
1135                 status = wine_server_call( req );
1136             }
1137             SERVER_END_REQ;
1138         }
1139         return status;
1140     case ThreadAffinityMask:
1141         {
1142             const ULONG_PTR affinity_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
1143             ULONG_PTR req_aff;
1144
1145             if (length != sizeof(ULONG_PTR)) return STATUS_INVALID_PARAMETER;
1146             req_aff = *(const ULONG_PTR *)data;
1147             if ((ULONG)req_aff == ~0u) req_aff = affinity_mask;
1148             else if (req_aff & ~affinity_mask) return STATUS_INVALID_PARAMETER;
1149             else if (!req_aff) return STATUS_INVALID_PARAMETER;
1150             SERVER_START_REQ( set_thread_info )
1151             {
1152                 req->handle   = wine_server_obj_handle( handle );
1153                 req->affinity = req_aff;
1154                 req->mask     = SET_THREAD_INFO_AFFINITY;
1155                 status = wine_server_call( req );
1156             }
1157             SERVER_END_REQ;
1158         }
1159         return status;
1160     case ThreadHideFromDebugger:
1161         /* pretend the call succeeded to satisfy some code protectors */
1162         return STATUS_SUCCESS;
1163
1164     case ThreadBasicInformation:
1165     case ThreadTimes:
1166     case ThreadPriority:
1167     case ThreadDescriptorTableEntry:
1168     case ThreadEnableAlignmentFaultFixup:
1169     case ThreadEventPair_Reusable:
1170     case ThreadQuerySetWin32StartAddress:
1171     case ThreadPerformanceCount:
1172     case ThreadAmILastThread:
1173     case ThreadIdealProcessor:
1174     case ThreadPriorityBoost:
1175     case ThreadSetTlsArrayAddress:
1176     case ThreadIsIoPending:
1177     default:
1178         FIXME( "info class %d not supported yet\n", class );
1179         return STATUS_NOT_IMPLEMENTED;
1180     }
1181 }