server: New scheme for cleaning up objects on server exit.
[wine] / server / request.c
1 /*
2  * Server-side request handling
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <assert.h>
25 #include <errno.h>
26 #include <fcntl.h>
27 #ifdef HAVE_PWD_H
28 #include <pwd.h>
29 #endif
30 #include <signal.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <stdarg.h>
34 #include <string.h>
35 #include <sys/stat.h>
36 #include <sys/time.h>
37 #include <sys/types.h>
38 #ifdef HAVE_SYS_SOCKET_H
39 # include <sys/socket.h>
40 #endif
41 #ifdef HAVE_SYS_WAIT_H
42 # include <sys/wait.h>
43 #endif
44 #ifdef HAVE_SYS_UIO_H
45 #include <sys/uio.h>
46 #endif
47 #ifdef HAVE_SYS_UN_H
48 #include <sys/un.h>
49 #endif
50 #include <unistd.h>
51 #ifdef HAVE_POLL_H
52 #include <poll.h>
53 #endif
54
55 #include "windef.h"
56 #include "winbase.h"
57 #include "wincon.h"
58 #include "wine/library.h"
59
60 #include "file.h"
61 #include "process.h"
62 #define WANT_REQUEST_HANDLERS
63 #include "request.h"
64
65 /* Some versions of glibc don't define this */
66 #ifndef SCM_RIGHTS
67 #define SCM_RIGHTS 1
68 #endif
69
70 /* path names for server master Unix socket */
71 static const char * const server_socket_name = "socket";   /* name of the socket file */
72 static const char * const server_lock_name = "lock";       /* name of the server lock file */
73
74 struct master_socket
75 {
76     struct object        obj;        /* object header */
77     struct fd           *fd;         /* file descriptor of the master socket */
78     struct timeout_user *timeout;    /* timeout on last process exit */
79 };
80
81 static void master_socket_dump( struct object *obj, int verbose );
82 static void master_socket_destroy( struct object *obj );
83 static void master_socket_poll_event( struct fd *fd, int event );
84
85 static const struct object_ops master_socket_ops =
86 {
87     sizeof(struct master_socket),  /* size */
88     master_socket_dump,            /* dump */
89     no_add_queue,                  /* add_queue */
90     NULL,                          /* remove_queue */
91     NULL,                          /* signaled */
92     NULL,                          /* satisfied */
93     no_signal,                     /* signal */
94     no_get_fd,                     /* get_fd */
95     no_map_access,                 /* map_access */
96     no_lookup_name,                /* lookup_name */
97     no_close_handle,               /* close_handle */
98     master_socket_destroy          /* destroy */
99 };
100
101 static const struct fd_ops master_socket_fd_ops =
102 {
103     NULL,                          /* get_poll_events */
104     master_socket_poll_event,      /* poll_event */
105     no_flush,                      /* flush */
106     no_get_file_info,              /* get_file_info */
107     no_queue_async,                /* queue_async */
108     no_cancel_async                /* cancel_async */
109 };
110
111
112 struct thread *current = NULL;  /* thread handling the current request */
113 unsigned int global_error = 0;  /* global error code for when no thread is current */
114 time_t server_start_time = 0;  /* server startup time */
115
116 static struct master_socket *master_socket;  /* the master socket object */
117
118 /* socket communication static structures */
119 static struct iovec myiovec;
120 static struct msghdr msghdr;
121 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
122 struct cmsg_fd
123 {
124     struct
125     {
126         size_t len;   /* size of structure */
127         int    level; /* SOL_SOCKET */
128         int    type;  /* SCM_RIGHTS */
129     } header;
130     int fd;           /* fd to pass */
131 };
132 static struct cmsg_fd cmsg = { { sizeof(cmsg.header) + sizeof(cmsg.fd), SOL_SOCKET, SCM_RIGHTS }, -1 };
133 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
134
135 /* complain about a protocol error and terminate the client connection */
136 void fatal_protocol_error( struct thread *thread, const char *err, ... )
137 {
138     va_list args;
139
140     va_start( args, err );
141     fprintf( stderr, "Protocol error:%p: ", thread );
142     vfprintf( stderr, err, args );
143     va_end( args );
144     thread->exit_code = 1;
145     kill_thread( thread, 1 );
146 }
147
148 /* complain about a protocol error and terminate the client connection */
149 void fatal_protocol_perror( struct thread *thread, const char *err, ... )
150 {
151     va_list args;
152
153     va_start( args, err );
154     fprintf( stderr, "Protocol error:%p: ", thread );
155     vfprintf( stderr, err, args );
156     perror( " " );
157     va_end( args );
158     thread->exit_code = 1;
159     kill_thread( thread, 1 );
160 }
161
162 /* die on a fatal error */
163 void fatal_error( const char *err, ... )
164 {
165     va_list args;
166
167     va_start( args, err );
168     fprintf( stderr, "wineserver: " );
169     vfprintf( stderr, err, args );
170     va_end( args );
171     exit(1);
172 }
173
174 /* die on a fatal error */
175 void fatal_perror( const char *err, ... )
176 {
177     va_list args;
178
179     va_start( args, err );
180     fprintf( stderr, "wineserver: " );
181     vfprintf( stderr, err, args );
182     perror( " " );
183     va_end( args );
184     exit(1);
185 }
186
187 /* allocate the reply data */
188 void *set_reply_data_size( size_t size )
189 {
190     assert( size <= get_reply_max_size() );
191     if (size && !(current->reply_data = mem_alloc( size ))) size = 0;
192     current->reply_size = size;
193     return current->reply_data;
194 }
195
196 /* write the remaining part of the reply */
197 void write_reply( struct thread *thread )
198 {
199     int ret;
200
201     if ((ret = write( get_unix_fd( thread->reply_fd ),
202                       (char *)thread->reply_data + thread->reply_size - thread->reply_towrite,
203                       thread->reply_towrite )) >= 0)
204     {
205         if (!(thread->reply_towrite -= ret))
206         {
207             free( thread->reply_data );
208             thread->reply_data = NULL;
209             /* sent everything, can go back to waiting for requests */
210             set_fd_events( thread->request_fd, POLLIN );
211             set_fd_events( thread->reply_fd, 0 );
212         }
213         return;
214     }
215     if (errno == EPIPE)
216         kill_thread( thread, 0 );  /* normal death */
217     else if (errno != EWOULDBLOCK && errno != EAGAIN)
218         fatal_protocol_perror( thread, "reply write" );
219 }
220
221 /* send a reply to the current thread */
222 static void send_reply( union generic_reply *reply )
223 {
224     int ret;
225
226     if (!current->reply_size)
227     {
228         if ((ret = write( get_unix_fd( current->reply_fd ),
229                           reply, sizeof(*reply) )) != sizeof(*reply)) goto error;
230     }
231     else
232     {
233         struct iovec vec[2];
234
235         vec[0].iov_base = (void *)reply;
236         vec[0].iov_len  = sizeof(*reply);
237         vec[1].iov_base = current->reply_data;
238         vec[1].iov_len  = current->reply_size;
239
240         if ((ret = writev( get_unix_fd( current->reply_fd ), vec, 2 )) < sizeof(*reply)) goto error;
241
242         if ((current->reply_towrite = current->reply_size - (ret - sizeof(*reply))))
243         {
244             /* couldn't write it all, wait for POLLOUT */
245             set_fd_events( current->reply_fd, POLLOUT );
246             set_fd_events( current->request_fd, 0 );
247             return;
248         }
249     }
250     if (current->reply_data)
251     {
252         free( current->reply_data );
253         current->reply_data = NULL;
254     }
255     return;
256
257  error:
258     if (ret >= 0)
259         fatal_protocol_error( current, "partial write %d\n", ret );
260     else if (errno == EPIPE)
261         kill_thread( current, 0 );  /* normal death */
262     else
263         fatal_protocol_perror( current, "reply write" );
264 }
265
266 /* call a request handler */
267 static void call_req_handler( struct thread *thread )
268 {
269     union generic_reply reply;
270     enum request req = thread->req.request_header.req;
271
272     current = thread;
273     current->reply_size = 0;
274     clear_error();
275     memset( &reply, 0, sizeof(reply) );
276
277     if (debug_level) trace_request();
278
279     if (req < REQ_NB_REQUESTS)
280     {
281         req_handlers[req]( &current->req, &reply );
282         if (current)
283         {
284             if (current->reply_fd)
285             {
286                 reply.reply_header.error = current->error;
287                 reply.reply_header.reply_size = current->reply_size;
288                 if (debug_level) trace_reply( req, &reply );
289                 send_reply( &reply );
290             }
291             else fatal_protocol_error( current, "no reply fd for request %d\n", req );
292         }
293         current = NULL;
294         return;
295     }
296     fatal_protocol_error( current, "bad request %d\n", req );
297 }
298
299 /* read a request from a thread */
300 void read_request( struct thread *thread )
301 {
302     int ret;
303
304     if (!thread->req_toread)  /* no pending request */
305     {
306         if ((ret = read( get_unix_fd( thread->request_fd ), &thread->req,
307                          sizeof(thread->req) )) != sizeof(thread->req)) goto error;
308         if (!(thread->req_toread = thread->req.request_header.request_size))
309         {
310             /* no data, handle request at once */
311             call_req_handler( thread );
312             return;
313         }
314         if (!(thread->req_data = malloc( thread->req_toread )))
315             fatal_protocol_error( thread, "no memory for %d bytes request\n", thread->req_toread );
316     }
317
318     /* read the variable sized data */
319     for (;;)
320     {
321         ret = read( get_unix_fd( thread->request_fd ),
322                     (char *)thread->req_data + thread->req.request_header.request_size
323                       - thread->req_toread,
324                     thread->req_toread );
325         if (ret <= 0) break;
326         if (!(thread->req_toread -= ret))
327         {
328             call_req_handler( thread );
329             free( thread->req_data );
330             thread->req_data = NULL;
331             return;
332         }
333     }
334
335 error:
336     if (!ret)  /* closed pipe */
337         kill_thread( thread, 0 );
338     else if (ret > 0)
339         fatal_protocol_error( thread, "partial read %d\n", ret );
340     else if (errno != EWOULDBLOCK && errno != EAGAIN)
341         fatal_protocol_perror( thread, "read" );
342 }
343
344 /* receive a file descriptor on the process socket */
345 int receive_fd( struct process *process )
346 {
347     struct send_fd data;
348     int fd, ret;
349
350 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
351     msghdr.msg_accrightslen = sizeof(int);
352     msghdr.msg_accrights = (void *)&fd;
353 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
354     msghdr.msg_control    = &cmsg;
355     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
356     cmsg.fd = -1;
357 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
358
359     myiovec.iov_base = (void *)&data;
360     myiovec.iov_len  = sizeof(data);
361
362     ret = recvmsg( get_unix_fd( process->msg_fd ), &msghdr, 0 );
363 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
364     fd = cmsg.fd;
365 #endif
366
367     if (ret == sizeof(data))
368     {
369         struct thread *thread;
370
371         if (data.tid) thread = get_thread_from_id( data.tid );
372         else thread = (struct thread *)grab_object( get_process_first_thread( process ));
373
374         if (!thread || thread->process != process || thread->state == TERMINATED)
375         {
376             if (debug_level)
377                 fprintf( stderr, "%04x: *fd* %d <- %d bad thread id\n",
378                          data.tid, data.fd, fd );
379             close( fd );
380         }
381         else
382         {
383             if (debug_level)
384                 fprintf( stderr, "%04x: *fd* %d <- %d\n",
385                          thread->id, data.fd, fd );
386             thread_add_inflight_fd( thread, data.fd, fd );
387         }
388         if (thread) release_object( thread );
389         return 0;
390     }
391
392     if (ret >= 0)
393     {
394         if (ret > 0)
395             fprintf( stderr, "Protocol error: process %p: partial recvmsg %d for fd\n",
396                      process, ret );
397         kill_process( process, NULL, 1 );
398     }
399     else
400     {
401         if (errno != EWOULDBLOCK && errno != EAGAIN)
402         {
403             fprintf( stderr, "Protocol error: process %p: ", process );
404             perror( "recvmsg" );
405             kill_process( process, NULL, 1 );
406         }
407     }
408     return -1;
409 }
410
411 /* send an fd to a client */
412 int send_client_fd( struct process *process, int fd, obj_handle_t handle )
413 {
414     int ret;
415
416     if (debug_level)
417         fprintf( stderr, "%04x: *fd* %p -> %d\n",
418                  current ? current->id : process->id, handle, fd );
419
420 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
421     msghdr.msg_accrightslen = sizeof(fd);
422     msghdr.msg_accrights = (void *)&fd;
423 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
424     msghdr.msg_control    = &cmsg;
425     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
426     cmsg.fd = fd;
427 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
428
429     myiovec.iov_base = (void *)&handle;
430     myiovec.iov_len  = sizeof(handle);
431
432     ret = sendmsg( get_unix_fd( process->msg_fd ), &msghdr, 0 );
433
434     if (ret == sizeof(handle)) return 0;
435
436     if (ret >= 0)
437     {
438         fprintf( stderr, "Protocol error: process %p: partial sendmsg %d\n", process, ret );
439         kill_process( process, NULL, 1 );
440     }
441     else if (errno == EPIPE)
442     {
443         kill_process( process, NULL, 0 );
444     }
445     else
446     {
447         fprintf( stderr, "Protocol error: process %p: ", process );
448         perror( "sendmsg" );
449         kill_process( process, NULL, 1 );
450     }
451     return -1;
452 }
453
454 /* get current tick count to return to client */
455 unsigned int get_tick_count(void)
456 {
457     struct timeval t;
458     gettimeofday( &t, NULL );
459     return ((t.tv_sec - server_start_time) * 1000) + (t.tv_usec / 1000);
460 }
461
462 static void master_socket_dump( struct object *obj, int verbose )
463 {
464     struct master_socket *sock = (struct master_socket *)obj;
465     assert( obj->ops == &master_socket_ops );
466     fprintf( stderr, "Master socket fd=%p\n", sock->fd );
467 }
468
469 static void master_socket_destroy( struct object *obj )
470 {
471     struct master_socket *sock = (struct master_socket *)obj;
472     assert( obj->ops == &master_socket_ops );
473     release_object( sock->fd );
474 }
475
476 /* handle a socket event */
477 static void master_socket_poll_event( struct fd *fd, int event )
478 {
479     struct master_socket *sock = get_fd_user( fd );
480     assert( master_socket->obj.ops == &master_socket_ops );
481
482     assert( sock == master_socket );  /* there is only one master socket */
483
484     if (event & (POLLERR | POLLHUP))
485     {
486         /* this is not supposed to happen */
487         fprintf( stderr, "wineserver: Error on master socket\n" );
488         release_object( sock );
489     }
490     else if (event & POLLIN)
491     {
492         struct sockaddr_un dummy;
493         unsigned int len = sizeof(dummy);
494         int client = accept( get_unix_fd( master_socket->fd ), (struct sockaddr *) &dummy, &len );
495         if (client == -1) return;
496         if (sock->timeout)
497         {
498             remove_timeout_user( sock->timeout );
499             sock->timeout = NULL;
500         }
501         fcntl( client, F_SETFL, O_NONBLOCK );
502         create_process( client );
503     }
504 }
505
506 /* remove the socket upon exit */
507 static void socket_cleanup(void)
508 {
509     static int do_it_once;
510     if (!do_it_once++) unlink( server_socket_name );
511 }
512
513 /* create a directory and check its permissions */
514 static void create_dir( const char *name, struct stat *st )
515 {
516     if (lstat( name, st ) == -1)
517     {
518         if (errno != ENOENT) fatal_perror( "lstat %s", name );
519         if (mkdir( name, 0700 ) == -1 && errno != EEXIST) fatal_perror( "mkdir %s", name );
520         if (lstat( name, st ) == -1) fatal_perror( "lstat %s", name );
521     }
522     if (!S_ISDIR(st->st_mode)) fatal_error( "%s is not a directory\n", name );
523     if (st->st_uid != getuid()) fatal_error( "%s is not owned by you\n", name );
524     if (st->st_mode & 077) fatal_error( "%s must not be accessible by other users\n", name );
525 }
526
527 /* create the server directory and chdir to it */
528 static void create_server_dir(void)
529 {
530     char *p, *server_dir;
531     struct stat st, st2;
532
533     if (!(server_dir = strdup( wine_get_server_dir() ))) fatal_error( "out of memory\n" );
534
535     /* first create the base directory if needed */
536
537     p = strrchr( server_dir, '/' );
538     *p = 0;
539     create_dir( server_dir, &st );
540
541     /* now create the server directory */
542
543     *p = '/';
544     create_dir( server_dir, &st );
545
546     if (chdir( server_dir ) == -1) fatal_perror( "chdir %s", server_dir );
547     if (stat( ".", &st2 ) == -1) fatal_perror( "stat %s", server_dir );
548     if (st.st_dev != st2.st_dev || st.st_ino != st2.st_ino)
549         fatal_error( "chdir did not end up in %s\n", server_dir );
550
551     free( server_dir );
552 }
553
554 /* create the lock file and return its file descriptor */
555 static int create_server_lock(void)
556 {
557     struct stat st;
558     int fd;
559
560     if (lstat( server_lock_name, &st ) == -1)
561     {
562         if (errno != ENOENT)
563             fatal_perror( "lstat %s/%s", wine_get_server_dir(), server_lock_name );
564     }
565     else
566     {
567         if (!S_ISREG(st.st_mode))
568             fatal_error( "%s/%s is not a regular file\n", wine_get_server_dir(), server_lock_name );
569     }
570
571     if ((fd = open( server_lock_name, O_CREAT|O_TRUNC|O_WRONLY, 0600 )) == -1)
572         fatal_perror( "error creating %s/%s", wine_get_server_dir(), server_lock_name );
573     return fd;
574 }
575
576 /* wait for the server lock */
577 int wait_for_lock(void)
578 {
579     int fd, r;
580     struct flock fl;
581
582     create_server_dir();
583     fd = create_server_lock();
584
585     fl.l_type   = F_WRLCK;
586     fl.l_whence = SEEK_SET;
587     fl.l_start  = 0;
588     fl.l_len    = 1;
589     r = fcntl( fd, F_SETLKW, &fl );
590     close(fd);
591
592     return r;
593 }
594
595 /* kill the wine server holding the lock */
596 int kill_lock_owner( int sig )
597 {
598     int fd, i, ret = 0;
599     pid_t pid = 0;
600     struct flock fl;
601
602     create_server_dir();
603     fd = create_server_lock();
604
605     for (i = 0; i < 10; i++)
606     {
607         fl.l_type   = F_WRLCK;
608         fl.l_whence = SEEK_SET;
609         fl.l_start  = 0;
610         fl.l_len    = 1;
611         if (fcntl( fd, F_GETLK, &fl ) == -1) goto done;
612         if (fl.l_type != F_WRLCK) goto done;  /* the file is not locked */
613         if (!pid)  /* first time around */
614         {
615             if (!(pid = fl.l_pid)) goto done;  /* shouldn't happen */
616             if (sig == -1)
617             {
618                 if (kill( pid, SIGINT ) == -1) goto done;
619                 kill( pid, SIGCONT );
620                 ret = 1;
621             }
622             else  /* just send the specified signal and return */
623             {
624                 ret = (kill( pid, sig ) != -1);
625                 goto done;
626             }
627         }
628         else if (fl.l_pid != pid) goto done;  /* no longer the same process */
629         sleep( 1 );
630     }
631     /* waited long enough, now kill it */
632     kill( pid, SIGKILL );
633
634  done:
635     close( fd );
636     return ret;
637 }
638
639 /* acquire the main server lock */
640 static void acquire_lock(void)
641 {
642     struct sockaddr_un addr;
643     struct stat st;
644     struct flock fl;
645     int fd, slen, got_lock = 0;
646
647     fd = create_server_lock();
648
649     fl.l_type   = F_WRLCK;
650     fl.l_whence = SEEK_SET;
651     fl.l_start  = 0;
652     fl.l_len    = 1;
653     if (fcntl( fd, F_SETLK, &fl ) != -1)
654     {
655         /* check for crashed server */
656         if (stat( server_socket_name, &st ) != -1 &&   /* there is a leftover socket */
657             stat( "core", &st ) != -1 && st.st_size)   /* and there is a non-empty core file */
658         {
659             fprintf( stderr,
660                      "Warning: a previous instance of the wine server seems to have crashed.\n"
661                      "Please run 'gdb %s %s/core',\n"
662                      "type 'backtrace' at the gdb prompt and report the results. Thanks.\n\n",
663                      server_argv0, wine_get_server_dir() );
664         }
665         unlink( server_socket_name ); /* we got the lock, we can safely remove the socket */
666         got_lock = 1;
667         /* in that case we reuse fd without closing it, this ensures
668          * that we hold the lock until the process exits */
669     }
670     else
671     {
672         switch(errno)
673         {
674         case ENOLCK:
675             break;
676         case EACCES:
677             /* check whether locks work at all on this file system */
678             if (fcntl( fd, F_GETLK, &fl ) == -1) break;
679             /* fall through */
680         case EAGAIN:
681             exit(2); /* we didn't get the lock, exit with special status */
682         default:
683             fatal_perror( "fcntl %s/%s", wine_get_server_dir(), server_lock_name );
684         }
685         /* it seems we can't use locks on this fs, so we will use the socket existence as lock */
686         close( fd );
687     }
688
689     if ((fd = socket( AF_UNIX, SOCK_STREAM, 0 )) == -1) fatal_perror( "socket" );
690     addr.sun_family = AF_UNIX;
691     strcpy( addr.sun_path, server_socket_name );
692     slen = sizeof(addr) - sizeof(addr.sun_path) + strlen(addr.sun_path) + 1;
693 #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
694     addr.sun_len = slen;
695 #endif
696     if (bind( fd, (struct sockaddr *)&addr, slen ) == -1)
697     {
698         if ((errno == EEXIST) || (errno == EADDRINUSE))
699         {
700             if (got_lock)
701                 fatal_error( "couldn't bind to the socket even though we hold the lock\n" );
702             exit(2); /* we didn't get the lock, exit with special status */
703         }
704         fatal_perror( "bind" );
705     }
706     atexit( socket_cleanup );
707     chmod( server_socket_name, 0600 );  /* make sure no other user can connect */
708     if (listen( fd, 5 ) == -1) fatal_perror( "listen" );
709
710     if (!(master_socket = alloc_object( &master_socket_ops )) ||
711         !(master_socket->fd = create_anonymous_fd( &master_socket_fd_ops, fd, &master_socket->obj )))
712         fatal_error( "out of memory\n" );
713     master_socket->timeout = NULL;
714     set_fd_events( master_socket->fd, POLLIN );
715     make_object_static( &master_socket->obj );
716 }
717
718 /* open the master server socket and start waiting for new clients */
719 void open_master_socket(void)
720 {
721     int fd, pid, status, sync_pipe[2];
722     char dummy;
723
724     /* make sure no request is larger than the maximum size */
725     assert( sizeof(union generic_request) == sizeof(struct request_max_size) );
726     assert( sizeof(union generic_reply) == sizeof(struct request_max_size) );
727
728     create_server_dir();
729
730     if (!foreground)
731     {
732         if (pipe( sync_pipe ) == -1) fatal_perror( "pipe" );
733         pid = fork();
734         switch( pid )
735         {
736         case 0:  /* child */
737             setsid();
738             close( sync_pipe[0] );
739
740             acquire_lock();
741
742             /* close stdin and stdout */
743             if ((fd = open( "/dev/null", O_RDWR )) != -1)
744             {
745                 dup2( fd, 0 );
746                 dup2( fd, 1 );
747                 close( fd );
748             }
749
750             /* signal parent */
751             dummy = 0;
752             write( sync_pipe[1], &dummy, 1 );
753             close( sync_pipe[1] );
754             break;
755
756         case -1:
757             fatal_perror( "fork" );
758             break;
759
760         default:  /* parent */
761             close( sync_pipe[1] );
762
763             /* wait for child to signal us and then exit */
764             if (read( sync_pipe[0], &dummy, 1 ) == 1) _exit(0);
765
766             /* child terminated, propagate exit status */
767             wait4( pid, &status, 0, NULL );
768             if (WIFEXITED(status)) _exit( WEXITSTATUS(status) );
769             _exit(1);
770         }
771     }
772     else  /* remain in the foreground */
773     {
774         acquire_lock();
775     }
776
777     /* setup msghdr structure constant fields */
778     msghdr.msg_name    = NULL;
779     msghdr.msg_namelen = 0;
780     msghdr.msg_iov     = &myiovec;
781     msghdr.msg_iovlen  = 1;
782
783     /* init startup time */
784     server_start_time = time(NULL);
785 }
786
787 /* master socket timer expiration handler */
788 static void close_socket_timeout( void *arg )
789 {
790     master_socket->timeout = NULL;
791     flush_registry();
792
793     /* if a new client is waiting, we keep on running */
794     if (check_fd_events( master_socket->fd, POLLIN )) return;
795
796     if (debug_level) fprintf( stderr, "wineserver: exiting (pid=%ld)\n", (long) getpid() );
797
798 #ifdef DEBUG_OBJECTS
799     close_objects();  /* shut down everything properly */
800 #endif
801     exit(0);
802 }
803
804 /* close the master socket and stop waiting for new clients */
805 void close_master_socket(void)
806 {
807     struct timeval when;
808
809     if (master_socket_timeout == -1) return;  /* just keep running forever */
810
811     if (master_socket_timeout)
812     {
813         gettimeofday( &when, NULL );
814         add_timeout( &when, master_socket_timeout * 1000 );
815         master_socket->timeout = add_timeout_user( &when, close_socket_timeout, NULL );
816     }
817     else close_socket_timeout( NULL );  /* close it right away */
818 }
819
820 /* lock/unlock the master socket to stop accepting new clients */
821 void lock_master_socket( int locked )
822 {
823     set_fd_events( master_socket->fd, locked ? 0 : POLLIN );
824 }