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