/*
**  MasqMail
**  Copyright (C) 2000 Oliver Kurth
**
**  This program is free software; you can redistribute it and/or modify
**  it under the terms of the GNU General Public License as published by
**  the Free Software Foundation; either version 2 of the License, or
**  (at your option) any later version.
**
**  This program is distributed in the hope that it will be useful,
**  but WITHOUT ANY WARRANTY; without even the implied warranty of
**  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
**  GNU General Public License for more details.
**
**  You should have received a copy of the GNU General Public License
**  along with this program; if not, write to the Free Software
**  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/

#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <setjmp.h>
#include <unistd.h>
#include <ctype.h>

#include "readsock.h"

jmp_buf jmp_timeout;

static void
sig_timeout_handler(int sig)
{
	longjmp(jmp_timeout, 1);
}

static struct sigaction old_sa_alrm;

static void
alarm_on(int timeout)
{
	struct sigaction sa;

	sa.sa_handler = sig_timeout_handler;
	sigemptyset(&(sa.sa_mask));
	sa.sa_flags = 0;
	sigaction(SIGALRM, &sa, &old_sa_alrm);

	if (timeout > 0)
		alarm(timeout);
}

static void
alarm_off()
{
	alarm(0);

	sigaction(SIGALRM, &old_sa_alrm, NULL);
}

static void
_read_chug(FILE *in)
{
	int c = 0;

	c = fgetc(in);
	while (isspace(c) && (c != EOF))
		c = fgetc(in);
	ungetc(c, in);
}

static int
_read_line(FILE *in, char *buf, int buf_len, int timeout)
{
	int p = 0;
	int c = 0;

	c = fgetc(in);
	while ((c != '\n') && (c != EOF) && (p < buf_len - 1)) {
		buf[p++] = c;
		c = fgetc(in);
	}

	buf[p] = '\0';

	if (c == EOF)
		return -1;
	else if (p >= buf_len) {
		ungetc(c, in);
		return -2;
	}

	buf[p++] = c;  /* \n */
	buf[p] = '\0';

	return p;
}

int
read_sockline(FILE *in, char *buf, int buf_len, int timeout, unsigned int flags)
{
	int p = 0;

	if (setjmp(jmp_timeout) != 0) {
		alarm_off();
		return -3;
	}

	alarm_on(timeout);

	/* strip leading spaces */
	if (flags & READSOCKL_CHUG) {
		_read_chug(in);
	}

	p = _read_line(in, buf, buf_len, timeout);

	alarm_off();

	if (p > 1) {
		/* here we are sure that buf[p-1] == '\n' */
		if (flags & READSOCKL_CVT_CRLF) {
			if ((buf[p - 2] == '\r') && (buf[p - 1] == '\n')) {
				buf[p - 2] = '\n';
				buf[p - 1] = 0;
				p--;
			}
		}
	}
	return p;
}

int
read_sockline1(FILE *in, char **pbuf, int *buf_len, int timeout,
		unsigned int flags)
{
	int p = 0, size = *buf_len;
	char *buf;

	if (setjmp(jmp_timeout) != 0) {
		alarm_off();
		return -3;
	}

	alarm_on(timeout);

	/* strip leading spaces */
	if (flags & READSOCKL_CHUG) {
		_read_chug(in);
	}

	if (!*pbuf) {
		*pbuf = (char *) malloc(size);
		if (!*pbuf) {
			fprintf(stderr, "Out of memory.\n");
			exit(1);
		}
	}
	buf = *pbuf;

	while (1) {
		int pp;

		pp = _read_line(in, buf, size, timeout);
		if (pp == -2) {
			*pbuf = realloc(*pbuf, *buf_len + size);
			buf = *pbuf + *buf_len;
			*buf_len += size;
			p += size;
		} else {
			if (pp > 0)
				p += pp;
			else
				p = pp;
			break;
		}
	}

	alarm_off();

	if (p > 1) {
		buf = *pbuf;
		/* here we are sure that buf[p-1] == '\n' */
		if (flags & READSOCKL_CVT_CRLF) {
			if ((buf[p - 2] == '\r') && (buf[p - 1] == '\n')) {
				buf[p - 2] = '\n';
				buf[p - 1] = '\0';
				p--;
			}
		}
	}
	return p;
}