nonblock.c 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. /***************************************************************************
  2. * _ _ ____ _
  3. * Project ___| | | | _ \| |
  4. * / __| | | | |_) | |
  5. * | (__| |_| | _ <| |___
  6. * \___|\___/|_| \_\_____|
  7. *
  8. * Copyright (C) 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 https://curl.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. * SPDX-License-Identifier: curl
  22. *
  23. ***************************************************************************/
  24. #include "curl_setup.h"
  25. #ifdef HAVE_SYS_IOCTL_H
  26. #include <sys/ioctl.h>
  27. #endif
  28. #ifdef HAVE_FCNTL_H
  29. #include <fcntl.h>
  30. #endif
  31. #ifdef __VMS
  32. #include <in.h>
  33. #include <inet.h>
  34. #endif
  35. #include "nonblock.h"
  36. /*
  37. * curlx_nonblock() set the given socket to either blocking or non-blocking
  38. * mode based on the 'nonblock' boolean argument. This function is highly
  39. * portable.
  40. */
  41. int curlx_nonblock(curl_socket_t sockfd, /* operate on this */
  42. int nonblock /* TRUE or FALSE */)
  43. {
  44. #if defined(HAVE_FCNTL_O_NONBLOCK)
  45. /* most recent unix versions */
  46. int flags;
  47. flags = sfcntl(sockfd, F_GETFL, 0);
  48. if(nonblock)
  49. return sfcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
  50. return sfcntl(sockfd, F_SETFL, flags & (~O_NONBLOCK));
  51. #elif defined(HAVE_IOCTL_FIONBIO)
  52. /* older unix versions */
  53. int flags = nonblock ? 1 : 0;
  54. return ioctl(sockfd, FIONBIO, &flags);
  55. #elif defined(HAVE_IOCTLSOCKET_FIONBIO)
  56. /* Windows */
  57. unsigned long flags = nonblock ? 1UL : 0UL;
  58. return ioctlsocket(sockfd, (long)FIONBIO, &flags);
  59. #elif defined(HAVE_IOCTLSOCKET_CAMEL_FIONBIO)
  60. /* Amiga */
  61. long flags = nonblock ? 1L : 0L;
  62. return IoctlSocket(sockfd, FIONBIO, (char *)&flags);
  63. #elif defined(HAVE_SETSOCKOPT_SO_NONBLOCK)
  64. /* Orbis OS */
  65. long b = nonblock ? 1L : 0L;
  66. return setsockopt(sockfd, SOL_SOCKET, SO_NONBLOCK, &b, sizeof(b));
  67. #else
  68. # error "no non-blocking method was found/used/set"
  69. #endif
  70. }