[pkgsrc/trunk]: pkgsrc/net/libfetch/files Convert fetch_read from select(2) t...

[joerg <joerg%pkgsrc.org@localhost>]

details:   https://anonhg.NetBSD.org/pkgsrc/rev/55af1375d267
branches:  trunk
changeset: 354156:55af1375d267
user:      joerg <joerg%pkgsrc.org@localhost>
date:      Thu Oct 20 21:22:18 2016 +0000

description:
Convert fetch_read from select(2) to poll(2) based sleep.

diffstat:

 net/libfetch/files/common.c |  69 ++++++++++++++++++++++++++------------------
 1 files changed, 40 insertions(+), 29 deletions(-)

diffs (107 lines):

diff -r f3db93f8959e -r 55af1375d267 net/libfetch/files/common.c
--- a/net/libfetch/files/common.c       Thu Oct 20 21:21:25 2016 +0000
+++ b/net/libfetch/files/common.c       Thu Oct 20 21:22:18 2016 +0000
@@ -1,4 +1,4 @@
-/*     $NetBSD: common.c,v 1.29 2014/01/08 20:25:34 joerg Exp $        */
+/*     $NetBSD: common.c,v 1.30 2016/10/20 21:22:18 joerg Exp $        */
 /*-
  * Copyright (c) 1998-2004 Dag-Erling Coïdan Smørgrav
  * Copyright (c) 2008, 2010 Joerg Sonnenberger <joerg%NetBSD.org@localhost>
@@ -41,7 +41,11 @@
 #include <sys/socket.h>
 #include <sys/time.h>
 #include <sys/uio.h>
-
+#if HAVE_POLL_H
+#include <poll.h>
+#elif HAVE_SYS_POLL_H
+#include <sys/poll.h>
+#endif
 #include <netinet/in.h>
 #include <arpa/inet.h>
 
@@ -492,6 +496,16 @@
 #endif
 }
 
+static int
+compute_timeout(const struct timeval *tv)
+{
+       struct timeval cur;
+       int timeout;
+
+       gettimeofday(&cur, NULL);
+       timeout = (tv->tv_sec - cur.tv_sec) * 1000 + (tv->tv_usec - cur.tv_usec) / 1000;
+       return timeout;
+}
 
 /*
  * Read a character from a connection w/ timeout
@@ -499,8 +513,9 @@
 ssize_t
 fetch_read(conn_t *conn, char *buf, size_t len)
 {
-       struct timeval now, timeout, waittv;
-       fd_set readfds;
+       struct timeval timeout_end;
+       struct pollfd pfd;
+       int timeout_cur;
        ssize_t rlen;
        int r;
 
@@ -517,34 +532,30 @@
        }
 
        if (fetchTimeout) {
-               FD_ZERO(&readfds);
-               gettimeofday(&timeout, NULL);
-               timeout.tv_sec += fetchTimeout;
+               gettimeofday(&timeout_end, NULL);
+               timeout_end.tv_sec += fetchTimeout;
        }
 
+       pfd.fd = conn->sd;
+       pfd.events = POLLIN;
        for (;;) {
-               while (fetchTimeout && !FD_ISSET(conn->sd, &readfds)) {
-                       FD_SET(conn->sd, &readfds);
-                       gettimeofday(&now, NULL);
-                       waittv.tv_sec = timeout.tv_sec - now.tv_sec;
-                       waittv.tv_usec = timeout.tv_usec - now.tv_usec;
-                       if (waittv.tv_usec < 0) {
-                               waittv.tv_usec += 1000000;
-                               waittv.tv_sec--;
-                       }
-                       if (waittv.tv_sec < 0) {
-                               errno = ETIMEDOUT;
-                               fetch_syserr();
-                               return (-1);
-                       }
-                       errno = 0;
-                       r = select(conn->sd + 1, &readfds, NULL, NULL, &waittv);
-                       if (r == -1) {
-                               if (errno == EINTR && fetchRestartCalls)
-                                       continue;
-                               fetch_syserr();
-                               return (-1);
-                       }
+               if (fetchTimeout) {
+                       do {
+                               timeout_cur = compute_timeout(&timeout_end);
+                               if (timeout_cur < 0) {
+                                       errno = ETIMEDOUT;
+                                       fetch_syserr();
+                                       return (-1);
+                               }
+                               errno = 0;
+                               r = poll(&pfd, 1, timeout_cur);
+                               if (r == -1) {
+                                       if (errno == EINTR && fetchRestartCalls)
+                                               continue;
+                                       fetch_syserr();
+                                       return (-1);
+                               }
+                       } while (pfd.revents == 0);
                }
 #ifdef WITH_SSL
                if (conn->ssl != NULL)