server/sock.c

   1 /*
   2  * Server-side socket management
   3  *
   4  * Copyright (C) 1999 Marcus Meissner, Ove Kåven
   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  * FIXME: we use read|write access in all cases. Shouldn't we depend that
  21  * on the access of the current handle?
  22  */
  23
  24 #include "config.h"
  25
  26 #include <assert.h>
  27 #include <fcntl.h>
  28 #include <stdarg.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <stdlib.h>
  32 #include <errno.h>
  33 #ifdef HAVE_POLL_H
  34 # include <poll.h>
  35 #endif
  36 #include <sys/time.h>
  37 #include <sys/types.h>
  38 #ifdef HAVE_SYS_SOCKET_H
  39 # include <sys/socket.h>
  40 #endif
  41 #ifdef HAVE_SYS_IOCTL_H
  42 #include <sys/ioctl.h>
  43 #endif
  44 #ifdef HAVE_SYS_FILIO_H
  45 # include <sys/filio.h>
  46 #endif
  47 #include <time.h>
  48 #include <unistd.h>
  49
  50 #include "ntstatus.h"
  51 #define WIN32_NO_STATUS
  52 #include "windef.h"
  53 #include "winternl.h"
  54 #include "winerror.h"
  55
  56 #include "process.h"
  57 #include "file.h"
  58 #include "handle.h"
  59 #include "thread.h"
  60 #include "request.h"
  61 #include "user.h"
  62
  63 /* From winsock.h */
  64 #define FD_MAX_EVENTS              10
  65 #define FD_READ_BIT                0
  66 #define FD_WRITE_BIT               1
  67 #define FD_OOB_BIT                 2
  68 #define FD_ACCEPT_BIT              3
  69 #define FD_CONNECT_BIT             4
  70 #define FD_CLOSE_BIT               5
  71
  72 /*
  73  * Define flags to be used with the WSAAsyncSelect() call.
  74  */
  75 #define FD_READ                    0x00000001
  76 #define FD_WRITE                   0x00000002
  77 #define FD_OOB                     0x00000004
  78 #define FD_ACCEPT                  0x00000008
  79 #define FD_CONNECT                 0x00000010
  80 #define FD_CLOSE                   0x00000020
  81
  82 /* internal per-socket flags */
  83 #define FD_WINE_LISTENING          0x10000000
  84 #define FD_WINE_NONBLOCKING        0x20000000
  85 #define FD_WINE_CONNECTED          0x40000000
  86 #define FD_WINE_RAW                0x80000000
  87 #define FD_WINE_INTERNAL           0xFFFF0000
  88
  89 /* Constants for WSAIoctl() */
  90 #define WSA_FLAG_OVERLAPPED        0x01
  91
  92 struct sock
  93 {
  94     struct object       obj;         /* object header */
  95     struct fd          *fd;          /* socket file descriptor */
  96     unsigned int        state;       /* status bits */
  97     unsigned int        mask;        /* event mask */
  98     unsigned int        hmask;       /* held (blocked) events */
  99     unsigned int        pmask;       /* pending events */
 100     unsigned int        flags;       /* socket flags */
 101     int                 polling;     /* is socket being polled? */
 102     unsigned short      proto;       /* socket protocol */
 103     unsigned short      type;        /* socket type */
 104     unsigned short      family;      /* socket family */
 105     struct event       *event;       /* event object */
 106     user_handle_t       window;      /* window to send the message to */
 107     unsigned int        message;     /* message to send */
 108     obj_handle_t        wparam;      /* message wparam (socket handle) */
 109     int                 errors[FD_MAX_EVENTS]; /* event errors */
 110     timeout_t           connect_time;/* time the socket was connected */
 111     struct sock        *deferred;    /* socket that waits for a deferred accept */
 112     struct async_queue *read_q;      /* queue for asynchronous reads */
 113     struct async_queue *write_q;     /* queue for asynchronous writes */
 114 };
 115
 116 static void sock_dump( struct object *obj, int verbose );
 117 static int sock_signaled( struct object *obj, struct wait_queue_entry *entry );
 118 static struct fd *sock_get_fd( struct object *obj );
 119 static void sock_destroy( struct object *obj );
 120
 121 static int sock_get_poll_events( struct fd *fd );
 122 static void sock_poll_event( struct fd *fd, int event );
 123 static enum server_fd_type sock_get_fd_type( struct fd *fd );
 124 static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
 125 static void sock_reselect_async( struct fd *fd, struct async_queue *queue );
 126 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb );
 127
 128 static int sock_get_ntstatus( int err );
 129 static int sock_get_error( int err );
 130 static void sock_set_error(void);
 131
 132 static const struct object_ops sock_ops =
 133 {
 134     sizeof(struct sock),          /* size */
 135     sock_dump,                    /* dump */
 136     no_get_type,                  /* get_type */
 137     add_queue,                    /* add_queue */
 138     remove_queue,                 /* remove_queue */
 139     sock_signaled,                /* signaled */
 140     no_satisfied,                 /* satisfied */
 141     no_signal,                    /* signal */
 142     sock_get_fd,                  /* get_fd */
 143     default_fd_map_access,        /* map_access */
 144     default_get_sd,               /* get_sd */
 145     default_set_sd,               /* set_sd */
 146     no_lookup_name,               /* lookup_name */
 147     no_open_file,                 /* open_file */
 148     fd_close_handle,              /* close_handle */
 149     sock_destroy                  /* destroy */
 150 };
 151
 152 static const struct fd_ops sock_fd_ops =
 153 {
 154     sock_get_poll_events,         /* get_poll_events */
 155     sock_poll_event,              /* poll_event */
 156     no_flush,                     /* flush */
 157     sock_get_fd_type,             /* get_fd_type */
 158     default_fd_ioctl,             /* ioctl */
 159     sock_queue_async,             /* queue_async */
 160     sock_reselect_async,          /* reselect_async */
 161     sock_cancel_async             /* cancel_async */
 162 };
 163
 164
 165 /* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
 166  * we post messages if there are multiple events.  Used to send
 167  * messages.  The problem is if there is both a FD_CONNECT event and,
 168  * say, an FD_READ event available on the same socket, we want to
 169  * notify the app of the connect event first.  Otherwise it may
 170  * discard the read event because it thinks it hasn't connected yet.
 171  */
 172 static const int event_bitorder[FD_MAX_EVENTS] =
 173 {
 174     FD_CONNECT_BIT,
 175     FD_ACCEPT_BIT,
 176     FD_OOB_BIT,
 177     FD_WRITE_BIT,
 178     FD_READ_BIT,
 179     FD_CLOSE_BIT,
 180     6, 7, 8, 9  /* leftovers */
 181 };
 182
 183 /* Flags that make sense only for SOCK_STREAM sockets */
 184 #define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
 185
 186 typedef enum {
 187     SOCK_SHUTDOWN_ERROR = -1,
 188     SOCK_SHUTDOWN_EOF = 0,
 189     SOCK_SHUTDOWN_POLLHUP = 1
 190 } sock_shutdown_t;
 191
 192 static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
 193
 194 static sock_shutdown_t sock_check_pollhup(void)
 195 {
 196     sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
 197     int fd[2], n;
 198     struct pollfd pfd;
 199     char dummy;
 200
 201     if ( socketpair( AF_UNIX, SOCK_STREAM, 0, fd ) ) return ret;
 202     if ( shutdown( fd[0], 1 ) ) goto out;
 203
 204     pfd.fd = fd[1];
 205     pfd.events = POLLIN;
 206     pfd.revents = 0;
 207
 208     /* Solaris' poll() sometimes returns nothing if given a 0ms timeout here */
 209     n = poll( &pfd, 1, 1 );
 210     if ( n != 1 ) goto out; /* error or timeout */
 211     if ( pfd.revents & POLLHUP )
 212         ret = SOCK_SHUTDOWN_POLLHUP;
 213     else if ( pfd.revents & POLLIN &&
 214               read( fd[1], &dummy, 1 ) == 0 )
 215         ret = SOCK_SHUTDOWN_EOF;
 216
 217 out:
 218     close( fd[0] );
 219     close( fd[1] );
 220     return ret;
 221 }
 222
 223 void sock_init(void)
 224 {
 225     sock_shutdown_type = sock_check_pollhup();
 226
 227     switch ( sock_shutdown_type )
 228     {
 229     case SOCK_SHUTDOWN_EOF:
 230         if (debug_level) fprintf( stderr, "sock_init: shutdown() causes EOF\n" );
 231         break;
 232     case SOCK_SHUTDOWN_POLLHUP:
 233         if (debug_level) fprintf( stderr, "sock_init: shutdown() causes POLLHUP\n" );
 234         break;
 235     default:
 236         fprintf( stderr, "sock_init: ERROR in sock_check_pollhup()\n" );
 237         sock_shutdown_type = SOCK_SHUTDOWN_EOF;
 238     }
 239 }
 240
 241 static int sock_reselect( struct sock *sock )
 242 {
 243     int ev = sock_get_poll_events( sock->fd );
 244
 245     if (debug_level)
 246         fprintf(stderr,"sock_reselect(%p): new mask %x\n", sock, ev);
 247
 248     if (!sock->polling)  /* FIXME: should find a better way to do this */
 249     {
 250         /* previously unconnected socket, is this reselect supposed to connect it? */
 251         if (!(sock->state & ~FD_WINE_NONBLOCKING)) return 0;
 252         /* ok, it is, attach it to the wineserver's main poll loop */
 253         sock->polling = 1;
 254         allow_fd_caching( sock->fd );
 255     }
 256     /* update condition mask */
 257     set_fd_events( sock->fd, ev );
 258     return ev;
 259 }
 260
 261 /* wake anybody waiting on the socket event or send the associated message */
 262 static void sock_wake_up( struct sock *sock )
 263 {
 264     unsigned int events = sock->pmask & sock->mask;
 265     int i;
 266
 267     if ( !events ) return;
 268
 269     if (sock->event)
 270     {
 271         if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
 272         set_event( sock->event );
 273     }
 274     if (sock->window)
 275     {
 276         if (debug_level) fprintf(stderr, "signalling events %x win %08x\n", events, sock->window );
 277         for (i = 0; i < FD_MAX_EVENTS; i++)
 278         {
 279             int event = event_bitorder[i];
 280             if (sock->pmask & (1 << event))
 281             {
 282                 lparam_t lparam = (1 << event) | (sock_get_error(sock->errors[event]) << 16);
 283                 post_message( sock->window, sock->message, sock->wparam, lparam );
 284             }
 285         }
 286         sock->pmask = 0;
 287         sock_reselect( sock );
 288     }
 289 }
 290
 291 static inline int sock_error( struct fd *fd )
 292 {
 293     unsigned int optval = 0;
 294     socklen_t optlen = sizeof(optval);
 295
 296     getsockopt( get_unix_fd(fd), SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
 297     return optval;
 298 }
 299
 300 static int sock_dispatch_asyncs( struct sock *sock, int event, int error )
 301 {
 302     if ( sock->flags & WSA_FLAG_OVERLAPPED )
 303     {
 304         if ( event & (POLLIN|POLLPRI) && async_waiting( sock->read_q ) )
 305         {
 306             if (debug_level) fprintf( stderr, "activating read queue for socket %p\n", sock );
 307             async_wake_up( sock->read_q, STATUS_ALERTED );
 308             event &= ~(POLLIN|POLLPRI);
 309         }
 310         if ( event & POLLOUT && async_waiting( sock->write_q ) )
 311         {
 312             if (debug_level) fprintf( stderr, "activating write queue for socket %p\n", sock );
 313             async_wake_up( sock->write_q, STATUS_ALERTED );
 314             event &= ~POLLOUT;
 315         }
 316         if ( event & (POLLERR|POLLHUP) )
 317         {
 318             int status = sock_get_ntstatus( error );
 319
 320             if ( !(sock->state & FD_READ) )
 321                 async_wake_up( sock->read_q, status );
 322             if ( !(sock->state & FD_WRITE) )
 323                 async_wake_up( sock->write_q, status );
 324         }
 325     }
 326     return event;
 327 }
 328
 329 static void sock_dispatch_events( struct sock *sock, int prevstate, int event, int error )
 330 {
 331     if (prevstate & FD_CONNECT)
 332     {
 333         sock->pmask |= FD_CONNECT;
 334         sock->hmask |= FD_CONNECT;
 335         sock->errors[FD_CONNECT_BIT] = error;
 336         goto end;
 337     }
 338     if (prevstate & FD_WINE_LISTENING)
 339     {
 340         sock->pmask |= FD_ACCEPT;
 341         sock->hmask |= FD_ACCEPT;
 342         sock->errors[FD_ACCEPT_BIT] = error;
 343         goto end;
 344     }
 345
 346     if (event & POLLIN)
 347     {
 348         sock->pmask |= FD_READ;
 349         sock->hmask |= FD_READ;
 350         sock->errors[FD_READ_BIT] = 0;
 351     }
 352
 353     if (event & POLLOUT)
 354     {
 355         sock->pmask |= FD_WRITE;
 356         sock->hmask |= FD_WRITE;
 357         sock->errors[FD_WRITE_BIT] = 0;
 358     }
 359
 360     if (event & POLLPRI)
 361     {
 362         sock->pmask |= FD_OOB;
 363         sock->hmask |= FD_OOB;
 364         sock->errors[FD_OOB_BIT] = 0;
 365     }
 366
 367     if (event & (POLLERR|POLLHUP))
 368     {
 369         sock->pmask |= FD_CLOSE;
 370         sock->hmask |= FD_CLOSE;
 371         sock->errors[FD_CLOSE_BIT] = error;
 372     }
 373 end:
 374     sock_wake_up( sock );
 375 }
 376
 377 static void sock_poll_event( struct fd *fd, int event )
 378 {
 379     struct sock *sock = get_fd_user( fd );
 380     int hangup_seen = 0;
 381     int prevstate = sock->state;
 382     int error = 0;
 383
 384     assert( sock->obj.ops == &sock_ops );
 385     if (debug_level)
 386         fprintf(stderr, "socket %p select event: %x\n", sock, event);
 387
 388     /* we may change event later, remove from loop here */
 389     if (event & (POLLERR|POLLHUP)) set_fd_events( sock->fd, -1 );
 390
 391     if (sock->state & FD_CONNECT)
 392     {
 393         if (event & (POLLERR|POLLHUP))
 394         {
 395             /* we didn't get connected? */
 396             sock->state &= ~FD_CONNECT;
 397             event &= ~POLLOUT;
 398             error = sock_error( fd );
 399         }
 400         else if (event & POLLOUT)
 401         {
 402             /* we got connected */
 403             sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 404             sock->state &= ~FD_CONNECT;
 405             sock->connect_time = current_time;
 406         }
 407     }
 408     else if (sock->state & FD_WINE_LISTENING)
 409     {
 410         /* listening */
 411         if (event & (POLLERR|POLLHUP))
 412             error = sock_error( fd );
 413     }
 414     else
 415     {
 416         /* normal data flow */
 417         if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
 418         {
 419             char dummy;
 420             int nr;
 421
 422             /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
 423              * has been closed, so we need to check for it explicitly here */
 424             nr  = recv( get_unix_fd( fd ), &dummy, 1, MSG_PEEK );
 425             if ( nr == 0 )
 426             {
 427                 hangup_seen = 1;
 428                 event &= ~POLLIN;
 429             }
 430             else if ( nr < 0 )
 431             {
 432                 event &= ~POLLIN;
 433                 /* EAGAIN can happen if an async recv() falls between the server's poll()
 434                    call and the invocation of this routine */
 435                 if ( errno != EAGAIN )
 436                 {
 437                     error = errno;
 438                     event |= POLLERR;
 439                     if ( debug_level )
 440                         fprintf( stderr, "recv error on socket %p: %d\n", sock, errno );
 441                 }
 442             }
 443         }
 444
 445         if ( (hangup_seen || event & (POLLHUP|POLLERR)) && (sock->state & (FD_READ|FD_WRITE)) )
 446         {
 447             error = error ? error : sock_error( fd );
 448             if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
 449                 sock->state &= ~FD_WRITE;
 450             sock->state &= ~FD_READ;
 451
 452             if (debug_level)
 453                 fprintf(stderr, "socket %p aborted by error %d, event: %x\n", sock, error, event);
 454         }
 455
 456         if (hangup_seen)
 457             event |= POLLHUP;
 458     }
 459
 460     event = sock_dispatch_asyncs( sock, event, error );
 461     sock_dispatch_events( sock, prevstate, event, error );
 462
 463     /* if anyone is stupid enough to wait on the socket object itself,
 464      * maybe we should wake them up too, just in case? */
 465     wake_up( &sock->obj, 0 );
 466
 467     sock_reselect( sock );
 468 }
 469
 470 static void sock_dump( struct object *obj, int verbose )
 471 {
 472     struct sock *sock = (struct sock *)obj;
 473     assert( obj->ops == &sock_ops );
 474     fprintf( stderr, "Socket fd=%p, state=%x, mask=%x, pending=%x, held=%x\n",
 475             sock->fd, sock->state,
 476             sock->mask, sock->pmask, sock->hmask );
 477 }
 478
 479 static int sock_signaled( struct object *obj, struct wait_queue_entry *entry )
 480 {
 481     struct sock *sock = (struct sock *)obj;
 482     assert( obj->ops == &sock_ops );
 483
 484     return check_fd_events( sock->fd, sock_get_poll_events( sock->fd ) ) != 0;
 485 }
 486
 487 static int sock_get_poll_events( struct fd *fd )
 488 {
 489     struct sock *sock = get_fd_user( fd );
 490     unsigned int mask = sock->mask & ~sock->hmask;
 491     unsigned int smask = sock->state & mask;
 492     int ev = 0;
 493
 494     assert( sock->obj.ops == &sock_ops );
 495
 496     if (sock->state & FD_CONNECT)
 497         /* connecting, wait for writable */
 498         return POLLOUT;
 499
 500     if ( async_queued( sock->read_q ) )
 501     {
 502         if ( async_waiting( sock->read_q ) ) ev |= POLLIN | POLLPRI;
 503     }
 504     else if (smask & FD_READ || (sock->state & FD_WINE_LISTENING && mask & FD_ACCEPT))
 505         ev |= POLLIN | POLLPRI;
 506     /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
 507     else if ( sock->type == SOCK_STREAM && sock->state & FD_READ && mask & FD_CLOSE &&
 508               !(sock->hmask & FD_READ) )
 509         ev |= POLLIN;
 510
 511     if ( async_queued( sock->write_q ) )
 512     {
 513         if ( async_waiting( sock->write_q ) ) ev |= POLLOUT;
 514     }
 515     else if (smask & FD_WRITE)
 516         ev |= POLLOUT;
 517
 518     return ev;
 519 }
 520
 521 static enum server_fd_type sock_get_fd_type( struct fd *fd )
 522 {
 523     return FD_TYPE_SOCKET;
 524 }
 525
 526 static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
 527 {
 528     struct sock *sock = get_fd_user( fd );
 529     struct async *async;
 530     struct async_queue *queue;
 531
 532     assert( sock->obj.ops == &sock_ops );
 533
 534     switch (type)
 535     {
 536     case ASYNC_TYPE_READ:
 537         if (!sock->read_q && !(sock->read_q = create_async_queue( sock->fd ))) return;
 538         queue = sock->read_q;
 539         break;
 540     case ASYNC_TYPE_WRITE:
 541         if (!sock->write_q && !(sock->write_q = create_async_queue( sock->fd ))) return;
 542         queue = sock->write_q;
 543         break;
 544     default:
 545         set_error( STATUS_INVALID_PARAMETER );
 546         return;
 547     }
 548
 549     if ( ( !( sock->state & (FD_READ|FD_CONNECT|FD_WINE_LISTENING) ) && type == ASYNC_TYPE_READ  ) ||
 550          ( !( sock->state & (FD_WRITE|FD_CONNECT) ) && type == ASYNC_TYPE_WRITE ) )
 551     {
 552         set_error( STATUS_PIPE_DISCONNECTED );
 553         return;
 554     }
 555
 556     if (!(async = create_async( current, queue, data ))) return;
 557     release_object( async );
 558
 559     sock_reselect( sock );
 560
 561     set_error( STATUS_PENDING );
 562 }
 563
 564 static void sock_reselect_async( struct fd *fd, struct async_queue *queue )
 565 {
 566     struct sock *sock = get_fd_user( fd );
 567     sock_reselect( sock );
 568 }
 569
 570 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb )
 571 {
 572     struct sock *sock = get_fd_user( fd );
 573     int n = 0;
 574     assert( sock->obj.ops == &sock_ops );
 575
 576     n += async_wake_up_by( sock->read_q, process, thread, iosb, STATUS_CANCELLED );
 577     n += async_wake_up_by( sock->write_q, process, thread, iosb, STATUS_CANCELLED );
 578     if (!n && iosb)
 579         set_error( STATUS_NOT_FOUND );
 580 }
 581
 582 static struct fd *sock_get_fd( struct object *obj )
 583 {
 584     struct sock *sock = (struct sock *)obj;
 585     return (struct fd *)grab_object( sock->fd );
 586 }
 587
 588 static void sock_destroy( struct object *obj )
 589 {
 590     struct sock *sock = (struct sock *)obj;
 591     assert( obj->ops == &sock_ops );
 592
 593     /* FIXME: special socket shutdown stuff? */
 594
 595     if ( sock->deferred )
 596         release_object( sock->deferred );
 597
 598     free_async_queue( sock->read_q );
 599     free_async_queue( sock->write_q );
 600     if (sock->event) release_object( sock->event );
 601     if (sock->fd)
 602     {
 603         /* shut the socket down to force pending poll() calls in the client to return */
 604         shutdown( get_unix_fd(sock->fd), SHUT_RDWR );
 605         release_object( sock->fd );
 606     }
 607 }
 608
 609 static void init_sock(struct sock *sock)
 610 {
 611     sock->state = 0;
 612     sock->mask    = 0;
 613     sock->hmask   = 0;
 614     sock->pmask   = 0;
 615     sock->polling = 0;
 616     sock->flags   = 0;
 617     sock->type    = 0;
 618     sock->family  = 0;
 619     sock->event   = NULL;
 620     sock->window  = 0;
 621     sock->message = 0;
 622     sock->wparam  = 0;
 623     sock->connect_time = 0;
 624     sock->deferred = NULL;
 625     sock->read_q  = NULL;
 626     sock->write_q = NULL;
 627     memset( sock->errors, 0, sizeof(sock->errors) );
 628 }
 629
 630 /* create a new and unconnected socket */
 631 static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
 632 {
 633     struct sock *sock;
 634     int sockfd;
 635
 636     sockfd = socket( family, type, protocol );
 637     if (debug_level)
 638         fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
 639     if (sockfd == -1)
 640     {
 641         sock_set_error();
 642         return NULL;
 643     }
 644     fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 645     if (!(sock = alloc_object( &sock_ops )))
 646     {
 647         close( sockfd );
 648         return NULL;
 649     }
 650     init_sock( sock );
 651     sock->state  = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
 652     sock->flags  = flags;
 653     sock->proto  = protocol;
 654     sock->type   = type;
 655     sock->family = family;
 656
 657     if (!(sock->fd = create_anonymous_fd( &sock_fd_ops, sockfd, &sock->obj,
 658                             (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT )))
 659     {
 660         release_object( sock );
 661         return NULL;
 662     }
 663     sock_reselect( sock );
 664     clear_error();
 665     return &sock->obj;
 666 }
 667
 668 /* accepts a socket and inits it */
 669 static int accept_new_fd( struct sock *sock )
 670 {
 671
 672     /* Try to accept(2). We can't be safe that this an already connected socket
 673      * or that accept() is allowed on it. In those cases we will get -1/errno
 674      * return.
 675      */
 676     int acceptfd;
 677     struct sockaddr saddr;
 678     socklen_t slen = sizeof(saddr);
 679     acceptfd = accept( get_unix_fd(sock->fd), &saddr, &slen);
 680     if (acceptfd == -1)
 681     {
 682         sock_set_error();
 683         return acceptfd;
 684     }
 685
 686     fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 687     return acceptfd;
 688 }
 689
 690 /* accept a socket (creates a new fd) */
 691 static struct sock *accept_socket( obj_handle_t handle )
 692 {
 693     struct sock *acceptsock;
 694     struct sock *sock;
 695     int acceptfd;
 696
 697     sock = (struct sock *)get_handle_obj( current->process, handle, FILE_READ_DATA, &sock_ops );
 698     if (!sock)
 699         return NULL;
 700
 701     if ( sock->deferred )
 702     {
 703         acceptsock = sock->deferred;
 704         sock->deferred = NULL;
 705     }
 706     else
 707     {
 708         if ((acceptfd = accept_new_fd( sock )) == -1)
 709         {
 710             release_object( sock );
 711             return NULL;
 712         }
 713         if (!(acceptsock = alloc_object( &sock_ops )))
 714         {
 715             close( acceptfd );
 716             release_object( sock );
 717             return NULL;
 718         }
 719
 720         init_sock( acceptsock );
 721         /* newly created socket gets the same properties of the listening socket */
 722         acceptsock->state  = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 723         if (sock->state & FD_WINE_NONBLOCKING)
 724             acceptsock->state |= FD_WINE_NONBLOCKING;
 725         acceptsock->mask    = sock->mask;
 726         acceptsock->proto   = sock->proto;
 727         acceptsock->type    = sock->type;
 728         acceptsock->family  = sock->family;
 729         acceptsock->window  = sock->window;
 730         acceptsock->message = sock->message;
 731         acceptsock->connect_time = current_time;
 732         if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
 733         acceptsock->flags = sock->flags;
 734         if (!(acceptsock->fd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
 735                                                     get_fd_options( sock->fd ) )))
 736         {
 737             release_object( acceptsock );
 738             release_object( sock );
 739             return NULL;
 740         }
 741     }
 742     clear_error();
 743     sock->pmask &= ~FD_ACCEPT;
 744     sock->hmask &= ~FD_ACCEPT;
 745     sock_reselect( sock );
 746     release_object( sock );
 747     return acceptsock;
 748 }
 749
 750 static int accept_into_socket( struct sock *sock, struct sock *acceptsock )
 751 {
 752     int acceptfd;
 753     struct fd *newfd;
 754     if ( sock->deferred )
 755     {
 756         newfd = dup_fd_object( sock->deferred->fd, 0, 0,
 757                                get_fd_options( acceptsock->fd ) );
 758         if ( !newfd )
 759             return FALSE;
 760
 761         set_fd_user( newfd, &sock_fd_ops, &acceptsock->obj );
 762
 763         release_object( sock->deferred );
 764         sock->deferred = NULL;
 765     }
 766     else
 767     {
 768         if ((acceptfd = accept_new_fd( sock )) == -1)
 769             return FALSE;
 770
 771         if (!(newfd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
 772                                             get_fd_options( acceptsock->fd ) )))
 773             return FALSE;
 774     }
 775
 776     acceptsock->state  |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 777     acceptsock->hmask   = 0;
 778     acceptsock->pmask   = 0;
 779     acceptsock->polling = 0;
 780     acceptsock->proto   = sock->proto;
 781     acceptsock->type    = sock->type;
 782     acceptsock->family  = sock->family;
 783     acceptsock->wparam  = 0;
 784     acceptsock->deferred = NULL;
 785     acceptsock->connect_time = current_time;
 786     release_object( acceptsock->fd );
 787     acceptsock->fd = newfd;
 788
 789     clear_error();
 790     sock->pmask &= ~FD_ACCEPT;
 791     sock->hmask &= ~FD_ACCEPT;
 792     sock_reselect( sock );
 793
 794     return TRUE;
 795 }
 796
 797 /* return an errno value mapped to a WSA error */
 798 static int sock_get_error( int err )
 799 {
 800     switch (err)
 801     {
 802         case EINTR:             return WSAEINTR;
 803         case EBADF:             return WSAEBADF;
 804         case EPERM:
 805         case EACCES:            return WSAEACCES;
 806         case EFAULT:            return WSAEFAULT;
 807         case EINVAL:            return WSAEINVAL;
 808         case EMFILE:            return WSAEMFILE;
 809         case EWOULDBLOCK:       return WSAEWOULDBLOCK;
 810         case EINPROGRESS:       return WSAEINPROGRESS;
 811         case EALREADY:          return WSAEALREADY;
 812         case ENOTSOCK:          return WSAENOTSOCK;
 813         case EDESTADDRREQ:      return WSAEDESTADDRREQ;
 814         case EMSGSIZE:          return WSAEMSGSIZE;
 815         case EPROTOTYPE:        return WSAEPROTOTYPE;
 816         case ENOPROTOOPT:       return WSAENOPROTOOPT;
 817         case EPROTONOSUPPORT:   return WSAEPROTONOSUPPORT;
 818         case ESOCKTNOSUPPORT:   return WSAESOCKTNOSUPPORT;
 819         case EOPNOTSUPP:        return WSAEOPNOTSUPP;
 820         case EPFNOSUPPORT:      return WSAEPFNOSUPPORT;
 821         case EAFNOSUPPORT:      return WSAEAFNOSUPPORT;
 822         case EADDRINUSE:        return WSAEADDRINUSE;
 823         case EADDRNOTAVAIL:     return WSAEADDRNOTAVAIL;
 824         case ENETDOWN:          return WSAENETDOWN;
 825         case ENETUNREACH:       return WSAENETUNREACH;
 826         case ENETRESET:         return WSAENETRESET;
 827         case ECONNABORTED:      return WSAECONNABORTED;
 828         case EPIPE:
 829         case ECONNRESET:        return WSAECONNRESET;
 830         case ENOBUFS:           return WSAENOBUFS;
 831         case EISCONN:           return WSAEISCONN;
 832         case ENOTCONN:          return WSAENOTCONN;
 833         case ESHUTDOWN:         return WSAESHUTDOWN;
 834         case ETOOMANYREFS:      return WSAETOOMANYREFS;
 835         case ETIMEDOUT:         return WSAETIMEDOUT;
 836         case ECONNREFUSED:      return WSAECONNREFUSED;
 837         case ELOOP:             return WSAELOOP;
 838         case ENAMETOOLONG:      return WSAENAMETOOLONG;
 839         case EHOSTDOWN:         return WSAEHOSTDOWN;
 840         case EHOSTUNREACH:      return WSAEHOSTUNREACH;
 841         case ENOTEMPTY:         return WSAENOTEMPTY;
 842 #ifdef EPROCLIM
 843         case EPROCLIM:          return WSAEPROCLIM;
 844 #endif
 845 #ifdef EUSERS
 846         case EUSERS:            return WSAEUSERS;
 847 #endif
 848 #ifdef EDQUOT
 849         case EDQUOT:            return WSAEDQUOT;
 850 #endif
 851 #ifdef ESTALE
 852         case ESTALE:            return WSAESTALE;
 853 #endif
 854 #ifdef EREMOTE
 855         case EREMOTE:           return WSAEREMOTE;
 856 #endif
 857
 858         case 0:                 return 0;
 859         default:
 860             errno = err;
 861             perror("wineserver: sock_get_error() can't map error");
 862             return WSAEFAULT;
 863     }
 864 }
 865
 866 static int sock_get_ntstatus( int err )
 867 {
 868     switch ( err )
 869     {
 870         case EBADF:             return STATUS_INVALID_HANDLE;
 871         case EBUSY:             return STATUS_DEVICE_BUSY;
 872         case EPERM:
 873         case EACCES:            return STATUS_ACCESS_DENIED;
 874         case EFAULT:            return STATUS_NO_MEMORY;
 875         case EINVAL:            return STATUS_INVALID_PARAMETER;
 876         case ENFILE:
 877         case EMFILE:            return STATUS_TOO_MANY_OPENED_FILES;
 878         case EWOULDBLOCK:       return STATUS_CANT_WAIT;
 879         case EINPROGRESS:       return STATUS_PENDING;
 880         case EALREADY:          return STATUS_NETWORK_BUSY;
 881         case ENOTSOCK:          return STATUS_OBJECT_TYPE_MISMATCH;
 882         case EDESTADDRREQ:      return STATUS_INVALID_PARAMETER;
 883         case EMSGSIZE:          return STATUS_BUFFER_OVERFLOW;
 884         case EPROTONOSUPPORT:
 885         case ESOCKTNOSUPPORT:
 886         case EPFNOSUPPORT:
 887         case EAFNOSUPPORT:
 888         case EPROTOTYPE:        return STATUS_NOT_SUPPORTED;
 889         case ENOPROTOOPT:       return STATUS_INVALID_PARAMETER;
 890         case EOPNOTSUPP:        return STATUS_NOT_SUPPORTED;
 891         case EADDRINUSE:        return STATUS_ADDRESS_ALREADY_ASSOCIATED;
 892         case EADDRNOTAVAIL:     return STATUS_INVALID_PARAMETER;
 893         case ECONNREFUSED:      return STATUS_CONNECTION_REFUSED;
 894         case ESHUTDOWN:         return STATUS_PIPE_DISCONNECTED;
 895         case ENOTCONN:          return STATUS_CONNECTION_DISCONNECTED;
 896         case ETIMEDOUT:         return STATUS_IO_TIMEOUT;
 897         case ENETUNREACH:       return STATUS_NETWORK_UNREACHABLE;
 898         case EHOSTUNREACH:      return STATUS_HOST_UNREACHABLE;
 899         case ENETDOWN:          return STATUS_NETWORK_BUSY;
 900         case EPIPE:
 901         case ECONNRESET:        return STATUS_CONNECTION_RESET;
 902         case ECONNABORTED:      return STATUS_CONNECTION_ABORTED;
 903
 904         case 0:                 return STATUS_SUCCESS;
 905         default:
 906             errno = err;
 907             perror("wineserver: sock_get_ntstatus() can't map error");
 908             return STATUS_UNSUCCESSFUL;
 909     }
 910 }
 911
 912 /* set the last error depending on errno */
 913 static void sock_set_error(void)
 914 {
 915     set_error( sock_get_ntstatus( errno ) );
 916 }
 917
 918 /* create a socket */
 919 DECL_HANDLER(create_socket)
 920 {
 921     struct object *obj;
 922
 923     reply->handle = 0;
 924     if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
 925     {
 926         reply->handle = alloc_handle( current->process, obj, req->access, req->attributes );
 927         release_object( obj );
 928     }
 929 }
 930
 931 /* accept a socket */
 932 DECL_HANDLER(accept_socket)
 933 {
 934     struct sock *sock;
 935
 936     reply->handle = 0;
 937     if ((sock = accept_socket( req->lhandle )) != NULL)
 938     {
 939         reply->handle = alloc_handle( current->process, &sock->obj, req->access, req->attributes );
 940         sock->wparam = reply->handle;  /* wparam for message is the socket handle */
 941         sock_reselect( sock );
 942         release_object( &sock->obj );
 943     }
 944 }
 945
 946 /* accept a socket into an initialized socket */
 947 DECL_HANDLER(accept_into_socket)
 948 {
 949     struct sock *sock, *acceptsock;
 950     const int all_attributes = FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES|FILE_READ_DATA;
 951
 952     if (!(sock = (struct sock *)get_handle_obj( current->process, req->lhandle,
 953                                                 all_attributes, &sock_ops)))
 954         return;
 955
 956     if (!(acceptsock = (struct sock *)get_handle_obj( current->process, req->ahandle,
 957                                                       all_attributes, &sock_ops)))
 958     {
 959         release_object( sock );
 960         return;
 961     }
 962
 963     if (accept_into_socket( sock, acceptsock ))
 964     {
 965         acceptsock->wparam = req->ahandle;  /* wparam for message is the socket handle */
 966         sock_reselect( acceptsock );
 967     }
 968     release_object( acceptsock );
 969     release_object( sock );
 970 }
 971
 972 /* set socket event parameters */
 973 DECL_HANDLER(set_socket_event)
 974 {
 975     struct sock *sock;
 976     struct event *old_event;
 977
 978     if (!(sock = (struct sock *)get_handle_obj( current->process, req->handle,
 979                                                 FILE_WRITE_ATTRIBUTES, &sock_ops))) return;
 980     old_event = sock->event;
 981     sock->mask    = req->mask;
 982     sock->hmask   &= ~req->mask; /* re-enable held events */
 983     sock->event   = NULL;
 984     sock->window  = req->window;
 985     sock->message = req->msg;
 986     sock->wparam  = req->handle;  /* wparam is the socket handle */
 987     if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
 988
 989     if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
 990
 991     sock_reselect( sock );
 992
 993     sock->state |= FD_WINE_NONBLOCKING;
 994
 995     /* if a network event is pending, signal the event object
 996        it is possible that FD_CONNECT or FD_ACCEPT network events has happened
 997        before a WSAEventSelect() was done on it.
 998        (when dealing with Asynchronous socket)  */
 999     sock_wake_up( sock );
1000
1001     if (old_event) release_object( old_event ); /* we're through with it */
1002     release_object( &sock->obj );
1003 }
1004
1005 /* get socket event parameters */
1006 DECL_HANDLER(get_socket_event)
1007 {
1008     struct sock *sock;
1009     int i;
1010     int errors[FD_MAX_EVENTS];
1011
1012     sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
1013     if (!sock)
1014     {
1015         reply->mask  = 0;
1016         reply->pmask = 0;
1017         reply->state = 0;
1018         return;
1019     }
1020     reply->mask  = sock->mask;
1021     reply->pmask = sock->pmask;
1022     reply->state = sock->state;
1023     for (i = 0; i < FD_MAX_EVENTS; i++)
1024         errors[i] = sock_get_ntstatus(sock->errors[i]);
1025
1026     set_reply_data( errors, min( get_reply_max_size(), sizeof(errors) ));
1027
1028     if (req->service)
1029     {
1030         if (req->c_event)
1031         {
1032             struct event *cevent = get_event_obj( current->process, req->c_event,
1033                                                   EVENT_MODIFY_STATE );
1034             if (cevent)
1035             {
1036                 reset_event( cevent );
1037                 release_object( cevent );
1038             }
1039         }
1040         sock->pmask = 0;
1041         sock_reselect( sock );
1042     }
1043     release_object( &sock->obj );
1044 }
1045
1046 /* re-enable pending socket events */
1047 DECL_HANDLER(enable_socket_event)
1048 {
1049     struct sock *sock;
1050
1051     if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
1052                                                FILE_WRITE_ATTRIBUTES, &sock_ops)))
1053         return;
1054
1055     /* for event-based notification, windows erases stale events */
1056     sock->pmask &= ~req->mask;
1057
1058     sock->hmask &= ~req->mask;
1059     sock->state |= req->sstate;
1060     sock->state &= ~req->cstate;
1061     if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
1062
1063     sock_reselect( sock );
1064
1065     release_object( &sock->obj );
1066 }
1067
1068 DECL_HANDLER(set_socket_deferred)
1069 {
1070     struct sock *sock, *acceptsock;
1071
1072     sock=(struct sock *)get_handle_obj( current->process, req->handle, FILE_WRITE_ATTRIBUTES, &sock_ops );
1073     if ( !sock )
1074         return;
1075
1076     acceptsock = (struct sock *)get_handle_obj( current->process, req->deferred, 0, &sock_ops );
1077     if ( !acceptsock )
1078     {
1079         release_object( sock );
1080         return;
1081     }
1082     sock->deferred = acceptsock;
1083     release_object( sock );
1084 }
1085
1086 DECL_HANDLER(get_socket_info)
1087 {
1088     struct sock *sock;
1089
1090     sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
1091     if (!sock) return;
1092
1093     reply->family   = sock->family;
1094     reply->type     = sock->type;
1095     reply->protocol = sock->proto;
1096
1097     release_object( &sock->obj );
1098 }