?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
hidclass.sys: Correct input vs output parameters for IOCTL_HID_SET_FEATURE.
[wine.git] / dlls / ntdll / server.c
blob356d6319f1546287b5f3daa282a90637cd0b35c4
1 /*
2 * Wine server communication
3 *
4 * Copyright (C) 1998 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 <ctype.h>
26 #ifdef HAVE_DIRENT_H
27 # include <dirent.h>
28 #endif
29 #include <errno.h>
30 #include <fcntl.h>
31 #ifdef HAVE_LWP_H
32 #include <lwp.h>
33 #endif
34 #ifdef HAVE_PTHREAD_NP_H
35 # include <pthread_np.h>
36 #endif
37 #include <signal.h>
38 #include <stdarg.h>
39 #include <stdio.h>
40 #include <string.h>
41 #include <sys/types.h>
42 #ifdef HAVE_SYS_SOCKET_H
43 # include <sys/socket.h>
44 #endif
45 #ifdef HAVE_SYS_WAIT_H
46 #include <sys/wait.h>
47 #endif
48 #ifdef HAVE_SYS_UN_H
49 #include <sys/un.h>
50 #endif
51 #ifdef HAVE_SYS_MMAN_H
52 #include <sys/mman.h>
53 #endif
54 #ifdef HAVE_SYS_PRCTL_H
55 # include <sys/prctl.h>
56 #endif
57 #ifdef HAVE_SYS_STAT_H
58 # include <sys/stat.h>
59 #endif
60 #ifdef HAVE_SYS_SYSCALL_H
61 # include <sys/syscall.h>
62 #endif
63 #ifdef HAVE_SYS_UIO_H
64 #include <sys/uio.h>
65 #endif
66 #ifdef HAVE_SYS_UCONTEXT_H
67 # include <sys/ucontext.h>
68 #endif
69 #ifdef HAVE_SYS_THR_H
70 #include <sys/thr.h>
71 #endif
72 #ifdef HAVE_UNISTD_H
73 # include <unistd.h>
74 #endif
75
76 #include "ntstatus.h"
77 #define WIN32_NO_STATUS
78 #include "windef.h"
79 #include "winnt.h"
80 #include "wine/library.h"
81 #include "wine/server.h"
82 #include "wine/debug.h"
83 #include "ntdll_misc.h"
84
85 WINE_DEFAULT_DEBUG_CHANNEL(server);
86
87 /* Some versions of glibc don't define this */
88 #ifndef SCM_RIGHTS
89 #define SCM_RIGHTS 1
90 #endif
91
92 #ifndef MSG_CMSG_CLOEXEC
93 #define MSG_CMSG_CLOEXEC 0
94 #endif
95
96 #define SOCKETNAME "socket" /* name of the socket file */
97 #define LOCKNAME "lock" /* name of the lock file */
98
99 #ifdef __i386__
100 static const enum cpu_type client_cpu = CPU_x86;
101 #elif defined(__x86_64__)
102 static const enum cpu_type client_cpu = CPU_x86_64;
103 #elif defined(__powerpc__)
104 static const enum cpu_type client_cpu = CPU_POWERPC;
105 #elif defined(__arm__)
106 static const enum cpu_type client_cpu = CPU_ARM;
107 #elif defined(__aarch64__)
108 static const enum cpu_type client_cpu = CPU_ARM64;
109 #else
110 #error Unsupported CPU
111 #endif
112
113 unsigned int server_cpus = 0;
114 BOOL is_wow64 = FALSE;
115
116 timeout_t server_start_time = 0; /* time of server startup */
117
118 sigset_t server_block_set; /* signals to block during server calls */
119 static int fd_socket = -1; /* socket to exchange file descriptors with the server */
120 static pid_t server_pid;
121
122 static RTL_CRITICAL_SECTION fd_cache_section;
123 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
124 {
125 0, 0, &fd_cache_section,
126 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
127 0, 0, { (DWORD_PTR)(__FILE__ ": fd_cache_section") }
128 };
129 static RTL_CRITICAL_SECTION fd_cache_section = { &critsect_debug, -1, 0, 0, 0, 0 };
130
131 /* atomically exchange a 64-bit value */
132 static inline LONG64 interlocked_xchg64( LONG64 *dest, LONG64 val )
133 {
134 #ifdef _WIN64
135 return (LONG64)interlocked_xchg_ptr( (void **)dest, (void *)val );
136 #else
137 LONG64 tmp = *dest;
138 while (interlocked_cmpxchg64( dest, val, tmp ) != tmp) tmp = *dest;
139 return tmp;
140 #endif
141 }
142
143 #ifdef __GNUC__
144 static void fatal_error( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
145 static void fatal_perror( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
146 static void server_connect_error( const char *serverdir ) __attribute__((noreturn));
147 #endif
148
149 /* die on a fatal error; use only during initialization */
150 static void fatal_error( const char *err, ... )
151 {
152 va_list args;
153
154 va_start( args, err );
155 fprintf( stderr, "wine: " );
156 vfprintf( stderr, err, args );
157 va_end( args );
158 exit(1);
159 }
160
161 /* die on a fatal error; use only during initialization */
162 static void fatal_perror( const char *err, ... )
163 {
164 va_list args;
165
166 va_start( args, err );
167 fprintf( stderr, "wine: " );
168 vfprintf( stderr, err, args );
169 perror( " " );
170 va_end( args );
171 exit(1);
172 }
173
174
175 /***********************************************************************
176 * server_protocol_error
177 */
178 static DECLSPEC_NORETURN void server_protocol_error( const char *err, ... )
179 {
180 va_list args;
181
182 va_start( args, err );
183 fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
184 vfprintf( stderr, err, args );
185 va_end( args );
186 abort_thread(1);
187 }
188
189
190 /***********************************************************************
191 * server_protocol_perror
192 */
193 static DECLSPEC_NORETURN void server_protocol_perror( const char *err )
194 {
195 fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
196 perror( err );
197 abort_thread(1);
198 }
199
200
201 /***********************************************************************
202 * send_request
203 *
204 * Send a request to the server.
205 */
206 static unsigned int send_request( const struct __server_request_info *req )
207 {
208 unsigned int i;
209 int ret;
210
211 if (!req->u.req.request_header.request_size)
212 {
213 if ((ret = write( ntdll_get_thread_data()->request_fd, &req->u.req,
214 sizeof(req->u.req) )) == sizeof(req->u.req)) return STATUS_SUCCESS;
215
216 }
217 else
218 {
219 struct iovec vec[__SERVER_MAX_DATA+1];
220
221 vec[0].iov_base = (void *)&req->u.req;
222 vec[0].iov_len = sizeof(req->u.req);
223 for (i = 0; i < req->data_count; i++)
224 {
225 vec[i+1].iov_base = (void *)req->data[i].ptr;
226 vec[i+1].iov_len = req->data[i].size;
227 }
228 if ((ret = writev( ntdll_get_thread_data()->request_fd, vec, i+1 )) ==
229 req->u.req.request_header.request_size + sizeof(req->u.req)) return STATUS_SUCCESS;
230 }
231
232 if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
233 if (errno == EPIPE) abort_thread(0);
234 if (errno == EFAULT) return STATUS_ACCESS_VIOLATION;
235 server_protocol_perror( "write" );
236 }
237
238
239 /***********************************************************************
240 * read_reply_data
241 *
242 * Read data from the reply buffer; helper for wait_reply.
243 */
244 static void read_reply_data( void *buffer, size_t size )
245 {
246 int ret;
247
248 for (;;)
249 {
250 if ((ret = read( ntdll_get_thread_data()->reply_fd, buffer, size )) > 0)
251 {
252 if (!(size -= ret)) return;
253 buffer = (char *)buffer + ret;
254 continue;
255 }
256 if (!ret) break;
257 if (errno == EINTR) continue;
258 if (errno == EPIPE) break;
259 server_protocol_perror("read");
260 }
261 /* the server closed the connection; time to die... */
262 abort_thread(0);
263 }
264
265
266 /***********************************************************************
267 * wait_reply
268 *
269 * Wait for a reply from the server.
270 */
271 static inline unsigned int wait_reply( struct __server_request_info *req )
272 {
273 read_reply_data( &req->u.reply, sizeof(req->u.reply) );
274 if (req->u.reply.reply_header.reply_size)
275 read_reply_data( req->reply_data, req->u.reply.reply_header.reply_size );
276 return req->u.reply.reply_header.error;
277 }
278
279
280 /***********************************************************************
281 * wine_server_call (NTDLL.@)
282 *
283 * Perform a server call.
284 *
285 * PARAMS
286 * req_ptr [I/O] Function dependent data
287 *
288 * RETURNS
289 * Depends on server function being called, but usually an NTSTATUS code.
290 *
291 * NOTES
292 * Use the SERVER_START_REQ and SERVER_END_REQ to help you fill out the
293 * server request structure for the particular call. E.g:
294 *| SERVER_START_REQ( event_op )
295 *| {
296 *| req->handle = handle;
297 *| req->op = SET_EVENT;
298 *| ret = wine_server_call( req );
299 *| }
300 *| SERVER_END_REQ;
301 */
302 unsigned int wine_server_call( void *req_ptr )
303 {
304 struct __server_request_info * const req = req_ptr;
305 sigset_t old_set;
306 unsigned int ret;
307
308 pthread_sigmask( SIG_BLOCK, &server_block_set, &old_set );
309 ret = send_request( req );
310 if (!ret) ret = wait_reply( req );
311 pthread_sigmask( SIG_SETMASK, &old_set, NULL );
312 return ret;
313 }
314
315
316 /***********************************************************************
317 * server_enter_uninterrupted_section
318 */
319 void server_enter_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
320 {
321 pthread_sigmask( SIG_BLOCK, &server_block_set, sigset );
322 RtlEnterCriticalSection( cs );
323 }
324
325
326 /***********************************************************************
327 * server_leave_uninterrupted_section
328 */
329 void server_leave_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
330 {
331 RtlLeaveCriticalSection( cs );
332 pthread_sigmask( SIG_SETMASK, sigset, NULL );
333 }
334
335
336 /***********************************************************************
337 * wait_select_reply
338 *
339 * Wait for a reply on the waiting pipe of the current thread.
340 */
341 static int wait_select_reply( void *cookie )
342 {
343 int signaled;
344 struct wake_up_reply reply;
345 for (;;)
346 {
347 int ret;
348 ret = read( ntdll_get_thread_data()->wait_fd[0], &reply, sizeof(reply) );
349 if (ret == sizeof(reply))
350 {
351 if (!reply.cookie) abort_thread( reply.signaled ); /* thread got killed */
352 if (wine_server_get_ptr(reply.cookie) == cookie) return reply.signaled;
353 /* we stole another reply, wait for the real one */
354 signaled = wait_select_reply( cookie );
355 /* and now put the wrong one back in the pipe */
356 for (;;)
357 {
358 ret = write( ntdll_get_thread_data()->wait_fd[1], &reply, sizeof(reply) );
359 if (ret == sizeof(reply)) break;
360 if (ret >= 0) server_protocol_error( "partial wakeup write %d\n", ret );
361 if (errno == EINTR) continue;
362 server_protocol_perror("wakeup write");
363 }
364 return signaled;
365 }
366 if (ret >= 0) server_protocol_error( "partial wakeup read %d\n", ret );
367 if (errno == EINTR) continue;
368 server_protocol_perror("wakeup read");
369 }
370 }
371
372
373 /***********************************************************************
374 * invoke_apc
375 *
376 * Invoke a single APC. Return TRUE if a user APC has been run.
377 */
378 static BOOL invoke_apc( const apc_call_t *call, apc_result_t *result )
379 {
380 BOOL user_apc = FALSE;
381 SIZE_T size;
382 void *addr;
383
384 memset( result, 0, sizeof(*result) );
385
386 switch (call->type)
387 {
388 case APC_NONE:
389 break;
390 case APC_USER:
391 {
392 void (WINAPI *func)(ULONG_PTR,ULONG_PTR,ULONG_PTR) = wine_server_get_ptr( call->user.func );
393 func( call->user.args[0], call->user.args[1], call->user.args[2] );
394 user_apc = TRUE;
395 break;
396 }
397 case APC_TIMER:
398 {
399 void (WINAPI *func)(void*, unsigned int, unsigned int) = wine_server_get_ptr( call->timer.func );
400 func( wine_server_get_ptr( call->timer.arg ),
401 (DWORD)call->timer.time, (DWORD)(call->timer.time >> 32) );
402 user_apc = TRUE;
403 break;
404 }
405 case APC_ASYNC_IO:
406 {
407 void *apc = NULL, *arg = NULL;
408 IO_STATUS_BLOCK *iosb = wine_server_get_ptr( call->async_io.sb );
409 NTSTATUS (*func)(void *, IO_STATUS_BLOCK *, NTSTATUS, void **, void **) = wine_server_get_ptr( call->async_io.func );
410 result->type = call->type;
411 result->async_io.status = func( wine_server_get_ptr( call->async_io.user ),
412 iosb, call->async_io.status, &apc, &arg );
413 if (result->async_io.status != STATUS_PENDING)
414 {
415 result->async_io.total = iosb->Information;
416 result->async_io.apc = wine_server_client_ptr( apc );
417 result->async_io.arg = wine_server_client_ptr( arg );
418 }
419 break;
420 }
421 case APC_VIRTUAL_ALLOC:
422 result->type = call->type;
423 addr = wine_server_get_ptr( call->virtual_alloc.addr );
424 size = call->virtual_alloc.size;
425 if ((ULONG_PTR)addr == call->virtual_alloc.addr && size == call->virtual_alloc.size)
426 {
427 result->virtual_alloc.status = NtAllocateVirtualMemory( NtCurrentProcess(), &addr,
428 call->virtual_alloc.zero_bits, &size,
429 call->virtual_alloc.op_type,
430 call->virtual_alloc.prot );
431 result->virtual_alloc.addr = wine_server_client_ptr( addr );
432 result->virtual_alloc.size = size;
433 }
434 else result->virtual_alloc.status = STATUS_WORKING_SET_LIMIT_RANGE;
435 break;
436 case APC_VIRTUAL_FREE:
437 result->type = call->type;
438 addr = wine_server_get_ptr( call->virtual_free.addr );
439 size = call->virtual_free.size;
440 if ((ULONG_PTR)addr == call->virtual_free.addr && size == call->virtual_free.size)
441 {
442 result->virtual_free.status = NtFreeVirtualMemory( NtCurrentProcess(), &addr, &size,
443 call->virtual_free.op_type );
444 result->virtual_free.addr = wine_server_client_ptr( addr );
445 result->virtual_free.size = size;
446 }
447 else result->virtual_free.status = STATUS_INVALID_PARAMETER;
448 break;
449 case APC_VIRTUAL_QUERY:
450 {
451 MEMORY_BASIC_INFORMATION info;
452 result->type = call->type;
453 addr = wine_server_get_ptr( call->virtual_query.addr );
454 if ((ULONG_PTR)addr == call->virtual_query.addr)
455 result->virtual_query.status = NtQueryVirtualMemory( NtCurrentProcess(),
456 addr, MemoryBasicInformation, &info,
457 sizeof(info), NULL );
458 else
459 result->virtual_query.status = STATUS_WORKING_SET_LIMIT_RANGE;
460
461 if (result->virtual_query.status == STATUS_SUCCESS)
462 {
463 result->virtual_query.base = wine_server_client_ptr( info.BaseAddress );
464 result->virtual_query.alloc_base = wine_server_client_ptr( info.AllocationBase );
465 result->virtual_query.size = info.RegionSize;
466 result->virtual_query.prot = info.Protect;
467 result->virtual_query.alloc_prot = info.AllocationProtect;
468 result->virtual_query.state = info.State >> 12;
469 result->virtual_query.alloc_type = info.Type >> 16;
470 }
471 break;
472 }
473 case APC_VIRTUAL_PROTECT:
474 result->type = call->type;
475 addr = wine_server_get_ptr( call->virtual_protect.addr );
476 size = call->virtual_protect.size;
477 if ((ULONG_PTR)addr == call->virtual_protect.addr && size == call->virtual_protect.size)
478 {
479 result->virtual_protect.status = NtProtectVirtualMemory( NtCurrentProcess(), &addr, &size,
480 call->virtual_protect.prot,
481 &result->virtual_protect.prot );
482 result->virtual_protect.addr = wine_server_client_ptr( addr );
483 result->virtual_protect.size = size;
484 }
485 else result->virtual_protect.status = STATUS_INVALID_PARAMETER;
486 break;
487 case APC_VIRTUAL_FLUSH:
488 result->type = call->type;
489 addr = wine_server_get_ptr( call->virtual_flush.addr );
490 size = call->virtual_flush.size;
491 if ((ULONG_PTR)addr == call->virtual_flush.addr && size == call->virtual_flush.size)
492 {
493 result->virtual_flush.status = NtFlushVirtualMemory( NtCurrentProcess(),
494 (const void **)&addr, &size, 0 );
495 result->virtual_flush.addr = wine_server_client_ptr( addr );
496 result->virtual_flush.size = size;
497 }
498 else result->virtual_flush.status = STATUS_INVALID_PARAMETER;
499 break;
500 case APC_VIRTUAL_LOCK:
501 result->type = call->type;
502 addr = wine_server_get_ptr( call->virtual_lock.addr );
503 size = call->virtual_lock.size;
504 if ((ULONG_PTR)addr == call->virtual_lock.addr && size == call->virtual_lock.size)
505 {
506 result->virtual_lock.status = NtLockVirtualMemory( NtCurrentProcess(), &addr, &size, 0 );
507 result->virtual_lock.addr = wine_server_client_ptr( addr );
508 result->virtual_lock.size = size;
509 }
510 else result->virtual_lock.status = STATUS_INVALID_PARAMETER;
511 break;
512 case APC_VIRTUAL_UNLOCK:
513 result->type = call->type;
514 addr = wine_server_get_ptr( call->virtual_unlock.addr );
515 size = call->virtual_unlock.size;
516 if ((ULONG_PTR)addr == call->virtual_unlock.addr && size == call->virtual_unlock.size)
517 {
518 result->virtual_unlock.status = NtUnlockVirtualMemory( NtCurrentProcess(), &addr, &size, 0 );
519 result->virtual_unlock.addr = wine_server_client_ptr( addr );
520 result->virtual_unlock.size = size;
521 }
522 else result->virtual_unlock.status = STATUS_INVALID_PARAMETER;
523 break;
524 case APC_MAP_VIEW:
525 result->type = call->type;
526 addr = wine_server_get_ptr( call->map_view.addr );
527 size = call->map_view.size;
528 if ((ULONG_PTR)addr == call->map_view.addr && size == call->map_view.size)
529 {
530 LARGE_INTEGER offset;
531 offset.QuadPart = call->map_view.offset;
532 result->map_view.status = NtMapViewOfSection( wine_server_ptr_handle(call->map_view.handle),
533 NtCurrentProcess(), &addr,
534 call->map_view.zero_bits, 0,
535 &offset, &size, ViewShare,
536 call->map_view.alloc_type, call->map_view.prot );
537 result->map_view.addr = wine_server_client_ptr( addr );
538 result->map_view.size = size;
539 }
540 else result->map_view.status = STATUS_INVALID_PARAMETER;
541 NtClose( wine_server_ptr_handle(call->map_view.handle) );
542 break;
543 case APC_UNMAP_VIEW:
544 result->type = call->type;
545 addr = wine_server_get_ptr( call->unmap_view.addr );
546 if ((ULONG_PTR)addr == call->unmap_view.addr)
547 result->unmap_view.status = NtUnmapViewOfSection( NtCurrentProcess(), addr );
548 else
549 result->unmap_view.status = STATUS_INVALID_PARAMETER;
550 break;
551 case APC_CREATE_THREAD:
552 {
553 CLIENT_ID id;
554 HANDLE handle;
555 SIZE_T reserve = call->create_thread.reserve;
556 SIZE_T commit = call->create_thread.commit;
557 void *func = wine_server_get_ptr( call->create_thread.func );
558 void *arg = wine_server_get_ptr( call->create_thread.arg );
559
560 result->type = call->type;
561 if (reserve == call->create_thread.reserve && commit == call->create_thread.commit &&
562 (ULONG_PTR)func == call->create_thread.func && (ULONG_PTR)arg == call->create_thread.arg)
563 {
564 result->create_thread.status = RtlCreateUserThread( NtCurrentProcess(), NULL,
565 call->create_thread.suspend, NULL,
566 reserve, commit, func, arg, &handle, &id );
567 result->create_thread.handle = wine_server_obj_handle( handle );
568 result->create_thread.tid = HandleToULong(id.UniqueThread);
569 }
570 else result->create_thread.status = STATUS_INVALID_PARAMETER;
571 break;
572 }
573 default:
574 server_protocol_error( "get_apc_request: bad type %d\n", call->type );
575 break;
576 }
577 return user_apc;
578 }
579
580
581 /***********************************************************************
582 * server_select
583 */
584 unsigned int server_select( const select_op_t *select_op, data_size_t size, UINT flags,
585 const LARGE_INTEGER *timeout )
586 {
587 unsigned int ret;
588 int cookie;
589 BOOL user_apc = FALSE;
590 obj_handle_t apc_handle = 0;
591 apc_call_t call;
592 apc_result_t result;
593 timeout_t abs_timeout = timeout ? timeout->QuadPart : TIMEOUT_INFINITE;
594
595 memset( &result, 0, sizeof(result) );
596
597 for (;;)
598 {
599 SERVER_START_REQ( select )
600 {
601 req->flags = flags;
602 req->cookie = wine_server_client_ptr( &cookie );
603 req->prev_apc = apc_handle;
604 req->timeout = abs_timeout;
605 wine_server_add_data( req, &result, sizeof(result) );
606 wine_server_add_data( req, select_op, size );
607 ret = wine_server_call( req );
608 abs_timeout = reply->timeout;
609 apc_handle = reply->apc_handle;
610 call = reply->call;
611 }
612 SERVER_END_REQ;
613 if (ret == STATUS_PENDING) ret = wait_select_reply( &cookie );
614 if (ret != STATUS_USER_APC) break;
615 if (invoke_apc( &call, &result ))
616 {
617 /* if we ran a user apc we have to check once more if additional apcs are queued,
618 * but we don't want to wait */
619 abs_timeout = 0;
620 user_apc = TRUE;
621 size = 0;
622 }
623
624 /* don't signal multiple times */
625 if (size >= sizeof(select_op->signal_and_wait) && select_op->op == SELECT_SIGNAL_AND_WAIT)
626 size = offsetof( select_op_t, signal_and_wait.signal );
627 }
628
629 if (ret == STATUS_TIMEOUT && user_apc) ret = STATUS_USER_APC;
630
631 /* A test on Windows 2000 shows that Windows always yields during
632 a wait, but a wait that is hit by an event gets a priority
633 boost as well. This seems to model that behavior the closest. */
634 if (ret == STATUS_TIMEOUT) NtYieldExecution();
635
636 return ret;
637 }
638
639
640 /***********************************************************************
641 * server_queue_process_apc
642 */
643 unsigned int server_queue_process_apc( HANDLE process, const apc_call_t *call, apc_result_t *result )
644 {
645 for (;;)
646 {
647 unsigned int ret;
648 HANDLE handle = 0;
649 BOOL self = FALSE;
650
651 SERVER_START_REQ( queue_apc )
652 {
653 req->handle = wine_server_obj_handle( process );
654 req->call = *call;
655 if (!(ret = wine_server_call( req )))
656 {
657 handle = wine_server_ptr_handle( reply->handle );
658 self = reply->self;
659 }
660 }
661 SERVER_END_REQ;
662 if (ret != STATUS_SUCCESS) return ret;
663
664 if (self)
665 {
666 invoke_apc( call, result );
667 }
668 else
669 {
670 NtWaitForSingleObject( handle, FALSE, NULL );
671
672 SERVER_START_REQ( get_apc_result )
673 {
674 req->handle = wine_server_obj_handle( handle );
675 if (!(ret = wine_server_call( req ))) *result = reply->result;
676 }
677 SERVER_END_REQ;
678
679 if (!ret && result->type == APC_NONE) continue; /* APC didn't run, try again */
680 }
681 return ret;
682 }
683 }
684
685
686 /***********************************************************************
687 * wine_server_send_fd (NTDLL.@)
688 *
689 * Send a file descriptor to the server.
690 *
691 * PARAMS
692 * fd [I] file descriptor to send
693 *
694 * RETURNS
695 * nothing
696 */
697 void CDECL wine_server_send_fd( int fd )
698 {
699 struct send_fd data;
700 struct msghdr msghdr;
701 struct iovec vec;
702 int ret;
703
704 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
705 msghdr.msg_accrights = (void *)&fd;
706 msghdr.msg_accrightslen = sizeof(fd);
707 #else /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
708 char cmsg_buffer[256];
709 struct cmsghdr *cmsg;
710 msghdr.msg_control = cmsg_buffer;
711 msghdr.msg_controllen = sizeof(cmsg_buffer);
712 msghdr.msg_flags = 0;
713 cmsg = CMSG_FIRSTHDR( &msghdr );
714 cmsg->cmsg_len = CMSG_LEN( sizeof(fd) );
715 cmsg->cmsg_level = SOL_SOCKET;
716 cmsg->cmsg_type = SCM_RIGHTS;
717 *(int *)CMSG_DATA(cmsg) = fd;
718 msghdr.msg_controllen = cmsg->cmsg_len;
719 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
720
721 msghdr.msg_name = NULL;
722 msghdr.msg_namelen = 0;
723 msghdr.msg_iov = &vec;
724 msghdr.msg_iovlen = 1;
725
726 vec.iov_base = (void *)&data;
727 vec.iov_len = sizeof(data);
728
729 data.tid = GetCurrentThreadId();
730 data.fd = fd;
731
732 for (;;)
733 {
734 if ((ret = sendmsg( fd_socket, &msghdr, 0 )) == sizeof(data)) return;
735 if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
736 if (errno == EINTR) continue;
737 if (errno == EPIPE) abort_thread(0);
738 server_protocol_perror( "sendmsg" );
739 }
740 }
741
742
743 /***********************************************************************
744 * receive_fd
745 *
746 * Receive a file descriptor passed from the server.
747 */
748 static int receive_fd( obj_handle_t *handle )
749 {
750 struct iovec vec;
751 struct msghdr msghdr;
752 int ret, fd = -1;
753
754 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
755 msghdr.msg_accrights = (void *)&fd;
756 msghdr.msg_accrightslen = sizeof(fd);
757 #else /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
758 char cmsg_buffer[256];
759 msghdr.msg_control = cmsg_buffer;
760 msghdr.msg_controllen = sizeof(cmsg_buffer);
761 msghdr.msg_flags = 0;
762 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
763
764 msghdr.msg_name = NULL;
765 msghdr.msg_namelen = 0;
766 msghdr.msg_iov = &vec;
767 msghdr.msg_iovlen = 1;
768 vec.iov_base = (void *)handle;
769 vec.iov_len = sizeof(*handle);
770
771 for (;;)
772 {
773 if ((ret = recvmsg( fd_socket, &msghdr, MSG_CMSG_CLOEXEC )) > 0)
774 {
775 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
776 struct cmsghdr *cmsg;
777 for (cmsg = CMSG_FIRSTHDR( &msghdr ); cmsg; cmsg = CMSG_NXTHDR( &msghdr, cmsg ))
778 {
779 if (cmsg->cmsg_level != SOL_SOCKET) continue;
780 if (cmsg->cmsg_type == SCM_RIGHTS) fd = *(int *)CMSG_DATA(cmsg);
781 #ifdef SCM_CREDENTIALS
782 else if (cmsg->cmsg_type == SCM_CREDENTIALS)
783 {
784 struct ucred *ucred = (struct ucred *)CMSG_DATA(cmsg);
785 server_pid = ucred->pid;
786 }
787 #endif
788 }
789 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
790 if (fd != -1) fcntl( fd, F_SETFD, FD_CLOEXEC ); /* in case MSG_CMSG_CLOEXEC is not supported */
791 return fd;
792 }
793 if (!ret) break;
794 if (errno == EINTR) continue;
795 if (errno == EPIPE) break;
796 server_protocol_perror("recvmsg");
797 }
798 /* the server closed the connection; time to die... */
799 abort_thread(0);
800 }
801
802
803 /***********************************************************************/
804 /* fd cache support */
805
806 #include "pshpack1.h"
807 union fd_cache_entry
808 {
809 LONG64 data;
810 struct
811 {
812 int fd;
813 enum server_fd_type type : 5;
814 unsigned int access : 3;
815 unsigned int options : 24;
816 } s;
817 };
818 #include "poppack.h"
819
820 C_ASSERT( sizeof(union fd_cache_entry) == sizeof(LONG64) );
821
822 #define FD_CACHE_BLOCK_SIZE (65536 / sizeof(union fd_cache_entry))
823 #define FD_CACHE_ENTRIES 128
824
825 static union fd_cache_entry *fd_cache[FD_CACHE_ENTRIES];
826 static union fd_cache_entry fd_cache_initial_block[FD_CACHE_BLOCK_SIZE];
827
828 static inline unsigned int handle_to_index( HANDLE handle, unsigned int *entry )
829 {
830 unsigned int idx = (wine_server_obj_handle(handle) >> 2) - 1;
831 *entry = idx / FD_CACHE_BLOCK_SIZE;
832 return idx % FD_CACHE_BLOCK_SIZE;
833 }
834
835
836 /***********************************************************************
837 * add_fd_to_cache
838 *
839 * Caller must hold fd_cache_section.
840 */
841 static BOOL add_fd_to_cache( HANDLE handle, int fd, enum server_fd_type type,
842 unsigned int access, unsigned int options )
843 {
844 unsigned int entry, idx = handle_to_index( handle, &entry );
845 union fd_cache_entry cache;
846
847 if (entry >= FD_CACHE_ENTRIES)
848 {
849 FIXME( "too many allocated handles, not caching %p\n", handle );
850 return FALSE;
851 }
852
853 if (!fd_cache[entry]) /* do we need to allocate a new block of entries? */
854 {
855 if (!entry) fd_cache[0] = fd_cache_initial_block;
856 else
857 {
858 void *ptr = wine_anon_mmap( NULL, FD_CACHE_BLOCK_SIZE * sizeof(union fd_cache_entry),
859 PROT_READ | PROT_WRITE, 0 );
860 if (ptr == MAP_FAILED) return FALSE;
861 fd_cache[entry] = ptr;
862 }
863 }
864
865 /* store fd+1 so that 0 can be used as the unset value */
866 cache.s.fd = fd + 1;
867 cache.s.type = type;
868 cache.s.access = access;
869 cache.s.options = options;
870 cache.data = interlocked_xchg64( &fd_cache[entry][idx].data, cache.data );
871 assert( !cache.s.fd );
872 return TRUE;
873 }
874
875
876 /***********************************************************************
877 * get_cached_fd
878 */
879 static inline int get_cached_fd( HANDLE handle, enum server_fd_type *type,
880 unsigned int *access, unsigned int *options )
881 {
882 unsigned int entry, idx = handle_to_index( handle, &entry );
883 int fd = -1;
884
885 if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
886 {
887 union fd_cache_entry cache;
888 cache.data = interlocked_cmpxchg64( &fd_cache[entry][idx].data, 0, 0 );
889 fd = cache.s.fd - 1;
890 if (type) *type = cache.s.type;
891 if (access) *access = cache.s.access;
892 if (options) *options = cache.s.options;
893 }
894 return fd;
895 }
896
897
898 /***********************************************************************
899 * server_remove_fd_from_cache
900 */
901 int server_remove_fd_from_cache( HANDLE handle )
902 {
903 unsigned int entry, idx = handle_to_index( handle, &entry );
904 int fd = -1;
905
906 if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
907 {
908 union fd_cache_entry cache;
909 cache.data = interlocked_xchg64( &fd_cache[entry][idx].data, 0 );
910 fd = cache.s.fd - 1;
911 }
912
913 return fd;
914 }
915
916
917 /***********************************************************************
918 * server_get_unix_fd
919 *
920 * The returned unix_fd should be closed iff needs_close is non-zero.
921 */
922 int server_get_unix_fd( HANDLE handle, unsigned int wanted_access, int *unix_fd,
923 int *needs_close, enum server_fd_type *type, unsigned int *options )
924 {
925 sigset_t sigset;
926 obj_handle_t fd_handle;
927 int ret = 0, fd;
928 unsigned int access = 0;
929
930 *unix_fd = -1;
931 *needs_close = 0;
932 wanted_access &= FILE_READ_DATA | FILE_WRITE_DATA | FILE_APPEND_DATA;
933
934 fd = get_cached_fd( handle, type, &access, options );
935 if (fd != -1) goto done;
936
937 server_enter_uninterrupted_section( &fd_cache_section, &sigset );
938 fd = get_cached_fd( handle, type, &access, options );
939 if (fd == -1)
940 {
941 SERVER_START_REQ( get_handle_fd )
942 {
943 req->handle = wine_server_obj_handle( handle );
944 if (!(ret = wine_server_call( req )))
945 {
946 if (type) *type = reply->type;
947 if (options) *options = reply->options;
948 access = reply->access;
949 if ((fd = receive_fd( &fd_handle )) != -1)
950 {
951 assert( wine_server_ptr_handle(fd_handle) == handle );
952 *needs_close = (!reply->cacheable ||
953 !add_fd_to_cache( handle, fd, reply->type,
954 reply->access, reply->options ));
955 }
956 else ret = STATUS_TOO_MANY_OPENED_FILES;
957 }
958 }
959 SERVER_END_REQ;
960 }
961 server_leave_uninterrupted_section( &fd_cache_section, &sigset );
962
963 done:
964 if (!ret && ((access & wanted_access) != wanted_access))
965 {
966 ret = STATUS_ACCESS_DENIED;
967 if (*needs_close) close( fd );
968 }
969 if (!ret) *unix_fd = fd;
970 return ret;
971 }
972
973
974 /***********************************************************************
975 * wine_server_fd_to_handle (NTDLL.@)
976 *
977 * Allocate a file handle for a Unix file descriptor.
978 *
979 * PARAMS
980 * fd [I] Unix file descriptor.
981 * access [I] Win32 access flags.
982 * attributes [I] Object attributes.
983 * handle [O] Address where Wine file handle will be stored.
984 *
985 * RETURNS
986 * NTSTATUS code
987 */
988 int CDECL wine_server_fd_to_handle( int fd, unsigned int access, unsigned int attributes, HANDLE *handle )
989 {
990 int ret;
991
992 *handle = 0;
993 wine_server_send_fd( fd );
994
995 SERVER_START_REQ( alloc_file_handle )
996 {
997 req->access = access;
998 req->attributes = attributes;
999 req->fd = fd;
1000 if (!(ret = wine_server_call( req ))) *handle = wine_server_ptr_handle( reply->handle );
1001 }
1002 SERVER_END_REQ;
1003 return ret;
1004 }
1005
1006
1007 /***********************************************************************
1008 * wine_server_handle_to_fd (NTDLL.@)
1009 *
1010 * Retrieve the file descriptor corresponding to a file handle.
1011 *
1012 * PARAMS
1013 * handle [I] Wine file handle.
1014 * access [I] Win32 file access rights requested.
1015 * unix_fd [O] Address where Unix file descriptor will be stored.
1016 * options [O] Address where the file open options will be stored. Optional.
1017 *
1018 * RETURNS
1019 * NTSTATUS code
1020 */
1021 int CDECL wine_server_handle_to_fd( HANDLE handle, unsigned int access, int *unix_fd,
1022 unsigned int *options )
1023 {
1024 int needs_close, ret = server_get_unix_fd( handle, access, unix_fd, &needs_close, NULL, options );
1025
1026 if (!ret && !needs_close)
1027 {
1028 if ((*unix_fd = dup(*unix_fd)) == -1) ret = FILE_GetNtStatus();
1029 }
1030 return ret;
1031 }
1032
1033
1034 /***********************************************************************
1035 * wine_server_release_fd (NTDLL.@)
1036 *
1037 * Release the Unix file descriptor returned by wine_server_handle_to_fd.
1038 *
1039 * PARAMS
1040 * handle [I] Wine file handle.
1041 * unix_fd [I] Unix file descriptor to release.
1042 *
1043 * RETURNS
1044 * nothing
1045 */
1046 void CDECL wine_server_release_fd( HANDLE handle, int unix_fd )
1047 {
1048 close( unix_fd );
1049 }
1050
1051
1052 /***********************************************************************
1053 * server_pipe
1054 *
1055 * Create a pipe for communicating with the server.
1056 */
1057 int server_pipe( int fd[2] )
1058 {
1059 int ret;
1060 #ifdef HAVE_PIPE2
1061 static BOOL have_pipe2 = TRUE;
1062
1063 if (have_pipe2)
1064 {
1065 if (!(ret = pipe2( fd, O_CLOEXEC ))) return ret;
1066 if (errno == ENOSYS || errno == EINVAL) have_pipe2 = FALSE; /* don't try again */
1067 }
1068 #endif
1069 if (!(ret = pipe( fd )))
1070 {
1071 fcntl( fd[0], F_SETFD, FD_CLOEXEC );
1072 fcntl( fd[1], F_SETFD, FD_CLOEXEC );
1073 }
1074 return ret;
1075 }
1076
1077
1078 /***********************************************************************
1079 * start_server
1080 *
1081 * Start a new wine server.
1082 */
1083 static void start_server(void)
1084 {
1085 static BOOL started; /* we only try once */
1086 char *argv[3];
1087 static char wineserver[] = "server/wineserver";
1088 static char debug[] = "-d";
1089
1090 if (!started)
1091 {
1092 int status;
1093 int pid = fork();
1094 if (pid == -1) fatal_perror( "fork" );
1095 if (!pid)
1096 {
1097 argv[0] = wineserver;
1098 argv[1] = TRACE_ON(server) ? debug : NULL;
1099 argv[2] = NULL;
1100 wine_exec_wine_binary( argv[0], argv, getenv("WINESERVER") );
1101 fatal_error( "could not exec wineserver\n" );
1102 }
1103 waitpid( pid, &status, 0 );
1104 status = WIFEXITED(status) ? WEXITSTATUS(status) : 1;
1105 if (status == 2) return; /* server lock held by someone else, will retry later */
1106 if (status) exit(status); /* server failed */
1107 started = TRUE;
1108 }
1109 }
1110
1111
1112 /***********************************************************************
1113 * setup_config_dir
1114 *
1115 * Setup the wine configuration dir.
1116 */
1117 static void setup_config_dir(void)
1118 {
1119 const char *p, *config_dir = wine_get_config_dir();
1120
1121 if (chdir( config_dir ) == -1)
1122 {
1123 if (errno != ENOENT) fatal_perror( "chdir to %s\n", config_dir );
1124
1125 if ((p = strrchr( config_dir, '/' )) && p != config_dir)
1126 {
1127 struct stat st;
1128 char *tmp_dir;
1129
1130 if (!(tmp_dir = malloc( p + 1 - config_dir ))) fatal_error( "out of memory\n" );
1131 memcpy( tmp_dir, config_dir, p - config_dir );
1132 tmp_dir[p - config_dir] = 0;
1133 if (!stat( tmp_dir, &st ) && st.st_uid != getuid())
1134 fatal_error( "'%s' is not owned by you, refusing to create a configuration directory there\n",
1135 tmp_dir );
1136 free( tmp_dir );
1137 }
1138
1139 mkdir( config_dir, 0777 );
1140 if (chdir( config_dir ) == -1) fatal_perror( "chdir to %s\n", config_dir );
1141
1142 if ((p = getenv( "WINEARCH" )) && !strcmp( p, "win32" ))
1143 {
1144 /* force creation of a 32-bit prefix */
1145 int fd = open( "system.reg", O_WRONLY | O_CREAT | O_EXCL, 0666 );
1146 if (fd != -1)
1147 {
1148 static const char regfile[] = "WINE REGISTRY Version 2\n\n#arch=win32\n";
1149 write( fd, regfile, sizeof(regfile) - 1 );
1150 close( fd );
1151 }
1152 }
1153 MESSAGE( "wine: created the configuration directory '%s'\n", config_dir );
1154 }
1155
1156 if (mkdir( "dosdevices", 0777 ) == -1)
1157 {
1158 if (errno == EEXIST) return;
1159 fatal_perror( "cannot create %s/dosdevices\n", config_dir );
1160 }
1161
1162 /* create the drive symlinks */
1163
1164 mkdir( "drive_c", 0777 );
1165 symlink( "../drive_c", "dosdevices/c:" );
1166 symlink( "/", "dosdevices/z:" );
1167 }
1168
1169
1170 /***********************************************************************
1171 * server_connect_error
1172 *
1173 * Try to display a meaningful explanation of why we couldn't connect
1174 * to the server.
1175 */
1176 static void server_connect_error( const char *serverdir )
1177 {
1178 int fd;
1179 struct flock fl;
1180
1181 if ((fd = open( LOCKNAME, O_WRONLY )) == -1)
1182 fatal_error( "for some mysterious reason, the wine server never started.\n" );
1183
1184 fl.l_type = F_WRLCK;
1185 fl.l_whence = SEEK_SET;
1186 fl.l_start = 0;
1187 fl.l_len = 1;
1188 if (fcntl( fd, F_GETLK, &fl ) != -1)
1189 {
1190 if (fl.l_type == F_WRLCK) /* the file is locked */
1191 fatal_error( "a wine server seems to be running, but I cannot connect to it.\n"
1192 " You probably need to kill that process (it might be pid %d).\n",
1193 (int)fl.l_pid );
1194 fatal_error( "for some mysterious reason, the wine server failed to run.\n" );
1195 }
1196 fatal_error( "the file system of '%s' doesn't support locks,\n"
1197 " and there is a 'socket' file in that directory that prevents wine from starting.\n"
1198 " You should make sure no wine server is running, remove that file and try again.\n",
1199 serverdir );
1200 }
1201
1202
1203 /***********************************************************************
1204 * server_connect
1205 *
1206 * Attempt to connect to an existing server socket.
1207 * We need to be in the server directory already.
1208 */
1209 static int server_connect(void)
1210 {
1211 const char *serverdir;
1212 struct sockaddr_un addr;
1213 struct stat st;
1214 int s, slen, retry, fd_cwd;
1215
1216 /* retrieve the current directory */
1217 fd_cwd = open( ".", O_RDONLY );
1218 if (fd_cwd != -1) fcntl( fd_cwd, F_SETFD, 1 ); /* set close on exec flag */
1219
1220 setup_config_dir();
1221 serverdir = wine_get_server_dir();
1222
1223 /* chdir to the server directory */
1224 if (chdir( serverdir ) == -1)
1225 {
1226 if (errno != ENOENT) fatal_perror( "chdir to %s", serverdir );
1227 start_server();
1228 if (chdir( serverdir ) == -1) fatal_perror( "chdir to %s", serverdir );
1229 }
1230
1231 /* make sure we are at the right place */
1232 if (stat( ".", &st ) == -1) fatal_perror( "stat %s", serverdir );
1233 if (st.st_uid != getuid()) fatal_error( "'%s' is not owned by you\n", serverdir );
1234 if (st.st_mode & 077) fatal_error( "'%s' must not be accessible by other users\n", serverdir );
1235
1236 for (retry = 0; retry < 6; retry++)
1237 {
1238 /* if not the first try, wait a bit to leave the previous server time to exit */
1239 if (retry)
1240 {
1241 usleep( 100000 * retry * retry );
1242 start_server();
1243 if (lstat( SOCKETNAME, &st ) == -1) continue; /* still no socket, wait a bit more */
1244 }
1245 else if (lstat( SOCKETNAME, &st ) == -1) /* check for an already existing socket */
1246 {
1247 if (errno != ENOENT) fatal_perror( "lstat %s/%s", serverdir, SOCKETNAME );
1248 start_server();
1249 if (lstat( SOCKETNAME, &st ) == -1) continue; /* still no socket, wait a bit more */
1250 }
1251
1252 /* make sure the socket is sane (ISFIFO needed for Solaris) */
1253 if (!S_ISSOCK(st.st_mode) && !S_ISFIFO(st.st_mode))
1254 fatal_error( "'%s/%s' is not a socket\n", serverdir, SOCKETNAME );
1255 if (st.st_uid != getuid())
1256 fatal_error( "'%s/%s' is not owned by you\n", serverdir, SOCKETNAME );
1257
1258 /* try to connect to it */
1259 addr.sun_family = AF_UNIX;
1260 strcpy( addr.sun_path, SOCKETNAME );
1261 slen = sizeof(addr) - sizeof(addr.sun_path) + strlen(addr.sun_path) + 1;
1262 #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
1263 addr.sun_len = slen;
1264 #endif
1265 if ((s = socket( AF_UNIX, SOCK_STREAM, 0 )) == -1) fatal_perror( "socket" );
1266 #ifdef SO_PASSCRED
1267 else
1268 {
1269 int enable = 1;
1270 setsockopt( s, SOL_SOCKET, SO_PASSCRED, &enable, sizeof(enable) );
1271 }
1272 #endif
1273 if (connect( s, (struct sockaddr *)&addr, slen ) != -1)
1274 {
1275 /* switch back to the starting directory */
1276 if (fd_cwd != -1)
1277 {
1278 fchdir( fd_cwd );
1279 close( fd_cwd );
1280 }
1281 fcntl( s, F_SETFD, 1 ); /* set close on exec flag */
1282 return s;
1283 }
1284 close( s );
1285 }
1286 server_connect_error( serverdir );
1287 }
1288
1289
1290 #ifdef __APPLE__
1291 #include <mach/mach.h>
1292 #include <mach/mach_error.h>
1293 #include <servers/bootstrap.h>
1294
1295 /* send our task port to the server */
1296 static void send_server_task_port(void)
1297 {
1298 mach_port_t bootstrap_port, wineserver_port;
1299 kern_return_t kret;
1300
1301 struct {
1302 mach_msg_header_t header;
1303 mach_msg_body_t body;
1304 mach_msg_port_descriptor_t task_port;
1305 } msg;
1306
1307 if (task_get_bootstrap_port(mach_task_self(), &bootstrap_port) != KERN_SUCCESS) return;
1308
1309 kret = bootstrap_look_up(bootstrap_port, (char*)wine_get_server_dir(), &wineserver_port);
1310 if (kret != KERN_SUCCESS)
1311 fatal_error( "cannot find the server port: 0x%08x\n", kret );
1312
1313 mach_port_deallocate(mach_task_self(), bootstrap_port);
1314
1315 msg.header.msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, 0) | MACH_MSGH_BITS_COMPLEX;
1316 msg.header.msgh_size = sizeof(msg);
1317 msg.header.msgh_remote_port = wineserver_port;
1318 msg.header.msgh_local_port = MACH_PORT_NULL;
1319
1320 msg.body.msgh_descriptor_count = 1;
1321 msg.task_port.name = mach_task_self();
1322 msg.task_port.disposition = MACH_MSG_TYPE_COPY_SEND;
1323 msg.task_port.type = MACH_MSG_PORT_DESCRIPTOR;
1324
1325 kret = mach_msg_send(&msg.header);
1326 if (kret != KERN_SUCCESS)
1327 server_protocol_error( "mach_msg_send failed: 0x%08x\n", kret );
1328
1329 mach_port_deallocate(mach_task_self(), wineserver_port);
1330 }
1331 #endif /* __APPLE__ */
1332
1333
1334 /***********************************************************************
1335 * get_unix_tid
1336 *
1337 * Retrieve the Unix tid to use on the server side for the current thread.
1338 */
1339 static int get_unix_tid(void)
1340 {
1341 int ret = -1;
1342 #ifdef HAVE_PTHREAD_GETTHREADID_NP
1343 ret = pthread_getthreadid_np();
1344 #elif defined(linux)
1345 ret = syscall( __NR_gettid );
1346 #elif defined(__sun)
1347 ret = pthread_self();
1348 #elif defined(__APPLE__)
1349 ret = mach_thread_self();
1350 mach_port_deallocate(mach_task_self(), ret);
1351 #elif defined(__NetBSD__)
1352 ret = _lwp_self();
1353 #elif defined(__FreeBSD__)
1354 long lwpid;
1355 thr_self( &lwpid );
1356 ret = lwpid;
1357 #elif defined(__DragonFly__)
1358 ret = lwp_gettid();
1359 #endif
1360 return ret;
1361 }
1362
1363
1364 /***********************************************************************
1365 * server_init_process
1366 *
1367 * Start the server and create the initial socket pair.
1368 */
1369 void server_init_process(void)
1370 {
1371 obj_handle_t version;
1372 const char *env_socket = getenv( "WINESERVERSOCKET" );
1373
1374 server_pid = -1;
1375 if (env_socket)
1376 {
1377 fd_socket = atoi( env_socket );
1378 if (fcntl( fd_socket, F_SETFD, 1 ) == -1)
1379 fatal_perror( "Bad server socket %d", fd_socket );
1380 unsetenv( "WINESERVERSOCKET" );
1381 }
1382 else fd_socket = server_connect();
1383
1384 /* setup the signal mask */
1385 sigemptyset( &server_block_set );
1386 sigaddset( &server_block_set, SIGALRM );
1387 sigaddset( &server_block_set, SIGIO );
1388 sigaddset( &server_block_set, SIGINT );
1389 sigaddset( &server_block_set, SIGHUP );
1390 sigaddset( &server_block_set, SIGUSR1 );
1391 sigaddset( &server_block_set, SIGUSR2 );
1392 sigaddset( &server_block_set, SIGCHLD );
1393 pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
1394
1395 /* receive the first thread request fd on the main socket */
1396 ntdll_get_thread_data()->request_fd = receive_fd( &version );
1397
1398 #ifdef SO_PASSCRED
1399 /* now that we hopefully received the server_pid, disable SO_PASSCRED */
1400 {
1401 int enable = 0;
1402 setsockopt( fd_socket, SOL_SOCKET, SO_PASSCRED, &enable, sizeof(enable) );
1403 }
1404 #endif
1405
1406 if (version != SERVER_PROTOCOL_VERSION)
1407 server_protocol_error( "version mismatch %d/%d.\n"
1408 "Your %s binary was not upgraded correctly,\n"
1409 "or you have an older one somewhere in your PATH.\n"
1410 "Or maybe the wrong wineserver is still running?\n",
1411 version, SERVER_PROTOCOL_VERSION,
1412 (version > SERVER_PROTOCOL_VERSION) ? "wine" : "wineserver" );
1413 #ifdef __APPLE__
1414 send_server_task_port();
1415 #endif
1416 #if defined(__linux__) && defined(HAVE_PRCTL)
1417 /* work around Ubuntu's ptrace breakage */
1418 if (server_pid != -1) prctl( 0x59616d61 /* PR_SET_PTRACER */, server_pid );
1419 #endif
1420 }
1421
1422
1423 /***********************************************************************
1424 * server_init_process_done
1425 */
1426 NTSTATUS server_init_process_done(void)
1427 {
1428 PEB *peb = NtCurrentTeb()->Peb;
1429 IMAGE_NT_HEADERS *nt = RtlImageNtHeader( peb->ImageBaseAddress );
1430 NTSTATUS status;
1431
1432 /* Install signal handlers; this cannot be done earlier, since we cannot
1433 * send exceptions to the debugger before the create process event that
1434 * is sent by REQ_INIT_PROCESS_DONE.
1435 * We do need the handlers in place by the time the request is over, so
1436 * we set them up here. If we segfault between here and the server call
1437 * something is very wrong... */
1438 signal_init_process();
1439
1440 /* Signal the parent process to continue */
1441 SERVER_START_REQ( init_process_done )
1442 {
1443 req->module = wine_server_client_ptr( peb->ImageBaseAddress );
1444 #ifdef __i386__
1445 req->ldt_copy = wine_server_client_ptr( &wine_ldt_copy );
1446 #endif
1447 req->entry = wine_server_client_ptr( (char *)peb->ImageBaseAddress + nt->OptionalHeader.AddressOfEntryPoint );
1448 req->gui = (nt->OptionalHeader.Subsystem != IMAGE_SUBSYSTEM_WINDOWS_CUI);
1449 status = wine_server_call( req );
1450 }
1451 SERVER_END_REQ;
1452
1453 return status;
1454 }
1455
1456
1457 /***********************************************************************
1458 * server_init_thread
1459 *
1460 * Send an init thread request. Return 0 if OK.
1461 */
1462 size_t server_init_thread( void *entry_point )
1463 {
1464 static const char *cpu_names[] = { "x86", "x86_64", "PowerPC", "ARM", "ARM64" };
1465 static const BOOL is_win64 = (sizeof(void *) > sizeof(int));
1466 const char *arch = getenv( "WINEARCH" );
1467 int ret;
1468 int reply_pipe[2];
1469 struct sigaction sig_act;
1470 size_t info_size;
1471
1472 sig_act.sa_handler = SIG_IGN;
1473 sig_act.sa_flags = 0;
1474 sigemptyset( &sig_act.sa_mask );
1475
1476 /* ignore SIGPIPE so that we get an EPIPE error instead */
1477 sigaction( SIGPIPE, &sig_act, NULL );
1478
1479 /* create the server->client communication pipes */
1480 if (server_pipe( reply_pipe ) == -1) server_protocol_perror( "pipe" );
1481 if (server_pipe( ntdll_get_thread_data()->wait_fd ) == -1) server_protocol_perror( "pipe" );
1482 wine_server_send_fd( reply_pipe[1] );
1483 wine_server_send_fd( ntdll_get_thread_data()->wait_fd[1] );
1484 ntdll_get_thread_data()->reply_fd = reply_pipe[0];
1485 close( reply_pipe[1] );
1486
1487 SERVER_START_REQ( init_thread )
1488 {
1489 req->unix_pid = getpid();
1490 req->unix_tid = get_unix_tid();
1491 req->teb = wine_server_client_ptr( NtCurrentTeb() );
1492 req->entry = wine_server_client_ptr( entry_point );
1493 req->reply_fd = reply_pipe[1];
1494 req->wait_fd = ntdll_get_thread_data()->wait_fd[1];
1495 req->debug_level = (TRACE_ON(server) != 0);
1496 req->cpu = client_cpu;
1497 ret = wine_server_call( req );
1498 NtCurrentTeb()->ClientId.UniqueProcess = ULongToHandle(reply->pid);
1499 NtCurrentTeb()->ClientId.UniqueThread = ULongToHandle(reply->tid);
1500 info_size = reply->info_size;
1501 server_start_time = reply->server_start;
1502 server_cpus = reply->all_cpus;
1503 }
1504 SERVER_END_REQ;
1505
1506 is_wow64 = !is_win64 && (server_cpus & ((1 << CPU_x86_64) | (1 << CPU_ARM64))) != 0;
1507 ntdll_get_thread_data()->wow64_redir = is_wow64;
1508
1509 switch (ret)
1510 {
1511 case STATUS_SUCCESS:
1512 if (arch)
1513 {
1514 if (!strcmp( arch, "win32" ) && (is_win64 || is_wow64))
1515 fatal_error( "WINEARCH set to win32 but '%s' is a 64-bit installation.\n",
1516 wine_get_config_dir() );
1517 if (!strcmp( arch, "win64" ) && !is_win64 && !is_wow64)
1518 fatal_error( "WINEARCH set to win64 but '%s' is a 32-bit installation.\n",
1519 wine_get_config_dir() );
1520 }
1521 return info_size;
1522 case STATUS_INVALID_IMAGE_WIN_64:
1523 fatal_error( "'%s' is a 32-bit installation, it cannot support 64-bit applications.\n",
1524 wine_get_config_dir() );
1525 case STATUS_NOT_SUPPORTED:
1526 fatal_error( "'%s' is a 64-bit installation, it cannot be used with a 32-bit wineserver.\n",
1527 wine_get_config_dir() );
1528 case STATUS_INVALID_IMAGE_FORMAT:
1529 fatal_error( "wineserver doesn't support the %s architecture\n", cpu_names[client_cpu] );
1530 default:
1531 server_protocol_error( "init_thread failed with status %x\n", ret );
1532 }
1533 }
Windows API implementation