qse/ase/test/awk/awk.c

488 lines
9.6 KiB
C
Raw Normal View History

2006-03-31 16:35:37 +00:00
/*
2006-06-29 14:38:01 +00:00
* $Id: awk.c,v 1.46 2006-06-29 14:38:01 bacon Exp $
2006-03-31 16:35:37 +00:00
*/
2005-11-14 15:23:54 +00:00
#include <xp/awk/awk.h>
2006-03-31 18:13:31 +00:00
#include <stdio.h>
2006-04-10 09:26:17 +00:00
#include <string.h>
2006-06-19 04:38:51 +00:00
#ifdef XP_CHAR_IS_WCHAR
2006-04-30 18:05:07 +00:00
#include <wchar.h>
#endif
2006-01-18 16:12:59 +00:00
2006-05-06 16:05:12 +00:00
#ifndef __STAND_ALONE
#include <xp/bas/stdio.h>
#include <xp/bas/string.h>
2006-04-16 04:31:38 +00:00
#endif
2006-01-18 16:12:59 +00:00
2006-06-19 04:38:51 +00:00
#ifdef _WIN32
#include <tchar.h>
#endif
2006-04-16 04:31:38 +00:00
#ifdef __STAND_ALONE
2006-05-06 16:05:12 +00:00
#define xp_printf xp_awk_printf
extern int xp_awk_printf (const xp_char_t* fmt, ...);
#define xp_strcmp xp_awk_strcmp
extern int xp_awk_strcmp (const xp_char_t* s1, const xp_char_t* s2);
2006-06-19 04:38:51 +00:00
#define xp_strlen xp_awk_strlen
extern int xp_awk_strlen (const xp_char_t* s);
2006-04-16 04:31:38 +00:00
#endif
2006-04-10 09:22:05 +00:00
2006-05-12 09:39:20 +00:00
#if defined(_WIN32) && defined(__STAND_ALONE) && defined(_DEBUG)
#define _CRTDBG_MAP_ALLOC
#include <crtdbg.h>
#endif
2006-05-12 11:10:26 +00:00
#if defined(__linux) && defined(_DEBUG)
2006-05-12 09:39:20 +00:00
#include <mcheck.h>
#endif
2006-06-29 14:38:01 +00:00
struct src_io
{
const xp_char_t* input_file;
FILE* input_handle;
};
struct data_io
{
const char* input_file;
FILE* input_handle;
};
2006-04-16 04:31:38 +00:00
static xp_ssize_t process_source (
2006-06-22 04:25:44 +00:00
int cmd, int opt, void* arg, xp_char_t* data, xp_size_t size)
2006-01-18 16:12:59 +00:00
{
2006-06-29 14:38:01 +00:00
struct src_io* src_io = (struct src_io*)arg;
2006-03-29 16:39:04 +00:00
xp_char_t c;
2006-01-18 16:12:59 +00:00
2006-06-13 04:26:24 +00:00
switch (cmd)
{
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_OPEN:
2006-06-29 14:38:01 +00:00
{
if (src_io->input_file == XP_NULL) return 0;
src_io->input_handle = _tfopen (src_io->input_file, _T("r"));
if (src_io->input_handle == NULL) return -1;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_CLOSE:
2006-06-29 14:38:01 +00:00
{
if (src_io->input_file == XP_NULL) return 0;
fclose ((FILE*)src_io->input_handle);
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_NEXT:
2006-06-13 04:26:24 +00:00
{
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_READ:
2006-06-13 04:26:24 +00:00
{
if (size <= 0) return -1;
#ifdef XP_CHAR_IS_MCHAR
2006-06-29 14:38:01 +00:00
c = fgetc ((FILE*)src_io->input_handle);
2006-06-13 04:26:24 +00:00
#else
2006-06-29 14:38:01 +00:00
c = fgetwc ((FILE*)src_io->input_handle);
2006-06-13 04:26:24 +00:00
#endif
if (c == XP_CHAR_EOF) return 0;
*data = c;
return 1;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_WRITE:
2006-06-13 04:26:24 +00:00
{
2006-06-29 14:38:01 +00:00
xp_printf (XP_T("XP_AWK_IO_WRITE CALLED FOR SOURCE\n"));
2006-06-22 04:25:44 +00:00
return -1;
2006-06-13 04:26:24 +00:00
}
2006-04-22 13:54:53 +00:00
}
return -1;
}
2006-06-22 04:25:44 +00:00
2006-04-22 13:54:53 +00:00
static xp_ssize_t process_data (
2006-06-22 04:25:44 +00:00
int cmd, int opt, void* arg, xp_char_t* data, xp_size_t size)
2006-04-22 13:54:53 +00:00
{
struct data_io* io = (struct data_io*)arg;
xp_char_t c;
2006-06-13 04:26:24 +00:00
switch (cmd)
{
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_OPEN:
2006-06-13 04:26:24 +00:00
{
io->input_handle = fopen (io->input_file, "r");
if (io->input_handle == NULL) return -1;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_CLOSE:
2006-06-13 04:26:24 +00:00
{
fclose (io->input_handle);
io->input_handle = NULL;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_READ:
2006-06-13 04:26:24 +00:00
{
if (size <= 0) return -1;
#ifdef XP_CHAR_IS_MCHAR
c = fgetc (io->input_handle);
#else
c = fgetwc (io->input_handle);
#endif
if (c == XP_CHAR_EOF) return 0;
*data = c;
return 1;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_WRITE:
2006-06-13 04:26:24 +00:00
{
return -1;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_NEXT:
{
/* input switching not supported for the time being... */
return -1;
}
2006-01-18 16:12:59 +00:00
}
return -1;
}
2006-06-19 04:38:51 +00:00
static xp_ssize_t process_extio_pipe (
2006-06-22 04:25:44 +00:00
int cmd, int opt, void* arg, xp_char_t* data, xp_size_t size)
2006-06-19 04:38:51 +00:00
{
2006-06-19 09:10:57 +00:00
xp_awk_extio_t* epa = (xp_awk_extio_t*)arg;
2006-06-19 04:38:51 +00:00
switch (cmd)
{
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_OPEN:
2006-06-19 04:38:51 +00:00
{
FILE* handle;
2006-06-22 04:25:44 +00:00
const xp_char_t* mode;
if (opt == XP_AWK_IO_PIPE_READ)
mode = XP_T("r");
else if (opt == XP_AWK_IO_PIPE_WRITE)
mode = XP_T("w");
else return -1; /* TODO: any way to set the error number? */
2006-06-28 08:56:59 +00:00
xp_printf (XP_TEXT("opending %s of type %d (pipe)\n"), epa->name, epa->type);
2006-06-22 04:25:44 +00:00
handle = _tpopen (epa->name, mode);
2006-06-19 04:38:51 +00:00
if (handle == NULL) return -1;
epa->handle = (void*)handle;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_CLOSE:
2006-06-19 04:38:51 +00:00
{
2006-06-28 08:56:59 +00:00
xp_printf (XP_TEXT("closing %s of type (pipe) %d\n"), epa->name, epa->type);
2006-06-19 04:38:51 +00:00
fclose ((FILE*)epa->handle);
epa->handle = NULL;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_READ:
2006-06-19 04:38:51 +00:00
{
2006-06-22 04:25:44 +00:00
if (_fgetts (data, size, (FILE*)epa->handle) == XP_NULL)
2006-06-19 04:38:51 +00:00
return 0;
return xp_strlen(data);
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_WRITE:
2006-06-19 04:38:51 +00:00
{
2006-06-22 04:25:44 +00:00
/*
if (_fputts (data, size, (FILE*)epa->handle) == XP_NULL)
return 0;
return size;
*/
2006-06-19 04:38:51 +00:00
return -1;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_NEXT:
2006-06-19 15:43:27 +00:00
{
return -1;
}
}
2006-06-22 04:25:44 +00:00
return -1;
2006-06-19 04:38:51 +00:00
}
2006-06-19 09:10:57 +00:00
static xp_ssize_t process_extio_file (
2006-06-22 04:25:44 +00:00
int cmd, int opt, void* arg, xp_char_t* data, xp_size_t size)
2006-06-19 09:10:57 +00:00
{
xp_awk_extio_t* epa = (xp_awk_extio_t*)arg;
switch (cmd)
{
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_OPEN:
2006-06-19 09:10:57 +00:00
{
FILE* handle;
2006-06-22 04:25:44 +00:00
const xp_char_t* mode;
if (opt == XP_AWK_IO_FILE_READ)
mode = XP_T("r");
else if (opt == XP_AWK_IO_FILE_WRITE)
mode = XP_T("w");
else if (opt == XP_AWK_IO_FILE_APPEND)
mode = XP_T("a");
else return -1; /* TODO: any way to set the error number? */
2006-06-28 08:56:59 +00:00
xp_printf (XP_TEXT("opending %s of type %d (file)\n"), epa->name, epa->type);
2006-06-22 04:25:44 +00:00
handle = _tfopen (epa->name, mode);
2006-06-19 09:10:57 +00:00
if (handle == NULL) return -1;
epa->handle = (void*)handle;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_CLOSE:
2006-06-19 09:10:57 +00:00
{
2006-06-28 08:56:59 +00:00
xp_printf (XP_TEXT("closing %s of type %d (file)\n"), epa->name, epa->type);
2006-06-19 09:10:57 +00:00
fclose ((FILE*)epa->handle);
epa->handle = NULL;
return 0;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_READ:
2006-06-19 09:10:57 +00:00
{
2006-06-22 04:25:44 +00:00
if (_fgetts (data, size, (FILE*)epa->handle) == XP_NULL)
2006-06-19 09:10:57 +00:00
return 0;
return xp_strlen(data);
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_WRITE:
2006-06-19 09:10:57 +00:00
{
2006-06-27 10:53:04 +00:00
/* TODO: how to return error or 0 */
_fputts (data, /*size,*/ (FILE*)epa->handle);
2006-06-19 09:10:57 +00:00
return -1;
}
2006-06-22 04:25:44 +00:00
case XP_AWK_IO_NEXT:
2006-06-19 09:10:57 +00:00
{
return -1;
}
2006-06-19 15:43:27 +00:00
2006-06-19 09:10:57 +00:00
}
2006-06-22 04:25:44 +00:00
return -1;
2006-06-19 09:10:57 +00:00
}
2006-06-25 15:26:57 +00:00
static xp_ssize_t process_extio_console (
int cmd, int opt, void* arg, xp_char_t* data, xp_size_t size)
{
xp_awk_extio_t* epa = (xp_awk_extio_t*)arg;
switch (cmd)
{
case XP_AWK_IO_OPEN:
{
/* TODO: OpenConsole in GUI APPLICATION */
/* opt: XP_AWK_IO_CONSOLE_READ,
* XP_AWK_IO_CONSOLE_WRITE */
2006-06-29 14:38:01 +00:00
xp_printf (XP_TEXT("opening [%s] of type %d (console)\n"), epa->name, epa->type);
2006-06-25 15:26:57 +00:00
if (opt == XP_AWK_IO_CONSOLE_READ)
epa->handle = stdin;
else if (opt == XP_AWK_IO_CONSOLE_WRITE)
epa->handle = stdout;
else return -1;
2006-06-29 14:38:01 +00:00
return 0;
2006-06-25 15:26:57 +00:00
}
case XP_AWK_IO_CLOSE:
{
2006-06-29 14:38:01 +00:00
xp_printf (XP_TEXT("closing [%s] of type %d (console)\n"), epa->name, epa->type);
2006-06-25 15:26:57 +00:00
/* TODO: CloseConsole in GUI APPLICATION */
2006-06-26 15:09:28 +00:00
return 0;
2006-06-25 15:26:57 +00:00
}
case XP_AWK_IO_READ:
{
2006-06-29 14:38:01 +00:00
if (_fgetts (data, size, (FILE*)epa->handle) == XP_NULL)
//if (_fgetts (data, size, stdin) == XP_NULL)
{
2006-06-25 15:26:57 +00:00
return 0;
2006-06-29 14:38:01 +00:00
}
2006-06-25 15:26:57 +00:00
return xp_strlen(data);
}
case XP_AWK_IO_WRITE:
{
2006-06-26 15:09:28 +00:00
/* TODO: how to return error or 0 */
_fputts (data, /*size,*/ (FILE*)epa->handle);
2006-06-29 14:38:01 +00:00
//_fputts (data, /*size,*/ stdout);
2006-06-25 15:26:57 +00:00
return size;
}
case XP_AWK_IO_NEXT:
{
return -1;
}
}
return -1;
}
2006-06-19 09:10:57 +00:00
2006-04-18 16:27:29 +00:00
#if defined(__STAND_ALONE) && !defined(_WIN32)
2006-05-12 09:39:20 +00:00
static int __main (int argc, char* argv[])
2006-04-14 11:13:06 +00:00
#else
2006-05-12 09:39:20 +00:00
static int __main (int argc, xp_char_t* argv[])
2006-04-14 11:13:06 +00:00
#endif
2005-11-14 15:23:54 +00:00
{
2006-03-31 16:35:37 +00:00
xp_awk_t* awk;
2006-04-22 13:54:53 +00:00
struct data_io data_io = { "awk.in", NULL };
2006-06-29 14:38:01 +00:00
struct src_io src_io = { NULL, NULL };
2005-11-14 15:23:54 +00:00
2006-04-10 09:22:05 +00:00
if ((awk = xp_awk_open()) == XP_NULL)
{
2006-05-06 16:05:12 +00:00
xp_printf (XP_T("Error: cannot open awk\n"));
2005-11-14 15:23:54 +00:00
return -1;
}
2006-06-19 04:38:51 +00:00
/* TODO: */
2006-06-19 09:10:57 +00:00
if (xp_awk_setextio (awk,
XP_AWK_EXTIO_PIPE, process_extio_pipe, XP_NULL) == -1)
2006-06-19 04:38:51 +00:00
{
xp_awk_close (awk);
2006-06-19 09:10:57 +00:00
xp_printf (XP_T("Error: cannot set extio pipe\n"));
return -1;
}
/* TODO: */
if (xp_awk_setextio (awk,
XP_AWK_EXTIO_FILE, process_extio_file, XP_NULL) == -1)
2006-06-25 15:26:57 +00:00
{
xp_awk_close (awk);
xp_printf (XP_T("Error: cannot set extio file\n"));
return -1;
}
if (xp_awk_setextio (awk,
XP_AWK_EXTIO_CONSOLE, process_extio_console, XP_NULL) == -1)
2006-06-19 09:10:57 +00:00
{
xp_awk_close (awk);
xp_printf (XP_T("Error: cannot set extio file\n"));
2006-06-19 04:38:51 +00:00
return -1;
}
2006-03-31 16:35:37 +00:00
xp_awk_setparseopt (awk,
2006-06-21 13:52:15 +00:00
XP_AWK_EXPLICIT | XP_AWK_UNIQUE | XP_AWK_DBLSLASHES |
2006-06-18 13:43:28 +00:00
XP_AWK_SHADING | XP_AWK_IMPLICIT | XP_AWK_SHIFT | XP_AWK_EXTIO);
2006-03-05 17:07:33 +00:00
2006-04-10 09:22:05 +00:00
if (argc == 2)
{
2006-04-18 16:27:29 +00:00
#if defined(__STAND_ALONE) && !defined(_WIN32)
2006-04-14 11:13:06 +00:00
if (strcmp(argv[1], "-m") == 0)
#else
2006-05-06 16:05:12 +00:00
if (xp_strcmp(argv[1], XP_T("-m")) == 0)
2006-06-29 14:38:01 +00:00
#endif
{
xp_awk_close (awk);
xp_printf (XP_T("Usage: %s [-m] source_file\n"), argv[0]);
return -1;
}
src_io.input_file = argv[1];
}
else if (argc == 3)
{
#if defined(__STAND_ALONE) && !defined(_WIN32)
if (strcmp(argv[1], "-m") == 0)
#else
if (xp_strcmp(argv[1], XP_T("-m")) == 0)
2006-04-14 11:13:06 +00:00
#endif
2006-04-10 09:22:05 +00:00
{
xp_awk_setrunopt (awk, XP_AWK_RUNMAIN);
}
2006-06-29 14:38:01 +00:00
else
{
xp_awk_close (awk);
xp_printf (XP_T("Usage: %s [-m] source_file\n"), argv[0]);
return -1;
}
src_io.input_file = argv[2];
}
else
{
xp_awk_close (awk);
xp_printf (XP_T("Usage: %s [-m] source_file\n"), argv[0]);
return -1;
}
if (xp_awk_attsrc(awk, process_source, (void*)&src_io) == -1)
{
xp_awk_close (awk);
xp_printf (XP_T("Error: cannot attach source\n"));
return -1;
2006-04-10 09:22:05 +00:00
}
2006-04-09 15:34:38 +00:00
2006-04-10 09:22:05 +00:00
if (xp_awk_parse(awk) == -1)
{
2006-04-18 16:27:29 +00:00
#if defined(__STAND_ALONE) && !defined(_WIN32) && defined(XP_CHAR_IS_WCHAR)
2006-04-14 16:26:00 +00:00
xp_printf (
2006-05-13 16:33:07 +00:00
XP_T("error: cannot parse program - line %u [%d] %ls\n"),
(unsigned int)xp_awk_getsrcline(awk),
2006-04-14 16:26:00 +00:00
xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
#else
2006-01-31 16:57:45 +00:00
xp_printf (
2006-05-13 16:33:07 +00:00
XP_T("error: cannot parse program - line %u [%d] %s\n"),
(unsigned int)xp_awk_getsrcline(awk),
2006-03-31 16:35:37 +00:00
xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
2006-04-14 16:26:00 +00:00
#endif
2006-03-31 16:35:37 +00:00
xp_awk_close (awk);
2005-11-14 15:23:54 +00:00
return -1;
}
2006-04-22 13:54:53 +00:00
if (xp_awk_run (awk, process_data, (void*)&data_io) == -1)
2006-04-10 09:22:05 +00:00
{
2006-04-18 16:27:29 +00:00
#if defined(__STAND_ALONE) && !defined(_WIN32) && defined(XP_CHAR_IS_WCHAR)
2006-04-14 16:26:00 +00:00
xp_printf (
2006-05-06 16:05:12 +00:00
XP_T("error: cannot run program - [%d] %ls\n"),
2006-04-14 16:26:00 +00:00
xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
#else
2006-03-05 17:07:33 +00:00
xp_printf (
2006-05-06 16:05:12 +00:00
XP_T("error: cannot run program - [%d] %s\n"),
2006-03-31 16:35:37 +00:00
xp_awk_geterrnum(awk), xp_awk_geterrstr(awk));
2006-04-14 16:26:00 +00:00
#endif
2006-03-31 16:35:37 +00:00
xp_awk_close (awk);
2006-04-06 16:25:37 +00:00
return -1;
2006-03-05 17:07:33 +00:00
}
2006-03-31 16:35:37 +00:00
xp_awk_close (awk);
2006-05-12 09:39:20 +00:00
return 0;
}
2006-01-30 17:50:38 +00:00
2006-05-12 09:39:20 +00:00
#if defined(__STAND_ALONE) && !defined(_WIN32)
int main (int argc, char* argv[])
#else
int xp_main (int argc, xp_char_t* argv[])
#endif
{
int n;
2006-05-12 11:10:26 +00:00
#if defined(__linux) && defined(_DEBUG)
2006-05-12 09:39:20 +00:00
mtrace ();
#endif
2006-05-12 10:56:18 +00:00
/*#if defined(_WIN32) && defined(__STAND_ALONE) && defined(_DEBUG)
_CrtSetDbgFlag (_CRTDBG_LEAK_CHECK_DF | _CRTDBG_ALLOC_MEM_DF | _CRTDBG_CHECK_ALWAYS_DF);
#endif*/
2006-05-12 09:39:20 +00:00
n = __main (argc, argv);
2006-05-12 11:10:26 +00:00
#if defined(__linux) && defined(_DEBUG)
2006-01-30 17:50:38 +00:00
muntrace ();
#endif
2006-05-12 09:39:20 +00:00
#if defined(_WIN32) && defined(__STAND_ALONE) && defined(_DEBUG)
_CrtDumpMemoryLeaks ();
2006-05-13 15:51:43 +00:00
wprintf (L"Press ENTER to quit\n");
getchar ();
2006-05-12 09:39:20 +00:00
#endif
2006-05-13 15:51:43 +00:00
2006-05-12 09:39:20 +00:00
return n;
2005-11-14 15:23:54 +00:00
}
2006-05-12 09:39:20 +00:00