moparisthebest
/
wget
mirror of https://github.com/moparisthebest/wget
1
0
Fork 0
Code Issues Releases Wiki Activity

gnutls: Fix a problem with multiple peek read calls.

This commit is contained in:
Giuseppe Scrivano 2010-07-29 12:24:34 +02:00
parent 05503232c0
commit 75c6ca0f5d
2 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/ChangeLog
View File

@ -1,3 +1,8 @@
2010-07-29 Giuseppe Scrivano <gscrivano@gnu.org>
* gnutls.c (wgnutls_peek): Don't read more data if the buffered peek
data is sufficient.
2010-07-28 Giuseppe Scrivano <gscrivano@gnu.org>
* http.h (http_loop): Add new argument `original_url'

33
src/gnutls.c
View File

@ -173,28 +173,33 @@ wgnutls_peek (int fd, char *buf, int bufsize, void *arg)
{
int ret = 0;
struct wgnutls_transport_context *ctx = arg;
int offset = ctx->peeklen;
int offset = MIN (bufsize, ctx->peeklen);
if (bufsize > sizeof ctx->peekbuf)
bufsize = sizeof ctx->peekbuf;
if (offset)
if (ctx->peeklen)
memcpy (buf, ctx->peekbuf, offset);
do
if (bufsize > offset)
{
if (gnutls_record_check_pending (ctx->session)
|| select_fd (fd, 0, WAIT_FOR_READ))
ret = gnutls_record_recv (ctx->session, buf + offset, bufsize - offset);
}
while (ret == GNUTLS_E_INTERRUPTED);
do
{
if (gnutls_record_check_pending (ctx->session)
|| select_fd (fd, 0, WAIT_FOR_READ))
ret = gnutls_record_recv (ctx->session, buf + offset,
bufsize - offset);
}
while (ret == GNUTLS_E_INTERRUPTED);
if (ret > 0)
{
memcpy (ctx->peekbuf + offset, buf + offset, ret);
ctx->peeklen += ret;
if (ret > 0)
{
memcpy (ctx->peekbuf + offset, buf + offset,
ret);
ctx->peeklen += ret;
}
}
return ctx->peeklen;
return offset + ret;
}
static const char *