?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
wined3d: Explicitly calculate the sub-resource level in wined3d_surface_upload_data().
[wine.git] / dlls / ntdll / thread.c
blob9ec94d678010f76cd7a34a0cebc713a2b03c3c0d
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 <limits.h>
27 #include <sys/types.h>
28 #ifdef HAVE_SYS_MMAN_H
29 #include <sys/mman.h>
30 #endif
31 #ifdef HAVE_SYS_TIMES_H
32 #include <sys/times.h>
33 #endif
34 #ifdef HAVE_SYS_SYSCALL_H
35 #include <sys/syscall.h>
36 #endif
37
38 #define NONAMELESSUNION
39 #include "ntstatus.h"
40 #define WIN32_NO_STATUS
41 #include "winternl.h"
42 #include "wine/library.h"
43 #include "wine/server.h"
44 #include "wine/debug.h"
45 #include "ntdll_misc.h"
46 #include "ddk/wdm.h"
47 #include "wine/exception.h"
48
49 WINE_DEFAULT_DEBUG_CHANNEL(thread);
50
51 #ifndef PTHREAD_STACK_MIN
52 #define PTHREAD_STACK_MIN 16384
53 #endif
54
55 struct _KUSER_SHARED_DATA *user_shared_data = NULL;
56 static const WCHAR default_windirW[] = {'C',':','\\','w','i','n','d','o','w','s',0};
57
58 PUNHANDLED_EXCEPTION_FILTER unhandled_exception_filter = NULL;
59 void (WINAPI *kernel32_start_process)(LPTHREAD_START_ROUTINE,void*) = NULL;
60
61 /* info passed to a starting thread */
62 struct startup_info
63 {
64 TEB *teb;
65 PRTL_THREAD_START_ROUTINE entry_point;
66 void *entry_arg;
67 };
68
69 static PEB *peb;
70 static PEB_LDR_DATA ldr;
71 static RTL_USER_PROCESS_PARAMETERS params; /* default parameters if no parent */
72 static WCHAR current_dir[MAX_NT_PATH_LENGTH];
73 static RTL_BITMAP tls_bitmap;
74 static RTL_BITMAP tls_expansion_bitmap;
75 static RTL_BITMAP fls_bitmap;
76 static int nb_threads = 1;
77
78 static RTL_CRITICAL_SECTION peb_lock;
79 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
80 {
81 0, 0, &peb_lock,
82 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
83 0, 0, { (DWORD_PTR)(__FILE__ ": peb_lock") }
84 };
85 static RTL_CRITICAL_SECTION peb_lock = { &critsect_debug, -1, 0, 0, 0, 0 };
86
87 /***********************************************************************
88 * get_unicode_string
89 *
90 * Copy a unicode string from the startup info.
91 */
92 static inline void get_unicode_string( UNICODE_STRING *str, WCHAR **src, WCHAR **dst, UINT len )
93 {
94 str->Buffer = *dst;
95 str->Length = len;
96 str->MaximumLength = len + sizeof(WCHAR);
97 memcpy( str->Buffer, *src, len );
98 str->Buffer[len / sizeof(WCHAR)] = 0;
99 *src += len / sizeof(WCHAR);
100 *dst += len / sizeof(WCHAR) + 1;
101 }
102
103 /***********************************************************************
104 * init_user_process_params
105 *
106 * Fill the RTL_USER_PROCESS_PARAMETERS structure from the server.
107 */
108 static NTSTATUS init_user_process_params( SIZE_T data_size, HANDLE *exe_file )
109 {
110 void *ptr;
111 WCHAR *src, *dst;
112 SIZE_T info_size, env_size, size, alloc_size;
113 NTSTATUS status;
114 startup_info_t *info;
115 RTL_USER_PROCESS_PARAMETERS *params = NULL;
116
117 if (!(info = RtlAllocateHeap( GetProcessHeap(), 0, data_size )))
118 return STATUS_NO_MEMORY;
119
120 SERVER_START_REQ( get_startup_info )
121 {
122 wine_server_set_reply( req, info, data_size );
123 if (!(status = wine_server_call( req )))
124 {
125 data_size = wine_server_reply_size( reply );
126 info_size = reply->info_size;
127 env_size = data_size - info_size;
128 *exe_file = wine_server_ptr_handle( reply->exe_file );
129 }
130 }
131 SERVER_END_REQ;
132 if (status != STATUS_SUCCESS) goto done;
133
134 size = sizeof(*params);
135 size += MAX_NT_PATH_LENGTH * sizeof(WCHAR);
136 size += info->dllpath_len + sizeof(WCHAR);
137 size += info->imagepath_len + sizeof(WCHAR);
138 size += info->cmdline_len + sizeof(WCHAR);
139 size += info->title_len + sizeof(WCHAR);
140 size += info->desktop_len + sizeof(WCHAR);
141 size += info->shellinfo_len + sizeof(WCHAR);
142 size += info->runtime_len + sizeof(WCHAR);
143
144 alloc_size = size;
145 status = NtAllocateVirtualMemory( NtCurrentProcess(), (void **)&params, 0, &alloc_size,
146 MEM_COMMIT, PAGE_READWRITE );
147 if (status != STATUS_SUCCESS) goto done;
148
149 NtCurrentTeb()->Peb->ProcessParameters = params;
150 params->AllocationSize = alloc_size;
151 params->Size = size;
152 params->Flags = PROCESS_PARAMS_FLAG_NORMALIZED;
153 params->DebugFlags = info->debug_flags;
154 params->ConsoleHandle = wine_server_ptr_handle( info->console );
155 params->ConsoleFlags = info->console_flags;
156 params->hStdInput = wine_server_ptr_handle( info->hstdin );
157 params->hStdOutput = wine_server_ptr_handle( info->hstdout );
158 params->hStdError = wine_server_ptr_handle( info->hstderr );
159 params->dwX = info->x;
160 params->dwY = info->y;
161 params->dwXSize = info->xsize;
162 params->dwYSize = info->ysize;
163 params->dwXCountChars = info->xchars;
164 params->dwYCountChars = info->ychars;
165 params->dwFillAttribute = info->attribute;
166 params->dwFlags = info->flags;
167 params->wShowWindow = info->show;
168
169 src = (WCHAR *)(info + 1);
170 dst = (WCHAR *)(params + 1);
171
172 /* current directory needs more space */
173 get_unicode_string( &params->CurrentDirectory.DosPath, &src, &dst, info->curdir_len );
174 params->CurrentDirectory.DosPath.MaximumLength = MAX_NT_PATH_LENGTH * sizeof(WCHAR);
175 dst = (WCHAR *)(params + 1) + MAX_NT_PATH_LENGTH;
176
177 get_unicode_string( &params->DllPath, &src, &dst, info->dllpath_len );
178 get_unicode_string( &params->ImagePathName, &src, &dst, info->imagepath_len );
179 get_unicode_string( &params->CommandLine, &src, &dst, info->cmdline_len );
180 get_unicode_string( &params->WindowTitle, &src, &dst, info->title_len );
181 get_unicode_string( &params->Desktop, &src, &dst, info->desktop_len );
182 get_unicode_string( &params->ShellInfo, &src, &dst, info->shellinfo_len );
183
184 /* runtime info isn't a real string */
185 params->RuntimeInfo.Buffer = dst;
186 params->RuntimeInfo.Length = params->RuntimeInfo.MaximumLength = info->runtime_len;
187 memcpy( dst, src, info->runtime_len );
188
189 /* environment needs to be a separate memory block */
190 ptr = NULL;
191 alloc_size = max( 1, env_size );
192 status = NtAllocateVirtualMemory( NtCurrentProcess(), &ptr, 0, &alloc_size,
193 MEM_COMMIT, PAGE_READWRITE );
194 if (status != STATUS_SUCCESS) goto done;
195 memcpy( ptr, (char *)info + info_size, env_size );
196 params->Environment = ptr;
197
198 done:
199 RtlFreeHeap( GetProcessHeap(), 0, info );
200 return status;
201 }
202
203 #ifdef __linux__
204
205 #ifdef HAVE_ELF_H
206 # include <elf.h>
207 #endif
208 #ifdef HAVE_LINK_H
209 # include <link.h>
210 #endif
211 #ifdef HAVE_SYS_AUXV_H
212 # include <sys/auxv.h>
213 #endif
214 #ifndef HAVE_GETAUXVAL
215 static unsigned long getauxval( unsigned long id )
216 {
217 extern char **__wine_main_environ;
218 char **ptr = __wine_main_environ;
219 ElfW(auxv_t) *auxv;
220
221 while (*ptr) ptr++;
222 while (!*ptr) ptr++;
223 for (auxv = (ElfW(auxv_t) *)ptr; auxv->a_type; auxv++)
224 if (auxv->a_type == id) return auxv->a_un.a_val;
225 return 0;
226 }
227 #endif
228
229 static ULONG_PTR get_image_addr(void)
230 {
231 ULONG_PTR size, num, phdr_addr = getauxval( AT_PHDR );
232 ElfW(Phdr) *phdr;
233
234 if (!phdr_addr) return 0;
235 phdr = (ElfW(Phdr) *)phdr_addr;
236 size = getauxval( AT_PHENT );
237 num = getauxval( AT_PHNUM );
238 while (num--)
239 {
240 if (phdr->p_type == PT_PHDR) return phdr_addr - phdr->p_offset;
241 phdr = (ElfW(Phdr) *)((char *)phdr + size);
242 }
243 return 0;
244 }
245
246 #elif defined(__APPLE__)
247 #include <mach/mach.h>
248 #include <mach/mach_error.h>
249
250 static ULONG_PTR get_image_addr(void)
251 {
252 ULONG_PTR ret = 0;
253 #ifdef TASK_DYLD_INFO
254 struct task_dyld_info dyld_info;
255 mach_msg_type_number_t size = TASK_DYLD_INFO_COUNT;
256 if (task_info(mach_task_self(), TASK_DYLD_INFO, (task_info_t)&dyld_info, &size) == KERN_SUCCESS)
257 ret = dyld_info.all_image_info_addr;
258 #endif
259 return ret;
260 }
261
262 #else
263 static ULONG_PTR get_image_addr(void)
264 {
265 return 0;
266 }
267 #endif
268
269 /***********************************************************************
270 * thread_init
271 *
272 * Setup the initial thread.
273 *
274 * NOTES: The first allocated TEB on NT is at 0x7ffde000.
275 */
276 HANDLE thread_init(void)
277 {
278 TEB *teb;
279 void *addr;
280 BOOL suspend;
281 SIZE_T size, info_size;
282 HANDLE exe_file = 0;
283 LARGE_INTEGER now;
284 NTSTATUS status;
285 struct ntdll_thread_data *thread_data;
286 static struct debug_info debug_info; /* debug info for initial thread */
287
288 virtual_init();
289
290 /* reserve space for shared user data */
291
292 addr = (void *)0x7ffe0000;
293 size = 0x10000;
294 status = NtAllocateVirtualMemory( NtCurrentProcess(), &addr, 0, &size,
295 MEM_RESERVE|MEM_COMMIT, PAGE_READWRITE );
296 if (status)
297 {
298 MESSAGE( "wine: failed to map the shared user data: %08x\n", status );
299 exit(1);
300 }
301 user_shared_data = addr;
302 memcpy( user_shared_data->NtSystemRoot, default_windirW, sizeof(default_windirW) );
303
304 /* allocate and initialize the PEB */
305
306 addr = NULL;
307 size = sizeof(*peb);
308 NtAllocateVirtualMemory( NtCurrentProcess(), &addr, 1, &size,
309 MEM_COMMIT | MEM_TOP_DOWN, PAGE_READWRITE );
310 peb = addr;
311
312 peb->FastPebLock = &peb_lock;
313 peb->ProcessParameters = &params;
314 peb->TlsBitmap = &tls_bitmap;
315 peb->TlsExpansionBitmap = &tls_expansion_bitmap;
316 peb->FlsBitmap = &fls_bitmap;
317 peb->LdrData = &ldr;
318 peb->OSMajorVersion = 5;
319 peb->OSMinorVersion = 1;
320 peb->OSBuildNumber = 0xA28;
321 peb->OSPlatformId = VER_PLATFORM_WIN32_NT;
322 params.CurrentDirectory.DosPath.Buffer = current_dir;
323 params.CurrentDirectory.DosPath.MaximumLength = sizeof(current_dir);
324 params.wShowWindow = 1; /* SW_SHOWNORMAL */
325 ldr.Length = sizeof(ldr);
326 ldr.Initialized = TRUE;
327 RtlInitializeBitMap( &tls_bitmap, peb->TlsBitmapBits, sizeof(peb->TlsBitmapBits) * 8 );
328 RtlInitializeBitMap( &tls_expansion_bitmap, peb->TlsExpansionBitmapBits,
329 sizeof(peb->TlsExpansionBitmapBits) * 8 );
330 RtlInitializeBitMap( &fls_bitmap, peb->FlsBitmapBits, sizeof(peb->FlsBitmapBits) * 8 );
331 RtlSetBits( peb->TlsBitmap, 0, 1 ); /* TLS index 0 is reserved and should be initialized to NULL. */
332 RtlSetBits( peb->FlsBitmap, 0, 1 );
333 InitializeListHead( &peb->FlsListHead );
334 InitializeListHead( &ldr.InLoadOrderModuleList );
335 InitializeListHead( &ldr.InMemoryOrderModuleList );
336 InitializeListHead( &ldr.InInitializationOrderModuleList );
337 *(ULONG_PTR *)peb->Reserved = get_image_addr();
338
339 /*
340 * Starting with Vista, the first user to log on has session id 1.
341 * Session id 0 is for processes that don't interact with the user (like services).
342 */
343 peb->SessionId = 1;
344
345 /* allocate and initialize the initial TEB */
346
347 signal_alloc_thread( &teb );
348 teb->Peb = peb;
349 teb->Tib.StackBase = (void *)~0UL;
350 teb->StaticUnicodeString.Buffer = teb->StaticUnicodeBuffer;
351 teb->StaticUnicodeString.MaximumLength = sizeof(teb->StaticUnicodeBuffer);
352
353 thread_data = (struct ntdll_thread_data *)&teb->GdiTebBatch;
354 thread_data->request_fd = -1;
355 thread_data->reply_fd = -1;
356 thread_data->wait_fd[0] = -1;
357 thread_data->wait_fd[1] = -1;
358 thread_data->debug_info = &debug_info;
359
360 signal_init_thread( teb );
361 virtual_init_threading();
362
363 debug_info.str_pos = debug_info.strings;
364 debug_info.out_pos = debug_info.output;
365 debug_init();
366
367 /* setup the server connection */
368 server_init_process();
369 info_size = server_init_thread( peb, &suspend );
370
371 /* create the process heap */
372 if (!(peb->ProcessHeap = RtlCreateHeap( HEAP_GROWABLE, NULL, 0, 0, NULL, NULL )))
373 {
374 MESSAGE( "wine: failed to create the process heap\n" );
375 exit(1);
376 }
377
378 /* allocate user parameters */
379 if (info_size)
380 {
381 init_user_process_params( info_size, &exe_file );
382 }
383 else
384 {
385 if (isatty(0) || isatty(1) || isatty(2))
386 params.ConsoleHandle = (HANDLE)2; /* see kernel32/kernel_private.h */
387 if (!isatty(0))
388 wine_server_fd_to_handle( 0, GENERIC_READ|SYNCHRONIZE, OBJ_INHERIT, &params.hStdInput );
389 if (!isatty(1))
390 wine_server_fd_to_handle( 1, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, &params.hStdOutput );
391 if (!isatty(2))
392 wine_server_fd_to_handle( 2, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, &params.hStdError );
393 }
394
395 /* initialize time values in user_shared_data */
396 NtQuerySystemTime( &now );
397 user_shared_data->SystemTime.LowPart = now.u.LowPart;
398 user_shared_data->SystemTime.High1Time = user_shared_data->SystemTime.High2Time = now.u.HighPart;
399 user_shared_data->u.TickCountQuad = (now.QuadPart - server_start_time) / 10000;
400 user_shared_data->u.TickCount.High2Time = user_shared_data->u.TickCount.High1Time;
401 user_shared_data->TickCountLowDeprecated = user_shared_data->u.TickCount.LowPart;
402 user_shared_data->TickCountMultiplier = 1 << 24;
403
404 fill_cpu_info();
405
406 NtCreateKeyedEvent( &keyed_event, GENERIC_READ | GENERIC_WRITE, NULL, 0 );
407
408 return exe_file;
409 }
410
411
412 /***********************************************************************
413 * free_thread_data
414 */
415 static void free_thread_data( TEB *teb )
416 {
417 struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)&teb->GdiTebBatch;
418 SIZE_T size;
419
420 if (teb->DeallocationStack)
421 {
422 size = 0;
423 NtFreeVirtualMemory( GetCurrentProcess(), &teb->DeallocationStack, &size, MEM_RELEASE );
424 }
425 if (thread_data->start_stack)
426 {
427 size = 0;
428 NtFreeVirtualMemory( GetCurrentProcess(), &thread_data->start_stack, &size, MEM_RELEASE );
429 }
430 signal_free_thread( teb );
431 }
432
433
434 /***********************************************************************
435 * abort_thread
436 */
437 void abort_thread( int status )
438 {
439 pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
440 if (interlocked_xchg_add( &nb_threads, -1 ) <= 1) _exit( status );
441 signal_exit_thread( status );
442 }
443
444
445 /***********************************************************************
446 * exit_thread
447 */
448 void exit_thread( int status )
449 {
450 close( ntdll_get_thread_data()->wait_fd[0] );
451 close( ntdll_get_thread_data()->wait_fd[1] );
452 close( ntdll_get_thread_data()->reply_fd );
453 close( ntdll_get_thread_data()->request_fd );
454 pthread_exit( UIntToPtr(status) );
455 }
456
457
458 /***********************************************************************
459 * RtlExitUserThread (NTDLL.@)
460 */
461 void WINAPI RtlExitUserThread( ULONG status )
462 {
463 static void *prev_teb;
464 TEB *teb;
465
466 if (status) /* send the exit code to the server (0 is already the default) */
467 {
468 SERVER_START_REQ( terminate_thread )
469 {
470 req->handle = wine_server_obj_handle( GetCurrentThread() );
471 req->exit_code = status;
472 wine_server_call( req );
473 }
474 SERVER_END_REQ;
475 }
476
477 if (interlocked_xchg_add( &nb_threads, -1 ) <= 1)
478 {
479 LdrShutdownProcess();
480 pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
481 signal_exit_process( status );
482 }
483
484 LdrShutdownThread();
485 RtlFreeThreadActivationContextStack();
486
487 pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
488
489 if ((teb = interlocked_xchg_ptr( &prev_teb, NtCurrentTeb() )))
490 {
491 struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)&teb->GdiTebBatch;
492
493 if (thread_data->pthread_id)
494 {
495 pthread_join( thread_data->pthread_id, NULL );
496 free_thread_data( teb );
497 }
498 }
499
500 signal_exit_thread( status );
501 }
502
503
504 /***********************************************************************
505 * start_thread
506 *
507 * Startup routine for a newly created thread.
508 */
509 static void start_thread( struct startup_info *info )
510 {
511 BOOL suspend;
512 TEB *teb = info->teb;
513 struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)&teb->GdiTebBatch;
514 struct debug_info debug_info;
515
516 debug_info.str_pos = debug_info.strings;
517 debug_info.out_pos = debug_info.output;
518 thread_data->debug_info = &debug_info;
519 thread_data->pthread_id = pthread_self();
520
521 signal_init_thread( teb );
522 server_init_thread( info->entry_point, &suspend );
523 signal_start_thread( (LPTHREAD_START_ROUTINE)info->entry_point, info->entry_arg, suspend );
524 }
525
526
527 /***********************************************************************
528 * RtlCreateUserThread (NTDLL.@)
529 */
530 NTSTATUS WINAPI RtlCreateUserThread( HANDLE process, const SECURITY_DESCRIPTOR *descr,
531 BOOLEAN suspended, PVOID stack_addr,
532 SIZE_T stack_reserve, SIZE_T stack_commit,
533 PRTL_THREAD_START_ROUTINE start, void *param,
534 HANDLE *handle_ptr, CLIENT_ID *id )
535 {
536 sigset_t sigset;
537 pthread_t pthread_id;
538 pthread_attr_t attr;
539 struct ntdll_thread_data *thread_data;
540 struct startup_info *info;
541 HANDLE handle = 0, actctx = 0;
542 TEB *teb = NULL;
543 DWORD tid = 0;
544 int request_pipe[2];
545 NTSTATUS status;
546 SIZE_T extra_stack = PTHREAD_STACK_MIN;
547
548 if (process != NtCurrentProcess())
549 {
550 apc_call_t call;
551 apc_result_t result;
552
553 memset( &call, 0, sizeof(call) );
554
555 call.create_thread.type = APC_CREATE_THREAD;
556 call.create_thread.func = wine_server_client_ptr( start );
557 call.create_thread.arg = wine_server_client_ptr( param );
558 call.create_thread.reserve = stack_reserve;
559 call.create_thread.commit = stack_commit;
560 call.create_thread.suspend = suspended;
561 status = server_queue_process_apc( process, &call, &result );
562 if (status != STATUS_SUCCESS) return status;
563
564 if (result.create_thread.status == STATUS_SUCCESS)
565 {
566 if (id) id->UniqueThread = ULongToHandle(result.create_thread.tid);
567 if (handle_ptr) *handle_ptr = wine_server_ptr_handle( result.create_thread.handle );
568 else NtClose( wine_server_ptr_handle( result.create_thread.handle ));
569 }
570 return result.create_thread.status;
571 }
572
573 if (server_pipe( request_pipe ) == -1) return STATUS_TOO_MANY_OPENED_FILES;
574 wine_server_send_fd( request_pipe[0] );
575
576 SERVER_START_REQ( new_thread )
577 {
578 req->access = THREAD_ALL_ACCESS;
579 req->attributes = 0; /* FIXME */
580 req->suspend = suspended;
581 req->request_fd = request_pipe[0];
582 if (!(status = wine_server_call( req )))
583 {
584 handle = wine_server_ptr_handle( reply->handle );
585 tid = reply->tid;
586 }
587 close( request_pipe[0] );
588 }
589 SERVER_END_REQ;
590
591 if (status)
592 {
593 close( request_pipe[1] );
594 return status;
595 }
596
597 pthread_sigmask( SIG_BLOCK, &server_block_set, &sigset );
598
599 if ((status = signal_alloc_thread( &teb ))) goto error;
600
601 teb->Peb = NtCurrentTeb()->Peb;
602 teb->ClientId.UniqueProcess = ULongToHandle(GetCurrentProcessId());
603 teb->ClientId.UniqueThread = ULongToHandle(tid);
604 teb->StaticUnicodeString.Buffer = teb->StaticUnicodeBuffer;
605 teb->StaticUnicodeString.MaximumLength = sizeof(teb->StaticUnicodeBuffer);
606
607 /* create default activation context frame for new thread */
608 RtlGetActiveActivationContext(&actctx);
609 if (actctx)
610 {
611 RTL_ACTIVATION_CONTEXT_STACK_FRAME *frame;
612
613 frame = RtlAllocateHeap(GetProcessHeap(), 0, sizeof(*frame));
614 frame->Previous = NULL;
615 frame->ActivationContext = actctx;
616 frame->Flags = 0;
617 teb->ActivationContextStack.ActiveFrame = frame;
618 }
619
620 info = (struct startup_info *)(teb + 1);
621 info->teb = teb;
622 info->entry_point = start;
623 info->entry_arg = param;
624
625 if ((status = virtual_alloc_thread_stack( teb, stack_reserve, stack_commit, &extra_stack )))
626 goto error;
627
628 thread_data = (struct ntdll_thread_data *)&teb->GdiTebBatch;
629 thread_data->request_fd = request_pipe[1];
630 thread_data->reply_fd = -1;
631 thread_data->wait_fd[0] = -1;
632 thread_data->wait_fd[1] = -1;
633 thread_data->start_stack = (char *)teb->Tib.StackBase;
634
635 pthread_attr_init( &attr );
636 pthread_attr_setstack( &attr, teb->DeallocationStack,
637 (char *)teb->Tib.StackBase + extra_stack - (char *)teb->DeallocationStack );
638 pthread_attr_setscope( &attr, PTHREAD_SCOPE_SYSTEM ); /* force creating a kernel thread */
639 interlocked_xchg_add( &nb_threads, 1 );
640 if (pthread_create( &pthread_id, &attr, (void * (*)(void *))start_thread, info ))
641 {
642 interlocked_xchg_add( &nb_threads, -1 );
643 pthread_attr_destroy( &attr );
644 status = STATUS_NO_MEMORY;
645 goto error;
646 }
647 pthread_attr_destroy( &attr );
648 pthread_sigmask( SIG_SETMASK, &sigset, NULL );
649
650 if (id) id->UniqueThread = ULongToHandle(tid);
651 if (handle_ptr) *handle_ptr = handle;
652 else NtClose( handle );
653
654 return STATUS_SUCCESS;
655
656 error:
657 if (teb) free_thread_data( teb );
658 if (handle) NtClose( handle );
659 pthread_sigmask( SIG_SETMASK, &sigset, NULL );
660 close( request_pipe[1] );
661 return status;
662 }
663
664
665 /******************************************************************************
666 * RtlGetNtGlobalFlags (NTDLL.@)
667 */
668 ULONG WINAPI RtlGetNtGlobalFlags(void)
669 {
670 if (!peb) return 0; /* init not done yet */
671 return peb->NtGlobalFlag;
672 }
673
674
675 /***********************************************************************
676 * NtOpenThread (NTDLL.@)
677 * ZwOpenThread (NTDLL.@)
678 */
679 NTSTATUS WINAPI NtOpenThread( HANDLE *handle, ACCESS_MASK access,
680 const OBJECT_ATTRIBUTES *attr, const CLIENT_ID *id )
681 {
682 NTSTATUS ret;
683
684 SERVER_START_REQ( open_thread )
685 {
686 req->tid = HandleToULong(id->UniqueThread);
687 req->access = access;
688 req->attributes = attr ? attr->Attributes : 0;
689 ret = wine_server_call( req );
690 *handle = wine_server_ptr_handle( reply->handle );
691 }
692 SERVER_END_REQ;
693 return ret;
694 }
695
696
697 /******************************************************************************
698 * NtSuspendThread (NTDLL.@)
699 * ZwSuspendThread (NTDLL.@)
700 */
701 NTSTATUS WINAPI NtSuspendThread( HANDLE handle, PULONG count )
702 {
703 NTSTATUS ret;
704
705 SERVER_START_REQ( suspend_thread )
706 {
707 req->handle = wine_server_obj_handle( handle );
708 if (!(ret = wine_server_call( req )))
709 {
710 if (count) *count = reply->count;
711 }
712 }
713 SERVER_END_REQ;
714 return ret;
715 }
716
717
718 /******************************************************************************
719 * NtResumeThread (NTDLL.@)
720 * ZwResumeThread (NTDLL.@)
721 */
722 NTSTATUS WINAPI NtResumeThread( HANDLE handle, PULONG count )
723 {
724 NTSTATUS ret;
725
726 SERVER_START_REQ( resume_thread )
727 {
728 req->handle = wine_server_obj_handle( handle );
729 if (!(ret = wine_server_call( req )))
730 {
731 if (count) *count = reply->count;
732 }
733 }
734 SERVER_END_REQ;
735 return ret;
736 }
737
738
739 /******************************************************************************
740 * NtAlertResumeThread (NTDLL.@)
741 * ZwAlertResumeThread (NTDLL.@)
742 */
743 NTSTATUS WINAPI NtAlertResumeThread( HANDLE handle, PULONG count )
744 {
745 FIXME( "stub: should alert thread %p\n", handle );
746 return NtResumeThread( handle, count );
747 }
748
749
750 /******************************************************************************
751 * NtAlertThread (NTDLL.@)
752 * ZwAlertThread (NTDLL.@)
753 */
754 NTSTATUS WINAPI NtAlertThread( HANDLE handle )
755 {
756 FIXME( "stub: %p\n", handle );
757 return STATUS_NOT_IMPLEMENTED;
758 }
759
760
761 /******************************************************************************
762 * NtTerminateThread (NTDLL.@)
763 * ZwTerminateThread (NTDLL.@)
764 */
765 NTSTATUS WINAPI NtTerminateThread( HANDLE handle, LONG exit_code )
766 {
767 NTSTATUS ret;
768 BOOL self;
769
770 SERVER_START_REQ( terminate_thread )
771 {
772 req->handle = wine_server_obj_handle( handle );
773 req->exit_code = exit_code;
774 ret = wine_server_call( req );
775 self = !ret && reply->self;
776 }
777 SERVER_END_REQ;
778
779 if (self) abort_thread( exit_code );
780 return ret;
781 }
782
783
784 /******************************************************************************
785 * NtQueueApcThread (NTDLL.@)
786 */
787 NTSTATUS WINAPI NtQueueApcThread( HANDLE handle, PNTAPCFUNC func, ULONG_PTR arg1,
788 ULONG_PTR arg2, ULONG_PTR arg3 )
789 {
790 NTSTATUS ret;
791 SERVER_START_REQ( queue_apc )
792 {
793 req->handle = wine_server_obj_handle( handle );
794 if (func)
795 {
796 req->call.type = APC_USER;
797 req->call.user.func = wine_server_client_ptr( func );
798 req->call.user.args[0] = arg1;
799 req->call.user.args[1] = arg2;
800 req->call.user.args[2] = arg3;
801 }
802 else req->call.type = APC_NONE; /* wake up only */
803 ret = wine_server_call( req );
804 }
805 SERVER_END_REQ;
806 return ret;
807 }
808
809
810 /******************************************************************************
811 * RtlPushFrame (NTDLL.@)
812 */
813 void WINAPI RtlPushFrame( TEB_ACTIVE_FRAME *frame )
814 {
815 frame->Previous = NtCurrentTeb()->ActiveFrame;
816 NtCurrentTeb()->ActiveFrame = frame;
817 }
818
819
820 /******************************************************************************
821 * RtlPopFrame (NTDLL.@)
822 */
823 void WINAPI RtlPopFrame( TEB_ACTIVE_FRAME *frame )
824 {
825 NtCurrentTeb()->ActiveFrame = frame->Previous;
826 }
827
828
829 /******************************************************************************
830 * RtlGetFrame (NTDLL.@)
831 */
832 TEB_ACTIVE_FRAME * WINAPI RtlGetFrame(void)
833 {
834 return NtCurrentTeb()->ActiveFrame;
835 }
836
837
838 /***********************************************************************
839 * set_thread_context
840 */
841 NTSTATUS set_thread_context( HANDLE handle, const CONTEXT *context, BOOL *self )
842 {
843 NTSTATUS ret;
844 DWORD dummy, i;
845 context_t server_context;
846
847 context_to_server( &server_context, context );
848
849 SERVER_START_REQ( set_thread_context )
850 {
851 req->handle = wine_server_obj_handle( handle );
852 req->suspend = 1;
853 wine_server_add_data( req, &server_context, sizeof(server_context) );
854 ret = wine_server_call( req );
855 *self = reply->self;
856 }
857 SERVER_END_REQ;
858
859 if (ret == STATUS_PENDING)
860 {
861 for (i = 0; i < 100; i++)
862 {
863 SERVER_START_REQ( set_thread_context )
864 {
865 req->handle = wine_server_obj_handle( handle );
866 req->suspend = 0;
867 wine_server_add_data( req, &server_context, sizeof(server_context) );
868 ret = wine_server_call( req );
869 }
870 SERVER_END_REQ;
871 if (ret == STATUS_PENDING)
872 {
873 LARGE_INTEGER timeout;
874 timeout.QuadPart = -10000;
875 NtDelayExecution( FALSE, &timeout );
876 }
877 else break;
878 }
879 NtResumeThread( handle, &dummy );
880 if (ret == STATUS_PENDING) ret = STATUS_ACCESS_DENIED;
881 }
882
883 return ret;
884 }
885
886
887 /* convert CPU-specific flags to generic server flags */
888 static inline unsigned int get_server_context_flags( DWORD flags )
889 {
890 unsigned int ret = 0;
891
892 flags &= 0x3f; /* mask CPU id flags */
893 if (flags & CONTEXT_CONTROL) ret |= SERVER_CTX_CONTROL;
894 if (flags & CONTEXT_INTEGER) ret |= SERVER_CTX_INTEGER;
895 #ifdef CONTEXT_SEGMENTS
896 if (flags & CONTEXT_SEGMENTS) ret |= SERVER_CTX_SEGMENTS;
897 #endif
898 #ifdef CONTEXT_FLOATING_POINT
899 if (flags & CONTEXT_FLOATING_POINT) ret |= SERVER_CTX_FLOATING_POINT;
900 #endif
901 #ifdef CONTEXT_DEBUG_REGISTERS
902 if (flags & CONTEXT_DEBUG_REGISTERS) ret |= SERVER_CTX_DEBUG_REGISTERS;
903 #endif
904 #ifdef CONTEXT_EXTENDED_REGISTERS
905 if (flags & CONTEXT_EXTENDED_REGISTERS) ret |= SERVER_CTX_EXTENDED_REGISTERS;
906 #endif
907 return ret;
908 }
909
910 /***********************************************************************
911 * get_thread_context
912 */
913 NTSTATUS get_thread_context( HANDLE handle, CONTEXT *context, BOOL *self )
914 {
915 NTSTATUS ret;
916 DWORD dummy, i;
917 unsigned int server_flags = get_server_context_flags( context->ContextFlags );
918 context_t server_context;
919
920 SERVER_START_REQ( get_thread_context )
921 {
922 req->handle = wine_server_obj_handle( handle );
923 req->flags = server_flags;
924 req->suspend = 1;
925 wine_server_set_reply( req, &server_context, sizeof(server_context) );
926 ret = wine_server_call( req );
927 *self = reply->self;
928 }
929 SERVER_END_REQ;
930
931 if (ret == STATUS_PENDING)
932 {
933 for (i = 0; i < 100; i++)
934 {
935 SERVER_START_REQ( get_thread_context )
936 {
937 req->handle = wine_server_obj_handle( handle );
938 req->flags = server_flags;
939 req->suspend = 0;
940 wine_server_set_reply( req, &server_context, sizeof(server_context) );
941 ret = wine_server_call( req );
942 }
943 SERVER_END_REQ;
944 if (ret == STATUS_PENDING)
945 {
946 LARGE_INTEGER timeout;
947 timeout.QuadPart = -10000;
948 NtDelayExecution( FALSE, &timeout );
949 }
950 else break;
951 }
952 NtResumeThread( handle, &dummy );
953 if (ret == STATUS_PENDING) ret = STATUS_ACCESS_DENIED;
954 }
955 if (!ret) ret = context_from_server( context, &server_context );
956 return ret;
957 }
958
959
960 /******************************************************************************
961 * NtQueryInformationThread (NTDLL.@)
962 * ZwQueryInformationThread (NTDLL.@)
963 */
964 NTSTATUS WINAPI NtQueryInformationThread( HANDLE handle, THREADINFOCLASS class,
965 void *data, ULONG length, ULONG *ret_len )
966 {
967 NTSTATUS status;
968
969 switch(class)
970 {
971 case ThreadBasicInformation:
972 {
973 THREAD_BASIC_INFORMATION info;
974 const ULONG_PTR affinity_mask = get_system_affinity_mask();
975
976 SERVER_START_REQ( get_thread_info )
977 {
978 req->handle = wine_server_obj_handle( handle );
979 req->tid_in = 0;
980 if (!(status = wine_server_call( req )))
981 {
982 info.ExitStatus = reply->exit_code;
983 info.TebBaseAddress = wine_server_get_ptr( reply->teb );
984 info.ClientId.UniqueProcess = ULongToHandle(reply->pid);
985 info.ClientId.UniqueThread = ULongToHandle(reply->tid);
986 info.AffinityMask = reply->affinity & affinity_mask;
987 info.Priority = reply->priority;
988 info.BasePriority = reply->priority; /* FIXME */
989 }
990 }
991 SERVER_END_REQ;
992 if (status == STATUS_SUCCESS)
993 {
994 if (data) memcpy( data, &info, min( length, sizeof(info) ));
995 if (ret_len) *ret_len = min( length, sizeof(info) );
996 }
997 }
998 return status;
999 case ThreadAffinityMask:
1000 {
1001 const ULONG_PTR affinity_mask = get_system_affinity_mask();
1002 ULONG_PTR affinity = 0;
1003
1004 SERVER_START_REQ( get_thread_info )
1005 {
1006 req->handle = wine_server_obj_handle( handle );
1007 req->tid_in = 0;
1008 if (!(status = wine_server_call( req )))
1009 affinity = reply->affinity & affinity_mask;
1010 }
1011 SERVER_END_REQ;
1012 if (status == STATUS_SUCCESS)
1013 {
1014 if (data) memcpy( data, &affinity, min( length, sizeof(affinity) ));
1015 if (ret_len) *ret_len = min( length, sizeof(affinity) );
1016 }
1017 }
1018 return status;
1019 case ThreadTimes:
1020 {
1021 KERNEL_USER_TIMES kusrt;
1022
1023 SERVER_START_REQ( get_thread_times )
1024 {
1025 req->handle = wine_server_obj_handle( handle );
1026 status = wine_server_call( req );
1027 if (status == STATUS_SUCCESS)
1028 {
1029 kusrt.CreateTime.QuadPart = reply->creation_time;
1030 kusrt.ExitTime.QuadPart = reply->exit_time;
1031 }
1032 }
1033 SERVER_END_REQ;
1034 if (status == STATUS_SUCCESS)
1035 {
1036 /* We call times(2) for kernel time or user time */
1037 /* We can only (portably) do this for the current thread */
1038 if (handle == GetCurrentThread())
1039 {
1040 struct tms time_buf;
1041 long clocks_per_sec = sysconf(_SC_CLK_TCK);
1042
1043 times(&time_buf);
1044 kusrt.KernelTime.QuadPart = (ULONGLONG)time_buf.tms_stime * 10000000 / clocks_per_sec;
1045 kusrt.UserTime.QuadPart = (ULONGLONG)time_buf.tms_utime * 10000000 / clocks_per_sec;
1046 }
1047 else
1048 {
1049 static BOOL reported = FALSE;
1050
1051 kusrt.KernelTime.QuadPart = 0;
1052 kusrt.UserTime.QuadPart = 0;
1053 if (reported)
1054 TRACE("Cannot get kerneltime or usertime of other threads\n");
1055 else
1056 {
1057 FIXME("Cannot get kerneltime or usertime of other threads\n");
1058 reported = TRUE;
1059 }
1060 }
1061 if (data) memcpy( data, &kusrt, min( length, sizeof(kusrt) ));
1062 if (ret_len) *ret_len = min( length, sizeof(kusrt) );
1063 }
1064 }
1065 return status;
1066 case ThreadDescriptorTableEntry:
1067 {
1068 #ifdef __i386__
1069 THREAD_DESCRIPTOR_INFORMATION* tdi = data;
1070 if (length < sizeof(*tdi))
1071 status = STATUS_INFO_LENGTH_MISMATCH;
1072 else if (!(tdi->Selector & 4)) /* GDT selector */
1073 {
1074 unsigned sel = LOWORD(tdi->Selector) & ~3; /* ignore RPL */
1075 status = STATUS_SUCCESS;
1076 if (!sel) /* null selector */
1077 memset( &tdi->Entry, 0, sizeof(tdi->Entry) );
1078 else
1079 {
1080 tdi->Entry.BaseLow = 0;
1081 tdi->Entry.HighWord.Bits.BaseMid = 0;
1082 tdi->Entry.HighWord.Bits.BaseHi = 0;
1083 tdi->Entry.LimitLow = 0xffff;
1084 tdi->Entry.HighWord.Bits.LimitHi = 0xf;
1085 tdi->Entry.HighWord.Bits.Dpl = 3;
1086 tdi->Entry.HighWord.Bits.Sys = 0;
1087 tdi->Entry.HighWord.Bits.Pres = 1;
1088 tdi->Entry.HighWord.Bits.Granularity = 1;
1089 tdi->Entry.HighWord.Bits.Default_Big = 1;
1090 tdi->Entry.HighWord.Bits.Type = 0x12;
1091 tdi->Entry.HighWord.Bits.Reserved_0 = 0;
1092 /* it has to be one of the system GDT selectors */
1093 if (sel != (wine_get_ds() & ~3) && sel != (wine_get_ss() & ~3))
1094 {
1095 if (sel == (wine_get_cs() & ~3))
1096 tdi->Entry.HighWord.Bits.Type |= 8; /* code segment */
1097 else if (sel == (wine_get_fs() & ~3))
1098 {
1099 ULONG_PTR fs_base = (ULONG_PTR)NtCurrentTeb();
1100 tdi->Entry.BaseLow = fs_base & 0xffff;
1101 tdi->Entry.HighWord.Bits.BaseMid = (fs_base >> 16) & 0xff;
1102 tdi->Entry.HighWord.Bits.BaseHi = (fs_base >> 24) & 0xff;
1103 tdi->Entry.LimitLow = 0x0fff;
1104 tdi->Entry.HighWord.Bits.LimitHi = 0;
1105 tdi->Entry.HighWord.Bits.Granularity = 0;
1106 }
1107 else status = STATUS_ACCESS_DENIED;
1108 }
1109 }
1110 }
1111 else
1112 {
1113 SERVER_START_REQ( get_selector_entry )
1114 {
1115 req->handle = wine_server_obj_handle( handle );
1116 req->entry = LOWORD(tdi->Selector) >> 3;
1117 status = wine_server_call( req );
1118 if (!status)
1119 {
1120 if (!(reply->flags & WINE_LDT_FLAGS_ALLOCATED))
1121 status = STATUS_ACCESS_VIOLATION;
1122 else
1123 {
1124 wine_ldt_set_base ( &tdi->Entry, (void *)reply->base );
1125 wine_ldt_set_limit( &tdi->Entry, reply->limit );
1126 wine_ldt_set_flags( &tdi->Entry, reply->flags );
1127 }
1128 }
1129 }
1130 SERVER_END_REQ;
1131 }
1132 if (status == STATUS_SUCCESS && ret_len)
1133 /* yes, that's a bit strange, but it's the way it is */
1134 *ret_len = sizeof(LDT_ENTRY);
1135 #else
1136 status = STATUS_NOT_IMPLEMENTED;
1137 #endif
1138 return status;
1139 }
1140 case ThreadAmILastThread:
1141 {
1142 SERVER_START_REQ(get_thread_info)
1143 {
1144 req->handle = wine_server_obj_handle( handle );
1145 req->tid_in = 0;
1146 status = wine_server_call( req );
1147 if (status == STATUS_SUCCESS)
1148 {
1149 BOOLEAN last = reply->last;
1150 if (data) memcpy( data, &last, min( length, sizeof(last) ));
1151 if (ret_len) *ret_len = min( length, sizeof(last) );
1152 }
1153 }
1154 SERVER_END_REQ;
1155 return status;
1156 }
1157 case ThreadQuerySetWin32StartAddress:
1158 {
1159 SERVER_START_REQ( get_thread_info )
1160 {
1161 req->handle = wine_server_obj_handle( handle );
1162 req->tid_in = 0;
1163 status = wine_server_call( req );
1164 if (status == STATUS_SUCCESS)
1165 {
1166 PRTL_THREAD_START_ROUTINE entry = wine_server_get_ptr( reply->entry_point );
1167 if (data) memcpy( data, &entry, min( length, sizeof(entry) ) );
1168 if (ret_len) *ret_len = min( length, sizeof(entry) );
1169 }
1170 }
1171 SERVER_END_REQ;
1172 return status;
1173 }
1174 case ThreadGroupInformation:
1175 {
1176 const ULONG_PTR affinity_mask = get_system_affinity_mask();
1177 GROUP_AFFINITY affinity;
1178
1179 memset(&affinity, 0, sizeof(affinity));
1180 affinity.Group = 0; /* Wine only supports max 64 processors */
1181
1182 SERVER_START_REQ( get_thread_info )
1183 {
1184 req->handle = wine_server_obj_handle( handle );
1185 req->tid_in = 0;
1186 if (!(status = wine_server_call( req )))
1187 affinity.Mask = reply->affinity & affinity_mask;
1188 }
1189 SERVER_END_REQ;
1190 if (status == STATUS_SUCCESS)
1191 {
1192 if (data) memcpy( data, &affinity, min( length, sizeof(affinity) ));
1193 if (ret_len) *ret_len = min( length, sizeof(affinity) );
1194 }
1195 }
1196 return status;
1197 case ThreadIsIoPending:
1198 FIXME( "ThreadIsIoPending info class not supported yet\n" );
1199 if (length != sizeof(BOOL)) return STATUS_INFO_LENGTH_MISMATCH;
1200 if (!data) return STATUS_ACCESS_DENIED;
1201
1202 *(BOOL*)data = FALSE;
1203 if (ret_len) *ret_len = sizeof(BOOL);
1204 return STATUS_SUCCESS;
1205 case ThreadPriority:
1206 case ThreadBasePriority:
1207 case ThreadImpersonationToken:
1208 case ThreadEnableAlignmentFaultFixup:
1209 case ThreadEventPair_Reusable:
1210 case ThreadZeroTlsCell:
1211 case ThreadPerformanceCount:
1212 case ThreadIdealProcessor:
1213 case ThreadPriorityBoost:
1214 case ThreadSetTlsArrayAddress:
1215 default:
1216 FIXME( "info class %d not supported yet\n", class );
1217 return STATUS_NOT_IMPLEMENTED;
1218 }
1219 }
1220
1221
1222 /******************************************************************************
1223 * NtSetInformationThread (NTDLL.@)
1224 * ZwSetInformationThread (NTDLL.@)
1225 */
1226 NTSTATUS WINAPI NtSetInformationThread( HANDLE handle, THREADINFOCLASS class,
1227 LPCVOID data, ULONG length )
1228 {
1229 NTSTATUS status;
1230 switch(class)
1231 {
1232 case ThreadZeroTlsCell:
1233 if (handle == GetCurrentThread())
1234 {
1235 LIST_ENTRY *entry;
1236 DWORD index;
1237
1238 if (length != sizeof(DWORD)) return STATUS_INVALID_PARAMETER;
1239 index = *(const DWORD *)data;
1240 if (index < TLS_MINIMUM_AVAILABLE)
1241 {
1242 RtlAcquirePebLock();
1243 for (entry = tls_links.Flink; entry != &tls_links; entry = entry->Flink)
1244 {
1245 TEB *teb = CONTAINING_RECORD(entry, TEB, TlsLinks);
1246 teb->TlsSlots[index] = 0;
1247 }
1248 RtlReleasePebLock();
1249 }
1250 else
1251 {
1252 index -= TLS_MINIMUM_AVAILABLE;
1253 if (index >= 8 * sizeof(NtCurrentTeb()->Peb->TlsExpansionBitmapBits))
1254 return STATUS_INVALID_PARAMETER;
1255 RtlAcquirePebLock();
1256 for (entry = tls_links.Flink; entry != &tls_links; entry = entry->Flink)
1257 {
1258 TEB *teb = CONTAINING_RECORD(entry, TEB, TlsLinks);
1259 if (teb->TlsExpansionSlots) teb->TlsExpansionSlots[index] = 0;
1260 }
1261 RtlReleasePebLock();
1262 }
1263 return STATUS_SUCCESS;
1264 }
1265 FIXME( "ZeroTlsCell not supported on other threads\n" );
1266 return STATUS_NOT_IMPLEMENTED;
1267
1268 case ThreadImpersonationToken:
1269 {
1270 const HANDLE *phToken = data;
1271 if (length != sizeof(HANDLE)) return STATUS_INVALID_PARAMETER;
1272 TRACE("Setting ThreadImpersonationToken handle to %p\n", *phToken );
1273 SERVER_START_REQ( set_thread_info )
1274 {
1275 req->handle = wine_server_obj_handle( handle );
1276 req->token = wine_server_obj_handle( *phToken );
1277 req->mask = SET_THREAD_INFO_TOKEN;
1278 status = wine_server_call( req );
1279 }
1280 SERVER_END_REQ;
1281 }
1282 return status;
1283 case ThreadBasePriority:
1284 {
1285 const DWORD *pprio = data;
1286 if (length != sizeof(DWORD)) return STATUS_INVALID_PARAMETER;
1287 SERVER_START_REQ( set_thread_info )
1288 {
1289 req->handle = wine_server_obj_handle( handle );
1290 req->priority = *pprio;
1291 req->mask = SET_THREAD_INFO_PRIORITY;
1292 status = wine_server_call( req );
1293 }
1294 SERVER_END_REQ;
1295 }
1296 return status;
1297 case ThreadAffinityMask:
1298 {
1299 const ULONG_PTR affinity_mask = get_system_affinity_mask();
1300 ULONG_PTR req_aff;
1301
1302 if (length != sizeof(ULONG_PTR)) return STATUS_INVALID_PARAMETER;
1303 req_aff = *(const ULONG_PTR *)data;
1304 if ((ULONG)req_aff == ~0u) req_aff = affinity_mask;
1305 else if (req_aff & ~affinity_mask) return STATUS_INVALID_PARAMETER;
1306 else if (!req_aff) return STATUS_INVALID_PARAMETER;
1307 SERVER_START_REQ( set_thread_info )
1308 {
1309 req->handle = wine_server_obj_handle( handle );
1310 req->affinity = req_aff;
1311 req->mask = SET_THREAD_INFO_AFFINITY;
1312 status = wine_server_call( req );
1313 }
1314 SERVER_END_REQ;
1315 }
1316 return status;
1317 case ThreadHideFromDebugger:
1318 /* pretend the call succeeded to satisfy some code protectors */
1319 return STATUS_SUCCESS;
1320 case ThreadQuerySetWin32StartAddress:
1321 {
1322 const PRTL_THREAD_START_ROUTINE *entry = data;
1323 if (length != sizeof(PRTL_THREAD_START_ROUTINE)) return STATUS_INVALID_PARAMETER;
1324 SERVER_START_REQ( set_thread_info )
1325 {
1326 req->handle = wine_server_obj_handle( handle );
1327 req->mask = SET_THREAD_INFO_ENTRYPOINT;
1328 req->entry_point = wine_server_client_ptr( *entry );
1329 status = wine_server_call( req );
1330 }
1331 SERVER_END_REQ;
1332 }
1333 return status;
1334 case ThreadGroupInformation:
1335 {
1336 const ULONG_PTR affinity_mask = get_system_affinity_mask();
1337 const GROUP_AFFINITY *req_aff;
1338
1339 if (length != sizeof(*req_aff)) return STATUS_INVALID_PARAMETER;
1340 if (!data) return STATUS_ACCESS_VIOLATION;
1341 req_aff = data;
1342
1343 /* On Windows the request fails if the reserved fields are set */
1344 if (req_aff->Reserved[0] || req_aff->Reserved[1] || req_aff->Reserved[2])
1345 return STATUS_INVALID_PARAMETER;
1346
1347 /* Wine only supports max 64 processors */
1348 if (req_aff->Group) return STATUS_INVALID_PARAMETER;
1349 if (req_aff->Mask & ~affinity_mask) return STATUS_INVALID_PARAMETER;
1350 if (!req_aff->Mask) return STATUS_INVALID_PARAMETER;
1351 SERVER_START_REQ( set_thread_info )
1352 {
1353 req->handle = wine_server_obj_handle( handle );
1354 req->affinity = req_aff->Mask;
1355 req->mask = SET_THREAD_INFO_AFFINITY;
1356 status = wine_server_call( req );
1357 }
1358 SERVER_END_REQ;
1359 }
1360 return status;
1361 case ThreadBasicInformation:
1362 case ThreadTimes:
1363 case ThreadPriority:
1364 case ThreadDescriptorTableEntry:
1365 case ThreadEnableAlignmentFaultFixup:
1366 case ThreadEventPair_Reusable:
1367 case ThreadPerformanceCount:
1368 case ThreadAmILastThread:
1369 case ThreadIdealProcessor:
1370 case ThreadPriorityBoost:
1371 case ThreadSetTlsArrayAddress:
1372 case ThreadIsIoPending:
1373 default:
1374 FIXME( "info class %d not supported yet\n", class );
1375 return STATUS_NOT_IMPLEMENTED;
1376 }
1377 }
1378
1379 /******************************************************************************
1380 * NtGetCurrentProcessorNumber (NTDLL.@)
1381 *
1382 * Return the processor, on which the thread is running
1383 *
1384 */
1385 ULONG WINAPI NtGetCurrentProcessorNumber(void)
1386 {
1387 ULONG processor;
1388
1389 #if defined(__linux__) && defined(__NR_getcpu)
1390 int res = syscall(__NR_getcpu, &processor, NULL, NULL);
1391 if (res != -1) return processor;
1392 #endif
1393
1394 if (NtCurrentTeb()->Peb->NumberOfProcessors > 1)
1395 {
1396 ULONG_PTR thread_mask, processor_mask;
1397 NTSTATUS status;
1398
1399 status = NtQueryInformationThread(GetCurrentThread(), ThreadAffinityMask,
1400 &thread_mask, sizeof(thread_mask), NULL);
1401 if (status == STATUS_SUCCESS)
1402 {
1403 for (processor = 0; processor < NtCurrentTeb()->Peb->NumberOfProcessors; processor++)
1404 {
1405 processor_mask = (1 << processor);
1406 if (thread_mask & processor_mask)
1407 {
1408 if (thread_mask != processor_mask)
1409 FIXME("need multicore support (%d processors)\n",
1410 NtCurrentTeb()->Peb->NumberOfProcessors);
1411 return processor;
1412 }
1413 }
1414 }
1415 }
1416
1417 /* fallback to the first processor */
1418 return 0;
1419 }
Windows API implementation