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