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