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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 <stdio.h>
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <errno.h>
  32 #ifdef HAVE_SYS_ERRNO_H
  33 # include <sys/errno.h>
  34 #endif
  35 #include <sys/time.h>
  36 #include <sys/types.h>
  37 #ifdef HAVE_SYS_SOCKET_H
  38 # include <sys/socket.h>
  39 #endif
  40 #ifdef HAVE_SYS_IOCTL_H
  41 #include <sys/ioctl.h>
  42 #endif
  43 #ifdef HAVE_SYS_FILIO_H
  44 # include <sys/filio.h>
  45 #endif
  46 #include <time.h>
  47 #include <unistd.h>
  48
  49 #include "winbase.h"
  50
  51 #include "process.h"
  52 #include "file.h"
  53 #include "handle.h"
  54 #include "thread.h"
  55 #include "request.h"
  56 #include "user.h"
  57 #include "async.h"
  58
  59 /* To avoid conflicts with the Unix socket headers. Plus we only need a few
  60  * macros anyway.
  61  */
  62 #define USE_WS_PREFIX
  63 #include "winsock2.h"
  64
  65 struct sock
  66 {
  67     struct object       obj;         /* object header */
  68     unsigned int        state;       /* status bits */
  69     unsigned int        mask;        /* event mask */
  70     unsigned int        hmask;       /* held (blocked) events */
  71     unsigned int        pmask;       /* pending events */
  72     unsigned int        flags;       /* socket flags */
  73     unsigned short      type;        /* socket type */
  74     unsigned short      family;      /* socket family */
  75     struct event       *event;       /* event object */
  76     user_handle_t       window;      /* window to send the message to */
  77     unsigned int        message;     /* message to send */
  78     obj_handle_t        wparam;      /* message wparam (socket handle) */
  79     int                 errors[FD_MAX_EVENTS]; /* event errors */
  80     struct sock*        deferred;    /* socket that waits for a deferred accept */
  81     struct async_queue  read_q;      /* Queue for asynchronous reads */
  82     struct async_queue  write_q;     /* Queue for asynchronous writes */
  83 };
  84
  85 static void sock_dump( struct object *obj, int verbose );
  86 static int sock_signaled( struct object *obj, struct thread *thread );
  87 static int sock_get_poll_events( struct object *obj );
  88 static void sock_poll_event( struct object *obj, int event );
  89 static int sock_get_info( struct object *obj, struct get_file_info_reply *reply, int *flags );
  90 static void sock_destroy( struct object *obj );
  91 static int sock_get_error( int err );
  92 static void sock_set_error(void);
  93 static void sock_queue_async(struct object *obj, void *ptr, unsigned int status, int type, int count);
  94
  95 static const struct object_ops sock_ops =
  96 {
  97     sizeof(struct sock),          /* size */
  98     sock_dump,                    /* dump */
  99     add_queue,                    /* add_queue */
 100     remove_queue,                 /* remove_queue */
 101     sock_signaled,                /* signaled */
 102     no_satisfied,                 /* satisfied */
 103     default_get_fd,               /* get_fd */
 104     sock_get_info,                /* get_file_info */
 105     sock_destroy                  /* destroy */
 106 };
 107
 108 static const struct fd_ops sock_fd_ops =
 109 {
 110     sock_get_poll_events,         /* get_poll_events */
 111     sock_poll_event,              /* poll_event */
 112     no_flush,                     /* flush */
 113     sock_get_info,                /* get_file_info */
 114     sock_queue_async              /* queue_async */
 115 };
 116
 117
 118 /* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
 119  * we post messages if there are multiple events.  Used to send
 120  * messages.  The problem is if there is both a FD_CONNECT event and,
 121  * say, an FD_READ event available on the same socket, we want to
 122  * notify the app of the connect event first.  Otherwise it may
 123  * discard the read event because it thinks it hasn't connected yet.
 124  */
 125 static const int event_bitorder[FD_MAX_EVENTS] =
 126 {
 127     FD_CONNECT_BIT,
 128     FD_ACCEPT_BIT,
 129     FD_OOB_BIT,
 130     FD_WRITE_BIT,
 131     FD_READ_BIT,
 132     FD_CLOSE_BIT,
 133     6, 7, 8, 9  /* leftovers */
 134 };
 135
 136 /* Flags that make sense only for SOCK_STREAM sockets */
 137 #define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
 138
 139 typedef enum {
 140     SOCK_SHUTDOWN_ERROR = -1,
 141     SOCK_SHUTDOWN_EOF = 0,
 142     SOCK_SHUTDOWN_POLLHUP = 1
 143 } sock_shutdown_t;
 144
 145 static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
 146
 147 static sock_shutdown_t sock_check_pollhup (void)
 148 {
 149     sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
 150     int fd[2], n;
 151     struct pollfd pfd;
 152     char dummy;
 153
 154     if ( socketpair ( AF_UNIX, SOCK_STREAM, 0, fd ) ) goto out;
 155     if ( shutdown ( fd[0], 1 ) ) goto out;
 156
 157     pfd.fd = fd[1];
 158     pfd.events = POLLIN;
 159     pfd.revents = 0;
 160
 161     n = poll ( &pfd, 1, 0 );
 162     if ( n != 1 ) goto out; /* error or timeout */
 163     if ( pfd.revents & POLLHUP )
 164         ret = SOCK_SHUTDOWN_POLLHUP;
 165     else if ( pfd.revents & POLLIN &&
 166               read ( fd[1], &dummy, 1 ) == 0 )
 167         ret = SOCK_SHUTDOWN_EOF;
 168
 169 out:
 170     close ( fd[0] );
 171     close ( fd[1] );
 172     return ret;
 173 }
 174
 175 void sock_init(void)
 176 {
 177     sock_shutdown_type = sock_check_pollhup ();
 178
 179     switch ( sock_shutdown_type )
 180     {
 181     case SOCK_SHUTDOWN_EOF:
 182         if (debug_level) fprintf ( stderr, "sock_init: shutdown() causes EOF\n" );
 183         break;
 184     case SOCK_SHUTDOWN_POLLHUP:
 185         if (debug_level) fprintf ( stderr, "sock_init: shutdown() causes POLLHUP\n" );
 186         break;
 187     default:
 188         fprintf ( stderr, "sock_init: ERROR in sock_check_pollhup()\n" );
 189         sock_shutdown_type = SOCK_SHUTDOWN_EOF;
 190     }
 191 }
 192
 193 static int sock_reselect( struct sock *sock )
 194 {
 195     int ev = sock_get_poll_events( &sock->obj );
 196
 197     if (debug_level)
 198         fprintf(stderr,"sock_reselect(%d): new mask %x\n", sock->obj.fd, ev);
 199
 200     if (sock->obj.select == -1) {
 201         /* previously unconnected socket, is this reselect supposed to connect it? */
 202         if (!(sock->state & ~FD_WINE_NONBLOCKING)) return 0;
 203         /* ok, it is, attach it to the wineserver's main poll loop */
 204         add_select_user( &sock->obj );
 205     }
 206     /* update condition mask */
 207     set_select_events( &sock->obj, ev );
 208     return ev;
 209 }
 210
 211 /* After POLLHUP is received, the socket will no longer be in the main select loop.
 212    This function is used to signal pending events nevertheless */
 213 static void sock_try_event ( struct sock *sock, int event )
 214 {
 215     struct pollfd pfd;
 216
 217     pfd.fd = sock->obj.fd;
 218     pfd.events = event;
 219     pfd.revents = 0;
 220     poll (&pfd, 1, 0);
 221
 222     if ( pfd.revents )
 223     {
 224         if ( debug_level ) fprintf ( stderr, "sock_try_event: %x\n", pfd.revents );
 225         sock_poll_event ( &sock->obj, pfd.revents );
 226     }
 227 }
 228
 229 /* wake anybody waiting on the socket event or send the associated message */
 230 static void sock_wake_up( struct sock *sock, int pollev )
 231 {
 232     unsigned int events = sock->pmask & sock->mask;
 233     int i;
 234     int async_active = 0;
 235
 236     if ( sock->flags & FD_FLAG_OVERLAPPED )
 237     {
 238         if( pollev & (POLLIN|POLLPRI) && IS_READY( sock->read_q ) )
 239         {
 240             if (debug_level) fprintf ( stderr, "activating read queue for socket %p\n", sock );
 241             async_notify( sock->read_q.head, STATUS_ALERTED );
 242             async_active = 1;
 243         }
 244         if( pollev & POLLOUT && IS_READY( sock->write_q ) )
 245         {
 246             if (debug_level) fprintf ( stderr, "activating write queue for socket %p\n", sock );
 247             async_notify( sock->write_q.head, STATUS_ALERTED );
 248             async_active = 1;
 249         }
 250     }
 251
 252     /* Do not signal events if there are still pending asynchronous IO requests */
 253     /* We need this to delay FD_CLOSE events until all pending overlapped requests are processed */
 254     if ( !events || async_active ) return;
 255
 256     if (sock->event)
 257     {
 258         if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
 259         set_event( sock->event );
 260     }
 261     if (sock->window)
 262     {
 263         if (debug_level) fprintf(stderr, "signalling events %x win %p\n", events, sock->window );
 264         for (i = 0; i < FD_MAX_EVENTS; i++)
 265         {
 266             int event = event_bitorder[i];
 267             if (sock->pmask & (1 << event))
 268             {
 269                 unsigned int lparam = (1 << event) | (sock->errors[event] << 16);
 270                 post_message( sock->window, sock->message, (unsigned int)sock->wparam, lparam );
 271             }
 272         }
 273         sock->pmask = 0;
 274         sock_reselect( sock );
 275     }
 276 }
 277
 278 inline static int sock_error(int s)
 279 {
 280     unsigned int optval = 0, optlen;
 281
 282     optlen = sizeof(optval);
 283     getsockopt(s, SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
 284     return optval ? sock_get_error(optval) : 0;
 285 }
 286
 287 static void sock_poll_event( struct object *obj, int event )
 288 {
 289     struct sock *sock = (struct sock *)obj;
 290     int hangup_seen = 0;
 291
 292     assert( sock->obj.ops == &sock_ops );
 293     if (debug_level)
 294         fprintf(stderr, "socket %d select event: %x\n", sock->obj.fd, event);
 295     if (sock->state & FD_CONNECT)
 296     {
 297         /* connecting */
 298         if (event & POLLOUT)
 299         {
 300             /* we got connected */
 301             sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 302             sock->state &= ~FD_CONNECT;
 303             sock->pmask |= FD_CONNECT;
 304             sock->errors[FD_CONNECT_BIT] = 0;
 305             if (debug_level)
 306                 fprintf(stderr, "socket %d connection success\n", sock->obj.fd);
 307         }
 308         else if (event & (POLLERR|POLLHUP))
 309         {
 310             /* we didn't get connected? */
 311             sock->state &= ~FD_CONNECT;
 312             sock->pmask |= FD_CONNECT;
 313             sock->errors[FD_CONNECT_BIT] = sock_error( sock->obj.fd );
 314             if (debug_level)
 315                 fprintf(stderr, "socket %d connection failure\n", sock->obj.fd);
 316         }
 317     } else
 318     if (sock->state & FD_WINE_LISTENING)
 319     {
 320         /* listening */
 321         if (event & POLLIN)
 322         {
 323             /* incoming connection */
 324             sock->pmask |= FD_ACCEPT;
 325             sock->errors[FD_ACCEPT_BIT] = 0;
 326             sock->hmask |= FD_ACCEPT;
 327         }
 328         else if (event & (POLLERR|POLLHUP))
 329         {
 330             /* failed incoming connection? */
 331             sock->pmask |= FD_ACCEPT;
 332             sock->errors[FD_ACCEPT_BIT] = sock_error( sock->obj.fd );
 333             sock->hmask |= FD_ACCEPT;
 334         }
 335     } else
 336     {
 337         /* normal data flow */
 338         if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
 339         {
 340             char dummy;
 341             int nr;
 342
 343             /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
 344              * has been closed, so we need to check for it explicitly here */
 345             nr  = recv( sock->obj.fd, &dummy, 1, MSG_PEEK );
 346             if ( nr > 0 )
 347             {
 348                 /* incoming data */
 349                 sock->pmask |= FD_READ;
 350                 sock->hmask |= (FD_READ|FD_CLOSE);
 351                 sock->errors[FD_READ_BIT] = 0;
 352                 if (debug_level)
 353                     fprintf(stderr, "socket %d is readable\n", sock->obj.fd );
 354             }
 355             else if ( nr == 0 )
 356                 hangup_seen = 1;
 357             else
 358             {
 359                 /* EAGAIN can happen if an async recv() falls between the server's poll()
 360                    call and the invocation of this routine */
 361                 if ( errno == EAGAIN )
 362                     event &= ~POLLIN;
 363                 else
 364                 {
 365                     if ( debug_level )
 366                         fprintf ( stderr, "recv error on socket %d: %d\n", sock->obj.fd, errno );
 367                     event = POLLERR;
 368                 }
 369             }
 370
 371         }
 372         else if ( sock_shutdown_type == SOCK_SHUTDOWN_POLLHUP && (event & POLLHUP) )
 373         {
 374             hangup_seen = 1;
 375         }
 376         else if ( event & POLLIN ) /* POLLIN for non-stream socket */
 377         {
 378             sock->pmask |= FD_READ;
 379             sock->hmask |= (FD_READ|FD_CLOSE);
 380             sock->errors[FD_READ_BIT] = 0;
 381             if (debug_level)
 382                 fprintf(stderr, "socket %d is readable\n", sock->obj.fd );
 383
 384         }
 385
 386         if (event & POLLOUT)
 387         {
 388             sock->pmask |= FD_WRITE;
 389             sock->hmask |= FD_WRITE;
 390             sock->errors[FD_WRITE_BIT] = 0;
 391             if (debug_level)
 392                 fprintf(stderr, "socket %d is writable\n", sock->obj.fd);
 393         }
 394         if (event & POLLPRI)
 395         {
 396             sock->pmask |= FD_OOB;
 397             sock->hmask |= FD_OOB;
 398             sock->errors[FD_OOB_BIT] = 0;
 399             if (debug_level)
 400                 fprintf(stderr, "socket %d got OOB data\n", sock->obj.fd);
 401         }
 402         /* According to WS2 specs, FD_CLOSE is only delivered when there is
 403            no more data to be read (i.e. hangup_seen = 1) */
 404         else if ( hangup_seen && (sock->state & (FD_READ|FD_WRITE) ))
 405         {
 406             sock->errors[FD_CLOSE_BIT] = sock_error( sock->obj.fd );
 407             if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
 408                 sock->state &= ~FD_WRITE;
 409             sock->pmask |= FD_CLOSE;
 410             sock->hmask |= FD_CLOSE;
 411             if (debug_level)
 412                 fprintf(stderr, "socket %d aborted by error %d, event: %x - removing from select loop\n",
 413                         sock->obj.fd, sock->errors[FD_CLOSE_BIT], event);
 414         }
 415     }
 416
 417     if ( sock->pmask & FD_CLOSE || event & (POLLERR|POLLHUP) )
 418     {
 419         if ( debug_level )
 420             fprintf ( stderr, "removing socket %d from select loop\n", sock->obj.fd );
 421         set_select_events( &sock->obj, -1 );
 422     }
 423     else
 424         sock_reselect( sock );
 425
 426     /* wake up anyone waiting for whatever just happened */
 427     if ( sock->pmask & sock->mask || sock->flags & FD_FLAG_OVERLAPPED ) sock_wake_up( sock, event );
 428
 429     /* if anyone is stupid enough to wait on the socket object itself,
 430      * maybe we should wake them up too, just in case? */
 431     wake_up( &sock->obj, 0 );
 432 }
 433
 434 static void sock_dump( struct object *obj, int verbose )
 435 {
 436     struct sock *sock = (struct sock *)obj;
 437     assert( obj->ops == &sock_ops );
 438     printf( "Socket fd=%d, state=%x, mask=%x, pending=%x, held=%x\n",
 439             sock->obj.fd, sock->state,
 440             sock->mask, sock->pmask, sock->hmask );
 441 }
 442
 443 static int sock_signaled( struct object *obj, struct thread *thread )
 444 {
 445     struct sock *sock = (struct sock *)obj;
 446     assert( obj->ops == &sock_ops );
 447
 448     return check_select_events( get_unix_fd(obj), sock_get_poll_events( &sock->obj ) );
 449 }
 450
 451 static int sock_get_poll_events( struct object *obj )
 452 {
 453     struct sock *sock = (struct sock *)obj;
 454     unsigned int mask = sock->mask & sock->state & ~sock->hmask;
 455     int ev = 0;
 456
 457     assert( obj->ops == &sock_ops );
 458
 459     if (sock->state & FD_CONNECT)
 460         /* connecting, wait for writable */
 461         return POLLOUT;
 462     if (sock->state & FD_WINE_LISTENING)
 463         /* listening, wait for readable */
 464         return (sock->hmask & FD_ACCEPT) ? 0 : POLLIN;
 465
 466     if (mask & (FD_READ) || (sock->flags & WSA_FLAG_OVERLAPPED && IS_READY (sock->read_q)))
 467         ev |= POLLIN | POLLPRI;
 468     if (mask & FD_WRITE || (sock->flags & WSA_FLAG_OVERLAPPED && IS_READY (sock->write_q)))
 469         ev |= POLLOUT;
 470     /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
 471     if ( sock->type == SOCK_STREAM && ( sock->mask & ~sock->hmask & FD_CLOSE) )
 472         ev |= POLLIN;
 473
 474     return ev;
 475 }
 476
 477 static int sock_get_info( struct object *obj, struct get_file_info_reply *reply, int *flags )
 478 {
 479     struct sock *sock = (struct sock*) obj;
 480     assert ( obj->ops == &sock_ops );
 481
 482     if (reply)
 483     {
 484         reply->type        = FILE_TYPE_PIPE;
 485         reply->attr        = 0;
 486         reply->access_time = 0;
 487         reply->write_time  = 0;
 488         reply->size_high   = 0;
 489         reply->size_low    = 0;
 490         reply->links       = 0;
 491         reply->index_high  = 0;
 492         reply->index_low   = 0;
 493         reply->serial      = 0;
 494     }
 495     *flags = 0;
 496     if (sock->flags & WSA_FLAG_OVERLAPPED) *flags |= FD_FLAG_OVERLAPPED;
 497     if ( sock->type != SOCK_STREAM || sock->state & FD_WINE_CONNECTED )
 498     {
 499         if ( !(sock->state & FD_READ  ) ) *flags |= FD_FLAG_RECV_SHUTDOWN;
 500         if ( !(sock->state & FD_WRITE ) ) *flags |= FD_FLAG_SEND_SHUTDOWN;
 501     }
 502     return FD_TYPE_SOCKET;
 503 }
 504
 505 static void sock_queue_async(struct object *obj, void *ptr, unsigned int status, int type, int count)
 506 {
 507     struct sock *sock = (struct sock *)obj;
 508     struct async_queue *q;
 509     struct async *async;
 510     int pollev;
 511
 512     assert( obj->ops == &sock_ops );
 513
 514     if ( !(sock->flags & WSA_FLAG_OVERLAPPED) )
 515     {
 516         set_error ( STATUS_INVALID_HANDLE );
 517         return;
 518     }
 519
 520     switch( type )
 521     {
 522     case ASYNC_TYPE_READ:
 523         q = &sock->read_q;
 524         sock->hmask &= ~FD_CLOSE;
 525         break;
 526     case ASYNC_TYPE_WRITE:
 527         q = &sock->write_q;
 528         break;
 529     default:
 530         set_error( STATUS_INVALID_PARAMETER );
 531         return;
 532     }
 533
 534     async = find_async ( q, current, ptr );
 535
 536     if ( status == STATUS_PENDING )
 537     {
 538         if ( ( !( sock->state & FD_READ ) && type == ASYNC_TYPE_READ  ) ||
 539              ( !( sock->state & FD_WRITE ) && type == ASYNC_TYPE_WRITE ) )
 540         {
 541             set_error ( STATUS_PIPE_DISCONNECTED );
 542             if ( async ) destroy_async ( async );
 543         }
 544         else
 545         {
 546             if ( !async )
 547                 async = create_async ( obj, current, ptr );
 548             if ( !async )
 549                 return;
 550
 551             async->status = STATUS_PENDING;
 552             if ( !async->q )
 553                 async_insert ( q, async );
 554         }
 555     }
 556     else if ( async ) destroy_async ( async );
 557     else set_error ( STATUS_INVALID_PARAMETER );
 558
 559     pollev = sock_reselect ( sock );
 560     if ( pollev ) sock_try_event ( sock, pollev );
 561 }
 562
 563 static void sock_destroy( struct object *obj )
 564 {
 565     struct sock *sock = (struct sock *)obj;
 566     assert( obj->ops == &sock_ops );
 567
 568     /* FIXME: special socket shutdown stuff? */
 569
 570     if ( sock->deferred )
 571         release_object ( sock->deferred );
 572
 573     if ( sock->flags & WSA_FLAG_OVERLAPPED )
 574     {
 575         destroy_async_queue ( &sock->read_q );
 576         destroy_async_queue ( &sock->write_q );
 577     }
 578     if (sock->event) release_object( sock->event );
 579 }
 580
 581 /* create a new and unconnected socket */
 582 static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
 583 {
 584     struct sock *sock;
 585     int sockfd;
 586
 587     sockfd = socket( family, type, protocol );
 588     if (debug_level)
 589         fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
 590     if (sockfd == -1) {
 591         sock_set_error();
 592         return NULL;
 593     }
 594     fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 595     if (!(sock = alloc_fd_object( &sock_ops, &sock_fd_ops, -1 ))) return NULL;
 596     set_unix_fd( &sock->obj, sockfd );
 597     sock->state = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
 598     sock->mask    = 0;
 599     sock->hmask   = 0;
 600     sock->pmask   = 0;
 601     sock->flags   = flags;
 602     sock->type    = type;
 603     sock->family  = family;
 604     sock->event   = NULL;
 605     sock->window  = 0;
 606     sock->message = 0;
 607     sock->wparam  = 0;
 608     sock->deferred = NULL;
 609     if (sock->flags & WSA_FLAG_OVERLAPPED)
 610     {
 611         init_async_queue (&sock->read_q);
 612         init_async_queue (&sock->write_q);
 613     }
 614     sock_reselect( sock );
 615     clear_error();
 616     return &sock->obj;
 617 }
 618
 619 /* accept a socket (creates a new fd) */
 620 static struct sock *accept_socket( obj_handle_t handle )
 621 {
 622     struct sock *acceptsock;
 623     struct sock *sock;
 624     int acceptfd;
 625     struct sockaddr     saddr;
 626     int                 slen;
 627
 628     sock=(struct sock*)get_handle_obj(current->process,handle,
 629                                       GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE,&sock_ops);
 630     if (!sock)
 631         return NULL;
 632
 633     if ( sock->deferred ) {
 634         acceptsock = sock->deferred;
 635         sock->deferred = NULL;
 636     } else {
 637
 638         /* Try to accept(2). We can't be safe that this an already connected socket
 639          * or that accept() is allowed on it. In those cases we will get -1/errno
 640          * return.
 641          */
 642         slen = sizeof(saddr);
 643         acceptfd = accept( get_unix_fd(&sock->obj), &saddr, &slen);
 644         if (acceptfd==-1) {
 645             sock_set_error();
 646             release_object( sock );
 647             return NULL;
 648         }
 649         if (!(acceptsock = alloc_fd_object( &sock_ops, &sock_fd_ops, acceptfd )))
 650         {
 651             release_object( sock );
 652             return NULL;
 653         }
 654
 655         /* newly created socket gets the same properties of the listening socket */
 656         fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
 657         acceptsock->state  = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
 658         if (sock->state & FD_WINE_NONBLOCKING)
 659             acceptsock->state |= FD_WINE_NONBLOCKING;
 660         acceptsock->mask    = sock->mask;
 661         acceptsock->hmask   = 0;
 662         acceptsock->pmask   = 0;
 663         acceptsock->type    = sock->type;
 664         acceptsock->family  = sock->family;
 665         acceptsock->event   = NULL;
 666         acceptsock->window  = sock->window;
 667         acceptsock->message = sock->message;
 668         acceptsock->wparam  = 0;
 669         if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
 670         acceptsock->flags = sock->flags;
 671         acceptsock->deferred = 0;
 672         if ( acceptsock->flags & WSA_FLAG_OVERLAPPED )
 673         {
 674             init_async_queue ( &acceptsock->read_q );
 675             init_async_queue ( &acceptsock->write_q );
 676         }
 677     }
 678     clear_error();
 679     sock->pmask &= ~FD_ACCEPT;
 680     sock->hmask &= ~FD_ACCEPT;
 681     sock_reselect( sock );
 682     release_object( sock );
 683     return acceptsock;
 684 }
 685
 686 /* set the last error depending on errno */
 687 static int sock_get_error( int err )
 688 {
 689     switch (err)
 690     {
 691         case EINTR:             return WSAEINTR; break;
 692         case EBADF:             return WSAEBADF; break;
 693         case EPERM:
 694         case EACCES:            return WSAEACCES; break;
 695         case EFAULT:            return WSAEFAULT; break;
 696         case EINVAL:            return WSAEINVAL; break;
 697         case EMFILE:            return WSAEMFILE; break;
 698         case EWOULDBLOCK:       return WSAEWOULDBLOCK; break;
 699         case EINPROGRESS:       return WSAEINPROGRESS; break;
 700         case EALREADY:          return WSAEALREADY; break;
 701         case ENOTSOCK:          return WSAENOTSOCK; break;
 702         case EDESTADDRREQ:      return WSAEDESTADDRREQ; break;
 703         case EMSGSIZE:          return WSAEMSGSIZE; break;
 704         case EPROTOTYPE:        return WSAEPROTOTYPE; break;
 705         case ENOPROTOOPT:       return WSAENOPROTOOPT; break;
 706         case EPROTONOSUPPORT:   return WSAEPROTONOSUPPORT; break;
 707         case ESOCKTNOSUPPORT:   return WSAESOCKTNOSUPPORT; break;
 708         case EOPNOTSUPP:        return WSAEOPNOTSUPP; break;
 709         case EPFNOSUPPORT:      return WSAEPFNOSUPPORT; break;
 710         case EAFNOSUPPORT:      return WSAEAFNOSUPPORT; break;
 711         case EADDRINUSE:        return WSAEADDRINUSE; break;
 712         case EADDRNOTAVAIL:     return WSAEADDRNOTAVAIL; break;
 713         case ENETDOWN:          return WSAENETDOWN; break;
 714         case ENETUNREACH:       return WSAENETUNREACH; break;
 715         case ENETRESET:         return WSAENETRESET; break;
 716         case ECONNABORTED:      return WSAECONNABORTED; break;
 717         case EPIPE:
 718         case ECONNRESET:        return WSAECONNRESET; break;
 719         case ENOBUFS:           return WSAENOBUFS; break;
 720         case EISCONN:           return WSAEISCONN; break;
 721         case ENOTCONN:          return WSAENOTCONN; break;
 722         case ESHUTDOWN:         return WSAESHUTDOWN; break;
 723         case ETOOMANYREFS:      return WSAETOOMANYREFS; break;
 724         case ETIMEDOUT:         return WSAETIMEDOUT; break;
 725         case ECONNREFUSED:      return WSAECONNREFUSED; break;
 726         case ELOOP:             return WSAELOOP; break;
 727         case ENAMETOOLONG:      return WSAENAMETOOLONG; break;
 728         case EHOSTDOWN:         return WSAEHOSTDOWN; break;
 729         case EHOSTUNREACH:      return WSAEHOSTUNREACH; break;
 730         case ENOTEMPTY:         return WSAENOTEMPTY; break;
 731 #ifdef EPROCLIM
 732         case EPROCLIM:          return WSAEPROCLIM; break;
 733 #endif
 734 #ifdef EUSERS
 735         case EUSERS:            return WSAEUSERS; break;
 736 #endif
 737 #ifdef EDQUOT
 738         case EDQUOT:            return WSAEDQUOT; break;
 739 #endif
 740 #ifdef ESTALE
 741         case ESTALE:            return WSAESTALE; break;
 742 #endif
 743 #ifdef EREMOTE
 744         case EREMOTE:           return WSAEREMOTE; break;
 745 #endif
 746     default: errno=err; perror("sock_set_error"); return ERROR_UNKNOWN; break;
 747     }
 748 }
 749
 750 /* set the last error depending on errno */
 751 static void sock_set_error(void)
 752 {
 753     set_error( sock_get_error( errno ) );
 754 }
 755
 756 /* create a socket */
 757 DECL_HANDLER(create_socket)
 758 {
 759     struct object *obj;
 760
 761     reply->handle = 0;
 762     if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
 763     {
 764         reply->handle = alloc_handle( current->process, obj, req->access, req->inherit );
 765         release_object( obj );
 766     }
 767 }
 768
 769 /* accept a socket */
 770 DECL_HANDLER(accept_socket)
 771 {
 772     struct sock *sock;
 773
 774     reply->handle = 0;
 775     if ((sock = accept_socket( req->lhandle )) != NULL)
 776     {
 777         reply->handle = alloc_handle( current->process, &sock->obj, req->access, req->inherit );
 778         sock->wparam = reply->handle;  /* wparam for message is the socket handle */
 779         sock_reselect( sock );
 780         release_object( &sock->obj );
 781     }
 782 }
 783
 784 /* set socket event parameters */
 785 DECL_HANDLER(set_socket_event)
 786 {
 787     struct sock *sock;
 788     struct event *old_event;
 789     int pollev;
 790
 791     if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
 792                                                GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE, &sock_ops)))
 793         return;
 794     old_event = sock->event;
 795     sock->mask    = req->mask;
 796     sock->event   = NULL;
 797     sock->window  = req->window;
 798     sock->message = req->msg;
 799     sock->wparam  = req->handle;  /* wparam is the socket handle */
 800     if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
 801
 802     if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
 803
 804     pollev = sock_reselect( sock );
 805     if ( pollev ) sock_try_event ( sock, pollev );
 806
 807     if (sock->mask)
 808         sock->state |= FD_WINE_NONBLOCKING;
 809
 810     /* if a network event is pending, signal the event object
 811        it is possible that FD_CONNECT or FD_ACCEPT network events has happened
 812        before a WSAEventSelect() was done on it.
 813        (when dealing with Asynchronous socket)  */
 814     if (sock->pmask & sock->mask) sock_wake_up( sock, pollev );
 815
 816     if (old_event) release_object( old_event ); /* we're through with it */
 817     release_object( &sock->obj );
 818 }
 819
 820 /* get socket event parameters */
 821 DECL_HANDLER(get_socket_event)
 822 {
 823     struct sock *sock;
 824
 825     sock=(struct sock*)get_handle_obj(current->process,req->handle,GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE,&sock_ops);
 826     if (!sock)
 827     {
 828         reply->mask  = 0;
 829         reply->pmask = 0;
 830         reply->state = 0;
 831         set_error( WSAENOTSOCK );
 832         return;
 833     }
 834     reply->mask  = sock->mask;
 835     reply->pmask = sock->pmask;
 836     reply->state = sock->state;
 837     set_reply_data( sock->errors, min( get_reply_max_size(), sizeof(sock->errors) ));
 838
 839     if (req->service)
 840     {
 841         if (req->c_event)
 842         {
 843             struct event *cevent = get_event_obj( current->process, req->c_event,
 844                                                   EVENT_MODIFY_STATE );
 845             if (cevent)
 846             {
 847                 reset_event( cevent );
 848                 release_object( cevent );
 849             }
 850         }
 851         sock->pmask = 0;
 852         sock_reselect( sock );
 853     }
 854     release_object( &sock->obj );
 855 }
 856
 857 /* re-enable pending socket events */
 858 DECL_HANDLER(enable_socket_event)
 859 {
 860     struct sock *sock;
 861     int pollev;
 862
 863     if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
 864                                                GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE, &sock_ops)))
 865         return;
 866
 867     sock->pmask &= ~req->mask; /* is this safe? */
 868     sock->hmask &= ~req->mask;
 869     if ( req->mask & FD_READ )
 870         sock->hmask &= ~FD_CLOSE;
 871     sock->state |= req->sstate;
 872     sock->state &= ~req->cstate;
 873     if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
 874
 875     pollev = sock_reselect( sock );
 876     if ( pollev ) sock_try_event ( sock, pollev );
 877
 878     release_object( &sock->obj );
 879 }
 880
 881 DECL_HANDLER(set_socket_deferred)
 882 {
 883     struct sock *sock, *acceptsock;
 884
 885     sock=(struct sock*)get_handle_obj( current->process,req->handle,
 886                                        GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE,&sock_ops );
 887     if ( !sock )
 888     {
 889         set_error ( WSAENOTSOCK );
 890         return;
 891     }
 892     acceptsock = (struct sock*)get_handle_obj( current->process,req->deferred,
 893                                                GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE,&sock_ops );
 894     if ( !acceptsock )
 895     {
 896         release_object ( sock );
 897         set_error ( WSAENOTSOCK );
 898         return;
 899     }
 900     sock->deferred = acceptsock;
 901     release_object ( sock );
 902 }