qse/samples/awk/awk10.c

156 lines
3.9 KiB
C

#include <qse/awk/stdawk.h>
#include <qse/si/sio.h>
#include <qse/cmn/main.h>
#include <qse/cmn/mem.h>
#include <qse/cmn/path.h>
#include "awk00.h"
static const qse_char_t* src = QSE_T(
"BEGIN { if (basename(\"/etc/passwd\", base) <= -1) print \"ERROR\"; else print base; }"
);
static int fnc_basename (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
{
qse_awk_val_t* a0, * rv = QSE_NULL;
qse_char_t* ptr;
qse_size_t len;
/* note that this implementation doesn't care if the parameter
* contains a null character like "/etc/p\0asswd" */
/* get the value of the first parameter */
a0 = qse_awk_rtx_getarg (rtx, 0);
if (qse_awk_rtx_getvaltype (rtx, a0) == QSE_AWK_VAL_STR)
{
/* if it is a string value, don't duplicate the value */
ptr = ((qse_awk_val_str_t*)a0)->val.ptr;
len = ((qse_awk_val_str_t*)a0)->val.len;
/* make a string value with the base name */
rv = qse_awk_rtx_makestrvalwithstr (rtx, qse_basename (ptr));
}
else
{
/* if it is a string value, convert the value to a string
* with duplication */
ptr = qse_awk_rtx_valtostrdup (rtx, a0, &len);
if (ptr)
{
/* make a string value with the base name */
rv = qse_awk_rtx_makestrvalwithstr (rtx, qse_basename (ptr));
/* free the duplicated string */
qse_awk_rtx_freemem (rtx, ptr);
}
}
if (rv)
{
/* change the value of the second parameter passed by reference */
qse_awk_rtx_setrefval (rtx, qse_awk_rtx_getarg (rtx, 1), rv);
/* set the return value without error checks because
* qse_awk_rtx_makeintval() for 0 never fails */
qse_awk_rtx_setretval (rtx, qse_awk_rtx_makeintval (rtx, 0));
}
else
{
/* set the return value without error checks because
* qse_awk_rtx_makeintval() for -1 never fails */
qse_awk_rtx_setretval (rtx, qse_awk_rtx_makeintval (rtx, -1));
}
/* implementation success */
return 0;
}
static int awk_main (int argc, qse_char_t* argv[])
{
qse_awk_t* awk = QSE_NULL;
qse_awk_rtx_t* rtx = QSE_NULL;
qse_awk_parsestd_t psin[2];
qse_awk_val_t* rtv;
int ret = -1;
qse_awk_fnc_spec_t spec;
/* create an awk object */
awk = qse_awk_openstd (0, QSE_NULL);
if (awk == QSE_NULL)
{
qse_fprintf (QSE_STDERR, QSE_T("ERROR: cannot open awk\n"));
goto oops;
}
/* add a built-in function basename() */
qse_memset (&spec, 0, QSE_SIZEOF(spec));
spec.arg.min = 2; /* limit the number of arguments to 1 */
spec.arg.max = 2;
spec.arg.spec = QSE_T("vr"); /* pass the second argument by reference */
spec.impl = fnc_basename; /* specify the actual implementation */
if (qse_awk_addfnc (awk, QSE_T("basename"), &spec) == QSE_NULL)
{
qse_fprintf (QSE_STDERR, QSE_T("ERROR: %s\n"), qse_awk_geterrmsg(awk));
goto oops;
}
/* prepare an awk script to parse */
psin[0].type = QSE_AWK_PARSESTD_STR;
psin[0].u.str.ptr = src;
psin[0].u.str.len = qse_strlen(src);
psin[1].type = QSE_AWK_PARSESTD_NULL;
/* parse the script */
if (qse_awk_parsestd (awk, psin, QSE_NULL) <= -1)
{
qse_fprintf (QSE_STDERR, QSE_T("ERROR: %s\n"), qse_awk_geterrmsg(awk));
goto oops;
}
/* create a runtime context */
rtx = qse_awk_rtx_openstd (
awk,
0,
QSE_T("awk10"),
QSE_NULL, /* stdin */
QSE_NULL, /* stdout */
QSE_NULL /* default cmgr */
);
if (rtx == QSE_NULL)
{
qse_fprintf (QSE_STDERR, QSE_T("ERROR: %s\n"), qse_awk_geterrmsg(awk));
goto oops;
}
/* execute the script over the standard data streams */
rtv = qse_awk_rtx_loop (rtx);
if (rtv == QSE_NULL)
{
qse_fprintf (QSE_STDERR, QSE_T("ERROR: %s\n"), qse_awk_rtx_geterrmsg(rtx));
goto oops;
}
/* clear the return value */
qse_awk_rtx_refdownval (rtx, rtv);
ret = 0;
oops:
/* destroy a runtime context */
if (rtx) qse_awk_rtx_close (rtx);
/* destroy the processor */
if (awk) qse_awk_close (awk);
return ret;
}
int qse_main (int argc, qse_achar_t* argv[])
{
int x;
qse_open_stdsios ();
init_awk_sample_locale ();
x = qse_run_main (argc, argv, awk_main);
qse_close_stdsios ();
return x;
}