msi: Downgrade a few messages to WARN.
[wine.git] / dlls / ntdll / server.c
blobddabb676d84331d6da7606cac4cbb6df69d12a8f
1 /*
2 * Wine server communication
4 * Copyright (C) 1998 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "config.h"
22 #include "wine/port.h"
24 #include <assert.h>
25 #include <ctype.h>
26 #include <dirent.h>
27 #include <errno.h>
28 #include <fcntl.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <string.h>
33 #include <sys/types.h>
34 #ifdef HAVE_SYS_SOCKET_H
35 # include <sys/socket.h>
36 #endif
37 #ifdef HAVE_SYS_WAIT_H
38 #include <sys/wait.h>
39 #endif
40 #ifdef HAVE_SYS_UN_H
41 #include <sys/un.h>
42 #endif
43 #ifdef HAVE_SYS_MMAN_H
44 #include <sys/mman.h>
45 #endif
46 #ifdef HAVE_SYS_STAT_H
47 # include <sys/stat.h>
48 #endif
49 #ifdef HAVE_SYS_UIO_H
50 #include <sys/uio.h>
51 #endif
52 #ifdef HAVE_UNISTD_H
53 # include <unistd.h>
54 #endif
56 #include "ntstatus.h"
57 #define WIN32_NO_STATUS
58 #include "wine/library.h"
59 #include "wine/pthread.h"
60 #include "wine/server.h"
61 #include "wine/debug.h"
62 #include "ntdll_misc.h"
64 WINE_DEFAULT_DEBUG_CHANNEL(server);
66 /* Some versions of glibc don't define this */
67 #ifndef SCM_RIGHTS
68 #define SCM_RIGHTS 1
69 #endif
71 #define SOCKETNAME "socket" /* name of the socket file */
72 #define LOCKNAME "lock" /* name of the lock file */
74 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
75 /* data structure used to pass an fd with sendmsg/recvmsg */
76 struct cmsg_fd
78 struct
80 size_t len; /* size of structure */
81 int level; /* SOL_SOCKET */
82 int type; /* SCM_RIGHTS */
83 } header;
84 int fd; /* fd to pass */
86 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
88 timeout_t server_start_time = 0; /* time of server startup */
90 extern struct wine_pthread_functions pthread_functions;
92 sigset_t server_block_set; /* signals to block during server calls */
93 static int fd_socket = -1; /* socket to exchange file descriptors with the server */
95 static RTL_CRITICAL_SECTION fd_cache_section;
96 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
98 0, 0, &fd_cache_section,
99 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
100 0, 0, { (DWORD_PTR)(__FILE__ ": fd_cache_section") }
102 static RTL_CRITICAL_SECTION fd_cache_section = { &critsect_debug, -1, 0, 0, 0, 0 };
105 #ifdef __GNUC__
106 static void fatal_error( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
107 static void fatal_perror( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
108 static void server_connect_error( const char *serverdir ) __attribute__((noreturn));
109 #endif
111 /* die on a fatal error; use only during initialization */
112 static void fatal_error( const char *err, ... )
114 va_list args;
116 va_start( args, err );
117 fprintf( stderr, "wine: " );
118 vfprintf( stderr, err, args );
119 va_end( args );
120 exit(1);
123 /* die on a fatal error; use only during initialization */
124 static void fatal_perror( const char *err, ... )
126 va_list args;
128 va_start( args, err );
129 fprintf( stderr, "wine: " );
130 vfprintf( stderr, err, args );
131 perror( " " );
132 va_end( args );
133 exit(1);
137 /***********************************************************************
138 * server_exit_thread
140 void server_exit_thread( int status )
142 struct wine_pthread_thread_info info;
143 SIZE_T size;
144 int fds[4];
146 RtlAcquirePebLock();
147 RemoveEntryList( &NtCurrentTeb()->TlsLinks );
148 RtlReleasePebLock();
149 RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->FlsSlots );
150 RtlFreeHeap( GetProcessHeap(), 0, NtCurrentTeb()->TlsExpansionSlots );
152 info.stack_base = NtCurrentTeb()->DeallocationStack;
153 info.teb_base = NtCurrentTeb();
154 info.teb_sel = wine_get_fs();
155 info.exit_status = status;
157 fds[0] = ntdll_get_thread_data()->wait_fd[0];
158 fds[1] = ntdll_get_thread_data()->wait_fd[1];
159 fds[2] = ntdll_get_thread_data()->reply_fd;
160 fds[3] = ntdll_get_thread_data()->request_fd;
161 pthread_functions.sigprocmask( SIG_BLOCK, &server_block_set, NULL );
163 size = 0;
164 NtFreeVirtualMemory( GetCurrentProcess(), &info.stack_base, &size, MEM_RELEASE | MEM_SYSTEM );
165 info.stack_size = size;
167 size = 0;
168 NtFreeVirtualMemory( GetCurrentProcess(), &info.teb_base, &size, MEM_RELEASE | MEM_SYSTEM );
169 info.teb_size = size;
171 close( fds[0] );
172 close( fds[1] );
173 close( fds[2] );
174 close( fds[3] );
175 pthread_functions.exit_thread( &info );
179 /***********************************************************************
180 * server_abort_thread
182 void server_abort_thread( int status )
184 pthread_functions.sigprocmask( SIG_BLOCK, &server_block_set, NULL );
185 close( ntdll_get_thread_data()->wait_fd[0] );
186 close( ntdll_get_thread_data()->wait_fd[1] );
187 close( ntdll_get_thread_data()->reply_fd );
188 close( ntdll_get_thread_data()->request_fd );
189 pthread_functions.abort_thread( status );
193 /***********************************************************************
194 * server_protocol_error
196 void server_protocol_error( const char *err, ... )
198 va_list args;
200 va_start( args, err );
201 fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
202 vfprintf( stderr, err, args );
203 va_end( args );
204 server_abort_thread(1);
208 /***********************************************************************
209 * server_protocol_perror
211 void server_protocol_perror( const char *err )
213 fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
214 perror( err );
215 server_abort_thread(1);
219 /***********************************************************************
220 * send_request
222 * Send a request to the server.
224 static unsigned int send_request( const struct __server_request_info *req )
226 unsigned int i;
227 int ret;
229 if (!req->u.req.request_header.request_size)
231 if ((ret = write( ntdll_get_thread_data()->request_fd, &req->u.req,
232 sizeof(req->u.req) )) == sizeof(req->u.req)) return STATUS_SUCCESS;
235 else
237 struct iovec vec[__SERVER_MAX_DATA+1];
239 vec[0].iov_base = (void *)&req->u.req;
240 vec[0].iov_len = sizeof(req->u.req);
241 for (i = 0; i < req->data_count; i++)
243 vec[i+1].iov_base = (void *)req->data[i].ptr;
244 vec[i+1].iov_len = req->data[i].size;
246 if ((ret = writev( ntdll_get_thread_data()->request_fd, vec, i+1 )) ==
247 req->u.req.request_header.request_size + sizeof(req->u.req)) return STATUS_SUCCESS;
250 if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
251 if (errno == EPIPE) server_abort_thread(0);
252 if (errno == EFAULT) return STATUS_ACCESS_VIOLATION;
253 server_protocol_perror( "write" );
257 /***********************************************************************
258 * read_reply_data
260 * Read data from the reply buffer; helper for wait_reply.
262 static void read_reply_data( void *buffer, size_t size )
264 int ret;
266 for (;;)
268 if ((ret = read( ntdll_get_thread_data()->reply_fd, buffer, size )) > 0)
270 if (!(size -= ret)) return;
271 buffer = (char *)buffer + ret;
272 continue;
274 if (!ret) break;
275 if (errno == EINTR) continue;
276 if (errno == EPIPE) break;
277 server_protocol_perror("read");
279 /* the server closed the connection; time to die... */
280 server_abort_thread(0);
284 /***********************************************************************
285 * wait_reply
287 * Wait for a reply from the server.
289 static inline unsigned int wait_reply( struct __server_request_info *req )
291 read_reply_data( &req->u.reply, sizeof(req->u.reply) );
292 if (req->u.reply.reply_header.reply_size)
293 read_reply_data( req->reply_data, req->u.reply.reply_header.reply_size );
294 return req->u.reply.reply_header.error;
298 /***********************************************************************
299 * wine_server_call (NTDLL.@)
301 * Perform a server call.
303 * PARAMS
304 * req_ptr [I/O] Function dependent data
306 * RETURNS
307 * Depends on server function being called, but usually an NTSTATUS code.
309 * NOTES
310 * Use the SERVER_START_REQ and SERVER_END_REQ to help you fill out the
311 * server request structure for the particular call. E.g:
312 *| SERVER_START_REQ( event_op )
313 *| {
314 *| req->handle = handle;
315 *| req->op = SET_EVENT;
316 *| ret = wine_server_call( req );
317 *| }
318 *| SERVER_END_REQ;
320 unsigned int wine_server_call( void *req_ptr )
322 struct __server_request_info * const req = req_ptr;
323 sigset_t old_set;
324 unsigned int ret;
326 pthread_functions.sigprocmask( SIG_BLOCK, &server_block_set, &old_set );
327 ret = send_request( req );
328 if (!ret) ret = wait_reply( req );
329 pthread_functions.sigprocmask( SIG_SETMASK, &old_set, NULL );
330 return ret;
334 /***********************************************************************
335 * server_enter_uninterrupted_section
337 void server_enter_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
339 pthread_functions.sigprocmask( SIG_BLOCK, &server_block_set, sigset );
340 RtlEnterCriticalSection( cs );
344 /***********************************************************************
345 * server_leave_uninterrupted_section
347 void server_leave_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
349 RtlLeaveCriticalSection( cs );
350 pthread_functions.sigprocmask( SIG_SETMASK, sigset, NULL );
354 /***********************************************************************
355 * wine_server_send_fd (NTDLL.@)
357 * Send a file descriptor to the server.
359 * PARAMS
360 * fd [I] file descriptor to send
362 * RETURNS
363 * nothing
365 void wine_server_send_fd( int fd )
367 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
368 struct cmsg_fd cmsg;
369 #endif
370 struct send_fd data;
371 struct msghdr msghdr;
372 struct iovec vec;
373 int ret;
375 vec.iov_base = (void *)&data;
376 vec.iov_len = sizeof(data);
378 msghdr.msg_name = NULL;
379 msghdr.msg_namelen = 0;
380 msghdr.msg_iov = &vec;
381 msghdr.msg_iovlen = 1;
383 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
384 msghdr.msg_accrights = (void *)&fd;
385 msghdr.msg_accrightslen = sizeof(fd);
386 #else /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
387 cmsg.header.len = sizeof(cmsg.header) + sizeof(fd);
388 cmsg.header.level = SOL_SOCKET;
389 cmsg.header.type = SCM_RIGHTS;
390 cmsg.fd = fd;
391 msghdr.msg_control = &cmsg;
392 msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
393 msghdr.msg_flags = 0;
394 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
396 data.tid = GetCurrentThreadId();
397 data.fd = fd;
399 for (;;)
401 if ((ret = sendmsg( fd_socket, &msghdr, 0 )) == sizeof(data)) return;
402 if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
403 if (errno == EINTR) continue;
404 if (errno == EPIPE) server_abort_thread(0);
405 server_protocol_perror( "sendmsg" );
410 /***********************************************************************
411 * receive_fd
413 * Receive a file descriptor passed from the server.
415 static int receive_fd( obj_handle_t *handle )
417 struct iovec vec;
418 int ret, fd;
420 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
421 struct msghdr msghdr;
423 fd = -1;
424 msghdr.msg_accrights = (void *)&fd;
425 msghdr.msg_accrightslen = sizeof(fd);
426 #else /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
427 struct msghdr msghdr;
428 struct cmsg_fd cmsg;
430 cmsg.header.len = sizeof(cmsg.header) + sizeof(fd);
431 cmsg.header.level = SOL_SOCKET;
432 cmsg.header.type = SCM_RIGHTS;
433 cmsg.fd = -1;
434 msghdr.msg_control = &cmsg;
435 msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
436 msghdr.msg_flags = 0;
437 #endif /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
439 msghdr.msg_name = NULL;
440 msghdr.msg_namelen = 0;
441 msghdr.msg_iov = &vec;
442 msghdr.msg_iovlen = 1;
443 vec.iov_base = (void *)handle;
444 vec.iov_len = sizeof(*handle);
446 for (;;)
448 if ((ret = recvmsg( fd_socket, &msghdr, 0 )) > 0)
450 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
451 fd = cmsg.fd;
452 #endif
453 if (fd != -1) fcntl( fd, F_SETFD, 1 ); /* set close on exec flag */
454 return fd;
456 if (!ret) break;
457 if (errno == EINTR) continue;
458 if (errno == EPIPE) break;
459 server_protocol_perror("recvmsg");
461 /* the server closed the connection; time to die... */
462 server_abort_thread(0);
466 /***********************************************************************/
467 /* fd cache support */
469 struct fd_cache_entry
471 int fd;
472 enum server_fd_type type : 6;
473 unsigned int access : 2;
474 unsigned int options : 24;
477 #define FD_CACHE_BLOCK_SIZE (65536 / sizeof(struct fd_cache_entry))
478 #define FD_CACHE_ENTRIES 128
480 static struct fd_cache_entry *fd_cache[FD_CACHE_ENTRIES];
481 static struct fd_cache_entry fd_cache_initial_block[FD_CACHE_BLOCK_SIZE];
483 static inline unsigned int handle_to_index( obj_handle_t handle, unsigned int *entry )
485 unsigned long idx = ((unsigned long)handle >> 2) - 1;
486 *entry = idx / FD_CACHE_BLOCK_SIZE;
487 return idx % FD_CACHE_BLOCK_SIZE;
491 /***********************************************************************
492 * add_fd_to_cache
494 * Caller must hold fd_cache_section.
496 static int add_fd_to_cache( obj_handle_t handle, int fd, enum server_fd_type type,
497 unsigned int access, unsigned int options )
499 unsigned int entry, idx = handle_to_index( handle, &entry );
500 int prev_fd;
502 if (entry >= FD_CACHE_ENTRIES)
504 FIXME( "too many allocated handles, not caching %p\n", handle );
505 return 0;
508 if (!fd_cache[entry]) /* do we need to allocate a new block of entries? */
510 if (!entry) fd_cache[0] = fd_cache_initial_block;
511 else
513 void *ptr = wine_anon_mmap( NULL, FD_CACHE_BLOCK_SIZE * sizeof(struct fd_cache_entry),
514 PROT_READ | PROT_WRITE, 0 );
515 if (ptr == MAP_FAILED) return 0;
516 fd_cache[entry] = ptr;
519 /* store fd+1 so that 0 can be used as the unset value */
520 prev_fd = interlocked_xchg( &fd_cache[entry][idx].fd, fd + 1 ) - 1;
521 fd_cache[entry][idx].type = type;
522 fd_cache[entry][idx].access = access;
523 fd_cache[entry][idx].options = options;
524 if (prev_fd != -1) close( prev_fd );
525 return 1;
529 /***********************************************************************
530 * get_cached_fd
532 * Caller must hold fd_cache_section.
534 static inline int get_cached_fd( obj_handle_t handle, enum server_fd_type *type,
535 unsigned int *access, unsigned int *options )
537 unsigned int entry, idx = handle_to_index( handle, &entry );
538 int fd = -1;
540 if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
542 fd = fd_cache[entry][idx].fd - 1;
543 if (type) *type = fd_cache[entry][idx].type;
544 if (access) *access = fd_cache[entry][idx].access;
545 if (options) *options = fd_cache[entry][idx].options;
547 return fd;
551 /***********************************************************************
552 * server_remove_fd_from_cache
554 int server_remove_fd_from_cache( obj_handle_t handle )
556 unsigned int entry, idx = handle_to_index( handle, &entry );
557 int fd = -1;
559 if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
560 fd = interlocked_xchg( &fd_cache[entry][idx].fd, 0 ) - 1;
562 return fd;
566 /***********************************************************************
567 * server_get_unix_fd
569 * The returned unix_fd should be closed iff needs_close is non-zero.
571 int server_get_unix_fd( obj_handle_t handle, unsigned int wanted_access, int *unix_fd,
572 int *needs_close, enum server_fd_type *type, unsigned int *options )
574 sigset_t sigset;
575 obj_handle_t fd_handle;
576 int ret = 0, fd;
577 unsigned int access = 0;
579 *unix_fd = -1;
580 *needs_close = 0;
581 wanted_access &= FILE_READ_DATA | FILE_WRITE_DATA;
583 server_enter_uninterrupted_section( &fd_cache_section, &sigset );
585 fd = get_cached_fd( handle, type, &access, options );
586 if (fd != -1) goto done;
588 SERVER_START_REQ( get_handle_fd )
590 req->handle = handle;
591 if (!(ret = wine_server_call( req )))
593 if (type) *type = reply->type;
594 if (options) *options = reply->options;
595 access = reply->access;
596 if ((fd = receive_fd( &fd_handle )) != -1)
598 assert( fd_handle == handle );
599 *needs_close = (reply->removable ||
600 !add_fd_to_cache( handle, fd, reply->type,
601 reply->access, reply->options ));
603 else ret = STATUS_TOO_MANY_OPENED_FILES;
606 SERVER_END_REQ;
608 done:
609 server_leave_uninterrupted_section( &fd_cache_section, &sigset );
610 if (!ret && ((access & wanted_access) != wanted_access))
612 ret = STATUS_ACCESS_DENIED;
613 if (*needs_close) close( fd );
615 if (!ret) *unix_fd = fd;
616 return ret;
620 /***********************************************************************
621 * wine_server_fd_to_handle (NTDLL.@)
623 * Allocate a file handle for a Unix file descriptor.
625 * PARAMS
626 * fd [I] Unix file descriptor.
627 * access [I] Win32 access flags.
628 * attributes [I] Object attributes.
629 * handle [O] Address where Wine file handle will be stored.
631 * RETURNS
632 * NTSTATUS code
634 int wine_server_fd_to_handle( int fd, unsigned int access, unsigned int attributes, obj_handle_t *handle )
636 int ret;
638 *handle = 0;
639 wine_server_send_fd( fd );
641 SERVER_START_REQ( alloc_file_handle )
643 req->access = access;
644 req->attributes = attributes;
645 req->fd = fd;
646 if (!(ret = wine_server_call( req ))) *handle = reply->handle;
648 SERVER_END_REQ;
649 return ret;
653 /***********************************************************************
654 * wine_server_handle_to_fd (NTDLL.@)
656 * Retrieve the file descriptor corresponding to a file handle.
658 * PARAMS
659 * handle [I] Wine file handle.
660 * access [I] Win32 file access rights requested.
661 * unix_fd [O] Address where Unix file descriptor will be stored.
662 * options [O] Address where the file open options will be stored. Optional.
664 * RETURNS
665 * NTSTATUS code
667 int wine_server_handle_to_fd( obj_handle_t handle, unsigned int access, int *unix_fd,
668 unsigned int *options )
670 int needs_close, ret = server_get_unix_fd( handle, access, unix_fd, &needs_close, NULL, options );
672 if (!ret && !needs_close)
674 if ((*unix_fd = dup(*unix_fd)) == -1) ret = FILE_GetNtStatus();
676 return ret;
680 /***********************************************************************
681 * wine_server_release_fd (NTDLL.@)
683 * Release the Unix file descriptor returned by wine_server_handle_to_fd.
685 * PARAMS
686 * handle [I] Wine file handle.
687 * unix_fd [I] Unix file descriptor to release.
689 * RETURNS
690 * nothing
692 void wine_server_release_fd( obj_handle_t handle, int unix_fd )
694 close( unix_fd );
698 /***********************************************************************
699 * start_server
701 * Start a new wine server.
703 static void start_server(void)
705 static int started; /* we only try once */
706 char *argv[3];
707 static char wineserver[] = "server/wineserver";
708 static char debug[] = "-d";
710 if (!started)
712 int status;
713 int pid = fork();
714 if (pid == -1) fatal_perror( "fork" );
715 if (!pid)
717 argv[0] = wineserver;
718 argv[1] = TRACE_ON(server) ? debug : NULL;
719 argv[2] = NULL;
720 wine_exec_wine_binary( argv[0], argv, getenv("WINESERVER") );
721 fatal_error( "could not exec wineserver\n" );
723 waitpid( pid, &status, 0 );
724 status = WIFEXITED(status) ? WEXITSTATUS(status) : 1;
725 if (status == 2) return; /* server lock held by someone else, will retry later */
726 if (status) exit(status); /* server failed */
727 started = 1;
732 /***********************************************************************
733 * server_connect_error
735 * Try to display a meaningful explanation of why we couldn't connect
736 * to the server.
738 static void server_connect_error( const char *serverdir )
740 int fd;
741 struct flock fl;
743 if ((fd = open( LOCKNAME, O_WRONLY )) == -1)
744 fatal_error( "for some mysterious reason, the wine server never started.\n" );
746 fl.l_type = F_WRLCK;
747 fl.l_whence = SEEK_SET;
748 fl.l_start = 0;
749 fl.l_len = 1;
750 if (fcntl( fd, F_GETLK, &fl ) != -1)
752 if (fl.l_type == F_WRLCK) /* the file is locked */
753 fatal_error( "a wine server seems to be running, but I cannot connect to it.\n"
754 " You probably need to kill that process (it might be pid %d).\n",
755 (int)fl.l_pid );
756 fatal_error( "for some mysterious reason, the wine server failed to run.\n" );
758 fatal_error( "the file system of '%s' doesn't support locks,\n"
759 " and there is a 'socket' file in that directory that prevents wine from starting.\n"
760 " You should make sure no wine server is running, remove that file and try again.\n",
761 serverdir );
765 /***********************************************************************
766 * server_connect
768 * Attempt to connect to an existing server socket.
769 * We need to be in the server directory already.
771 static int server_connect( const char *serverdir )
773 struct sockaddr_un addr;
774 struct stat st;
775 int s, slen, retry, fd_cwd;
777 /* retrieve the current directory */
778 fd_cwd = open( ".", O_RDONLY );
779 if (fd_cwd != -1) fcntl( fd_cwd, F_SETFD, 1 ); /* set close on exec flag */
781 /* chdir to the server directory */
782 if (chdir( serverdir ) == -1)
784 if (errno != ENOENT) fatal_perror( "chdir to %s", serverdir );
785 start_server();
786 if (chdir( serverdir ) == -1) fatal_perror( "chdir to %s", serverdir );
789 /* make sure we are at the right place */
790 if (stat( ".", &st ) == -1) fatal_perror( "stat %s", serverdir );
791 if (st.st_uid != getuid()) fatal_error( "'%s' is not owned by you\n", serverdir );
792 if (st.st_mode & 077) fatal_error( "'%s' must not be accessible by other users\n", serverdir );
794 for (retry = 0; retry < 6; retry++)
796 /* if not the first try, wait a bit to leave the previous server time to exit */
797 if (retry)
799 usleep( 100000 * retry * retry );
800 start_server();
801 if (lstat( SOCKETNAME, &st ) == -1) continue; /* still no socket, wait a bit more */
803 else if (lstat( SOCKETNAME, &st ) == -1) /* check for an already existing socket */
805 if (errno != ENOENT) fatal_perror( "lstat %s/%s", serverdir, SOCKETNAME );
806 start_server();
807 if (lstat( SOCKETNAME, &st ) == -1) continue; /* still no socket, wait a bit more */
810 /* make sure the socket is sane (ISFIFO needed for Solaris) */
811 if (!S_ISSOCK(st.st_mode) && !S_ISFIFO(st.st_mode))
812 fatal_error( "'%s/%s' is not a socket\n", serverdir, SOCKETNAME );
813 if (st.st_uid != getuid())
814 fatal_error( "'%s/%s' is not owned by you\n", serverdir, SOCKETNAME );
816 /* try to connect to it */
817 addr.sun_family = AF_UNIX;
818 strcpy( addr.sun_path, SOCKETNAME );
819 slen = sizeof(addr) - sizeof(addr.sun_path) + strlen(addr.sun_path) + 1;
820 #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
821 addr.sun_len = slen;
822 #endif
823 if ((s = socket( AF_UNIX, SOCK_STREAM, 0 )) == -1) fatal_perror( "socket" );
824 if (connect( s, (struct sockaddr *)&addr, slen ) != -1)
826 /* switch back to the starting directory */
827 if (fd_cwd != -1)
829 fchdir( fd_cwd );
830 close( fd_cwd );
832 fcntl( s, F_SETFD, 1 ); /* set close on exec flag */
833 return s;
835 close( s );
837 server_connect_error( serverdir );
841 /***********************************************************************
842 * rm_rf
844 * Remove a directory and all its contents; helper for create_config_dir.
846 static void rm_rf( const char *path )
848 int err = errno; /* preserve errno */
849 DIR *dir;
850 char *buffer, *p;
851 struct stat st;
852 struct dirent *de;
854 if (!(buffer = malloc( strlen(path) + 256 + 1 ))) goto done;
855 strcpy( buffer, path );
856 p = buffer + strlen(buffer);
857 *p++ = '/';
859 if ((dir = opendir( path )))
861 while ((de = readdir( dir )))
863 if (!strcmp( de->d_name, "." ) || !strcmp( de->d_name, ".." )) continue;
864 strcpy( p, de->d_name );
865 if (unlink( buffer ) != -1) continue;
866 if (errno == EISDIR ||
867 (errno == EPERM && !lstat( buffer, &st ) && S_ISDIR(st.st_mode)))
869 /* recurse in the sub-directory */
870 rm_rf( buffer );
873 closedir( dir );
875 free( buffer );
876 rmdir( path );
877 done:
878 errno = err;
882 /***********************************************************************
883 * create_config_dir
885 * Create the wine configuration dir (~/.wine).
887 static void create_config_dir(void)
889 const char *p, *config_dir = wine_get_config_dir();
890 char *tmp_dir;
891 int fd;
892 pid_t pid, wret;
894 if (!(tmp_dir = malloc( strlen(config_dir) + sizeof("-XXXXXX") )))
895 fatal_error( "out of memory\n" );
897 if ((p = strrchr( config_dir, '/' )) && p != config_dir)
899 struct stat st;
901 memcpy( tmp_dir, config_dir, p - config_dir );
902 tmp_dir[p - config_dir] = 0;
903 if (!stat( tmp_dir, &st ) && st.st_uid != getuid())
904 fatal_error( "'%s' is not owned by you, refusing to create a configuration directory there\n",
905 tmp_dir );
907 strcpy( tmp_dir, config_dir );
908 strcat( tmp_dir, "-XXXXXX" );
909 if ((fd = mkstemps( tmp_dir, 0 )) == -1)
910 fatal_perror( "can't get temp file name for %s", config_dir );
911 close( fd );
912 unlink( tmp_dir );
913 if (mkdir( tmp_dir, 0777 ) == -1)
914 fatal_perror( "cannot create temp dir %s", tmp_dir );
916 MESSAGE( "wine: creating configuration directory '%s'...\n", config_dir );
917 pid = fork();
918 if (pid == -1)
920 rmdir( tmp_dir );
921 fatal_perror( "fork" );
923 if (!pid)
925 char *argv[6];
926 static char argv0[] = "tools/wineprefixcreate",
927 argv1[] = "--quiet",
928 argv2[] = "--wait",
929 argv3[] = "--prefix";
931 argv[0] = argv0;
932 argv[1] = argv1;
933 argv[2] = argv2;
934 argv[3] = argv3;
935 argv[4] = tmp_dir;
936 argv[5] = NULL;
937 wine_exec_wine_binary( argv[0], argv, NULL );
938 rmdir( tmp_dir );
939 fatal_perror( "could not exec wineprefixcreate" );
941 else
943 int status;
945 while ((wret = waitpid( pid, &status, 0 )) != pid)
947 if (wret == -1 && errno != EINTR) fatal_perror( "wait4" );
949 if (!WIFEXITED(status) || WEXITSTATUS(status))
951 rm_rf( tmp_dir );
952 fatal_error( "wineprefixcreate failed while creating '%s'.\n", config_dir );
955 if (rename( tmp_dir, config_dir ) == -1)
957 rm_rf( tmp_dir );
958 if (errno != EEXIST && errno != ENOTEMPTY)
959 fatal_perror( "rename '%s' to '%s'", tmp_dir, config_dir );
960 /* else it was probably created by a concurrent wine process */
962 free( tmp_dir );
963 MESSAGE( "wine: '%s' created successfully.\n", config_dir );
967 #ifdef __APPLE__
968 #include <mach/mach.h>
969 #include <mach/mach_error.h>
970 #include <servers/bootstrap.h>
972 /* send our task port to the server */
973 static void send_server_task_port(void)
975 mach_port_t bootstrap_port, wineserver_port;
976 kern_return_t kret;
978 struct {
979 mach_msg_header_t header;
980 mach_msg_body_t body;
981 mach_msg_port_descriptor_t task_port;
982 } msg;
984 if (task_get_bootstrap_port(mach_task_self(), &bootstrap_port) != KERN_SUCCESS) return;
986 kret = bootstrap_look_up(bootstrap_port, (char*)wine_get_server_dir(), &wineserver_port);
987 if (kret != KERN_SUCCESS)
988 fatal_error( "cannot find the server port: 0x%08x\n", kret );
990 mach_port_deallocate(mach_task_self(), bootstrap_port);
992 msg.header.msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, 0) | MACH_MSGH_BITS_COMPLEX;
993 msg.header.msgh_size = sizeof(msg);
994 msg.header.msgh_remote_port = wineserver_port;
995 msg.header.msgh_local_port = MACH_PORT_NULL;
997 msg.body.msgh_descriptor_count = 1;
998 msg.task_port.name = mach_task_self();
999 msg.task_port.disposition = MACH_MSG_TYPE_COPY_SEND;
1000 msg.task_port.type = MACH_MSG_PORT_DESCRIPTOR;
1002 kret = mach_msg_send(&msg.header);
1003 if (kret != KERN_SUCCESS)
1004 server_protocol_error( "mach_msg_send failed: 0x%08x\n", kret );
1006 mach_port_deallocate(mach_task_self(), wineserver_port);
1008 #endif /* __APPLE__ */
1010 /***********************************************************************
1011 * server_init_process
1013 * Start the server and create the initial socket pair.
1015 void server_init_process(void)
1017 obj_handle_t dummy_handle;
1018 const char *env_socket = getenv( "WINESERVERSOCKET" );
1020 if (env_socket)
1022 fd_socket = atoi( env_socket );
1023 if (fcntl( fd_socket, F_SETFD, 1 ) == -1)
1024 fatal_perror( "Bad server socket %d", fd_socket );
1025 unsetenv( "WINESERVERSOCKET" );
1027 else
1029 const char *server_dir = wine_get_server_dir();
1031 if (!server_dir) /* this means the config dir doesn't exist */
1033 create_config_dir();
1034 server_dir = wine_get_server_dir();
1037 /* connect to the server */
1038 fd_socket = server_connect( server_dir );
1041 /* setup the signal mask */
1042 sigemptyset( &server_block_set );
1043 sigaddset( &server_block_set, SIGALRM );
1044 sigaddset( &server_block_set, SIGIO );
1045 sigaddset( &server_block_set, SIGINT );
1046 sigaddset( &server_block_set, SIGHUP );
1047 sigaddset( &server_block_set, SIGUSR1 );
1048 sigaddset( &server_block_set, SIGUSR2 );
1049 sigaddset( &server_block_set, SIGCHLD );
1050 pthread_functions.sigprocmask( SIG_BLOCK, &server_block_set, NULL );
1052 /* receive the first thread request fd on the main socket */
1053 ntdll_get_thread_data()->request_fd = receive_fd( &dummy_handle );
1055 #ifdef __APPLE__
1056 send_server_task_port();
1057 #endif
1061 /***********************************************************************
1062 * server_init_process_done
1064 NTSTATUS server_init_process_done(void)
1066 PEB *peb = NtCurrentTeb()->Peb;
1067 IMAGE_NT_HEADERS *nt = RtlImageNtHeader( peb->ImageBaseAddress );
1068 NTSTATUS status;
1070 /* Install signal handlers; this cannot be done earlier, since we cannot
1071 * send exceptions to the debugger before the create process event that
1072 * is sent by REQ_INIT_PROCESS_DONE.
1073 * We do need the handlers in place by the time the request is over, so
1074 * we set them up here. If we segfault between here and the server call
1075 * something is very wrong... */
1076 if (!SIGNAL_Init()) exit(1);
1078 /* Signal the parent process to continue */
1079 SERVER_START_REQ( init_process_done )
1081 req->module = peb->ImageBaseAddress;
1082 req->entry = (char *)peb->ImageBaseAddress + nt->OptionalHeader.AddressOfEntryPoint;
1083 req->gui = (nt->OptionalHeader.Subsystem != IMAGE_SUBSYSTEM_WINDOWS_CUI);
1084 status = wine_server_call( req );
1086 SERVER_END_REQ;
1088 return status;
1092 /***********************************************************************
1093 * server_init_thread
1095 * Send an init thread request. Return 0 if OK.
1097 size_t server_init_thread( int unix_pid, int unix_tid, void *entry_point )
1099 int version, ret;
1100 int reply_pipe[2];
1101 struct sigaction sig_act;
1102 size_t info_size;
1104 sig_act.sa_handler = SIG_IGN;
1105 sig_act.sa_flags = 0;
1106 sigemptyset( &sig_act.sa_mask );
1108 /* ignore SIGPIPE so that we get an EPIPE error instead */
1109 sigaction( SIGPIPE, &sig_act, NULL );
1110 /* automatic child reaping to avoid zombies */
1111 #ifdef SA_NOCLDWAIT
1112 sig_act.sa_flags |= SA_NOCLDWAIT;
1113 #endif
1114 sigaction( SIGCHLD, &sig_act, NULL );
1116 /* create the server->client communication pipes */
1117 if (pipe( reply_pipe ) == -1) server_protocol_perror( "pipe" );
1118 if (pipe( ntdll_get_thread_data()->wait_fd ) == -1) server_protocol_perror( "pipe" );
1119 wine_server_send_fd( reply_pipe[1] );
1120 wine_server_send_fd( ntdll_get_thread_data()->wait_fd[1] );
1121 ntdll_get_thread_data()->reply_fd = reply_pipe[0];
1122 close( reply_pipe[1] );
1124 /* set close on exec flag */
1125 fcntl( ntdll_get_thread_data()->reply_fd, F_SETFD, 1 );
1126 fcntl( ntdll_get_thread_data()->wait_fd[0], F_SETFD, 1 );
1127 fcntl( ntdll_get_thread_data()->wait_fd[1], F_SETFD, 1 );
1129 SERVER_START_REQ( init_thread )
1131 req->unix_pid = unix_pid;
1132 req->unix_tid = unix_tid;
1133 req->teb = NtCurrentTeb();
1134 req->peb = NtCurrentTeb()->Peb;
1135 req->entry = entry_point;
1136 req->ldt_copy = &wine_ldt_copy;
1137 req->reply_fd = reply_pipe[1];
1138 req->wait_fd = ntdll_get_thread_data()->wait_fd[1];
1139 req->debug_level = (TRACE_ON(server) != 0);
1140 ret = wine_server_call( req );
1141 NtCurrentTeb()->ClientId.UniqueProcess = ULongToHandle(reply->pid);
1142 NtCurrentTeb()->ClientId.UniqueThread = ULongToHandle(reply->tid);
1143 info_size = reply->info_size;
1144 version = reply->version;
1145 server_start_time = reply->server_start;
1147 SERVER_END_REQ;
1149 if (ret) server_protocol_error( "init_thread failed with status %x\n", ret );
1150 if (version != SERVER_PROTOCOL_VERSION)
1151 server_protocol_error( "version mismatch %d/%d.\n"
1152 "Your %s binary was not upgraded correctly,\n"
1153 "or you have an older one somewhere in your PATH.\n"
1154 "Or maybe the wrong wineserver is still running?\n",
1155 version, SERVER_PROTOCOL_VERSION,
1156 (version > SERVER_PROTOCOL_VERSION) ? "wine" : "wineserver" );
1157 return info_size;