ntdll: Store the fd type in the cache and return it in server_get_unix_fd.
[wine] / dlls / ntdll / server.c
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 #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
55
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"
63
64 WINE_DEFAULT_DEBUG_CHANNEL(server);
65
66 /* Some versions of glibc don't define this */
67 #ifndef SCM_RIGHTS
68 #define SCM_RIGHTS 1
69 #endif
70
71 #define SOCKETNAME "socket"        /* name of the socket file */
72 #define LOCKNAME   "lock"          /* name of the lock file */
73
74 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
75 /* data structure used to pass an fd with sendmsg/recvmsg */
76 struct cmsg_fd
77 {
78     struct
79     {
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 */
85 };
86 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
87
88 abs_time_t server_start_time = { 0, 0 };  /* time of server startup */
89
90 extern struct wine_pthread_functions pthread_functions;
91
92 static sigset_t block_set;  /* signals to block during server calls */
93 static int fd_socket = -1;  /* socket to exchange file descriptors with the server */
94
95 static RTL_CRITICAL_SECTION fd_cache_section;
96 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
97 {
98     0, 0, &fd_cache_section,
99     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
100       0, 0, { (DWORD_PTR)(__FILE__ ": fd_cache_section") }
101 };
102 static RTL_CRITICAL_SECTION fd_cache_section = { &critsect_debug, -1, 0, 0, 0, 0 };
103
104
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
110
111 /* die on a fatal error; use only during initialization */
112 static void fatal_error( const char *err, ... )
113 {
114     va_list args;
115
116     va_start( args, err );
117     fprintf( stderr, "wine: " );
118     vfprintf( stderr, err, args );
119     va_end( args );
120     exit(1);
121 }
122
123 /* die on a fatal error; use only during initialization */
124 static void fatal_perror( const char *err, ... )
125 {
126     va_list args;
127
128     va_start( args, err );
129     fprintf( stderr, "wine: " );
130     vfprintf( stderr, err, args );
131     perror( " " );
132     va_end( args );
133     exit(1);
134 }
135
136
137 /***********************************************************************
138  *           server_exit_thread
139  */
140 void server_exit_thread( int status )
141 {
142     struct wine_pthread_thread_info info;
143     SIZE_T size;
144     int fds[4];
145
146     RtlAcquirePebLock();
147     RemoveEntryList( &NtCurrentTeb()->TlsLinks );
148     RtlReleasePebLock();
149
150     info.stack_base  = NtCurrentTeb()->DeallocationStack;
151     info.teb_base    = NtCurrentTeb();
152     info.teb_sel     = wine_get_fs();
153     info.exit_status = status;
154
155     fds[0] = ntdll_get_thread_data()->wait_fd[0];
156     fds[1] = ntdll_get_thread_data()->wait_fd[1];
157     fds[2] = ntdll_get_thread_data()->reply_fd;
158     fds[3] = ntdll_get_thread_data()->request_fd;
159     pthread_functions.sigprocmask( SIG_BLOCK, &block_set, NULL );
160
161     size = 0;
162     NtFreeVirtualMemory( GetCurrentProcess(), &info.stack_base, &size, MEM_RELEASE | MEM_SYSTEM );
163     info.stack_size = size;
164
165     size = 0;
166     NtFreeVirtualMemory( GetCurrentProcess(), &info.teb_base, &size, MEM_RELEASE | MEM_SYSTEM );
167     info.teb_size = size;
168
169     close( fds[0] );
170     close( fds[1] );
171     close( fds[2] );
172     close( fds[3] );
173     pthread_functions.exit_thread( &info );
174 }
175
176
177 /***********************************************************************
178  *           server_abort_thread
179  */
180 void server_abort_thread( int status )
181 {
182     pthread_functions.sigprocmask( SIG_BLOCK, &block_set, NULL );
183     close( ntdll_get_thread_data()->wait_fd[0] );
184     close( ntdll_get_thread_data()->wait_fd[1] );
185     close( ntdll_get_thread_data()->reply_fd );
186     close( ntdll_get_thread_data()->request_fd );
187     pthread_functions.abort_thread( status );
188 }
189
190
191 /***********************************************************************
192  *           server_protocol_error
193  */
194 void server_protocol_error( const char *err, ... )
195 {
196     va_list args;
197
198     va_start( args, err );
199     fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
200     vfprintf( stderr, err, args );
201     va_end( args );
202     server_abort_thread(1);
203 }
204
205
206 /***********************************************************************
207  *           server_protocol_perror
208  */
209 void server_protocol_perror( const char *err )
210 {
211     fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
212     perror( err );
213     server_abort_thread(1);
214 }
215
216
217 /***********************************************************************
218  *           send_request
219  *
220  * Send a request to the server.
221  */
222 static void send_request( const struct __server_request_info *req )
223 {
224     unsigned int i;
225     int ret;
226
227     if (!req->u.req.request_header.request_size)
228     {
229         if ((ret = write( ntdll_get_thread_data()->request_fd, &req->u.req,
230                           sizeof(req->u.req) )) == sizeof(req->u.req)) return;
231
232     }
233     else
234     {
235         struct iovec vec[__SERVER_MAX_DATA+1];
236
237         vec[0].iov_base = (void *)&req->u.req;
238         vec[0].iov_len = sizeof(req->u.req);
239         for (i = 0; i < req->data_count; i++)
240         {
241             vec[i+1].iov_base = (void *)req->data[i].ptr;
242             vec[i+1].iov_len = req->data[i].size;
243         }
244         if ((ret = writev( ntdll_get_thread_data()->request_fd, vec, i+1 )) ==
245             req->u.req.request_header.request_size + sizeof(req->u.req)) return;
246     }
247
248     if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
249     if (errno == EPIPE) server_abort_thread(0);
250     server_protocol_perror( "write" );
251 }
252
253
254 /***********************************************************************
255  *           read_reply_data
256  *
257  * Read data from the reply buffer; helper for wait_reply.
258  */
259 static void read_reply_data( void *buffer, size_t size )
260 {
261     int ret;
262
263     for (;;)
264     {
265         if ((ret = read( ntdll_get_thread_data()->reply_fd, buffer, size )) > 0)
266         {
267             if (!(size -= ret)) return;
268             buffer = (char *)buffer + ret;
269             continue;
270         }
271         if (!ret) break;
272         if (errno == EINTR) continue;
273         if (errno == EPIPE) break;
274         server_protocol_perror("read");
275     }
276     /* the server closed the connection; time to die... */
277     server_abort_thread(0);
278 }
279
280
281 /***********************************************************************
282  *           wait_reply
283  *
284  * Wait for a reply from the server.
285  */
286 inline static void wait_reply( struct __server_request_info *req )
287 {
288     read_reply_data( &req->u.reply, sizeof(req->u.reply) );
289     if (req->u.reply.reply_header.reply_size)
290         read_reply_data( req->reply_data, req->u.reply.reply_header.reply_size );
291 }
292
293
294 /***********************************************************************
295  *           wine_server_call (NTDLL.@)
296  *
297  * Perform a server call.
298  *
299  * PARAMS
300  *     req_ptr [I/O] Function dependent data
301  *
302  * RETURNS
303  *     Depends on server function being called, but usually an NTSTATUS code.
304  *
305  * NOTES
306  *     Use the SERVER_START_REQ and SERVER_END_REQ to help you fill out the
307  *     server request structure for the particular call. E.g:
308  *|     SERVER_START_REQ( event_op )
309  *|     {
310  *|         req->handle = handle;
311  *|         req->op     = SET_EVENT;
312  *|         ret = wine_server_call( req );
313  *|     }
314  *|     SERVER_END_REQ;
315  */
316 unsigned int wine_server_call( void *req_ptr )
317 {
318     struct __server_request_info * const req = req_ptr;
319     sigset_t old_set;
320
321     pthread_functions.sigprocmask( SIG_BLOCK, &block_set, &old_set );
322     send_request( req );
323     wait_reply( req );
324     pthread_functions.sigprocmask( SIG_SETMASK, &old_set, NULL );
325     return req->u.reply.reply_header.error;
326 }
327
328
329 /***********************************************************************
330  *           wine_server_send_fd   (NTDLL.@)
331  *
332  * Send a file descriptor to the server.
333  *
334  * PARAMS
335  *     fd [I] file descriptor to send
336  *
337  * RETURNS
338  *     nothing
339  */
340 void wine_server_send_fd( int fd )
341 {
342 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
343     struct cmsg_fd cmsg;
344 #endif
345     struct send_fd data;
346     struct msghdr msghdr;
347     struct iovec vec;
348     int ret;
349
350     vec.iov_base = (void *)&data;
351     vec.iov_len  = sizeof(data);
352
353     msghdr.msg_name    = NULL;
354     msghdr.msg_namelen = 0;
355     msghdr.msg_iov     = &vec;
356     msghdr.msg_iovlen  = 1;
357
358 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
359     msghdr.msg_accrights    = (void *)&fd;
360     msghdr.msg_accrightslen = sizeof(fd);
361 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
362     cmsg.header.len   = sizeof(cmsg.header) + sizeof(fd);
363     cmsg.header.level = SOL_SOCKET;
364     cmsg.header.type  = SCM_RIGHTS;
365     cmsg.fd           = fd;
366     msghdr.msg_control    = &cmsg;
367     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
368     msghdr.msg_flags      = 0;
369 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
370
371     data.tid = GetCurrentThreadId();
372     data.fd  = fd;
373
374     for (;;)
375     {
376         if ((ret = sendmsg( fd_socket, &msghdr, 0 )) == sizeof(data)) return;
377         if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
378         if (errno == EINTR) continue;
379         if (errno == EPIPE) server_abort_thread(0);
380         server_protocol_perror( "sendmsg" );
381     }
382 }
383
384
385 /***********************************************************************
386  *           receive_fd
387  *
388  * Receive a file descriptor passed from the server.
389  */
390 static int receive_fd( obj_handle_t *handle )
391 {
392     struct iovec vec;
393     int ret, fd;
394
395 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
396     struct msghdr msghdr;
397
398     fd = -1;
399     msghdr.msg_accrights    = (void *)&fd;
400     msghdr.msg_accrightslen = sizeof(fd);
401 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
402     struct msghdr msghdr;
403     struct cmsg_fd cmsg;
404
405     cmsg.header.len   = sizeof(cmsg.header) + sizeof(fd);
406     cmsg.header.level = SOL_SOCKET;
407     cmsg.header.type  = SCM_RIGHTS;
408     cmsg.fd           = -1;
409     msghdr.msg_control    = &cmsg;
410     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
411     msghdr.msg_flags      = 0;
412 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
413
414     msghdr.msg_name    = NULL;
415     msghdr.msg_namelen = 0;
416     msghdr.msg_iov     = &vec;
417     msghdr.msg_iovlen  = 1;
418     vec.iov_base = (void *)handle;
419     vec.iov_len  = sizeof(*handle);
420
421     for (;;)
422     {
423         if ((ret = recvmsg( fd_socket, &msghdr, 0 )) > 0)
424         {
425 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
426             fd = cmsg.fd;
427 #endif
428             if (fd != -1) fcntl( fd, F_SETFD, 1 ); /* set close on exec flag */
429             return fd;
430         }
431         if (!ret) break;
432         if (errno == EINTR) continue;
433         if (errno == EPIPE) break;
434         server_protocol_perror("recvmsg");
435     }
436     /* the server closed the connection; time to die... */
437     server_abort_thread(0);
438 }
439
440
441 inline static unsigned int handle_to_index( obj_handle_t handle )
442 {
443     return ((unsigned long)handle >> 2) - 1;
444 }
445
446 struct fd_cache_entry
447 {
448     int fd;
449     enum server_fd_type type;
450 };
451
452 static struct fd_cache_entry *fd_cache;
453 static unsigned int fd_cache_size;
454
455 /***********************************************************************
456  *           add_fd_to_cache
457  *
458  * Caller must hold fd_cache_section.
459  */
460 static int add_fd_to_cache( obj_handle_t handle, int fd, enum server_fd_type type )
461 {
462     unsigned int idx = handle_to_index( handle );
463
464     if (idx >= fd_cache_size)
465     {
466         unsigned int i, size = max( 32, fd_cache_size * 2 );
467         struct fd_cache_entry *new_cache;
468
469         if (size <= idx) size = idx + 1;
470         if (fd_cache)
471             new_cache = RtlReAllocateHeap( GetProcessHeap(), 0, fd_cache, size*sizeof(fd_cache[0]) );
472         else
473             new_cache = RtlAllocateHeap( GetProcessHeap(), 0, size*sizeof(fd_cache[0]) );
474
475         if (new_cache)
476         {
477             for (i = fd_cache_size; i < size; i++) new_cache[i].fd = -1;
478             fd_cache = new_cache;
479             fd_cache_size = size;
480         }
481     }
482     if (idx < fd_cache_size)
483     {
484         assert( fd_cache[idx].fd == -1 );
485         fd_cache[idx].fd   = fd;
486         fd_cache[idx].type = type;
487         TRACE("added %p (%d) type %d to cache\n", handle, fd, type );
488         return 1;
489     }
490     return 0;
491 }
492
493
494 /***********************************************************************
495  *           get_cached_fd
496  *
497  * Caller must hold fd_cache_section.
498  */
499 static inline int get_cached_fd( obj_handle_t handle, enum server_fd_type *type )
500 {
501     unsigned int idx = handle_to_index( handle );
502     int fd = -1;
503
504     if (idx < fd_cache_size)
505     {
506         fd = fd_cache[idx].fd;
507         if (type) *type = fd_cache[idx].type;
508     }
509     return fd;
510 }
511
512
513 /***********************************************************************
514  *           server_remove_fd_from_cache
515  */
516 int server_remove_fd_from_cache( obj_handle_t handle )
517 {
518     unsigned int idx = handle_to_index( handle );
519     int fd = -1;
520
521     RtlEnterCriticalSection( &fd_cache_section );
522     if (idx < fd_cache_size)
523     {
524         fd = fd_cache[idx].fd;
525         fd_cache[idx].fd = -1;
526     }
527     RtlLeaveCriticalSection( &fd_cache_section );
528
529     if (fd != -1)
530     {
531         close( fd );
532         TRACE("removed %p (%d) from cache\n", handle, fd );
533     }
534     return fd;
535 }
536
537
538 /***********************************************************************
539  *           server_get_unix_fd
540  *
541  * The returned unix_fd should be closed iff needs_close is non-zero.
542  */
543 int server_get_unix_fd( obj_handle_t handle, unsigned int access, int *unix_fd,
544                         int *needs_close, enum server_fd_type *type, int *flags )
545 {
546     obj_handle_t fd_handle;
547     int ret = 0, removable = 0, fd;
548
549     *unix_fd = -1;
550     *needs_close = 0;
551
552     RtlEnterCriticalSection( &fd_cache_section );
553
554     fd = get_cached_fd( handle, type );
555     if (fd != -1 && !flags) goto done;
556
557     SERVER_START_REQ( get_handle_fd )
558     {
559         req->handle = handle;
560         req->access = access;
561         req->cached = (fd != -1);
562         if (!(ret = wine_server_call( req )))
563         {
564             removable = reply->flags & FD_FLAG_REMOVABLE;
565             if (type) *type = reply->type;
566             if (flags) *flags = reply->flags;
567             if (fd == -1)
568             {
569                 if ((fd = receive_fd( &fd_handle )) != -1)
570                 {
571                     assert( fd_handle == handle );
572                     if (!removable) add_fd_to_cache( handle, fd, reply->type );
573                 }
574                 else ret = STATUS_TOO_MANY_OPENED_FILES;
575             }
576         }
577     }
578     SERVER_END_REQ;
579
580 done:
581     RtlLeaveCriticalSection( &fd_cache_section );
582     if (!ret) *unix_fd = fd;
583     return ret;
584 }
585
586
587 /***********************************************************************
588  *           wine_server_fd_to_handle   (NTDLL.@)
589  *
590  * Allocate a file handle for a Unix file descriptor.
591  *
592  * PARAMS
593  *     fd      [I] Unix file descriptor.
594  *     access  [I] Win32 access flags.
595  *     attributes [I] Object attributes.
596  *     handle  [O] Address where Wine file handle will be stored.
597  *
598  * RETURNS
599  *     NTSTATUS code
600  */
601 int wine_server_fd_to_handle( int fd, unsigned int access, unsigned int attributes, obj_handle_t *handle )
602 {
603     int ret;
604
605     *handle = 0;
606     wine_server_send_fd( fd );
607
608     SERVER_START_REQ( alloc_file_handle )
609     {
610         req->access     = access;
611         req->attributes = attributes;
612         req->fd         = fd;
613         if (!(ret = wine_server_call( req ))) *handle = reply->handle;
614     }
615     SERVER_END_REQ;
616     return ret;
617 }
618
619
620 /***********************************************************************
621  *           wine_server_handle_to_fd   (NTDLL.@)
622  *
623  * Retrieve the file descriptor corresponding to a file handle.
624  *
625  * PARAMS
626  *     handle  [I] Wine file handle.
627  *     access  [I] Win32 file access rights requested.
628  *     unix_fd [O] Address where Unix file descriptor will be stored.
629  *     flags   [O] Address where the Unix flags associated with file will be stored. Optional.
630  *
631  * RETURNS
632  *     NTSTATUS code
633  */
634 int wine_server_handle_to_fd( obj_handle_t handle, unsigned int access, int *unix_fd, int *flags )
635 {
636     int needs_close, ret = server_get_unix_fd( handle, access, unix_fd, &needs_close, NULL, flags );
637
638     if (!ret && !needs_close)
639     {
640         if ((*unix_fd = dup(*unix_fd)) == -1) ret = FILE_GetNtStatus();
641     }
642     return ret;
643 }
644
645
646 /***********************************************************************
647  *           wine_server_release_fd   (NTDLL.@)
648  *
649  * Release the Unix file descriptor returned by wine_server_handle_to_fd.
650  *
651  * PARAMS
652  *     handle  [I] Wine file handle.
653  *     unix_fd [I] Unix file descriptor to release.
654  *
655  * RETURNS
656  *     nothing
657  */
658 void wine_server_release_fd( obj_handle_t handle, int unix_fd )
659 {
660     close( unix_fd );
661 }
662
663
664 /***********************************************************************
665  *           start_server
666  *
667  * Start a new wine server.
668  */
669 static void start_server(void)
670 {
671     static int started;  /* we only try once */
672     char *argv[3];
673     static char wineserver[] = "server/wineserver";
674     static char debug[] = "-d";
675
676     if (!started)
677     {
678         int status;
679         int pid = fork();
680         if (pid == -1) fatal_perror( "fork" );
681         if (!pid)
682         {
683             argv[0] = wineserver;
684             argv[1] = TRACE_ON(server) ? debug : NULL;
685             argv[2] = NULL;
686             wine_exec_wine_binary( argv[0], argv, getenv("WINESERVER") );
687             fatal_error( "could not exec wineserver\n" );
688         }
689         waitpid( pid, &status, 0 );
690         status = WIFEXITED(status) ? WEXITSTATUS(status) : 1;
691         if (status == 2) return;  /* server lock held by someone else, will retry later */
692         if (status) exit(status);  /* server failed */
693         started = 1;
694     }
695 }
696
697
698 /***********************************************************************
699  *           server_connect_error
700  *
701  * Try to display a meaningful explanation of why we couldn't connect
702  * to the server.
703  */
704 static void server_connect_error( const char *serverdir )
705 {
706     int fd;
707     struct flock fl;
708
709     if ((fd = open( LOCKNAME, O_WRONLY )) == -1)
710         fatal_error( "for some mysterious reason, the wine server never started.\n" );
711
712     fl.l_type   = F_WRLCK;
713     fl.l_whence = SEEK_SET;
714     fl.l_start  = 0;
715     fl.l_len    = 1;
716     if (fcntl( fd, F_GETLK, &fl ) != -1)
717     {
718         if (fl.l_type == F_WRLCK)  /* the file is locked */
719             fatal_error( "a wine server seems to be running, but I cannot connect to it.\n"
720                          "   You probably need to kill that process (it might be pid %d).\n",
721                          (int)fl.l_pid );
722         fatal_error( "for some mysterious reason, the wine server failed to run.\n" );
723     }
724     fatal_error( "the file system of '%s' doesn't support locks,\n"
725           "   and there is a 'socket' file in that directory that prevents wine from starting.\n"
726           "   You should make sure no wine server is running, remove that file and try again.\n",
727                  serverdir );
728 }
729
730
731 /***********************************************************************
732  *           server_connect
733  *
734  * Attempt to connect to an existing server socket.
735  * We need to be in the server directory already.
736  */
737 static int server_connect( const char *serverdir )
738 {
739     struct sockaddr_un addr;
740     struct stat st;
741     int s, slen, retry, fd_cwd;
742
743     /* retrieve the current directory */
744     fd_cwd = open( ".", O_RDONLY );
745     if (fd_cwd != -1) fcntl( fd_cwd, F_SETFD, 1 ); /* set close on exec flag */
746
747     /* chdir to the server directory */
748     if (chdir( serverdir ) == -1)
749     {
750         if (errno != ENOENT) fatal_perror( "chdir to %s", serverdir );
751         start_server();
752         if (chdir( serverdir ) == -1) fatal_perror( "chdir to %s", serverdir );
753     }
754
755     /* make sure we are at the right place */
756     if (stat( ".", &st ) == -1) fatal_perror( "stat %s", serverdir );
757     if (st.st_uid != getuid()) fatal_error( "'%s' is not owned by you\n", serverdir );
758     if (st.st_mode & 077) fatal_error( "'%s' must not be accessible by other users\n", serverdir );
759
760     for (retry = 0; retry < 6; retry++)
761     {
762         /* if not the first try, wait a bit to leave the previous server time to exit */
763         if (retry)
764         {
765             usleep( 100000 * retry * retry );
766             start_server();
767             if (lstat( SOCKETNAME, &st ) == -1) continue;  /* still no socket, wait a bit more */
768         }
769         else if (lstat( SOCKETNAME, &st ) == -1) /* check for an already existing socket */
770         {
771             if (errno != ENOENT) fatal_perror( "lstat %s/%s", serverdir, SOCKETNAME );
772             start_server();
773             if (lstat( SOCKETNAME, &st ) == -1) continue;  /* still no socket, wait a bit more */
774         }
775
776         /* make sure the socket is sane (ISFIFO needed for Solaris) */
777         if (!S_ISSOCK(st.st_mode) && !S_ISFIFO(st.st_mode))
778             fatal_error( "'%s/%s' is not a socket\n", serverdir, SOCKETNAME );
779         if (st.st_uid != getuid())
780             fatal_error( "'%s/%s' is not owned by you\n", serverdir, SOCKETNAME );
781
782         /* try to connect to it */
783         addr.sun_family = AF_UNIX;
784         strcpy( addr.sun_path, SOCKETNAME );
785         slen = sizeof(addr) - sizeof(addr.sun_path) + strlen(addr.sun_path) + 1;
786 #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
787         addr.sun_len = slen;
788 #endif
789         if ((s = socket( AF_UNIX, SOCK_STREAM, 0 )) == -1) fatal_perror( "socket" );
790         if (connect( s, (struct sockaddr *)&addr, slen ) != -1)
791         {
792             /* switch back to the starting directory */
793             if (fd_cwd != -1)
794             {
795                 fchdir( fd_cwd );
796                 close( fd_cwd );
797             }
798             fcntl( s, F_SETFD, 1 ); /* set close on exec flag */
799             return s;
800         }
801         close( s );
802     }
803     server_connect_error( serverdir );
804 }
805
806
807 /***********************************************************************
808  *           rm_rf
809  *
810  * Remove a directory and all its contents; helper for create_config_dir.
811  */
812 static void rm_rf( const char *path )
813 {
814     int err = errno;  /* preserve errno */
815     DIR *dir;
816     char *buffer, *p;
817     struct stat st;
818     struct dirent *de;
819
820     if (!(buffer = malloc( strlen(path) + 256 + 1 ))) goto done;
821     strcpy( buffer, path );
822     p = buffer + strlen(buffer);
823     *p++ = '/';
824
825     if ((dir = opendir( path )))
826     {
827         while ((de = readdir( dir )))
828         {
829             if (!strcmp( de->d_name, "." ) || !strcmp( de->d_name, ".." )) continue;
830             strcpy( p, de->d_name );
831             if (unlink( buffer ) != -1) continue;
832             if (errno == EISDIR ||
833                 (errno == EPERM && !lstat( buffer, &st ) && S_ISDIR(st.st_mode)))
834             {
835                 /* recurse in the sub-directory */
836                 rm_rf( buffer );
837             }
838         }
839         closedir( dir );
840     }
841     free( buffer );
842     rmdir( path );
843 done:
844     errno = err;
845 }
846
847
848 /***********************************************************************
849  *           create_config_dir
850  *
851  * Create the wine configuration dir (~/.wine).
852  */
853 static void create_config_dir(void)
854 {
855     const char *config_dir = wine_get_config_dir();
856     char *tmp_dir;
857     int fd;
858     pid_t pid, wret;
859
860     if (!(tmp_dir = malloc( strlen(config_dir) + sizeof("-XXXXXX") )))
861         fatal_error( "out of memory\n" );
862     strcpy( tmp_dir, config_dir );
863     strcat( tmp_dir, "-XXXXXX" );
864     if ((fd = mkstemps( tmp_dir, 0 )) == -1)
865         fatal_perror( "can't get temp file name for %s", config_dir );
866     close( fd );
867     unlink( tmp_dir );
868     if (mkdir( tmp_dir, 0777 ) == -1)
869         fatal_perror( "cannot create temp dir %s", tmp_dir );
870
871     MESSAGE( "wine: creating configuration directory '%s'...\n", config_dir );
872     pid = fork();
873     if (pid == -1)
874     {
875         rmdir( tmp_dir );
876         fatal_perror( "fork" );
877     }
878     if (!pid)
879     {
880         char *argv[6];
881         static char argv0[] = "tools/wineprefixcreate",
882                     argv1[] = "--quiet",
883                     argv2[] = "--wait",
884                     argv3[] = "--prefix";
885
886         argv[0] = argv0;
887         argv[1] = argv1;
888         argv[2] = argv2;
889         argv[3] = argv3;
890         argv[4] = tmp_dir;
891         argv[5] = NULL;
892         wine_exec_wine_binary( argv[0], argv, NULL );
893         rmdir( tmp_dir );
894         fatal_perror( "could not exec wineprefixcreate" );
895     }
896     else
897     {
898         int status;
899
900         while ((wret = waitpid( pid, &status, 0 )) != pid)
901         {
902             if (wret == -1 && errno != EINTR) fatal_perror( "wait4" );
903         }
904         if (!WIFEXITED(status) || WEXITSTATUS(status))
905         {
906             rm_rf( tmp_dir );
907             fatal_error( "wineprefixcreate failed while creating '%s'.\n", config_dir );
908         }
909     }
910     if (rename( tmp_dir, config_dir ) == -1)
911     {
912         rm_rf( tmp_dir );
913         if (errno != EEXIST && errno != ENOTEMPTY)
914             fatal_perror( "rename '%s' to '%s'", tmp_dir, config_dir );
915         /* else it was probably created by a concurrent wine process */
916     }
917     free( tmp_dir );
918     MESSAGE( "wine: '%s' created successfully.\n", config_dir );
919 }
920
921
922 /***********************************************************************
923  *           server_init_process
924  *
925  * Start the server and create the initial socket pair.
926  */
927 void server_init_process(void)
928 {
929     obj_handle_t dummy_handle;
930     const char *env_socket = getenv( "WINESERVERSOCKET" );
931
932     if (env_socket)
933     {
934         fd_socket = atoi( env_socket );
935         if (fcntl( fd_socket, F_SETFD, 1 ) == -1)
936             fatal_perror( "Bad server socket %d", fd_socket );
937     }
938     else
939     {
940         const char *server_dir = wine_get_server_dir();
941
942         if (!server_dir)  /* this means the config dir doesn't exist */
943         {
944             create_config_dir();
945             server_dir = wine_get_server_dir();
946         }
947
948         /* connect to the server */
949         fd_socket = server_connect( server_dir );
950     }
951
952     /* setup the signal mask */
953     sigemptyset( &block_set );
954     sigaddset( &block_set, SIGALRM );
955     sigaddset( &block_set, SIGIO );
956     sigaddset( &block_set, SIGINT );
957     sigaddset( &block_set, SIGHUP );
958     sigaddset( &block_set, SIGUSR1 );
959     sigaddset( &block_set, SIGUSR2 );
960     sigaddset( &block_set, SIGCHLD );
961
962     /* receive the first thread request fd on the main socket */
963     ntdll_get_thread_data()->request_fd = receive_fd( &dummy_handle );
964 }
965
966
967 /***********************************************************************
968  *           server_init_thread
969  *
970  * Send an init thread request. Return 0 if OK.
971  */
972 size_t server_init_thread( int unix_pid, int unix_tid, void *entry_point )
973 {
974     int version, ret;
975     int reply_pipe[2];
976     struct sigaction sig_act;
977     size_t info_size;
978
979     sig_act.sa_handler = SIG_IGN;
980     sig_act.sa_flags   = 0;
981     sigemptyset( &sig_act.sa_mask );
982
983     /* ignore SIGPIPE so that we get an EPIPE error instead  */
984     sigaction( SIGPIPE, &sig_act, NULL );
985     /* automatic child reaping to avoid zombies */
986 #ifdef SA_NOCLDWAIT
987     sig_act.sa_flags |= SA_NOCLDWAIT;
988 #endif
989     sigaction( SIGCHLD, &sig_act, NULL );
990
991     /* create the server->client communication pipes */
992     if (pipe( reply_pipe ) == -1) server_protocol_perror( "pipe" );
993     if (pipe( ntdll_get_thread_data()->wait_fd ) == -1) server_protocol_perror( "pipe" );
994     wine_server_send_fd( reply_pipe[1] );
995     wine_server_send_fd( ntdll_get_thread_data()->wait_fd[1] );
996     ntdll_get_thread_data()->reply_fd = reply_pipe[0];
997     close( reply_pipe[1] );
998
999     /* set close on exec flag */
1000     fcntl( ntdll_get_thread_data()->reply_fd, F_SETFD, 1 );
1001     fcntl( ntdll_get_thread_data()->wait_fd[0], F_SETFD, 1 );
1002     fcntl( ntdll_get_thread_data()->wait_fd[1], F_SETFD, 1 );
1003
1004     SERVER_START_REQ( init_thread )
1005     {
1006         req->unix_pid    = unix_pid;
1007         req->unix_tid    = unix_tid;
1008         req->teb         = NtCurrentTeb();
1009         req->peb         = NtCurrentTeb()->Peb;
1010         req->entry       = entry_point;
1011         req->ldt_copy    = &wine_ldt_copy;
1012         req->reply_fd    = reply_pipe[1];
1013         req->wait_fd     = ntdll_get_thread_data()->wait_fd[1];
1014         req->debug_level = (TRACE_ON(server) != 0);
1015         ret = wine_server_call( req );
1016         NtCurrentTeb()->ClientId.UniqueProcess = (HANDLE)reply->pid;
1017         NtCurrentTeb()->ClientId.UniqueThread  = (HANDLE)reply->tid;
1018         info_size         = reply->info_size;
1019         version           = reply->version;
1020         server_start_time = reply->server_start;
1021     }
1022     SERVER_END_REQ;
1023
1024     if (ret) server_protocol_error( "init_thread failed with status %x\n", ret );
1025     if (version != SERVER_PROTOCOL_VERSION)
1026         server_protocol_error( "version mismatch %d/%d.\n"
1027                                "Your %s binary was not upgraded correctly,\n"
1028                                "or you have an older one somewhere in your PATH.\n"
1029                                "Or maybe the wrong wineserver is still running?\n",
1030                                version, SERVER_PROTOCOL_VERSION,
1031                                (version > SERVER_PROTOCOL_VERSION) ? "wine" : "wineserver" );
1032     return info_size;
1033 }