2
0

select.h 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. #ifndef HEADER_CURL_SELECT_H
  2. #define HEADER_CURL_SELECT_H
  3. /***************************************************************************
  4. * _ _ ____ _
  5. * Project ___| | | | _ \| |
  6. * / __| | | | |_) | |
  7. * | (__| |_| | _ <| |___
  8. * \___|\___/|_| \_\_____|
  9. *
  10. * Copyright (C) 1998 - 2022, Daniel Stenberg, <daniel@haxx.se>, et al.
  11. *
  12. * This software is licensed as described in the file COPYING, which
  13. * you should have received as part of this distribution. The terms
  14. * are also available at https://curl.se/docs/copyright.html.
  15. *
  16. * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  17. * copies of the Software, and permit persons to whom the Software is
  18. * furnished to do so, under the terms of the COPYING file.
  19. *
  20. * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  21. * KIND, either express or implied.
  22. *
  23. * SPDX-License-Identifier: curl
  24. *
  25. ***************************************************************************/
  26. #include "curl_setup.h"
  27. #ifdef HAVE_POLL_H
  28. #include <poll.h>
  29. #elif defined(HAVE_SYS_POLL_H)
  30. #include <sys/poll.h>
  31. #endif
  32. /*
  33. * Definition of pollfd struct and constants for platforms lacking them.
  34. */
  35. #if !defined(HAVE_STRUCT_POLLFD) && \
  36. !defined(HAVE_SYS_POLL_H) && \
  37. !defined(HAVE_POLL_H) && \
  38. !defined(POLLIN)
  39. #define POLLIN 0x01
  40. #define POLLPRI 0x02
  41. #define POLLOUT 0x04
  42. #define POLLERR 0x08
  43. #define POLLHUP 0x10
  44. #define POLLNVAL 0x20
  45. struct pollfd
  46. {
  47. curl_socket_t fd;
  48. short events;
  49. short revents;
  50. };
  51. #endif
  52. #ifndef POLLRDNORM
  53. #define POLLRDNORM POLLIN
  54. #endif
  55. #ifndef POLLWRNORM
  56. #define POLLWRNORM POLLOUT
  57. #endif
  58. #ifndef POLLRDBAND
  59. #define POLLRDBAND POLLPRI
  60. #endif
  61. /* there are three CSELECT defines that are defined in the public header that
  62. are exposed to users, but this *IN2 bit is only ever used internally and
  63. therefore defined here */
  64. #define CURL_CSELECT_IN2 (CURL_CSELECT_ERR << 1)
  65. int Curl_socket_check(curl_socket_t readfd, curl_socket_t readfd2,
  66. curl_socket_t writefd,
  67. timediff_t timeout_ms);
  68. #define SOCKET_READABLE(x,z) \
  69. Curl_socket_check(x, CURL_SOCKET_BAD, CURL_SOCKET_BAD, z)
  70. #define SOCKET_WRITABLE(x,z) \
  71. Curl_socket_check(CURL_SOCKET_BAD, CURL_SOCKET_BAD, x, z)
  72. int Curl_poll(struct pollfd ufds[], unsigned int nfds, timediff_t timeout_ms);
  73. int Curl_wait_ms(timediff_t timeout_ms);
  74. /*
  75. With Winsock the valid range is [0..INVALID_SOCKET-1] according to
  76. https://docs.microsoft.com/en-us/windows/win32/winsock/socket-data-type-2
  77. */
  78. #ifdef USE_WINSOCK
  79. #define VALID_SOCK(s) ((s) < INVALID_SOCKET)
  80. #define FDSET_SOCK(x) 1
  81. #define VERIFY_SOCK(x) do { \
  82. if(!VALID_SOCK(x)) { \
  83. SET_SOCKERRNO(WSAEINVAL); \
  84. return -1; \
  85. } \
  86. } while(0)
  87. #else
  88. #define VALID_SOCK(s) ((s) >= 0)
  89. /* If the socket is small enough to get set or read from an fdset */
  90. #define FDSET_SOCK(s) ((s) < FD_SETSIZE)
  91. #define VERIFY_SOCK(x) do { \
  92. if(!VALID_SOCK(x) || !FDSET_SOCK(x)) { \
  93. SET_SOCKERRNO(EINVAL); \
  94. return -1; \
  95. } \
  96. } while(0)
  97. #endif
  98. #endif /* HEADER_CURL_SELECT_H */