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 #include <limits.h>
  50 #ifdef HAVE_LINUX_RTNETLINK_H
  51 # include <linux/rtnetlink.h>
  52 #endif
  53
  54 #include "ntstatus.h"
  55 #define WIN32_NO_STATUS
  56 #include "windef.h"
  57 #include "winternl.h"
  58 #include "winerror.h"
  59 #define USE_WS_PREFIX
  60 #include "winsock2.h"
  61
  62 #include "process.h"
  63 #include "file.h"
  64 #include "handle.h"
  65 #include "thread.h"
  66 #include "request.h"
  67 #include "user.h"
  68
  69 /* From winsock.h */
  70 #define FD_MAX_EVENTS              10
  71 #define FD_READ_BIT                0
  72 #define FD_WRITE_BIT               1
  73 #define FD_OOB_BIT                 2
  74 #define FD_ACCEPT_BIT              3
  75 #define FD_CONNECT_BIT             4
  76 #define FD_CLOSE_BIT               5
  77
  78 /*
  79  * Define flags to be used with the WSAAsyncSelect() call.
  80  */
  81 #define FD_READ                    0x00000001
  82 #define FD_WRITE                   0x00000002
  83 #define FD_OOB                     0x00000004
  84 #define FD_ACCEPT                  0x00000008
  85 #define FD_CONNECT                 0x00000010
  86 #define FD_CLOSE                   0x00000020
  87
  88 /* internal per-socket flags */
  89 #define FD_WINE_LISTENING          0x10000000
  90 #define FD_WINE_NONBLOCKING        0x20000000
  91 #define FD_WINE_CONNECTED          0x40000000
  92 #define FD_WINE_RAW                0x80000000
  93 #define FD_WINE_INTERNAL           0xFFFF0000
  94
  95 struct sock
  96 {
  97     struct object       obj;         /* object header */
  98     struct fd          *fd;          /* socket file descriptor */
  99     unsigned int        state;       /* status bits */
 100     unsigned int        mask;        /* event mask */
 101     unsigned int        hmask;       /* held (blocked) events */
 102     unsigned int        pmask;       /* pending events */
 103     unsigned int        flags;       /* socket flags */
 104     int                 polling;     /* is socket being polled? */
 105     unsigned short      proto;       /* socket protocol */
 106     unsigned short      type;        /* socket type */
 107     unsigned short      family;      /* socket family */
 108     struct event       *event;       /* event object */
 109     user_handle_t       window;      /* window to send the message to */
 110     unsigned int        message;     /* message to send */
 111     obj_handle_t        wparam;      /* message wparam (socket handle) */
 112     int                 errors[FD_MAX_EVENTS]; /* event errors */
 113     timeout_t           connect_time;/* time the socket was connected */
 114     struct sock        *deferred;    /* socket that waits for a deferred accept */
 115     struct async_queue  read_q;      /* queue for asynchronous reads */
 116     struct async_queue  write_q;     /* queue for asynchronous writes */
 117     struct async_queue  ifchange_q;  /* queue for interface change notifications */
 118     struct object      *ifchange_obj; /* the interface change notification object */
 119     struct list         ifchange_entry; /* entry in ifchange notification list */
 120 };
 121
 122 static void sock_dump( struct object *obj, int verbose );
 123 static int sock_signaled( struct object *obj, struct wait_queue_entry *entry );
 124 static struct fd *sock_get_fd( struct object *obj );
 125 static void sock_destroy( struct object *obj );
 126 static struct object *sock_get_ifchange( struct sock *sock );
 127 static void sock_release_ifchange( struct sock *sock );
 128
 129 static int sock_get_poll_events( struct fd *fd );
 130 static void sock_poll_event( struct fd *fd, int event );
 131 static enum server_fd_type sock_get_fd_type( struct fd *fd );
 132 static int sock_ioctl( struct fd *fd, ioctl_code_t code, struct async *async );
 133 static void sock_queue_async( struct fd *fd, struct async *async, int type, int count );
 134 static void sock_reselect_async( struct fd *fd, struct async_queue *queue );
 135
 136 static int sock_get_ntstatus( int err );
 137 static int sock_get_error( int err );
 138 static void sock_set_error(void);
 139
 140 static const struct object_ops sock_ops =
 141 {
 142     sizeof(struct sock),          /* size */
 143     sock_dump,                    /* dump */
 144     no_get_type,                  /* get_type */
 145     add_queue,                    /* add_queue */
 146     remove_queue,                 /* remove_queue */
 147     sock_signaled,                /* signaled */
 148     no_satisfied,                 /* satisfied */
 149     no_signal,                    /* signal */
 150     sock_get_fd,                  /* get_fd */
 151     default_fd_map_access,        /* map_access */
 152     default_get_sd,               /* get_sd */
 153     default_set_sd,               /* set_sd */
 154     no_lookup_name,               /* lookup_name */
 155     no_link_name,                 /* link_name */
 156     NULL,                         /* unlink_name */
 157     no_open_file,                 /* open_file */
 158     fd_close_handle,              /* close_handle */
 159     sock_destroy                  /* destroy */
 160 };
 161
 162 static const struct fd_ops sock_fd_ops =
 163 {
 164     sock_get_poll_events,         /* get_poll_events */
 165     sock_poll_event,              /* poll_event */
 166     sock_get_fd_type,             /* get_fd_type */
 167     no_fd_read,                   /* read */
 168     no_fd_write,                  /* write */
 169     no_fd_flush,                  /* flush */
 170     no_fd_get_volume_info,        /* get_volume_info */
 171     sock_ioctl,                   /* ioctl */
 172     sock_queue_async,             /* queue_async */
 173     sock_reselect_async           /* reselect_async */
 174 };
 175
 176
 177 /* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
 178  * we post messages if there are multiple events.  Used to send
 179  * messages.  The problem is if there is both a FD_CONNECT event and,
 180  * say, an FD_READ event available on the same socket, we want to
 181  * notify the app of the connect event first.  Otherwise it may
 182  * discard the read event because it thinks it hasn't connected yet.
 183  */
 184 static const int event_bitorder[FD_MAX_EVENTS] =
 185 {
 186     FD_CONNECT_BIT,
 187     FD_ACCEPT_BIT,
 188     FD_OOB_BIT,
 189     FD_WRITE_BIT,
 190     FD_READ_BIT,
 191     FD_CLOSE_BIT,
 192     6, 7, 8, 9  /* leftovers */
 193 };
 194
 195 /* Flags that make sense only for SOCK_STREAM sockets */
 196 #define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
 197
 198 typedef enum {
 199     SOCK_SHUTDOWN_ERROR = -1,
 200     SOCK_SHUTDOWN_EOF = 0,
 201     SOCK_SHUTDOWN_POLLHUP = 1
 202 } sock_shutdown_t;
 203
 204 static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
 205
 206 static sock_shutdown_t sock_check_pollhup(void)
 207 {
 208     sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
 209     int fd[2], n;
 210     struct pollfd pfd;
 211     char dummy;
 212
 213     if ( socketpair( AF_UNIX, SOCK_STREAM, 0, fd ) ) return ret;
 214     if ( shutdown( fd[0], 1 ) ) goto out;
 215
 216     pfd.fd = fd[1];
 217     pfd.events = POLLIN;
 218     pfd.revents = 0;
 219
 220     /* Solaris' poll() sometimes returns nothing if given a 0ms timeout here */
 221     n = poll( &pfd, 1, 1 );
 222     if ( n != 1 ) goto out; /* error or timeout */
 223     if ( pfd.revents & POLLHUP )
 224         ret = SOCK_SHUTDOWN_POLLHUP;
 225     else if ( pfd.revents & POLLIN &&
 226               read( fd[1], &dummy, 1 ) == 0 )
 227         ret = SOCK_SHUTDOWN_EOF;
 228
 229 out:
 230     close( fd[0] );
 231     close( fd[1] );
 232     return ret;
 233 }
 234
 235 void sock_init(void)
 236 {
 237     sock_shutdown_type = sock_check_pollhup();
 238
 239     switch ( sock_shutdown_type )
 240     {
 241     case SOCK_SHUTDOWN_EOF:
 242         if (debug_level) fprintf( stderr, "sock_init: shutdown() causes EOF\n" );
 243         break;
 244     case SOCK_SHUTDOWN_POLLHUP:
 245         if (debug_level) fprintf( stderr, "sock_init: shutdown() causes POLLHUP\n" );
 246         break;
 247     default:
 248         fprintf( stderr, "sock_init: ERROR in sock_check_pollhup()\n" );
 249         sock_shutdown_type = SOCK_SHUTDOWN_EOF;
 250     }
 251 }
 252
 253 static int sock_reselect( struct sock *sock )
 254 {
 255     int ev = sock_get_poll_events( sock->fd );
 256
 257     if (debug_level)
 258         fprintf(stderr,"sock_reselect(%p): new mask %x\n", sock, ev);
 259
 260     if (!sock->polling)  /* FIXME: should find a better way to do this */
 261     {
 262         /* previously unconnected socket, is this reselect supposed to connect it? */
 263         if (!(sock->state & ~FD_WINE_NONBLOCKING)) return 0;
 264         /* ok, it is, attach it to the wineserver's main poll loop */
 265         sock->polling = 1;
 266         allow_fd_caching( sock->fd );
 267     }
 268     /* update condition mask */
 269     set_fd_events( sock->fd, ev );
 270     return ev;
 271 }
 272
 273 /* wake anybody waiting on the socket event or send the associated message */
 274 static void sock_wake_up( struct sock *sock )
 275 {
 276     unsigned int events = sock->pmask & sock->mask;
 277     int i;
 278
 279     if ( !events ) return;
 280
 281     if (sock->event)
 282     {
 283         if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
 284         set_event( sock->event );
 285     }
 286     if (sock->window)
 287     {
 288         if (debug_level) fprintf(stderr, "signalling events %x win %08x\n", events, sock->window );
 289         for (i = 0; i < FD_MAX_EVENTS; i++)
 290         {
 291             int event = event_bitorder[i];
 292             if (sock->pmask & (1 << event))
 293             {
 294                 lparam_t lparam = (1 << event) | (sock_get_error(sock->errors[event]) << 16);
 295                 post_message( sock->window, sock->message, sock->wparam, lparam );
 296             }
 297         }
 298         sock->pmask = 0;
 299         sock_reselect( sock );
 300     }
 301 }
 302
 303 static inline int sock_error( struct fd *fd )
 304 {
 305     unsigned int optval = 0;
 306     socklen_t optlen = sizeof(optval);
 307
 308     getsockopt( get_unix_fd(fd), SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
 309     return optval;
 310 }
 311
 312 static int sock_dispatch_asyncs( struct sock *sock, int event, int error )
 313 {
 314     if ( sock->flags & WSA_FLAG_OVERLAPPED )
 315     {
 316         if (event & (POLLIN|POLLPRI) && async_waiting( &sock->read_q ))
 317         {
 318             if (debug_level) fprintf( stderr, "activating read queue for socket %p\n", sock );
 319             async_wake_up( &sock->read_q, STATUS_ALERTED );
 320             event &= ~(POLLIN|POLLPRI);
 321         }
 322         if (event & POLLOUT && async_waiting( &sock->write_q ))
 323         {
 324             if (debug_level) fprintf( stderr, "activating write queue for socket %p\n", sock );
 325             async_wake_up( &sock->write_q, STATUS_ALERTED );
 326             event &= ~POLLOUT;
 327         }
 328         if ( event & (POLLERR|POLLHUP) )
 329         {
 330             int status = sock_get_ntstatus( error );
 331
 332             if ( !(sock->state & FD_READ) )
 333                 async_wake_up( &sock->read_q, status );
 334             if ( !(sock->state & FD_WRITE) )
 335                 async_wake_up( &sock->write_q, status );
 336         }
 337     }
 338     return event;
 339 }
 340
 341 static void sock_dispatch_events( struct sock *sock, int prevstate, int event, int error )
 342 {
 343     if (prevstate & FD_CONNECT)
 344     {
 345         sock->pmask |= FD_CONNECT;
 346         sock->hmask |= FD_CONNECT;
 347         sock->errors[FD_CONNECT_BIT] = error;
 348         goto end;
 349     }
 350     if (prevstate & FD_WINE_LISTENING)
 351     {
 352         sock->pmask |= FD_ACCEPT;
 353         sock->hmask |= FD_ACCEPT;
 354         sock->errors[FD_ACCEPT_BIT] = error;
 355         goto end;
 356     }
 357
 358     if (event & POLLIN)
 359     {
 360         sock->pmask |= FD_READ;
 361         sock->hmask |= FD_READ;
 362         sock->errors[FD_READ_BIT] = 0;
 363     }
 364
 365     if (event & POLLOUT)
 366     {
 367         sock->pmask |= FD_WRITE;
 368         sock->hmask |= FD_WRITE;
 369         sock->errors[FD_WRITE_BIT] = 0;
 370     }
 371
 372     if (event & POLLPRI)
 373     {
 374         sock->pmask |= FD_OOB;
 375         sock->hmask |= FD_OOB;
 376         sock->errors[FD_OOB_BIT] = 0;
 377     }
 378
 379     if (event & (POLLERR|POLLHUP))
 380     {
 381         sock->pmask |= FD_CLOSE;
 382         sock->hmask |= FD_CLOSE;
 383         sock->errors[FD_CLOSE_BIT] = error;
 384     }
 385 end:
 386     sock_wake_up( sock );
 387 }
 388
 389 static void sock_poll_event( struct fd *fd, int event )
 390 {
 391     struct sock *sock = get_fd_user( fd );
 392     int hangup_seen = 0;
 393     int prevstate = sock->state;
 394     int error = 0;
 395
 396     assert( sock->obj.ops == &sock_ops );
 397     if (debug_level)
 398         fprintf(stderr, "socket %p select event: %x\n", sock, event);
 399
 400     /* we may change event later, remove from loop here */
 401     if (event & (POLLERR|POLLHUP)) set_fd_events( sock->fd, -1 );
 402
 403     if (sock->state & FD_CONNECT)
 404     {
 405         if (event & (POLLERR|POLLHUP))
 406         {
 407             /* we didn't get connected? */
 408             sock->state &= ~FD_CONNECT;
 409             event &= ~POLLOUT;
 410             error = sock_error( fd );
 411         }
 412         else if (event & POLLOUT)
 413         {
 414             /* we got connected */
 415             sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 416             sock->state &= ~FD_CONNECT;
 417             sock->connect_time = current_time;
 418         }
 419     }
 420     else if (sock->state & FD_WINE_LISTENING)
 421     {
 422         /* listening */
 423         if (event & (POLLERR|POLLHUP))
 424             error = sock_error( fd );
 425     }
 426     else
 427     {
 428         /* normal data flow */
 429         if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
 430         {
 431             char dummy;
 432             int nr;
 433
 434             /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
 435              * has been closed, so we need to check for it explicitly here */
 436             nr  = recv( get_unix_fd( fd ), &dummy, 1, MSG_PEEK );
 437             if ( nr == 0 )
 438             {
 439                 hangup_seen = 1;
 440                 event &= ~POLLIN;
 441             }
 442             else if ( nr < 0 )
 443             {
 444                 event &= ~POLLIN;
 445                 /* EAGAIN can happen if an async recv() falls between the server's poll()
 446                    call and the invocation of this routine */
 447                 if ( errno != EAGAIN )
 448                 {
 449                     error = errno;
 450                     event |= POLLERR;
 451                     if ( debug_level )
 452                         fprintf( stderr, "recv error on socket %p: %d\n", sock, errno );
 453                 }
 454             }
 455         }
 456
 457         if ( (hangup_seen || event & (POLLHUP|POLLERR)) && (sock->state & (FD_READ|FD_WRITE)) )
 458         {
 459             error = error ? error : sock_error( fd );
 460             if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
 461                 sock->state &= ~FD_WRITE;
 462             sock->state &= ~FD_READ;
 463
 464             if (debug_level)
 465                 fprintf(stderr, "socket %p aborted by error %d, event: %x\n", sock, error, event);
 466         }
 467
 468         if (hangup_seen)
 469             event |= POLLHUP;
 470     }
 471
 472     event = sock_dispatch_asyncs( sock, event, error );
 473     sock_dispatch_events( sock, prevstate, event, error );
 474
 475     /* if anyone is stupid enough to wait on the socket object itself,
 476      * maybe we should wake them up too, just in case? */
 477     wake_up( &sock->obj, 0 );
 478
 479     sock_reselect( sock );
 480 }
 481
 482 static void sock_dump( struct object *obj, int verbose )
 483 {
 484     struct sock *sock = (struct sock *)obj;
 485     assert( obj->ops == &sock_ops );
 486     fprintf( stderr, "Socket fd=%p, state=%x, mask=%x, pending=%x, held=%x\n",
 487             sock->fd, sock->state,
 488             sock->mask, sock->pmask, sock->hmask );
 489 }
 490
 491 static int sock_signaled( struct object *obj, struct wait_queue_entry *entry )
 492 {
 493     struct sock *sock = (struct sock *)obj;
 494     assert( obj->ops == &sock_ops );
 495
 496     return check_fd_events( sock->fd, sock_get_poll_events( sock->fd ) ) != 0;
 497 }
 498
 499 static int sock_get_poll_events( struct fd *fd )
 500 {
 501     struct sock *sock = get_fd_user( fd );
 502     unsigned int mask = sock->mask & ~sock->hmask;
 503     unsigned int smask = sock->state & mask;
 504     int ev = 0;
 505
 506     assert( sock->obj.ops == &sock_ops );
 507
 508     if (sock->state & FD_CONNECT)
 509         /* connecting, wait for writable */
 510         return POLLOUT;
 511
 512     if (async_queued( &sock->read_q ))
 513     {
 514         if (async_waiting( &sock->read_q )) ev |= POLLIN | POLLPRI;
 515     }
 516     else if (smask & FD_READ || (sock->state & FD_WINE_LISTENING && mask & FD_ACCEPT))
 517         ev |= POLLIN | POLLPRI;
 518     /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
 519     else if ( sock->type == SOCK_STREAM && sock->state & FD_READ && mask & FD_CLOSE &&
 520               !(sock->hmask & FD_READ) )
 521         ev |= POLLIN;
 522
 523     if (async_queued( &sock->write_q ))
 524     {
 525         if (async_waiting( &sock->write_q )) ev |= POLLOUT;
 526     }
 527     else if (smask & FD_WRITE)
 528         ev |= POLLOUT;
 529
 530     return ev;
 531 }
 532
 533 static enum server_fd_type sock_get_fd_type( struct fd *fd )
 534 {
 535     return FD_TYPE_SOCKET;
 536 }
 537
 538 static int sock_ioctl( struct fd *fd, ioctl_code_t code, struct async *async )
 539 {
 540     struct sock *sock = get_fd_user( fd );
 541
 542     assert( sock->obj.ops == &sock_ops );
 543
 544     switch(code)
 545     {
 546     case WS_SIO_ADDRESS_LIST_CHANGE:
 547         if ((sock->state & FD_WINE_NONBLOCKING) && async_is_blocking( async ))
 548         {
 549             set_error( STATUS_CANT_WAIT );
 550             return 0;
 551         }
 552         if (!sock_get_ifchange( sock )) return 0;
 553         queue_async( &sock->ifchange_q, async );
 554         set_error( STATUS_PENDING );
 555         return 1;
 556     default:
 557         set_error( STATUS_NOT_SUPPORTED );
 558         return 0;
 559     }
 560 }
 561
 562 static void sock_queue_async( struct fd *fd, struct async *async, int type, int count )
 563 {
 564     struct sock *sock = get_fd_user( fd );
 565     struct async_queue *queue;
 566
 567     assert( sock->obj.ops == &sock_ops );
 568
 569     switch (type)
 570     {
 571     case ASYNC_TYPE_READ:
 572         queue = &sock->read_q;
 573         break;
 574     case ASYNC_TYPE_WRITE:
 575         queue = &sock->write_q;
 576         break;
 577     default:
 578         set_error( STATUS_INVALID_PARAMETER );
 579         return;
 580     }
 581
 582     if ( ( !( sock->state & (FD_READ|FD_CONNECT|FD_WINE_LISTENING) ) && type == ASYNC_TYPE_READ  ) ||
 583          ( !( sock->state & (FD_WRITE|FD_CONNECT) ) && type == ASYNC_TYPE_WRITE ) )
 584     {
 585         set_error( STATUS_PIPE_DISCONNECTED );
 586         return;
 587     }
 588
 589     queue_async( queue, async );
 590     sock_reselect( sock );
 591
 592     set_error( STATUS_PENDING );
 593 }
 594
 595 static void sock_reselect_async( struct fd *fd, struct async_queue *queue )
 596 {
 597     struct sock *sock = get_fd_user( fd );
 598     /* ignore reselect on ifchange queue */
 599     if (&sock->ifchange_q != queue)
 600         sock_reselect( sock );
 601 }
 602
 603 static struct fd *sock_get_fd( struct object *obj )
 604 {
 605     struct sock *sock = (struct sock *)obj;
 606     return (struct fd *)grab_object( sock->fd );
 607 }
 608
 609 static void sock_destroy( struct object *obj )
 610 {
 611     struct sock *sock = (struct sock *)obj;
 612     assert( obj->ops == &sock_ops );
 613
 614     /* FIXME: special socket shutdown stuff? */
 615
 616     if ( sock->deferred )
 617         release_object( sock->deferred );
 618
 619     async_wake_up( &sock->ifchange_q, STATUS_CANCELLED );
 620     sock_release_ifchange( sock );
 621     free_async_queue( &sock->read_q );
 622     free_async_queue( &sock->write_q );
 623     free_async_queue( &sock->ifchange_q );
 624     if (sock->event) release_object( sock->event );
 625     if (sock->fd)
 626     {
 627         /* shut the socket down to force pending poll() calls in the client to return */
 628         shutdown( get_unix_fd(sock->fd), SHUT_RDWR );
 629         release_object( sock->fd );
 630     }
 631 }
 632
 633 static void init_sock(struct sock *sock)
 634 {
 635     sock->state = 0;
 636     sock->mask    = 0;
 637     sock->hmask   = 0;
 638     sock->pmask   = 0;
 639     sock->polling = 0;
 640     sock->flags   = 0;
 641     sock->type    = 0;
 642     sock->family  = 0;
 643     sock->event   = NULL;
 644     sock->window  = 0;
 645     sock->message = 0;
 646     sock->wparam  = 0;
 647     sock->connect_time = 0;
 648     sock->deferred = NULL;
 649     sock->ifchange_obj = NULL;
 650     init_async_queue( &sock->read_q );
 651     init_async_queue( &sock->write_q );
 652     init_async_queue( &sock->ifchange_q );
 653     memset( sock->errors, 0, sizeof(sock->errors) );
 654 }
 655
 656 /* create a new and unconnected socket */
 657 static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
 658 {
 659     struct sock *sock;
 660     int sockfd;
 661
 662     sockfd = socket( family, type, protocol );
 663     if (debug_level)
 664         fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
 665     if (sockfd == -1)
 666     {
 667         sock_set_error();
 668         return NULL;
 669     }
 670     fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 671     if (!(sock = alloc_object( &sock_ops )))
 672     {
 673         close( sockfd );
 674         return NULL;
 675     }
 676     init_sock( sock );
 677     sock->state  = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
 678     sock->flags  = flags;
 679     sock->proto  = protocol;
 680     sock->type   = type;
 681     sock->family = family;
 682
 683     if (!(sock->fd = create_anonymous_fd( &sock_fd_ops, sockfd, &sock->obj,
 684                             (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT )))
 685     {
 686         release_object( sock );
 687         return NULL;
 688     }
 689     sock_reselect( sock );
 690     clear_error();
 691     return &sock->obj;
 692 }
 693
 694 /* accepts a socket and inits it */
 695 static int accept_new_fd( struct sock *sock )
 696 {
 697
 698     /* Try to accept(2). We can't be safe that this an already connected socket
 699      * or that accept() is allowed on it. In those cases we will get -1/errno
 700      * return.
 701      */
 702     int acceptfd;
 703     struct sockaddr saddr;
 704     socklen_t slen = sizeof(saddr);
 705     acceptfd = accept( get_unix_fd(sock->fd), &saddr, &slen);
 706     if (acceptfd == -1)
 707     {
 708         sock_set_error();
 709         return acceptfd;
 710     }
 711
 712     fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 713     return acceptfd;
 714 }
 715
 716 /* accept a socket (creates a new fd) */
 717 static struct sock *accept_socket( obj_handle_t handle )
 718 {
 719     struct sock *acceptsock;
 720     struct sock *sock;
 721     int acceptfd;
 722
 723     sock = (struct sock *)get_handle_obj( current->process, handle, FILE_READ_DATA, &sock_ops );
 724     if (!sock)
 725         return NULL;
 726
 727     if ( sock->deferred )
 728     {
 729         acceptsock = sock->deferred;
 730         sock->deferred = NULL;
 731     }
 732     else
 733     {
 734         if ((acceptfd = accept_new_fd( sock )) == -1)
 735         {
 736             release_object( sock );
 737             return NULL;
 738         }
 739         if (!(acceptsock = alloc_object( &sock_ops )))
 740         {
 741             close( acceptfd );
 742             release_object( sock );
 743             return NULL;
 744         }
 745
 746         init_sock( acceptsock );
 747         /* newly created socket gets the same properties of the listening socket */
 748         acceptsock->state  = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 749         if (sock->state & FD_WINE_NONBLOCKING)
 750             acceptsock->state |= FD_WINE_NONBLOCKING;
 751         acceptsock->mask    = sock->mask;
 752         acceptsock->proto   = sock->proto;
 753         acceptsock->type    = sock->type;
 754         acceptsock->family  = sock->family;
 755         acceptsock->window  = sock->window;
 756         acceptsock->message = sock->message;
 757         acceptsock->connect_time = current_time;
 758         if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
 759         acceptsock->flags = sock->flags;
 760         if (!(acceptsock->fd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
 761                                                     get_fd_options( sock->fd ) )))
 762         {
 763             release_object( acceptsock );
 764             release_object( sock );
 765             return NULL;
 766         }
 767     }
 768     clear_error();
 769     sock->pmask &= ~FD_ACCEPT;
 770     sock->hmask &= ~FD_ACCEPT;
 771     sock_reselect( sock );
 772     release_object( sock );
 773     return acceptsock;
 774 }
 775
 776 static int accept_into_socket( struct sock *sock, struct sock *acceptsock )
 777 {
 778     int acceptfd;
 779     struct fd *newfd;
 780     if ( sock->deferred )
 781     {
 782         newfd = dup_fd_object( sock->deferred->fd, 0, 0,
 783                                get_fd_options( acceptsock->fd ) );
 784         if ( !newfd )
 785             return FALSE;
 786
 787         set_fd_user( newfd, &sock_fd_ops, &acceptsock->obj );
 788
 789         release_object( sock->deferred );
 790         sock->deferred = NULL;
 791     }
 792     else
 793     {
 794         if ((acceptfd = accept_new_fd( sock )) == -1)
 795             return FALSE;
 796
 797         if (!(newfd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
 798                                             get_fd_options( acceptsock->fd ) )))
 799             return FALSE;
 800     }
 801
 802     acceptsock->state  |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 803     acceptsock->hmask   = 0;
 804     acceptsock->pmask   = 0;
 805     acceptsock->polling = 0;
 806     acceptsock->proto   = sock->proto;
 807     acceptsock->type    = sock->type;
 808     acceptsock->family  = sock->family;
 809     acceptsock->wparam  = 0;
 810     acceptsock->deferred = NULL;
 811     acceptsock->connect_time = current_time;
 812     fd_copy_completion( acceptsock->fd, newfd );
 813     release_object( acceptsock->fd );
 814     acceptsock->fd = newfd;
 815
 816     clear_error();
 817     sock->pmask &= ~FD_ACCEPT;
 818     sock->hmask &= ~FD_ACCEPT;
 819     sock_reselect( sock );
 820
 821     return TRUE;
 822 }
 823
 824 /* return an errno value mapped to a WSA error */
 825 static int sock_get_error( int err )
 826 {
 827     switch (err)
 828     {
 829         case EINTR:             return WSAEINTR;
 830         case EBADF:             return WSAEBADF;
 831         case EPERM:
 832         case EACCES:            return WSAEACCES;
 833         case EFAULT:            return WSAEFAULT;
 834         case EINVAL:            return WSAEINVAL;
 835         case EMFILE:            return WSAEMFILE;
 836         case EWOULDBLOCK:       return WSAEWOULDBLOCK;
 837         case EINPROGRESS:       return WSAEINPROGRESS;
 838         case EALREADY:          return WSAEALREADY;
 839         case ENOTSOCK:          return WSAENOTSOCK;
 840         case EDESTADDRREQ:      return WSAEDESTADDRREQ;
 841         case EMSGSIZE:          return WSAEMSGSIZE;
 842         case EPROTOTYPE:        return WSAEPROTOTYPE;
 843         case ENOPROTOOPT:       return WSAENOPROTOOPT;
 844         case EPROTONOSUPPORT:   return WSAEPROTONOSUPPORT;
 845         case ESOCKTNOSUPPORT:   return WSAESOCKTNOSUPPORT;
 846         case EOPNOTSUPP:        return WSAEOPNOTSUPP;
 847         case EPFNOSUPPORT:      return WSAEPFNOSUPPORT;
 848         case EAFNOSUPPORT:      return WSAEAFNOSUPPORT;
 849         case EADDRINUSE:        return WSAEADDRINUSE;
 850         case EADDRNOTAVAIL:     return WSAEADDRNOTAVAIL;
 851         case ENETDOWN:          return WSAENETDOWN;
 852         case ENETUNREACH:       return WSAENETUNREACH;
 853         case ENETRESET:         return WSAENETRESET;
 854         case ECONNABORTED:      return WSAECONNABORTED;
 855         case EPIPE:
 856         case ECONNRESET:        return WSAECONNRESET;
 857         case ENOBUFS:           return WSAENOBUFS;
 858         case EISCONN:           return WSAEISCONN;
 859         case ENOTCONN:          return WSAENOTCONN;
 860         case ESHUTDOWN:         return WSAESHUTDOWN;
 861         case ETOOMANYREFS:      return WSAETOOMANYREFS;
 862         case ETIMEDOUT:         return WSAETIMEDOUT;
 863         case ECONNREFUSED:      return WSAECONNREFUSED;
 864         case ELOOP:             return WSAELOOP;
 865         case ENAMETOOLONG:      return WSAENAMETOOLONG;
 866         case EHOSTDOWN:         return WSAEHOSTDOWN;
 867         case EHOSTUNREACH:      return WSAEHOSTUNREACH;
 868         case ENOTEMPTY:         return WSAENOTEMPTY;
 869 #ifdef EPROCLIM
 870         case EPROCLIM:          return WSAEPROCLIM;
 871 #endif
 872 #ifdef EUSERS
 873         case EUSERS:            return WSAEUSERS;
 874 #endif
 875 #ifdef EDQUOT
 876         case EDQUOT:            return WSAEDQUOT;
 877 #endif
 878 #ifdef ESTALE
 879         case ESTALE:            return WSAESTALE;
 880 #endif
 881 #ifdef EREMOTE
 882         case EREMOTE:           return WSAEREMOTE;
 883 #endif
 884
 885         case 0:                 return 0;
 886         default:
 887             errno = err;
 888             perror("wineserver: sock_get_error() can't map error");
 889             return WSAEFAULT;
 890     }
 891 }
 892
 893 static int sock_get_ntstatus( int err )
 894 {
 895     switch ( err )
 896     {
 897         case EBADF:             return STATUS_INVALID_HANDLE;
 898         case EBUSY:             return STATUS_DEVICE_BUSY;
 899         case EPERM:
 900         case EACCES:            return STATUS_ACCESS_DENIED;
 901         case EFAULT:            return STATUS_NO_MEMORY;
 902         case EINVAL:            return STATUS_INVALID_PARAMETER;
 903         case ENFILE:
 904         case EMFILE:            return STATUS_TOO_MANY_OPENED_FILES;
 905         case EWOULDBLOCK:       return STATUS_CANT_WAIT;
 906         case EINPROGRESS:       return STATUS_PENDING;
 907         case EALREADY:          return STATUS_NETWORK_BUSY;
 908         case ENOTSOCK:          return STATUS_OBJECT_TYPE_MISMATCH;
 909         case EDESTADDRREQ:      return STATUS_INVALID_PARAMETER;
 910         case EMSGSIZE:          return STATUS_BUFFER_OVERFLOW;
 911         case EPROTONOSUPPORT:
 912         case ESOCKTNOSUPPORT:
 913         case EPFNOSUPPORT:
 914         case EAFNOSUPPORT:
 915         case EPROTOTYPE:        return STATUS_NOT_SUPPORTED;
 916         case ENOPROTOOPT:       return STATUS_INVALID_PARAMETER;
 917         case EOPNOTSUPP:        return STATUS_NOT_SUPPORTED;
 918         case EADDRINUSE:        return STATUS_ADDRESS_ALREADY_ASSOCIATED;
 919         case EADDRNOTAVAIL:     return STATUS_INVALID_PARAMETER;
 920         case ECONNREFUSED:      return STATUS_CONNECTION_REFUSED;
 921         case ESHUTDOWN:         return STATUS_PIPE_DISCONNECTED;
 922         case ENOTCONN:          return STATUS_CONNECTION_DISCONNECTED;
 923         case ETIMEDOUT:         return STATUS_IO_TIMEOUT;
 924         case ENETUNREACH:       return STATUS_NETWORK_UNREACHABLE;
 925         case EHOSTUNREACH:      return STATUS_HOST_UNREACHABLE;
 926         case ENETDOWN:          return STATUS_NETWORK_BUSY;
 927         case EPIPE:
 928         case ECONNRESET:        return STATUS_CONNECTION_RESET;
 929         case ECONNABORTED:      return STATUS_CONNECTION_ABORTED;
 930
 931         case 0:                 return STATUS_SUCCESS;
 932         default:
 933             errno = err;
 934             perror("wineserver: sock_get_ntstatus() can't map error");
 935             return STATUS_UNSUCCESSFUL;
 936     }
 937 }
 938
 939 /* set the last error depending on errno */
 940 static void sock_set_error(void)
 941 {
 942     set_error( sock_get_ntstatus( errno ) );
 943 }
 944
 945 #ifdef HAVE_LINUX_RTNETLINK_H
 946
 947 /* only keep one ifchange object around, all sockets waiting for wakeups will look to it */
 948 static struct object *ifchange_object;
 949
 950 static void ifchange_dump( struct object *obj, int verbose );
 951 static struct fd *ifchange_get_fd( struct object *obj );
 952 static void ifchange_destroy( struct object *obj );
 953
 954 static int ifchange_get_poll_events( struct fd *fd );
 955 static void ifchange_poll_event( struct fd *fd, int event );
 956
 957 struct ifchange
 958 {
 959     struct object       obj;     /* object header */
 960     struct fd          *fd;      /* interface change file descriptor */
 961     struct list         sockets; /* list of sockets to send interface change notifications */
 962 };
 963
 964 static const struct object_ops ifchange_ops =
 965 {
 966     sizeof(struct ifchange), /* size */
 967     ifchange_dump,           /* dump */
 968     no_get_type,             /* get_type */
 969     add_queue,               /* add_queue */
 970     NULL,                    /* remove_queue */
 971     NULL,                    /* signaled */
 972     no_satisfied,            /* satisfied */
 973     no_signal,               /* signal */
 974     ifchange_get_fd,         /* get_fd */
 975     default_fd_map_access,   /* map_access */
 976     default_get_sd,          /* get_sd */
 977     default_set_sd,          /* set_sd */
 978     no_lookup_name,          /* lookup_name */
 979     no_link_name,            /* link_name */
 980     NULL,                    /* unlink_name */
 981     no_open_file,            /* open_file */
 982     no_close_handle,         /* close_handle */
 983     ifchange_destroy         /* destroy */
 984 };
 985
 986 static const struct fd_ops ifchange_fd_ops =
 987 {
 988     ifchange_get_poll_events, /* get_poll_events */
 989     ifchange_poll_event,      /* poll_event */
 990     NULL,                     /* get_fd_type */
 991     no_fd_read,               /* read */
 992     no_fd_write,              /* write */
 993     no_fd_flush,              /* flush */
 994     no_fd_get_volume_info,    /* get_volume_info */
 995     no_fd_ioctl,              /* ioctl */
 996     NULL,                     /* queue_async */
 997     NULL                      /* reselect_async */
 998 };
 999
