4 * Copyright 1996, 2003 Alexandre Julliard
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.
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.
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
22 #include "wine/port.h"
26 #include <sys/types.h>
27 #ifdef HAVE_SYS_MMAN_H
30 #ifdef HAVE_SYS_TIMES_H
31 #include <sys/times.h>
34 #define NONAMELESSUNION
36 #define WIN32_NO_STATUS
38 #include "wine/library.h"
39 #include "wine/server.h"
40 #include "wine/debug.h"
41 #include "ntdll_misc.h"
43 #include "wine/exception.h"
45 WINE_DEFAULT_DEBUG_CHANNEL(thread
);
46 WINE_DECLARE_DEBUG_CHANNEL(relay
);
48 struct _KUSER_SHARED_DATA
*user_shared_data
= NULL
;
50 PUNHANDLED_EXCEPTION_FILTER unhandled_exception_filter
= NULL
;
52 /* info passed to a starting thread */
56 PRTL_THREAD_START_ROUTINE entry_point
;
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;
70 /***********************************************************************
73 * Copy a unicode string from the startup info.
75 static inline void get_unicode_string( UNICODE_STRING
*str
, WCHAR
**src
, WCHAR
**dst
, UINT 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;
86 /***********************************************************************
87 * init_user_process_params
89 * Fill the RTL_USER_PROCESS_PARAMETERS structure from the server.
91 static NTSTATUS
init_user_process_params( SIZE_T data_size
, HANDLE
*exe_file
)
95 SIZE_T info_size
, env_size
, size
, alloc_size
;
98 RTL_USER_PROCESS_PARAMETERS
*params
= NULL
;
100 if (!(info
= RtlAllocateHeap( GetProcessHeap(), 0, data_size
)))
101 return STATUS_NO_MEMORY
;
103 SERVER_START_REQ( get_startup_info
)
105 wine_server_set_reply( req
, info
, data_size
);
106 if (!(status
= wine_server_call( req
)))
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
);
115 if (status
!= STATUS_SUCCESS
) goto done
;
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
);
128 status
= NtAllocateVirtualMemory( NtCurrentProcess(), (void **)¶ms
, 0, &alloc_size
,
129 MEM_COMMIT
, PAGE_READWRITE
);
130 if (status
!= STATUS_SUCCESS
) goto done
;
132 NtCurrentTeb()->Peb
->ProcessParameters
= params
;
133 params
->AllocationSize
= alloc_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
;
152 src
= (WCHAR
*)(info
+ 1);
153 dst
= (WCHAR
*)(params
+ 1);
155 /* current directory needs more space */
156 get_unicode_string( ¶ms
->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
;
160 get_unicode_string( ¶ms
->DllPath
, &src
, &dst
, info
->dllpath_len
);
161 get_unicode_string( ¶ms
->ImagePathName
, &src
, &dst
, info
->imagepath_len
);
162 get_unicode_string( ¶ms
->CommandLine
, &src
, &dst
, info
->cmdline_len
);
163 get_unicode_string( ¶ms
->WindowTitle
, &src
, &dst
, info
->title_len
);
164 get_unicode_string( ¶ms
->Desktop
, &src
, &dst
, info
->desktop_len
);
165 get_unicode_string( ¶ms
->ShellInfo
, &src
, &dst
, info
->shellinfo_len
);
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
);
172 /* environment needs to be a separate memory block */
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
;
182 RtlFreeHeap( GetProcessHeap(), 0, info
);
186 /***********************************************************************
189 * Setup the initial thread.
191 * NOTES: The first allocated TEB on NT is at 0x7ffde000.
193 HANDLE
thread_init(void)
197 SIZE_T size
, info_size
;
200 struct ntdll_thread_data
*thread_data
;
201 static struct debug_info debug_info
; /* debug info for initial thread */
205 /* reserve space for shared user data */
207 addr
= (void *)0x7ffe0000;
209 NtAllocateVirtualMemory( NtCurrentProcess(), &addr
, 0, &size
, MEM_RESERVE
|MEM_COMMIT
, PAGE_READWRITE
);
210 user_shared_data
= addr
;
212 /* allocate and initialize the PEB */
216 NtAllocateVirtualMemory( NtCurrentProcess(), &addr
, 1, &size
,
217 MEM_COMMIT
| MEM_TOP_DOWN
, PAGE_READWRITE
);
220 peb
->ProcessParameters
= ¶ms
;
221 peb
->TlsBitmap
= &tls_bitmap
;
222 peb
->TlsExpansionBitmap
= &tls_expansion_bitmap
;
223 peb
->FlsBitmap
= &fls_bitmap
;
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
);
238 /* allocate and initialize the initial TEB */
240 signal_alloc_thread( &teb
);
242 teb
->Tib
.StackBase
= (void *)~0UL;
243 teb
->StaticUnicodeString
.Buffer
= teb
->StaticUnicodeBuffer
;
244 teb
->StaticUnicodeString
.MaximumLength
= sizeof(teb
->StaticUnicodeBuffer
);
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
);
254 signal_init_thread( teb
);
255 virtual_init_threading();
257 debug_info
.str_pos
= debug_info
.strings
;
258 debug_info
.out_pos
= debug_info
.output
;
261 /* setup the server connection */
262 server_init_process();
263 info_size
= server_init_thread( peb
);
265 /* create the process heap */
266 if (!(peb
->ProcessHeap
= RtlCreateHeap( HEAP_GROWABLE
, NULL
, 0, 0, NULL
, NULL
)))
268 MESSAGE( "wine: failed to create the process heap\n" );
272 /* allocate user parameters */
275 init_user_process_params( info_size
, &exe_file
);
279 if (isatty(0) || isatty(1) || isatty(2))
280 params
.ConsoleHandle
= (HANDLE
)2; /* see kernel32/kernel_private.h */
282 wine_server_fd_to_handle( 0, GENERIC_READ
|SYNCHRONIZE
, OBJ_INHERIT
, ¶ms
.hStdInput
);
284 wine_server_fd_to_handle( 1, GENERIC_WRITE
|SYNCHRONIZE
, OBJ_INHERIT
, ¶ms
.hStdOutput
);
286 wine_server_fd_to_handle( 2, GENERIC_WRITE
|SYNCHRONIZE
, OBJ_INHERIT
, ¶ms
.hStdError
);
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;
305 /***********************************************************************
308 void terminate_thread( int status
)
310 pthread_sigmask( SIG_BLOCK
, &server_block_set
, NULL
);
311 if (interlocked_xchg_add( &nb_threads
, -1 ) <= 1) _exit( status
);
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
) );
321 /***********************************************************************
324 void exit_thread( int status
)
326 static void *prev_teb
;
329 if (status
) /* send the exit code to the server (0 is already the default) */
331 SERVER_START_REQ( terminate_thread
)
333 req
->handle
= wine_server_obj_handle( GetCurrentThread() );
334 req
->exit_code
= status
;
335 wine_server_call( req
);
340 if (interlocked_xchg_add( &nb_threads
, -1 ) <= 1)
342 LdrShutdownProcess();
348 RemoveEntryList( &NtCurrentTeb()->TlsLinks
);
350 RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->FlsSlots
);
351 RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->TlsExpansionSlots
);
353 pthread_sigmask( SIG_BLOCK
, &server_block_set
, NULL
);
355 if ((teb
= interlocked_xchg_ptr( &prev_teb
, NtCurrentTeb() )))
357 struct ntdll_thread_data
*thread_data
= (struct ntdll_thread_data
*)teb
->SpareBytes1
;
359 pthread_join( thread_data
->pthread_id
, NULL
);
360 signal_free_thread( teb
);
363 close( ntdll_get_thread_data()->wait_fd
[0] );
364 close( ntdll_get_thread_data()->wait_fd
[1] );
365 close( ntdll_get_thread_data()->reply_fd
);
366 close( ntdll_get_thread_data()->request_fd
);
367 pthread_exit( UIntToPtr(status
) );
371 /***********************************************************************
374 * Startup routine for a newly created thread.
376 static void start_thread( struct startup_info
*info
)
378 TEB
*teb
= info
->teb
;
379 struct ntdll_thread_data
*thread_data
= (struct ntdll_thread_data
*)teb
->SpareBytes1
;
380 PRTL_THREAD_START_ROUTINE func
= info
->entry_point
;
381 void *arg
= info
->entry_arg
;
382 struct debug_info debug_info
;
384 debug_info
.str_pos
= debug_info
.strings
;
385 debug_info
.out_pos
= debug_info
.output
;
386 thread_data
->debug_info
= &debug_info
;
387 thread_data
->pthread_id
= pthread_self();
389 signal_init_thread( teb
);
390 server_init_thread( func
);
391 pthread_sigmask( SIG_UNBLOCK
, &server_block_set
, NULL
);
394 InsertHeadList( &tls_links
, &teb
->TlsLinks
);
397 MODULE_DllThreadAttach( NULL
);
400 DPRINTF( "%04x:Starting thread proc %p (arg=%p)\n", GetCurrentThreadId(), func
, arg
);
402 call_thread_entry_point( (LPTHREAD_START_ROUTINE
)func
, arg
);
406 /***********************************************************************
407 * RtlCreateUserThread (NTDLL.@)
409 NTSTATUS WINAPI
RtlCreateUserThread( HANDLE process
, const SECURITY_DESCRIPTOR
*descr
,
410 BOOLEAN suspended
, PVOID stack_addr
,
411 SIZE_T stack_reserve
, SIZE_T stack_commit
,
412 PRTL_THREAD_START_ROUTINE start
, void *param
,
413 HANDLE
*handle_ptr
, CLIENT_ID
*id
)
416 pthread_t pthread_id
;
418 struct ntdll_thread_data
*thread_data
;
419 struct startup_info
*info
= NULL
;
426 if (process
!= NtCurrentProcess())
431 memset( &call
, 0, sizeof(call
) );
433 call
.create_thread
.type
= APC_CREATE_THREAD
;
434 call
.create_thread
.func
= wine_server_client_ptr( start
);
435 call
.create_thread
.arg
= wine_server_client_ptr( param
);
436 call
.create_thread
.reserve
= stack_reserve
;
437 call
.create_thread
.commit
= stack_commit
;
438 call
.create_thread
.suspend
= suspended
;
439 status
= NTDLL_queue_process_apc( process
, &call
, &result
);
440 if (status
!= STATUS_SUCCESS
) return status
;
442 if (result
.create_thread
.status
== STATUS_SUCCESS
)
444 if (id
) id
->UniqueThread
= ULongToHandle(result
.create_thread
.tid
);
445 if (handle_ptr
) *handle_ptr
= wine_server_ptr_handle( result
.create_thread
.handle
);
446 else NtClose( wine_server_ptr_handle( result
.create_thread
.handle
));
448 return result
.create_thread
.status
;
451 if (server_pipe( request_pipe
) == -1) return STATUS_TOO_MANY_OPENED_FILES
;
452 wine_server_send_fd( request_pipe
[0] );
454 SERVER_START_REQ( new_thread
)
456 req
->access
= THREAD_ALL_ACCESS
;
457 req
->attributes
= 0; /* FIXME */
458 req
->suspend
= suspended
;
459 req
->request_fd
= request_pipe
[0];
460 if (!(status
= wine_server_call( req
)))
462 handle
= wine_server_ptr_handle( reply
->handle
);
465 close( request_pipe
[0] );
471 close( request_pipe
[1] );
475 pthread_sigmask( SIG_BLOCK
, &server_block_set
, &sigset
);
477 if ((status
= signal_alloc_thread( &teb
))) goto error
;
479 teb
->Peb
= NtCurrentTeb()->Peb
;
480 teb
->ClientId
.UniqueProcess
= ULongToHandle(GetCurrentProcessId());
481 teb
->ClientId
.UniqueThread
= ULongToHandle(tid
);
482 teb
->StaticUnicodeString
.Buffer
= teb
->StaticUnicodeBuffer
;
483 teb
->StaticUnicodeString
.MaximumLength
= sizeof(teb
->StaticUnicodeBuffer
);
485 info
= (struct startup_info
*)(teb
+ 1);
487 info
->entry_point
= start
;
488 info
->entry_arg
= param
;
490 thread_data
= (struct ntdll_thread_data
*)teb
->SpareBytes1
;
491 thread_data
->request_fd
= request_pipe
[1];
492 thread_data
->reply_fd
= -1;
493 thread_data
->wait_fd
[0] = -1;
494 thread_data
->wait_fd
[1] = -1;
496 if ((status
= virtual_alloc_thread_stack( teb
, stack_reserve
, stack_commit
))) goto error
;
498 pthread_attr_init( &attr
);
499 pthread_attr_setstack( &attr
, teb
->DeallocationStack
,
500 (char *)teb
->Tib
.StackBase
- (char *)teb
->DeallocationStack
);
501 pthread_attr_setscope( &attr
, PTHREAD_SCOPE_SYSTEM
); /* force creating a kernel thread */
502 interlocked_xchg_add( &nb_threads
, 1 );
503 if (pthread_create( &pthread_id
, &attr
, (void * (*)(void *))start_thread
, info
))
505 interlocked_xchg_add( &nb_threads
, -1 );
506 pthread_attr_destroy( &attr
);
507 status
= STATUS_NO_MEMORY
;
510 pthread_attr_destroy( &attr
);
511 pthread_sigmask( SIG_SETMASK
, &sigset
, NULL
);
513 if (id
) id
->UniqueThread
= ULongToHandle(tid
);
514 if (handle_ptr
) *handle_ptr
= handle
;
515 else NtClose( handle
);
517 return STATUS_SUCCESS
;
520 if (teb
) signal_free_thread( teb
);
521 if (handle
) NtClose( handle
);
522 pthread_sigmask( SIG_SETMASK
, &sigset
, NULL
);
523 close( request_pipe
[1] );
528 /******************************************************************************
529 * RtlGetNtGlobalFlags (NTDLL.@)
531 ULONG WINAPI
RtlGetNtGlobalFlags(void)
533 if (!peb
) return 0; /* init not done yet */
534 return peb
->NtGlobalFlag
;
538 /***********************************************************************
539 * NtOpenThread (NTDLL.@)
540 * ZwOpenThread (NTDLL.@)
542 NTSTATUS WINAPI
NtOpenThread( HANDLE
*handle
, ACCESS_MASK access
,
543 const OBJECT_ATTRIBUTES
*attr
, const CLIENT_ID
*id
)
547 SERVER_START_REQ( open_thread
)
549 req
->tid
= HandleToULong(id
->UniqueThread
);
550 req
->access
= access
;
551 req
->attributes
= attr
? attr
->Attributes
: 0;
552 ret
= wine_server_call( req
);
553 *handle
= wine_server_ptr_handle( reply
->handle
);
560 /******************************************************************************
561 * NtSuspendThread (NTDLL.@)
562 * ZwSuspendThread (NTDLL.@)
564 NTSTATUS WINAPI
NtSuspendThread( HANDLE handle
, PULONG count
)
568 SERVER_START_REQ( suspend_thread
)
570 req
->handle
= wine_server_obj_handle( handle
);
571 if (!(ret
= wine_server_call( req
))) *count
= reply
->count
;
578 /******************************************************************************
579 * NtResumeThread (NTDLL.@)
580 * ZwResumeThread (NTDLL.@)
582 NTSTATUS WINAPI
NtResumeThread( HANDLE handle
, PULONG count
)
586 SERVER_START_REQ( resume_thread
)
588 req
->handle
= wine_server_obj_handle( handle
);
589 if (!(ret
= wine_server_call( req
))) *count
= reply
->count
;
596 /******************************************************************************
597 * NtAlertResumeThread (NTDLL.@)
598 * ZwAlertResumeThread (NTDLL.@)
600 NTSTATUS WINAPI
NtAlertResumeThread( HANDLE handle
, PULONG count
)
602 FIXME( "stub: should alert thread %p\n", handle
);
603 return NtResumeThread( handle
, count
);
607 /******************************************************************************
608 * NtAlertThread (NTDLL.@)
609 * ZwAlertThread (NTDLL.@)
611 NTSTATUS WINAPI
NtAlertThread( HANDLE handle
)
613 FIXME( "stub: %p\n", handle
);
614 return STATUS_NOT_IMPLEMENTED
;
618 /******************************************************************************
619 * NtTerminateThread (NTDLL.@)
620 * ZwTerminateThread (NTDLL.@)
622 NTSTATUS WINAPI
NtTerminateThread( HANDLE handle
, LONG exit_code
)
627 SERVER_START_REQ( terminate_thread
)
629 req
->handle
= wine_server_obj_handle( handle
);
630 req
->exit_code
= exit_code
;
631 ret
= wine_server_call( req
);
632 self
= !ret
&& reply
->self
;
636 if (self
) abort_thread( exit_code
);
641 /******************************************************************************
642 * NtQueueApcThread (NTDLL.@)
644 NTSTATUS WINAPI
NtQueueApcThread( HANDLE handle
, PNTAPCFUNC func
, ULONG_PTR arg1
,
645 ULONG_PTR arg2
, ULONG_PTR arg3
)
648 SERVER_START_REQ( queue_apc
)
650 req
->handle
= wine_server_obj_handle( handle
);
653 req
->call
.type
= APC_USER
;
654 req
->call
.user
.func
= wine_server_client_ptr( func
);
655 req
->call
.user
.args
[0] = arg1
;
656 req
->call
.user
.args
[1] = arg2
;
657 req
->call
.user
.args
[2] = arg3
;
659 else req
->call
.type
= APC_NONE
; /* wake up only */
660 ret
= wine_server_call( req
);
667 /***********************************************************************
668 * NtSetContextThread (NTDLL.@)
669 * ZwSetContextThread (NTDLL.@)
671 NTSTATUS WINAPI
NtSetContextThread( HANDLE handle
, const CONTEXT
*context
)
678 /* on i386 debug registers always require a server call */
679 self
= (handle
== GetCurrentThread());
680 if (self
&& (context
->ContextFlags
& (CONTEXT_DEBUG_REGISTERS
& ~CONTEXT_i386
)))
682 self
= (ntdll_get_thread_data()->dr0
== context
->Dr0
&&
683 ntdll_get_thread_data()->dr1
== context
->Dr1
&&
684 ntdll_get_thread_data()->dr2
== context
->Dr2
&&
685 ntdll_get_thread_data()->dr3
== context
->Dr3
&&
686 ntdll_get_thread_data()->dr6
== context
->Dr6
&&
687 ntdll_get_thread_data()->dr7
== context
->Dr7
);
693 context_t server_context
;
695 context_to_server( &server_context
, context
);
697 SERVER_START_REQ( set_thread_context
)
699 req
->handle
= wine_server_obj_handle( handle
);
701 wine_server_add_data( req
, &server_context
, sizeof(server_context
) );
702 ret
= wine_server_call( req
);
707 if (ret
== STATUS_PENDING
)
709 if (NtSuspendThread( handle
, &dummy
) == STATUS_SUCCESS
)
711 for (i
= 0; i
< 100; i
++)
713 SERVER_START_REQ( set_thread_context
)
715 req
->handle
= wine_server_obj_handle( handle
);
717 wine_server_add_data( req
, &server_context
, sizeof(server_context
) );
718 ret
= wine_server_call( req
);
721 if (ret
== STATUS_PENDING
)
723 LARGE_INTEGER timeout
;
724 timeout
.QuadPart
= -10000;
725 NtDelayExecution( FALSE
, &timeout
);
729 NtResumeThread( handle
, &dummy
);
731 if (ret
== STATUS_PENDING
) ret
= STATUS_ACCESS_DENIED
;
737 if (self
) set_cpu_context( context
);
738 return STATUS_SUCCESS
;
742 /* convert CPU-specific flags to generic server flags */
743 static inline unsigned int get_server_context_flags( DWORD flags
)
745 unsigned int ret
= 0;
747 flags
&= 0x3f; /* mask CPU id flags */
748 if (flags
& CONTEXT_CONTROL
) ret
|= SERVER_CTX_CONTROL
;
749 if (flags
& CONTEXT_INTEGER
) ret
|= SERVER_CTX_INTEGER
;
750 #ifdef CONTEXT_SEGMENTS
751 if (flags
& CONTEXT_SEGMENTS
) ret
|= SERVER_CTX_SEGMENTS
;
753 #ifdef CONTEXT_FLOATING_POINT
754 if (flags
& CONTEXT_FLOATING_POINT
) ret
|= SERVER_CTX_FLOATING_POINT
;
756 #ifdef CONTEXT_DEBUG_REGISTERS
757 if (flags
& CONTEXT_DEBUG_REGISTERS
) ret
|= SERVER_CTX_DEBUG_REGISTERS
;
759 #ifdef CONTEXT_EXTENDED_REGISTERS
760 if (flags
& CONTEXT_EXTENDED_REGISTERS
) ret
|= SERVER_CTX_EXTENDED_REGISTERS
;
765 /***********************************************************************
766 * NtGetContextThread (NTDLL.@)
767 * ZwGetContextThread (NTDLL.@)
769 NTSTATUS WINAPI
NtGetContextThread( HANDLE handle
, CONTEXT
*context
)
773 DWORD needed_flags
= context
->ContextFlags
;
774 BOOL self
= (handle
== GetCurrentThread());
777 /* on i386 debug registers always require a server call */
778 if (context
->ContextFlags
& (CONTEXT_DEBUG_REGISTERS
& ~CONTEXT_i386
)) self
= FALSE
;
783 unsigned int server_flags
= get_server_context_flags( context
->ContextFlags
);
784 context_t server_context
;
786 SERVER_START_REQ( get_thread_context
)
788 req
->handle
= wine_server_obj_handle( handle
);
789 req
->flags
= server_flags
;
791 wine_server_set_reply( req
, &server_context
, sizeof(server_context
) );
792 ret
= wine_server_call( req
);
797 if (ret
== STATUS_PENDING
)
799 if (NtSuspendThread( handle
, &dummy
) == STATUS_SUCCESS
)
801 for (i
= 0; i
< 100; i
++)
803 SERVER_START_REQ( get_thread_context
)
805 req
->handle
= wine_server_obj_handle( handle
);
806 req
->flags
= server_flags
;
808 wine_server_set_reply( req
, &server_context
, sizeof(server_context
) );
809 ret
= wine_server_call( req
);
812 if (ret
== STATUS_PENDING
)
814 LARGE_INTEGER timeout
;
815 timeout
.QuadPart
= -10000;
816 NtDelayExecution( FALSE
, &timeout
);
820 NtResumeThread( handle
, &dummy
);
822 if (ret
== STATUS_PENDING
) ret
= STATUS_ACCESS_DENIED
;
824 if (!ret
) ret
= context_from_server( context
, &server_context
);
826 needed_flags
&= ~context
->ContextFlags
;
834 RtlCaptureContext( &ctx
);
835 copy_context( context
, &ctx
, ctx
.ContextFlags
& needed_flags
);
836 context
->ContextFlags
|= ctx
.ContextFlags
& needed_flags
;
839 /* update the cached version of the debug registers */
840 if (context
->ContextFlags
& (CONTEXT_DEBUG_REGISTERS
& ~CONTEXT_i386
))
842 ntdll_get_thread_data()->dr0
= context
->Dr0
;
843 ntdll_get_thread_data()->dr1
= context
->Dr1
;
844 ntdll_get_thread_data()->dr2
= context
->Dr2
;
845 ntdll_get_thread_data()->dr3
= context
->Dr3
;
846 ntdll_get_thread_data()->dr6
= context
->Dr6
;
847 ntdll_get_thread_data()->dr7
= context
->Dr7
;
851 return STATUS_SUCCESS
;
855 /******************************************************************************
856 * NtQueryInformationThread (NTDLL.@)
857 * ZwQueryInformationThread (NTDLL.@)
859 NTSTATUS WINAPI
NtQueryInformationThread( HANDLE handle
, THREADINFOCLASS
class,
860 void *data
, ULONG length
, ULONG
*ret_len
)
866 case ThreadBasicInformation
:
868 THREAD_BASIC_INFORMATION info
;
869 const ULONG_PTR affinity_mask
= ((ULONG_PTR
)1 << NtCurrentTeb()->Peb
->NumberOfProcessors
) - 1;
871 SERVER_START_REQ( get_thread_info
)
873 req
->handle
= wine_server_obj_handle( handle
);
875 if (!(status
= wine_server_call( req
)))
877 info
.ExitStatus
= reply
->exit_code
;
878 info
.TebBaseAddress
= wine_server_get_ptr( reply
->teb
);
879 info
.ClientId
.UniqueProcess
= ULongToHandle(reply
->pid
);
880 info
.ClientId
.UniqueThread
= ULongToHandle(reply
->tid
);
881 info
.AffinityMask
= reply
->affinity
& affinity_mask
;
882 info
.Priority
= reply
->priority
;
883 info
.BasePriority
= reply
->priority
; /* FIXME */
887 if (status
== STATUS_SUCCESS
)
889 if (data
) memcpy( data
, &info
, min( length
, sizeof(info
) ));
890 if (ret_len
) *ret_len
= min( length
, sizeof(info
) );
894 case ThreadAffinityMask
:
896 const ULONG_PTR affinity_mask
= ((ULONG_PTR
)1 << NtCurrentTeb()->Peb
->NumberOfProcessors
) - 1;
897 ULONG_PTR affinity
= 0;
899 SERVER_START_REQ( get_thread_info
)
901 req
->handle
= wine_server_obj_handle( handle
);
903 if (!(status
= wine_server_call( req
)))
904 affinity
= reply
->affinity
& affinity_mask
;
907 if (status
== STATUS_SUCCESS
)
909 if (data
) memcpy( data
, &affinity
, min( length
, sizeof(affinity
) ));
910 if (ret_len
) *ret_len
= min( length
, sizeof(affinity
) );
916 KERNEL_USER_TIMES kusrt
;
917 /* We need to do a server call to get the creation time or exit time */
918 /* This works on any thread */
919 SERVER_START_REQ( get_thread_info
)
921 req
->handle
= wine_server_obj_handle( handle
);
923 status
= wine_server_call( req
);
924 if (status
== STATUS_SUCCESS
)
926 kusrt
.CreateTime
.QuadPart
= reply
->creation_time
;
927 kusrt
.ExitTime
.QuadPart
= reply
->exit_time
;
931 if (status
== STATUS_SUCCESS
)
933 /* We call times(2) for kernel time or user time */
934 /* We can only (portably) do this for the current thread */
935 if (handle
== GetCurrentThread())
938 long clocks_per_sec
= sysconf(_SC_CLK_TCK
);
941 kusrt
.KernelTime
.QuadPart
= (ULONGLONG
)time_buf
.tms_stime
* 10000000 / clocks_per_sec
;
942 kusrt
.UserTime
.QuadPart
= (ULONGLONG
)time_buf
.tms_utime
* 10000000 / clocks_per_sec
;
946 static BOOL reported
= FALSE
;
948 kusrt
.KernelTime
.QuadPart
= 0;
949 kusrt
.UserTime
.QuadPart
= 0;
951 TRACE("Cannot get kerneltime or usertime of other threads\n");
954 FIXME("Cannot get kerneltime or usertime of other threads\n");
958 if (data
) memcpy( data
, &kusrt
, min( length
, sizeof(kusrt
) ));
959 if (ret_len
) *ret_len
= min( length
, sizeof(kusrt
) );
963 case ThreadDescriptorTableEntry
:
966 THREAD_DESCRIPTOR_INFORMATION
* tdi
= data
;
967 if (length
< sizeof(*tdi
))
968 status
= STATUS_INFO_LENGTH_MISMATCH
;
969 else if (!(tdi
->Selector
& 4)) /* GDT selector */
971 unsigned sel
= tdi
->Selector
& ~3; /* ignore RPL */
972 status
= STATUS_SUCCESS
;
973 if (!sel
) /* null selector */
974 memset( &tdi
->Entry
, 0, sizeof(tdi
->Entry
) );
977 tdi
->Entry
.BaseLow
= 0;
978 tdi
->Entry
.HighWord
.Bits
.BaseMid
= 0;
979 tdi
->Entry
.HighWord
.Bits
.BaseHi
= 0;
980 tdi
->Entry
.LimitLow
= 0xffff;
981 tdi
->Entry
.HighWord
.Bits
.LimitHi
= 0xf;
982 tdi
->Entry
.HighWord
.Bits
.Dpl
= 3;
983 tdi
->Entry
.HighWord
.Bits
.Sys
= 0;
984 tdi
->Entry
.HighWord
.Bits
.Pres
= 1;
985 tdi
->Entry
.HighWord
.Bits
.Granularity
= 1;
986 tdi
->Entry
.HighWord
.Bits
.Default_Big
= 1;
987 tdi
->Entry
.HighWord
.Bits
.Type
= 0x12;
988 /* it has to be one of the system GDT selectors */
989 if (sel
!= (wine_get_ds() & ~3) && sel
!= (wine_get_ss() & ~3))
991 if (sel
== (wine_get_cs() & ~3))
992 tdi
->Entry
.HighWord
.Bits
.Type
|= 8; /* code segment */
993 else status
= STATUS_ACCESS_DENIED
;
999 SERVER_START_REQ( get_selector_entry
)
1001 req
->handle
= wine_server_obj_handle( handle
);
1002 req
->entry
= tdi
->Selector
>> 3;
1003 status
= wine_server_call( req
);
1006 if (!(reply
->flags
& WINE_LDT_FLAGS_ALLOCATED
))
1007 status
= STATUS_ACCESS_VIOLATION
;
1010 wine_ldt_set_base ( &tdi
->Entry
, (void *)reply
->base
);
1011 wine_ldt_set_limit( &tdi
->Entry
, reply
->limit
);
1012 wine_ldt_set_flags( &tdi
->Entry
, reply
->flags
);
1018 if (status
== STATUS_SUCCESS
&& ret_len
)
1019 /* yes, that's a bit strange, but it's the way it is */
1020 *ret_len
= sizeof(LDT_ENTRY
);
1022 status
= STATUS_NOT_IMPLEMENTED
;
1026 case ThreadAmILastThread
:
1028 SERVER_START_REQ(get_thread_info
)
1030 req
->handle
= wine_server_obj_handle( handle
);
1032 status
= wine_server_call( req
);
1033 if (status
== STATUS_SUCCESS
)
1035 BOOLEAN last
= reply
->last
;
1036 if (data
) memcpy( data
, &last
, min( length
, sizeof(last
) ));
1037 if (ret_len
) *ret_len
= min( length
, sizeof(last
) );
1043 case ThreadPriority
:
1044 case ThreadBasePriority
:
1045 case ThreadImpersonationToken
:
1046 case ThreadEnableAlignmentFaultFixup
:
1047 case ThreadEventPair_Reusable
:
1048 case ThreadQuerySetWin32StartAddress
:
1049 case ThreadZeroTlsCell
:
1050 case ThreadPerformanceCount
:
1051 case ThreadIdealProcessor
:
1052 case ThreadPriorityBoost
:
1053 case ThreadSetTlsArrayAddress
:
1054 case ThreadIsIoPending
:
1056 FIXME( "info class %d not supported yet\n", class );
1057 return STATUS_NOT_IMPLEMENTED
;
1062 /******************************************************************************
1063 * NtSetInformationThread (NTDLL.@)
1064 * ZwSetInformationThread (NTDLL.@)
1066 NTSTATUS WINAPI
NtSetInformationThread( HANDLE handle
, THREADINFOCLASS
class,
1067 LPCVOID data
, ULONG length
)
1072 case ThreadZeroTlsCell
:
1073 if (handle
== GetCurrentThread())
1078 if (length
!= sizeof(DWORD
)) return STATUS_INVALID_PARAMETER
;
1079 index
= *(const DWORD
*)data
;
1080 if (index
< TLS_MINIMUM_AVAILABLE
)
1082 RtlAcquirePebLock();
1083 for (entry
= tls_links
.Flink
; entry
!= &tls_links
; entry
= entry
->Flink
)
1085 TEB
*teb
= CONTAINING_RECORD(entry
, TEB
, TlsLinks
);
1086 teb
->TlsSlots
[index
] = 0;
1088 RtlReleasePebLock();
1092 index
-= TLS_MINIMUM_AVAILABLE
;
1093 if (index
>= 8 * sizeof(NtCurrentTeb()->Peb
->TlsExpansionBitmapBits
))
1094 return STATUS_INVALID_PARAMETER
;
1095 RtlAcquirePebLock();
1096 for (entry
= tls_links
.Flink
; entry
!= &tls_links
; entry
= entry
->Flink
)
1098 TEB
*teb
= CONTAINING_RECORD(entry
, TEB
, TlsLinks
);
1099 if (teb
->TlsExpansionSlots
) teb
->TlsExpansionSlots
[index
] = 0;
1101 RtlReleasePebLock();
1103 return STATUS_SUCCESS
;
1105 FIXME( "ZeroTlsCell not supported on other threads\n" );
1106 return STATUS_NOT_IMPLEMENTED
;
1108 case ThreadImpersonationToken
:
1110 const HANDLE
*phToken
= data
;
1111 if (length
!= sizeof(HANDLE
)) return STATUS_INVALID_PARAMETER
;
1112 TRACE("Setting ThreadImpersonationToken handle to %p\n", *phToken
);
1113 SERVER_START_REQ( set_thread_info
)
1115 req
->handle
= wine_server_obj_handle( handle
);
1116 req
->token
= wine_server_obj_handle( *phToken
);
1117 req
->mask
= SET_THREAD_INFO_TOKEN
;
1118 status
= wine_server_call( req
);
1123 case ThreadBasePriority
:
1125 const DWORD
*pprio
= data
;
1126 if (length
!= sizeof(DWORD
)) return STATUS_INVALID_PARAMETER
;
1127 SERVER_START_REQ( set_thread_info
)
1129 req
->handle
= wine_server_obj_handle( handle
);
1130 req
->priority
= *pprio
;
1131 req
->mask
= SET_THREAD_INFO_PRIORITY
;
1132 status
= wine_server_call( req
);
1137 case ThreadAffinityMask
:
1139 const ULONG_PTR affinity_mask
= ((ULONG_PTR
)1 << NtCurrentTeb()->Peb
->NumberOfProcessors
) - 1;
1142 if (length
!= sizeof(ULONG_PTR
)) return STATUS_INVALID_PARAMETER
;
1143 req_aff
= *(const ULONG_PTR
*)data
;
1144 if ((ULONG
)req_aff
== ~0u) req_aff
= affinity_mask
;
1145 else if (req_aff
& ~affinity_mask
) return STATUS_INVALID_PARAMETER
;
1146 else if (!req_aff
) return STATUS_INVALID_PARAMETER
;
1147 SERVER_START_REQ( set_thread_info
)
1149 req
->handle
= wine_server_obj_handle( handle
);
1150 req
->affinity
= req_aff
;
1151 req
->mask
= SET_THREAD_INFO_AFFINITY
;
1152 status
= wine_server_call( req
);
1157 case ThreadHideFromDebugger
:
1158 /* pretend the call succeeded to satisfy some code protectors */
1159 return STATUS_SUCCESS
;
1161 case ThreadBasicInformation
:
1163 case ThreadPriority
:
1164 case ThreadDescriptorTableEntry
:
1165 case ThreadEnableAlignmentFaultFixup
:
1166 case ThreadEventPair_Reusable
:
1167 case ThreadQuerySetWin32StartAddress
:
1168 case ThreadPerformanceCount
:
1169 case ThreadAmILastThread
:
1170 case ThreadIdealProcessor
:
1171 case ThreadPriorityBoost
:
1172 case ThreadSetTlsArrayAddress
:
1173 case ThreadIsIoPending
:
1175 FIXME( "info class %d not supported yet\n", class );
1176 return STATUS_NOT_IMPLEMENTED
;