echoping/SRC/readline.c

/*
 * Read a line from a descriptor with fgets 
 *
 * $Id$
 *
 */

#include "echoping.h"

int
readline(fs, ptr, maxlen, ln)
	FILE           *fs;
	char           *ptr;
	int             maxlen;
	unsigned short  ln;
{
	int             n = 1;
	char           *rc;
	int             r;

	/* Reading with fgets or fread instead of read one-character-at-a-time is
	 * more than ten times faster, on a local server. */
	if (ln) {
		rc = fgets(ptr, maxlen + 1, fs);
		if (rc == NULL) {
			if (timeout_flag)
				return n;
			return (-1);
		}
		n = strlen(rc);
		return n;
	} else {
		while (n < maxlen) {
			r = fread(ptr, 1, maxlen, fs);
			if (timeout_flag)
				return r;
			if (r == 0) {
				if (n == 1)
					return (0);	/* EOF, no data read */
				else
					break;	/* EOF, some data was read */
			}
			n = n + r;
		}
	}
	return (n);
}

#ifdef OPENSSL

char            SSL_buffer[MAXTOREAD];
int             buf_ptr;
int             buf_end;

int
SSL_readline(sslh, ptr, maxlen, ln)
	SSL            *sslh;
	char           *ptr;
	int             maxlen;
	unsigned short  ln;
{
	int             rc = 0;
	int             n = 0;
	int             i, oi;
	if (ln) {
		/* Empty buffer */
		if (buf_end == 0) {
			rc = SSL_read(sslh, SSL_buffer, maxlen);
			if (rc == -1)
				return rc;
			buf_end = rc;
			buf_ptr = 0;
		}
		/* No more data in the buffer */
		else if (buf_ptr == buf_end) {
			buf_ptr = 0;
			rc = SSL_read(sslh, SSL_buffer, maxlen);
			if (rc == -1)
				return rc;
			buf_end = rc;
		} else if (SSL_buffer[buf_end] != '\n') {
			/* We have a probleme here is the first SSL_read sent back a 
			 * text not finished by a \n. See www.SSL.de for an example. 
			 * We get more data. See bug #230384 */
			rc = SSL_read(sslh, SSL_buffer + buf_end, maxlen);
			if (rc == -1)
				return rc;
			buf_end = buf_end + rc;
		}
		for (oi = buf_ptr, i = buf_ptr;
		     i <= buf_end && SSL_buffer[i] != '\n'; i++) {
			*ptr++ = SSL_buffer[i];
			buf_ptr++;
		}
		if (SSL_buffer[i] == '\n')
			buf_ptr++;
		*ptr = '\0';
		/* if (ln) printf ("SSL_readline returns %d (%s)\n", i - oi,
		 * SSL_buffer); */
		return (i - oi);
	} else {
		/* OpenSSL reads at most 4096 characters */
		rc = 1;		/* Just to avoid exiting too soon */
		while (n < maxlen && rc != 0) {
			if ((buf_end == 0) || (buf_ptr > buf_end)) {
				rc = SSL_read(sslh, ptr, maxlen);
				buf_end = 0;
				buf_ptr = 0;
			} else {
				for (i = buf_ptr; i < maxlen && i <= buf_end; i++) {
					*ptr++ = SSL_buffer[i];
					rc++;
				}
				buf_ptr = i;
			}
			n = n + rc;
		}
		return n;
	}
}
#endif

#ifdef GNUTLS

char            TLS_buffer[MAXTOREAD];
int             buf_ptr;
int             buf_end;

int
TLS_readline(session, ptr, maxlen, ln)
	gnutls_session  session;
	char           *ptr;
	int             maxlen;
	unsigned short  ln;
{
	int             rc = 0;
	int             n = 0;
	int             i, oi;
	if (ln) {
		/* Empty buffer */
		if (buf_end == 0) {
			rc = gnutls_record_recv(session, TLS_buffer, maxlen);
			if (rc == -1)
				return rc;
			buf_end = rc;
			buf_ptr = 0;
		}
		/* No more data in the buffer */
		else if (buf_ptr == buf_end) {
			buf_ptr = 0;
			rc = gnutls_record_recv(session, TLS_buffer, maxlen);
			if (rc == -1)
				return rc;
			buf_end = rc;
		} else if (TLS_buffer[buf_end] != '\n') {
			rc = gnutls_record_recv(session, TLS_buffer + buf_end,
						maxlen);
			if (rc == -1)
				return rc;
			buf_end = buf_end + rc;
		}
		for (oi = buf_ptr, i = buf_ptr;
		     i <= buf_end && TLS_buffer[i] != '\n'; i++) {
			*ptr++ = TLS_buffer[i];
			buf_ptr++;
		}
		if (TLS_buffer[i] == '\n')
			buf_ptr++;
		*ptr = '\0';
		/* printf ("DEBUG: TLS_readline returns %d (%s)\n", i - oi,
		 * TLS_buffer); */
		return (i - oi);
	} else {
		rc = 1;		/* Just to avoid exiting too soon */
		while (n < maxlen && rc > 0) {
			if ((buf_end == 0) || (buf_ptr > buf_end)) {
				rc = gnutls_record_recv(session, ptr, maxlen);
				buf_end = 0;
				buf_ptr = 0;
			} else {
				for (i = buf_ptr; i < maxlen && i <= buf_end; i++) {
					*ptr++ = TLS_buffer[i];
					rc++;
					/* printf ("DEBUG: Now %d chars read\n",
					 * rc); */
				}
				buf_ptr = i;
			}
			if (rc > 0)
				n = n + rc;
			/* printf ("DEBUG: Now %d chars to send\n", n); */
		}
		return n;
	}
}
#endif