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