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 #ifdef HAVE_DIRENT_H
  27 # include <dirent.h>
  28 #endif
  29 #include <errno.h>
  30 #include <fcntl.h>
  31 #include <signal.h>
  32 #include <stdarg.h>
  33 #include <stdio.h>
  34 #include <string.h>
  35 #include <sys/types.h>
  36 #ifdef HAVE_SYS_SOCKET_H
  37 # include <sys/socket.h>
  38 #endif
  39 #ifdef HAVE_SYS_WAIT_H
  40 #include <sys/wait.h>
  41 #endif
  42 #ifdef HAVE_SYS_UN_H
  43 #include <sys/un.h>
  44 #endif
  45 #ifdef HAVE_SYS_MMAN_H
  46 #include <sys/mman.h>
  47 #endif
  48 #ifdef HAVE_SYS_STAT_H
  49 # include <sys/stat.h>
  50 #endif
  51 #ifdef HAVE_SYS_UIO_H
  52 #include <sys/uio.h>
  53 #endif
  54 #ifdef HAVE_SYS_THR_H
  55 #include <sys/ucontext.h>
  56 #include <sys/thr.h>
  57 #endif
  58 #ifdef HAVE_UNISTD_H
  59 # include <unistd.h>
  60 #endif
  61
  62 #include "ntstatus.h"
  63 #define WIN32_NO_STATUS
  64 #include "wine/library.h"
  65 #include "wine/server.h"
  66 #include "wine/debug.h"
  67 #include "ntdll_misc.h"
  68
  69 WINE_DEFAULT_DEBUG_CHANNEL(server);
  70
  71 /* Some versions of glibc don't define this */
  72 #ifndef SCM_RIGHTS
  73 #define SCM_RIGHTS 1
  74 #endif
  75
  76 #ifndef MSG_CMSG_CLOEXEC
  77 #define MSG_CMSG_CLOEXEC 0
  78 #endif
  79
  80 #define SOCKETNAME "socket"        /* name of the socket file */
  81 #define LOCKNAME   "lock"          /* name of the lock file */
  82
  83 #ifdef __i386__
  84 static const enum cpu_type client_cpu = CPU_x86;
  85 #elif defined(__x86_64__)
  86 static const enum cpu_type client_cpu = CPU_x86_64;
  87 #elif defined(__ALPHA__)
  88 static const enum cpu_type client_cpu = CPU_ALPHA;
  89 #elif defined(__powerpc__)
  90 static const enum cpu_type client_cpu = CPU_POWERPC;
  91 #elif defined(__sparc__)
  92 static const enum cpu_type client_cpu = CPU_SPARC;
  93 #else
  94 #error Unsupported CPU
  95 #endif
  96
  97 unsigned int server_cpus = 0;
  98 int is_wow64 = FALSE;
  99
 100 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
 101 /* data structure used to pass an fd with sendmsg/recvmsg */
 102 struct cmsg_fd
 103 {
 104     struct
 105     {
 106         size_t len;   /* size of structure */
 107         int    level; /* SOL_SOCKET */
 108         int    type;  /* SCM_RIGHTS */
 109     } header;
 110     int fd;          /* fd to pass */
 111 };
 112 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
 113
 114 timeout_t server_start_time = 0;  /* time of server startup */
 115
 116 sigset_t server_block_set;  /* signals to block during server calls */
 117 static int fd_socket = -1;  /* socket to exchange file descriptors with the server */
 118
 119 static RTL_CRITICAL_SECTION fd_cache_section;
 120 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
 121 {
 122     0, 0, &fd_cache_section,
 123     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
 124       0, 0, { (DWORD_PTR)(__FILE__ ": fd_cache_section") }
 125 };
 126 static RTL_CRITICAL_SECTION fd_cache_section = { &critsect_debug, -1, 0, 0, 0, 0 };
 127
 128
 129 #ifdef __GNUC__
 130 static void fatal_error( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
 131 static void fatal_perror( const char *err, ... ) __attribute__((noreturn, format(printf,1,2)));
 132 static void server_connect_error( const char *serverdir ) __attribute__((noreturn));
 133 #endif
 134
 135 /* die on a fatal error; use only during initialization */
 136 static void fatal_error( const char *err, ... )
 137 {
 138     va_list args;
 139
 140     va_start( args, err );
 141     fprintf( stderr, "wine: " );
 142     vfprintf( stderr, err, args );
 143     va_end( args );
 144     exit(1);
 145 }
 146
 147 /* die on a fatal error; use only during initialization */
 148 static void fatal_perror( const char *err, ... )
 149 {
 150     va_list args;
 151
 152     va_start( args, err );
 153     fprintf( stderr, "wine: " );
 154     vfprintf( stderr, err, args );
 155     perror( " " );
 156     va_end( args );
 157     exit(1);
 158 }
 159
 160
 161 /***********************************************************************
 162  *           server_protocol_error
 163  */
 164 void server_protocol_error( const char *err, ... )
 165 {
 166     va_list args;
 167
 168     va_start( args, err );
 169     fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
 170     vfprintf( stderr, err, args );
 171     va_end( args );
 172     abort_thread(1);
 173 }
 174
 175
 176 /***********************************************************************
 177  *           server_protocol_perror
 178  */
 179 void server_protocol_perror( const char *err )
 180 {
 181     fprintf( stderr, "wine client error:%x: ", GetCurrentThreadId() );
 182     perror( err );
 183     abort_thread(1);
 184 }
 185
 186
 187 /***********************************************************************
 188  *           send_request
 189  *
 190  * Send a request to the server.
 191  */
 192 static unsigned int send_request( const struct __server_request_info *req )
 193 {
 194     unsigned int i;
 195     int ret;
 196
 197     if (!req->u.req.request_header.request_size)
 198     {
 199         if ((ret = write( ntdll_get_thread_data()->request_fd, &req->u.req,
 200                           sizeof(req->u.req) )) == sizeof(req->u.req)) return STATUS_SUCCESS;
 201
 202     }
 203     else
 204     {
 205         struct iovec vec[__SERVER_MAX_DATA+1];
 206
 207         vec[0].iov_base = (void *)&req->u.req;
 208         vec[0].iov_len = sizeof(req->u.req);
 209         for (i = 0; i < req->data_count; i++)
 210         {
 211             vec[i+1].iov_base = (void *)req->data[i].ptr;
 212             vec[i+1].iov_len = req->data[i].size;
 213         }
 214         if ((ret = writev( ntdll_get_thread_data()->request_fd, vec, i+1 )) ==
 215             req->u.req.request_header.request_size + sizeof(req->u.req)) return STATUS_SUCCESS;
 216     }
 217
 218     if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
 219     if (errno == EPIPE) abort_thread(0);
 220     if (errno == EFAULT) return STATUS_ACCESS_VIOLATION;
 221     server_protocol_perror( "write" );
 222 }
 223
 224
 225 /***********************************************************************
 226  *           read_reply_data
 227  *
 228  * Read data from the reply buffer; helper for wait_reply.
 229  */
 230 static void read_reply_data( void *buffer, size_t size )
 231 {
 232     int ret;
 233
 234     for (;;)
 235     {
 236         if ((ret = read( ntdll_get_thread_data()->reply_fd, buffer, size )) > 0)
 237         {
 238             if (!(size -= ret)) return;
 239             buffer = (char *)buffer + ret;
 240             continue;
 241         }
 242         if (!ret) break;
 243         if (errno == EINTR) continue;
 244         if (errno == EPIPE) break;
 245         server_protocol_perror("read");
 246     }
 247     /* the server closed the connection; time to die... */
 248     abort_thread(0);
 249 }
 250
 251
 252 /***********************************************************************
 253  *           wait_reply
 254  *
 255  * Wait for a reply from the server.
 256  */
 257 static inline unsigned int wait_reply( struct __server_request_info *req )
 258 {
 259     read_reply_data( &req->u.reply, sizeof(req->u.reply) );
 260     if (req->u.reply.reply_header.reply_size)
 261         read_reply_data( req->reply_data, req->u.reply.reply_header.reply_size );
 262     return req->u.reply.reply_header.error;
 263 }
 264
 265
 266 /***********************************************************************
 267  *           wine_server_call (NTDLL.@)
 268  *
 269  * Perform a server call.
 270  *
 271  * PARAMS
 272  *     req_ptr [I/O] Function dependent data
 273  *
 274  * RETURNS
 275  *     Depends on server function being called, but usually an NTSTATUS code.
 276  *
 277  * NOTES
 278  *     Use the SERVER_START_REQ and SERVER_END_REQ to help you fill out the
 279  *     server request structure for the particular call. E.g:
 280  *|     SERVER_START_REQ( event_op )
 281  *|     {
 282  *|         req->handle = handle;
 283  *|         req->op     = SET_EVENT;
 284  *|         ret = wine_server_call( req );
 285  *|     }
 286  *|     SERVER_END_REQ;
 287  */
 288 unsigned int wine_server_call( void *req_ptr )
 289 {
 290     struct __server_request_info * const req = req_ptr;
 291     sigset_t old_set;
 292     unsigned int ret;
 293
 294     pthread_sigmask( SIG_BLOCK, &server_block_set, &old_set );
 295     ret = send_request( req );
 296     if (!ret) ret = wait_reply( req );
 297     pthread_sigmask( SIG_SETMASK, &old_set, NULL );
 298     return ret;
 299 }
 300
 301
 302 /***********************************************************************
 303  *           server_enter_uninterrupted_section
 304  */
 305 void server_enter_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
 306 {
 307     pthread_sigmask( SIG_BLOCK, &server_block_set, sigset );
 308     RtlEnterCriticalSection( cs );
 309 }
 310
 311
 312 /***********************************************************************
 313  *           server_leave_uninterrupted_section
 314  */
 315 void server_leave_uninterrupted_section( RTL_CRITICAL_SECTION *cs, sigset_t *sigset )
 316 {
 317     RtlLeaveCriticalSection( cs );
 318     pthread_sigmask( SIG_SETMASK, sigset, NULL );
 319 }
 320
 321
 322 /***********************************************************************
 323  *           wine_server_send_fd   (NTDLL.@)
 324  *
 325  * Send a file descriptor to the server.
 326  *
 327  * PARAMS
 328  *     fd [I] file descriptor to send
 329  *
 330  * RETURNS
 331  *     nothing
 332  */
 333 void CDECL wine_server_send_fd( int fd )
 334 {
 335 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
 336     struct cmsg_fd cmsg;
 337 #endif
 338     struct send_fd data;
 339     struct msghdr msghdr;
 340     struct iovec vec;
 341     int ret;
 342
 343     vec.iov_base = (void *)&data;
 344     vec.iov_len  = sizeof(data);
 345
 346     msghdr.msg_name    = NULL;
 347     msghdr.msg_namelen = 0;
 348     msghdr.msg_iov     = &vec;
 349     msghdr.msg_iovlen  = 1;
 350
 351 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
 352     msghdr.msg_accrights    = (void *)&fd;
 353     msghdr.msg_accrightslen = sizeof(fd);
 354 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
 355     cmsg.header.len   = sizeof(cmsg.header) + sizeof(fd);
 356     cmsg.header.level = SOL_SOCKET;
 357     cmsg.header.type  = SCM_RIGHTS;
 358     cmsg.fd           = fd;
 359     msghdr.msg_control    = &cmsg;
 360     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
 361     msghdr.msg_flags      = 0;
 362 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
 363
 364     data.tid = GetCurrentThreadId();
 365     data.fd  = fd;
 366
 367     for (;;)
 368     {
 369         if ((ret = sendmsg( fd_socket, &msghdr, 0 )) == sizeof(data)) return;
 370         if (ret >= 0) server_protocol_error( "partial write %d\n", ret );
 371         if (errno == EINTR) continue;
 372         if (errno == EPIPE) abort_thread(0);
 373         server_protocol_perror( "sendmsg" );
 374     }
 375 }
 376
 377
 378 /***********************************************************************
 379  *           receive_fd
 380  *
 381  * Receive a file descriptor passed from the server.
 382  */
 383 static int receive_fd( obj_handle_t *handle )
 384 {
 385     struct iovec vec;
 386     int ret, fd;
 387
 388 #ifdef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
 389     struct msghdr msghdr;
 390
 391     fd = -1;
 392     msghdr.msg_accrights    = (void *)&fd;
 393     msghdr.msg_accrightslen = sizeof(fd);
 394 #else  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
 395     struct msghdr msghdr;
 396     struct cmsg_fd cmsg;
 397
 398     cmsg.header.len   = sizeof(cmsg.header) + sizeof(fd);
 399     cmsg.header.level = SOL_SOCKET;
 400     cmsg.header.type  = SCM_RIGHTS;
 401     cmsg.fd           = -1;
 402     msghdr.msg_control    = &cmsg;
 403     msghdr.msg_controllen = sizeof(cmsg.header) + sizeof(fd);
 404     msghdr.msg_flags      = 0;
 405 #endif  /* HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS */
 406
 407     msghdr.msg_name    = NULL;
 408     msghdr.msg_namelen = 0;
 409     msghdr.msg_iov     = &vec;
 410     msghdr.msg_iovlen  = 1;
 411     vec.iov_base = (void *)handle;
 412     vec.iov_len  = sizeof(*handle);
 413
 414     for (;;)
 415     {
 416         if ((ret = recvmsg( fd_socket, &msghdr, MSG_CMSG_CLOEXEC )) > 0)
 417         {
 418 #ifndef HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS
 419             fd = cmsg.fd;
 420 #endif
 421             if (fd != -1) fcntl( fd, F_SETFD, FD_CLOEXEC ); /* in case MSG_CMSG_CLOEXEC is not supported */
 422             return fd;
 423         }
 424         if (!ret) break;
 425         if (errno == EINTR) continue;
 426         if (errno == EPIPE) break;
 427         server_protocol_perror("recvmsg");
 428     }
 429     /* the server closed the connection; time to die... */
 430     abort_thread(0);
 431 }
 432
 433
 434 /***********************************************************************/
 435 /* fd cache support */
 436
 437 struct fd_cache_entry
 438 {
 439     int fd;
 440     enum server_fd_type type : 6;
 441     unsigned int        access : 2;
 442     unsigned int        options : 24;
 443 };
 444
 445 #define FD_CACHE_BLOCK_SIZE  (65536 / sizeof(struct fd_cache_entry))
 446 #define FD_CACHE_ENTRIES     128
 447
 448 static struct fd_cache_entry *fd_cache[FD_CACHE_ENTRIES];
 449 static struct fd_cache_entry fd_cache_initial_block[FD_CACHE_BLOCK_SIZE];
 450
 451 static inline unsigned int handle_to_index( HANDLE handle, unsigned int *entry )
 452 {
 453     unsigned int idx = (wine_server_obj_handle(handle) >> 2) - 1;
 454     *entry = idx / FD_CACHE_BLOCK_SIZE;
 455     return idx % FD_CACHE_BLOCK_SIZE;
 456 }
 457
 458
 459 /***********************************************************************
 460  *           add_fd_to_cache
 461  *
 462  * Caller must hold fd_cache_section.
 463  */
 464 static int add_fd_to_cache( HANDLE handle, int fd, enum server_fd_type type,
 465                             unsigned int access, unsigned int options )
 466 {
 467     unsigned int entry, idx = handle_to_index( handle, &entry );
 468     int prev_fd;
 469
 470     if (entry >= FD_CACHE_ENTRIES)
 471     {
 472         FIXME( "too many allocated handles, not caching %p\n", handle );
 473         return 0;
 474     }
 475
 476     if (!fd_cache[entry])  /* do we need to allocate a new block of entries? */
 477     {
 478         if (!entry) fd_cache[0] = fd_cache_initial_block;
 479         else
 480         {
 481             void *ptr = wine_anon_mmap( NULL, FD_CACHE_BLOCK_SIZE * sizeof(struct fd_cache_entry),
 482                                         PROT_READ | PROT_WRITE, 0 );
 483             if (ptr == MAP_FAILED) return 0;
 484             fd_cache[entry] = ptr;
 485         }
 486     }
 487     /* store fd+1 so that 0 can be used as the unset value */
 488     prev_fd = interlocked_xchg( &fd_cache[entry][idx].fd, fd + 1 ) - 1;
 489     fd_cache[entry][idx].type = type;
 490     fd_cache[entry][idx].access = access;
 491     fd_cache[entry][idx].options = options;
 492     if (prev_fd != -1) close( prev_fd );
 493     return 1;
 494 }
 495
 496
 497 /***********************************************************************
 498  *           get_cached_fd
 499  *
 500  * Caller must hold fd_cache_section.
 501  */
 502 static inline int get_cached_fd( HANDLE handle, enum server_fd_type *type,
 503                                  unsigned int *access, unsigned int *options )
 504 {
 505     unsigned int entry, idx = handle_to_index( handle, &entry );
 506     int fd = -1;
 507
 508     if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
 509     {
 510         fd = fd_cache[entry][idx].fd - 1;
 511         if (type) *type = fd_cache[entry][idx].type;
 512         if (access) *access = fd_cache[entry][idx].access;
 513         if (options) *options = fd_cache[entry][idx].options;
 514     }
 515     return fd;
 516 }
 517
 518
 519 /***********************************************************************
 520  *           server_remove_fd_from_cache
 521  */
 522 int server_remove_fd_from_cache( HANDLE handle )
 523 {
 524     unsigned int entry, idx = handle_to_index( handle, &entry );
 525     int fd = -1;
 526
 527     if (entry < FD_CACHE_ENTRIES && fd_cache[entry])
 528         fd = interlocked_xchg( &fd_cache[entry][idx].fd, 0 ) - 1;
 529
 530     return fd;
 531 }
 532
 533
 534 /***********************************************************************
 535  *           server_get_unix_fd
 536  *
 537  * The returned unix_fd should be closed iff needs_close is non-zero.
 538  */
 539 int server_get_unix_fd( HANDLE handle, unsigned int wanted_access, int *unix_fd,
 540                         int *needs_close, enum server_fd_type *type, unsigned int *options )
 541 {
 542     sigset_t sigset;
 543     obj_handle_t fd_handle;
 544     int ret = 0, fd;
 545     unsigned int access = 0;
 546
 547     *unix_fd = -1;
 548     *needs_close = 0;
 549     wanted_access &= FILE_READ_DATA | FILE_WRITE_DATA;
 550
 551     server_enter_uninterrupted_section( &fd_cache_section, &sigset );
 552
 553     fd = get_cached_fd( handle, type, &access, options );
 554     if (fd != -1) goto done;
 555
 556     SERVER_START_REQ( get_handle_fd )
 557     {
 558         req->handle = wine_server_obj_handle( handle );
 559         if (!(ret = wine_server_call( req )))
 560         {
 561             if (type) *type = reply->type;
 562             if (options) *options = reply->options;
 563             access = reply->access;
 564             if ((fd = receive_fd( &fd_handle )) != -1)
 565             {
 566                 assert( wine_server_ptr_handle(fd_handle) == handle );
 567                 *needs_close = (reply->removable ||
 568                                 !add_fd_to_cache( handle, fd, reply->type,
 569                                                   reply->access, reply->options ));
 570             }
 571             else ret = STATUS_TOO_MANY_OPENED_FILES;
 572         }
 573     }
 574     SERVER_END_REQ;
 575
 576 done:
 577     server_leave_uninterrupted_section( &fd_cache_section, &sigset );
 578     if (!ret && ((access & wanted_access) != wanted_access))
 579     {
 580         ret = STATUS_ACCESS_DENIED;
 581         if (*needs_close) close( fd );
 582     }
 583     if (!ret) *unix_fd = fd;
 584     return ret;
 585 }
 586
 587
 588 /***********************************************************************
 589  *           wine_server_fd_to_handle   (NTDLL.@)
 590  *
 591  * Allocate a file handle for a Unix file descriptor.
 592  *
 593  * PARAMS
 594  *     fd      [I] Unix file descriptor.
 595  *     access  [I] Win32 access flags.
 596  *     attributes [I] Object attributes.
 597  *     handle  [O] Address where Wine file handle will be stored.
 598  *
 599  * RETURNS
 600  *     NTSTATUS code
 601  */
 602 int CDECL wine_server_fd_to_handle( int fd, unsigned int access, unsigned int attributes, HANDLE *handle )
 603 {
 604     int ret;
 605
 606     *handle = 0;
 607     wine_server_send_fd( fd );
 608
 609     SERVER_START_REQ( alloc_file_handle )
 610     {
 611         req->access     = access;
 612         req->attributes = attributes;
 613         req->fd         = fd;
 614         if (!(ret = wine_server_call( req ))) *handle = wine_server_ptr_handle( reply->handle );
 615     }
 616     SERVER_END_REQ;
 617     return ret;
 618 }
 619
 620
 621 /***********************************************************************
 622  *           wine_server_handle_to_fd   (NTDLL.@)
 623  *
 624  * Retrieve the file descriptor corresponding to a file handle.
 625  *
 626  * PARAMS
 627  *     handle  [I] Wine file handle.
 628  *     access  [I] Win32 file access rights requested.
 629  *     unix_fd [O] Address where Unix file descriptor will be stored.
 630  *     options [O] Address where the file open options will be stored. Optional.
 631  *
 632  * RETURNS
 633  *     NTSTATUS code
 634  */
 635 int CDECL wine_server_handle_to_fd( HANDLE handle, unsigned int access, int *unix_fd,
 636                               unsigned int *options )
 637 {
 638     int needs_close, ret = server_get_unix_fd( handle, access, unix_fd, &needs_close, NULL, options );
 639
 640     if (!ret && !needs_close)
 641     {
 642         if ((*unix_fd = dup(*unix_fd)) == -1) ret = FILE_GetNtStatus();
 643     }
 644     return ret;
 645 }
 646
 647
 648 /***********************************************************************
 649  *           wine_server_release_fd   (NTDLL.@)
 650  *
 651  * Release the Unix file descriptor returned by wine_server_handle_to_fd.
 652  *
 653  * PARAMS
 654  *     handle  [I] Wine file handle.
 655  *     unix_fd [I] Unix file descriptor to release.
 656  *
 657  * RETURNS
 658  *     nothing
 659  */
 660 void CDECL wine_server_release_fd( HANDLE handle, int unix_fd )
 661 {
 662     close( unix_fd );
 663 }
 664
 665
 666 /***********************************************************************
 667  *           server_pipe
 668  *
 669  * Create a pipe for communicating with the server.
 670  */
 671 int server_pipe( int fd[2] )
 672 {
 673     int ret;
 674 #ifdef HAVE_PIPE2
 675     static int have_pipe2 = 1;
 676
 677     if (have_pipe2)
 678     {
 679         if (!(ret = pipe2( fd, O_CLOEXEC ))) return ret;
 680         if (errno == ENOSYS || errno == EINVAL) have_pipe2 = 0;  /* don't try again */
 681     }
 682 #endif
 683     if (!(ret = pipe( fd )))
 684     {
 685         fcntl( fd[0], F_SETFD, FD_CLOEXEC );
 686         fcntl( fd[1], F_SETFD, FD_CLOEXEC );
 687     }
 688     return ret;
 689 }
 690
 691
 692 /***********************************************************************
 693  *           start_server
 694  *
 695  * Start a new wine server.
 696  */
 697 static void start_server(void)
 698 {
 699     static int started;  /* we only try once */
 700     char *argv[3];
 701     static char wineserver[] = "server/wineserver";
 702     static char debug[] = "-d";
 703
 704     if (!started)
 705     {
 706         int status;
 707         int pid = fork();
 708         if (pid == -1) fatal_perror( "fork" );
 709         if (!pid)
 710         {
 711             argv[0] = wineserver;
 712             argv[1] = TRACE_ON(server) ? debug : NULL;
 713             argv[2] = NULL;
 714             wine_exec_wine_binary( argv[0], argv, getenv("WINESERVER") );
 715             fatal_error( "could not exec wineserver\n" );
 716         }
 717         waitpid( pid, &status, 0 );
 718         status = WIFEXITED(status) ? WEXITSTATUS(status) : 1;
 719         if (status == 2) return;  /* server lock held by someone else, will retry later */
 720         if (status) exit(status);  /* server failed */
 721         started = 1;
 722     }
 723 }
 724
 725
 726 /***********************************************************************
 727  *           setup_config_dir
 728  *
 729  * Setup the wine configuration dir.
 730  */
 731 static void setup_config_dir(void)
 732 {
 733     const char *p, *config_dir = wine_get_config_dir();
 734
 735     if (chdir( config_dir ) == -1)
 736     {
 737         if (errno != ENOENT) fatal_perror( "chdir to %s\n", config_dir );
 738
 739         if ((p = strrchr( config_dir, '/' )) && p != config_dir)
 740         {
 741             struct stat st;
 742             char *tmp_dir;
 743
 744             if (!(tmp_dir = malloc( p + 1 - config_dir ))) fatal_error( "out of memory\n" );
 745             memcpy( tmp_dir, config_dir, p - config_dir );
 746             tmp_dir[p - config_dir] = 0;
 747             if (!stat( tmp_dir, &st ) && st.st_uid != getuid())
 748                 fatal_error( "'%s' is not owned by you, refusing to create a configuration directory there\n",
 749                              tmp_dir );
 750             free( tmp_dir );
 751         }
 752
 753         mkdir( config_dir, 0777 );
 754         if (chdir( config_dir ) == -1) fatal_perror( "chdir to %s\n", config_dir );
 755
 756         if ((p = getenv( "WINEARCH" )) && !strcmp( p, "win32" ))
 757         {
 758             /* force creation of a 32-bit prefix */
 759             int fd = open( "system.reg", O_WRONLY | O_CREAT | O_EXCL, 0666 );
 760             if (fd != -1)
 761             {
 762                 static const char regfile[] = "WINE REGISTRY Version 2\n\n#arch=win32\n";
 763                 write( fd, regfile, sizeof(regfile) - 1 );
 764                 close( fd );
 765             }
 766         }
 767         MESSAGE( "wine: created the configuration directory '%s'\n", config_dir );
 768     }
 769
 770     if (mkdir( "dosdevices", 0777 ) == -1)
 771     {
 772         if (errno == EEXIST) return;
 773         fatal_perror( "cannot create %s/dosdevices\n", config_dir );
 774     }
 775
 776     /* create the drive symlinks */
 777
 778     mkdir( "drive_c", 0777 );
 779     symlink( "../drive_c", "dosdevices/c:" );
 780     symlink( "/", "dosdevices/z:" );
 781 }
 782
 783
 784 /***********************************************************************
 785  *           server_connect_error
 786  *
 787  * Try to display a meaningful explanation of why we couldn't connect
 788  * to the server.
 789  */
 790 static void server_connect_error( const char *serverdir )
 791 {
 792     int fd;
 793     struct flock fl;
 794
 795     if ((fd = open( LOCKNAME, O_WRONLY )) == -1)
 796         fatal_error( "for some mysterious reason, the wine server never started.\n" );
 797
 798     fl.l_type   = F_WRLCK;
 799     fl.l_whence = SEEK_SET;
 800     fl.l_start  = 0;
 801     fl.l_len    = 1;
 802     if (fcntl( fd, F_GETLK, &fl ) != -1)
 803     {
 804         if (fl.l_type == F_WRLCK)  /* the file is locked */
 805             fatal_error( "a wine server seems to be running, but I cannot connect to it.\n"
 806                          "   You probably need to kill that process (it might be pid %d).\n",
 807                          (int)fl.l_pid );
 808         fatal_error( "for some mysterious reason, the wine server failed to run.\n" );
 809     }
 810     fatal_error( "the file system of '%s' doesn't support locks,\n"
 811           "   and there is a 'socket' file in that directory that prevents wine from starting.\n"
 812           "   You should make sure no wine server is running, remove that file and try again.\n",
 813                  serverdir );
 814 }
 815
 816
 817 /***********************************************************************
 818  *           server_connect
 819  *
 820  * Attempt to connect to an existing server socket.
 821  * We need to be in the server directory already.
 822  */
 823 static int server_connect(void)
 824 {
 825     const char *serverdir;
 826     struct sockaddr_un addr;
 827     struct stat st;
 828     int s, slen, retry, fd_cwd;
 829
 830     /* retrieve the current directory */
 831     fd_cwd = open( ".", O_RDONLY );
 832     if (fd_cwd != -1) fcntl( fd_cwd, F_SETFD, 1 ); /* set close on exec flag */
 833
 834     setup_config_dir();
 835     serverdir = wine_get_server_dir();
 836
 837     /* chdir to the server directory */
 838     if (chdir( serverdir ) == -1)
 839     {
 840         if (errno != ENOENT) fatal_perror( "chdir to %s", serverdir );
 841         start_server();
 842         if (chdir( serverdir ) == -1) fatal_perror( "chdir to %s", serverdir );
 843     }
 844
 845     /* make sure we are at the right place */
 846     if (stat( ".", &st ) == -1) fatal_perror( "stat %s", serverdir );
 847     if (st.st_uid != getuid()) fatal_error( "'%s' is not owned by you\n", serverdir );
 848     if (st.st_mode & 077) fatal_error( "'%s' must not be accessible by other users\n", serverdir );
 849
 850     for (retry = 0; retry < 6; retry++)
 851     {
 852         /* if not the first try, wait a bit to leave the previous server time to exit */
 853         if (retry)
 854         {
 855             usleep( 100000 * retry * retry );
 856             start_server();
 857             if (lstat( SOCKETNAME, &st ) == -1) continue;  /* still no socket, wait a bit more */
 858         }
 859         else if (lstat( SOCKETNAME, &st ) == -1) /* check for an already existing socket */
 860         {
 861             if (errno != ENOENT) fatal_perror( "lstat %s/%s", serverdir, SOCKETNAME );
 862             start_server();
 863             if (lstat( SOCKETNAME, &st ) == -1) continue;  /* still no socket, wait a bit more */
 864         }
 865
 866         /* make sure the socket is sane (ISFIFO needed for Solaris) */
 867         if (!S_ISSOCK(st.st_mode) && !S_ISFIFO(st.st_mode))
 868             fatal_error( "'%s/%s' is not a socket\n", serverdir, SOCKETNAME );
 869         if (st.st_uid != getuid())
 870             fatal_error( "'%s/%s' is not owned by you\n", serverdir, SOCKETNAME );
 871
 872         /* try to connect to it */
 873         addr.sun_family = AF_UNIX;
 874         strcpy( addr.sun_path, SOCKETNAME );
 875         slen = sizeof(addr) - sizeof(addr.sun_path) + strlen(addr.sun_path) + 1;
 876 #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
 877         addr.sun_len = slen;
 878 #endif
 879         if ((s = socket( AF_UNIX, SOCK_STREAM, 0 )) == -1) fatal_perror( "socket" );
 880         if (connect( s, (struct sockaddr *)&addr, slen ) != -1)
 881         {
 882             /* switch back to the starting directory */
 883             if (fd_cwd != -1)
 884             {
 885                 fchdir( fd_cwd );
 886                 close( fd_cwd );
 887             }
 888             fcntl( s, F_SETFD, 1 ); /* set close on exec flag */
 889             return s;
 890         }
 891         close( s );
 892     }
 893     server_connect_error( serverdir );
 894 }
 895
 896
 897 #ifdef __APPLE__
 898 #include <mach/mach.h>
 899 #include <mach/mach_error.h>
 900 #include <servers/bootstrap.h>
 901
 902 /* send our task port to the server */
 903 static void send_server_task_port(void)
 904 {
 905     mach_port_t bootstrap_port, wineserver_port;
 906     kern_return_t kret;
 907
 908     struct {
 909         mach_msg_header_t           header;
 910         mach_msg_body_t             body;
 911         mach_msg_port_descriptor_t  task_port;
 912     } msg;
 913
 914     if (task_get_bootstrap_port(mach_task_self(), &bootstrap_port) != KERN_SUCCESS) return;
 915
 916     kret = bootstrap_look_up(bootstrap_port, (char*)wine_get_server_dir(), &wineserver_port);
 917     if (kret != KERN_SUCCESS)
 918         fatal_error( "cannot find the server port: 0x%08x\n", kret );
 919
 920     mach_port_deallocate(mach_task_self(), bootstrap_port);
 921
 922     msg.header.msgh_bits        = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, 0) | MACH_MSGH_BITS_COMPLEX;
 923     msg.header.msgh_size        = sizeof(msg);
 924     msg.header.msgh_remote_port = wineserver_port;
 925     msg.header.msgh_local_port  = MACH_PORT_NULL;
 926
 927     msg.body.msgh_descriptor_count  = 1;
 928     msg.task_port.name              = mach_task_self();
 929     msg.task_port.disposition       = MACH_MSG_TYPE_COPY_SEND;
 930     msg.task_port.type              = MACH_MSG_PORT_DESCRIPTOR;
 931
 932     kret = mach_msg_send(&msg.header);
 933     if (kret != KERN_SUCCESS)
 934         server_protocol_error( "mach_msg_send failed: 0x%08x\n", kret );
 935
 936     mach_port_deallocate(mach_task_self(), wineserver_port);
 937 }
 938 #endif  /* __APPLE__ */
 939
 940
 941 /***********************************************************************
 942  *           get_unix_tid
 943  *
 944  * Retrieve the Unix tid to use on the server side for the current thread.
 945  */
 946 static int get_unix_tid(void)
 947 {
 948     int ret = -1;
 949 #if defined(linux) && defined(__i386__)
 950     __asm__("int $0x80" : "=a" (ret) : "0" (224) /* SYS_gettid */);
 951 #elif defined(linux) && defined(__x86_64__)
 952     __asm__("syscall" : "=a" (ret) : "0" (186) /* SYS_gettid */);
 953 #elif defined(__sun)
 954     ret = pthread_self();
 955 #elif defined(__APPLE__)
 956     ret = mach_thread_self();
 957     mach_port_deallocate(mach_task_self(), ret);
 958 #elif defined(__FreeBSD__)
 959     long lwpid;
 960     thr_self( &lwpid );
 961     ret = lwpid;
 962 #endif
 963     return ret;
 964 }
 965
 966
 967 /***********************************************************************
 968  *           server_init_process
 969  *
 970  * Start the server and create the initial socket pair.
 971  */
 972 void server_init_process(void)
 973 {
 974     obj_handle_t version;
 975     const char *env_socket = getenv( "WINESERVERSOCKET" );
 976
 977     if (env_socket)
 978     {
 979         fd_socket = atoi( env_socket );
 980         if (fcntl( fd_socket, F_SETFD, 1 ) == -1)
 981             fatal_perror( "Bad server socket %d", fd_socket );
 982         unsetenv( "WINESERVERSOCKET" );
 983     }
 984     else fd_socket = server_connect();
 985
 986     /* setup the signal mask */
 987     sigemptyset( &server_block_set );
 988     sigaddset( &server_block_set, SIGALRM );
 989     sigaddset( &server_block_set, SIGIO );
 990     sigaddset( &server_block_set, SIGINT );
 991     sigaddset( &server_block_set, SIGHUP );
 992     sigaddset( &server_block_set, SIGUSR1 );
 993     sigaddset( &server_block_set, SIGUSR2 );
 994     sigaddset( &server_block_set, SIGCHLD );
 995     pthread_sigmask( SIG_BLOCK, &server_block_set, NULL );
 996
 997     /* receive the first thread request fd on the main socket */
 998     ntdll_get_thread_data()->request_fd = receive_fd( &version );
 999
1000     if (version != SERVER_PROTOCOL_VERSION)
1001         server_protocol_error( "version mismatch %d/%d.\n"
1002                                "Your %s binary was not upgraded correctly,\n"
1003                                "or you have an older one somewhere in your PATH.\n"
1004                                "Or maybe the wrong wineserver is still running?\n",
1005                                version, SERVER_PROTOCOL_VERSION,
1006                                (version > SERVER_PROTOCOL_VERSION) ? "wine" : "wineserver" );
1007 #ifdef __APPLE__
1008     send_server_task_port();
1009 #endif
1010 }
1011
1012
1013 /***********************************************************************
1014  *           server_init_process_done
1015  */
1016 NTSTATUS server_init_process_done(void)
1017 {
1018     PEB *peb = NtCurrentTeb()->Peb;
1019     IMAGE_NT_HEADERS *nt = RtlImageNtHeader( peb->ImageBaseAddress );
1020     NTSTATUS status;
1021
1022     /* Install signal handlers; this cannot be done earlier, since we cannot
1023      * send exceptions to the debugger before the create process event that
1024      * is sent by REQ_INIT_PROCESS_DONE.
1025      * We do need the handlers in place by the time the request is over, so
1026      * we set them up here. If we segfault between here and the server call
1027      * something is very wrong... */
1028     signal_init_process();
1029
1030     /* Signal the parent process to continue */
1031     SERVER_START_REQ( init_process_done )
1032     {
1033         req->module   = wine_server_client_ptr( peb->ImageBaseAddress );
1034 #ifdef __i386__
1035         req->ldt_copy = wine_server_client_ptr( &wine_ldt_copy );
1036 #endif
1037         req->entry    = wine_server_client_ptr( (char *)peb->ImageBaseAddress + nt->OptionalHeader.AddressOfEntryPoint );
1038         req->gui      = (nt->OptionalHeader.Subsystem != IMAGE_SUBSYSTEM_WINDOWS_CUI);
1039         status = wine_server_call( req );
1040     }
1041     SERVER_END_REQ;
1042
1043     return status;
1044 }
1045
1046
1047 /***********************************************************************
1048  *           server_init_thread
1049  *
1050  * Send an init thread request. Return 0 if OK.
1051  */
1052 size_t server_init_thread( void *entry_point )
1053 {
1054     static const int is_win64 = (sizeof(void *) > sizeof(int));
1055     const char *arch = getenv( "WINEARCH" );
1056     int ret;
1057     int reply_pipe[2];
1058     struct sigaction sig_act;
1059     size_t info_size;
1060
1061     sig_act.sa_handler = SIG_IGN;
1062     sig_act.sa_flags   = 0;
1063     sigemptyset( &sig_act.sa_mask );
1064
1065     /* ignore SIGPIPE so that we get an EPIPE error instead  */
1066     sigaction( SIGPIPE, &sig_act, NULL );
1067     /* automatic child reaping to avoid zombies */
1068 #ifdef SA_NOCLDWAIT
1069     sig_act.sa_flags |= SA_NOCLDWAIT;
1070 #endif
1071     sigaction( SIGCHLD, &sig_act, NULL );
1072
1073     /* create the server->client communication pipes */
1074     if (server_pipe( reply_pipe ) == -1) server_protocol_perror( "pipe" );
1075     if (server_pipe( ntdll_get_thread_data()->wait_fd ) == -1) server_protocol_perror( "pipe" );
1076     wine_server_send_fd( reply_pipe[1] );
1077     wine_server_send_fd( ntdll_get_thread_data()->wait_fd[1] );
1078     ntdll_get_thread_data()->reply_fd = reply_pipe[0];
1079     close( reply_pipe[1] );
1080
1081     SERVER_START_REQ( init_thread )
1082     {
1083         req->unix_pid    = getpid();
1084         req->unix_tid    = get_unix_tid();
1085         req->teb         = wine_server_client_ptr( NtCurrentTeb() );
1086         req->entry       = wine_server_client_ptr( entry_point );
1087         req->reply_fd    = reply_pipe[1];
1088         req->wait_fd     = ntdll_get_thread_data()->wait_fd[1];
1089         req->debug_level = (TRACE_ON(server) != 0);
1090         req->cpu         = client_cpu;
1091         ret = wine_server_call( req );
1092         NtCurrentTeb()->ClientId.UniqueProcess = ULongToHandle(reply->pid);
1093         NtCurrentTeb()->ClientId.UniqueThread  = ULongToHandle(reply->tid);
1094         info_size         = reply->info_size;
1095         server_start_time = reply->server_start;
1096         server_cpus       = reply->all_cpus;
1097     }
1098     SERVER_END_REQ;
1099
1100     is_wow64 = !is_win64 && (server_cpus & (1 << CPU_x86_64)) != 0;
1101     ntdll_get_thread_data()->wow64_redir = is_wow64;
1102
1103     switch (ret)
1104     {
1105     case STATUS_SUCCESS:
1106         if (arch)
1107         {
1108             if (!strcmp( arch, "win32" ) && (is_win64 || is_wow64))
1109                 fatal_error( "WINEARCH set to win32 but '%s' is a 64-bit installation.\n",
1110                              wine_get_config_dir() );
1111             if (!strcmp( arch, "win64" ) && !is_wow64)
1112                 fatal_error( "WINEARCH set to win64 but '%s' is a 32-bit installation.\n",
1113                              wine_get_config_dir() );
1114         }
1115         return info_size;
1116     case STATUS_NOT_REGISTRY_FILE:
1117         fatal_error( "'%s' is a 32-bit installation, it cannot support 64-bit applications.\n",
1118                      wine_get_config_dir() );
1119     case STATUS_NOT_SUPPORTED:
1120         if (is_win64)
1121             fatal_error( "wineserver is 32-bit, it cannot support 64-bit applications.\n" );
1122         else
1123             fatal_error( "'%s' is a 64-bit installation, it cannot be used with a 32-bit wineserver.\n",
1124                          wine_get_config_dir() );
1125     default:
1126         server_protocol_error( "init_thread failed with status %x\n", ret );
1127     }
1128 }