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