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