moparisthebest/wget
1
0
Fork 0
mirror of https://github.com/moparisthebest/wget synced 2024-07-03 16:38:41 -04:00
Code Issues Releases Wiki Activity

gnutls: Honor read timeout.

Browse Source
This commit is contained in:
Giuseppe Scrivano 2011-08-26 17:13:56 +02:00
parent 547bcb0d3f
commit 5bcd75d32f
2 changed files with 51 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/ChangeLog
View File

@ -1,5 +1,8 @@
2011-08-26 Giuseppe Scrivano <gscrivano@gnu.org> 2011-08-26 Giuseppe Scrivano <gscrivano@gnu.org>
* gnutls.c: Include "ptimer.h".
(wgnutls_read): Honor read timeout.
* openssl.c (ssl_init): Make `meth' const. * openssl.c (ssl_init): Make `meth' const.
2011-08-25 Giuseppe Scrivano <gscrivano@gnu.org> 2011-08-25 Giuseppe Scrivano <gscrivano@gnu.org>

50
src/gnutls.c
View File

@ -45,6 +45,7 @@ as that of the covered work. */
#include "utils.h" #include "utils.h"
#include "connect.h" #include "connect.h"
#include "url.h" #include "url.h"
#include "ptimer.h"
#include "ssl.h" #include "ssl.h"
#ifdef WIN32 #ifdef WIN32
@ -125,7 +126,11 @@ struct wgnutls_transport_context
static int static int
wgnutls_read (int fd, char *buf, int bufsize, void *arg) wgnutls_read (int fd, char *buf, int bufsize, void *arg)
{ {
#ifdef F_GETFL
int flags = 0;
#endif
int ret = 0; int ret = 0;
struct ptimer *timer;
struct wgnutls_transport_context *ctx = arg; struct wgnutls_transport_context *ctx = arg;
if (ctx->peeklen) if (ctx->peeklen)
@ -140,9 +145,50 @@ wgnutls_read (int fd, char *buf, int bufsize, void *arg)
return copysize; return copysize;
} }
if (opt.read_timeout)
{
#ifdef F_GETFL
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 one = 1;
ret = ioctl (fd, FIONBIO, &one);
if (ret < 0)
return ret;
#endif
timer = ptimer_new ();
if (timer == 0)
return -1;
}
do do
ret = gnutls_record_recv (ctx->session, buf, bufsize); {
while (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN); do
ret = gnutls_record_recv (ctx->session, buf, bufsize);
while (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN);
}
while (opt.read_timeout == 0 || ptimer_measure (timer) < opt.read_timeout);
if (opt.read_timeout)
{
ptimer_destroy (timer);
#ifdef F_GETFL
ret = fcntl (fd, F_SETFL, flags);
if (ret < 0)
return ret;
#else
const int zero = 0;
ret = ioctl (fd, FIONBIO, &zero);
if (ret < 0)
return ret;
#endif
}
if (ret < 0) if (ret < 0)
ctx->last_error = ret; ctx->last_error = ret;