Fix build under Windows.

[wget] / src / gnutls.c

diff --git a/src/gnutls.c b/src/gnutls.c
index 9e5c733b4c90c536ac46e4b6e3dfe0d5180220f0..2e5f89a781888f68af3362562a4539e81263cfa0 100644 (file)
--- a/src/gnutls.c
+++ b/src/gnutls.c
@@ -40,6 +40,8 @@ as that of the covered work.  */
 
 #include <gnutls/gnutls.h>
 #include <gnutls/x509.h>
+#include <fcntl.h>
+#include <sys/ioctl.h>
 
 #include "utils.h"
 #include "connect.h"
@@ -182,6 +184,22 @@ wgnutls_peek (int fd, char *buf, int bufsize, void *arg)
 
   if (bufsize > offset)
     {
+#ifdef F_GETFL
+      int flags;
+      flags = fcntl (fd, F_GETFL, 0);
+      if (flags < 0)
+        return ret;
+
+      ret = fcntl (fd, F_SETFL, flags | O_NONBLOCK);
+      if (ret < 0)
+        return ret;
+#else
+      /* XXX: Assume it was blocking before.  */
+      const int zero = 0;
+      ret = ioctl (fd, FIONBIO, &zero);
+      if (ret < 0)
+        return ret;
+#endif
       do
         {
           ret = gnutls_record_recv (ctx->session, buf + offset,
@@ -203,6 +221,17 @@ wgnutls_peek (int fd, char *buf, int bufsize, void *arg)
                   ret);
           ctx->peeklen += ret;
         }
+
+#ifdef F_GETFL
+      fcntl (fd, F_SETFL, flags);
+      if (ret < 0)
+        return ret;
+#else
+      const int one = 1;
+      ret = ioctl (fd, FIONBIO, &one);
+      if (ret < 0)
+        return ret;
+#endif
     }
 
   return offset + ret;