qse/qse/lib/awk/std.c

2017 lines
42 KiB
C
Raw Normal View History

2008-08-04 08:14:52 +00:00
/*
* $Id: std.c 538 2011-08-09 16:08:26Z hyunghwan.chung $
2008-12-27 04:35:14 +00:00
*
Copyright 2006-2012 Chung, Hyung-Hwan.
2009-09-16 04:01:02 +00:00
This file is part of QSE.
2008-12-27 04:35:14 +00:00
2009-09-16 04:01:02 +00:00
QSE is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as
published by the Free Software Foundation, either version 3 of
the License, or (at your option) any later version.
2008-12-27 04:35:14 +00:00
2009-09-16 04:01:02 +00:00
QSE 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 Lesser General Public License for more details.
2008-12-27 04:35:14 +00:00
2009-09-16 04:01:02 +00:00
You should have received a copy of the GNU Lesser General Public
License along with QSE. If not, see <http://www.gnu.org/licenses/>.
2008-08-04 08:14:52 +00:00
*/
2008-08-21 03:17:25 +00:00
#include "awk.h"
#include <qse/awk/std.h>
2008-12-21 21:35:07 +00:00
#include <qse/cmn/sio.h>
#include <qse/cmn/pio.h>
2012-04-30 09:46:58 +00:00
#include <qse/cmn/nwio.h>
2008-12-21 21:35:07 +00:00
#include <qse/cmn/str.h>
2011-12-31 15:24:48 +00:00
#include <qse/cmn/mbwc.h>
2008-12-21 21:35:07 +00:00
#include <qse/cmn/time.h>
2011-10-14 22:57:41 +00:00
#include <qse/cmn/path.h>
#include <qse/cmn/htb.h>
2009-06-04 15:50:32 +00:00
#include <qse/cmn/stdio.h> /* TODO: remove dependency on qse_vsprintf */
#include "../cmn/mem.h"
2008-12-11 04:19:59 +00:00
#include <stdarg.h>
#include <stdlib.h>
2009-08-29 05:58:05 +00:00
#include <math.h>
2008-08-04 08:13:03 +00:00
2011-04-18 09:28:22 +00:00
#if defined(_WIN32)
# include <tchar.h>
#endif
#ifndef QSE_HAVE_CONFIG_H
2011-05-04 08:00:38 +00:00
# if defined(_WIN32) || defined(__OS2__) || defined(__DOS__)
2011-04-18 09:28:22 +00:00
# define HAVE_POW
2011-07-25 08:42:23 +00:00
# define HAVE_FMOD
2011-04-18 09:28:22 +00:00
# define HAVE_SIN
# define HAVE_COS
# define HAVE_TAN
# define HAVE_ATAN
# define HAVE_ATAN2
# define HAVE_LOG
2012-02-20 14:45:49 +00:00
# define HAVE_LOG10
2011-04-18 09:28:22 +00:00
# define HAVE_EXP
# define HAVE_SQRT
# endif
#endif
2008-12-16 03:56:48 +00:00
typedef struct xtn_t
{
struct
{
2009-02-16 08:31:34 +00:00
struct
{
qse_awk_parsestd_t* x;
2009-02-16 08:31:34 +00:00
union
{
struct
{
qse_sio_t* sio; /* the handle to an open file */
qse_cstr_t dir;
} file;
2009-02-22 06:28:02 +00:00
struct
{
const qse_char_t* ptr;
const qse_char_t* end;
} str;
2009-02-18 07:55:48 +00:00
} u;
2009-02-16 08:31:34 +00:00
} in;
struct
{
qse_awk_parsestd_t* x;
2009-02-21 23:06:45 +00:00
union
{
struct
{
qse_sio_t* sio;
} file;
2009-02-22 06:28:02 +00:00
struct
{
qse_str_t* buf;
} str;
2009-02-21 23:06:45 +00:00
} u;
2009-02-16 08:31:34 +00:00
} out;
2012-04-27 14:33:14 +00:00
} s; /* script/source handling */
2009-02-16 08:31:34 +00:00
} xtn_t;
typedef struct rxtn_t
{
unsigned int seed;
2009-02-16 08:31:34 +00:00
struct
{
struct {
2009-02-18 07:55:48 +00:00
const qse_char_t*const* files;
qse_size_t index;
qse_size_t count;
2009-02-16 08:31:34 +00:00
} in;
struct
{
2009-02-18 07:55:48 +00:00
const qse_char_t*const* files;
qse_size_t index;
qse_size_t count;
} out;
qse_cmgr_t* cmgr;
2009-02-16 08:31:34 +00:00
} c; /* console */
int cmgrtab_inited;
qse_htb_t cmgrtab;
} rxtn_t;
typedef struct ioattr_t
{
qse_cmgr_t* cmgr;
qse_char_t cmgr_name[64]; /* i assume that the cmgr name never exceeds this length */
qse_long_t timeout[3];
} ioattr_t;
static qse_flt_t custom_awk_pow (qse_awk_t* awk, qse_flt_t x, qse_flt_t y)
2008-08-04 08:13:03 +00:00
{
#if defined(HAVE_POWL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return powl (x, y);
#elif defined(HAVE_POW)
2008-08-04 08:13:03 +00:00
return pow (x, y);
#elif defined(HAVE_POWF)
return powf (x, y);
#else
#error ### no pow function available ###
#endif
2008-08-04 08:13:03 +00:00
}
static qse_flt_t custom_awk_mod (qse_awk_t* awk, qse_flt_t x, qse_flt_t y)
{
#if defined(HAVE_FMODL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return fmodl (x, y);
#elif defined(HAVE_FMOD)
return fmod (x, y);
#elif defined(HAVE_FMODF)
return fmodf (x, y);
#else
#error ### no fmod function available ###
#endif
}
static qse_flt_t custom_awk_sin (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_SINL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return sinl (x);
#elif defined(HAVE_SIN)
return sin (x);
#elif defined(HAVE_SINF)
return sinf (x);
#else
#error ### no sin function available ###
#endif
}
static qse_flt_t custom_awk_cos (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_COSL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return cosl (x);
#elif defined(HAVE_COS)
return cos (x);
#elif defined(HAVE_COSF)
return cosf (x);
#else
#error ### no cos function available ###
#endif
}
static qse_flt_t custom_awk_tan (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_TANL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return tanl (x);
#elif defined(HAVE_TAN)
return tan (x);
#elif defined(HAVE_TANF)
return tanf (x);
#else
#error ### no tan function available ###
#endif
}
static qse_flt_t custom_awk_atan (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_ATANL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return atanl (x);
#elif defined(HAVE_ATAN)
return atan (x);
#elif defined(HAVE_ATANF)
return atanf (x);
#else
#error ### no atan function available ###
#endif
}
static qse_flt_t custom_awk_atan2 (qse_awk_t* awk, qse_flt_t x, qse_flt_t y)
{
#if defined(HAVE_ATAN2L) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return atan2l (x, y);
#elif defined(HAVE_ATAN2)
return atan2 (x, y);
#elif defined(HAVE_ATAN2F)
return atan2f (x, y);
#else
#error ### no atan2 function available ###
#endif
}
static qse_flt_t custom_awk_log (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_LOGL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return logl (x);
#elif defined(HAVE_LOG)
return log (x);
#elif defined(HAVE_LOGF)
return logf (x);
#else
#error ### no log function available ###
#endif
}
2012-01-26 14:19:48 +00:00
static qse_flt_t custom_awk_log10 (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_LOG10L) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return log10l (x);
#elif defined(HAVE_LOG10)
return log10 (x);
#elif defined(HAVE_LOG10F)
return log10f (x);
#else
#error ### no log10 function available ###
#endif
}
static qse_flt_t custom_awk_exp (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_EXPL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return expl (x);
#elif defined(HAVE_EXP)
return exp (x);
#elif defined(HAVE_EXPF)
return expf (x);
#else
#error ### no exp function available ###
#endif
}
static qse_flt_t custom_awk_sqrt (qse_awk_t* awk, qse_flt_t x)
{
#if defined(HAVE_SQRTL) && (QSE_SIZEOF_LONG_DOUBLE > QSE_SIZEOF_DOUBLE)
return sqrtl (x);
#elif defined(HAVE_SQRT)
return sqrt (x);
#elif defined(HAVE_SQRTF)
return sqrtf (x);
#else
#error ### no sqrt function available ###
#endif
}
2008-08-04 08:13:03 +00:00
static int custom_awk_sprintf (
2009-02-16 08:31:34 +00:00
qse_awk_t* awk, qse_char_t* buf, qse_size_t size,
2008-12-21 21:35:07 +00:00
const qse_char_t* fmt, ...)
2008-08-04 08:13:03 +00:00
{
int n;
va_list ap;
va_start (ap, fmt);
2008-12-21 21:35:07 +00:00
n = qse_vsprintf (buf, size, fmt, ap);
2008-08-04 08:13:03 +00:00
va_end (ap);
return n;
}
2008-12-21 21:35:07 +00:00
static int add_functions (qse_awk_t* awk);
qse_awk_t* qse_awk_openstd (qse_size_t xtnsize)
2010-07-25 06:43:26 +00:00
{
return qse_awk_openstdwithmmgr (QSE_MMGR_GETDFL(), xtnsize);
}
qse_awk_t* qse_awk_openstdwithmmgr (qse_mmgr_t* mmgr, qse_size_t xtnsize)
2008-08-04 08:13:03 +00:00
{
2008-12-21 21:35:07 +00:00
qse_awk_t* awk;
2009-02-16 08:31:34 +00:00
qse_awk_prm_t prm;
xtn_t* xtn;
2008-08-04 08:13:03 +00:00
prm.sprintf = custom_awk_sprintf;
prm.math.pow = custom_awk_pow;
prm.math.mod = custom_awk_mod;
prm.math.sin = custom_awk_sin;
prm.math.cos = custom_awk_cos;
prm.math.tan = custom_awk_tan;
prm.math.atan = custom_awk_atan;
prm.math.atan2 = custom_awk_atan2;
prm.math.log = custom_awk_log;
2012-01-26 14:19:48 +00:00
prm.math.log10 = custom_awk_log10;
prm.math.exp = custom_awk_exp;
prm.math.sqrt = custom_awk_sqrt;
2009-02-17 02:11:31 +00:00
2009-02-16 08:31:34 +00:00
/* create an object */
2010-07-25 06:43:26 +00:00
awk = qse_awk_open (mmgr, QSE_SIZEOF(xtn_t) + xtnsize, &prm);
2009-02-16 08:31:34 +00:00
if (awk == QSE_NULL) return QSE_NULL;
2008-08-04 08:13:03 +00:00
2009-02-16 08:31:34 +00:00
/* initialize extension */
2009-02-17 02:11:31 +00:00
xtn = (xtn_t*) QSE_XTN (awk);
2009-02-16 08:31:34 +00:00
QSE_MEMSET (xtn, 0, QSE_SIZEOF(xtn_t));
2008-12-16 03:56:48 +00:00
2009-02-16 08:31:34 +00:00
/* add intrinsic functions */
2011-05-26 08:42:26 +00:00
if (add_functions (awk) <= -1)
{
2008-12-21 21:35:07 +00:00
qse_awk_close (awk);
return QSE_NULL;
}
2008-08-04 08:13:03 +00:00
return awk;
}
2008-10-01 05:14:20 +00:00
void* qse_awk_getxtnstd (qse_awk_t* awk)
{
return (void*)((xtn_t*)QSE_XTN(awk) + 1);
}
static qse_sio_t* open_sio (qse_awk_t* awk, const qse_char_t* file, int flags)
{
qse_sio_t* sio;
sio = qse_sio_open (awk->mmgr, 0, file, flags);
if (sio == QSE_NULL)
{
qse_cstr_t errarg;
errarg.ptr = file;
errarg.len = qse_strlen(file);
qse_awk_seterrnum (awk, QSE_AWK_EOPEN, &errarg);
}
return sio;
}
static qse_sio_t* open_sio_rtx (qse_awk_rtx_t* rtx, const qse_char_t* file, int flags)
{
qse_sio_t* sio;
sio = qse_sio_open (rtx->awk->mmgr, 0, file, flags);
if (sio == QSE_NULL)
{
qse_cstr_t errarg;
errarg.ptr = file;
errarg.len = qse_strlen(file);
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EOPEN, &errarg);
}
return sio;
}
struct sio_std_name_t
{
const qse_char_t* ptr;
qse_size_t len;
};
static struct sio_std_name_t sio_std_names[] =
{
{ QSE_T("stdin"), 5 },
{ QSE_T("stdout"), 6 },
{ QSE_T("stderr"), 6 }
};
2008-12-11 04:19:59 +00:00
static qse_sio_t* open_sio_std (qse_awk_t* awk, qse_sio_std_t std, int flags)
{
qse_sio_t* sio;
sio = qse_sio_openstd (awk->mmgr, 0, std, flags);
if (sio == QSE_NULL)
{
qse_cstr_t ea;
ea.ptr = sio_std_names[std].ptr;
ea.len = sio_std_names[std].len;
qse_awk_seterrnum (awk, QSE_AWK_EOPEN, &ea);
}
return sio;
}
static qse_sio_t* open_sio_std_rtx (qse_awk_rtx_t* rtx, qse_sio_std_t std, int flags)
{
qse_sio_t* sio;
sio = qse_sio_openstd (rtx->awk->mmgr, 0, std, flags);
if (sio == QSE_NULL)
{
qse_cstr_t ea;
ea.ptr = sio_std_names[std].ptr;
ea.len = sio_std_names[std].len;
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EOPEN, &ea);
}
return sio;
}
/*** PARSESTD ***/
2009-07-27 20:31:58 +00:00
static qse_ssize_t sf_in_open (
qse_awk_t* awk, qse_awk_sio_arg_t* arg, xtn_t* xtn)
2008-10-01 05:14:20 +00:00
{
2009-07-27 20:31:58 +00:00
if (arg == QSE_NULL || arg->name == QSE_NULL)
2008-10-01 05:14:20 +00:00
{
switch (xtn->s.in.x->type)
2008-10-14 05:32:58 +00:00
{
case QSE_AWK_PARSESTD_FILE:
if (xtn->s.in.x->u.file.path == QSE_NULL ||
(xtn->s.in.x->u.file.path[0] == QSE_T('-') &&
xtn->s.in.x->u.file.path[1] == QSE_T('\0')))
2009-06-24 07:29:18 +00:00
{
/* special file name '-' */
xtn->s.in.u.file.sio = open_sio_std (
awk, QSE_SIO_STDIN, QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR);
if (xtn->s.in.u.file.sio == QSE_NULL) return -1;
2009-06-24 07:29:18 +00:00
}
else
{
2009-08-10 21:29:59 +00:00
const qse_char_t* base;
xtn->s.in.u.file.sio = open_sio (
awk, xtn->s.in.x->u.file.path,
QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR
2009-06-24 07:29:18 +00:00
);
if (xtn->s.in.u.file.sio == QSE_NULL) return -1;
2009-08-10 21:29:59 +00:00
base = qse_basename (xtn->s.in.x->u.file.path);
if (base != xtn->s.in.x->u.file.path)
2009-08-10 21:29:59 +00:00
{
xtn->s.in.u.file.dir.ptr = xtn->s.in.x->u.file.path;
xtn->s.in.u.file.dir.len = base - xtn->s.in.x->u.file.path;
2009-08-10 21:29:59 +00:00
}
2009-06-24 07:29:18 +00:00
}
if (xtn->s.in.x->u.file.cmgr)
qse_sio_setcmgr (xtn->s.in.u.file.sio, xtn->s.in.x->u.file.cmgr);
2009-02-22 06:28:02 +00:00
return 1;
2008-12-13 03:42:32 +00:00
case QSE_AWK_PARSESTD_STR:
xtn->s.in.u.str.ptr = xtn->s.in.x->u.str.ptr;
xtn->s.in.u.str.end = xtn->s.in.x->u.str.ptr + xtn->s.in.x->u.str.len;
2009-02-22 06:28:02 +00:00
return 1;
default:
/* this should never happen */
qse_awk_seterrnum (awk, QSE_AWK_EINTERN, QSE_NULL);
return -1;
2009-02-22 06:28:02 +00:00
}
2011-05-26 08:42:26 +00:00
2008-10-01 05:14:20 +00:00
}
2009-07-27 20:31:58 +00:00
else
{
2009-08-10 21:29:59 +00:00
const qse_char_t* file = arg->name;
qse_char_t fbuf[64];
qse_char_t* dbuf = QSE_NULL;
if (xtn->s.in.u.file.dir.len > 0 && arg->name[0] != QSE_T('/'))
2009-08-10 21:29:59 +00:00
{
qse_size_t tmplen, totlen;
totlen = qse_strlen(arg->name) + xtn->s.in.u.file.dir.len;
2009-08-10 21:29:59 +00:00
if (totlen >= QSE_COUNTOF(fbuf))
{
dbuf = QSE_MMGR_ALLOC (
awk->mmgr,
QSE_SIZEOF(qse_char_t) * (totlen + 1)
);
if (dbuf == QSE_NULL)
{
qse_awk_seterrnum (awk, QSE_AWK_ENOMEM, QSE_NULL);
2009-08-10 21:29:59 +00:00
return -1;
}
file = dbuf;
}
else file = fbuf;
tmplen = qse_strncpy (
(qse_char_t*)file,
xtn->s.in.u.file.dir.ptr,
xtn->s.in.u.file.dir.len
2009-08-10 21:29:59 +00:00
);
qse_strcpy ((qse_char_t*)file + tmplen, arg->name);
}
2009-07-27 20:31:58 +00:00
arg->handle = qse_sio_open (
awk->mmgr, 0, file, QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR
2009-07-27 20:31:58 +00:00
);
2009-08-10 21:29:59 +00:00
if (dbuf != QSE_NULL) QSE_MMGR_FREE (awk->mmgr, dbuf);
if (arg->handle == QSE_NULL)
2009-07-27 20:31:58 +00:00
{
qse_cstr_t ea;
ea.ptr = arg->name;
ea.len = qse_strlen(ea.ptr);
qse_awk_seterrnum (awk, QSE_AWK_EOPEN, &ea);
2009-07-27 20:31:58 +00:00
return -1;
}
return 1;
}
}
static qse_ssize_t sf_in_close (
qse_awk_t* awk, qse_awk_sio_arg_t* arg, xtn_t* xtn)
{
if (arg == QSE_NULL || arg->name == QSE_NULL)
2008-10-01 05:14:20 +00:00
{
switch (xtn->s.in.x->type)
{
case QSE_AWK_PARSESTD_FILE:
qse_sio_close (xtn->s.in.u.file.sio);
break;
case QSE_AWK_PARSESTD_STR:
/* nothing to close */
break;
default:
/* nothing to close */
break;
}
2008-10-01 05:14:20 +00:00
}
2009-07-27 20:31:58 +00:00
else
{
QSE_ASSERT (arg->handle != QSE_NULL);
2009-07-27 20:31:58 +00:00
qse_sio_close (arg->handle);
}
return 0;
}
static qse_ssize_t sf_in_read (
qse_awk_t* awk, qse_awk_sio_arg_t* arg,
qse_char_t* data, qse_size_t size, xtn_t* xtn)
{
if (arg == QSE_NULL || arg->name == QSE_NULL)
2008-10-01 05:14:20 +00:00
{
switch (xtn->s.in.x->type)
2008-12-11 04:19:59 +00:00
{
case QSE_AWK_PARSESTD_FILE:
2009-07-25 23:18:42 +00:00
{
qse_ssize_t n;
QSE_ASSERT (xtn->s.in.u.file.sio != QSE_NULL);
n = qse_sio_getstrn (xtn->s.in.u.file.sio, data, size);
if (n <= -1)
2009-07-25 23:18:42 +00:00
{
2009-07-27 20:31:58 +00:00
qse_cstr_t ea;
if (xtn->s.in.x->u.file.path)
{
ea.ptr = xtn->s.in.x->u.file.path;
ea.len = qse_strlen(ea.ptr);
}
else
{
ea.ptr = sio_std_names[QSE_SIO_STDIN].ptr;
ea.len = sio_std_names[QSE_SIO_STDIN].len;
}
qse_awk_seterrnum (awk, QSE_AWK_EREAD, &ea);
2009-07-25 23:18:42 +00:00
}
return n;
}
2008-12-13 03:42:32 +00:00
case QSE_AWK_PARSESTD_STR:
2008-12-11 04:19:59 +00:00
{
2009-02-22 06:28:02 +00:00
qse_size_t n = 0;
while (n < size && xtn->s.in.u.str.ptr < xtn->s.in.u.str.end)
2009-02-22 06:28:02 +00:00
{
data[n++] = *xtn->s.in.u.str.ptr++;
2009-02-22 06:28:02 +00:00
}
return n;
2008-12-11 04:19:59 +00:00
}
default:
/* this should never happen */
qse_awk_seterrnum (awk, QSE_AWK_EINTERN, QSE_NULL);
return -1;
2008-10-01 05:14:20 +00:00
}
2011-05-26 08:42:26 +00:00
2008-10-01 05:14:20 +00:00
}
2009-07-27 20:31:58 +00:00
else
{
qse_ssize_t n;
2008-10-01 05:14:20 +00:00
2009-07-27 20:31:58 +00:00
QSE_ASSERT (arg->handle != QSE_NULL);
n = qse_sio_getstrn (arg->handle, data, size);
if (n <= -1)
2009-07-27 20:31:58 +00:00
{
qse_cstr_t ea;
ea.ptr = arg->name;
ea.len = qse_strlen(ea.ptr);
qse_awk_seterrnum (awk, QSE_AWK_EREAD, &ea);
2009-07-27 20:31:58 +00:00
}
return n;
}
}
static qse_ssize_t sf_in (
qse_awk_t* awk, qse_awk_sio_cmd_t cmd,
qse_awk_sio_arg_t* arg, qse_char_t* data, qse_size_t size)
{
xtn_t* xtn = QSE_XTN (awk);
switch (cmd)
{
case QSE_AWK_SIO_OPEN:
return sf_in_open (awk, arg, xtn);
case QSE_AWK_SIO_CLOSE:
return sf_in_close (awk, arg, xtn);
case QSE_AWK_SIO_READ:
return sf_in_read (awk, arg, data, size, xtn);
default:
qse_awk_seterrnum (awk, QSE_AWK_EINTERN, QSE_NULL);
return -1;
2009-07-27 20:31:58 +00:00
}
2008-10-01 05:14:20 +00:00
}
2009-02-17 02:11:31 +00:00
static qse_ssize_t sf_out (
2009-07-25 23:18:42 +00:00
qse_awk_t* awk, qse_awk_sio_cmd_t cmd,
qse_awk_sio_arg_t* arg, qse_char_t* data, qse_size_t size)
2008-10-01 05:14:20 +00:00
{
2009-02-17 02:11:31 +00:00
xtn_t* xtn = QSE_XTN (awk);
2008-10-14 05:32:58 +00:00
2011-05-26 08:42:26 +00:00
switch (cmd)
2008-10-14 05:32:58 +00:00
{
2011-05-26 08:42:26 +00:00
case QSE_AWK_SIO_OPEN:
2008-10-14 05:32:58 +00:00
{
switch (xtn->s.out.x->type)
2011-05-26 08:42:26 +00:00
{
case QSE_AWK_PARSESTD_FILE:
if (xtn->s.out.x->u.file.path == QSE_NULL ||
(xtn->s.out.x->u.file.path[0] == QSE_T('-') &&
xtn->s.out.x->u.file.path[1] == QSE_T('\0')))
2009-07-25 23:18:42 +00:00
{
2011-05-26 08:42:26 +00:00
/* special file name '-' */
xtn->s.out.u.file.sio = open_sio_std (
awk, QSE_SIO_STDOUT,
QSE_SIO_WRITE | QSE_SIO_IGNOREMBWCERR
);
if (xtn->s.out.u.file.sio == QSE_NULL) return -1;
2009-07-25 23:18:42 +00:00
}
2011-05-26 08:42:26 +00:00
else
{
xtn->s.out.u.file.sio = open_sio (
awk, xtn->s.out.x->u.file.path,
QSE_SIO_WRITE | QSE_SIO_CREATE |
QSE_SIO_TRUNCATE | QSE_SIO_IGNOREMBWCERR
2011-05-26 08:42:26 +00:00
);
if (xtn->s.out.u.file.sio == QSE_NULL) return -1;
2011-05-26 08:42:26 +00:00
}
2009-02-22 06:28:02 +00:00
if (xtn->s.out.x->u.file.cmgr)
qse_sio_setcmgr (xtn->s.out.u.file.sio, xtn->s.out.x->u.file.cmgr);
2011-05-26 08:42:26 +00:00
return 1;
2009-02-22 06:28:02 +00:00
case QSE_AWK_PARSESTD_STR:
xtn->s.out.u.str.buf = qse_str_open (awk->mmgr, 0, 512);
if (xtn->s.out.u.str.buf == QSE_NULL)
{
qse_awk_seterrnum (awk, QSE_AWK_ENOMEM, QSE_NULL);
return -1;
}
2011-05-26 08:42:26 +00:00
return 1;
}
2008-12-12 04:05:28 +00:00
2011-05-26 08:42:26 +00:00
break;
}
2009-02-22 06:28:02 +00:00
2011-05-26 08:42:26 +00:00
case QSE_AWK_SIO_CLOSE:
2009-02-22 06:28:02 +00:00
{
switch (xtn->s.out.x->type)
2009-07-25 23:18:42 +00:00
{
2011-05-26 08:42:26 +00:00
case QSE_AWK_PARSESTD_FILE:
qse_sio_close (xtn->s.out.u.file.sio);
2011-05-26 08:42:26 +00:00
return 0;
2009-07-25 23:18:42 +00:00
case QSE_AWK_PARSESTD_STR:
/* i don't close xtn->s.out.u.str.buf intentionally here.
* it will be closed at the end of qse_awk_parsestd() */
2011-05-26 08:42:26 +00:00
return 0;
2009-07-25 23:18:42 +00:00
}
2009-02-22 06:28:02 +00:00
2011-05-26 08:42:26 +00:00
break;
}
case QSE_AWK_SIO_WRITE:
{
switch (xtn->s.out.x->type)
2009-02-22 06:28:02 +00:00
{
2011-05-26 08:42:26 +00:00
case QSE_AWK_PARSESTD_FILE:
2009-02-22 06:28:02 +00:00
{
2011-05-26 08:42:26 +00:00
qse_ssize_t n;
QSE_ASSERT (xtn->s.out.u.file.sio != QSE_NULL);
n = qse_sio_putstrn (xtn->s.out.u.file.sio, data, size);
if (n <= -1)
2011-05-26 08:42:26 +00:00
{
qse_cstr_t ea;
if (xtn->s.out.x->u.file.path)
{
ea.ptr = xtn->s.out.x->u.file.path;
ea.len = qse_strlen(ea.ptr);
}
else
{
ea.ptr = sio_std_names[QSE_SIO_STDOUT].ptr;
ea.len = sio_std_names[QSE_SIO_STDOUT].len;
}
2011-05-26 08:42:26 +00:00
qse_awk_seterrnum (awk, QSE_AWK_EWRITE, &ea);
}
return n;
}
case QSE_AWK_PARSESTD_STR:
2011-05-26 08:42:26 +00:00
{
if (size > QSE_TYPE_MAX(qse_ssize_t)) size = QSE_TYPE_MAX(qse_ssize_t);
if (qse_str_ncat (xtn->s.out.u.str.buf, data, size) == (qse_size_t)-1)
2011-05-26 08:42:26 +00:00
{
qse_awk_seterrnum (awk, QSE_AWK_ENOMEM, QSE_NULL);
return -1;
2011-05-26 08:42:26 +00:00
}
return size;
2009-02-22 06:28:02 +00:00
}
}
2011-05-26 08:42:26 +00:00
break;
2009-02-22 06:28:02 +00:00
}
2012-04-30 09:46:58 +00:00
default:
/* other code must not trigger this function */
break;
2008-10-01 05:14:20 +00:00
}
2011-05-26 08:42:26 +00:00
qse_awk_seterrnum (awk, QSE_AWK_EINTERN, QSE_NULL);
2008-10-01 05:14:20 +00:00
return -1;
}
int qse_awk_parsestd (
qse_awk_t* awk, qse_awk_parsestd_t* in, qse_awk_parsestd_t* out)
2008-10-01 05:14:20 +00:00
{
qse_awk_sio_t sio;
2009-02-17 02:11:31 +00:00
xtn_t* xtn = (xtn_t*) QSE_XTN (awk);
int n;
2008-10-01 05:14:20 +00:00
2009-02-22 08:16:35 +00:00
if (in == QSE_NULL)
2008-10-14 05:32:58 +00:00
{
2009-02-22 08:16:35 +00:00
/* the input is a must */
qse_awk_seterrnum (awk, QSE_AWK_EINVAL, QSE_NULL);
2009-02-22 08:16:35 +00:00
return -1;
2009-02-21 23:06:45 +00:00
}
2009-02-22 06:28:02 +00:00
if (in->type != QSE_AWK_PARSESTD_FILE &&
in->type != QSE_AWK_PARSESTD_STR)
2008-12-11 04:19:59 +00:00
{
qse_awk_seterrnum (awk, QSE_AWK_EINVAL, QSE_NULL);
return -1;
2008-12-11 04:19:59 +00:00
}
2011-05-26 08:42:26 +00:00
sio.in = sf_in;
xtn->s.in.x = in;
2008-12-13 03:42:32 +00:00
2009-02-22 08:16:35 +00:00
if (out == QSE_NULL) sio.out = QSE_NULL;
else
{
if (out->type != QSE_AWK_PARSESTD_FILE &&
out->type != QSE_AWK_PARSESTD_STR)
2009-02-22 08:16:35 +00:00
{
qse_awk_seterrnum (awk, QSE_AWK_EINVAL, QSE_NULL);
return -1;
2009-02-22 08:16:35 +00:00
}
sio.out = sf_out;
xtn->s.out.x = out;
2009-02-22 08:16:35 +00:00
}
2008-10-01 05:14:20 +00:00
n = qse_awk_parse (awk, &sio);
if (out && out->type == QSE_AWK_PARSESTD_STR)
{
if (n >= 0)
{
QSE_ASSERT (xtn->s.out.u.str.buf != QSE_NULL);
qse_str_yield (xtn->s.out.u.str.buf, &out->u.str, 0);
}
if (xtn->s.out.u.str.buf) qse_str_close (xtn->s.out.u.str.buf);
}
return n;
2008-10-01 05:14:20 +00:00
}
/*** RTX_OPENSTD ***/
2012-04-27 14:33:14 +00:00
2012-04-30 09:46:58 +00:00
static qse_ssize_t nwio_handler_open (
qse_awk_rtx_t* rtx, qse_awk_rio_arg_t* riod, int flags, qse_nwad_t* nwad)
{
qse_nwio_t* handle;
handle = qse_nwio_open (
qse_awk_rtx_getmmgr(rtx), 0, nwad,
flags | QSE_NWIO_TEXT | QSE_NWIO_IGNOREMBWCERR |
QSE_NWIO_REUSEADDR | QSE_NWIO_READNORETRY | QSE_NWIO_WRITENORETRY
2012-04-30 09:46:58 +00:00
);
if (handle == QSE_NULL) return -1;
#if defined(QSE_CHAR_IS_WCHAR)
{
qse_cmgr_t* cmgr = qse_awk_rtx_getcmgrstd (rtx, riod->name);
if (cmgr) qse_nwio_setcmgr (handle, cmgr);
2012-04-30 09:46:58 +00:00
}
#endif
riod->handle2 = (void*)handle;
return 1;
}
static qse_ssize_t nwio_handler_rest (
qse_awk_rtx_t* rtx, qse_awk_rio_cmd_t cmd, qse_awk_rio_arg_t* riod,
2009-02-16 08:31:34 +00:00
qse_char_t* data, qse_size_t size)
2008-12-12 04:05:28 +00:00
{
switch (cmd)
{
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_OPEN:
2008-12-12 04:05:28 +00:00
{
2012-04-30 09:46:58 +00:00
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
return -1;
}
2008-12-12 04:05:28 +00:00
2012-04-30 09:46:58 +00:00
case QSE_AWK_RIO_CLOSE:
{
qse_nwio_close ((qse_nwio_t*)riod->handle2);
riod->handle2 = QSE_NULL;
return 0;
}
case QSE_AWK_RIO_READ:
{
return qse_nwio_read ((qse_nwio_t*)riod->handle2, data, size);
}
case QSE_AWK_RIO_WRITE:
{
return qse_nwio_write ((qse_nwio_t*)riod->handle2, data, size);
}
case QSE_AWK_RIO_FLUSH:
{
/*if (riod->mode == QSE_AWK_RIO_PIPE_READ) return -1;*/
return qse_nwio_flush ((qse_nwio_t*)riod->handle2);
}
case QSE_AWK_RIO_NEXT:
break;
}
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
return -1;
}
static int parse_rwpipe_uri (const qse_char_t* uri, int* flags, qse_nwad_t* nwad)
{
static struct
{
qse_char_t* prefix;
qse_size_t len;
int flags;
} x[] =
{
{ QSE_T("tcp://"), 6, QSE_NWIO_TCP },
{ QSE_T("udp://"), 6, QSE_NWIO_UDP },
{ QSE_T("tcpd://"), 7, QSE_NWIO_TCP | QSE_NWIO_PASSIVE },
{ QSE_T("udpd://"), 7, QSE_NWIO_UDP | QSE_NWIO_PASSIVE }
};
int i;
2008-12-12 04:05:28 +00:00
2012-04-30 09:46:58 +00:00
for (i = 0; i < QSE_COUNTOF(x); i++)
{
if (qse_strzcmp (uri, x[i].prefix, x[i].len) == 0)
{
if (qse_strtonwad (uri + x[i].len, nwad) <= -1) return -1;
*flags = x[i].flags;
return 0;
}
}
return -1;
}
static qse_ssize_t pio_handler_open (
qse_awk_rtx_t* rtx, qse_awk_rio_arg_t* riod)
{
qse_pio_t* handle;
int flags;
if (riod->mode == QSE_AWK_RIO_PIPE_READ)
{
/* TODO: should ERRTOOUT be unset? */
flags = QSE_PIO_READOUT |
QSE_PIO_ERRTOOUT;
}
else if (riod->mode == QSE_AWK_RIO_PIPE_WRITE)
{
flags = QSE_PIO_WRITEIN;
}
else if (riod->mode == QSE_AWK_RIO_PIPE_RW)
{
flags = QSE_PIO_READOUT |
QSE_PIO_ERRTOOUT |
QSE_PIO_WRITEIN;
}
else
{
/* this must not happen */
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
return -1;
}
handle = qse_pio_open (
rtx->awk->mmgr,
0,
riod->name,
QSE_NULL,
flags|QSE_PIO_SHELL|QSE_PIO_TEXT|QSE_PIO_IGNOREMBWCERR
);
if (handle == QSE_NULL) return -1;
#if defined(QSE_CHAR_IS_WCHAR)
2012-04-30 09:46:58 +00:00
{
qse_cmgr_t* cmgr = qse_awk_rtx_getcmgrstd (rtx, riod->name);
if (cmgr)
{
qse_pio_setcmgr (handle, QSE_PIO_IN, cmgr);
qse_pio_setcmgr (handle, QSE_PIO_OUT, cmgr);
qse_pio_setcmgr (handle, QSE_PIO_ERR, cmgr);
}
}
#endif
2012-04-30 09:46:58 +00:00
riod->handle = (void*)handle;
return 1;
}
static qse_ssize_t pio_handler_rest (
qse_awk_rtx_t* rtx, qse_awk_rio_cmd_t cmd, qse_awk_rio_arg_t* riod,
qse_char_t* data, qse_size_t size)
{
switch (cmd)
{
case QSE_AWK_RIO_OPEN:
{
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
return -1;
2008-12-12 04:05:28 +00:00
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_CLOSE:
2008-12-12 04:05:28 +00:00
{
2009-08-28 06:52:20 +00:00
qse_pio_t* pio = (qse_pio_t*)riod->handle;
if (riod->mode == QSE_AWK_RIO_PIPE_RW)
{
/* specialy treatment is needef for rwpipe.
* inspect rwcmode to see if partial closing is
2009-08-28 06:52:20 +00:00
* requested. */
if (riod->rwcmode == QSE_AWK_RIO_CLOSE_READ)
2009-08-28 06:52:20 +00:00
{
qse_pio_end (pio, QSE_PIO_IN);
return 0;
}
if (riod->rwcmode == QSE_AWK_RIO_CLOSE_WRITE)
2009-08-28 06:52:20 +00:00
{
qse_pio_end (pio, QSE_PIO_OUT);
return 0;
}
}
qse_pio_close (pio);
2009-02-16 08:31:34 +00:00
riod->handle = QSE_NULL;
2008-12-12 04:05:28 +00:00
return 0;
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_READ:
2008-12-12 04:05:28 +00:00
{
return qse_pio_read (
2009-02-16 08:31:34 +00:00
(qse_pio_t*)riod->handle,
QSE_PIO_OUT, data, size
2008-12-12 04:05:28 +00:00
);
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_WRITE:
2008-12-12 04:05:28 +00:00
{
return qse_pio_write (
2009-02-16 08:31:34 +00:00
(qse_pio_t*)riod->handle,
QSE_PIO_IN, data, size
2008-12-12 04:05:28 +00:00
);
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_FLUSH:
2008-12-12 04:05:28 +00:00
{
2009-02-16 08:31:34 +00:00
/*if (riod->mode == QSE_AWK_RIO_PIPE_READ) return -1;*/
return qse_pio_flush ((qse_pio_t*)riod->handle, QSE_PIO_IN);
2008-12-12 04:05:28 +00:00
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_NEXT:
2012-04-30 09:46:58 +00:00
break;
2008-12-12 04:05:28 +00:00
}
2012-04-30 09:46:58 +00:00
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
2008-12-12 04:05:28 +00:00
return -1;
}
2012-04-30 09:46:58 +00:00
static qse_ssize_t awk_rio_pipe (
qse_awk_rtx_t* rtx, qse_awk_rio_cmd_t cmd, qse_awk_rio_arg_t* riod,
qse_char_t* data, qse_size_t size)
{
if (cmd == QSE_AWK_RIO_OPEN)
{
int flags;
qse_nwad_t nwad;
if (riod->mode != QSE_AWK_RIO_PIPE_RW ||
parse_rwpipe_uri (riod->name, &flags, &nwad) <= -1)
return pio_handler_open (rtx, riod);
else
return nwio_handler_open (rtx, riod, flags, &nwad);
}
else if (riod->handle2)
return nwio_handler_rest (rtx, cmd, riod, data, size);
else
return pio_handler_rest (rtx, cmd, riod, data, size);
}
2009-02-16 08:31:34 +00:00
static qse_ssize_t awk_rio_file (
qse_awk_rtx_t* rtx, qse_awk_rio_cmd_t cmd, qse_awk_rio_arg_t* riod,
2009-02-16 08:31:34 +00:00
qse_char_t* data, qse_size_t size)
2008-12-12 04:05:28 +00:00
{
switch (cmd)
{
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_OPEN:
2008-12-12 04:05:28 +00:00
{
qse_sio_t* handle;
int flags = QSE_SIO_IGNOREMBWCERR;
2008-12-12 04:05:28 +00:00
switch (riod->mode)
2008-12-12 04:05:28 +00:00
{
case QSE_AWK_RIO_FILE_READ:
flags |= QSE_SIO_READ;
break;
case QSE_AWK_RIO_FILE_WRITE:
flags |= QSE_SIO_WRITE |
QSE_SIO_CREATE |
QSE_SIO_TRUNCATE;
break;
case QSE_AWK_RIO_FILE_APPEND:
flags |= QSE_SIO_APPEND |
QSE_SIO_CREATE;
break;
default:
/* this must not happen */
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
return -1;
2008-12-12 04:05:28 +00:00
}
handle = qse_sio_open (rtx->awk->mmgr, 0, riod->name, flags);
2008-12-21 21:35:07 +00:00
if (handle == QSE_NULL)
2008-12-12 04:05:28 +00:00
{
2008-12-21 21:35:07 +00:00
qse_cstr_t errarg;
2009-02-16 08:31:34 +00:00
errarg.ptr = riod->name;
errarg.len = qse_strlen(riod->name);
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EOPEN, &errarg);
2008-12-12 04:05:28 +00:00
return -1;
}
#if defined(QSE_CHAR_IS_WCHAR)
{
2012-04-27 14:33:14 +00:00
qse_cmgr_t* cmgr = qse_awk_rtx_getcmgrstd (rtx, riod->name);
if (cmgr) qse_sio_setcmgr (handle, cmgr);
}
#endif
2009-02-16 08:31:34 +00:00
riod->handle = (void*)handle;
2008-12-12 04:05:28 +00:00
return 1;
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_CLOSE:
2008-12-12 04:05:28 +00:00
{
qse_sio_close ((qse_sio_t*)riod->handle);
2009-02-16 08:31:34 +00:00
riod->handle = QSE_NULL;
2008-12-12 04:05:28 +00:00
return 0;
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_READ:
2008-12-12 04:05:28 +00:00
{
return qse_sio_getstrn (
(qse_sio_t*)riod->handle,
2008-12-12 04:05:28 +00:00
data,
size
);
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_WRITE:
2008-12-12 04:05:28 +00:00
{
return qse_sio_putstrn (
(qse_sio_t*)riod->handle,
2008-12-12 04:05:28 +00:00
data,
size
);
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_FLUSH:
2008-12-12 04:05:28 +00:00
{
return qse_sio_flush ((qse_sio_t*)riod->handle);
2008-12-12 04:05:28 +00:00
}
2009-02-17 02:11:31 +00:00
case QSE_AWK_RIO_NEXT:
2008-12-12 04:05:28 +00:00
{
return -1;
}
}
return -1;
}
static int open_rio_console (qse_awk_rtx_t* rtx, qse_awk_rio_arg_t* riod)
{
2009-02-17 02:11:31 +00:00
rxtn_t* rxtn = (rxtn_t*) QSE_XTN (rtx);
qse_sio_t* sio;
2009-02-16 08:31:34 +00:00
if (riod->mode == QSE_AWK_RIO_CONSOLE_READ)
{
if (rxtn->c.in.files == QSE_NULL)
{
/* if no input files is specified,
* open the standard input */
QSE_ASSERT (rxtn->c.in.index == 0);
if (rxtn->c.in.count == 0)
{
sio = open_sio_std_rtx (
rtx, QSE_SIO_STDIN,
QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR
);
if (sio == QSE_NULL) return -1;
if (rxtn->c.cmgr) qse_sio_setcmgr (sio, rxtn->c.cmgr);
riod->handle = sio;
rxtn->c.in.count++;
return 1;
}
return 0;
}
else
{
/* a temporary variable sio is used here not to change
2009-02-16 08:31:34 +00:00
* any fields of riod when the open operation fails */
const qse_char_t* file;
qse_awk_val_t* argv;
qse_htb_t* map;
qse_htb_pair_t* pair;
qse_char_t ibuf[128];
qse_size_t ibuflen;
qse_awk_val_t* v;
qse_awk_rtx_valtostr_out_t out;
nextfile:
file = rxtn->c.in.files[rxtn->c.in.index];
if (file == QSE_NULL)
{
/* no more input file */
if (rxtn->c.in.count == 0)
{
/* all ARGVs are empty strings.
* so no console files were opened.
* open the standard input here.
*
* 'BEGIN { ARGV[1]=""; ARGV[2]=""; }
* { print $0; }' file1 file2
*/
sio = open_sio_std_rtx (
rtx, QSE_SIO_STDIN,
QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR
);
if (sio == QSE_NULL) return -1;
if (rxtn->c.cmgr) qse_sio_setcmgr (sio, rxtn->c.cmgr);
riod->handle = sio;
rxtn->c.in.count++;
return 1;
}
return 0;
}
/* handle special case when ARGV[x] has been altered.
* so from here down, the file name gotten from
* rxtn->c.in.files is not important and is overridden
* from ARGV.
* 'BEGIN { ARGV[1]="file3"; }
* { print $0; }' file1 file2
*/
argv = qse_awk_rtx_getgbl (rtx, QSE_AWK_GBL_ARGV);
QSE_ASSERT (argv != QSE_NULL);
QSE_ASSERT (argv->type == QSE_AWK_VAL_MAP);
map = ((qse_awk_val_map_t*)argv)->map;
QSE_ASSERT (map != QSE_NULL);
ibuflen = qse_awk_longtostr (
rtx->awk, rxtn->c.in.index + 1, 10, QSE_NULL,
ibuf, QSE_COUNTOF(ibuf));
pair = qse_htb_search (map, ibuf, ibuflen);
QSE_ASSERT (pair != QSE_NULL);
v = QSE_HTB_VPTR(pair);
QSE_ASSERT (v != QSE_NULL);
out.type = QSE_AWK_RTX_VALTOSTR_CPLDUP;
if (qse_awk_rtx_valtostr (rtx, v, &out) <= -1) return -1;
if (out.u.cpldup.len == 0)
{
/* the name is empty */
qse_awk_rtx_freemem (rtx, out.u.cpldup.ptr);
rxtn->c.in.index++;
goto nextfile;
}
if (qse_strlen(out.u.cpldup.ptr) < out.u.cpldup.len)
{
/* the name contains one or more '\0' */
qse_cstr_t errarg;
errarg.ptr = out.u.cpldup.ptr;
/* use this length not to contains '\0'
* in an error message */
errarg.len = qse_strlen(out.u.cpldup.ptr);
qse_awk_rtx_seterrnum (
rtx, QSE_AWK_EIONMNL, &errarg);
qse_awk_rtx_freemem (rtx, out.u.cpldup.ptr);
return -1;
}
file = out.u.cpldup.ptr;
sio = (file[0] == QSE_T('-') && file[1] == QSE_T('\0'))?
open_sio_std_rtx (rtx, QSE_SIO_STDIN,
QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR):
open_sio_rtx (rtx, file,
QSE_SIO_READ | QSE_SIO_IGNOREMBWCERR);
if (sio == QSE_NULL)
{
qse_awk_rtx_freemem (rtx, out.u.cpldup.ptr);
return -1;
}
if (rxtn->c.cmgr) qse_sio_setcmgr (sio, rxtn->c.cmgr);
if (qse_awk_rtx_setfilename (
2011-04-18 09:28:22 +00:00
rtx, file, qse_strlen(file)) <= -1)
{
qse_sio_close (sio);
qse_awk_rtx_freemem (rtx, out.u.cpldup.ptr);
return -1;
}
qse_awk_rtx_freemem (rtx, out.u.cpldup.ptr);
riod->handle = sio;
/* increment the counter of files successfully opened */
rxtn->c.in.count++;
rxtn->c.in.index++;
return 1;
}
}
2009-02-16 08:31:34 +00:00
else if (riod->mode == QSE_AWK_RIO_CONSOLE_WRITE)
{
if (rxtn->c.out.files == QSE_NULL)
{
QSE_ASSERT (rxtn->c.out.index == 0);
if (rxtn->c.out.count == 0)
{
sio = open_sio_std_rtx (
rtx, QSE_SIO_STDOUT,
QSE_SIO_WRITE | QSE_SIO_IGNOREMBWCERR
);
if (sio == QSE_NULL) return -1;
if (rxtn->c.cmgr) qse_sio_setcmgr (sio, rxtn->c.cmgr);
riod->handle = sio;
rxtn->c.out.count++;
return 1;
}
return 0;
}
else
{
/* a temporary variable sio is used here not to change
* any fields of riod when the open operation fails */
qse_sio_t* sio;
const qse_char_t* file;
file = rxtn->c.out.files[rxtn->c.out.index];
if (file == QSE_NULL)
{
/* no more input file */
return 0;
}
sio = (file[0] == QSE_T('-') && file[1] == QSE_T('\0'))?
open_sio_std_rtx (
rtx, QSE_SIO_STDOUT,
QSE_SIO_WRITE | QSE_SIO_IGNOREMBWCERR):
open_sio_rtx (
rtx, file,
QSE_SIO_WRITE | QSE_SIO_CREATE |
QSE_SIO_TRUNCATE | QSE_SIO_IGNOREMBWCERR);
if (sio == QSE_NULL) return -1;
if (rxtn->c.cmgr) qse_sio_setcmgr (sio, rxtn->c.cmgr);
2009-06-24 07:29:18 +00:00
if (qse_awk_rtx_setofilename (
2011-04-18 09:28:22 +00:00
rtx, file, qse_strlen(file)) <= -1)
{
qse_sio_close (sio);
return -1;
}
riod->handle = sio;
rxtn->c.out.index++;
rxtn->c.out.count++;
return 1;
}
}
return -1;
}
2008-12-12 04:05:28 +00:00
2009-02-16 08:31:34 +00:00
static qse_ssize_t awk_rio_console (
qse_awk_rtx_t* rtx, qse_awk_rio_cmd_t cmd, qse_awk_rio_arg_t* riod,
2009-02-16 08:31:34 +00:00
qse_char_t* data, qse_size_t size)
2008-12-12 04:05:28 +00:00
{
2011-05-26 08:42:26 +00:00
switch (cmd)
2008-12-12 04:05:28 +00:00
{
2011-05-26 08:42:26 +00:00
case QSE_AWK_RIO_OPEN:
return open_rio_console (rtx, riod);
case QSE_AWK_RIO_CLOSE:
if (riod->handle) qse_sio_close ((qse_sio_t*)riod->handle);
2011-05-26 08:42:26 +00:00
return 0;
case QSE_AWK_RIO_READ:
{
2011-05-26 08:42:26 +00:00
qse_ssize_t nn;
while ((nn = qse_sio_getstrn((qse_sio_t*)riod->handle,data,size)) == 0)
2011-05-26 08:42:26 +00:00
{
int n;
qse_sio_t* sio = (qse_sio_t*)riod->handle;
n = open_rio_console (rtx, riod);
if (n <= -1) return -1;
if (n == 0)
{
/* no more input console */
return 0;
}
if (sio) qse_sio_close (sio);
2011-05-26 08:42:26 +00:00
}
return nn;
}
2011-05-26 08:42:26 +00:00
case QSE_AWK_RIO_WRITE:
return qse_sio_putstrn (
2011-05-26 08:42:26 +00:00
(qse_sio_t*)riod->handle,
data,
size
);
2008-12-12 04:05:28 +00:00
2011-05-26 08:42:26 +00:00
case QSE_AWK_RIO_FLUSH:
return qse_sio_flush ((qse_sio_t*)riod->handle);
case QSE_AWK_RIO_NEXT:
2008-12-12 04:05:28 +00:00
{
int n;
qse_sio_t* sio = (qse_sio_t*)riod->handle;
2011-05-26 08:42:26 +00:00
n = open_rio_console (rtx, riod);
2011-05-26 08:42:26 +00:00
if (n <= -1) return -1;
if (n == 0)
2008-12-12 04:05:28 +00:00
{
2011-05-26 08:42:26 +00:00
/* if there is no more file, keep the previous handle */
2008-12-12 04:05:28 +00:00
return 0;
}
if (sio) qse_sio_close (sio);
2011-05-26 08:42:26 +00:00
return n;
2008-12-12 04:05:28 +00:00
}
}
2011-05-26 08:42:26 +00:00
qse_awk_rtx_seterrnum (rtx, QSE_AWK_EINTERN, QSE_NULL);
2008-12-12 04:05:28 +00:00
return -1;
}
static void fini_rxtn (qse_awk_rtx_t* rtx, void* ctx)
{
rxtn_t* rxtn = (rxtn_t*) QSE_XTN (rtx);
if (rxtn->cmgrtab_inited)
{
qse_htb_fini (&rxtn->cmgrtab);
rxtn->cmgrtab_inited = 0;
}
}
qse_awk_rtx_t* qse_awk_rtx_openstd (
2011-04-18 09:28:22 +00:00
qse_awk_t* awk,
qse_size_t xtnsize,
const qse_char_t* id,
2009-08-29 05:58:05 +00:00
const qse_char_t*const icf[],
const qse_char_t*const ocf[],
qse_cmgr_t* cmgr)
2009-02-13 04:55:25 +00:00
{
static qse_awk_rcb_t rcb =
{
fini_rxtn,
QSE_NULL,
QSE_NULL
};
qse_awk_rtx_t* rtx;
2009-02-13 04:55:25 +00:00
qse_awk_rio_t rio;
rxtn_t* rxtn;
qse_ntime_t now;
const qse_char_t*const* p;
qse_size_t argc = 0;
qse_cstr_t argv[16];
qse_cstr_t* argvp = QSE_NULL, * p2;
2009-08-29 05:58:05 +00:00
rio.pipe = awk_rio_pipe;
rio.file = awk_rio_file;
2009-02-16 08:31:34 +00:00
rio.console = awk_rio_console;
2009-02-13 04:55:25 +00:00
if (icf)
{
for (p = icf; *p != QSE_NULL; p++);
argc = p - icf;
}
argc++; /* for id */
if (argc < QSE_COUNTOF(argv)) argvp = argv;
else
{
argvp = QSE_AWK_ALLOC (
awk, QSE_SIZEOF(*argvp) * (argc + 1));
if (argvp == QSE_NULL)
{
qse_awk_seterrnum (awk, QSE_AWK_ENOMEM, QSE_NULL);
return QSE_NULL;
}
}
p2 = argvp;
p2->ptr = id;
p2->len = qse_strlen(id);
p2++;
if (icf)
{
for (p = icf; *p; p++, p2++)
{
p2->ptr = *p;
p2->len = qse_strlen(*p);
}
}
p2->ptr = QSE_NULL;
p2->len = 0;
2009-02-13 04:55:25 +00:00
rtx = qse_awk_rtx_open (
awk,
QSE_SIZEOF(rxtn_t) + xtnsize,
2009-02-13 04:55:25 +00:00
&rio,
argvp
2009-02-13 04:55:25 +00:00
);
if (argvp && argvp != argv) QSE_AWK_FREE (awk, argvp);
2009-02-13 04:55:25 +00:00
if (rtx == QSE_NULL) return QSE_NULL;
2009-02-17 02:11:31 +00:00
rxtn = (rxtn_t*) QSE_XTN (rtx);
2009-02-16 08:31:34 +00:00
QSE_MEMSET (rxtn, 0, QSE_SIZEOF(rxtn_t));
if (rtx->awk->option & QSE_AWK_RIO)
{
if (qse_htb_init (
&rxtn->cmgrtab, awk->mmgr, 256, 70,
QSE_SIZEOF(qse_char_t), 0) <= -1)
{
qse_awk_rtx_close (rtx);
qse_awk_seterrnum (awk, QSE_AWK_ENOMEM, QSE_NULL);
return QSE_NULL;
}
qse_htb_setmancbs (&rxtn->cmgrtab,
qse_gethtbmancbs(QSE_HTB_MANCBS_INLINE_COPIERS));
rxtn->cmgrtab_inited = 1;
}
qse_awk_rtx_pushrcb (rtx, &rcb);
2011-05-26 08:42:26 +00:00
if (qse_gettime (&now) <= -1) rxtn->seed = 0;
else rxtn->seed = (unsigned int) now;
2009-02-13 04:55:25 +00:00
srand (rxtn->seed);
2009-02-16 08:31:34 +00:00
rxtn->c.in.files = icf;
rxtn->c.in.index = 0;
rxtn->c.in.count = 0;
rxtn->c.out.files = ocf;
rxtn->c.out.index = 0;
rxtn->c.out.count = 0;
rxtn->c.cmgr = cmgr;
/* FILENAME can be set when the input console is opened.
* so we skip setting it here even if an explicit console file
* is specified. So the value of FILENAME is empty in the
* BEGIN block unless getline is ever met.
*
* However, OFILENAME is different. The output
* console is treated as if it's already open upon start-up.
* otherwise, 'BEGIN { print OFILENAME; }' prints an empty
* string regardless of output console files specified.
* That's because OFILENAME is evaluated before the output
* console file is opened.
*/
2011-04-18 09:28:22 +00:00
if (ocf && ocf[0])
{
if (qse_awk_rtx_setofilename (rtx, ocf[0], qse_strlen(ocf[0])) <= -1)
{
awk->errinf = rtx->errinf; /* transfer error info */
qse_awk_rtx_close (rtx);
return QSE_NULL;
}
}
2009-02-13 04:55:25 +00:00
return rtx;
}
2008-12-16 03:56:48 +00:00
void* qse_awk_rtx_getxtnstd (qse_awk_rtx_t* rtx)
{
return (void*)((rxtn_t*)QSE_XTN(rtx) + 1);
}
static int fnc_rand (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
2008-12-21 21:35:07 +00:00
qse_awk_val_t* r;
2008-12-16 03:56:48 +00:00
/*
rxtn_t* rxtn;
rxtn = (rxtn_t*) QSE_XTN (rtx);
r = qse_awk_rtx_makefltval (
rtx, (qse_flt_t)(rand_r(rxtn->seed) % RAND_MAX) / RAND_MAX );
*/
r = qse_awk_rtx_makefltval (
rtx, (qse_flt_t)(rand() % RAND_MAX) / RAND_MAX);
if (r == QSE_NULL) return -1;
2008-12-16 03:56:48 +00:00
qse_awk_rtx_setretval (rtx, r);
2008-12-16 03:56:48 +00:00
return 0;
}
static int fnc_srand (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
2008-12-21 21:35:07 +00:00
qse_size_t nargs;
qse_awk_val_t* a0;
qse_long_t lv;
qse_awk_val_t* r;
int n;
unsigned int prev;
rxtn_t* rxtn;
rxtn = (rxtn_t*) QSE_XTN (rtx);
nargs = qse_awk_rtx_getnargs (rtx);
2008-12-21 21:35:07 +00:00
QSE_ASSERT (nargs == 0 || nargs == 1);
prev = rxtn->seed;
if (nargs == 1)
{
a0 = qse_awk_rtx_getarg (rtx, 0);
n = qse_awk_rtx_valtolong (rtx, a0, &lv);
2009-07-17 06:43:47 +00:00
if (n <= -1) return -1;
rxtn->seed = lv;
}
else
{
2008-12-21 21:35:07 +00:00
qse_ntime_t now;
2009-07-17 06:43:47 +00:00
if (qse_gettime(&now) <= -1) rxtn->seed >>= 1;
else rxtn->seed = (unsigned int)now;
}
srand (rxtn->seed);
r = qse_awk_rtx_makeintval (rtx, prev);
if (r == QSE_NULL) return -1;
qse_awk_rtx_setretval (rtx, r);
return 0;
}
static int fnc_system (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
qse_size_t nargs;
qse_awk_val_t* v;
qse_char_t* str;
qse_size_t len;
int n = 0;
nargs = qse_awk_rtx_getnargs (rtx);
QSE_ASSERT (nargs == 1);
v = qse_awk_rtx_getarg (rtx, 0);
if (v->type == QSE_AWK_VAL_STR)
{
2011-05-19 08:36:40 +00:00
str = ((qse_awk_val_str_t*)v)->val.ptr;
len = ((qse_awk_val_str_t*)v)->val.len;
}
2008-12-18 02:39:15 +00:00
else
{
str = qse_awk_rtx_valtocpldup (rtx, v, &len);
if (str == QSE_NULL) return -1;
}
/* the target name contains a null character.
* make system return -1 */
if (qse_strxchr (str, len, QSE_T('\0')))
{
n = -1;
goto skip_system;
}
#if defined(_WIN32)
n = _tsystem (str);
#elif defined(QSE_CHAR_IS_MCHAR)
n = system (str);
#else
{
qse_mchar_t* mbs;
mbs = qse_wcstombsdup (str, rtx->awk->mmgr);
if (mbs == QSE_NULL)
{
n = -1;
goto skip_system;
}
n = system (mbs);
QSE_AWK_FREE (rtx->awk, mbs);
}
#endif
skip_system:
if (v->type != QSE_AWK_VAL_STR) QSE_AWK_FREE (rtx->awk, str);
v = qse_awk_rtx_makeintval (rtx, (qse_long_t)n);
if (v == QSE_NULL) return -1;
qse_awk_rtx_setretval (rtx, v);
return 0;
}
2008-12-31 00:08:03 +00:00
static int fnc_time (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
qse_awk_val_t* r;
qse_ntime_t now;
if (qse_gettime (&now) <= -1) now = 0;
r = qse_awk_rtx_makeintval (rtx, now);
if (r == QSE_NULL) return -1;
qse_awk_rtx_setretval (rtx, r);
return 0;
}
2012-04-27 14:33:14 +00:00
qse_cmgr_t* qse_awk_rtx_getcmgrstd (
qse_awk_rtx_t* rtx, const qse_char_t* ioname)
{
#if defined(QSE_CHAR_IS_WCHAR)
rxtn_t* rxtn;
qse_htb_pair_t* pair;
ioattr_t* ioattr;
rxtn = (rxtn_t*) QSE_XTN (rtx);
QSE_ASSERT (rxtn->cmgrtab_inited == 1);
pair = qse_htb_search (&rxtn->cmgrtab, ioname, qse_strlen(ioname));
if (pair)
{
ioattr = (ioattr_t*)QSE_HTB_VPTR(pair);
return ioattr->cmgr;
}
#endif
return QSE_NULL;
}
static int timeout_code (const qse_char_t* name)
{
if (qse_strcmp (name, QSE_T("rtimeout")) == 0) return 0;
if (qse_strcmp (name, QSE_T("wtimeout")) == 0) return 1;
if (qse_strcmp (name, QSE_T("ctimeout")) == 0) return 2;
return -1;
}
static QSE_INLINE void init_ioattr (ioattr_t* ioattr)
{
int i;
QSE_MEMSET (ioattr, 0, QSE_SIZEOF(*ioattr));
for (i = 0; i < QSE_COUNTOF(ioattr->timeout); i++)
{
/* a negative number for no timeout */
ioattr->timeout[i] = -999;
}
}
static qse_htb_pair_t* find_or_make_ioattr (
qse_awk_rtx_t* rtx, qse_htb_t* tab, const qse_char_t* ptr, qse_size_t len)
{
qse_htb_pair_t* pair;
pair = qse_htb_search (tab, ptr, len);
if (pair == QSE_NULL)
{
ioattr_t ioattr;
init_ioattr (&ioattr);
pair = qse_htb_insert (tab, ptr, len, &ioattr, QSE_SIZEOF(ioattr));
if (pair == QSE_NULL)
{
qse_awk_rtx_seterrnum (rtx, QSE_AWK_ENOMEM, QSE_NULL);
}
}
return pair;
}
static int fnc_setioattr (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
rxtn_t* rxtn;
qse_size_t nargs;
qse_awk_val_t* v[3];
qse_char_t* ptr[3];
qse_size_t len[3];
int i, ret = 0, fret = 0;
qse_cmgr_t* cmgr;
int tmout;
rxtn = (rxtn_t*) QSE_XTN (rtx);
QSE_ASSERT (rxtn->cmgrtab_inited == 1);
nargs = qse_awk_rtx_getnargs (rtx);
QSE_ASSERT (nargs == 3);
for (i = 0; i < 3; i++)
{
v[i] = qse_awk_rtx_getarg (rtx, i);
if (v[i]->type == QSE_AWK_VAL_STR)
{
ptr[i] = ((qse_awk_val_str_t*)v[i])->val.ptr;
len[i] = ((qse_awk_val_str_t*)v[i])->val.len;
}
else
{
ptr[i] = qse_awk_rtx_valtocpldup (rtx, v[i], &len[i]);
if (ptr[i] == QSE_NULL)
{
ret = -1;
goto done;
}
}
if (qse_strxchr (ptr[i], len[i], QSE_T('\0')))
{
fret = -1;
goto done;
}
}
if ((tmout = timeout_code (ptr[1])) >= 0)
{
qse_htb_pair_t* pair;
ioattr_t* ioattr;
qse_long_t l;
qse_flt_t r;
int x;
/* no error is returned by qse_awk_rtx_strnum() if the second
* parameter is 0. so i don't check for an error */
x = qse_awk_rtx_strtonum (rtx, 0, ptr[2], len[2], &l, &r);
if (x >= 1) l = (qse_long_t)r;
pair = find_or_make_ioattr (rtx, &rxtn->cmgrtab, ptr[0], len[0]);
if (pair == QSE_NULL)
{
ret = -1;
goto done;
}
ioattr = QSE_HTB_VPTR(pair);
ioattr->timeout[tmout] = l;
}
#if defined(QSE_CHAR_IS_WCHAR)
else if (qse_strcmp (ptr[1], QSE_T("codepage")) == 0)
{
qse_htb_pair_t* pair;
ioattr_t* ioattr;
if (ptr[2][0] == QSE_T('\0'))
{
cmgr = QSE_NULL;
}
else
{
cmgr = qse_findcmgr (ptr[2]);
if (cmgr == QSE_NULL)
{
fret = -1;
goto done;
}
}
pair = find_or_make_ioattr (rtx, &rxtn->cmgrtab, ptr[0], len[0]);
if (pair == QSE_NULL)
{
ret = -1;
goto done;
}
ioattr = QSE_HTB_VPTR(pair);
ioattr->cmgr = cmgr;
qse_strxcpy (ioattr->cmgr_name, QSE_COUNTOF(ioattr->cmgr_name), ptr[2]);
}
#endif
else
{
/* unknown attribute name */
fret = -1;
goto done;
}
done:
while (i > 0)
{
i--;
if (v[i]->type != QSE_AWK_VAL_STR)
QSE_AWK_FREE (rtx->awk, ptr[i]);
}
if (ret >= 0)
{
v[0] = qse_awk_rtx_makeintval (rtx, (qse_long_t)fret);
if (v[0] == QSE_NULL) return -1;
qse_awk_rtx_setretval (rtx, v[0]);
}
return ret;
}
static int fnc_getioattr (qse_awk_rtx_t* rtx, const qse_cstr_t* fnm)
{
rxtn_t* rxtn;
qse_size_t nargs;
qse_awk_val_t* v[2];
qse_char_t* ptr[2];
qse_size_t len[2];
int i, ret = 0;
int tmout;
qse_awk_val_t* rv = QSE_NULL;
qse_htb_pair_t* pair;
ioattr_t* ioattr;
ioattr_t ioattr_buf;
rxtn = (rxtn_t*) QSE_XTN (rtx);
QSE_ASSERT (rxtn->cmgrtab_inited == 1);
nargs = qse_awk_rtx_getnargs (rtx);
QSE_ASSERT (nargs == 2);
for (i = 0; i < 2; i++)
{
v[i] = qse_awk_rtx_getarg (rtx, i);
if (v[i]->type == QSE_AWK_VAL_STR)
{
ptr[i] = ((qse_awk_val_str_t*)v[i])->val.ptr;
len[i] = ((qse_awk_val_str_t*)v[i])->val.len;
}
else
{
ptr[i] = qse_awk_rtx_valtocpldup (rtx, v[i], &len[i]);
if (ptr[i] == QSE_NULL)
{
ret = -1;
goto done;
}
}
if (qse_strxchr (ptr[i], len[i], QSE_T('\0'))) goto done;
}
pair = qse_htb_search (&rxtn->cmgrtab, ptr[0], len[0]);
if (pair == QSE_NULL)
{
init_ioattr (&ioattr_buf);
ioattr = &ioattr_buf;
}
else
{
ioattr = QSE_HTB_VPTR(pair);
}
if ((tmout = timeout_code (ptr[1])) >= 0)
{
rv = qse_awk_rtx_makeintval (rtx, ioattr->timeout[tmout]);
if (rv == QSE_NULL)
{
ret = -1;
goto done;
}
}
#if defined(QSE_CHAR_IS_WCHAR)
else if (qse_strcmp (ptr[1], QSE_T("codepage")) == 0)
{
rv = qse_awk_rtx_makestrval0 (rtx, ioattr->cmgr_name);
if (rv == QSE_NULL)
{
ret = -1;
goto done;
}
}
#endif
else
{
/* unknown attribute name */
goto done;
}
done:
while (i > 0)
{
i--;
if (v[i]->type != QSE_AWK_VAL_STR)
QSE_AWK_FREE (rtx->awk, ptr[i]);
}
if (ret >= 0)
{
/* the return value of -1 for an error may be confusing since
* a literal value of -1 can be returned for some attribute
* names */
if (rv == QSE_NULL) rv = qse_awk_rtx_makeintval (rtx, -1);;
qse_awk_rtx_setretval (rtx, rv);
}
return ret;
}
#define ADDFNC(awk,name,min,max,fnc,valid) \
if (qse_awk_addfnc (\
2008-12-21 21:35:07 +00:00
(awk), (name), qse_strlen(name), \
valid, (min), (max), QSE_NULL, (fnc)) == QSE_NULL) return -1;
2008-12-21 21:35:07 +00:00
static int add_functions (qse_awk_t* awk)
{
ADDFNC (awk, QSE_T("rand"), 0, 0, fnc_rand, 0);
ADDFNC (awk, QSE_T("srand"), 0, 1, fnc_srand, 0);
ADDFNC (awk, QSE_T("system"), 1, 1, fnc_system, 0);
ADDFNC (awk, QSE_T("time"), 0, 0, fnc_time, 0);
ADDFNC (awk, QSE_T("setioattr"), 3, 3, fnc_setioattr, QSE_AWK_RIO);
ADDFNC (awk, QSE_T("getioattr"), 2, 2, fnc_getioattr, QSE_AWK_RIO);
return 0;
}