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                 // https://bugzilla.novell.com/show_bug.cgi?id=MONO54565
 906                 socket_handle.protocol = 4;
 907                 fd = socket (AF_INET, SOCK_RAW, 4);
 908         }
 909
 910         if (fd == -1) {
 911                 gint errnum = errno;
 912                 DEBUG ("%s: socket error: %s", __func__, strerror (errno));
 913                 errnum = errno_to_WSA (errnum, __func__);
 914                 WSASetLastError (errnum);
 915
 916                 return(INVALID_SOCKET);
 917         }
 918
 919         if (fd >= _wapi_fd_reserve) {
 920                 DEBUG ("%s: File descriptor is too big (%d >= %d)",
 921                            __func__, fd, _wapi_fd_reserve);
 922
 923                 WSASetLastError (WSASYSCALLFAILURE);
 924                 close (fd);
 925
 926                 return(INVALID_SOCKET);
 927         }
 928
 929         /* .net seems to set this by default for SOCK_STREAM, not for
 930          * SOCK_DGRAM (see bug #36322)
 931          * https://bugzilla.novell.com/show_bug.cgi?id=MONO36322
 932          *
 933          * It seems winsock has a rather different idea of what
 934          * SO_REUSEADDR means.  If it's set, then a new socket can be
 935          * bound over an existing listening socket.  There's a new
 936          * windows-specific option called SO_EXCLUSIVEADDRUSE but
 937          * using that means the socket MUST be closed properly, or a
 938          * denial of service can occur.  Luckily for us, winsock
 939          * behaves as though any other system would when SO_REUSEADDR
 940          * is true, so we don't need to do anything else here.  See
 941          * bug 53992.
 942          * https://bugzilla.novell.com/show_bug.cgi?id=MONO53992
 943          */
 944         {
 945                 int ret, true = 1;
 946
 947                 ret = setsockopt (fd, SOL_SOCKET, SO_REUSEADDR, &true,
 948                                   sizeof (true));
 949                 if (ret == -1) {
 950                         int errnum = errno;
 951
 952                         DEBUG ("%s: Error setting SO_REUSEADDR", __func__);
 953
 954                         errnum = errno_to_WSA (errnum, __func__);
 955                         WSASetLastError (errnum);
 956
 957                         close (fd);
 958
 959                         return(INVALID_SOCKET);
 960                 }
 961         }
 962
 963
 964         mono_once (&socket_ops_once, socket_ops_init);
 965
 966         handle = _wapi_handle_new_fd (WAPI_HANDLE_SOCKET, fd, &socket_handle);
 967         if (handle == _WAPI_HANDLE_INVALID) {
 968                 g_warning ("%s: error creating socket handle", __func__);
 969                 WSASetLastError (WSASYSCALLFAILURE);
 970                 close (fd);
 971                 return(INVALID_SOCKET);
 972         }
 973
 974         DEBUG ("%s: returning socket handle %p", __func__, handle);
 975
 976         return(fd);
 977 }
 978
 979 struct hostent *_wapi_gethostbyname(const char *hostname)
 980 {
 981         struct hostent *he;
 982
 983         if (startup_count == 0) {
 984                 WSASetLastError (WSANOTINITIALISED);
 985                 return(NULL);
 986         }
 987
 988         he = gethostbyname (hostname);
 989         if (he == NULL) {
 990                 DEBUG ("%s: gethostbyname error: %s", __func__,
 991                            strerror (h_errno));
 992
 993                 switch(h_errno) {
 994                 case HOST_NOT_FOUND:
 995                         WSASetLastError (WSAHOST_NOT_FOUND);
 996                         break;
 997 #if NO_ADDRESS != NO_DATA
 998                 case NO_ADDRESS:
 999 #endif
1000                 case NO_DATA:
1001                         WSASetLastError (WSANO_DATA);
1002                         break;
1003                 case NO_RECOVERY:
1004                         WSASetLastError (WSANO_RECOVERY);
1005                         break;
1006                 case TRY_AGAIN:
1007                         WSASetLastError (WSATRY_AGAIN);
1008                         break;
1009                 default:
1010                         g_warning ("%s: Need to translate %d into winsock error", __func__, h_errno);
1011                         break;
1012                 }
1013         }
1014
1015         return(he);
1016 }
1017
1018 static gboolean socket_disconnect (guint32 fd)
1019 {
1020         struct _WapiHandle_socket *socket_handle;
1021         gboolean ok;
1022         gpointer handle = GUINT_TO_POINTER (fd);
1023         int newsock, ret;
1024
1025         ok = _wapi_lookup_handle (handle, WAPI_HANDLE_SOCKET,
1026                                   (gpointer *)&socket_handle);
1027         if (ok == FALSE) {
1028                 g_warning ("%s: error looking up socket handle %p", __func__,
1029                            handle);
1030                 WSASetLastError (WSAENOTSOCK);
1031                 return(FALSE);
1032         }
1033
1034         newsock = socket (socket_handle->domain, socket_handle->type,
1035                           socket_handle->protocol);
1036         if (newsock == -1) {
1037                 gint errnum = errno;
1038
1039                 DEBUG ("%s: socket error: %s", __func__, strerror (errno));
1040
1041                 errnum = errno_to_WSA (errnum, __func__);
1042                 WSASetLastError (errnum);
1043
1044                 return(FALSE);
1045         }
1046
1047         /* According to Stevens "Advanced Programming in the UNIX
1048          * Environment: UNIX File I/O" dup2() is atomic so there
1049          * should not be a race condition between the old fd being
1050          * closed and the new socket fd being copied over
1051          */
1052         do {
1053                 ret = dup2 (newsock, fd);
1054         } while (ret == -1 && errno == EAGAIN);
1055
1056         if (ret == -1) {
1057                 gint errnum = errno;
1058
1059                 DEBUG ("%s: dup2 error: %s", __func__, strerror (errno));
1060
1061                 errnum = errno_to_WSA (errnum, __func__);
1062                 WSASetLastError (errnum);
1063
1064                 return(FALSE);
1065         }
1066
1067         close (newsock);
1068
1069         return(TRUE);
1070 }
1071
1072 static gboolean wapi_disconnectex (guint32 fd, WapiOverlapped *overlapped,
1073                                    guint32 flags, guint32 reserved)
1074 {
1075         DEBUG ("%s: called on socket %d!", __func__, fd);
1076
1077         if (reserved != 0) {
1078                 WSASetLastError (WSAEINVAL);
1079                 return(FALSE);
1080         }
1081
1082         /* We could check the socket type here and fail unless its
1083          * SOCK_STREAM, SOCK_SEQPACKET or SOCK_RDM (according to msdn)
1084          * if we really wanted to
1085          */
1086
1087         return(socket_disconnect (fd));
1088 }
1089
1090 #define SF_BUFFER_SIZE  16384
1091 static gint
1092 wapi_sendfile (guint32 socket, gpointer fd, guint32 bytes_to_write, guint32 bytes_per_send, guint32 flags)
1093 {
1094 #if defined(HAVE_SENDFILE) && (defined(__linux__) || defined(DARWIN))
1095         gint file = GPOINTER_TO_INT (fd);
1096         gint n;
1097         gint errnum;
1098         gssize res;
1099         struct stat statbuf;
1100
1101         n = fstat (file, &statbuf);
1102         if (n == -1) {
1103                 errnum = errno;
1104                 errnum = errno_to_WSA (errnum, __func__);
1105                 WSASetLastError (errnum);
1106                 return SOCKET_ERROR;
1107         }
1108         do {
1109 #ifdef __linux__
1110                 res = sendfile (socket, file, NULL, statbuf.st_size);
1111 #elif defined(DARWIN)
1112                 /* TODO: header/tail could be sent in the 5th argument */
1113                 /* TODO: Might not send the entire file for non-blocking sockets */
1114                 res = sendfile (file, socket, 0, &statbuf.st_size, NULL, 0);
1115 #endif
1116         } while (res != -1 && (errno == EINTR || errno == EAGAIN) && !_wapi_thread_cur_apc_pending ());
1117         if (res == -1) {
1118                 errnum = errno;
1119                 errnum = errno_to_WSA (errnum, __func__);
1120                 WSASetLastError (errnum);
1121                 return SOCKET_ERROR;
1122         }
1123 #else
1124         /* Default implementation */
1125         gint file = GPOINTER_TO_INT (fd);
1126         gchar *buffer;
1127         gint n;
1128
1129         buffer = g_malloc (SF_BUFFER_SIZE);
1130         do {
1131                 do {
1132                         n = read (file, buffer, SF_BUFFER_SIZE);
1133                 } while (n == -1 && errno == EINTR && !_wapi_thread_cur_apc_pending ());
1134                 if (n == -1)
1135                         break;
1136                 if (n == 0) {
1137                         g_free (buffer);
1138                         return 0; /* We're done reading */
1139                 }
1140                 do {
1141                         n = send (socket, buffer, n, 0); /* short sends? enclose this in a loop? */
1142                 } while (n == -1 && errno == EINTR && !_wapi_thread_cur_apc_pending ());
1143         } while (n != -1);
1144
1145         if (n == -1) {
1146                 gint errnum = errno;
1147                 errnum = errno_to_WSA (errnum, __func__);
1148                 WSASetLastError (errnum);
1149                 g_free (buffer);
1150                 return SOCKET_ERROR;
1151         }
1152         g_free (buffer);
1153 #endif
1154         return 0;
1155 }
1156
1157 gboolean
1158 TransmitFile (guint32 socket, gpointer file, guint32 bytes_to_write, guint32 bytes_per_send, WapiOverlapped *ol,
1159                 WapiTransmitFileBuffers *buffers, guint32 flags)
1160 {
1161         gpointer sock = GUINT_TO_POINTER (socket);
1162         gint ret;
1163
1164         if (startup_count == 0) {
1165                 WSASetLastError (WSANOTINITIALISED);
1166                 return FALSE;
1167         }
1168
1169         if (_wapi_handle_type (sock) != WAPI_HANDLE_SOCKET) {
1170                 WSASetLastError (WSAENOTSOCK);
1171                 return FALSE;
1172         }
1173
1174         /* Write the header */
1175         if (buffers != NULL && buffers->Head != NULL && buffers->HeadLength > 0) {
1176                 ret = _wapi_send (socket, buffers->Head, buffers->HeadLength, 0);
1177                 if (ret == SOCKET_ERROR)
1178                         return FALSE;
1179         }
1180
1181         ret = wapi_sendfile (socket, file, bytes_to_write, bytes_per_send, flags);
1182         if (ret == SOCKET_ERROR)
1183                 return FALSE;
1184
1185         /* Write the tail */
1186         if (buffers != NULL && buffers->Tail != NULL && buffers->TailLength > 0) {
1187                 ret = _wapi_send (socket, buffers->Tail, buffers->TailLength, 0);
1188                 if (ret == SOCKET_ERROR)
1189                         return FALSE;
1190         }
1191
1192         if ((flags & TF_DISCONNECT) == TF_DISCONNECT)
1193                 closesocket (socket);
1194
1195         return TRUE;
1196 }
1197
1198 static struct
1199 {
1200         WapiGuid guid;
1201         gpointer func;
1202 } extension_functions[] = {
1203         {WSAID_DISCONNECTEX, wapi_disconnectex},
1204         {WSAID_TRANSMITFILE, TransmitFile},
1205         {{0}, NULL},
1206 };
1207
1208 int
1209 WSAIoctl (guint32 fd, gint32 command,
1210           gchar *input, gint i_len,
1211           gchar *output, gint o_len, glong *written,
1212           void *unused1, void *unused2)
1213 {
1214         gpointer handle = GUINT_TO_POINTER (fd);
1215         int ret;
1216         gchar *buffer = NULL;
1217
1218         if (startup_count == 0) {
1219                 WSASetLastError (WSANOTINITIALISED);
1220                 return(SOCKET_ERROR);
1221         }
1222
1223         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1224                 WSASetLastError (WSAENOTSOCK);
1225                 return SOCKET_ERROR;
1226         }
1227
1228         if (command == SIO_GET_EXTENSION_FUNCTION_POINTER) {
1229                 int i = 0;
1230                 WapiGuid *guid = (WapiGuid *)input;
1231
1232                 if (i_len < sizeof(WapiGuid)) {
1233                         /* As far as I can tell, windows doesn't
1234                          * actually set an error here...
1235                          */
1236                         WSASetLastError (WSAEINVAL);
1237                         return(SOCKET_ERROR);
1238                 }
1239
1240                 if (o_len < sizeof(gpointer)) {
1241                         /* Or here... */
1242                         WSASetLastError (WSAEINVAL);
1243                         return(SOCKET_ERROR);
1244                 }
1245
1246                 if (output == NULL) {
1247                         /* Or here */
1248                         WSASetLastError (WSAEINVAL);
1249                         return(SOCKET_ERROR);
1250                 }
1251
1252                 while(extension_functions[i].func != NULL) {
1253                         if (!memcmp (guid, &extension_functions[i].guid,
1254                                      sizeof(WapiGuid))) {
1255                                 memcpy (output, &extension_functions[i].func,
1256                                         sizeof(gpointer));
1257                                 *written = sizeof(gpointer);
1258                                 return(0);
1259                         }
1260
1261                         i++;
1262                 }
1263
1264                 WSASetLastError (WSAEINVAL);
1265                 return(SOCKET_ERROR);
1266         }
1267
1268         if (command == SIO_KEEPALIVE_VALS) {
1269                 uint32_t onoff;
1270                 uint32_t keepalivetime;
1271                 uint32_t keepaliveinterval;
1272
1273                 if (i_len < (3 * sizeof (uint32_t))) {
1274                         WSASetLastError (WSAEINVAL);
1275                         return SOCKET_ERROR;
1276                 }
1277                 memcpy (&onoff, input, sizeof (uint32_t));
1278                 memcpy (&keepalivetime, input + sizeof (uint32_t), sizeof (uint32_t));
1279                 memcpy (&keepaliveinterval, input + 2 * sizeof (uint32_t), sizeof (uint32_t));
1280                 ret = setsockopt (fd, SOL_SOCKET, SO_KEEPALIVE, &onoff, sizeof (uint32_t));
1281                 if (ret < 0) {
1282                         gint errnum = errno;
1283                         errnum = errno_to_WSA (errnum, __func__);
1284                         WSASetLastError (errnum);
1285                         return SOCKET_ERROR;
1286                 }
1287                 if (onoff != 0) {
1288 #if defined(TCP_KEEPIDLE) && defined(TCP_KEEPINTVL)
1289                         /* Values are in ms, but we need s */
1290                         uint32_t rem;
1291
1292                         /* keepalivetime and keepaliveinterval are > 0 (checked in managed code) */
1293                         rem = keepalivetime % 1000;
1294                         keepalivetime /= 1000;
1295                         if (keepalivetime == 0 || rem >= 500)
1296                                 keepalivetime++;
1297                         ret = setsockopt (fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepalivetime, sizeof (uint32_t));
1298                         if (ret == 0) {
1299                                 rem = keepaliveinterval % 1000;
1300                                 keepaliveinterval /= 1000;
1301                                 if (keepaliveinterval == 0 || rem >= 500)
1302                                         keepaliveinterval++;
1303                                 ret = setsockopt (fd, IPPROTO_TCP, TCP_KEEPINTVL, &keepaliveinterval, sizeof (uint32_t));
1304                         }
1305                         if (ret != 0) {
1306                                 gint errnum = errno;
1307                                 errnum = errno_to_WSA (errnum, __func__);
1308                                 WSASetLastError (errnum);
1309                                 return SOCKET_ERROR;
1310                         }
1311                         return 0;
1312 #endif
1313                 }
1314                 return 0;
1315         }
1316
1317         if (i_len > 0) {
1318                 buffer = g_memdup (input, i_len);
1319         }
1320
1321         ret = ioctl (fd, command, buffer);
1322         if (ret == -1) {
1323                 gint errnum = errno;
1324                 DEBUG("%s: WSAIoctl error: %s", __func__,
1325                           strerror (errno));
1326
1327                 errnum = errno_to_WSA (errnum, __func__);
1328                 WSASetLastError (errnum);
1329                 g_free (buffer);
1330
1331                 return(SOCKET_ERROR);
1332         }
1333
1334         if (buffer == NULL) {
1335                 *written = 0;
1336         } else {
1337                 /* We just copy the buffer to the output. Some ioctls
1338                  * don't even output any data, but, well...
1339                  *
1340                  * NB windows returns WSAEFAULT if o_len is too small
1341                  */
1342                 i_len = (i_len > o_len) ? o_len : i_len;
1343
1344                 if (i_len > 0 && output != NULL) {
1345                         memcpy (output, buffer, i_len);
1346                 }
1347
1348                 g_free (buffer);
1349                 *written = i_len;
1350         }
1351
1352         return(0);
1353 }
1354
1355 #ifndef PLATFORM_PORT_PROVIDES_IOCTLSOCKET
1356 int ioctlsocket(guint32 fd, gint32 command, gpointer arg)
1357 {
1358         gpointer handle = GUINT_TO_POINTER (fd);
1359         int ret;
1360
1361         if (startup_count == 0) {
1362                 WSASetLastError (WSANOTINITIALISED);
1363                 return(SOCKET_ERROR);
1364         }
1365
1366         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1367                 WSASetLastError (WSAENOTSOCK);
1368                 return(SOCKET_ERROR);
1369         }
1370
1371         switch(command){
1372                 case FIONBIO:
1373 #ifdef O_NONBLOCK
1374                         /* This works better than ioctl(...FIONBIO...)
1375                          * on Linux (it causes connect to return
1376                          * EINPROGRESS, but the ioctl doesn't seem to)
1377                          */
1378                         ret = fcntl(fd, F_GETFL, 0);
1379                         if (ret != -1) {
1380                                 if (*(gboolean *)arg) {
1381                                         ret |= O_NONBLOCK;
1382                                 } else {
1383                                         ret &= ~O_NONBLOCK;
1384                                 }
1385                                 ret = fcntl(fd, F_SETFL, ret);
1386                         }
1387                         break;
1388 #endif /* O_NONBLOCK */
1389                         /* Unused in Mono */
1390                 case SIOCATMARK:
1391                         ret = ioctl (fd, command, arg);
1392                         break;
1393
1394                 case FIONREAD:
1395                 {
1396 #if defined (PLATFORM_MACOSX)
1397
1398                         // ioctl (fd, FIONREAD, XXX) returns the size of
1399                         // the UDP header as well on
1400                         // Darwin.
1401                         //
1402                         // Use getsockopt SO_NREAD instead to get the
1403                         // right values for TCP and UDP.
1404                         //
1405                         // ai_canonname can be null in some cases on darwin, where the runtime assumes it will
1406                         // be the value of the ip buffer.
1407
1408                         socklen_t optlen = sizeof (int);
1409                         ret = getsockopt (fd, SOL_SOCKET, SO_NREAD, arg, &optlen);
1410 #else
1411                         ret = ioctl (fd, command, arg);
1412 #endif
1413                         break;
1414                 }
1415                 default:
1416                         WSASetLastError (WSAEINVAL);
1417                         return(SOCKET_ERROR);
1418         }
1419
1420         if (ret == -1) {
1421                 gint errnum = errno;
1422                 DEBUG ("%s: ioctl error: %s", __func__, strerror (errno));
1423
1424                 errnum = errno_to_WSA (errnum, __func__);
1425                 WSASetLastError (errnum);
1426
1427                 return(SOCKET_ERROR);
1428         }
1429
1430         return(0);
1431 }
1432
1433 int _wapi_select(int nfds G_GNUC_UNUSED, fd_set *readfds, fd_set *writefds,
1434                  fd_set *exceptfds, struct timeval *timeout)
1435 {
1436         int ret, maxfd;
1437
1438         if (startup_count == 0) {
1439                 WSASetLastError (WSANOTINITIALISED);
1440                 return(SOCKET_ERROR);
1441         }
1442
1443         for (maxfd = FD_SETSIZE-1; maxfd >= 0; maxfd--) {
1444                 if ((readfds && FD_ISSET (maxfd, readfds)) ||
1445                     (writefds && FD_ISSET (maxfd, writefds)) ||
1446                     (exceptfds && FD_ISSET (maxfd, exceptfds))) {
1447                         break;
1448                 }
1449         }
1450
1451         if (maxfd == -1) {
1452                 WSASetLastError (WSAEINVAL);
1453                 return(SOCKET_ERROR);
1454         }
1455
1456         do {
1457                 ret = select(maxfd + 1, readfds, writefds, exceptfds,
1458                              timeout);
1459         } while (ret == -1 && errno == EINTR &&
1460                  !_wapi_thread_cur_apc_pending ());
1461
1462         if (ret == -1) {
1463                 gint errnum = errno;
1464                 DEBUG ("%s: select error: %s", __func__, strerror (errno));
1465                 errnum = errno_to_WSA (errnum, __func__);
1466                 WSASetLastError (errnum);
1467
1468                 return(SOCKET_ERROR);
1469         }
1470
1471         return(ret);
1472 }
1473
1474 void _wapi_FD_CLR(guint32 fd, fd_set *set)
1475 {
1476         gpointer handle = GUINT_TO_POINTER (fd);
1477
1478         if (fd >= FD_SETSIZE) {
1479                 WSASetLastError (WSAEINVAL);
1480                 return;
1481         }
1482
1483         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1484                 WSASetLastError (WSAENOTSOCK);
1485                 return;
1486         }
1487
1488         FD_CLR (fd, set);
1489 }
1490
1491 int _wapi_FD_ISSET(guint32 fd, fd_set *set)
1492 {
1493         gpointer handle = GUINT_TO_POINTER (fd);
1494
1495         if (fd >= FD_SETSIZE) {
1496                 WSASetLastError (WSAEINVAL);
1497                 return(0);
1498         }
1499
1500         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1501                 WSASetLastError (WSAENOTSOCK);
1502                 return(0);
1503         }
1504
1505         return(FD_ISSET (fd, set));
1506 }
1507
1508 void _wapi_FD_SET(guint32 fd, fd_set *set)
1509 {
1510         gpointer handle = GUINT_TO_POINTER (fd);
1511
1512         if (fd >= FD_SETSIZE) {
1513                 WSASetLastError (WSAEINVAL);
1514                 return;
1515         }
1516
1517         if (_wapi_handle_type (handle) != WAPI_HANDLE_SOCKET) {
1518                 WSASetLastError (WSAENOTSOCK);
1519                 return;
1520         }
1521
1522         FD_SET (fd, set);
1523 }
1524 #endif
1525
1526 static void
1527 wsabuf_to_msghdr (WapiWSABuf *buffers, guint32 count, struct msghdr *hdr)
1528 {
1529         guint32 i;
1530
1531         memset (hdr, 0, sizeof (struct msghdr));
1532         hdr->msg_iovlen = count;
1533         hdr->msg_iov = g_new0 (struct iovec, count);
1534         for (i = 0; i < count; i++) {
1535                 hdr->msg_iov [i].iov_base = buffers [i].buf;
1536                 hdr->msg_iov [i].iov_len  = buffers [i].len;
1537         }
1538 }
1539
1540 static void
1541 msghdr_iov_free (struct msghdr *hdr)
1542 {
1543         g_free (hdr->msg_iov);
1544 }
1545
1546 int WSARecv (guint32 fd, WapiWSABuf *buffers, guint32 count, guint32 *received,
1547              guint32 *flags, WapiOverlapped *overlapped,
1548              WapiOverlappedCB *complete)
1549 {
1550         int ret;
1551         struct msghdr hdr;
1552
1553         g_assert (overlapped == NULL);
1554         g_assert (complete == NULL);
1555
1556         wsabuf_to_msghdr (buffers, count, &hdr);
1557         ret = _wapi_recvmsg (fd, &hdr, *flags);
1558         msghdr_iov_free (&hdr);
1559
1560         if(ret == SOCKET_ERROR) {
1561                 return(ret);
1562         }
1563
1564         *received = ret;
1565         *flags = hdr.msg_flags;
1566
1567         return(0);
1568 }
1569
1570 int WSASend (guint32 fd, WapiWSABuf *buffers, guint32 count, guint32 *sent,
1571              guint32 flags, WapiOverlapped *overlapped,
1572              WapiOverlappedCB *complete)
1573 {
1574         int ret;
1575         struct msghdr hdr;
1576
1577         g_assert (overlapped == NULL);
1578         g_assert (complete == NULL);
1579
1580         wsabuf_to_msghdr (buffers, count, &hdr);
1581         ret = _wapi_sendmsg (fd, &hdr, flags);
1582         msghdr_iov_free (&hdr);
1583
1584         if(ret == SOCKET_ERROR)
1585                 return ret;
1586
1587         *sent = ret;
1588         return 0;
1589 }
1590
1591 #endif /* ifndef DISABLE_SOCKETS */