nonblock.c 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. /***************************************************************************
  2. * _ _ ____ _
  3. * Project ___| | | | _ \| |
  4. * / __| | | | |_) | |
  5. * | (__| |_| | _ <| |___
  6. * \___|\___/|_| \_\_____|
  7. *
  8. * Copyright (C) 1998 - 2012, Daniel Stenberg, <daniel@haxx.se>, et al.
  9. *
  10. * This software is licensed as described in the file COPYING, which
  11. * you should have received as part of this distribution. The terms
  12. * are also available at http://curl.haxx.se/docs/copyright.html.
  13. *
  14. * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  15. * copies of the Software, and permit persons to whom the Software is
  16. * furnished to do so, under the terms of the COPYING file.
  17. *
  18. * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  19. * KIND, either express or implied.
  20. *
  21. ***************************************************************************/
  22. #include "curl_setup.h"
  23. #ifdef HAVE_SYS_IOCTL_H
  24. #include <sys/ioctl.h>
  25. #endif
  26. #ifdef HAVE_FCNTL_H
  27. #include <fcntl.h>
  28. #endif
  29. #if (defined(HAVE_IOCTL_FIONBIO) && defined(NETWARE))
  30. #include <sys/filio.h>
  31. #endif
  32. #ifdef __VMS
  33. #include <in.h>
  34. #include <inet.h>
  35. #endif
  36. #include "nonblock.h"
  37. /*
  38. * curlx_nonblock() set the given socket to either blocking or non-blocking
  39. * mode based on the 'nonblock' boolean argument. This function is highly
  40. * portable.
  41. */
  42. int curlx_nonblock(curl_socket_t sockfd, /* operate on this */
  43. int nonblock /* TRUE or FALSE */)
  44. {
  45. #if defined(USE_BLOCKING_SOCKETS)
  46. return 0; /* returns success */
  47. #elif defined(HAVE_FCNTL_O_NONBLOCK)
  48. /* most recent unix versions */
  49. int flags;
  50. flags = sfcntl(sockfd, F_GETFL, 0);
  51. if(nonblock)
  52. return sfcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
  53. else
  54. return sfcntl(sockfd, F_SETFL, flags & (~O_NONBLOCK));
  55. #elif defined(HAVE_IOCTL_FIONBIO)
  56. /* older unix versions */
  57. int flags = nonblock ? 1 : 0;
  58. return ioctl(sockfd, FIONBIO, &flags);
  59. #elif defined(HAVE_IOCTLSOCKET_FIONBIO)
  60. /* Windows */
  61. unsigned long flags = nonblock ? 1UL : 0UL;
  62. return ioctlsocket(sockfd, FIONBIO, &flags);
  63. #elif defined(HAVE_IOCTLSOCKET_CAMEL_FIONBIO)
  64. /* Amiga */
  65. long flags = nonblock ? 1L : 0L;
  66. return IoctlSocket(sockfd, FIONBIO, flags);
  67. #elif defined(HAVE_SETSOCKOPT_SO_NONBLOCK)
  68. /* BeOS */
  69. long b = nonblock ? 1L : 0L;
  70. return setsockopt(sockfd, SOL_SOCKET, SO_NONBLOCK, &b, sizeof(b));
  71. #else
  72. # error "no non-blocking method was found/used/set"
  73. #endif
  74. }