Merge pull request #1 from perry-clarke/perry-clarke-patch-1

Fix crash related to incorrect buffer size
This commit is contained in:
Perry Clarke 2016-05-03 16:45:06 -07:00
commit 0f4eaf06e4

View File

@ -191,19 +191,19 @@ static int ssl_recv(void *ctx, char *data, size_t count, size_t *got,
{ {
int err; int err;
p_ssl ssl = (p_ssl)ctx; p_ssl ssl = (p_ssl)ctx;
*got = 0;
if (ssl->state != LSEC_STATE_CONNECTED) if (ssl->state != LSEC_STATE_CONNECTED)
return IO_CLOSED; return IO_CLOSED;
*got = 0;
for ( ; ; ) { for ( ; ; ) {
ERR_clear_error(); ERR_clear_error();
err = SSL_read(ssl->ssl, data, (int)count); err = SSL_read(ssl->ssl, data, (int)count);
ssl->error = SSL_get_error(ssl->ssl, err); ssl->error = SSL_get_error(ssl->ssl, err);
switch (ssl->error) { switch (ssl->error) {
case SSL_ERROR_NONE: case SSL_ERROR_NONE:
*got = err; *got = 0;
return IO_DONE; return IO_DONE;
case SSL_ERROR_ZERO_RETURN: case SSL_ERROR_ZERO_RETURN:
*got = err; *got = 0;
return IO_CLOSED; return IO_CLOSED;
case SSL_ERROR_WANT_READ: case SSL_ERROR_WANT_READ:
err = socket_waitfd(&ssl->sock, WAITFD_R, tm); err = socket_waitfd(&ssl->sock, WAITFD_R, tm);