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