mono/io-layer/sockets.c

   1 /*
   2  * sockets.c:  Socket handles
   3  *
   4  * Author:
   5  *      Dick Porter (dick@ximian.com)
   6  *
   7  * (C) 2002 Ximian, Inc.
   8  */
   9
  10 #include <config.h>
  11
  12 #ifndef DISABLE_SOCKETS
  13
  14 #include <glib.h>
  15 #include <pthread.h>
  16 #include <errno.h>
  17 #include <string.h>
  18 #include <sys/types.h>
  19 #include <sys/socket.h>
  20 #ifdef HAVE_SYS_UIO_H
  21 #  include <sys/uio.h>
  22 #endif
  23 #ifdef HAVE_SYS_IOCTL_H
  24 #  include <sys/ioctl.h>
  25 #endif
  26 #ifdef HAVE_SYS_FILIO_H
  27 #include <sys/filio.h>     /* defines FIONBIO and FIONREAD */
  28 #endif
  29 #ifdef HAVE_SYS_SOCKIO_H
  30 #include <sys/sockio.h>    /* defines SIOCATMARK */
  31 #endif
  32 #include <unistd.h>
  33 #include <fcntl.h>
  34
  35 #ifndef HAVE_MSG_NOSIGNAL
  36 #include <signal.h>
  37 #endif
  38
  39 #include <mono/io-layer/wapi.h>
  40 #include <mono/io-layer/wapi-private.h>
  41 #include <mono/io-layer/socket-private.h>
  42 #include <mono/io-layer/handles-private.h>
  43 #include <mono/io-layer/socket-wrappers.h>
  44 #include <mono/utils/mono-poll.h>
  45
  46 #include <netinet/in.h>
  47 #include <netinet/tcp.h>
  48 #include <netdb.h>
  49 #include <arpa/inet.h>
  50 #ifdef HAVE_SYS_SENDFILE_H
  51 #include <sys/sendfile.h>
  52 #endif
  53
  54 #if 0
  55 #define DEBUG(...) g_message(__VA_ARGS__)
  56 #else
  57 #define DEBUG(...)
  58 #endif
  59
  60 static guint32 startup_count=0;
  61 static guint32 in_cleanup = 0;
  62
  63 static void socket_close (gpointer handle, gpointer data);
  64
  65 struct _WapiHandleOps _wapi_socket_ops = {
  66         socket_close,           /* close */
  67         NULL,                   /* signal */
  68         NULL,                   /* own */
  69         NULL,                   /* is_owned */
  70         NULL,                   /* special_wait */
  71         NULL                    /* prewait */
  72 };
  73
  74 static mono_once_t socket_ops_once=MONO_ONCE_INIT;
  75
  76 static void socket_ops_init (void)
  77 {
  78         /* No capabilities to register */
  79 }
  80
  81 static void socket_close (gpointer handle, gpointer data)
  82 {
  83         int ret;
  84         struct _WapiHandle_socket *socket_handle = (struct _WapiHandle_socket *)data;
  85
  86         DEBUG ("%s: closing socket handle %p", __func__, handle);
  87
  88         if (startup_count == 0 && !in_cleanup) {
  89                 WSASetLastError (WSANOTINITIALISED);
  90                 return;
  91         }
  92
  93         /* Shutdown the socket for reading, to interrupt any potential
  94          * receives that may be blocking for data.  See bug 75705.
  95          */
  96         shutdown (GPOINTER_TO_UINT (handle), SHUT_RD);
  97
  98         do {
  99                 ret = close (GPOINTER_TO_UINT(handle));
 100         } while (ret == -1 && errno == EINTR &&
 101                  !_wapi_thread_cur_apc_pending ());
 102
 103         if (ret == -1) {
 104                 gint errnum = errno;
 105                 DEBUG ("%s: close error: %s", __func__, strerror (errno));
 106                 errnum = errno_to_WSA (errnum, __func__);
 107                 if (!in_cleanup)
 108                         WSASetLastError (errnum);
 109         }
 110
 111         if (!in_cleanup)
 112                 socket_handle->saved_error = 0;
 113 }
 114
 115 int WSAStartup(guint32 requested, WapiWSAData *data)
 116 {
 117         if (data == NULL) {
 118                 return(WSAEFAULT);
 119         }
 120
 121         /* Insist on v2.0+ */
 122         if (requested < MAKEWORD(2,0)) {
 123                 return(WSAVERNOTSUPPORTED);
 124         }
 125
 126         startup_count++;
 127
 128         /* I've no idea what is the minor version of the spec I read */
 129         data->wHighVersion = MAKEWORD(2,2);
 130
 131         data->wVersion = requested < data->wHighVersion? requested:
 132                 data->wHighVersion;
 133
 134         DEBUG ("%s: high version 0x%x", __func__, data->wHighVersion);
 135
 136         strncpy (data->szDescription, "WAPI", WSADESCRIPTION_LEN);
 137         strncpy (data->szSystemStatus, "groovy", WSASYS_STATUS_LEN);
 138
 139         return(0);
 140 }
 141
 142 static gboolean
 143 cleanup_close (gpointer handle, gpointer data)
 144 {
 145         _wapi_handle_ops_close (handle, NULL);
 146         return TRUE;
 147 }
 148
 149 int WSACleanup(void)
 150 {
 151         DEBUG ("%s: cleaning up", __func__);
 152
 153         if (--startup_count) {
 154                 /* Do nothing */
 155                 return(0);
 156         }
 157
 158         in_cleanup = 1;
 159         _wapi_handle_foreach (WAPI_HANDLE_SOCKET, cleanup_close, NULL);
 160         in_cleanup = 0;
 161         return(0);
 162 }
 163
 164 void WSASetLastError(int error)
 165 {
 166         SetLastError (error);
 167 }
 168
 169 int WSAGetLastError(void)
 170 {
 171         return(GetLastError ());
 172 }
 173
 174 int closesocket(guint32 fd)
 175 {
 176         gpointer handle = GUINT_TO_POINTER (fd);
 177
 178         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 179                 WSASetLastError (WSAENOTSOCK);
 180                 return(0);
 181         }
 182
 183         _wapi_handle_unref (handle);
 184         return(0);
 185 }
 186
 187 guint32 _wapi_accept(guint32 fd, struct sockaddr *addr, socklen_t *addrlen)
 188 {
 189         gpointer handle = GUINT_TO_POINTER (fd);
 190         gpointer new_handle;
 191         struct _WapiHandle_socket *socket_handle;
 192         struct _WapiHandle_socket new_socket_handle = {0};
 193         gboolean ok;
 194         int new_fd;
 195
 196         if (startup_count == 0) {
 197                 WSASetLastError (WSANOTINITIALISED);
 198                 return(INVALID_SOCKET);
 199         }
 200
 201         if (addr != NULL && *addrlen < sizeof(struct sockaddr)) {
 202                 WSASetLastError (WSAEFAULT);
 203                 return(INVALID_SOCKET);
 204         }
 205
 206         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 207                 WSASetLastError (WSAENOTSOCK);
 208                 return(INVALID_SOCKET);
 209         }
 210
 211         ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 212                                   (gpointer *)&socket_handle);
 213         if (ok == FALSE) {
 214                 g_warning ("%s: error looking up socket handle %p",
 215                            __func__, handle);
 216                 WSASetLastError (WSAENOTSOCK);
 217                 return(INVALID_SOCKET);
 218         }
 219
 220         do {
 221                 new_fd = accept (fd, addr, addrlen);
 222         } while (new_fd == -1 && errno == EINTR &&
 223                  !_wapi_thread_cur_apc_pending());
 224
 225         if (new_fd == -1) {
 226                 gint errnum = errno;
 227                 DEBUG ("%s: accept error: %s", __func__, strerror(errno));
 228
 229                 errnum = errno_to_WSA (errnum, __func__);
 230                 WSASetLastError (errnum);
 231
 232                 return(INVALID_SOCKET);
 233         }
 234
 235         if (new_fd >= _wapi_fd_reserve) {
 236                 DEBUG ("%s: File descriptor is too big", __func__);
 237
 238                 WSASetLastError (WSASYSCALLFAILURE);
 239
 240                 close (new_fd);
 241
 242                 return(INVALID_SOCKET);
 243         }
 244
 245         new_socket_handle.domain = socket_handle->domain;
 246         new_socket_handle.type = socket_handle->type;
 247         new_socket_handle.protocol = socket_handle->protocol;
 248         new_socket_handle.still_readable = 1;
 249
 250         new_handle = _wapi_handle_new_fd (WAPI_HANDLE_SOCKET, new_fd,
 251                                           &new_socket_handle);
 252         if(new_handle == _WAPI_HANDLE_INVALID) {
 253                 g_warning ("%s: error creating socket handle", __func__);
 254                 WSASetLastError (ERROR_GEN_FAILURE);
 255                 return(INVALID_SOCKET);
 256         }
 257
 258         DEBUG ("%s: returning newly accepted socket handle %p with",
 259                    __func__, new_handle);
 260
 261         return(new_fd);
 262 }
 263
 264 int _wapi_bind(guint32 fd, struct sockaddr *my_addr, socklen_t addrlen)
 265 {
 266         gpointer handle = GUINT_TO_POINTER (fd);
 267         int ret;
 268
 269         if (startup_count == 0) {
 270                 WSASetLastError (WSANOTINITIALISED);
 271                 return(SOCKET_ERROR);
 272         }
 273
 274         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 275                 WSASetLastError (WSAENOTSOCK);
 276                 return(SOCKET_ERROR);
 277         }
 278
 279         ret = bind (fd, my_addr, addrlen);
 280         if (ret == -1) {
 281                 gint errnum = errno;
 282                 DEBUG ("%s: bind error: %s", __func__, strerror(errno));
 283                 errnum = errno_to_WSA (errnum, __func__);
 284                 WSASetLastError (errnum);
 285
 286                 return(SOCKET_ERROR);
 287         }
 288         return(ret);
 289 }
 290
 291 int _wapi_connect(guint32 fd, const struct sockaddr *serv_addr,
 292                   socklen_t addrlen)
 293 {
 294         gpointer handle = GUINT_TO_POINTER (fd);
 295         struct _WapiHandle_socket *socket_handle;
 296         gboolean ok;
 297         gint errnum;
 298
 299         if (startup_count == 0) {
 300                 WSASetLastError (WSANOTINITIALISED);
 301                 return(SOCKET_ERROR);
 302         }
 303
 304         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 305                 WSASetLastError (WSAENOTSOCK);
 306                 return(SOCKET_ERROR);
 307         }
 308
 309         if (connect (fd, serv_addr, addrlen) == -1) {
 310                 mono_pollfd fds;
 311                 int so_error;
 312                 socklen_t len;
 313
 314                 errnum = errno;
 315
 316                 if (errno != EINTR) {
 317                         DEBUG ("%s: connect error: %s", __func__,
 318                                    strerror (errnum));
 319
 320                         errnum = errno_to_WSA (errnum, __func__);
 321                         if (errnum == WSAEINPROGRESS)
 322                                 errnum = WSAEWOULDBLOCK; /* see bug #73053 */
 323
 324                         WSASetLastError (errnum);
 325
 326                         /*
 327                          * On solaris x86 getsockopt (SO_ERROR) is not set after
 328                          * connect () fails so we need to save this error.
 329                          *
 330                          * But don't do this for EWOULDBLOCK (bug 317315)
 331                          */
 332                         if (errnum != WSAEWOULDBLOCK) {
 333                                 ok = _wapi_lookup_handle (handle,
 334                                                           WAPI_HANDLE_SOCKET,
 335                                                           (gpointer *)&socket_handle);
 336                                 if (ok == FALSE) {
 337                                         /* ECONNRESET means the socket was closed by another thread */
 338                                         /* Async close on mac raises ECONNABORTED. */
 339                                         if (errnum != WSAECONNRESET && errnum != WSAENETDOWN)
 340                                                 g_warning ("%s: error looking up socket handle %p (error %d)", __func__, handle, errnum);
 341                                 } else {
 342                                         socket_handle->saved_error = errnum;
 343                                 }
 344                         }
 345                         return(SOCKET_ERROR);
 346                 }
 347
 348                 fds.fd = fd;
 349                 fds.events = POLLOUT;
 350                 while (mono_poll (&fds, 1, -1) == -1 &&
 351                        !_wapi_thread_cur_apc_pending ()) {
 352                         if (errno != EINTR) {
 353                                 errnum = errno_to_WSA (errno, __func__);
 354
 355                                 DEBUG ("%s: connect poll error: %s",
 356                                            __func__, strerror (errno));
 357
 358                                 WSASetLastError (errnum);
 359                                 return(SOCKET_ERROR);
 360                         }
 361                 }
 362
 363                 len = sizeof(so_error);
 364                 if (getsockopt (fd, SOL_SOCKET, SO_ERROR, &so_error,
 365                                 &len) == -1) {
 366                         errnum = errno_to_WSA (errno, __func__);
 367
 368                         DEBUG ("%s: connect getsockopt error: %s",
 369                                    __func__, strerror (errno));
 370
 371                         WSASetLastError (errnum);
 372                         return(SOCKET_ERROR);
 373                 }
 374
 375                 if (so_error != 0) {
 376                         errnum = errno_to_WSA (so_error, __func__);
 377
 378                         /* Need to save this socket error */
 379                         ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 380                                                   (gpointer *)&socket_handle);
 381                         if (ok == FALSE) {
 382                                 g_warning ("%s: error looking up socket handle %p", __func__, handle);
 383                         } else {
 384                                 socket_handle->saved_error = errnum;
 385                         }
 386
 387                         DEBUG ("%s: connect getsockopt returned error: %s",
 388                                    __func__, strerror (so_error));
 389
 390                         WSASetLastError (errnum);
 391                         return(SOCKET_ERROR);
 392                 }
 393         }
 394
 395         return(0);
 396 }
 397
 398 int _wapi_getpeername(guint32 fd, struct sockaddr *name, socklen_t *namelen)
 399 {
 400         gpointer handle = GUINT_TO_POINTER (fd);
 401         int ret;
 402
 403         if (startup_count == 0) {
 404                 WSASetLastError (WSANOTINITIALISED);
 405                 return(SOCKET_ERROR);
 406         }
 407
 408         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 409                 WSASetLastError (WSAENOTSOCK);
 410                 return(SOCKET_ERROR);
 411         }
 412
 413         ret = getpeername (fd, name, namelen);
 414         if (ret == -1) {
 415                 gint errnum = errno;
 416                 DEBUG ("%s: getpeername error: %s", __func__,
 417                            strerror (errno));
 418
 419                 errnum = errno_to_WSA (errnum, __func__);
 420                 WSASetLastError (errnum);
 421
 422                 return(SOCKET_ERROR);
 423         }
 424
 425         return(ret);
 426 }
 427
 428 int _wapi_getsockname(guint32 fd, struct sockaddr *name, socklen_t *namelen)
 429 {
 430         gpointer handle = GUINT_TO_POINTER (fd);
 431         int ret;
 432
 433         if (startup_count == 0) {
 434                 WSASetLastError (WSANOTINITIALISED);
 435                 return(SOCKET_ERROR);
 436         }
 437
 438         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 439                 WSASetLastError (WSAENOTSOCK);
 440                 return(SOCKET_ERROR);
 441         }
 442
 443         ret = getsockname (fd, name, namelen);
 444         if (ret == -1) {
 445                 gint errnum = errno;
 446                 DEBUG ("%s: getsockname error: %s", __func__,
 447                            strerror (errno));
 448
 449                 errnum = errno_to_WSA (errnum, __func__);
 450                 WSASetLastError (errnum);
 451
 452                 return(SOCKET_ERROR);
 453         }
 454
 455         return(ret);
 456 }
 457
 458 int _wapi_getsockopt(guint32 fd, int level, int optname, void *optval,
 459                      socklen_t *optlen)
 460 {
 461         gpointer handle = GUINT_TO_POINTER (fd);
 462         int ret;
 463         struct timeval tv;
 464         void *tmp_val;
 465         struct _WapiHandle_socket *socket_handle;
 466         gboolean ok;
 467
 468         if (startup_count == 0) {
 469                 WSASetLastError (WSANOTINITIALISED);
 470                 return(SOCKET_ERROR);
 471         }
 472
 473         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 474                 WSASetLastError (WSAENOTSOCK);
 475                 return(SOCKET_ERROR);
 476         }
 477
 478         tmp_val = optval;
 479         if (level == SOL_SOCKET &&
 480             (optname == SO_RCVTIMEO || optname == SO_SNDTIMEO)) {
 481                 tmp_val = &tv;
 482                 *optlen = sizeof (tv);
 483         }
 484
 485         ret = getsockopt (fd, level, optname, tmp_val, optlen);
 486         if (ret == -1) {
 487                 gint errnum = errno;
 488                 DEBUG ("%s: getsockopt error: %s", __func__,
 489                            strerror (errno));
 490
 491                 errnum = errno_to_WSA (errnum, __func__);
 492                 WSASetLastError (errnum);
 493
 494                 return(SOCKET_ERROR);
 495         }
 496
 497         if (level == SOL_SOCKET &&
 498             (optname == SO_RCVTIMEO || optname == SO_SNDTIMEO)) {
 499                 *((int *) optval)  = tv.tv_sec * 1000 + (tv.tv_usec / 1000);    // milli from micro
 500                 *optlen = sizeof (int);
 501         }
 502
 503         if (optname == SO_ERROR) {
 504                 ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 505                                           (gpointer *)&socket_handle);
 506                 if (ok == FALSE) {
 507                         g_warning ("%s: error looking up socket handle %p",
 508                                    __func__, handle);
 509
 510                         /* can't extract the last error */
 511                         *((int *) optval) = errno_to_WSA (*((int *)optval),
 512                                                           __func__);
 513                 } else {
 514                         if (*((int *)optval) != 0) {
 515                                 *((int *) optval) = errno_to_WSA (*((int *)optval),
 516                                                                   __func__);
 517                                 socket_handle->saved_error = *((int *)optval);
 518                         } else {
 519                                 *((int *)optval) = socket_handle->saved_error;
 520                         }
 521                 }
 522         }
 523
 524         return(ret);
 525 }
 526
 527 int _wapi_listen(guint32 fd, int backlog)
 528 {
 529         gpointer handle = GUINT_TO_POINTER (fd);
 530         int ret;
 531
 532         if (startup_count == 0) {
 533                 WSASetLastError (WSANOTINITIALISED);
 534                 return(SOCKET_ERROR);
 535         }
 536
 537         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 538                 WSASetLastError (WSAENOTSOCK);
 539                 return(SOCKET_ERROR);
 540         }
 541
 542         ret = listen (fd, backlog);
 543         if (ret == -1) {
 544                 gint errnum = errno;
 545                 DEBUG ("%s: listen error: %s", __func__, strerror (errno));
 546
 547                 errnum = errno_to_WSA (errnum, __func__);
 548                 WSASetLastError (errnum);
 549
 550                 return(SOCKET_ERROR);
 551         }
 552
 553         return(0);
 554 }
 555
 556 int _wapi_recv(guint32 fd, void *buf, size_t len, int recv_flags)
 557 {
 558         return(_wapi_recvfrom (fd, buf, len, recv_flags, NULL, 0));
 559 }
 560
 561 int _wapi_recvfrom(guint32 fd, void *buf, size_t len, int recv_flags,
 562                    struct sockaddr *from, socklen_t *fromlen)
 563 {
 564         gpointer handle = GUINT_TO_POINTER (fd);
 565         struct _WapiHandle_socket *socket_handle;
 566         gboolean ok;
 567         int ret;
 568
 569         if (startup_count == 0) {
 570                 WSASetLastError (WSANOTINITIALISED);
 571                 return(SOCKET_ERROR);
 572         }
 573
 574         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 575                 WSASetLastError (WSAENOTSOCK);
 576                 return(SOCKET_ERROR);
 577         }
 578
 579         do {
 580                 ret = recvfrom (fd, buf, len, recv_flags, from, fromlen);
 581         } while (ret == -1 && errno == EINTR &&
 582                  !_wapi_thread_cur_apc_pending ());
 583
 584         if (ret == 0 && len > 0) {
 585                 /* According to the Linux man page, recvfrom only
 586                  * returns 0 when the socket has been shut down
 587                  * cleanly.  Turn this into an EINTR to simulate win32
 588                  * behaviour of returning EINTR when a socket is
 589                  * closed while the recvfrom is blocking (we use a
 590                  * shutdown() in socket_close() to trigger this.) See
 591                  * bug 75705.
 592                  */
 593                 /* Distinguish between the socket being shut down at
 594                  * the local or remote ends, and reads that request 0
 595                  * bytes to be read
 596                  */
 597
 598                 /* If this returns FALSE, it means the socket has been
 599                  * closed locally.  If it returns TRUE, but
 600                  * still_readable != 1 then shutdown
 601                  * (SHUT_RD|SHUT_RDWR) has been called locally.
 602                  */
 603                 ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 604                                           (gpointer *)&socket_handle);
 605                 if (ok == FALSE || socket_handle->still_readable != 1) {
 606                         ret = -1;
 607                         errno = EINTR;
 608                 }
 609         }
 610
 611         if (ret == -1) {
 612                 gint errnum = errno;
 613                 DEBUG ("%s: recv error: %s", __func__, strerror(errno));
 614
 615                 errnum = errno_to_WSA (errnum, __func__);
 616                 WSASetLastError (errnum);
 617
 618                 return(SOCKET_ERROR);
 619         }
 620         return(ret);
 621 }
 622
 623 static int
 624 _wapi_recvmsg(guint32 fd, struct msghdr *msg, int recv_flags)
 625 {
 626         gpointer handle = GUINT_TO_POINTER (fd);
 627         struct _WapiHandle_socket *socket_handle;
 628         gboolean ok;
 629         int ret;
 630
 631         if (startup_count == 0) {
 632                 WSASetLastError (WSANOTINITIALISED);
 633                 return(SOCKET_ERROR);
 634         }
 635
 636         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 637                 WSASetLastError (WSAENOTSOCK);
 638                 return(SOCKET_ERROR);
 639         }
 640
 641         do {
 642                 ret = recvmsg (fd, msg, recv_flags);
 643         } while (ret == -1 && errno == EINTR &&
 644                  !_wapi_thread_cur_apc_pending ());
 645
 646         if (ret == 0) {
 647                 /* see _wapi_recvfrom */
 648                 ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 649                                           (gpointer *)&socket_handle);
 650                 if (ok == FALSE || socket_handle->still_readable != 1) {
 651                         ret = -1;
 652                         errno = EINTR;
 653                 }
 654         }
 655
 656         if (ret == -1) {
 657                 gint errnum = errno;
 658                 DEBUG ("%s: recvmsg error: %s", __func__, strerror(errno));
 659
 660                 errnum = errno_to_WSA (errnum, __func__);
 661                 WSASetLastError (errnum);
 662
 663                 return(SOCKET_ERROR);
 664         }
 665         return(ret);
 666 }
 667
 668 int _wapi_send(guint32 fd, const void *msg, size_t len, int send_flags)
 669 {
 670         gpointer handle = GUINT_TO_POINTER (fd);
 671         int ret;
 672
 673         if (startup_count == 0) {
 674                 WSASetLastError (WSANOTINITIALISED);
 675                 return(SOCKET_ERROR);
 676         }
 677
 678         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 679                 WSASetLastError (WSAENOTSOCK);
 680                 return(SOCKET_ERROR);
 681         }
 682
 683         do {
 684                 ret = send (fd, msg, len, send_flags);
 685         } while (ret == -1 && errno == EINTR &&
 686                  !_wapi_thread_cur_apc_pending ());
 687
 688         if (ret == -1) {
 689                 gint errnum = errno;
 690                 DEBUG ("%s: send error: %s", __func__, strerror (errno));
 691
 692 #ifdef O_NONBLOCK
 693                 /* At least linux returns EAGAIN/EWOULDBLOCK when the timeout has been set on
 694                  * a blocking socket. See bug #599488 */
 695                 if (errnum == EAGAIN) {
 696                         ret = fcntl (fd, F_GETFL, 0);
 697                         if (ret != -1 && (ret & O_NONBLOCK) == 0)
 698                                 errnum = ETIMEDOUT;
 699                 }
 700 #endif /* O_NONBLOCK */
 701                 errnum = errno_to_WSA (errnum, __func__);
 702                 WSASetLastError (errnum);
 703
 704                 return(SOCKET_ERROR);
 705         }
 706         return(ret);
 707 }
 708
 709 int _wapi_sendto(guint32 fd, const void *msg, size_t len, int send_flags,
 710                  const struct sockaddr *to, socklen_t tolen)
 711 {
 712         gpointer handle = GUINT_TO_POINTER (fd);
 713         int ret;
 714
 715         if (startup_count == 0) {
 716                 WSASetLastError (WSANOTINITIALISED);
 717                 return(SOCKET_ERROR);
 718         }
 719
 720         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 721                 WSASetLastError (WSAENOTSOCK);
 722                 return(SOCKET_ERROR);
 723         }
 724
 725         do {
 726                 ret = sendto (fd, msg, len, send_flags, to, tolen);
 727         } while (ret == -1 && errno == EINTR &&
 728                  !_wapi_thread_cur_apc_pending ());
 729
 730         if (ret == -1) {
 731                 gint errnum = errno;
 732                 DEBUG ("%s: send error: %s", __func__, strerror (errno));
 733
 734                 errnum = errno_to_WSA (errnum, __func__);
 735                 WSASetLastError (errnum);
 736
 737                 return(SOCKET_ERROR);
 738         }
 739         return(ret);
 740 }
 741
 742 static int
 743 _wapi_sendmsg(guint32 fd,  const struct msghdr *msg, int send_flags)
 744 {
 745         gpointer handle = GUINT_TO_POINTER (fd);
 746         int ret;
 747
 748         if (startup_count == 0) {
 749                 WSASetLastError (WSANOTINITIALISED);
 750                 return(SOCKET_ERROR);
 751         }
 752
 753         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 754                 WSASetLastError (WSAENOTSOCK);
 755                 return(SOCKET_ERROR);
 756         }
 757
 758         do {
 759                 ret = sendmsg (fd, msg, send_flags);
 760         } while (ret == -1 && errno == EINTR &&
 761                  !_wapi_thread_cur_apc_pending ());
 762
 763         if (ret == -1) {
 764                 gint errnum = errno;
 765                 DEBUG ("%s: sendmsg error: %s", __func__, strerror (errno));
 766
 767                 errnum = errno_to_WSA (errnum, __func__);
 768                 WSASetLastError (errnum);
 769
 770                 return(SOCKET_ERROR);
 771         }
 772         return(ret);
 773 }
 774
 775 int _wapi_setsockopt(guint32 fd, int level, int optname,
 776                      const void *optval, socklen_t optlen)
 777 {
 778         gpointer handle = GUINT_TO_POINTER (fd);
 779         int ret;
 780         const void *tmp_val;
 781         struct timeval tv;
 782
 783         if (startup_count == 0) {
 784                 WSASetLastError (WSANOTINITIALISED);
 785                 return(SOCKET_ERROR);
 786         }
 787
 788         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 789                 WSASetLastError (WSAENOTSOCK);
 790                 return(SOCKET_ERROR);
 791         }
 792
 793         tmp_val = optval;
 794         if (level == SOL_SOCKET &&
 795             (optname == SO_RCVTIMEO || optname == SO_SNDTIMEO)) {
 796                 int ms = *((int *) optval);
 797                 tv.tv_sec = ms / 1000;
 798                 tv.tv_usec = (ms % 1000) * 1000;        // micro from milli
 799                 tmp_val = &tv;
 800                 optlen = sizeof (tv);
 801 #if defined (__linux__)
 802         } else if (level == SOL_SOCKET &&
 803                    (optname == SO_SNDBUF || optname == SO_RCVBUF)) {
 804                 /* According to socket(7) the Linux kernel doubles the
 805                  * buffer sizes "to allow space for bookkeeping
 806                  * overhead."
 807                  */
 808                 int bufsize = *((int *) optval);
 809
 810                 bufsize /= 2;
 811                 tmp_val = &bufsize;
 812 #endif
 813         }
 814
 815         ret = setsockopt (fd, level, optname, tmp_val, optlen);
 816         if (ret == -1) {
 817                 gint errnum = errno;
 818                 DEBUG ("%s: setsockopt error: %s", __func__,
 819                            strerror (errno));
 820
 821                 errnum = errno_to_WSA (errnum, __func__);
 822                 WSASetLastError (errnum);
 823
 824                 return(SOCKET_ERROR);
 825         }
 826
 827 #if defined (SO_REUSEPORT)
 828         /* BSD's and MacOS X multicast sockets also need SO_REUSEPORT when SO_REUSEADDR is requested.  */
 829         if (level == SOL_SOCKET && optname == SO_REUSEADDR) {
 830                 int type;
 831                 socklen_t type_len = sizeof (type);
 832
 833                 if (!getsockopt (fd, level, SO_TYPE, &type, &type_len)) {
 834                         if (type == SOCK_DGRAM)
 835                                 setsockopt (fd, level, SO_REUSEPORT, tmp_val, optlen);
 836                 }
 837         }
 838 #endif
 839
 840         return(ret);
 841 }
 842
 843 int _wapi_shutdown(guint32 fd, int how)
 844 {
 845         struct _WapiHandle_socket *socket_handle;
 846         gboolean ok;
 847         gpointer handle = GUINT_TO_POINTER (fd);
 848         int ret;
 849
 850         if (startup_count == 0) {
 851                 WSASetLastError (WSANOTINITIALISED);
 852                 return(SOCKET_ERROR);
 853         }
 854
 855         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
 856                 WSASetLastError (WSAENOTSOCK);
 857                 return(SOCKET_ERROR);
 858         }
 859
 860         if (how == SHUT_RD ||
 861             how == SHUT_RDWR) {
 862                 ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
 863                                           (gpointer *)&socket_handle);
 864                 if (ok == FALSE) {
 865                         g_warning ("%s: error looking up socket handle %p",
 866                                    __func__, handle);
 867                         WSASetLastError (WSAENOTSOCK);
 868                         return(SOCKET_ERROR);
 869                 }
 870
 871                 socket_handle->still_readable = 0;
 872         }
 873
 874         ret = shutdown (fd, how);
 875         if (ret == -1) {
 876                 gint errnum = errno;
 877                 DEBUG ("%s: shutdown error: %s", __func__,
 878                            strerror (errno));
 879
 880                 errnum = errno_to_WSA (errnum, __func__);
 881                 WSASetLastError (errnum);
 882
 883                 return(SOCKET_ERROR);
 884         }
 885
 886         return(ret);
 887 }
 888
 889 guint32 _wapi_socket(int domain, int type, int protocol, void *unused,
 890                      guint32 unused2, guint32 unused3)
 891 {
 892         struct _WapiHandle_socket socket_handle = {0};
 893         gpointer handle;
 894         int fd;
 895
 896         socket_handle.domain = domain;
 897         socket_handle.type = type;
 898         socket_handle.protocol = protocol;
 899         socket_handle.still_readable = 1;
 900
 901         fd = socket (domain, type, protocol);
 902         if (fd == -1 && domain == AF_INET && type == SOCK_RAW &&
 903             protocol == 0) {
 904                 /* Retry with protocol == 4 (see bug #54565) */
 905                 socket_handle.protocol = 4;
 906                 fd = socket (AF_INET, SOCK_RAW, 4);
 907         }
 908
 909         if (fd == -1) {
 910                 gint errnum = errno;
 911                 DEBUG ("%s: socket error: %s", __func__, strerror (errno));
 912                 errnum = errno_to_WSA (errnum, __func__);
 913                 WSASetLastError (errnum);
 914
 915                 return(INVALID_SOCKET);
 916         }
 917
 918         if (fd >= _wapi_fd_reserve) {
 919                 DEBUG ("%s: File descriptor is too big (%d >= %d)",
 920                            __func__, fd, _wapi_fd_reserve);
 921
 922                 WSASetLastError (WSASYSCALLFAILURE);
 923                 close (fd);
 924
 925                 return(INVALID_SOCKET);
 926         }
 927
 928         /* .net seems to set this by default for SOCK_STREAM, not for
 929          * SOCK_DGRAM (see bug #36322)
 930          *
 931          * It seems winsock has a rather different idea of what
 932          * SO_REUSEADDR means.  If it's set, then a new socket can be
 933          * bound over an existing listening socket.  There's a new
 934          * windows-specific option called SO_EXCLUSIVEADDRUSE but
 935          * using that means the socket MUST be closed properly, or a
 936          * denial of service can occur.  Luckily for us, winsock
 937          * behaves as though any other system would when SO_REUSEADDR
 938          * is true, so we don't need to do anything else here.  See
 939          * bug 53992.
 940          */
 941         {
 942                 int ret, true = 1;
 943
 944                 ret = setsockopt (fd, SOL_SOCKET, SO_REUSEADDR, &true,
 945                                   sizeof (true));
 946                 if (ret == -1) {
 947                         int errnum = errno;
 948
 949                         DEBUG ("%s: Error setting SO_REUSEADDR", __func__);
 950
 951                         errnum = errno_to_WSA (errnum, __func__);
 952                         WSASetLastError (errnum);
 953
 954                         close (fd);
 955
 956                         return(INVALID_SOCKET);
 957                 }
 958         }
 959
 960
 961         mono_once (&socket_ops_once, socket_ops_init);
 962
 963         handle = _wapi_handle_new_fd (WAPI_HANDLE_SOCKET, fd, &socket_handle);
 964         if (handle == _WAPI_HANDLE_INVALID) {
 965                 g_warning ("%s: error creating socket handle", __func__);
 966                 WSASetLastError (WSASYSCALLFAILURE);
 967                 close (fd);
 968                 return(INVALID_SOCKET);
 969         }
 970
 971         DEBUG ("%s: returning socket handle %p", __func__, handle);
 972
 973         return(fd);
 974 }
 975
 976 struct hostent *_wapi_gethostbyname(const char *hostname)
 977 {
 978         struct hostent *he;
 979
 980         if (startup_count == 0) {
 981                 WSASetLastError (WSANOTINITIALISED);
 982                 return(NULL);
 983         }
 984
 985         he = gethostbyname (hostname);
 986         if (he == NULL) {
 987                 DEBUG ("%s: gethostbyname error: %s", __func__,
 988                            strerror (h_errno));
 989
 990                 switch(h_errno) {
 991                 case HOST_NOT_FOUND:
 992                         WSASetLastError (WSAHOST_NOT_FOUND);
 993                         break;
 994 #if NO_ADDRESS != NO_DATA
 995                 case NO_ADDRESS:
 996 #endif
 997                 case NO_DATA:
 998                         WSASetLastError (WSANO_DATA);
 999                         break;
1000                 case NO_RECOVERY:
1001                         WSASetLastError (WSANO_RECOVERY);
1002                         break;
1003                 case TRY_AGAIN:
1004                         WSASetLastError (WSATRY_AGAIN);
1005                         break;
1006                 default:
1007                         g_warning ("%s: Need to translate %d into winsock error", __func__, h_errno);
1008                         break;
1009                 }
1010         }
1011
1012         return(he);
1013 }
1014
1015 static gboolean socket_disconnect (guint32 fd)
1016 {
1017         struct _WapiHandle_socket *socket_handle;
1018         gboolean ok;
1019         gpointer handle = GUINT_TO_POINTER (fd);
1020         int newsock, ret;
1021
1022         ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
1023                                   (gpointer *)&socket_handle);
1024         if (ok == FALSE) {
1025                 g_warning ("%s: error looking up socket handle %p", __func__,
1026                            handle);
1027                 WSASetLastError (WSAENOTSOCK);
1028                 return(FALSE);
1029         }
1030
1031         newsock = socket (socket_handle->domain, socket_handle->type,
1032                           socket_handle->protocol);
1033         if (newsock == -1) {
1034                 gint errnum = errno;
1035
1036                 DEBUG ("%s: socket error: %s", __func__, strerror (errno));
1037
1038                 errnum = errno_to_WSA (errnum, __func__);
1039                 WSASetLastError (errnum);
1040
1041                 return(FALSE);
1042         }
1043
1044         /* According to Stevens "Advanced Programming in the UNIX
1045          * Environment: UNIX File I/O" dup2() is atomic so there
1046          * should not be a race condition between the old fd being
1047          * closed and the new socket fd being copied over
1048          */
1049         do {
1050                 ret = dup2 (newsock, fd);
1051         } while (ret == -1 && errno == EAGAIN);
1052
1053         if (ret == -1) {
1054                 gint errnum = errno;
1055
1056                 DEBUG ("%s: dup2 error: %s", __func__, strerror (errno));
1057
1058                 errnum = errno_to_WSA (errnum, __func__);
1059                 WSASetLastError (errnum);
1060
1061                 return(FALSE);
1062         }
1063
1064         close (newsock);
1065
1066         return(TRUE);
1067 }
1068
1069 static gboolean wapi_disconnectex (guint32 fd, WapiOverlapped *overlapped,
1070                                    guint32 flags, guint32 reserved)
1071 {
1072         DEBUG ("%s: called on socket %d!", __func__, fd);
1073
1074         if (reserved != 0) {
1075                 WSASetLastError (WSAEINVAL);
1076                 return(FALSE);
1077         }
1078
1079         /* We could check the socket type here and fail unless its
1080          * SOCK_STREAM, SOCK_SEQPACKET or SOCK_RDM (according to msdn)
1081          * if we really wanted to
1082          */
1083
1084         return(socket_disconnect (fd));
1085 }
1086
1087 #define SF_BUFFER_SIZE  16384
1088 static gint
1089 wapi_sendfile (guint32 socket, gpointer fd, guint32 bytes_to_write, guint32 bytes_per_send, guint32 flags)
1090 {
1091 #if defined(HAVE_SENDFILE) && (defined(__linux__) || defined(DARWIN))
1092         gint file = GPOINTER_TO_INT (fd);
1093         gint n;
1094         gint errnum;
1095         gssize res;
1096         struct stat statbuf;
1097
1098         n = fstat (file, &statbuf);
1099         if (n == -1) {
1100                 errnum = errno;
1101                 errnum = errno_to_WSA (errnum, __func__);
1102                 WSASetLastError (errnum);
1103                 return SOCKET_ERROR;
1104         }
1105         do {
1106 #ifdef __linux__
1107                 res = sendfile (socket, file, NULL, statbuf.st_size);
1108 #elif defined(DARWIN)
1109                 /* TODO: header/tail could be sent in the 5th argument */
1110                 /* TODO: Might not send the entire file for non-blocking sockets */
1111                 res = sendfile (file, socket, 0, &statbuf.st_size, NULL, 0);
1112 #endif
1113         } while (res != -1 && (errno == EINTR || errno == EAGAIN) && !_wapi_thread_cur_apc_pending ());
1114         if (res == -1) {
1115                 errnum = errno;
1116                 errnum = errno_to_WSA (errnum, __func__);
1117                 WSASetLastError (errnum);
1118                 return SOCKET_ERROR;
1119         }
1120 #else
1121         /* Default implementation */
1122         gint file = GPOINTER_TO_INT (fd);
1123         gchar *buffer;
1124         gint n;
1125
1126         buffer = g_malloc (SF_BUFFER_SIZE);
1127         do {
1128                 do {
1129                         n = read (file, buffer, SF_BUFFER_SIZE);
1130                 } while (n == -1 && errno == EINTR && !_wapi_thread_cur_apc_pending ());
1131                 if (n == -1)
1132                         break;
1133                 if (n == 0) {
1134                         g_free (buffer);
1135                         return 0; /* We're done reading */
1136                 }
1137                 do {
1138                         n = send (socket, buffer, n, 0); /* short sends? enclose this in a loop? */
1139                 } while (n == -1 && errno == EINTR && !_wapi_thread_cur_apc_pending ());
1140         } while (n != -1);
1141
1142         if (n == -1) {
1143                 gint errnum = errno;
1144                 errnum = errno_to_WSA (errnum, __func__);
1145                 WSASetLastError (errnum);
1146                 g_free (buffer);
1147                 return SOCKET_ERROR;
1148         }
1149         g_free (buffer);
1150 #endif
1151         return 0;
1152 }
1153
1154 gboolean
1155 TransmitFile (guint32 socket, gpointer file, guint32 bytes_to_write, guint32 bytes_per_send, WapiOverlapped *ol,
1156                 WapiTransmitFileBuffers *buffers, guint32 flags)
1157 {
1158         gpointer sock = GUINT_TO_POINTER (socket);
1159         gint ret;
1160
1161         if (startup_count == 0) {
1162                 WSASetLastError (WSANOTINITIALISED);
1163                 return FALSE;
1164         }
1165
1166         if (_wapi_handle_type (sock) != WAPI_HANDLE_SOCKET) {
1167                 WSASetLastError (WSAENOTSOCK);
1168                 return FALSE;
1169         }
1170
1171         /* Write the header */
1172         if (buffers != NULL && buffers->Head != NULL && buffers->HeadLength > 0) {
1173                 ret = _wapi_send (socket, buffers->Head, buffers->HeadLength, 0);
1174                 if (ret == SOCKET_ERROR)
1175                         return FALSE;
1176         }
1177
1178         ret = wapi_sendfile (socket, file, bytes_to_write, bytes_per_send, flags);
1179         if (ret == SOCKET_ERROR)
1180                 return FALSE;
1181
1182         /* Write the tail */
1183         if (buffers != NULL && buffers->Tail != NULL && buffers->TailLength > 0) {
1184                 ret = _wapi_send (socket, buffers->Tail, buffers->TailLength, 0);
1185                 if (ret == SOCKET_ERROR)
1186                         return FALSE;
1187         }
1188
1189         if ((flags & TF_DISCONNECT) == TF_DISCONNECT)
1190                 closesocket (socket);
1191
1192         return TRUE;
1193 }
1194
1195 static struct
1196 {
1197         WapiGuid guid;
1198         gpointer func;
1199 } extension_functions[] = {
1200         {WSAID_DISCONNECTEX, wapi_disconnectex},
1201         {WSAID_TRANSMITFILE, TransmitFile},
1202         {{0}, NULL},
1203 };
1204
1205 int
1206 WSAIoctl (guint32 fd, gint32 command,
1207           gchar *input, gint i_len,
1208           gchar *output, gint o_len, glong *written,
1209           void *unused1, void *unused2)
1210 {
1211         gpointer handle = GUINT_TO_POINTER (fd);
1212         int ret;
1213         gchar *buffer = NULL;
1214
1215         if (startup_count == 0) {
1216                 WSASetLastError (WSANOTINITIALISED);
1217                 return(SOCKET_ERROR);
1218         }
1219
1220         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1221                 WSASetLastError (WSAENOTSOCK);
1222                 return SOCKET_ERROR;
1223         }
1224
1225         if (command == SIO_GET_EXTENSION_FUNCTION_POINTER) {
1226                 int i = 0;
1227                 WapiGuid *guid = (WapiGuid *)input;
1228
1229                 if (i_len < sizeof(WapiGuid)) {
1230                         /* As far as I can tell, windows doesn't
1231                          * actually set an error here...
1232                          */
1233                         WSASetLastError (WSAEINVAL);
1234                         return(SOCKET_ERROR);
1235                 }
1236
1237                 if (o_len < sizeof(gpointer)) {
1238                         /* Or here... */
1239                         WSASetLastError (WSAEINVAL);
1240                         return(SOCKET_ERROR);
1241                 }
1242
1243                 if (output == NULL) {
1244                         /* Or here */
1245                         WSASetLastError (WSAEINVAL);
1246                         return(SOCKET_ERROR);
1247                 }
1248
1249                 while(extension_functions[i].func != NULL) {
1250                         if (!memcmp (guid, &extension_functions[i].guid,
1251                                      sizeof(WapiGuid))) {
1252                                 memcpy (output, &extension_functions[i].func,
1253                                         sizeof(gpointer));
1254                                 *written = sizeof(gpointer);
1255                                 return(0);
1256                         }
1257
1258                         i++;
1259                 }
1260
1261                 WSASetLastError (WSAEINVAL);
1262                 return(SOCKET_ERROR);
1263         }
1264
1265         if (command == SIO_KEEPALIVE_VALS) {
1266                 uint32_t onoff;
1267                 uint32_t keepalivetime;
1268                 uint32_t keepaliveinterval;
1269
1270                 if (i_len < (3 * sizeof (uint32_t))) {
1271                         WSASetLastError (WSAEINVAL);
1272                         return SOCKET_ERROR;
1273                 }
1274                 memcpy (&onoff, input, sizeof (uint32_t));
1275                 memcpy (&keepalivetime, input + sizeof (uint32_t), sizeof (uint32_t));
1276                 memcpy (&keepaliveinterval, input + 2 * sizeof (uint32_t), sizeof (uint32_t));
1277                 ret = setsockopt (fd, SOL_SOCKET, SO_KEEPALIVE, &onoff, sizeof (uint32_t));
1278                 if (ret < 0) {
1279                         gint errnum = errno;
1280                         errnum = errno_to_WSA (errnum, __func__);
1281                         WSASetLastError (errnum);
1282                         return SOCKET_ERROR;
1283                 }
1284                 if (onoff != 0) {
1285 #if defined(TCP_KEEPIDLE) && defined(TCP_KEEPINTVL)
1286                         /* Values are in ms, but we need s */
1287                         uint32_t rem;
1288
1289                         /* keepalivetime and keepaliveinterval are > 0 (checked in managed code) */
1290                         rem = keepalivetime % 1000;
1291                         keepalivetime /= 1000;
1292                         if (keepalivetime == 0 || rem >= 500)
1293                                 keepalivetime++;
1294                         ret = setsockopt (fd, SOL_TCP, TCP_KEEPIDLE, &keepalivetime, sizeof (uint32_t));
1295                         if (ret == 0) {
1296                                 rem = keepaliveinterval % 1000;
1297                                 keepaliveinterval /= 1000;
1298                                 if (keepaliveinterval == 0 || rem >= 500)
1299                                         keepaliveinterval++;
1300                                 ret = setsockopt (fd, SOL_TCP, TCP_KEEPINTVL, &keepaliveinterval, sizeof (uint32_t));
1301                         }
1302                         if (ret != 0) {
1303                                 gint errnum = errno;
1304                                 errnum = errno_to_WSA (errnum, __func__);
1305                                 WSASetLastError (errnum);
1306                                 return SOCKET_ERROR;
1307                         }
1308                         return 0;
1309 #endif
1310                 }
1311                 return 0;
1312         }
1313
1314         if (i_len > 0) {
1315                 buffer = g_memdup (input, i_len);
1316         }
1317
1318         ret = ioctl (fd, command, buffer);
1319         if (ret == -1) {
1320                 gint errnum = errno;
1321                 DEBUG("%s: WSAIoctl error: %s", __func__,
1322                           strerror (errno));
1323
1324                 errnum = errno_to_WSA (errnum, __func__);
1325                 WSASetLastError (errnum);
1326                 g_free (buffer);
1327
1328                 return(SOCKET_ERROR);
1329         }
1330
1331         if (buffer == NULL) {
1332                 *written = 0;
1333         } else {
1334                 /* We just copy the buffer to the output. Some ioctls
1335                  * don't even output any data, but, well...
1336                  *
1337                  * NB windows returns WSAEFAULT if o_len is too small
1338                  */
1339                 i_len = (i_len > o_len) ? o_len : i_len;
1340
1341                 if (i_len > 0 && output != NULL) {
1342                         memcpy (output, buffer, i_len);
1343                 }
1344
1345                 g_free (buffer);
1346                 *written = i_len;
1347         }
1348
1349         return(0);
1350 }
1351
1352 #ifndef PLATFORM_PORT_PROVIDES_IOCTLSOCKET
1353 int ioctlsocket(guint32 fd, gint32 command, gpointer arg)
1354 {
1355         gpointer handle = GUINT_TO_POINTER (fd);
1356         int ret;
1357
1358         if (startup_count == 0) {
1359                 WSASetLastError (WSANOTINITIALISED);
1360                 return(SOCKET_ERROR);
1361         }
1362
1363         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1364                 WSASetLastError (WSAENOTSOCK);
1365                 return(SOCKET_ERROR);
1366         }
1367
1368         switch(command){
1369                 case FIONBIO:
1370 #ifdef O_NONBLOCK
1371                         /* This works better than ioctl(...FIONBIO...)
1372                          * on Linux (it causes connect to return
1373                          * EINPROGRESS, but the ioctl doesn't seem to)
1374                          */
1375                         ret = fcntl(fd, F_GETFL, 0);
1376                         if (ret != -1) {
1377                                 if (*(gboolean *)arg) {
1378                                         ret |= O_NONBLOCK;
1379                                 } else {
1380                                         ret &= ~O_NONBLOCK;
1381                                 }
1382                                 ret = fcntl(fd, F_SETFL, ret);
1383                         }
1384                         break;
1385 #endif /* O_NONBLOCK */
1386                         /* Unused in Mono */
1387                 case SIOCATMARK:
1388                         ret = ioctl (fd, command, arg);
1389                         break;
1390
1391                 case FIONREAD:
1392                 {
1393 #if defined (PLATFORM_MACOSX)
1394
1395                         // ioctl (fd, FIONREAD, XXX) returns the size of
1396                         // the UDP header as well on
1397                         // Darwin.
1398                         //
1399                         // Use getsockopt SO_NREAD instead to get the
1400                         // right values for TCP and UDP.
1401                         //
1402                         // ai_canonname can be null in some cases on darwin, where the runtime assumes it will
1403                         // be the value of the ip buffer.
1404
1405                         socklen_t optlen = sizeof (int);
1406                         ret = getsockopt (fd, SOL_SOCKET, SO_NREAD, arg, &optlen);
1407 #else
1408                         ret = ioctl (fd, command, arg);
1409 #endif
1410                         break;
1411                 }
1412                 default:
1413                         WSASetLastError (WSAEINVAL);
1414                         return(SOCKET_ERROR);
1415         }
1416
1417         if (ret == -1) {
1418                 gint errnum = errno;
1419                 DEBUG ("%s: ioctl error: %s", __func__, strerror (errno));
1420
1421                 errnum = errno_to_WSA (errnum, __func__);
1422                 WSASetLastError (errnum);
1423
1424                 return(SOCKET_ERROR);
1425         }
1426
1427         return(0);
1428 }
1429
1430 int _wapi_select(int nfds G_GNUC_UNUSED, fd_set *readfds, fd_set *writefds,
1431                  fd_set *exceptfds, struct timeval *timeout)
1432 {
1433         int ret, maxfd;
1434
1435         if (startup_count == 0) {
1436                 WSASetLastError (WSANOTINITIALISED);
1437                 return(SOCKET_ERROR);
1438         }
1439
1440         for (maxfd = FD_SETSIZE-1; maxfd >= 0; maxfd--) {
1441                 if ((readfds && FD_ISSET (maxfd, readfds)) ||
1442                     (writefds && FD_ISSET (maxfd, writefds)) ||
1443                     (exceptfds && FD_ISSET (maxfd, exceptfds))) {
1444                         break;
1445                 }
1446         }
1447
1448         if (maxfd == -1) {
1449                 WSASetLastError (WSAEINVAL);
1450                 return(SOCKET_ERROR);
1451         }
1452
1453         do {
1454                 ret = select(maxfd + 1, readfds, writefds, exceptfds,
1455                              timeout);
1456         } while (ret == -1 && errno == EINTR &&
1457                  !_wapi_thread_cur_apc_pending ());
1458
1459         if (ret == -1) {
1460                 gint errnum = errno;
1461                 DEBUG ("%s: select error: %s", __func__, strerror (errno));
1462                 errnum = errno_to_WSA (errnum, __func__);
1463                 WSASetLastError (errnum);
1464
1465                 return(SOCKET_ERROR);
1466         }
1467
1468         return(ret);
1469 }
1470
1471 void _wapi_FD_CLR(guint32 fd, fd_set *set)
1472 {
1473         gpointer handle = GUINT_TO_POINTER (fd);
1474
1475         if (fd >= FD_SETSIZE) {
1476                 WSASetLastError (WSAEINVAL);
1477                 return;
1478         }
1479
1480         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1481                 WSASetLastError (WSAENOTSOCK);
1482                 return;
1483         }
1484
1485         FD_CLR (fd, set);
1486 }
1487
1488 int _wapi_FD_ISSET(guint32 fd, fd_set *set)
1489 {
1490         gpointer handle = GUINT_TO_POINTER (fd);
1491
1492         if (fd >= FD_SETSIZE) {
1493                 WSASetLastError (WSAEINVAL);
1494                 return(0);
1495         }
1496
1497         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1498                 WSASetLastError (WSAENOTSOCK);
1499                 return(0);
1500         }
1501
1502         return(FD_ISSET (fd, set));
1503 }
1504
1505 void _wapi_FD_SET(guint32 fd, fd_set *set)
1506 {
1507         gpointer handle = GUINT_TO_POINTER (fd);
1508
1509         if (fd >= FD_SETSIZE) {
1510                 WSASetLastError (WSAEINVAL);
1511                 return;
1512         }
1513
1514         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1515                 WSASetLastError (WSAENOTSOCK);
1516                 return;
1517         }
1518
1519         FD_SET (fd, set);
1520 }
1521 #endif
1522
1523 static void
1524 wsabuf_to_msghdr (WapiWSABuf *buffers, guint32 count, struct msghdr *hdr)
1525 {
1526         guint32 i;
1527
1528         memset (hdr, 0, sizeof (struct msghdr));
1529         hdr->msg_iovlen = count;
1530         hdr->msg_iov = g_new0 (struct iovec, count);
1531         for (i = 0; i < count; i++) {
1532                 hdr->msg_iov [i].iov_base = buffers [i].buf;
1533                 hdr->msg_iov [i].iov_len  = buffers [i].len;
1534         }
1535 }
1536
1537 static void
1538 msghdr_iov_free (struct msghdr *hdr)
1539 {
1540         g_free (hdr->msg_iov);
1541 }
1542
1543 int WSARecv (guint32 fd, WapiWSABuf *buffers, guint32 count, guint32 *received,
1544              guint32 *flags, WapiOverlapped *overlapped,
1545              WapiOverlappedCB *complete)
1546 {
1547         int ret;
1548         struct msghdr hdr;
1549
1550         g_assert (overlapped == NULL);
1551         g_assert (complete == NULL);
1552
1553         wsabuf_to_msghdr (buffers, count, &hdr);
1554         ret = _wapi_recvmsg (fd, &hdr, *flags);
1555         msghdr_iov_free (&hdr);
1556
1557         if(ret == SOCKET_ERROR) {
1558                 return(ret);
1559         }
1560
1561         *received = ret;
1562         *flags = hdr.msg_flags;
1563
1564         return(0);
1565 }
1566
1567 int WSASend (guint32 fd, WapiWSABuf *buffers, guint32 count, guint32 *sent,
1568              guint32 flags, WapiOverlapped *overlapped,
1569              WapiOverlappedCB *complete)
1570 {
1571         int ret;
1572         struct msghdr hdr;
1573
1574         g_assert (overlapped == NULL);
1575         g_assert (complete == NULL);
1576
1577         wsabuf_to_msghdr (buffers, count, &hdr);
1578         ret = _wapi_sendmsg (fd, &hdr, flags);
1579         msghdr_iov_free (&hdr);
1580
1581         if(ret == SOCKET_ERROR)
1582                 return ret;
1583
1584         *sent = ret;
1585         return 0;
1586 }
1587
1588 #endif /* ifndef DISABLE_SOCKETS */