2 * Server-side socket management
4 * Copyright (C) 1999 Marcus Meissner, Ove Kåven
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.
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.
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
20 * FIXME: we use read|write access in all cases. Shouldn't we depend that
21 * on the access of the current handle?
33 #ifdef HAVE_SYS_ERRNO_H
34 # include <sys/errno.h>
37 #include <sys/types.h>
38 #ifdef HAVE_SYS_SOCKET_H
39 # include <sys/socket.h>
41 #ifdef HAVE_SYS_IOCTL_H
42 #include <sys/ioctl.h>
44 #ifdef HAVE_SYS_FILIO_H
45 # include <sys/filio.h>
51 #define WIN32_NO_STATUS
64 #define FD_MAX_EVENTS 10
66 #define FD_WRITE_BIT 1
68 #define FD_ACCEPT_BIT 3
69 #define FD_CONNECT_BIT 4
70 #define FD_CLOSE_BIT 5
73 * Define flags to be used with the WSAAsyncSelect() call.
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
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
89 /* Constants for WSAIoctl() */
90 #define WSA_FLAG_OVERLAPPED 0x01
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 */
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 );
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 void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
123 static void sock_reselect_async( struct fd *fd, struct async_queue *queue );
124 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb );
126 static int sock_get_ntstatus( int err );
127 static int sock_get_error( int err );
128 static void sock_set_error(void);
130 static const struct object_ops sock_ops =
132 sizeof(struct sock), /* size */
133 sock_dump, /* dump */
134 no_get_type, /* get_type */
135 add_queue, /* add_queue */
136 remove_queue, /* remove_queue */
137 sock_signaled, /* signaled */
138 no_satisfied, /* satisfied */
139 no_signal, /* signal */
140 sock_get_fd, /* get_fd */
141 default_fd_map_access, /* map_access */
142 default_get_sd, /* get_sd */
143 default_set_sd, /* set_sd */
144 no_lookup_name, /* lookup_name */
145 no_open_file, /* open_file */
146 fd_close_handle, /* close_handle */
147 sock_destroy /* destroy */
150 static const struct fd_ops sock_fd_ops =
152 sock_get_poll_events, /* get_poll_events */
153 sock_poll_event, /* poll_event */
154 no_flush, /* flush */
155 sock_get_fd_type, /* get_fd_type */
156 default_fd_ioctl, /* ioctl */
157 sock_queue_async, /* queue_async */
158 sock_reselect_async, /* reselect_async */
159 sock_cancel_async /* cancel_async */
163 /* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
164 * we post messages if there are multiple events. Used to send
165 * messages. The problem is if there is both a FD_CONNECT event and,
166 * say, an FD_READ event available on the same socket, we want to
167 * notify the app of the connect event first. Otherwise it may
168 * discard the read event because it thinks it hasn't connected yet.
170 static const int event_bitorder[FD_MAX_EVENTS] =
178 6, 7, 8, 9 /* leftovers */
181 /* Flags that make sense only for SOCK_STREAM sockets */
182 #define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
185 SOCK_SHUTDOWN_ERROR = -1,
186 SOCK_SHUTDOWN_EOF = 0,
187 SOCK_SHUTDOWN_POLLHUP = 1
190 static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
192 static sock_shutdown_t sock_check_pollhup(void)
194 sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
199 if ( socketpair( AF_UNIX, SOCK_STREAM, 0, fd ) ) goto out;
200 if ( shutdown( fd[0], 1 ) ) goto out;
206 n = poll( &pfd, 1, 0 );
207 if ( n != 1 ) goto out; /* error or timeout */
208 if ( pfd.revents & POLLHUP )
209 ret = SOCK_SHUTDOWN_POLLHUP;
210 else if ( pfd.revents & POLLIN &&
211 read( fd[1], &dummy, 1 ) == 0 )
212 ret = SOCK_SHUTDOWN_EOF;
222 sock_shutdown_type = sock_check_pollhup();
224 switch ( sock_shutdown_type )
226 case SOCK_SHUTDOWN_EOF:
227 if (debug_level) fprintf( stderr, "sock_init: shutdown() causes EOF\n" );
229 case SOCK_SHUTDOWN_POLLHUP:
230 if (debug_level) fprintf( stderr, "sock_init: shutdown() causes POLLHUP\n" );
233 fprintf( stderr, "sock_init: ERROR in sock_check_pollhup()\n" );
234 sock_shutdown_type = SOCK_SHUTDOWN_EOF;
238 static int sock_reselect( struct sock *sock )
240 int ev = sock_get_poll_events( sock->fd );
243 fprintf(stderr,"sock_reselect(%p): new mask %x\n", sock, ev);
245 if (!sock->polling) /* FIXME: should find a better way to do this */
247 /* previously unconnected socket, is this reselect supposed to connect it? */
248 if (!(sock->state & ~FD_WINE_NONBLOCKING)) return 0;
249 /* ok, it is, attach it to the wineserver's main poll loop */
251 allow_fd_caching( sock->fd );
253 /* update condition mask */
254 set_fd_events( sock->fd, ev );
258 /* wake anybody waiting on the socket event or send the associated message */
259 static void sock_wake_up( struct sock *sock )
261 unsigned int events = sock->pmask & sock->mask;
264 if ( !events ) return;
268 if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
269 set_event( sock->event );
273 if (debug_level) fprintf(stderr, "signalling events %x win %08x\n", events, sock->window );
274 for (i = 0; i < FD_MAX_EVENTS; i++)
276 int event = event_bitorder[i];
277 if (sock->pmask & (1 << event))
279 lparam_t lparam = (1 << event) | (sock_get_error(sock->errors[event]) << 16);
280 post_message( sock->window, sock->message, sock->wparam, lparam );
284 sock_reselect( sock );
288 static inline int sock_error( struct fd *fd )
290 unsigned int optval = 0, optlen;
292 optlen = sizeof(optval);
293 getsockopt( get_unix_fd(fd), SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
297 static int sock_dispatch_asyncs( struct sock *sock, int event, int error )
299 if ( sock->flags & WSA_FLAG_OVERLAPPED )
301 if ( event & (POLLIN|POLLPRI) && async_waiting( sock->read_q ) )
303 if (debug_level) fprintf( stderr, "activating read queue for socket %p\n", sock );
304 async_wake_up( sock->read_q, STATUS_ALERTED );
305 event &= ~(POLLIN|POLLPRI);
307 if ( event & POLLOUT && async_waiting( sock->write_q ) )
309 if (debug_level) fprintf( stderr, "activating write queue for socket %p\n", sock );
310 async_wake_up( sock->write_q, STATUS_ALERTED );
313 if ( event & (POLLERR|POLLHUP) )
315 int status = sock_get_ntstatus( error );
317 if ( !(sock->state & FD_READ) )
318 async_wake_up( sock->read_q, status );
319 if ( !(sock->state & FD_WRITE) )
320 async_wake_up( sock->write_q, status );
326 static void sock_dispatch_events( struct sock *sock, int prevstate, int event, int error )
328 if (prevstate & FD_CONNECT)
330 sock->pmask |= FD_CONNECT;
331 sock->hmask |= FD_CONNECT;
332 sock->errors[FD_CONNECT_BIT] = error;
335 if (prevstate & FD_WINE_LISTENING)
337 sock->pmask |= FD_ACCEPT;
338 sock->hmask |= FD_ACCEPT;
339 sock->errors[FD_ACCEPT_BIT] = error;
345 sock->pmask |= FD_READ;
346 sock->hmask |= FD_READ;
347 sock->errors[FD_READ_BIT] = 0;
352 sock->pmask |= FD_WRITE;
353 sock->hmask |= FD_WRITE;
354 sock->errors[FD_WRITE_BIT] = 0;
359 sock->pmask |= FD_OOB;
360 sock->hmask |= FD_OOB;
361 sock->errors[FD_OOB_BIT] = 0;
364 if (event & (POLLERR|POLLHUP))
366 sock->pmask |= FD_CLOSE;
367 sock->hmask |= FD_CLOSE;
368 sock->errors[FD_CLOSE_BIT] = error;
371 sock_wake_up( sock );
374 static void sock_poll_event( struct fd *fd, int event )
376 struct sock *sock = get_fd_user( fd );
378 int prevstate = sock->state;
381 assert( sock->obj.ops == &sock_ops );
383 fprintf(stderr, "socket %p select event: %x\n", sock, event);
385 /* we may change event later, remove from loop here */
386 if (event & (POLLERR|POLLHUP)) set_fd_events( sock->fd, -1 );
388 if (sock->state & FD_CONNECT)
393 /* we got connected */
394 sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
395 sock->state &= ~FD_CONNECT;
397 else if (event & (POLLERR|POLLHUP))
399 /* we didn't get connected? */
400 sock->state &= ~FD_CONNECT;
401 error = sock_error( fd );
404 else if (sock->state & FD_WINE_LISTENING)
407 if (event & (POLLERR|POLLHUP))
408 error = sock_error( fd );
412 /* normal data flow */
413 if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
418 /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
419 * has been closed, so we need to check for it explicitly here */
420 nr = recv( get_unix_fd( fd ), &dummy, 1, MSG_PEEK );
429 /* EAGAIN can happen if an async recv() falls between the server's poll()
430 call and the invocation of this routine */
431 if ( errno != EAGAIN )
436 fprintf( stderr, "recv error on socket %p: %d\n", sock, errno );
441 if ( (hangup_seen || event & (POLLHUP|POLLERR)) && (sock->state & (FD_READ|FD_WRITE)) )
443 error = error ? error : sock_error( fd );
444 if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
445 sock->state &= ~FD_WRITE;
446 sock->state &= ~FD_READ;
449 fprintf(stderr, "socket %p aborted by error %d, event: %x\n", sock, error, event);
456 event = sock_dispatch_asyncs( sock, event, error );
457 sock_dispatch_events( sock, prevstate, event, error );
459 /* if anyone is stupid enough to wait on the socket object itself,
460 * maybe we should wake them up too, just in case? */
461 wake_up( &sock->obj, 0 );
463 sock_reselect( sock );
466 static void sock_dump( struct object *obj, int verbose )
468 struct sock *sock = (struct sock *)obj;
469 assert( obj->ops == &sock_ops );
470 printf( "Socket fd=%p, state=%x, mask=%x, pending=%x, held=%x\n",
471 sock->fd, sock->state,
472 sock->mask, sock->pmask, sock->hmask );
475 static int sock_signaled( struct object *obj, struct thread *thread )
477 struct sock *sock = (struct sock *)obj;
478 assert( obj->ops == &sock_ops );
480 return check_fd_events( sock->fd, sock_get_poll_events( sock->fd ) ) != 0;
483 static int sock_get_poll_events( struct fd *fd )
485 struct sock *sock = get_fd_user( fd );
486 unsigned int mask = sock->mask & ~sock->hmask;
487 unsigned int smask = sock->state & mask;
490 assert( sock->obj.ops == &sock_ops );
492 if (sock->state & FD_CONNECT)
493 /* connecting, wait for writable */
495 if (sock->state & FD_WINE_LISTENING)
496 /* listening, wait for readable */
497 return (mask & FD_ACCEPT) ? POLLIN : 0;
499 if ( async_queued( sock->read_q ) )
501 if ( async_waiting( sock->read_q ) ) ev |= POLLIN | POLLPRI;
503 else if (smask & FD_READ)
504 ev |= POLLIN | POLLPRI;
505 /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
506 else if ( sock->type == SOCK_STREAM && sock->state & FD_READ && mask & FD_CLOSE &&
507 !(sock->hmask & FD_READ) )
510 if ( async_queued( sock->write_q ) )
512 if ( async_waiting( sock->write_q ) ) ev |= POLLOUT;
514 else if (smask & FD_WRITE)
520 static enum server_fd_type sock_get_fd_type( struct fd *fd )
522 return FD_TYPE_SOCKET;
525 static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
527 struct sock *sock = get_fd_user( fd );
529 struct async_queue *queue;
531 assert( sock->obj.ops == &sock_ops );
535 case ASYNC_TYPE_READ:
536 if (!sock->read_q && !(sock->read_q = create_async_queue( sock->fd ))) return;
537 queue = sock->read_q;
539 case ASYNC_TYPE_WRITE:
540 if (!sock->write_q && !(sock->write_q = create_async_queue( sock->fd ))) return;
541 queue = sock->write_q;
544 set_error( STATUS_INVALID_PARAMETER );
548 if ( ( !( sock->state & (FD_READ|FD_CONNECT) ) && type == ASYNC_TYPE_READ ) ||
549 ( !( sock->state & (FD_WRITE|FD_CONNECT) ) && type == ASYNC_TYPE_WRITE ) )
551 set_error( STATUS_PIPE_DISCONNECTED );
555 if (!(async = create_async( current, queue, data ))) return;
556 release_object( async );
558 sock_reselect( sock );
560 set_error( STATUS_PENDING );
563 static void sock_reselect_async( struct fd *fd, struct async_queue *queue )
565 struct sock *sock = get_fd_user( fd );
566 sock_reselect( sock );
569 static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb )
571 struct sock *sock = get_fd_user( fd );
573 assert( sock->obj.ops == &sock_ops );
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 );
578 set_error( STATUS_NOT_FOUND );
581 static struct fd *sock_get_fd( struct object *obj )
583 struct sock *sock = (struct sock *)obj;
584 return (struct fd *)grab_object( sock->fd );
587 static void sock_destroy( struct object *obj )
589 struct sock *sock = (struct sock *)obj;
590 assert( obj->ops == &sock_ops );
592 /* FIXME: special socket shutdown stuff? */
594 if ( sock->deferred )
595 release_object( sock->deferred );
597 free_async_queue( sock->read_q );
598 free_async_queue( sock->write_q );
599 if (sock->event) release_object( sock->event );
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 );
608 static void init_sock(struct sock *sock)
622 sock->deferred = NULL;
624 sock->write_q = NULL;
625 memset( sock->errors, 0, sizeof(sock->errors) );
628 /* create a new and unconnected socket */
629 static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
634 sockfd = socket( family, type, protocol );
636 fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
642 fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
643 if (!(sock = alloc_object( &sock_ops )))
649 sock->state = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
652 sock->family = family;
654 if (!(sock->fd = create_anonymous_fd( &sock_fd_ops, sockfd, &sock->obj,
655 (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT )))
657 release_object( sock );
660 sock_reselect( sock );
665 /* accepts a socket and inits it */
666 static int accept_new_fd( struct sock *sock )
669 /* Try to accept(2). We can't be safe that this an already connected socket
670 * or that accept() is allowed on it. In those cases we will get -1/errno
674 struct sockaddr saddr;
675 unsigned int slen = sizeof(saddr);
676 acceptfd = accept( get_unix_fd(sock->fd), &saddr, &slen);
683 fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
687 /* accept a socket (creates a new fd) */
688 static struct sock *accept_socket( obj_handle_t handle )
690 struct sock *acceptsock;
694 sock = (struct sock *)get_handle_obj( current->process, handle, FILE_READ_DATA, &sock_ops );
698 if ( sock->deferred )
700 acceptsock = sock->deferred;
701 sock->deferred = NULL;
705 if ((acceptfd = accept_new_fd( sock )) == -1)
707 release_object( sock );
710 if (!(acceptsock = alloc_object( &sock_ops )))
713 release_object( sock );
717 init_sock( acceptsock );
718 /* newly created socket gets the same properties of the listening socket */
719 acceptsock->state = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
720 if (sock->state & FD_WINE_NONBLOCKING)
721 acceptsock->state |= FD_WINE_NONBLOCKING;
722 acceptsock->mask = sock->mask;
723 acceptsock->type = sock->type;
724 acceptsock->family = sock->family;
725 acceptsock->window = sock->window;
726 acceptsock->message = sock->message;
727 if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
728 acceptsock->flags = sock->flags;
729 if (!(acceptsock->fd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
730 get_fd_options( sock->fd ) )))
732 release_object( acceptsock );
733 release_object( sock );
738 sock->pmask &= ~FD_ACCEPT;
739 sock->hmask &= ~FD_ACCEPT;
740 sock_reselect( sock );
741 release_object( sock );
745 static int accept_into_socket( struct sock *sock, struct sock *acceptsock )
749 if ( sock->deferred )
751 newfd = dup_fd_object( sock->deferred->fd, 0, 0,
752 get_fd_options( acceptsock->fd ) );
756 set_fd_user( newfd, &sock_fd_ops, &acceptsock->obj );
758 release_object( sock->deferred );
759 sock->deferred = NULL;
763 if ((acceptfd = accept_new_fd( sock )) == -1)
766 if (!(newfd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
767 get_fd_options( acceptsock->fd ) )))
774 acceptsock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
775 acceptsock->hmask = 0;
776 acceptsock->pmask = 0;
777 acceptsock->polling = 0;
778 acceptsock->type = sock->type;
779 acceptsock->family = sock->family;
780 acceptsock->wparam = 0;
781 acceptsock->deferred = NULL;
782 if (acceptsock->fd) release_object( acceptsock->fd );
783 acceptsock->fd = newfd;
786 sock->pmask &= ~FD_ACCEPT;
787 sock->hmask &= ~FD_ACCEPT;
788 sock_reselect( sock );
793 /* set the last error depending on errno */
794 static int sock_get_error( int err )
798 case EINTR: return WSAEINTR;
799 case EBADF: return WSAEBADF;
801 case EACCES: return WSAEACCES;
802 case EFAULT: return WSAEFAULT;
803 case EINVAL: return WSAEINVAL;
804 case EMFILE: return WSAEMFILE;
805 case EWOULDBLOCK: return WSAEWOULDBLOCK;
806 case EINPROGRESS: return WSAEINPROGRESS;
807 case EALREADY: return WSAEALREADY;
808 case ENOTSOCK: return WSAENOTSOCK;
809 case EDESTADDRREQ: return WSAEDESTADDRREQ;
810 case EMSGSIZE: return WSAEMSGSIZE;
811 case EPROTOTYPE: return WSAEPROTOTYPE;
812 case ENOPROTOOPT: return WSAENOPROTOOPT;
813 case EPROTONOSUPPORT: return WSAEPROTONOSUPPORT;
814 case ESOCKTNOSUPPORT: return WSAESOCKTNOSUPPORT;
815 case EOPNOTSUPP: return WSAEOPNOTSUPP;
816 case EPFNOSUPPORT: return WSAEPFNOSUPPORT;
817 case EAFNOSUPPORT: return WSAEAFNOSUPPORT;
818 case EADDRINUSE: return WSAEADDRINUSE;
819 case EADDRNOTAVAIL: return WSAEADDRNOTAVAIL;
820 case ENETDOWN: return WSAENETDOWN;
821 case ENETUNREACH: return WSAENETUNREACH;
822 case ENETRESET: return WSAENETRESET;
823 case ECONNABORTED: return WSAECONNABORTED;
825 case ECONNRESET: return WSAECONNRESET;
826 case ENOBUFS: return WSAENOBUFS;
827 case EISCONN: return WSAEISCONN;
828 case ENOTCONN: return WSAENOTCONN;
829 case ESHUTDOWN: return WSAESHUTDOWN;
830 case ETOOMANYREFS: return WSAETOOMANYREFS;
831 case ETIMEDOUT: return WSAETIMEDOUT;
832 case ECONNREFUSED: return WSAECONNREFUSED;
833 case ELOOP: return WSAELOOP;
834 case ENAMETOOLONG: return WSAENAMETOOLONG;
835 case EHOSTDOWN: return WSAEHOSTDOWN;
836 case EHOSTUNREACH: return WSAEHOSTUNREACH;
837 case ENOTEMPTY: return WSAENOTEMPTY;
839 case EPROCLIM: return WSAEPROCLIM;
842 case EUSERS: return WSAEUSERS;
845 case EDQUOT: return WSAEDQUOT;
848 case ESTALE: return WSAESTALE;
851 case EREMOTE: return WSAEREMOTE;
857 perror("wineserver: sock_get_error() can't map error");
862 static int sock_get_ntstatus( int err )
866 case EBADF: return STATUS_INVALID_HANDLE;
867 case EBUSY: return STATUS_DEVICE_BUSY;
869 case EACCES: return STATUS_ACCESS_DENIED;
870 case EFAULT: return STATUS_NO_MEMORY;
871 case EINVAL: return STATUS_INVALID_PARAMETER;
873 case EMFILE: return STATUS_TOO_MANY_OPENED_FILES;
874 case EWOULDBLOCK: return STATUS_CANT_WAIT;
875 case EINPROGRESS: return STATUS_PENDING;
876 case EALREADY: return STATUS_NETWORK_BUSY;
877 case ENOTSOCK: return STATUS_OBJECT_TYPE_MISMATCH;
878 case EDESTADDRREQ: return STATUS_INVALID_PARAMETER;
879 case EMSGSIZE: return STATUS_BUFFER_OVERFLOW;
880 case EPROTONOSUPPORT:
881 case ESOCKTNOSUPPORT:
884 case EPROTOTYPE: return STATUS_NOT_SUPPORTED;
885 case ENOPROTOOPT: return STATUS_INVALID_PARAMETER;
886 case EOPNOTSUPP: return STATUS_NOT_SUPPORTED;
887 case EADDRINUSE: return STATUS_ADDRESS_ALREADY_ASSOCIATED;
888 case EADDRNOTAVAIL: return STATUS_INVALID_PARAMETER;
889 case ECONNREFUSED: return STATUS_CONNECTION_REFUSED;
890 case ESHUTDOWN: return STATUS_PIPE_DISCONNECTED;
891 case ENOTCONN: return STATUS_CONNECTION_DISCONNECTED;
892 case ETIMEDOUT: return STATUS_IO_TIMEOUT;
893 case ENETUNREACH: return STATUS_NETWORK_UNREACHABLE;
894 case EHOSTUNREACH: return STATUS_HOST_UNREACHABLE;
895 case ENETDOWN: return STATUS_NETWORK_BUSY;
897 case ECONNRESET: return STATUS_CONNECTION_RESET;
898 case ECONNABORTED: return STATUS_CONNECTION_ABORTED;
900 case 0: return STATUS_SUCCESS;
903 perror("wineserver: sock_get_ntstatus() can't map error");
904 return STATUS_UNSUCCESSFUL;
908 /* set the last error depending on errno */
909 static void sock_set_error(void)
911 set_error( sock_get_ntstatus( errno ) );
914 /* create a socket */
915 DECL_HANDLER(create_socket)
920 if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
922 reply->handle = alloc_handle( current->process, obj, req->access, req->attributes );
923 release_object( obj );
927 /* accept a socket */
928 DECL_HANDLER(accept_socket)
933 if ((sock = accept_socket( req->lhandle )) != NULL)
935 reply->handle = alloc_handle( current->process, &sock->obj, req->access, req->attributes );
936 sock->wparam = reply->handle; /* wparam for message is the socket handle */
937 sock_reselect( sock );
938 release_object( &sock->obj );
942 /* accept a socket into an initialized socket */
943 DECL_HANDLER(accept_into_socket)
945 struct sock *sock, *acceptsock;
946 const int all_attributes = FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES|FILE_READ_DATA;
948 if (!(sock = (struct sock *)get_handle_obj( current->process, req->lhandle,
949 all_attributes, &sock_ops)))
952 if (!(acceptsock = (struct sock *)get_handle_obj( current->process, req->ahandle,
953 all_attributes, &sock_ops)))
955 release_object( sock );
959 if (accept_into_socket( sock, acceptsock ))
961 acceptsock->wparam = req->ahandle; /* wparam for message is the socket handle */
962 sock_reselect( acceptsock );
964 release_object( acceptsock );
965 release_object( sock );
968 /* set socket event parameters */
969 DECL_HANDLER(set_socket_event)
972 struct event *old_event;
975 if (!(sock = (struct sock *)get_handle_obj( current->process, req->handle,
976 FILE_WRITE_ATTRIBUTES, &sock_ops))) return;
977 old_event = sock->event;
978 sock->mask = req->mask;
979 sock->hmask &= ~req->mask; /* re-enable held events */
981 sock->window = req->window;
982 sock->message = req->msg;
983 sock->wparam = req->handle; /* wparam is the socket handle */
984 if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
986 if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
988 pollev = sock_reselect( sock );
991 sock->state |= FD_WINE_NONBLOCKING;
993 /* if a network event is pending, signal the event object
994 it is possible that FD_CONNECT or FD_ACCEPT network events has happened
995 before a WSAEventSelect() was done on it.
996 (when dealing with Asynchronous socket) */
997 sock_wake_up( sock );
999 if (old_event) release_object( old_event ); /* we're through with it */
1000 release_object( &sock->obj );
1003 /* get socket event parameters */
1004 DECL_HANDLER(get_socket_event)
1008 int errors[FD_MAX_EVENTS];
1010 sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
1018 reply->mask = sock->mask;
1019 reply->pmask = sock->pmask;
1020 reply->state = sock->state;
1021 for (i = 0; i < FD_MAX_EVENTS; i++)
1022 errors[i] = sock_get_ntstatus(sock->errors[i]);
1024 set_reply_data( errors, min( get_reply_max_size(), sizeof(errors) ));
1030 struct event *cevent = get_event_obj( current->process, req->c_event,
1031 EVENT_MODIFY_STATE );
1034 reset_event( cevent );
1035 release_object( cevent );
1039 sock_reselect( sock );
1041 release_object( &sock->obj );
1044 /* re-enable pending socket events */
1045 DECL_HANDLER(enable_socket_event)
1049 if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
1050 FILE_WRITE_ATTRIBUTES, &sock_ops)))
1053 /* for event-based notification, windows erases stale events */
1054 sock->pmask &= ~req->mask;
1056 sock->hmask &= ~req->mask;
1057 sock->state |= req->sstate;
1058 sock->state &= ~req->cstate;
1059 if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
1061 sock_reselect( sock );
1063 release_object( &sock->obj );
1066 DECL_HANDLER(set_socket_deferred)
1068 struct sock *sock, *acceptsock;
1070 sock=(struct sock *)get_handle_obj( current->process, req->handle, FILE_WRITE_ATTRIBUTES, &sock_ops );
1074 acceptsock = (struct sock *)get_handle_obj( current->process, req->deferred, 0, &sock_ops );
1077 release_object( sock );
1080 sock->deferred = acceptsock;
1081 release_object( sock );