412 lines
8.2 KiB
C
412 lines
8.2 KiB
C
/*
|
|
* $Id: extio.c,v 1.17 2006-07-01 16:07:06 bacon Exp $
|
|
*/
|
|
|
|
#include <xp/awk/awk_i.h>
|
|
|
|
#ifndef XP_AWK_STAND_ALONE
|
|
#include <xp/bas/assert.h>
|
|
#include <xp/bas/string.h>
|
|
#include <xp/bas/memory.h>
|
|
#endif
|
|
|
|
enum
|
|
{
|
|
__MASK_READ = 0x01FF,
|
|
__MASK_WRITE = 0x02FF,
|
|
__MASK_RDWR = 0x04FF,
|
|
__MASK_CLEAR = 0x00FF
|
|
};
|
|
|
|
int xp_awk_readextio (
|
|
xp_awk_run_t* run, int in_type,
|
|
const xp_char_t* name, xp_str_t* buf, int* errnum)
|
|
{
|
|
xp_awk_extio_t* p = run->extio;
|
|
xp_awk_io_t handler;
|
|
int extio_type, extio_opt, extio_mask;
|
|
|
|
static int __in_type_map[] =
|
|
{
|
|
/* the order should match the order of the
|
|
* XP_AWK_IN_XXX values in tree.h */
|
|
XP_AWK_EXTIO_PIPE,
|
|
XP_AWK_EXTIO_COPROC,
|
|
XP_AWK_EXTIO_FILE,
|
|
XP_AWK_EXTIO_CONSOLE
|
|
};
|
|
|
|
static int __in_opt_map[] =
|
|
{
|
|
/* the order should match the order of the
|
|
* XP_AWK_IN_XXX values in tree.h */
|
|
XP_AWK_IO_PIPE_READ,
|
|
0,
|
|
XP_AWK_IO_FILE_READ,
|
|
XP_AWK_IO_FILE_READ
|
|
};
|
|
|
|
static int __in_mask_map[] =
|
|
{
|
|
__MASK_READ,
|
|
__MASK_RDWR,
|
|
__MASK_READ,
|
|
__MASK_READ
|
|
};
|
|
|
|
xp_assert (in_type >= 0 && in_type <= xp_countof(__in_type_map));
|
|
xp_assert (in_type >= 0 && in_type <= xp_countof(__in_opt_map));
|
|
xp_assert (in_type >= 0 && in_type <= xp_countof(__in_mask_map));
|
|
|
|
/* translate the in_type into the relevant extio type and option */
|
|
extio_type = __in_type_map[in_type];
|
|
extio_opt = __in_opt_map[in_type];
|
|
extio_mask = __in_mask_map[in_type];
|
|
|
|
handler = run->awk->extio[extio_type];
|
|
if (handler == XP_NULL)
|
|
{
|
|
/* no io handler provided */
|
|
*errnum = XP_AWK_EIOIMPL; /* TODO: change the error code */
|
|
return -1;
|
|
}
|
|
|
|
while (p != XP_NULL)
|
|
{
|
|
if (p->type == (extio_type | extio_mask) &&
|
|
xp_strcmp(p->name,name) == 0) break;
|
|
p = p->next;
|
|
}
|
|
|
|
if (p == XP_NULL)
|
|
{
|
|
p = (xp_awk_extio_t*) xp_malloc (xp_sizeof(xp_awk_extio_t));
|
|
if (p == XP_NULL)
|
|
{
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
p->name = xp_strdup (name);
|
|
if (p->name == XP_NULL)
|
|
{
|
|
xp_free (p);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
p->type = (extio_type | extio_mask);
|
|
p->handle = XP_NULL;
|
|
|
|
p->in.buf[0] = XP_C('\0');
|
|
p->in.pos = 0;
|
|
p->in.len = 0;
|
|
p->in.eof = xp_false;
|
|
|
|
p->next = XP_NULL;
|
|
|
|
if (handler (XP_AWK_IO_OPEN, extio_opt, p, XP_NULL, 0) == -1)
|
|
{
|
|
xp_free (p->name);
|
|
xp_free (p);
|
|
|
|
/* TODO: set ERRNO */
|
|
*errnum = XP_AWK_ENOERR;
|
|
return -1;
|
|
}
|
|
|
|
if (p->handle == XP_NULL)
|
|
{
|
|
xp_free (p->name);
|
|
xp_free (p);
|
|
|
|
/* wrong implementation of user io handler.
|
|
* the correct io handler should set p->handle to
|
|
* non XP_NULL when it returns 0. */
|
|
*errnum = XP_AWK_EIOIMPL;
|
|
return -1;
|
|
}
|
|
|
|
/* chain it */
|
|
p->next = run->extio;
|
|
run->extio = p;
|
|
}
|
|
|
|
/* read a line */
|
|
xp_str_clear (buf);
|
|
|
|
while (1)
|
|
{
|
|
xp_char_t c;
|
|
|
|
if (p->in.pos >= p->in.len)
|
|
{
|
|
xp_ssize_t n;
|
|
|
|
if (p->in.eof)
|
|
{
|
|
if (XP_STR_LEN(buf) == 0) return 0;
|
|
break;
|
|
}
|
|
|
|
n = handler (XP_AWK_IO_READ, 0,
|
|
p, p->in.buf, xp_countof(p->in.buf));
|
|
if (n == -1)
|
|
{
|
|
/* handler error. getline should return -1 */
|
|
/* TODO: set ERRNO */
|
|
*errnum = XP_AWK_ENOERR;
|
|
return -1;
|
|
}
|
|
|
|
if (n == 0)
|
|
{
|
|
p->in.eof = xp_true;
|
|
if (XP_STR_LEN(buf) == 0) return 0;
|
|
break;
|
|
}
|
|
|
|
p->in.len = n;
|
|
p->in.pos = 0;
|
|
}
|
|
|
|
c = p->in.buf[p->in.pos++];
|
|
if (c == XP_C('\n')) break;
|
|
|
|
if (xp_str_ccat (buf, c) == (xp_size_t)-1)
|
|
{
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
int xp_awk_writeextio (
|
|
xp_awk_run_t* run, int out_type,
|
|
const xp_char_t* name, xp_awk_val_t* v, int* errnum)
|
|
{
|
|
xp_awk_extio_t* p = run->extio;
|
|
xp_awk_io_t handler;
|
|
xp_str_t buf;
|
|
int extio_type, extio_opt, extio_mask;
|
|
|
|
static int __out_type_map[] =
|
|
{
|
|
/* the order should match the order of the
|
|
* XP_AWK_OUT_XXX values in tree.h */
|
|
XP_AWK_EXTIO_PIPE,
|
|
XP_AWK_EXTIO_COPROC,
|
|
XP_AWK_EXTIO_FILE,
|
|
XP_AWK_EXTIO_FILE,
|
|
XP_AWK_EXTIO_CONSOLE
|
|
};
|
|
|
|
static int __out_opt_map[] =
|
|
{
|
|
/* the order should match the order of the
|
|
* XP_AWK_OUT_XXX values in tree.h */
|
|
XP_AWK_IO_PIPE_WRITE,
|
|
0,
|
|
XP_AWK_IO_FILE_WRITE,
|
|
XP_AWK_IO_FILE_APPEND,
|
|
XP_AWK_IO_CONSOLE_WRITE
|
|
};
|
|
|
|
static int __out_mask_map[] =
|
|
{
|
|
__MASK_WRITE,
|
|
__MASK_RDWR,
|
|
__MASK_WRITE,
|
|
__MASK_WRITE,
|
|
__MASK_WRITE
|
|
};
|
|
|
|
xp_assert (out_type >= 0 && out_type <= xp_countof(__out_type_map));
|
|
xp_assert (out_type >= 0 && out_type <= xp_countof(__out_opt_map));
|
|
xp_assert (out_type >= 0 && out_type <= xp_countof(__out_mask_map));
|
|
|
|
/* translate the out_type into the relevant extio type and option */
|
|
extio_type = __out_type_map[out_type];
|
|
extio_opt = __out_opt_map[out_type];
|
|
extio_mask = __out_mask_map[out_type];
|
|
|
|
handler = run->awk->extio[extio_type];
|
|
if (handler == XP_NULL)
|
|
{
|
|
/* no io handler provided */
|
|
*errnum = XP_AWK_EIOIMPL; /* TODO: change the error code */
|
|
return -1;
|
|
}
|
|
|
|
/* TODO: optimize the buffer management.
|
|
* each xp_awk_run_t may have a buffer for this. */
|
|
if (xp_str_open (&buf, 256) == XP_NULL)
|
|
{
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
/* convert the value to string representation first */
|
|
if (xp_awk_valtostr(v, errnum, &buf) == XP_NULL)
|
|
{
|
|
xp_str_close (&buf);
|
|
return -1;
|
|
}
|
|
|
|
/* look for the corresponding extio for name */
|
|
while (p != XP_NULL)
|
|
{
|
|
/* the file "1.tmp", in the following code snippets,
|
|
* would be opened by the first print statement, but not by
|
|
* the second print statement. this is because
|
|
* both XP_AWK_OUT_FILE and XP_AWK_OUT_FILE_APPEND are
|
|
* translated to XP_AWK_EXTIO_FILE and it is used to
|
|
* keep track of file handles..
|
|
*
|
|
* print "1111" >> "1.tmp"
|
|
* print "1111" > "1.tmp"
|
|
*/
|
|
|
|
if (p->type == (extio_type | extio_mask) &&
|
|
xp_strcmp(p->name,name) == 0) break;
|
|
p = p->next;
|
|
}
|
|
|
|
/* if there is not corresponding extio for name, create one */
|
|
if (p == XP_NULL)
|
|
{
|
|
p = (xp_awk_extio_t*) xp_malloc (xp_sizeof(xp_awk_extio_t));
|
|
if (p == XP_NULL)
|
|
{
|
|
xp_str_close (&buf);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
p->name = xp_strdup (name);
|
|
if (p->name == XP_NULL)
|
|
{
|
|
xp_free (p);
|
|
xp_str_close (&buf);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
p->type = (extio_type | extio_mask);
|
|
p->handle = XP_NULL;
|
|
p->next = XP_NULL;
|
|
|
|
if (handler (XP_AWK_IO_OPEN, extio_opt, p, XP_NULL, 0) == -1)
|
|
{
|
|
xp_free (p->name);
|
|
xp_free (p);
|
|
xp_str_close (&buf);
|
|
|
|
/* TODO: set ERRNO */
|
|
*errnum = XP_AWK_ENOERR;
|
|
return -1;
|
|
}
|
|
|
|
if (p->handle == XP_NULL)
|
|
{
|
|
xp_free (p->name);
|
|
xp_free (p);
|
|
xp_str_close (&buf);
|
|
|
|
/* wrong implementation of the user io handler.
|
|
* the correct io handler should set p->handle to
|
|
* non XP_NULL when it returns 0. */
|
|
*errnum = XP_AWK_EIOIMPL;
|
|
return -1;
|
|
}
|
|
|
|
/* chain it */
|
|
p->next = run->extio;
|
|
run->extio = p;
|
|
}
|
|
|
|
if (handler (XP_AWK_IO_WRITE, 0,
|
|
p, XP_STR_BUF(&buf), XP_STR_LEN(&buf)) == 0)
|
|
{
|
|
xp_str_close (&buf);
|
|
return 0;
|
|
}
|
|
|
|
xp_str_close (&buf);
|
|
return 1;
|
|
}
|
|
|
|
int xp_awk_closeextio (xp_awk_run_t* run, const xp_char_t* name, int* errnum)
|
|
{
|
|
xp_awk_extio_t* p = run->extio, * px = XP_NULL;
|
|
|
|
while (p != XP_NULL)
|
|
{
|
|
/* it handles the first that matches the given name
|
|
* regardless of the extio type */
|
|
if (xp_strcmp(p->name,name) == 0)
|
|
{
|
|
xp_awk_io_t handler;
|
|
|
|
handler = run->awk->extio[p->type & __MASK_CLEAR];
|
|
if (handler != NULL)
|
|
{
|
|
/* TODO: io command should not be XP_AWK_IO_CLOSE
|
|
* it should be more generic form than this... */
|
|
if (handler (XP_AWK_IO_CLOSE, 0, p, XP_NULL, 0) == -1)
|
|
{
|
|
/* this is not a run-time error.*/
|
|
*errnum = XP_AWK_ENOERR;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (px != XP_NULL) px->next = p->next;
|
|
else run->extio = p->next;
|
|
|
|
xp_free (p->name);
|
|
xp_free (p);
|
|
return 0;
|
|
}
|
|
|
|
px = p;
|
|
p = p->next;
|
|
}
|
|
|
|
/* this is not a run-time error */
|
|
*errnum = XP_AWK_ENOERR;
|
|
return -1;
|
|
}
|
|
|
|
void xp_awk_clearextio (xp_awk_run_t* run)
|
|
{
|
|
xp_awk_extio_t* next;
|
|
xp_awk_io_t handler;
|
|
int n;
|
|
|
|
while (run->extio != XP_NULL)
|
|
{
|
|
handler = run->awk->extio[run->extio->type & __MASK_CLEAR];
|
|
next = run->extio->next;
|
|
|
|
if (handler != XP_NULL)
|
|
{
|
|
/* TODO: io command should not be XP_AWK_INPUT_CLOSE
|
|
* it should be more generic form than this... */
|
|
n = handler (XP_AWK_IO_CLOSE, 0, run->extio, XP_NULL, 0);
|
|
if (n == -1)
|
|
{
|
|
/* TODO:
|
|
* some warning actions need to be taken */
|
|
}
|
|
}
|
|
|
|
xp_free (run->extio->name);
|
|
xp_free (run->extio);
|
|
|
|
run->extio = next;
|
|
}
|
|
}
|