blob: 71d8ccb42c86b64f56de30aa8612ed458062167a [file] [log] [blame]
Ove Kaaven019211f1999-10-13 16:05:37 +00001/*
2 * Server-side socket management
3 *
Alexandre Julliard7d5f7322008-10-18 19:22:26 +02004 * Copyright (C) 1999 Marcus Meissner, Ove Kåven
Ove Kaaven019211f1999-10-13 16:05:37 +00005 *
Alexandre Julliard0799c1a2002-03-09 23:29:33 +00006 * 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
Jonathan Ernst360a3f92006-05-18 14:49:52 +020018 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
Alexandre Julliard0799c1a2002-03-09 23:29:33 +000019 *
Ove Kaaven019211f1999-10-13 16:05:37 +000020 * FIXME: we use read|write access in all cases. Shouldn't we depend that
21 * on the access of the current handle?
22 */
23
Patrik Stridvall96336321999-10-24 22:13:47 +000024#include "config.h"
25
Ove Kaaven019211f1999-10-13 16:05:37 +000026#include <assert.h>
27#include <fcntl.h>
Alexandre Julliarde37c6e12003-09-05 23:08:26 +000028#include <stdarg.h>
Ove Kaaven019211f1999-10-13 16:05:37 +000029#include <stdio.h>
30#include <string.h>
31#include <stdlib.h>
32#include <errno.h>
Patrik Stridvall96336321999-10-24 22:13:47 +000033#ifdef HAVE_SYS_ERRNO_H
34# include <sys/errno.h>
35#endif
Ove Kaaven019211f1999-10-13 16:05:37 +000036#include <sys/time.h>
37#include <sys/types.h>
Patrik Stridvall96336321999-10-24 22:13:47 +000038#ifdef HAVE_SYS_SOCKET_H
39# include <sys/socket.h>
40#endif
Steven Edwards037c8a12003-02-11 22:27:13 +000041#ifdef HAVE_SYS_IOCTL_H
Ove Kaaven019211f1999-10-13 16:05:37 +000042#include <sys/ioctl.h>
Steven Edwards037c8a12003-02-11 22:27:13 +000043#endif
Patrik Stridvall96336321999-10-24 22:13:47 +000044#ifdef HAVE_SYS_FILIO_H
45# include <sys/filio.h>
46#endif
Ove Kaaven019211f1999-10-13 16:05:37 +000047#include <time.h>
48#include <unistd.h>
49
Ge van Geldorp1a1583a2005-11-28 17:32:54 +010050#include "ntstatus.h"
51#define WIN32_NO_STATUS
Alexandre Julliarde37c6e12003-09-05 23:08:26 +000052#include "windef.h"
Ge van Geldorp1a1583a2005-11-28 17:32:54 +010053#include "winternl.h"
Alexandre Julliard863637b2003-01-30 00:26:44 +000054
Ove Kaaven019211f1999-10-13 16:05:37 +000055#include "process.h"
Alexandre Julliard863637b2003-01-30 00:26:44 +000056#include "file.h"
Ove Kaaven019211f1999-10-13 16:05:37 +000057#include "handle.h"
58#include "thread.h"
59#include "request.h"
Alexandre Julliard81f2a732002-03-23 20:43:52 +000060#include "user.h"
Ove Kaaven019211f1999-10-13 16:05:37 +000061
Francois Gouget27202312001-11-14 21:26:23 +000062/* To avoid conflicts with the Unix socket headers. Plus we only need a few
63 * macros anyway.
64 */
65#define USE_WS_PREFIX
66#include "winsock2.h"
67
Ove Kaaven019211f1999-10-13 16:05:37 +000068struct sock
69{
70 struct object obj; /* object header */
Alexandre Julliarde66207e2003-02-19 00:33:32 +000071 struct fd *fd; /* socket file descriptor */
Ove Kaaven019211f1999-10-13 16:05:37 +000072 unsigned int state; /* status bits */
73 unsigned int mask; /* event mask */
74 unsigned int hmask; /* held (blocked) events */
75 unsigned int pmask; /* pending events */
Martin Wilckaa477052002-01-09 21:16:24 +000076 unsigned int flags; /* socket flags */
Alexandre Julliarde66207e2003-02-19 00:33:32 +000077 int polling; /* is socket being polled? */
Martin Wilck15a0f0e2002-07-03 21:04:09 +000078 unsigned short type; /* socket type */
79 unsigned short family; /* socket family */
Ove Kaaven019211f1999-10-13 16:05:37 +000080 struct event *event; /* event object */
Alexandre Julliard81f2a732002-03-23 20:43:52 +000081 user_handle_t window; /* window to send the message to */
82 unsigned int message; /* message to send */
Alexandre Julliard73755972002-07-31 19:26:03 +000083 obj_handle_t wparam; /* message wparam (socket handle) */
Ove Kaaven019211f1999-10-13 16:05:37 +000084 int errors[FD_MAX_EVENTS]; /* event errors */
Eric Pouech46344472005-01-14 19:54:38 +000085 struct sock *deferred; /* socket that waits for a deferred accept */
Alexandre Julliarddf09ac52007-04-02 20:09:29 +020086 struct async_queue *read_q; /* queue for asynchronous reads */
87 struct async_queue *write_q; /* queue for asynchronous writes */
Ove Kaaven019211f1999-10-13 16:05:37 +000088};
89
90static void sock_dump( struct object *obj, int verbose );
Ove Kaaven019211f1999-10-13 16:05:37 +000091static int sock_signaled( struct object *obj, struct thread *thread );
Alexandre Julliarde66207e2003-02-19 00:33:32 +000092static struct fd *sock_get_fd( struct object *obj );
Ove Kaaven019211f1999-10-13 16:05:37 +000093static void sock_destroy( struct object *obj );
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +000094
95static int sock_get_poll_events( struct fd *fd );
96static void sock_poll_event( struct fd *fd, int event );
Alexandre Julliard7a9363a2007-04-10 22:26:23 +020097static enum server_fd_type sock_get_fd_type( struct fd *fd );
Alexandre Julliard111610c2007-03-20 20:21:12 +010098static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
Alexandre Julliard72bff2e2007-04-10 17:07:27 +020099static void sock_reselect_async( struct fd *fd, struct async_queue *queue );
Mike Kaplinskiyb05774e2009-08-09 00:08:27 -0400100static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb );
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000101
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000102static int sock_get_error( int err );
Ove Kaaven019211f1999-10-13 16:05:37 +0000103static void sock_set_error(void);
104
105static const struct object_ops sock_ops =
106{
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000107 sizeof(struct sock), /* size */
108 sock_dump, /* dump */
Alexandre Julliard8382eb02007-12-05 18:16:42 +0100109 no_get_type, /* get_type */
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000110 add_queue, /* add_queue */
111 remove_queue, /* remove_queue */
112 sock_signaled, /* signaled */
113 no_satisfied, /* satisfied */
Mike McCormackf92fff62005-04-24 17:35:52 +0000114 no_signal, /* signal */
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000115 sock_get_fd, /* get_fd */
Alexandre Julliard24001e82007-10-02 14:20:15 +0200116 default_fd_map_access, /* map_access */
Rob Shearmanc1707d82007-10-03 13:10:37 +0100117 default_get_sd, /* get_sd */
118 default_set_sd, /* set_sd */
Vitaliy Margolenbaffcb92005-11-22 14:55:42 +0000119 no_lookup_name, /* lookup_name */
Alexandre Julliard7e71c1d2007-03-22 11:44:29 +0100120 no_open_file, /* open_file */
Alexandre Julliard715d78e2006-11-02 20:52:22 +0100121 fd_close_handle, /* close_handle */
Alexandre Julliard863637b2003-01-30 00:26:44 +0000122 sock_destroy /* destroy */
123};
124
125static const struct fd_ops sock_fd_ops =
126{
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000127 sock_get_poll_events, /* get_poll_events */
128 sock_poll_event, /* poll_event */
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000129 no_flush, /* flush */
Alexandre Julliard7a9363a2007-04-10 22:26:23 +0200130 sock_get_fd_type, /* get_file_info */
Alexandre Julliard63571432007-04-16 14:45:03 +0200131 default_fd_ioctl, /* ioctl */
Eric Pouech46344472005-01-14 19:54:38 +0000132 sock_queue_async, /* queue_async */
Alexandre Julliard72bff2e2007-04-10 17:07:27 +0200133 sock_reselect_async, /* reselect_async */
Eric Pouech46344472005-01-14 19:54:38 +0000134 sock_cancel_async /* cancel_async */
Ove Kaaven019211f1999-10-13 16:05:37 +0000135};
136
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000137
138/* Permutation of 0..FD_MAX_EVENTS - 1 representing the order in which
139 * we post messages if there are multiple events. Used to send
140 * messages. The problem is if there is both a FD_CONNECT event and,
141 * say, an FD_READ event available on the same socket, we want to
142 * notify the app of the connect event first. Otherwise it may
143 * discard the read event because it thinks it hasn't connected yet.
144 */
145static const int event_bitorder[FD_MAX_EVENTS] =
146{
147 FD_CONNECT_BIT,
148 FD_ACCEPT_BIT,
149 FD_OOB_BIT,
150 FD_WRITE_BIT,
151 FD_READ_BIT,
152 FD_CLOSE_BIT,
153 6, 7, 8, 9 /* leftovers */
154};
155
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000156/* Flags that make sense only for SOCK_STREAM sockets */
157#define STREAM_FLAG_MASK ((unsigned int) (FD_CONNECT | FD_ACCEPT | FD_WINE_LISTENING | FD_WINE_CONNECTED))
158
159typedef enum {
160 SOCK_SHUTDOWN_ERROR = -1,
161 SOCK_SHUTDOWN_EOF = 0,
162 SOCK_SHUTDOWN_POLLHUP = 1
163} sock_shutdown_t;
164
165static sock_shutdown_t sock_shutdown_type = SOCK_SHUTDOWN_ERROR;
166
Eric Pouech46344472005-01-14 19:54:38 +0000167static sock_shutdown_t sock_check_pollhup(void)
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000168{
169 sock_shutdown_t ret = SOCK_SHUTDOWN_ERROR;
170 int fd[2], n;
171 struct pollfd pfd;
172 char dummy;
173
Eric Pouech46344472005-01-14 19:54:38 +0000174 if ( socketpair( AF_UNIX, SOCK_STREAM, 0, fd ) ) goto out;
175 if ( shutdown( fd[0], 1 ) ) goto out;
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000176
177 pfd.fd = fd[1];
178 pfd.events = POLLIN;
179 pfd.revents = 0;
180
Eric Pouech46344472005-01-14 19:54:38 +0000181 n = poll( &pfd, 1, 0 );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000182 if ( n != 1 ) goto out; /* error or timeout */
183 if ( pfd.revents & POLLHUP )
184 ret = SOCK_SHUTDOWN_POLLHUP;
185 else if ( pfd.revents & POLLIN &&
Eric Pouech46344472005-01-14 19:54:38 +0000186 read( fd[1], &dummy, 1 ) == 0 )
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000187 ret = SOCK_SHUTDOWN_EOF;
188
189out:
Eric Pouech46344472005-01-14 19:54:38 +0000190 close( fd[0] );
191 close( fd[1] );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000192 return ret;
193}
194
195void sock_init(void)
196{
Eric Pouech46344472005-01-14 19:54:38 +0000197 sock_shutdown_type = sock_check_pollhup();
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000198
199 switch ( sock_shutdown_type )
200 {
201 case SOCK_SHUTDOWN_EOF:
Eric Pouech46344472005-01-14 19:54:38 +0000202 if (debug_level) fprintf( stderr, "sock_init: shutdown() causes EOF\n" );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000203 break;
204 case SOCK_SHUTDOWN_POLLHUP:
Eric Pouech46344472005-01-14 19:54:38 +0000205 if (debug_level) fprintf( stderr, "sock_init: shutdown() causes POLLHUP\n" );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000206 break;
207 default:
Eric Pouech46344472005-01-14 19:54:38 +0000208 fprintf( stderr, "sock_init: ERROR in sock_check_pollhup()\n" );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000209 sock_shutdown_type = SOCK_SHUTDOWN_EOF;
210 }
211}
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000212
Martin Wilckbff7e692002-04-25 21:31:11 +0000213static int sock_reselect( struct sock *sock )
Ove Kaaven019211f1999-10-13 16:05:37 +0000214{
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000215 int ev = sock_get_poll_events( sock->fd );
Ove Kaaven77c30552000-03-25 21:41:43 +0000216
Ove Kaavenf45608f1999-10-23 16:53:34 +0000217 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000218 fprintf(stderr,"sock_reselect(%p): new mask %x\n", sock, ev);
Ove Kaaven7a0f96c2000-09-06 19:40:23 +0000219
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000220 if (!sock->polling) /* FIXME: should find a better way to do this */
221 {
Ove Kaaven7a0f96c2000-09-06 19:40:23 +0000222 /* previously unconnected socket, is this reselect supposed to connect it? */
Martin Wilckbff7e692002-04-25 21:31:11 +0000223 if (!(sock->state & ~FD_WINE_NONBLOCKING)) return 0;
Ove Kaaven7a0f96c2000-09-06 19:40:23 +0000224 /* ok, it is, attach it to the wineserver's main poll loop */
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000225 sock->polling = 1;
Ove Kaaven7a0f96c2000-09-06 19:40:23 +0000226 }
227 /* update condition mask */
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000228 set_fd_events( sock->fd, ev );
Martin Wilckbff7e692002-04-25 21:31:11 +0000229 return ev;
230}
231
232/* After POLLHUP is received, the socket will no longer be in the main select loop.
233 This function is used to signal pending events nevertheless */
Eric Pouech46344472005-01-14 19:54:38 +0000234static void sock_try_event( struct sock *sock, int event )
Martin Wilckbff7e692002-04-25 21:31:11 +0000235{
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000236 event = check_fd_events( sock->fd, event );
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000237 if (event)
Martin Wilckbff7e692002-04-25 21:31:11 +0000238 {
Eric Pouech46344472005-01-14 19:54:38 +0000239 if ( debug_level ) fprintf( stderr, "sock_try_event: %x\n", event );
240 sock_poll_event( sock->fd, event );
Martin Wilckbff7e692002-04-25 21:31:11 +0000241 }
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000242}
Ove Kaaven77c30552000-03-25 21:41:43 +0000243
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000244/* wake anybody waiting on the socket event or send the associated message */
Martin Wilckbff7e692002-04-25 21:31:11 +0000245static void sock_wake_up( struct sock *sock, int pollev )
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000246{
247 unsigned int events = sock->pmask & sock->mask;
248 int i;
Martin Wilckbff7e692002-04-25 21:31:11 +0000249 int async_active = 0;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000250
Alexandre Julliardf85437c2007-04-10 22:25:07 +0200251 if ( pollev & (POLLIN|POLLPRI) && async_waiting( sock->read_q ))
Martin Wilckbff7e692002-04-25 21:31:11 +0000252 {
Alexandre Julliardf85437c2007-04-10 22:25:07 +0200253 if (debug_level) fprintf( stderr, "activating read queue for socket %p\n", sock );
254 async_wake_up( sock->read_q, STATUS_ALERTED );
255 async_active = 1;
256 }
257 if ( pollev & POLLOUT && async_waiting( sock->write_q ))
258 {
259 if (debug_level) fprintf( stderr, "activating write queue for socket %p\n", sock );
260 async_wake_up( sock->write_q, STATUS_ALERTED );
261 async_active = 1;
Martin Wilckbff7e692002-04-25 21:31:11 +0000262 }
263
264 /* Do not signal events if there are still pending asynchronous IO requests */
265 /* We need this to delay FD_CLOSE events until all pending overlapped requests are processed */
266 if ( !events || async_active ) return;
267
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000268 if (sock->event)
269 {
270 if (debug_level) fprintf(stderr, "signalling events %x ptr %p\n", events, sock->event );
271 set_event( sock->event );
272 }
273 if (sock->window)
274 {
Alexandre Julliardd7641072008-12-08 16:57:38 +0100275 if (debug_level) fprintf(stderr, "signalling events %x win %08x\n", events, sock->window );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000276 for (i = 0; i < FD_MAX_EVENTS; i++)
277 {
278 int event = event_bitorder[i];
279 if (sock->pmask & (1 << event))
280 {
Alexandre Julliard31282b32008-12-10 16:01:50 +0100281 lparam_t lparam = (1 << event) | (sock->errors[event] << 16);
282 post_message( sock->window, sock->message, sock->wparam, lparam );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000283 }
284 }
285 sock->pmask = 0;
286 sock_reselect( sock );
287 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000288}
289
Andrew Talbotb1788c82007-03-17 10:52:14 +0000290static inline int sock_error( struct fd *fd )
Ove Kaaven019211f1999-10-13 16:05:37 +0000291{
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000292 unsigned int optval = 0, optlen;
Vincent Béron9a624912002-05-31 23:06:46 +0000293
Ove Kaaven019211f1999-10-13 16:05:37 +0000294 optlen = sizeof(optval);
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000295 getsockopt( get_unix_fd(fd), SOL_SOCKET, SO_ERROR, (void *) &optval, &optlen);
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000296 return optval ? sock_get_error(optval) : 0;
Ove Kaaven019211f1999-10-13 16:05:37 +0000297}
298
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000299static void sock_poll_event( struct fd *fd, int event )
Ove Kaaven019211f1999-10-13 16:05:37 +0000300{
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000301 struct sock *sock = get_fd_user( fd );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000302 int hangup_seen = 0;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000303
Ove Kaaven019211f1999-10-13 16:05:37 +0000304 assert( sock->obj.ops == &sock_ops );
Ove Kaavenf45608f1999-10-23 16:53:34 +0000305 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000306 fprintf(stderr, "socket %p select event: %x\n", sock, event);
Francois Gouget126e6b82001-08-24 19:14:56 +0000307 if (sock->state & FD_CONNECT)
Ove Kaaven019211f1999-10-13 16:05:37 +0000308 {
309 /* connecting */
Alexandre Julliard247b8ae1999-12-13 00:16:44 +0000310 if (event & POLLOUT)
Ove Kaaven019211f1999-10-13 16:05:37 +0000311 {
312 /* we got connected */
Francois Gouget126e6b82001-08-24 19:14:56 +0000313 sock->state |= FD_WINE_CONNECTED|FD_READ|FD_WRITE;
314 sock->state &= ~FD_CONNECT;
Ove Kaaven019211f1999-10-13 16:05:37 +0000315 sock->pmask |= FD_CONNECT;
316 sock->errors[FD_CONNECT_BIT] = 0;
Ove Kaavenf45608f1999-10-23 16:53:34 +0000317 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000318 fprintf(stderr, "socket %p connection success\n", sock);
Ove Kaaven019211f1999-10-13 16:05:37 +0000319 }
Alexandre Julliard247b8ae1999-12-13 00:16:44 +0000320 else if (event & (POLLERR|POLLHUP))
Ove Kaaven019211f1999-10-13 16:05:37 +0000321 {
322 /* we didn't get connected? */
Francois Gouget126e6b82001-08-24 19:14:56 +0000323 sock->state &= ~FD_CONNECT;
Ove Kaaven019211f1999-10-13 16:05:37 +0000324 sock->pmask |= FD_CONNECT;
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000325 sock->errors[FD_CONNECT_BIT] = sock_error( fd );
Ove Kaavenf45608f1999-10-23 16:53:34 +0000326 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000327 fprintf(stderr, "socket %p connection failure\n", sock);
Ove Kaaven019211f1999-10-13 16:05:37 +0000328 }
Eric Pouech46344472005-01-14 19:54:38 +0000329 }
330 else if (sock->state & FD_WINE_LISTENING)
Ove Kaaven019211f1999-10-13 16:05:37 +0000331 {
332 /* listening */
Alexandre Julliard247b8ae1999-12-13 00:16:44 +0000333 if (event & POLLIN)
Ove Kaaven019211f1999-10-13 16:05:37 +0000334 {
335 /* incoming connection */
336 sock->pmask |= FD_ACCEPT;
337 sock->errors[FD_ACCEPT_BIT] = 0;
338 sock->hmask |= FD_ACCEPT;
339 }
Alexandre Julliard247b8ae1999-12-13 00:16:44 +0000340 else if (event & (POLLERR|POLLHUP))
Ove Kaaven019211f1999-10-13 16:05:37 +0000341 {
342 /* failed incoming connection? */
343 sock->pmask |= FD_ACCEPT;
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000344 sock->errors[FD_ACCEPT_BIT] = sock_error( fd );
Ove Kaaven019211f1999-10-13 16:05:37 +0000345 sock->hmask |= FD_ACCEPT;
346 }
Eric Pouech46344472005-01-14 19:54:38 +0000347 }
348 else
Ove Kaaven019211f1999-10-13 16:05:37 +0000349 {
350 /* normal data flow */
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000351 if ( sock->type == SOCK_STREAM && ( event & POLLIN ) )
Ove Kaaven019211f1999-10-13 16:05:37 +0000352 {
Daniel Walkerc86517f2001-08-23 23:25:33 +0000353 char dummy;
Martin Wilckbff7e692002-04-25 21:31:11 +0000354 int nr;
Daniel Walkerc86517f2001-08-23 23:25:33 +0000355
356 /* Linux 2.4 doesn't report POLLHUP if only one side of the socket
357 * has been closed, so we need to check for it explicitly here */
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000358 nr = recv( get_unix_fd( fd ), &dummy, 1, MSG_PEEK );
Martin Wilckbff7e692002-04-25 21:31:11 +0000359 if ( nr > 0 )
Daniel Walkerc86517f2001-08-23 23:25:33 +0000360 {
361 /* incoming data */
362 sock->pmask |= FD_READ;
Martin Wilck75242f92002-04-27 01:06:09 +0000363 sock->hmask |= (FD_READ|FD_CLOSE);
Daniel Walkerc86517f2001-08-23 23:25:33 +0000364 sock->errors[FD_READ_BIT] = 0;
365 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000366 fprintf(stderr, "socket %p is readable\n", sock );
Daniel Walkerc86517f2001-08-23 23:25:33 +0000367 }
Martin Wilckbff7e692002-04-25 21:31:11 +0000368 else if ( nr == 0 )
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000369 hangup_seen = 1;
Martin Wilckbff7e692002-04-25 21:31:11 +0000370 else
371 {
372 /* EAGAIN can happen if an async recv() falls between the server's poll()
373 call and the invocation of this routine */
374 if ( errno == EAGAIN )
375 event &= ~POLLIN;
376 else
377 {
378 if ( debug_level )
Eric Pouech46344472005-01-14 19:54:38 +0000379 fprintf( stderr, "recv error on socket %p: %d\n", sock, errno );
Martin Wilckbff7e692002-04-25 21:31:11 +0000380 event = POLLERR;
381 }
382 }
383
Ove Kaaven019211f1999-10-13 16:05:37 +0000384 }
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000385 else if ( sock_shutdown_type == SOCK_SHUTDOWN_POLLHUP && (event & POLLHUP) )
386 {
387 hangup_seen = 1;
388 }
389 else if ( event & POLLIN ) /* POLLIN for non-stream socket */
390 {
391 sock->pmask |= FD_READ;
392 sock->hmask |= (FD_READ|FD_CLOSE);
393 sock->errors[FD_READ_BIT] = 0;
394 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000395 fprintf(stderr, "socket %p is readable\n", sock );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000396
397 }
Alexandre Julliardcf5188b2002-06-21 19:06:28 +0000398
Alexandre Julliard247b8ae1999-12-13 00:16:44 +0000399 if (event & POLLOUT)
Ove Kaaven019211f1999-10-13 16:05:37 +0000400 {
401 sock->pmask |= FD_WRITE;
402 sock->hmask |= FD_WRITE;
403 sock->errors[FD_WRITE_BIT] = 0;
Ove Kaavenf45608f1999-10-23 16:53:34 +0000404 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000405 fprintf(stderr, "socket %p is writable\n", sock);
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000406 }
407 if (event & POLLPRI)
408 {
409 sock->pmask |= FD_OOB;
410 sock->hmask |= FD_OOB;
Ove Kaaven77c30552000-03-25 21:41:43 +0000411 sock->errors[FD_OOB_BIT] = 0;
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000412 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000413 fprintf(stderr, "socket %p got OOB data\n", sock);
Ove Kaaven019211f1999-10-13 16:05:37 +0000414 }
Martin Wilckbff7e692002-04-25 21:31:11 +0000415 /* According to WS2 specs, FD_CLOSE is only delivered when there is
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000416 no more data to be read (i.e. hangup_seen = 1) */
417 else if ( hangup_seen && (sock->state & (FD_READ|FD_WRITE) ))
Martin Wilckbff7e692002-04-25 21:31:11 +0000418 {
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000419 sock->errors[FD_CLOSE_BIT] = sock_error( fd );
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000420 if ( (event & POLLERR) || ( sock_shutdown_type == SOCK_SHUTDOWN_EOF && (event & POLLHUP) ))
Martin Wilckc422dae2003-01-28 00:24:04 +0000421 sock->state &= ~FD_WRITE;
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000422 sock->pmask |= FD_CLOSE;
Martin Wilck75242f92002-04-27 01:06:09 +0000423 sock->hmask |= FD_CLOSE;
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000424 if (debug_level)
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000425 fprintf(stderr, "socket %p aborted by error %d, event: %x - removing from select loop\n",
426 sock, sock->errors[FD_CLOSE_BIT], event);
Ove Kaaven019211f1999-10-13 16:05:37 +0000427 }
428 }
429
Martin Wilckff1f3202002-04-26 18:31:19 +0000430 if ( sock->pmask & FD_CLOSE || event & (POLLERR|POLLHUP) )
431 {
432 if ( debug_level )
Eric Pouech46344472005-01-14 19:54:38 +0000433 fprintf( stderr, "removing socket %p from select loop\n", sock );
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000434 set_fd_events( sock->fd, -1 );
Martin Wilckff1f3202002-04-26 18:31:19 +0000435 }
436 else
437 sock_reselect( sock );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000438
Ove Kaaven019211f1999-10-13 16:05:37 +0000439 /* wake up anyone waiting for whatever just happened */
Eric Pouech1ffddb42004-08-17 23:37:55 +0000440 if ( sock->pmask & sock->mask || sock->flags & WSA_FLAG_OVERLAPPED ) sock_wake_up( sock, event );
Ove Kaaven019211f1999-10-13 16:05:37 +0000441
442 /* if anyone is stupid enough to wait on the socket object itself,
443 * maybe we should wake them up too, just in case? */
444 wake_up( &sock->obj, 0 );
445}
446
447static void sock_dump( struct object *obj, int verbose )
448{
449 struct sock *sock = (struct sock *)obj;
450 assert( obj->ops == &sock_ops );
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000451 printf( "Socket fd=%p, state=%x, mask=%x, pending=%x, held=%x\n",
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000452 sock->fd, sock->state,
Ove Kaavenf45608f1999-10-23 16:53:34 +0000453 sock->mask, sock->pmask, sock->hmask );
Ove Kaaven019211f1999-10-13 16:05:37 +0000454}
455
Ove Kaaven019211f1999-10-13 16:05:37 +0000456static int sock_signaled( struct object *obj, struct thread *thread )
457{
458 struct sock *sock = (struct sock *)obj;
459 assert( obj->ops == &sock_ops );
460
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000461 return check_fd_events( sock->fd, sock_get_poll_events( sock->fd ) ) != 0;
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000462}
463
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000464static int sock_get_poll_events( struct fd *fd )
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000465{
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000466 struct sock *sock = get_fd_user( fd );
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000467 unsigned int mask = sock->mask & sock->state & ~sock->hmask;
468 int ev = 0;
469
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000470 assert( sock->obj.ops == &sock_ops );
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000471
Francois Gouget126e6b82001-08-24 19:14:56 +0000472 if (sock->state & FD_CONNECT)
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000473 /* connecting, wait for writable */
474 return POLLOUT;
Francois Gouget126e6b82001-08-24 19:14:56 +0000475 if (sock->state & FD_WINE_LISTENING)
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000476 /* listening, wait for readable */
477 return (sock->hmask & FD_ACCEPT) ? 0 : POLLIN;
478
Alexandre Julliardf85437c2007-04-10 22:25:07 +0200479 if (mask & FD_READ || async_waiting( sock->read_q )) ev |= POLLIN | POLLPRI;
480 if (mask & FD_WRITE || async_waiting( sock->write_q )) ev |= POLLOUT;
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000481 /* We use POLLIN with 0 bytes recv() as FD_CLOSE indication for stream sockets. */
482 if ( sock->type == SOCK_STREAM && ( sock->mask & ~sock->hmask & FD_CLOSE) )
Martin Wilckbff7e692002-04-25 21:31:11 +0000483 ev |= POLLIN;
484
Alexandre Julliard1dca5e22000-01-01 00:56:27 +0000485 return ev;
Ove Kaaven019211f1999-10-13 16:05:37 +0000486}
487
Alexandre Julliard7a9363a2007-04-10 22:26:23 +0200488static enum server_fd_type sock_get_fd_type( struct fd *fd )
Alexandre Julliard646d6212001-10-05 19:45:45 +0000489{
Alexandre Julliard89304272006-11-20 14:14:04 +0100490 return FD_TYPE_SOCKET;
Alexandre Julliard646d6212001-10-05 19:45:45 +0000491}
492
Alexandre Julliard111610c2007-03-20 20:21:12 +0100493static void sock_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
Martin Wilckbff7e692002-04-25 21:31:11 +0000494{
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000495 struct sock *sock = get_fd_user( fd );
Alexandre Julliarddf09ac52007-04-02 20:09:29 +0200496 struct async_queue *queue;
Martin Wilckbff7e692002-04-25 21:31:11 +0000497 int pollev;
Alexandre Julliarddd81ac52005-02-24 17:06:31 +0000498
Alexandre Julliardcf27a7f2003-02-14 20:27:09 +0000499 assert( sock->obj.ops == &sock_ops );
Martin Wilckbff7e692002-04-25 21:31:11 +0000500
Eric Pouech46344472005-01-14 19:54:38 +0000501 switch (type)
Martin Wilckbff7e692002-04-25 21:31:11 +0000502 {
503 case ASYNC_TYPE_READ:
Alexandre Julliarddf09ac52007-04-02 20:09:29 +0200504 if (!sock->read_q && !(sock->read_q = create_async_queue( sock->fd ))) return;
505 queue = sock->read_q;
Martin Wilck75242f92002-04-27 01:06:09 +0000506 sock->hmask &= ~FD_CLOSE;
Martin Wilckbff7e692002-04-25 21:31:11 +0000507 break;
508 case ASYNC_TYPE_WRITE:
Alexandre Julliarddf09ac52007-04-02 20:09:29 +0200509 if (!sock->write_q && !(sock->write_q = create_async_queue( sock->fd ))) return;
510 queue = sock->write_q;
Martin Wilckbff7e692002-04-25 21:31:11 +0000511 break;
512 default:
513 set_error( STATUS_INVALID_PARAMETER );
514 return;
515 }
516
Eric Pouech46344472005-01-14 19:54:38 +0000517 if ( ( !( sock->state & FD_READ ) && type == ASYNC_TYPE_READ ) ||
518 ( !( sock->state & FD_WRITE ) && type == ASYNC_TYPE_WRITE ) )
Martin Wilckbff7e692002-04-25 21:31:11 +0000519 {
Eric Pouech46344472005-01-14 19:54:38 +0000520 set_error( STATUS_PIPE_DISCONNECTED );
Martin Wilckbff7e692002-04-25 21:31:11 +0000521 }
Eric Pouech46344472005-01-14 19:54:38 +0000522 else
523 {
Alexandre Julliard0aae1ca2007-04-02 20:41:59 +0200524 struct async *async;
525 if (!(async = create_async( current, queue, data ))) return;
526 release_object( async );
Alexandre Julliard589ce2b2007-03-26 19:13:21 +0200527 set_error( STATUS_PENDING );
Eric Pouech46344472005-01-14 19:54:38 +0000528 }
Martin Wilckbff7e692002-04-25 21:31:11 +0000529
Eric Pouech46344472005-01-14 19:54:38 +0000530 pollev = sock_reselect( sock );
531 if ( pollev ) sock_try_event( sock, pollev );
532}
533
Alexandre Julliard72bff2e2007-04-10 17:07:27 +0200534static void sock_reselect_async( struct fd *fd, struct async_queue *queue )
535{
536 struct sock *sock = get_fd_user( fd );
537 int events = sock_reselect( sock );
538 if (events) sock_try_event( sock, events );
539}
540
Mike Kaplinskiyb05774e2009-08-09 00:08:27 -0400541static void sock_cancel_async( struct fd *fd, struct process *process, struct thread *thread, client_ptr_t iosb )
Eric Pouech46344472005-01-14 19:54:38 +0000542{
543 struct sock *sock = get_fd_user( fd );
Mike Kaplinskiyb05774e2009-08-09 00:08:27 -0400544 int n = 0;
Eric Pouech46344472005-01-14 19:54:38 +0000545 assert( sock->obj.ops == &sock_ops );
546
Mike Kaplinskiyb05774e2009-08-09 00:08:27 -0400547 n += async_wake_up_by( sock->read_q, process, thread, iosb, STATUS_CANCELLED );
548 n += async_wake_up_by( sock->write_q, process, thread, iosb, STATUS_CANCELLED );
549 if (!n && iosb)
550 set_error( STATUS_NOT_FOUND );
Martin Wilckbff7e692002-04-25 21:31:11 +0000551}
552
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000553static struct fd *sock_get_fd( struct object *obj )
554{
555 struct sock *sock = (struct sock *)obj;
556 return (struct fd *)grab_object( sock->fd );
557}
558
Ove Kaaven019211f1999-10-13 16:05:37 +0000559static void sock_destroy( struct object *obj )
560{
561 struct sock *sock = (struct sock *)obj;
562 assert( obj->ops == &sock_ops );
563
Ove Kaaven019211f1999-10-13 16:05:37 +0000564 /* FIXME: special socket shutdown stuff? */
Martin Wilckaa477052002-01-09 21:16:24 +0000565
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000566 if ( sock->deferred )
Eric Pouech46344472005-01-14 19:54:38 +0000567 release_object( sock->deferred );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000568
Alexandre Julliardb2cba952007-04-03 19:36:07 +0200569 free_async_queue( sock->read_q );
570 free_async_queue( sock->write_q );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000571 if (sock->event) release_object( sock->event );
Alexandre Julliard24b64532006-12-04 18:02:02 +0100572 if (sock->fd)
573 {
574 /* shut the socket down to force pending poll() calls in the client to return */
575 shutdown( get_unix_fd(sock->fd), SHUT_RDWR );
576 release_object( sock->fd );
577 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000578}
579
580/* create a new and unconnected socket */
Martin Wilckaa477052002-01-09 21:16:24 +0000581static struct object *create_socket( int family, int type, int protocol, unsigned int flags )
Ove Kaaven019211f1999-10-13 16:05:37 +0000582{
583 struct sock *sock;
Ove Kaaven3669b941999-11-07 05:23:02 +0000584 int sockfd;
Ove Kaaven019211f1999-10-13 16:05:37 +0000585
Ove Kaaven3669b941999-11-07 05:23:02 +0000586 sockfd = socket( family, type, protocol );
587 if (debug_level)
588 fprintf(stderr,"socket(%d,%d,%d)=%d\n",family,type,protocol,sockfd);
Eric Pouech46344472005-01-14 19:54:38 +0000589 if (sockfd == -1)
590 {
Ove Kaaven3669b941999-11-07 05:23:02 +0000591 sock_set_error();
Ove Kaaven019211f1999-10-13 16:05:37 +0000592 return NULL;
Ove Kaaven3669b941999-11-07 05:23:02 +0000593 }
594 fcntl(sockfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000595 if (!(sock = alloc_object( &sock_ops )))
596 {
597 close( sockfd );
598 return NULL;
599 }
Francois Gouget126e6b82001-08-24 19:14:56 +0000600 sock->state = (type != SOCK_STREAM) ? (FD_READ|FD_WRITE) : 0;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000601 sock->mask = 0;
602 sock->hmask = 0;
603 sock->pmask = 0;
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000604 sock->polling = 0;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000605 sock->flags = flags;
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000606 sock->type = type;
607 sock->family = family;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000608 sock->event = NULL;
609 sock->window = 0;
610 sock->message = 0;
611 sock->wparam = 0;
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000612 sock->deferred = NULL;
Alexandre Julliarddf09ac52007-04-02 20:09:29 +0200613 sock->read_q = NULL;
614 sock->write_q = NULL;
Alexandre Julliardf85437c2007-04-10 22:25:07 +0200615 if (!(sock->fd = create_anonymous_fd( &sock_fd_ops, sockfd, &sock->obj,
616 (flags & WSA_FLAG_OVERLAPPED) ? 0 : FILE_SYNCHRONOUS_IO_NONALERT )))
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000617 {
618 release_object( sock );
619 return NULL;
620 }
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000621 sock_reselect( sock );
622 clear_error();
Ove Kaaven019211f1999-10-13 16:05:37 +0000623 return &sock->obj;
624}
625
626/* accept a socket (creates a new fd) */
Alexandre Julliard51885742002-05-30 20:12:58 +0000627static struct sock *accept_socket( obj_handle_t handle )
Ove Kaaven019211f1999-10-13 16:05:37 +0000628{
629 struct sock *acceptsock;
630 struct sock *sock;
631 int acceptfd;
632 struct sockaddr saddr;
Ove Kaaven019211f1999-10-13 16:05:37 +0000633
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100634 sock = (struct sock *)get_handle_obj( current->process, handle, FILE_READ_DATA, &sock_ops );
Ove Kaaven019211f1999-10-13 16:05:37 +0000635 if (!sock)
636 return NULL;
Ove Kaaven019211f1999-10-13 16:05:37 +0000637
Eric Pouech46344472005-01-14 19:54:38 +0000638 if ( sock->deferred )
639 {
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000640 acceptsock = sock->deferred;
641 sock->deferred = NULL;
Eric Pouech46344472005-01-14 19:54:38 +0000642 }
643 else
644 {
Jeremy White942010a1999-11-07 19:15:19 +0000645
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000646 /* Try to accept(2). We can't be safe that this an already connected socket
647 * or that accept() is allowed on it. In those cases we will get -1/errno
648 * return.
649 */
Mike McCormacke659f1e2005-08-09 10:37:50 +0000650 unsigned int slen = sizeof(saddr);
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000651 acceptfd = accept( get_unix_fd(sock->fd), &saddr, &slen);
Eric Pouech46344472005-01-14 19:54:38 +0000652 if (acceptfd==-1)
653 {
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000654 sock_set_error();
655 release_object( sock );
656 return NULL;
657 }
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000658 if (!(acceptsock = alloc_object( &sock_ops )))
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000659 {
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000660 close( acceptfd );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000661 release_object( sock );
662 return NULL;
663 }
664
665 /* newly created socket gets the same properties of the listening socket */
666 fcntl(acceptfd, F_SETFL, O_NONBLOCK); /* make socket nonblocking */
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000667 acceptsock->state = FD_WINE_CONNECTED|FD_READ|FD_WRITE;
668 if (sock->state & FD_WINE_NONBLOCKING)
669 acceptsock->state |= FD_WINE_NONBLOCKING;
670 acceptsock->mask = sock->mask;
671 acceptsock->hmask = 0;
672 acceptsock->pmask = 0;
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000673 acceptsock->polling = 0;
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000674 acceptsock->type = sock->type;
675 acceptsock->family = sock->family;
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000676 acceptsock->event = NULL;
677 acceptsock->window = sock->window;
678 acceptsock->message = sock->message;
679 acceptsock->wparam = 0;
680 if (sock->event) acceptsock->event = (struct event *)grab_object( sock->event );
681 acceptsock->flags = sock->flags;
Robert Shearmanc5165712005-05-25 18:41:09 +0000682 acceptsock->deferred = NULL;
Alexandre Julliarddf09ac52007-04-02 20:09:29 +0200683 acceptsock->read_q = NULL;
684 acceptsock->write_q = NULL;
Alexandre Julliardf85437c2007-04-10 22:25:07 +0200685 if (!(acceptsock->fd = create_anonymous_fd( &sock_fd_ops, acceptfd, &acceptsock->obj,
686 get_fd_options( sock->fd ) )))
Alexandre Julliarde66207e2003-02-19 00:33:32 +0000687 {
688 release_object( acceptsock );
689 release_object( sock );
690 return NULL;
691 }
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000692 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000693 clear_error();
694 sock->pmask &= ~FD_ACCEPT;
695 sock->hmask &= ~FD_ACCEPT;
Stephane Lussier21b099f2000-10-13 17:04:52 +0000696 sock_reselect( sock );
Ove Kaaven019211f1999-10-13 16:05:37 +0000697 release_object( sock );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000698 return acceptsock;
Ove Kaaven019211f1999-10-13 16:05:37 +0000699}
700
701/* set the last error depending on errno */
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000702static int sock_get_error( int err )
Ove Kaaven019211f1999-10-13 16:05:37 +0000703{
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000704 switch (err)
Ove Kaaven019211f1999-10-13 16:05:37 +0000705 {
Eric Pouech46344472005-01-14 19:54:38 +0000706 case EINTR: return WSAEINTR;
707 case EBADF: return WSAEBADF;
Ove Kaaven019211f1999-10-13 16:05:37 +0000708 case EPERM:
Eric Pouech46344472005-01-14 19:54:38 +0000709 case EACCES: return WSAEACCES;
710 case EFAULT: return WSAEFAULT;
711 case EINVAL: return WSAEINVAL;
712 case EMFILE: return WSAEMFILE;
713 case EWOULDBLOCK: return WSAEWOULDBLOCK;
714 case EINPROGRESS: return WSAEINPROGRESS;
715 case EALREADY: return WSAEALREADY;
716 case ENOTSOCK: return WSAENOTSOCK;
717 case EDESTADDRREQ: return WSAEDESTADDRREQ;
718 case EMSGSIZE: return WSAEMSGSIZE;
719 case EPROTOTYPE: return WSAEPROTOTYPE;
720 case ENOPROTOOPT: return WSAENOPROTOOPT;
721 case EPROTONOSUPPORT: return WSAEPROTONOSUPPORT;
722 case ESOCKTNOSUPPORT: return WSAESOCKTNOSUPPORT;
723 case EOPNOTSUPP: return WSAEOPNOTSUPP;
724 case EPFNOSUPPORT: return WSAEPFNOSUPPORT;
725 case EAFNOSUPPORT: return WSAEAFNOSUPPORT;
726 case EADDRINUSE: return WSAEADDRINUSE;
727 case EADDRNOTAVAIL: return WSAEADDRNOTAVAIL;
728 case ENETDOWN: return WSAENETDOWN;
729 case ENETUNREACH: return WSAENETUNREACH;
730 case ENETRESET: return WSAENETRESET;
731 case ECONNABORTED: return WSAECONNABORTED;
Ove Kaaven019211f1999-10-13 16:05:37 +0000732 case EPIPE:
Eric Pouech46344472005-01-14 19:54:38 +0000733 case ECONNRESET: return WSAECONNRESET;
734 case ENOBUFS: return WSAENOBUFS;
735 case EISCONN: return WSAEISCONN;
736 case ENOTCONN: return WSAENOTCONN;
737 case ESHUTDOWN: return WSAESHUTDOWN;
738 case ETOOMANYREFS: return WSAETOOMANYREFS;
739 case ETIMEDOUT: return WSAETIMEDOUT;
740 case ECONNREFUSED: return WSAECONNREFUSED;
741 case ELOOP: return WSAELOOP;
742 case ENAMETOOLONG: return WSAENAMETOOLONG;
743 case EHOSTDOWN: return WSAEHOSTDOWN;
744 case EHOSTUNREACH: return WSAEHOSTUNREACH;
745 case ENOTEMPTY: return WSAENOTEMPTY;
Ove Kaaven019211f1999-10-13 16:05:37 +0000746#ifdef EPROCLIM
Eric Pouech46344472005-01-14 19:54:38 +0000747 case EPROCLIM: return WSAEPROCLIM;
Ove Kaaven019211f1999-10-13 16:05:37 +0000748#endif
749#ifdef EUSERS
Eric Pouech46344472005-01-14 19:54:38 +0000750 case EUSERS: return WSAEUSERS;
Ove Kaaven019211f1999-10-13 16:05:37 +0000751#endif
752#ifdef EDQUOT
Eric Pouech46344472005-01-14 19:54:38 +0000753 case EDQUOT: return WSAEDQUOT;
Ove Kaaven019211f1999-10-13 16:05:37 +0000754#endif
755#ifdef ESTALE
Eric Pouech46344472005-01-14 19:54:38 +0000756 case ESTALE: return WSAESTALE;
Ove Kaaven019211f1999-10-13 16:05:37 +0000757#endif
758#ifdef EREMOTE
Eric Pouech46344472005-01-14 19:54:38 +0000759 case EREMOTE: return WSAEREMOTE;
Ove Kaaven019211f1999-10-13 16:05:37 +0000760#endif
Dmitry Timoshkov69025ba2007-09-04 22:51:39 +0900761 default:
762 errno = err;
763 perror("wineserver: sock_get_error() can't map error");
764 return WSAEFAULT;
Ove Kaaven019211f1999-10-13 16:05:37 +0000765 }
766}
767
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000768/* set the last error depending on errno */
769static void sock_set_error(void)
770{
771 set_error( sock_get_error( errno ) );
772}
773
Ove Kaaven019211f1999-10-13 16:05:37 +0000774/* create a socket */
775DECL_HANDLER(create_socket)
776{
777 struct object *obj;
Ove Kaaven019211f1999-10-13 16:05:37 +0000778
Alexandre Julliard9caa71e2001-11-30 18:46:42 +0000779 reply->handle = 0;
Martin Wilckaa477052002-01-09 21:16:24 +0000780 if ((obj = create_socket( req->family, req->type, req->protocol, req->flags )) != NULL)
Ove Kaaven019211f1999-10-13 16:05:37 +0000781 {
Alexandre Julliard24560e72005-12-09 13:58:25 +0100782 reply->handle = alloc_handle( current->process, obj, req->access, req->attributes );
Ove Kaaven019211f1999-10-13 16:05:37 +0000783 release_object( obj );
784 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000785}
786
787/* accept a socket */
788DECL_HANDLER(accept_socket)
789{
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000790 struct sock *sock;
Ove Kaaven019211f1999-10-13 16:05:37 +0000791
Alexandre Julliard9caa71e2001-11-30 18:46:42 +0000792 reply->handle = 0;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000793 if ((sock = accept_socket( req->lhandle )) != NULL)
Ove Kaaven019211f1999-10-13 16:05:37 +0000794 {
Alexandre Julliard24560e72005-12-09 13:58:25 +0100795 reply->handle = alloc_handle( current->process, &sock->obj, req->access, req->attributes );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000796 sock->wparam = reply->handle; /* wparam for message is the socket handle */
797 sock_reselect( sock );
798 release_object( &sock->obj );
Ove Kaaven019211f1999-10-13 16:05:37 +0000799 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000800}
801
802/* set socket event parameters */
803DECL_HANDLER(set_socket_event)
804{
805 struct sock *sock;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000806 struct event *old_event;
Martin Wilckbff7e692002-04-25 21:31:11 +0000807 int pollev;
Ove Kaaven019211f1999-10-13 16:05:37 +0000808
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100809 if (!(sock = (struct sock *)get_handle_obj( current->process, req->handle,
810 FILE_WRITE_ATTRIBUTES, &sock_ops))) return;
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000811 old_event = sock->event;
Ove Kaaven019211f1999-10-13 16:05:37 +0000812 sock->mask = req->mask;
Eric Poueche0477ce2006-10-21 08:49:40 +0200813 sock->hmask &= ~req->mask; /* re-enable held events */
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000814 sock->event = NULL;
815 sock->window = req->window;
816 sock->message = req->msg;
817 sock->wparam = req->handle; /* wparam is the socket handle */
818 if (req->event) sock->event = get_event_obj( current->process, req->event, EVENT_MODIFY_STATE );
819
Ove Kaavenf45608f1999-10-23 16:53:34 +0000820 if (debug_level && sock->event) fprintf(stderr, "event ptr: %p\n", sock->event);
Martin Wilckbff7e692002-04-25 21:31:11 +0000821
822 pollev = sock_reselect( sock );
Eric Pouech46344472005-01-14 19:54:38 +0000823 if ( pollev ) sock_try_event( sock, pollev );
Martin Wilckbff7e692002-04-25 21:31:11 +0000824
Ove Kaaven019211f1999-10-13 16:05:37 +0000825 if (sock->mask)
Francois Gouget126e6b82001-08-24 19:14:56 +0000826 sock->state |= FD_WINE_NONBLOCKING;
Stephane Lussier56adb252000-07-16 18:14:01 +0000827
Vincent Béron9a624912002-05-31 23:06:46 +0000828 /* if a network event is pending, signal the event object
Stephane Lussier56adb252000-07-16 18:14:01 +0000829 it is possible that FD_CONNECT or FD_ACCEPT network events has happened
Vincent Béron9a624912002-05-31 23:06:46 +0000830 before a WSAEventSelect() was done on it.
Stephane Lussier56adb252000-07-16 18:14:01 +0000831 (when dealing with Asynchronous socket) */
Martin Wilckbff7e692002-04-25 21:31:11 +0000832 if (sock->pmask & sock->mask) sock_wake_up( sock, pollev );
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000833
834 if (old_event) release_object( old_event ); /* we're through with it */
Ove Kaaven019211f1999-10-13 16:05:37 +0000835 release_object( &sock->obj );
836}
837
838/* get socket event parameters */
839DECL_HANDLER(get_socket_event)
840{
841 struct sock *sock;
842
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100843 sock = (struct sock *)get_handle_obj( current->process, req->handle, FILE_READ_ATTRIBUTES, &sock_ops );
Ove Kaaven019211f1999-10-13 16:05:37 +0000844 if (!sock)
845 {
Alexandre Julliard9caa71e2001-11-30 18:46:42 +0000846 reply->mask = 0;
847 reply->pmask = 0;
848 reply->state = 0;
849 set_error( WSAENOTSOCK );
850 return;
Ove Kaaven019211f1999-10-13 16:05:37 +0000851 }
Alexandre Julliard9caa71e2001-11-30 18:46:42 +0000852 reply->mask = sock->mask;
853 reply->pmask = sock->pmask;
854 reply->state = sock->state;
855 set_reply_data( sock->errors, min( get_reply_max_size(), sizeof(sock->errors) ));
Alexandre Julliard92643002000-08-31 01:59:51 +0000856
Ove Kaaven019211f1999-10-13 16:05:37 +0000857 if (req->service)
858 {
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000859 if (req->c_event)
Ove Kaaven019211f1999-10-13 16:05:37 +0000860 {
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000861 struct event *cevent = get_event_obj( current->process, req->c_event,
862 EVENT_MODIFY_STATE );
863 if (cevent)
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000864 {
Ove Kaavencb98a6e2000-03-26 18:21:16 +0000865 reset_event( cevent );
866 release_object( cevent );
867 }
Ove Kaaven019211f1999-10-13 16:05:37 +0000868 }
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000869 sock->pmask = 0;
870 sock_reselect( sock );
Ove Kaaven019211f1999-10-13 16:05:37 +0000871 }
872 release_object( &sock->obj );
873}
874
875/* re-enable pending socket events */
876DECL_HANDLER(enable_socket_event)
877{
878 struct sock *sock;
Martin Wilckbff7e692002-04-25 21:31:11 +0000879 int pollev;
Ove Kaaven019211f1999-10-13 16:05:37 +0000880
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000881 if (!(sock = (struct sock*)get_handle_obj( current->process, req->handle,
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100882 FILE_WRITE_ATTRIBUTES, &sock_ops)))
Alexandre Julliard81f2a732002-03-23 20:43:52 +0000883 return;
884
Ove Kaaven019211f1999-10-13 16:05:37 +0000885 sock->pmask &= ~req->mask; /* is this safe? */
886 sock->hmask &= ~req->mask;
Martin Wilck75242f92002-04-27 01:06:09 +0000887 if ( req->mask & FD_READ )
888 sock->hmask &= ~FD_CLOSE;
Ove Kaaven019211f1999-10-13 16:05:37 +0000889 sock->state |= req->sstate;
890 sock->state &= ~req->cstate;
Martin Wilck15a0f0e2002-07-03 21:04:09 +0000891 if ( sock->type != SOCK_STREAM ) sock->state &= ~STREAM_FLAG_MASK;
Martin Wilckbff7e692002-04-25 21:31:11 +0000892
893 pollev = sock_reselect( sock );
Eric Pouech46344472005-01-14 19:54:38 +0000894 if ( pollev ) sock_try_event( sock, pollev );
Martin Wilckbff7e692002-04-25 21:31:11 +0000895
Ove Kaaven019211f1999-10-13 16:05:37 +0000896 release_object( &sock->obj );
897}
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000898
899DECL_HANDLER(set_socket_deferred)
900{
901 struct sock *sock, *acceptsock;
902
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100903 sock=(struct sock *)get_handle_obj( current->process, req->handle, FILE_WRITE_ATTRIBUTES, &sock_ops );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000904 if ( !sock )
905 {
Eric Pouech46344472005-01-14 19:54:38 +0000906 set_error( WSAENOTSOCK );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000907 return;
908 }
Alexandre Julliarda510a7e2005-12-12 16:46:17 +0100909 acceptsock = (struct sock *)get_handle_obj( current->process, req->deferred, 0, &sock_ops );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000910 if ( !acceptsock )
911 {
Eric Pouech46344472005-01-14 19:54:38 +0000912 release_object( sock );
913 set_error( WSAENOTSOCK );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000914 return;
915 }
916 sock->deferred = acceptsock;
Eric Pouech46344472005-01-14 19:54:38 +0000917 release_object( sock );
Martin Wilckd15bf1c2002-04-23 22:03:42 +0000918}