From 2d517ebcebe326e79186e41ee7bbd1cf5ed1f2b9 Mon Sep 17 00:00:00 2001 From: Nikias Bassen Date: Tue, 26 Dec 2023 14:52:37 +0100 Subject: socket: Conditionally compile using poll or select based on availability --- configure.ac | 2 +- src/socket.c | 112 +++++++++++++++++++++++++++++++++++++++++++++-------------- 2 files changed, 86 insertions(+), 28 deletions(-) diff --git a/configure.ac b/configure.ac index ca69d32..04165c9 100644 --- a/configure.ac +++ b/configure.ac @@ -49,7 +49,7 @@ AC_TYPE_UINT32_T AC_TYPE_UINT8_T # Checks for library functions. -AC_CHECK_FUNCS([asprintf strcasecmp strdup strerror strndup stpcpy vasprintf getifaddrs]) +AC_CHECK_FUNCS([asprintf strcasecmp strdup strerror strndup stpcpy vasprintf getifaddrs poll]) AC_CHECK_HEADER(endian.h, [ac_cv_have_endian_h="yes"], [ac_cv_have_endian_h="no"]) if test "x$ac_cv_have_endian_h" = "xno"; then diff --git a/src/socket.c b/src/socket.c index 49a5ef4..c2ebf25 100644 --- a/src/socket.c +++ b/src/socket.c @@ -65,7 +65,9 @@ static int wsa_init = 0; #endif #include "common.h" #include "libimobiledevice-glue/socket.h" +#ifdef HAVE_POLL #include +#endif #define RECV_TIMEOUT 20000 #define SEND_TIMEOUT 10000 @@ -148,9 +150,10 @@ enum poll_status poll_status_error }; -// timeoutMilliseconds of -1 means infinity -static enum poll_status poll_wrapper(int fd, fd_mode mode, int timeoutMilliseconds) +// timeout of -1 means infinity +static inline __attribute__((always_inline)) enum poll_status poll_wrapper(int fd, fd_mode mode, int timeout) { +#ifdef HAVE_POLL // https://man7.org/linux/man-pages/man2/select.2.html // Correspondence between select() and poll() notifications // #define POLLIN_SET (EPOLLRDNORM | EPOLLRDBAND | EPOLLIN | @@ -163,8 +166,7 @@ static enum poll_status poll_wrapper(int fd, fd_mode mode, int timeoutMillisecon // /* Exceptional condition */ short events; - switch(mode) - { + switch (mode) { case FDM_READ: events = POLLRDNORM | POLLRDBAND | POLLIN | POLLHUP | POLLERR; break; @@ -179,20 +181,17 @@ static enum poll_status poll_wrapper(int fd, fd_mode mode, int timeoutMillisecon fprintf(stderr, "%s: fd_mode %d unsupported\n", __func__, mode); return poll_status_error; } - while(1) - { + while (1) { struct pollfd pfd = { .fd = fd, .events = events, }; - switch(poll(&pfd, 1, timeoutMilliseconds)) - { + switch (poll(&pfd, 1, timeout)) { case 1: if((pfd.revents & (POLLNVAL | POLLERR)) != 0) { if (verbose >= 2) - fprintf(stderr, "%s: poll unexpected events: %d\n", __func__, - (int)pfd.revents); + fprintf(stderr, "%s: poll unexpected events: %d\n", __func__, (int)pfd.revents); return poll_status_error; } return poll_status_success; @@ -208,11 +207,73 @@ static enum poll_status poll_wrapper(int fd, fd_mode mode, int timeoutMillisecon // fallthrough default: if (verbose >= 2) - fprintf(stderr, "%s: poll failed: %s\n", __func__, - strerror(errno)); + fprintf(stderr, "%s: poll failed: %s\n", __func__, strerror(errno)); return poll_status_error; } } +#else + fd_set fds; + int sret; + int eagain; + struct timeval to; + struct timeval *pto; + + FD_ZERO(&fds); + FD_SET(fd, &fds); + + sret = poll_status_error; + + do { + if (timeout > 0) { + to.tv_sec = (time_t) (timeout / 1000); + to.tv_usec = (time_t) ((timeout - (to.tv_sec * 1000)) * 1000); + pto = &to; + } else { + pto = NULL; + } + eagain = 0; + switch (mode) { + case FDM_READ: + sret = select(fd + 1, &fds, NULL, NULL, pto); + break; + case FDM_WRITE: + sret = select(fd + 1, NULL, &fds, NULL, pto); + break; + case FDM_EXCEPT: + sret = select(fd + 1, NULL, NULL, &fds, pto); + break; + default: + if (verbose >= 2) + fprintf(stderr, "%s: fd_mode %d unsupported\n", __func__, mode); + return poll_status_error; + } + + if (sret == 1) { + return poll_status_success; + } else if (sret == 0) { + return poll_status_timeout; + } else { + switch (errno) { + case EINTR: + // interrupt signal in select + if (verbose >= 2) + fprintf(stderr, "%s: EINTR\n", __func__); + eagain = 1; + break; + case EAGAIN: + if (verbose >= 2) + fprintf(stderr, "%s: EAGAIN\n", __func__); + break; + default: + if (verbose >= 2) + fprintf(stderr, "%s: select failed: %s\n", __func__, strerror(errno)); + return poll_status_error; + } + } + } while (eagain); + + return sret; +#endif } #ifndef WIN32 @@ -321,7 +382,7 @@ LIBIMOBILEDEVICE_GLUE_API int socket_connect_unix(const char *filename) break; } if (errno == EINPROGRESS) { - if(poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { + if (poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { int so_error; socklen_t len = sizeof(so_error); getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); @@ -998,7 +1059,7 @@ LIBIMOBILEDEVICE_GLUE_API int socket_connect_addr(struct sockaddr* addr, uint16_ if (errno == EINPROGRESS) #endif { - if(poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { + if (poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { int so_error; socklen_t len = sizeof(so_error); getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); @@ -1115,7 +1176,7 @@ LIBIMOBILEDEVICE_GLUE_API int socket_connect(const char *addr, uint16_t port) if (errno == EINPROGRESS) #endif { - if(poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { + if (poll_wrapper(sfd, FDM_WRITE, CONNECT_TIMEOUT) == poll_status_success) { int so_error; socklen_t len = sizeof(so_error); getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); @@ -1158,25 +1219,22 @@ LIBIMOBILEDEVICE_GLUE_API int socket_check_fd(int fd, fd_mode fdm, unsigned int return -1; } - int timeoutMilliseconds; - if(timeout > 0) - { - timeoutMilliseconds = (int)timeout; - if(timeoutMilliseconds <= 0) - timeoutMilliseconds = -1; - } - else - { - timeoutMilliseconds = -1; + int timeout_ms; + if (timeout > 0) { + timeout_ms = (int)timeout; + if (timeout_ms <= 0) + timeout_ms = -1; + } else { + timeout_ms = -1; } - switch(poll_wrapper(fd, fdm, timeoutMilliseconds)) - { + switch (poll_wrapper(fd, fdm, timeout_ms)) { case poll_status_success: return 1; case poll_status_timeout: return -ETIMEDOUT; case poll_status_error: + default: if (verbose >= 2) fprintf(stderr, "%s: poll_wrapper failed\n", __func__); return -1; -- cgit v1.1-32-gdbae