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