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_SYS_ERRNO_H
  34 # include <sys/errno.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      type;        /* socket type */
 103     unsigned short      family;      /* socket family */
 104     struct event       *event;       /* event object */
 105     user_handle_t       window;      /* window to send the message to */
 106     unsigned int        message;     /* message to send */
 107     obj_handle_t        wparam;      /* message wparam (socket handle) */
 108     int                 errors[FD_MAX_EVENTS]; /* event errors */
 109     struct sock        *deferred;    /* socket that waits for a deferred accept */
 110     struct async_queue *read_q;      /* queue for asynchronous reads */
 111     struct async_queue *write_q;     /* queue for asynchronous writes */
 112 };
 113
 114 static void sock_dump( struct object *obj, int verbose );
 115 static int sock_signaled( struct object *obj, struct thread *thread );
 116 static struct fd *sock_get_fd( struct object *obj );
 117 static void sock_destroy( struct object *obj );
 118
 119 static int sock_get_poll_events( struct fd *fd );
 120 static void sock_poll_event( struct fd *fd, int event );
 121 static enum server_fd_type sock_get_fd_type( struct fd *fd );
 122 static int sock_removable( struct fd *fd );
 123 static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
 124 static void sock_async_event( struct fd *fd, struct async_queue *queue, struct async *async, int status, int finished );
 125 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb );
 126
 127 static int sock_get_ntstatus( int err );
 128 static int sock_get_error( int err );
 129 static void sock_set_error(void);
 130
 131 static const struct object_ops sock_ops =
 132 {
 133     sizeof(struct sock),          /* size */
 134     sock_dump,                    /* dump */
 135     no_get_type,                  /* get_type */
 136     add_queue,                    /* add_queue */
 137     remove_queue,                 /* remove_queue */
 138     sock_signaled,                /* signaled */
 139     no_satisfied,                 /* satisfied */
 140     no_signal,                    /* signal */
 141     sock_get_fd,                  /* get_fd */
 142     default_fd_map_access,        /* map_access */
 143     default_get_sd,               /* get_sd */
 144     default_set_sd,               /* set_sd */
 145     no_lookup_name,               /* lookup_name */
 146     no_open_file,                 /* open_file */
 147     fd_close_handle,              /* close_handle */
 148     sock_destroy                  /* destroy */
 149 };
 150
 151 static const struct fd_ops sock_fd_ops =
 152 {
 153     sock_get_poll_events,         /* get_poll_events */
 154     sock_poll_event,              /* poll_event */
 155     no_flush,                     /* flush */
 156     sock_get_fd_type,             /* get_file_info */
 157     sock_removable,               /* removable */
 158     default_fd_ioctl,             /* ioctl */
 159     sock_queue_async,             /* queue_async */
 160     sock_async_event,             /* async_event */
 161     default_fd_async_terminated,  /* async_terminated */
 162     sock_cancel_async             /* cancel_async */
 163 };
 164
 165
 166 /* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
 167  * we post messages if there are multiple events.  Used to send
 168  * messages.  The problem is if there is both a FD_CONNECT event and,
 169  * say, an FD_READ event available on the same socket, we want to
 170  * notify the app of the connect event first.  Otherwise it may
 171  * discard the read event because it thinks it hasn't connected yet.
 172  */
 173 static const int event_bitorder[FD_MAX_EVENTS] =
 174 {
 175     FD_CONNECT_BIT,
 176     FD_ACCEPT_BIT,
 177     FD_OOB_BIT,
 178     FD_WRITE_BIT,
 179     FD_READ_BIT,
 180     FD_CLOSE_BIT,
 181     6, 7, 8, 9  /* leftovers */
 182 };
 183
 184 /* Flags that make sense only for SOCK_STREAM sockets */
 185 #define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
 186
 187 typedef enum {
 188     SOCK_SHUTDOWN_ERROR = -1,
 189     SOCK_SHUTDOWN_EOF = 0,
 190     SOCK_SHUTDOWN_POLLHUP = 1
 191 } sock_shutdown_t;
 192
 193 static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
 194
 195 static sock_shutdown_t sock_check_pollhup(void)
 196 {
 197     sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
 198     int fd[2], n;
 199     struct pollfd pfd;
 200     char dummy;
 201
 202     if ( socketpair( AF_UNIX, SOCK_STREAM, 0, fd ) ) goto out;
 203     if ( shutdown( fd[0], 1 ) ) goto out;
 204
 205     pfd.fd = fd[1];
 206     pfd.events = POLLIN;
 207     pfd.revents = 0;
 208
 209     n = poll( &pfd, 1, 0 );
 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     }
 255     /* update condition mask */
 256     set_fd_events( sock->fd, ev );
 257     return ev;
 258 }
 259
 260 /* wake anybody waiting on the socket event or send the associated message */
 261 static void sock_wake_up( struct sock *sock )
 262 {
 263     unsigned int events = sock->pmask & sock->mask;
 264     int i;
 265
 266     /* Do not signal events if there are still pending asynchronous IO requests */
 267     /* We need this to delay FD_CLOSE events until all pending overlapped requests are processed */
 268     if ( !events || async_queued( sock->read_q ) || async_queued( sock->write_q ) ) return;
 269
 270     if (sock->event)
 271     {
 272         if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
 273         set_event( sock->event );
 274     }
 275     if (sock->window)
 276     {
 277         if (debug_level) fprintf(stderr, "signalling events %x win %08x\n", events, sock->window );
 278         for (i = 0; i < FD_MAX_EVENTS; i++)
 279         {
 280             int event = event_bitorder[i];
 281             if (sock->pmask & (1 << event))
 282             {
 283                 lparam_t lparam = (1 << event) | (sock_get_error(sock->errors[event]) << 16);
 284                 post_message( sock->window, sock->message, sock->wparam, lparam );
 285             }
 286         }
 287         sock->pmask = 0;
 288         sock_reselect( sock );
 289     }
 290 }
 291
 292 static inline int sock_error( struct fd *fd )
 293 {
 294     unsigned int optval = 0, optlen;
 295
 296     optlen = sizeof(optval);
 297     getsockopt( get_unix_fd(fd), SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
 298     return optval;
 299 }
 300
 301 static void sock_dispatch_asyncs( struct sock *sock, int event )
 302 {
 303     if ( sock->flags & WSA_FLAG_OVERLAPPED )
 304     {
 305         if ( event & (POLLIN|POLLPRI|POLLERR|POLLHUP) && async_waiting( sock->read_q ))
 306         {
 307             if (debug_level) fprintf( stderr, "activating read queue for socket %p\n", sock );
 308             async_wake_up( sock->read_q, STATUS_ALERTED );
 309         }
 310         if ( event & (POLLOUT|POLLERR|POLLHUP) && 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         }
 315     }
 316 }
 317
 318 static void sock_poll_event( struct fd *fd, int event )
 319 {
 320     struct sock *sock = get_fd_user( fd );
 321     int hangup_seen = 0;
 322
 323     assert( sock->obj.ops == &sock_ops );
 324     if (debug_level)
 325         fprintf(stderr, "socket %p select event: %x\n", sock, event);
 326
 327     /* we may change event later, remove from loop here */
 328     if (event & (POLLERR|POLLHUP)) set_fd_events( sock->fd, -1 );
 329
 330     if (sock->state & FD_CONNECT)
 331     {
 332         /* connecting */
 333         if (event & POLLOUT)
 334         {
 335             /* we got connected */
 336             sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 337             sock->state &= ~FD_CONNECT;
 338             sock->pmask |= FD_CONNECT;
 339             sock->errors[FD_CONNECT_BIT] = 0;
 340             if (debug_level)
 341                 fprintf(stderr, "socket %p connection success\n", sock);
 342         }
 343         else if (event & (POLLERR|POLLHUP))
 344         {
 345             /* we didn't get connected? */
 346             sock->state &= ~FD_CONNECT;
 347             sock->pmask |= FD_CONNECT;
 348             sock->errors[FD_CONNECT_BIT] = sock_error( fd );
 349             if (debug_level)
 350                 fprintf(stderr, "socket %p connection failure\n", sock);
 351         }
 352     }
 353     else if (sock->state & FD_WINE_LISTENING)
 354     {
 355         /* listening */
 356         if (event & POLLIN)
 357         {
 358             /* incoming connection */
 359             sock->pmask |= FD_ACCEPT;
 360             sock->errors[FD_ACCEPT_BIT] = 0;
 361             sock->hmask |= FD_ACCEPT;
 362         }
 363         else if (event & (POLLERR|POLLHUP))
 364         {
 365             /* failed incoming connection? */
 366             sock->pmask |= FD_ACCEPT;
 367             sock->errors[FD_ACCEPT_BIT] = sock_error( fd );
 368             sock->hmask |= FD_ACCEPT;
 369         }
 370     }
 371     else
 372     {
 373         /* normal data flow */
 374         if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
 375         {
 376             char dummy;
 377             int nr;
 378
 379             /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
 380              * has been closed, so we need to check for it explicitly here */
 381             nr  = recv( get_unix_fd( fd ), &dummy, 1, MSG_PEEK );
 382             if ( nr == 0 )
 383             {
 384                 hangup_seen = 1;
 385                 event &= ~POLLIN;
 386             }
 387             else if ( nr < 0 )
 388             {
 389                 event &= ~POLLIN;
 390                 /* EAGAIN can happen if an async recv() falls between the server's poll()
 391                    call and the invocation of this routine */
 392                 if ( errno != EAGAIN )
 393                 {
 394                     if ( debug_level )
 395                         fprintf( stderr, "recv error on socket %p: %d\n", sock, errno );
 396                     event |= POLLERR;
 397                 }
 398             }
 399         }
 400
 401         if ( event & POLLIN )
 402         {
 403             sock->pmask |= FD_READ;
 404             sock->hmask |= FD_READ;
 405             sock->errors[FD_READ_BIT] = 0;
 406             if (debug_level)
 407                 fprintf(stderr, "socket %p is readable\n", sock );
 408         }
 409
 410         if (event & POLLOUT)
 411         {
 412             sock->pmask |= FD_WRITE;
 413             sock->hmask |= FD_WRITE;
 414             sock->errors[FD_WRITE_BIT] = 0;
 415             if (debug_level)
 416                 fprintf(stderr, "socket %p is writable\n", sock);
 417         }
 418         if (event & POLLPRI)
 419         {
 420             sock->pmask |= FD_OOB;
 421             sock->hmask |= FD_OOB;
 422             sock->errors[FD_OOB_BIT] = 0;
 423             if (debug_level)
 424                 fprintf(stderr, "socket %p got OOB data\n", sock);
 425         }
 426
 427         if ( (hangup_seen || event & (POLLHUP|POLLERR)) && (sock->state & (FD_READ|FD_WRITE)) )
 428         {
 429             sock->errors[FD_CLOSE_BIT] = sock_error( fd );
 430             if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
 431                 sock->state &= ~FD_WRITE;
 432             sock->state &= ~FD_READ;
 433
 434             sock->pmask |= FD_CLOSE;
 435             sock->hmask |= FD_CLOSE;
 436             if (debug_level)
 437                 fprintf(stderr, "socket %p aborted by error %d, event: %x - removing from select loop\n",
 438                         sock, sock->errors[FD_CLOSE_BIT], event);
 439         }
 440
 441         if (hangup_seen)
 442             event |= POLLHUP;
 443     }
 444
 445     sock_dispatch_asyncs( sock, event );
 446
 447     /* wake up anyone waiting for whatever just happened */
 448     sock_wake_up( sock );
 449
 450     /* if anyone is stupid enough to wait on the socket object itself,
 451      * maybe we should wake them up too, just in case? */
 452     wake_up( &sock->obj, 0 );
 453
 454     sock_reselect( sock );
 455 }
 456
 457 static void sock_dump( struct object *obj, int verbose )
 458 {
 459     struct sock *sock = (struct sock *)obj;
 460     assert( obj->ops == &sock_ops );
 461     printf( "Socket fd=%p, state=%x, mask=%x, pending=%x, held=%x\n",
 462             sock->fd, sock->state,
 463             sock->mask, sock->pmask, sock->hmask );
 464 }
 465
 466 static int sock_signaled( struct object *obj, struct thread *thread )
 467 {
 468     struct sock *sock = (struct sock *)obj;
 469     assert( obj->ops == &sock_ops );
 470
 471     return check_fd_events( sock->fd, sock_get_poll_events( sock->fd ) ) != 0;
 472 }
 473
 474 static int sock_get_poll_events( struct fd *fd )
 475 {
 476     struct sock *sock = get_fd_user( fd );
 477     unsigned int mask = sock->mask & ~sock->hmask;
 478     unsigned int smask = sock->state & mask;
 479     int ev = 0;
 480
 481     assert( sock->obj.ops == &sock_ops );
 482
 483     if (sock->state & FD_CONNECT)
 484         /* connecting, wait for writable */
 485         return POLLOUT;
 486     if (sock->state & FD_WINE_LISTENING)
 487         /* listening, wait for readable */
 488         return (mask & FD_ACCEPT) ? POLLIN : 0;
 489
 490     if ( async_queued( sock->read_q ) )
 491     {
 492         if ( async_waiting( sock->read_q ) ) ev |= POLLIN | POLLPRI;
 493     }
 494     else if (smask & FD_READ)
 495         ev |= POLLIN | POLLPRI;
 496     /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
 497     else if ( sock->type == SOCK_STREAM && sock->state & FD_READ && mask & FD_CLOSE &&
 498               !(sock->hmask & FD_READ) )
 499         ev |= POLLIN;
 500
 501     if ( async_queued( sock->write_q ) )
 502     {
 503         if ( async_waiting( sock->write_q ) ) ev |= POLLOUT;
 504     }
 505     else if (smask & FD_WRITE)
 506         ev |= POLLOUT;
 507
 508     return ev;
 509 }
 510
 511 static enum server_fd_type sock_get_fd_type( struct fd *fd )
 512 {
 513     return FD_TYPE_SOCKET;
 514 }
 515
 516 static int sock_removable( struct fd *fd )
 517 {
 518     struct sock *sock = get_fd_user( fd );
 519     assert( sock->obj.ops == &sock_ops );
 520
 521     /* Don't cache SOCK_STREAM sockets until connected (needed for AcceptEx) */
 522     return sock->type == SOCK_STREAM && !sock->polling;
 523 }
 524
 525 static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
 526 {
 527     struct sock *sock = get_fd_user( fd );
 528     struct async_queue *queue;
 529
 530     assert( sock->obj.ops == &sock_ops );
 531
 532     switch (type)
 533     {
 534     case ASYNC_TYPE_READ:
 535         if (!sock->read_q && !(sock->read_q = create_async_queue( sock->fd ))) return;
 536         queue = sock->read_q;
 537         break;
 538     case ASYNC_TYPE_WRITE:
 539         if (!sock->write_q && !(sock->write_q = create_async_queue( sock->fd ))) return;
 540         queue = sock->write_q;
 541         break;
 542     default:
 543         set_error( STATUS_INVALID_PARAMETER );
 544         return;
 545     }
 546
 547     if ( ( !( sock->state & FD_READ ) && type == ASYNC_TYPE_READ  ) ||
 548          ( !( sock->state & FD_WRITE ) && type == ASYNC_TYPE_WRITE ) )
 549     {
 550         set_error( STATUS_PIPE_DISCONNECTED );
 551     }
 552     else
 553     {
 554         struct async *async;
 555         if (!(async = create_async( current, queue, data ))) return;
 556         release_object( async );
 557         set_error( STATUS_PENDING );
 558     }
 559
 560     sock_reselect( sock );
 561 }
 562
 563 static void sock_async_event( struct fd *fd, struct async_queue *queue, struct async *async, int status, int finished )
 564 {
 565     struct sock *sock = get_fd_user( fd );
 566     sock_reselect( sock );
 567 }
 568
 569 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb )
 570 {
 571     struct sock *sock = get_fd_user( fd );
 572     int n = 0;
 573     assert( sock->obj.ops == &sock_ops );
 574
 575     n += async_wake_up_by( sock->read_q, process, thread, iosb, STATUS_CANCELLED );
 576     n += async_wake_up_by( sock->write_q, process, thread, iosb, STATUS_CANCELLED );
 577     if (!n && iosb)
 578         set_error( STATUS_NOT_FOUND );
 579 }
 580
 581 static struct fd *sock_get_fd( struct object *obj )
 582 {
 583     struct sock *sock = (struct sock *)obj;
 584     return (struct fd *)grab_object( sock->fd );
 585 }
 586
 587 static void sock_destroy( struct object *obj )
 588 {
 589     struct sock *sock = (struct sock *)obj;
 590     assert( obj->ops == &sock_ops );
 591
 592     /* FIXME: special socket shutdown stuff? */
 593
 594     if ( sock->deferred )
 595         release_object( sock->deferred );
 596
 597     free_async_queue( sock->read_q );
 598     free_async_queue( sock->write_q );
 599     if (sock->event) release_object( sock->event );
 600     if (sock->fd)
 601     {
 602         /* shut the socket down to force pending poll() calls in the client to return */
 603         shutdown( get_unix_fd(sock->fd), SHUT_RDWR );
 604         release_object( sock->fd );
 605     }
 606 }
 607
 608 static struct sock *alloc_sock(void)
 609 {
 610     struct sock *sock;
 611     if (!(sock = alloc_object( &sock_ops )))
 612         return NULL;
 613
 614     sock->state = 0;
 615     sock->mask    = 0;
 616     sock->hmask   = 0;
 617     sock->pmask   = 0;
 618     sock->polling = 0;
 619     sock->flags   = 0;
 620     sock->type    = 0;
 621     sock->family  = 0;
 622     sock->event   = NULL;
 623     sock->window  = 0;
 624     sock->message = 0;
 625     sock->wparam  = 0;
 626     sock->deferred = NULL;
 627     sock->read_q  = NULL;
 628     sock->write_q = NULL;
 629
 630     return sock;
 631 }
 632
 633 /* create a new and unconnected socket */
 634 static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
 635 {
 636     struct sock *sock;
 637     int sockfd;
 638
 639     sockfd = socket( family, type, protocol );
 640     if (debug_level)
 641         fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
 642     if (sockfd == -1)
 643     {
 644         sock_set_error();
 645         return NULL;
 646     }
 647     fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 648     if (!(sock = alloc_sock()))
 649     {
 650         close( sockfd );
 651         return NULL;
 652     }
 653     sock->state = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
 654     sock->flags   = flags;
 655     sock->type    = type;
 656     sock->family  = family;
 657     memset( sock->errors, 0, sizeof(sock->errors) );
 658     if (!(sock->fd = create_anonymous_fd( &sock_fd_ops, sockfd, &sock->obj,
 659                             (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT )))
 660     {
 661         release_object( sock );
 662         return NULL;
 663     }
 664     sock_reselect( sock );
 665     clear_error();
 666     return &sock->obj;
 667 }
 668
 669 /* accept a socket (creates a new fd) */
 670 static struct sock *accept_socket( struct sock *sock )
 671 {
 672     struct sock *acceptsock;
 673     int acceptfd;
 674     struct sockaddr     saddr;
 675
 676     if ( sock->deferred )
 677     {
 678         acceptsock = sock->deferred;
 679         sock->deferred = NULL;
 680     }
 681     else
 682     {
 683
 684         /* Try to accept(2). We can't be safe that this an already connected socket
 685          * or that accept() is allowed on it. In those cases we will get -1/errno
 686          * return.
 687          */
 688         unsigned int slen = sizeof(saddr);
 689         acceptfd = accept( get_unix_fd(sock->fd), &saddr, &slen);
 690         if (acceptfd==-1)
 691         {
 692             sock_set_error();
 693             return NULL;
 694         }
 695         if (!(acceptsock = alloc_sock()))
 696         {
 697             close( acceptfd );
 698             return NULL;
 699         }
 700
 701         /* newly created socket gets the same properties of the listening socket */
 702         fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 703         acceptsock->state  = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 704         if (sock->state & FD_WINE_NONBLOCKING)
 705             acceptsock->state |= FD_WINE_NONBLOCKING;
 706         acceptsock->mask    = sock->mask;
 707         acceptsock->type    = sock->type;
 708         acceptsock->family  = sock->family;
 709         acceptsock->window  = sock->window;
 710         acceptsock->message = sock->message;
 711         if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
 712         acceptsock->flags   = sock->flags;
 713         memset( acceptsock->errors, 0, sizeof(acceptsock->errors) );
 714         if (!(acceptsock->fd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
 715                                                     get_fd_options( sock->fd ) )))
 716         {
 717             release_object( acceptsock );
 718             return NULL;
 719         }
 720     }
 721     clear_error();
 722     sock->pmask &= ~FD_ACCEPT;
 723     sock->hmask &= ~FD_ACCEPT;
 724     sock_reselect( sock );
 725     return acceptsock;
 726 }
 727
 728 /* set the last error depending on errno */
 729 static int sock_get_error( int err )
 730 {
 731     switch (err)
 732     {
 733         case EINTR:             return WSAEINTR;
 734         case EBADF:             return WSAEBADF;
 735         case EPERM:
 736         case EACCES:            return WSAEACCES;
 737         case EFAULT:            return WSAEFAULT;
 738         case EINVAL:            return WSAEINVAL;
 739         case EMFILE:            return WSAEMFILE;
 740         case EWOULDBLOCK:       return WSAEWOULDBLOCK;
 741         case EINPROGRESS:       return WSAEINPROGRESS;
 742         case EALREADY:          return WSAEALREADY;
 743         case ENOTSOCK:          return WSAENOTSOCK;
 744         case EDESTADDRREQ:      return WSAEDESTADDRREQ;
 745         case EMSGSIZE:          return WSAEMSGSIZE;
 746         case EPROTOTYPE:        return WSAEPROTOTYPE;
 747         case ENOPROTOOPT:       return WSAENOPROTOOPT;
 748         case EPROTONOSUPPORT:   return WSAEPROTONOSUPPORT;
 749         case ESOCKTNOSUPPORT:   return WSAESOCKTNOSUPPORT;
 750         case EOPNOTSUPP:        return WSAEOPNOTSUPP;
 751         case EPFNOSUPPORT:      return WSAEPFNOSUPPORT;
 752         case EAFNOSUPPORT:      return WSAEAFNOSUPPORT;
 753         case EADDRINUSE:        return WSAEADDRINUSE;
 754         case EADDRNOTAVAIL:     return WSAEADDRNOTAVAIL;
 755         case ENETDOWN:          return WSAENETDOWN;
 756         case ENETUNREACH:       return WSAENETUNREACH;
 757         case ENETRESET:         return WSAENETRESET;
 758         case ECONNABORTED:      return WSAECONNABORTED;
 759         case EPIPE:
 760         case ECONNRESET:        return WSAECONNRESET;
 761         case ENOBUFS:           return WSAENOBUFS;
 762         case EISCONN:           return WSAEISCONN;
 763         case ENOTCONN:          return WSAENOTCONN;
 764         case ESHUTDOWN:         return WSAESHUTDOWN;
 765         case ETOOMANYREFS:      return WSAETOOMANYREFS;
 766         case ETIMEDOUT:         return WSAETIMEDOUT;
 767         case ECONNREFUSED:      return WSAECONNREFUSED;
 768         case ELOOP:             return WSAELOOP;
 769         case ENAMETOOLONG:      return WSAENAMETOOLONG;
 770         case EHOSTDOWN:         return WSAEHOSTDOWN;
 771         case EHOSTUNREACH:      return WSAEHOSTUNREACH;
 772         case ENOTEMPTY:         return WSAENOTEMPTY;
 773 #ifdef EPROCLIM
 774         case EPROCLIM:          return WSAEPROCLIM;
 775 #endif
 776 #ifdef EUSERS
 777         case EUSERS:            return WSAEUSERS;
 778 #endif
 779 #ifdef EDQUOT
 780         case EDQUOT:            return WSAEDQUOT;
 781 #endif
 782 #ifdef ESTALE
 783         case ESTALE:            return WSAESTALE;
 784 #endif
 785 #ifdef EREMOTE
 786         case EREMOTE:           return WSAEREMOTE;
 787 #endif
 788
 789         case 0:                 return 0;
 790         default:
 791             errno = err;
 792             perror("wineserver: sock_get_error() can't map error");
 793             return WSAEFAULT;
 794     }
 795 }
 796
 797 static int sock_get_ntstatus( int err )
 798 {
 799     switch ( err )
 800     {
 801         case EBADF:             return STATUS_INVALID_HANDLE;
 802         case EBUSY:             return STATUS_DEVICE_BUSY;
 803         case EPERM:
 804         case EACCES:            return STATUS_ACCESS_DENIED;
 805         case EFAULT:            return STATUS_NO_MEMORY;
 806         case EINVAL:            return STATUS_INVALID_PARAMETER;
 807         case ENFILE:
 808         case EMFILE:            return STATUS_TOO_MANY_OPENED_FILES;
 809         case EWOULDBLOCK:       return STATUS_CANT_WAIT;
 810         case EINPROGRESS:       return STATUS_PENDING;
 811         case EALREADY:          return STATUS_NETWORK_BUSY;
 812         case ENOTSOCK:          return STATUS_OBJECT_TYPE_MISMATCH;
 813         case EDESTADDRREQ:      return STATUS_INVALID_PARAMETER;
 814         case EMSGSIZE:          return STATUS_BUFFER_OVERFLOW;
 815         case EPROTONOSUPPORT:
 816         case ESOCKTNOSUPPORT:
 817         case EPFNOSUPPORT:
 818         case EAFNOSUPPORT:
 819         case EPROTOTYPE:        return STATUS_NOT_SUPPORTED;
 820         case ENOPROTOOPT:       return STATUS_INVALID_PARAMETER;
 821         case EOPNOTSUPP:        return STATUS_NOT_SUPPORTED;
 822         case EADDRINUSE:        return STATUS_ADDRESS_ALREADY_ASSOCIATED;
 823         case EADDRNOTAVAIL:     return STATUS_INVALID_PARAMETER;
 824         case ECONNREFUSED:      return STATUS_CONNECTION_REFUSED;
 825         case ESHUTDOWN:         return STATUS_PIPE_DISCONNECTED;
 826         case ENOTCONN:          return STATUS_CONNECTION_DISCONNECTED;
 827         case ETIMEDOUT:         return STATUS_IO_TIMEOUT;
 828         case ENETUNREACH:       return STATUS_NETWORK_UNREACHABLE;
 829         case EHOSTUNREACH:      return STATUS_HOST_UNREACHABLE;
 830         case ENETDOWN:          return STATUS_NETWORK_BUSY;
 831         case EPIPE:
 832         case ECONNRESET:        return STATUS_CONNECTION_RESET;
 833         case ECONNABORTED:      return STATUS_CONNECTION_ABORTED;
 834
 835         case 0:                 return STATUS_SUCCESS;
 836         default:
 837             errno = err;
 838             perror("wineserver: sock_get_ntstatus() can't map error");
 839             return STATUS_UNSUCCESSFUL;
 840     }
 841 }
 842
 843 /* set the last error depending on errno */
 844 static void sock_set_error(void)
 845 {
 846     set_error( sock_get_ntstatus( errno ) );
 847 }
 848
 849 /* create a socket */
 850 DECL_HANDLER(create_socket)
 851 {
 852     struct object *obj;
 853
 854     reply->handle = 0;
 855     if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
 856     {
 857         reply->handle = alloc_handle( current->process, obj, req->access, req->attributes );
 858         release_object( obj );
 859     }
 860 }
 861
 862 /* accept a socket */
 863 DECL_HANDLER(accept_socket)
 864 {
 865     struct sock *sock, *acceptsock;
 866
 867     reply->handle = 0;
 868     if (!(sock = (struct sock *)get_handle_obj( current->process, req->lhandle,
 869                                                 FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES|FILE_READ_DATA, &sock_ops)))
 870         return;
 871
 872     if ((acceptsock = accept_socket( sock )) != NULL)
 873     {
 874         reply->handle = alloc_handle( current->process, &acceptsock->obj, req->access, req->attributes );
 875         acceptsock->wparam = reply->handle;  /* wparam for message is the socket handle */
 876         sock_reselect( acceptsock );
 877
 878         release_object( acceptsock );
 879     }
 880     release_object( sock );
 881 }
 882
 883 /* set socket event parameters */
 884 DECL_HANDLER(set_socket_event)
 885 {
 886     struct sock *sock;
 887     struct event *old_event;
 888     int pollev;
 889
 890     if (!(sock = (struct sock *)get_handle_obj( current->process, req->handle,
 891                                                 FILE_WRITE_ATTRIBUTES, &sock_ops))) return;
 892     old_event = sock->event;
 893     sock->mask    = req->mask;
 894     sock->hmask   &= ~req->mask; /* re-enable held events */
 895     sock->event   = NULL;
 896     sock->window  = req->window;
 897     sock->message = req->msg;
 898     sock->wparam  = req->handle;  /* wparam is the socket handle */
 899     if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
 900
 901     if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
 902
 903     pollev = sock_reselect( sock );
 904
 905     if (sock->mask)
 906         sock->state |= FD_WINE_NONBLOCKING;
 907
 908     /* if a network event is pending, signal the event object
 909        it is possible that FD_CONNECT or FD_ACCEPT network events has happened
 910        before a WSAEventSelect() was done on it.
 911        (when dealing with Asynchronous socket)  */
 912     sock_wake_up( sock );
 913
 914     if (old_event) release_object( old_event ); /* we're through with it */
 915     release_object( &sock->obj );
 916 }
 917
 918 /* get socket event parameters */
 919 DECL_HANDLER(get_socket_event)
 920 {
 921     struct sock *sock;
 922     int i;
 923     int errors[FD_MAX_EVENTS];
 924
 925     sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
 926     if (!sock)
 927     {
 928         reply->mask  = 0;
 929         reply->pmask = 0;
 930         reply->state = 0;
 931         return;
 932     }
 933     reply->mask  = sock->mask;
 934     reply->pmask = sock->pmask;
 935     reply->state = sock->state;
 936     for (i = 0; i < FD_MAX_EVENTS; i++)
 937         errors[i] = sock_get_ntstatus(sock->errors[i]);
 938
 939     set_reply_data( errors, min( get_reply_max_size(), sizeof(errors) ));
 940
 941     if (req->service)
 942     {
 943         if (req->c_event)
 944         {
 945             struct event *cevent = get_event_obj( current->process, req->c_event,
 946                                                   EVENT_MODIFY_STATE );
 947             if (cevent)
 948             {
 949                 reset_event( cevent );
 950                 release_object( cevent );
 951             }
 952         }
 953         sock->pmask = 0;
 954         sock_reselect( sock );
 955     }
 956     release_object( &sock->obj );
 957 }
 958
 959 /* re-enable pending socket events */
 960 DECL_HANDLER(enable_socket_event)
 961 {
 962     struct sock *sock;
 963
 964     if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
 965                                                FILE_WRITE_ATTRIBUTES, &sock_ops)))
 966         return;
 967
 968     sock->pmask &= ~req->mask; /* is this safe? */
 969     sock->hmask &= ~req->mask;
 970     sock->state |= req->sstate;
 971     sock->state &= ~req->cstate;
 972     if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
 973
 974     sock_reselect( sock );
 975
 976     release_object( &sock->obj );
 977 }
 978
 979 DECL_HANDLER(set_socket_deferred)
 980 {
 981     struct sock *sock, *acceptsock;
 982
 983     sock=(struct sock *)get_handle_obj( current->process, req->handle, FILE_WRITE_ATTRIBUTES, &sock_ops );
 984     if ( !sock )
 985         return;
 986
 987     acceptsock = (struct sock *)get_handle_obj( current->process, req->deferred, 0, &sock_ops );
 988     if ( !acceptsock )
 989     {
 990         release_object( sock );
 991         return;
 992     }
 993     sock->deferred = acceptsock;
 994     release_object( sock );
 995 }