qse/ase/test/awk/awk.c

/*
 * $Id: awk.c,v 1.14 2006-03-31 16:35:37 bacon Exp $
 */

#include <xp/awk/awk.h>
#include <xp/bas/stdio.h>
#include <wchar.h>

#ifdef __STAND_ALONE

static xp_ssize_t process_source (int cmd, void* arg, xp_char_t* data, xp_size_t size)
{
	xp_char_t c;

	switch (cmd) {
	case XP_AWK_IO_OPEN:
	case XP_AWK_IO_CLOSE:
		return 0;

	case XP_AWK_IO_DATA:
		if (size <= 0) return -1;
#ifdef XP_CHAR_IS_MCHAR
		c = fgetc (stdin);
#else
		c = fgetwc (stdin);
#endif
		if (c == XP_CHAR_EOF) return 0;
		*data = c;
		return 1;
	}

	return -1;
}

#else

#include <xp/bas/stdio.h>
#include <xp/bas/sio.h>

static xp_ssize_t process_source (int cmd, void* arg, xp_char_t* data, xp_size_t size)
{
	xp_ssize_t n;

	switch (cmd) {
	case XP_AWK_IO_OPEN:
	case XP_AWK_IO_CLOSE:
		return 0;

	case XP_AWK_IO_DATA:
		if (size < 0) return -1;
		n = xp_sio_getc (xp_sio_in, data);
		if (n == 0) return 0;
		if (n != 1) return -1;
		return n;
	}

	return -1;
}
#endif

#ifdef __linux
#include <mcheck.h>
#endif

#ifdef _WIN32
int xp_main (int argc, xp_char_t* argv[])
#else
int xp_main (int argc, char* argv[])
#endif
{
	xp_awk_t* awk;

#ifdef __linux
	mtrace ();
#endif

#if 0
	if (argc != 2) {
		xp_printf (XP_TEXT("Usage: %s file\n"), argv[0]);
		return -1;
	}
#endif

	if ((awk = xp_awk_open()) == XP_NULL) {
		xp_printf (XP_TEXT("Error: cannot open awk\n"));
		return -1;
	}

	if (xp_awk_attsrc(awk, process_source, XP_NULL) == -1) {
		xp_awk_close (awk);
		xp_printf (XP_TEXT("error: cannot attach source\n"));
		return -1;
	}

	xp_awk_setparseopt (awk, 
		XP_AWK_EXPLICIT | XP_AWK_UNIQUE | 
		XP_AWK_SHADING | XP_AWK_IMPLICIT);

	if (xp_awk_parse(awk) == -1) {
		xp_printf (
			XP_TEXT("error: cannot parse program - [%d] %s\n"), 
			xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
		xp_awk_close (awk);
		return -1;
	}

	if (xp_awk_run(awk) == -1) {
		xp_printf (
			XP_TEXT("error: cannot run program - [%d] %s\n"), 
			xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
		xp_awk_close (awk);
	}

	xp_awk_close (awk);

#ifdef __linux
	muntrace ();
#endif
	return 0;
}