qse/ase/lib/cmn/tio_get.c

/*
 * $Id: tio_get.c,v 1.8 2005/12/26 07:41:48 bacon Exp $
 */

#include <ase/cmn/tio.h>
#include "mem.h"

#define STATUS_GETC_EILSEQ  (1 << 0)

ase_ssize_t ase_tio_getc (ase_tio_t* tio, ase_char_t* c)
{
	ase_size_t left = 0;
	ase_ssize_t n;
	ase_char_t curc;
#ifndef ASE_CHAR_IS_MCHAR
	ase_size_t seqlen;
#endif

	/* TODO: more efficient way to check this?
	 *       maybe better to use ASE_ASSERT 
	 * ASE_ASSERT (tio->input_func != ASE_NULL);
	 */
	if (tio->input_func == ASE_NULL) 
	{
		tio->errnum = ASE_TIO_ENOINF;
		return -1;
	}

	if (tio->input_status & STATUS_GETC_EILSEQ) 
	{
		tio->input_status &= ~STATUS_GETC_EILSEQ;
		tio->errnum = ASE_TIO_EILSEQ;
		return -1;
	}

	if (tio->inbuf_curp >= tio->inbuf_len) 
	{
	getc_conv:
		n = tio->input_func (
			ASE_TIO_IO_DATA, tio->input_arg,
			&tio->inbuf[left], ASE_COUNTOF(tio->inbuf) - left);
		if (n == 0) return 0;
		if (n <= -1) 
		{
			tio->errnum = ASE_TIO_EINPUT;
			return -1;
		}

		tio->inbuf_curp = 0;
		tio->inbuf_len = (ase_size_t)n + left;	
	}

#ifdef ASE_CHAR_IS_MCHAR
	curc = tio->inbuf[tio->inbuf_curp++];
#else
	left = tio->inbuf_len - tio->inbuf_curp;
	seqlen = ase_mblen (tio->inbuf[tio->inbuf_curp], left);
	if (seqlen == 0) 
	{
		/* illegal sequence */
		tio->inbuf_curp++;  /* skip one byte */
		tio->errnum = ASE_TIO_EILSEQ;
		return -1;
	}

	if (seqlen > left) 
	{
		/* incomplete sequence */
		if (tio->inbuf_curp > 0)
		{
			ASE_MEMCPY (tio->inbuf, &tio->inbuf[tio->inbuf_curp], left);
			tio->inbuf_curp = 0;
			tio->inbuf_len = left;
		}
		goto getc_conv;
	}
	
	n = ase_mbtowc (&tio->inbuf[tio->inbuf_curp], seqlen, &curc);
	if (n == 0) 
	{
		/* illegal sequence */
		tio->inbuf_curp++; /* skip one byte */
		tio->errnum = ASE_TIO_EILSEQ;
		return -1;
	}
	if (n > seqlen)
	{
		/* incomplete sequence - 
		 *  this check might not be needed because ase_mblen has
		 *  checked it. would ASE_ASSERT (n <= seqlen) be enough? */

		if (tio->inbuf_curp > 0)
		{
			ASE_MEMCPY (tio->inbuf, &tio->inbuf[tio->inbuf_curp], left);
			tio->inbuf_curp = 0;
			tio->inbuf_len = left;
		}
		goto getc_conv;
	}

	tio->inbuf_curp += n;
#endif

	*c = curc;
	return 1;
}

ase_ssize_t ase_tio_gets (ase_tio_t* tio, ase_char_t* buf, ase_size_t size)
{
	ase_ssize_t n;

	if (size <= 0) return 0;
	n = ase_tio_getsx (tio, buf, size - 1);
	if (n == -1) return -1;
	buf[n] = ASE_T('\0');
	return n;
}

ase_ssize_t ase_tio_getsx (ase_tio_t* tio, ase_char_t* buf, ase_size_t size)
{
	ase_ssize_t n;
	ase_char_t* p, * end, c;

	if (size <= 0) return 0;

	p = buf; end = buf + size;
	while (p < end) 
	{
		n = ase_tio_getc (tio, &c);
		if (n == -1) 
		{
			if (p > buf && tio->errnum == ASE_TIO_EILSEQ) 
			{
				tio->input_status |= STATUS_GETC_EILSEQ;
				break;
			}
			return -1;
		}
		if (n == 0) break;
		*p++ = c;

		if (c == ASE_T('\n')) break;
	}

	return p - buf;
}

ase_ssize_t ase_tio_getstr (ase_tio_t* tio, ase_str_t* buf)
{
	ase_ssize_t n;
	ase_char_t c;

	ase_str_clear (buf);

	for (;;) 
	{
		n = ase_tio_getc (tio, &c);
		if (n == -1) 
		{
			if (ASE_STR_LEN(buf) > 0 && tio->errnum == ASE_TIO_EILSEQ) 
			{
				tio->input_status |= STATUS_GETC_EILSEQ;
				break;
			}
			return -1;
		}
		if (n == 0) break;

		if (ase_str_ccat(buf, c) == (ase_size_t)-1) 
		{
			tio->errnum = ASE_TIO_ENOMEM;
			return -1;
		}

		if (c == ASE_T('\n')) break;
	}

	return ASE_STR_LEN(buf);
}