1000 static void ifchange_dump( struct object *obj, int verbose )
1001 {
1002     assert( obj->ops == &ifchange_ops );
1003     fprintf( stderr, "Interface change\n" );
1004 }
1005
1006 static struct fd *ifchange_get_fd( struct object *obj )
1007 {
1008     struct ifchange *ifchange = (struct ifchange *)obj;
1009     return (struct fd *)grab_object( ifchange->fd );
1010 }
1011
1012 static void ifchange_destroy( struct object *obj )
1013 {
1014     struct ifchange *ifchange = (struct ifchange *)obj;
1015     assert( obj->ops == &ifchange_ops );
1016
1017     release_object( ifchange->fd );
1018
1019     /* reset the global ifchange object so that it will be recreated if it is needed again */
1020     assert( obj == ifchange_object );
1021     ifchange_object = NULL;
1022 }
1023
1024 static int ifchange_get_poll_events( struct fd *fd )
1025 {
1026     return POLLIN;
1027 }
1028
1029 /* wake up all the sockets waiting for a change notification event */
1030 static void ifchange_wake_up( struct object *obj, unsigned int status )
1031 {
1032     struct ifchange *ifchange = (struct ifchange *)obj;
1033     struct list *ptr, *next;
1034     assert( obj->ops == &ifchange_ops );
1035     assert( obj == ifchange_object );
1036
1037     LIST_FOR_EACH_SAFE( ptr, next, &ifchange->sockets )
1038     {
1039         struct sock *sock = LIST_ENTRY( ptr, struct sock, ifchange_entry );
1040
1041         assert( sock->ifchange_obj );
1042         async_wake_up( &sock->ifchange_q, status ); /* issue ifchange notification for the socket */
1043         sock_release_ifchange( sock ); /* remove socket from list and decrement ifchange refcount */
1044     }
1045 }
1046
1047 static void ifchange_poll_event( struct fd *fd, int event )
1048 {
1049     struct object *ifchange = get_fd_user( fd );
1050     unsigned int status = STATUS_PENDING;
1051     char buffer[PIPE_BUF];
1052     int r;
1053
1054     r = recv( get_unix_fd(fd), buffer, sizeof(buffer), MSG_DONTWAIT );
1055     if (r < 0)
1056     {
1057         if (errno == EWOULDBLOCK || (EWOULDBLOCK != EAGAIN && errno == EAGAIN))
1058             return;  /* retry when poll() says the socket is ready */
1059         status = sock_get_ntstatus( errno );
1060     }
1061     else if (r > 0)
1062     {
1063         struct nlmsghdr *nlh;
1064
1065         for (nlh = (struct nlmsghdr *)buffer; NLMSG_OK(nlh, r); nlh = NLMSG_NEXT(nlh, r))
1066         {
1067             if (nlh->nlmsg_type == NLMSG_DONE)
1068                 break;
1069             if (nlh->nlmsg_type == RTM_NEWADDR || nlh->nlmsg_type == RTM_DELADDR)
1070                 status = STATUS_SUCCESS;
1071         }
1072     }
1073     else status = STATUS_CANCELLED;
1074
1075     if (status != STATUS_PENDING) ifchange_wake_up( ifchange, status );
1076 }
1077
1078 #endif
1079
1080 /* we only need one of these interface notification objects, all of the sockets dependent upon
1081  * it will wake up when a notification event occurs */
1082  static struct object *get_ifchange( void )
1083  {
1084 #ifdef HAVE_LINUX_RTNETLINK_H
1085     struct ifchange *ifchange;
1086     struct sockaddr_nl addr;
1087     int unix_fd;
1088
1089     if (ifchange_object)
1090     {
1091         /* increment the refcount for each socket that uses the ifchange object */
1092         return grab_object( ifchange_object );
1093     }
1094
1095     /* create the socket we need for processing interface change notifications */
1096     unix_fd = socket( PF_NETLINK, SOCK_RAW, NETLINK_ROUTE );
1097     if (unix_fd == -1)
1098     {
1099         sock_set_error();
1100         return NULL;
1101     }
1102     fcntl( unix_fd, F_SETFL, O_NONBLOCK ); /* make socket nonblocking */
1103     memset( &addr, 0, sizeof(addr) );
1104     addr.nl_family = AF_NETLINK;
1105     addr.nl_groups = RTMGRP_IPV4_IFADDR;
1106     /* bind the socket to the special netlink kernel interface */
1107     if (bind( unix_fd, (struct sockaddr *)&addr, sizeof(addr) ) == -1)
1108     {
1109         close( unix_fd );
1110         sock_set_error();
1111         return NULL;
1112     }
1113     if (!(ifchange = alloc_object( &ifchange_ops )))
1114     {
1115         close( unix_fd );
1116         set_error( STATUS_NO_MEMORY );
1117         return NULL;
1118     }
1119     list_init( &ifchange->sockets );
1120     if (!(ifchange->fd = create_anonymous_fd( &ifchange_fd_ops, unix_fd, &ifchange->obj, 0 )))
1121     {
1122         release_object( ifchange );
1123         set_error( STATUS_NO_MEMORY );
1124         return NULL;
1125     }
1126     set_fd_events( ifchange->fd, POLLIN ); /* enable read wakeup on the file descriptor */
1127
1128     /* the ifchange object is now successfully configured */
1129     ifchange_object = &ifchange->obj;
1130     return &ifchange->obj;
1131 #else
1132     set_error( STATUS_NOT_SUPPORTED );
1133     return NULL;
1134 #endif
1135 }
1136
1137 /* add the socket to the interface change notification list */
1138 static void ifchange_add_sock( struct object *obj, struct sock *sock )
1139 {
1140 #ifdef HAVE_LINUX_RTNETLINK_H
1141     struct ifchange *ifchange = (struct ifchange *)obj;
1142
1143     list_add_tail( &ifchange->sockets, &sock->ifchange_entry );
1144 #endif
1145 }
1146
1147 /* create a new ifchange queue for a specific socket or, if one already exists, reuse the existing one */
1148 static struct object *sock_get_ifchange( struct sock *sock )
1149 {
1150     struct object *ifchange;
1151
1152     if (sock->ifchange_obj) /* reuse existing ifchange_obj for this socket */
1153         return sock->ifchange_obj;
1154
1155     if (!(ifchange = get_ifchange()))
1156         return NULL;
1157
1158     /* add the socket to the ifchange notification list */
1159     ifchange_add_sock( ifchange, sock );
1160     sock->ifchange_obj = ifchange;
1161     return ifchange;
1162 }
1163
1164 /* destroy an existing ifchange queue for a specific socket */
1165 static void sock_release_ifchange( struct sock *sock )
1166 {
1167     if (sock->ifchange_obj)
1168     {
1169         list_remove( &sock->ifchange_entry );
1170         release_object( sock->ifchange_obj );
1171         sock->ifchange_obj = NULL;
1172     }
1173 }
1174
1175 /* create a socket */
1176 DECL_HANDLER(create_socket)
1177 {
1178     struct object *obj;
1179
1180     reply->handle = 0;
1181     if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
1182     {
1183         reply->handle = alloc_handle( current->process, obj, req->access, req->attributes );
1184         release_object( obj );
1185     }
1186 }
1187
1188 /* accept a socket */
1189 DECL_HANDLER(accept_socket)
1190 {
1191     struct sock *sock;
1192
1193     reply->handle = 0;
1194     if ((sock = accept_socket( req->lhandle )) != NULL)
1195     {
1196         reply->handle = alloc_handle( current->process, &sock->obj, req->access, req->attributes );
1197         sock->wparam = reply->handle;  /* wparam for message is the socket handle */
1198         sock_reselect( sock );
1199         release_object( &sock->obj );
1200     }
1201 }
1202
1203 /* accept a socket into an initialized socket */
1204 DECL_HANDLER(accept_into_socket)
1205 {
1206     struct sock *sock, *acceptsock;
1207     const int all_attributes = FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES|FILE_READ_DATA;
1208
1209     if (!(sock = (struct sock *)get_handle_obj( current->process, req->lhandle,
1210                                                 all_attributes, &sock_ops)))
1211         return;
1212
1213     if (!(acceptsock = (struct sock *)get_handle_obj( current->process, req->ahandle,
1214                                                       all_attributes, &sock_ops)))
1215     {
1216         release_object( sock );
1217         return;
1218     }
1219
1220     if (accept_into_socket( sock, acceptsock ))
1221     {
1222         acceptsock->wparam = req->ahandle;  /* wparam for message is the socket handle */
1223         sock_reselect( acceptsock );
1224     }
1225     release_object( acceptsock );
1226     release_object( sock );
1227 }
1228
1229 /* set socket event parameters */
1230 DECL_HANDLER(set_socket_event)
1231 {
1232     struct sock *sock;
1233     struct event *old_event;
1234
1235     if (!(sock = (struct sock *)get_handle_obj( current->process, req->handle,
1236                                                 FILE_WRITE_ATTRIBUTES, &sock_ops))) return;
1237     old_event = sock->event;
1238     sock->mask    = req->mask;
1239     sock->hmask   &= ~req->mask; /* re-enable held events */
1240     sock->event   = NULL;
1241     sock->window  = req->window;
1242     sock->message = req->msg;
1243     sock->wparam  = req->handle;  /* wparam is the socket handle */
1244     if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
1245
1246     if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
1247
1248     sock_reselect( sock );
1249
1250     sock->state |= FD_WINE_NONBLOCKING;
1251
1252     /* if a network event is pending, signal the event object
1253        it is possible that FD_CONNECT or FD_ACCEPT network events has happened
1254        before a WSAEventSelect() was done on it.
1255        (when dealing with Asynchronous socket)  */
1256     sock_wake_up( sock );
1257
1258     if (old_event) release_object( old_event ); /* we're through with it */
1259     release_object( &sock->obj );
1260 }
1261
1262 /* get socket event parameters */
1263 DECL_HANDLER(get_socket_event)
1264 {
1265     struct sock *sock;
1266     int i;
1267     int errors[FD_MAX_EVENTS];
1268
1269     sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
1270     if (!sock)
1271     {
1272         reply->mask  = 0;
1273         reply->pmask = 0;
1274         reply->state = 0;
1275         return;
1276     }
1277     reply->mask  = sock->mask;
1278     reply->pmask = sock->pmask;
1279     reply->state = sock->state;
1280     for (i = 0; i < FD_MAX_EVENTS; i++)
1281         errors[i] = sock_get_ntstatus(sock->errors[i]);
1282
1283     set_reply_data( errors, min( get_reply_max_size(), sizeof(errors) ));
1284
1285     if (req->service)
1286     {
1287         if (req->c_event)
1288         {
1289             struct event *cevent = get_event_obj( current->process, req->c_event,
1290                                                   EVENT_MODIFY_STATE );
1291             if (cevent)
1292             {
1293                 reset_event( cevent );
1294                 release_object( cevent );
1295             }
1296         }
1297         sock->pmask = 0;
1298         sock_reselect( sock );
1299     }
1300     release_object( &sock->obj );
1301 }
1302
1303 /* re-enable pending socket events */
1304 DECL_HANDLER(enable_socket_event)
1305 {
1306     struct sock *sock;
1307
1308     if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
1309                                                FILE_WRITE_ATTRIBUTES, &sock_ops)))
1310         return;
1311
1312     /* for event-based notification, windows erases stale events */
1313     sock->pmask &= ~req->mask;
1314
1315     sock->hmask &= ~req->mask;
1316     sock->state |= req->sstate;
1317     sock->state &= ~req->cstate;
1318     if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
1319
1320     sock_reselect( sock );
1321
1322     release_object( &sock->obj );
1323 }
1324
1325 DECL_HANDLER(set_socket_deferred)
1326 {
1327     struct sock *sock, *acceptsock;
1328
1329     sock=(struct sock *)get_handle_obj( current->process, req->handle, FILE_WRITE_ATTRIBUTES, &sock_ops );
1330     if ( !sock )
1331         return;
1332
1333     acceptsock = (struct sock *)get_handle_obj( current->process, req->deferred, 0, &sock_ops );
1334     if ( !acceptsock )
1335     {
1336         release_object( sock );
1337         return;
1338     }
1339     sock->deferred = acceptsock;
1340     release_object( sock );
1341 }
1342
1343 DECL_HANDLER(get_socket_info)
1344 {
1345     struct sock *sock;
1346
1347     sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
1348     if (!sock) return;
1349
1350     reply->family   = sock->family;
1351     reply->type     = sock->type;
1352     reply->protocol = sock->proto;
1353
1354     release_object( &sock->obj );
1355 }