3989 lines
92 KiB
C
3989 lines
92 KiB
C
/*
|
|
* $Id: run.c,v 1.135 2006-07-17 04:17:40 bacon Exp $
|
|
*/
|
|
|
|
#include <xp/awk/awk_i.h>
|
|
|
|
/* TODO: remove this dependency...*/
|
|
#include <math.h>
|
|
|
|
#ifndef XP_AWK_STAND_ALONE
|
|
#include <xp/bas/assert.h>
|
|
#include <xp/bas/string.h>
|
|
#include <xp/bas/memory.h>
|
|
#endif
|
|
|
|
#define DEF_BUF_CAPA 256
|
|
|
|
#define STACK_INCREMENT 512
|
|
|
|
#define STACK_AT(run,n) ((run)->stack[(run)->stack_base+(n)])
|
|
#define STACK_NARGS(run) (STACK_AT(run,3))
|
|
#define STACK_ARG(run,n) STACK_AT(run,3+1+(n))
|
|
#define STACK_LOCAL(run,n) STACK_AT(run,3+(xp_size_t)STACK_NARGS(run)+1+(n))
|
|
#define STACK_RETVAL(run) STACK_AT(run,2)
|
|
#define STACK_GLOBAL(run,n) ((run)->stack[(n)])
|
|
/*#define STACK_RETVAL_GLOBAL(run) ((run)->stack[(run)->nglobals+2])*/
|
|
#define STACK_RETVAL_GLOBAL(run) ((run)->stack[(run)->awk->tree.nglobals+2])
|
|
|
|
enum
|
|
{
|
|
EXIT_NONE,
|
|
EXIT_BREAK,
|
|
EXIT_CONTINUE,
|
|
EXIT_FUNCTION,
|
|
EXIT_NEXT,
|
|
EXIT_GLOBAL,
|
|
EXIT_ABORT,
|
|
};
|
|
|
|
#define PANIC(run,code) \
|
|
do { (run)->errnum = (code); return XP_NULL; } while (0)
|
|
#define PANIC_I(run,code) \
|
|
do { (run)->errnum = (code); return -1; } while (0)
|
|
|
|
static int __open_run (
|
|
xp_awk_run_t* run, xp_awk_t* awk, xp_awk_io_t txtio, void* txtio_arg);
|
|
static void __close_run (xp_awk_run_t* run);
|
|
|
|
static int __run_main (xp_awk_run_t* run);
|
|
static int __run_pattern_blocks (xp_awk_run_t* run);
|
|
static int __run_pattern_block_chain (xp_awk_run_t* run, xp_awk_chain_t* chain);
|
|
static int __run_block (xp_awk_run_t* run, xp_awk_nde_blk_t* nde);
|
|
static int __run_statement (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static int __run_if (xp_awk_run_t* run, xp_awk_nde_if_t* nde);
|
|
static int __run_while (xp_awk_run_t* run, xp_awk_nde_while_t* nde);
|
|
static int __run_for (xp_awk_run_t* run, xp_awk_nde_for_t* nde);
|
|
static int __run_foreach (xp_awk_run_t* run, xp_awk_nde_foreach_t* nde);
|
|
static int __run_break (xp_awk_run_t* run, xp_awk_nde_break_t* nde);
|
|
static int __run_continue (xp_awk_run_t* run, xp_awk_nde_continue_t* nde);
|
|
static int __run_return (xp_awk_run_t* run, xp_awk_nde_return_t* nde);
|
|
static int __run_exit (xp_awk_run_t* run, xp_awk_nde_exit_t* nde);
|
|
static int __run_next (xp_awk_run_t* run, xp_awk_nde_next_t* nde);
|
|
static int __run_nextfile (xp_awk_run_t* run, xp_awk_nde_nextfile_t* nde);
|
|
static int __run_delete (xp_awk_run_t* run, xp_awk_nde_delete_t* nde);
|
|
static int __run_print (xp_awk_run_t* run, xp_awk_nde_print_t* nde);
|
|
|
|
static xp_awk_val_t* __eval_expression (
|
|
xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
static xp_awk_val_t* __eval_group (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
static xp_awk_val_t* __eval_assignment (
|
|
xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __do_assignment (
|
|
xp_awk_run_t* run, xp_awk_nde_t* var, xp_awk_val_t* val);
|
|
static xp_awk_val_t* __do_assignment_scalar (
|
|
xp_awk_run_t* run, xp_awk_nde_var_t* var, xp_awk_val_t* val);
|
|
static xp_awk_val_t* __do_assignment_map (
|
|
xp_awk_run_t* run, xp_awk_nde_var_t* var, xp_awk_val_t* val);
|
|
static xp_awk_val_t* __do_assignment_pos (
|
|
xp_awk_run_t* run, xp_awk_nde_pos_t* pos, xp_awk_val_t* val);
|
|
|
|
static xp_awk_val_t* __eval_binary (
|
|
xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_binop_lor (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right);
|
|
static xp_awk_val_t* __eval_binop_land (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right);
|
|
static xp_awk_val_t* __eval_binop_in (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right);
|
|
static xp_awk_val_t* __eval_binop_bor (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_bxor (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_band (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_eq (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_ne (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_gt (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_ge (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_lt (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_le (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_lshift (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_rshift (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_plus (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_minus (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_mul (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_div (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_mod (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_exp (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_concat (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_ma (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
static xp_awk_val_t* __eval_binop_nm (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
|
|
static xp_awk_val_t* __eval_unary (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_incpre (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_incpst (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_cnd (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
static xp_awk_val_t* __eval_bfn (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_afn (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_call (
|
|
xp_awk_run_t* run, xp_awk_nde_t* nde, xp_awk_afn_t* afn);
|
|
|
|
static xp_awk_val_t* __eval_int (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_real (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_str (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_rex (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_named (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_global (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_local (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_arg (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_namedidx (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_globalidx (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_localidx (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_argidx (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_pos (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
static xp_awk_val_t* __eval_getline (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
static int __raw_push (xp_awk_run_t* run, void* val);
|
|
static void __raw_pop (xp_awk_run_t* run);
|
|
static void __raw_pop_times (xp_awk_run_t* run, xp_size_t times);
|
|
|
|
static int __read_record (xp_awk_run_t* run);
|
|
static int __split_record (xp_awk_run_t* run);
|
|
static void __clear_record (xp_awk_run_t* run, xp_bool_t noline);
|
|
static int __recomp_record_fields (xp_awk_run_t* run,
|
|
xp_size_t lv, xp_char_t* str, xp_size_t len, int* errnum);
|
|
|
|
static xp_char_t* __idxnde_to_str (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
typedef xp_awk_val_t* (*binop_func_t) (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right);
|
|
typedef xp_awk_val_t* (*eval_expr_t) (xp_awk_run_t* run, xp_awk_nde_t* nde);
|
|
|
|
/* TODO: remove this function */
|
|
static int __printval (xp_awk_pair_t* pair, void* arg)
|
|
{
|
|
xp_printf (XP_T("%s = "), (const xp_char_t*)pair->key);
|
|
xp_awk_printval ((xp_awk_val_t*)pair->val);
|
|
xp_printf (XP_T("\n"));
|
|
return 0;
|
|
}
|
|
|
|
xp_size_t xp_awk_getnargs (void* run)
|
|
{
|
|
return (xp_size_t) STACK_NARGS ((xp_awk_run_t*)run);
|
|
}
|
|
|
|
xp_awk_val_t* xp_awk_getarg (void* run, xp_size_t idx)
|
|
{
|
|
return STACK_ARG ((xp_awk_run_t*)run, idx);
|
|
}
|
|
|
|
void xp_awk_setretval (void* run, xp_awk_val_t* val)
|
|
{
|
|
xp_awk_run_t* r = (xp_awk_run_t*)run;
|
|
xp_awk_refdownval (r, STACK_RETVAL(r));
|
|
STACK_RETVAL(r) = val;
|
|
/* should use the same trick as __run_return_statement */
|
|
xp_awk_refupval (val);
|
|
}
|
|
|
|
void xp_awk_setglobal (void* run, xp_size_t idx, xp_awk_val_t* val)
|
|
{
|
|
xp_awk_run_t* r = (xp_awk_run_t*)run;
|
|
xp_awk_refdownval (run, STACK_GLOBAL(r,idx));
|
|
STACK_GLOBAL(r,idx) = val;
|
|
xp_awk_refupval (val);
|
|
}
|
|
|
|
void xp_awk_seterrnum (void* run, int errnum)
|
|
{
|
|
xp_awk_run_t* r = (xp_awk_run_t*)run;
|
|
r->errnum = errnum;
|
|
}
|
|
|
|
int xp_awk_run (xp_awk_t* awk, xp_awk_io_t txtio, void* txtio_arg)
|
|
{
|
|
xp_awk_run_t* run;
|
|
int n;
|
|
|
|
run = (xp_awk_run_t*) xp_malloc (xp_sizeof(xp_awk_run_t));
|
|
if (run == XP_NULL)
|
|
{
|
|
awk->errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
if (__open_run (run, awk, txtio, txtio_arg) == -1)
|
|
{
|
|
/* TODO: find a way to set the errnum into awk object in a thread-safe way */
|
|
awk->errnum = run->errnum;
|
|
xp_free (run);
|
|
return -1;
|
|
}
|
|
|
|
n = __run_main (run);
|
|
if (n == -1) awk->errnum = run->errnum;
|
|
|
|
__close_run (run);
|
|
xp_free (run);
|
|
|
|
return n;
|
|
}
|
|
|
|
static void __free_namedval (void* run, void* val)
|
|
{
|
|
xp_awk_refdownval ((xp_awk_run_t*)run, val);
|
|
}
|
|
|
|
static int __open_run (
|
|
xp_awk_run_t* run, xp_awk_t* awk, xp_awk_io_t txtio, void* txtio_arg)
|
|
{
|
|
xp_memset (run, 0, xp_sizeof(*run));
|
|
|
|
run->stack = XP_NULL;
|
|
run->stack_top = 0;
|
|
run->stack_base = 0;
|
|
run->stack_limit = 0;
|
|
|
|
run->exit_level = 0;
|
|
|
|
run->icache_count = 0;
|
|
run->rcache_count = 0;
|
|
|
|
run->txtio = txtio;
|
|
run->txtio_arg = txtio_arg;
|
|
|
|
run->opt = awk->opt.run;
|
|
run->errnum = XP_AWK_ENOERR;
|
|
/*run->tree = &awk->tree; */
|
|
/*run->nglobals = awk->tree.nglobals;*/
|
|
run->awk = awk;
|
|
|
|
run->inrec.buf_pos = 0;
|
|
run->inrec.buf_len = 0;
|
|
run->inrec.flds = XP_NULL;
|
|
run->inrec.nflds = 0;
|
|
run->inrec.maxflds = 0;
|
|
run->inrec.d0 = xp_awk_val_nil;
|
|
if (xp_str_open (&run->inrec.line, DEF_BUF_CAPA) == XP_NULL)
|
|
{
|
|
run->errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
if (xp_awk_map_open (&run->named,
|
|
run, DEF_BUF_CAPA, __free_namedval) == XP_NULL)
|
|
{
|
|
xp_str_close (&run->inrec.line);
|
|
run->errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
run->extio = XP_NULL;
|
|
return 0;
|
|
}
|
|
|
|
static void __close_run (xp_awk_run_t* run)
|
|
{
|
|
/* close all pending eio's */
|
|
/* TODO: what if this operation fails? */
|
|
xp_awk_clearextio (run);
|
|
|
|
/* destroy run stack */
|
|
if (run->stack != XP_NULL)
|
|
{
|
|
xp_assert (run->stack_top == 0);
|
|
|
|
xp_free (run->stack);
|
|
run->stack = XP_NULL;
|
|
run->stack_top = 0;
|
|
run->stack_base = 0;
|
|
run->stack_limit = 0;
|
|
}
|
|
|
|
/* destroy named variables */
|
|
xp_awk_map_close (&run->named);
|
|
|
|
/* destroy input record */
|
|
__clear_record (run, xp_false);
|
|
if (run->inrec.flds != XP_NULL)
|
|
{
|
|
xp_free (run->inrec.flds);
|
|
run->inrec.flds = XP_NULL;
|
|
run->inrec.maxflds = 0;
|
|
}
|
|
xp_str_close (&run->inrec.line);
|
|
|
|
/* destroy values in free list */
|
|
while (run->icache_count > 0)
|
|
{
|
|
xp_awk_val_int_t* tmp = run->icache[--run->icache_count];
|
|
xp_awk_freeval (run, (xp_awk_val_t*)tmp, xp_false);
|
|
}
|
|
|
|
while (run->rcache_count > 0)
|
|
{
|
|
xp_awk_val_real_t* tmp = run->rcache[--run->rcache_count];
|
|
xp_awk_freeval (run, (xp_awk_val_t*)tmp, xp_false);
|
|
}
|
|
}
|
|
|
|
static int __run_main (xp_awk_run_t* run)
|
|
{
|
|
xp_size_t nglobals, nargs, i;
|
|
xp_size_t saved_stack_top;
|
|
xp_awk_val_t* v;
|
|
int n = 0;
|
|
|
|
xp_assert (run->stack_base == 0 && run->stack_top == 0);
|
|
|
|
/* secure space for global variables */
|
|
saved_stack_top = run->stack_top;
|
|
|
|
/*nglobals = run->nglobals;*/
|
|
nglobals = run->awk->tree.nglobals;
|
|
|
|
while (nglobals > 0)
|
|
{
|
|
--nglobals;
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
/* restore the stack_top with the saved value
|
|
* instead of calling __raw_pop as many times as
|
|
* the successful __raw_push. it is ok because
|
|
* the values pushed so fare are all xp_awk_val_nil */
|
|
run->stack_top = saved_stack_top;
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
|
|
if (run->opt & XP_AWK_RUNMAIN)
|
|
{
|
|
/* TODO: should the main function be user-specifiable? */
|
|
static xp_char_t m_a_i_n[] =
|
|
{
|
|
XP_T('m'),
|
|
XP_T('a'),
|
|
XP_T('i'),
|
|
XP_T('n'),
|
|
XP_T('\0')
|
|
};
|
|
static xp_awk_nde_call_t nde =
|
|
{
|
|
XP_AWK_NDE_AFN, /* type */
|
|
XP_NULL, /* next */
|
|
m_a_i_n, /* name */
|
|
XP_NULL /* args */
|
|
};
|
|
|
|
run->exit_level = EXIT_NONE;
|
|
|
|
v = __eval_afn (run, (xp_awk_nde_t*)&nde);
|
|
if (v == XP_NULL) n = -1;
|
|
else
|
|
{
|
|
/* destroy the return value if necessary */
|
|
xp_awk_refupval (v);
|
|
xp_awk_refdownval (run, v);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
saved_stack_top = run->stack_top;
|
|
if (__raw_push(run,(void*)run->stack_base) == -1)
|
|
{
|
|
/* restore the stack top in a cheesy(?) way */
|
|
run->stack_top = saved_stack_top;
|
|
/* pops off global variables in a decent way */
|
|
/*__raw_pop_times (run, run->nglobals);*/
|
|
__raw_pop_times (run, run->awk->tree.nglobals);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
if (__raw_push(run,(void*)saved_stack_top) == -1)
|
|
{
|
|
run->stack_top = saved_stack_top;
|
|
/*__raw_pop_times (run, run->nglobals);*/
|
|
__raw_pop_times (run, run->awk->tree.nglobals);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
/* secure space for a return value */
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
run->stack_top = saved_stack_top;
|
|
/*__raw_pop_times (run, run->nglobals);*/
|
|
__raw_pop_times (run, run->awk->tree.nglobals);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
/* secure space for nargs */
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
run->stack_top = saved_stack_top;
|
|
/*__raw_pop_times (run, run->nglobals);*/
|
|
__raw_pop_times (run, run->awk->tree.nglobals);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
run->stack_base = saved_stack_top;
|
|
|
|
/* set nargs to zero */
|
|
nargs = 0;
|
|
STACK_NARGS(run) = (void*)nargs;
|
|
|
|
/* stack set up properly. ready to exeucte statement blocks */
|
|
if (n == 0 && run->awk->tree.begin != XP_NULL)
|
|
{
|
|
xp_awk_nde_blk_t* blk;
|
|
|
|
blk = (xp_awk_nde_blk_t*)run->awk->tree.begin;
|
|
xp_assert (blk->type == XP_AWK_NDE_BLK);
|
|
|
|
run->exit_level = EXIT_NONE;
|
|
if (__run_block (run, blk) == -1) n = -1;
|
|
}
|
|
|
|
if (n == 0 && run->txtio != XP_NULL)
|
|
{
|
|
if (__run_pattern_blocks (run) == -1) n = -1;
|
|
}
|
|
|
|
if (n == 0 && run->awk->tree.end != XP_NULL)
|
|
{
|
|
xp_awk_nde_blk_t* blk;
|
|
|
|
blk = (xp_awk_nde_blk_t*)run->awk->tree.end;
|
|
xp_assert (blk->type == XP_AWK_NDE_BLK);
|
|
|
|
run->exit_level = EXIT_NONE;
|
|
if (__run_block (run, blk) == -1) n = -1;
|
|
}
|
|
|
|
/* restore stack */
|
|
nargs = (xp_size_t)STACK_NARGS(run);
|
|
xp_assert (nargs == 0);
|
|
for (i = 0; i < nargs; i++)
|
|
{
|
|
xp_awk_refdownval (run, STACK_ARG(run,i));
|
|
}
|
|
|
|
v = STACK_RETVAL(run);
|
|
xp_printf (XP_T("Return Value - "));
|
|
xp_awk_printval (v);
|
|
xp_printf (XP_T("\n"));
|
|
/* the life of the global return value is over here
|
|
* unlike the return value of each function */
|
|
/*xp_awk_refdownval_nofree (awk, v);*/
|
|
xp_awk_refdownval (run, v);
|
|
|
|
run->stack_top =
|
|
(xp_size_t)run->stack[run->stack_base+1];
|
|
run->stack_base =
|
|
(xp_size_t)run->stack[run->stack_base+0];
|
|
}
|
|
|
|
/* pops off the global variables */
|
|
nglobals = run->awk->tree.nglobals; /*run->nglobals */
|
|
while (nglobals > 0)
|
|
{
|
|
--nglobals;
|
|
xp_awk_refdownval (run, STACK_GLOBAL(run,nglobals));
|
|
__raw_pop (run);
|
|
}
|
|
|
|
/* just reset the exit level */
|
|
run->exit_level = EXIT_NONE;
|
|
|
|
xp_printf (XP_T("-[VARIABLES]------------------------\n"));
|
|
xp_awk_map_walk (&run->named, __printval, XP_NULL);
|
|
xp_printf (XP_T("-[END VARIABLES]--------------------------\n"));
|
|
|
|
return n;
|
|
}
|
|
|
|
static int __run_pattern_blocks (xp_awk_run_t* run)
|
|
{
|
|
xp_ssize_t n;
|
|
|
|
xp_assert (run->txtio != XP_NULL);
|
|
|
|
n = run->txtio (XP_AWK_IO_OPEN, 0, run->txtio_arg, XP_NULL, 0);
|
|
if (n == -1) PANIC_I (run, XP_AWK_ETXTINOPEN);
|
|
|
|
run->inrec.buf_pos = 0;
|
|
run->inrec.buf_len = 0;
|
|
run->inrec.eof = xp_false;
|
|
|
|
while (run->exit_level != EXIT_GLOBAL &&
|
|
run->exit_level != EXIT_ABORT)
|
|
{
|
|
int x;
|
|
|
|
run->exit_level = EXIT_NONE;
|
|
|
|
x = __read_record (run);
|
|
if (x == -1)
|
|
{
|
|
/* don't care about the result of input close */
|
|
run->txtio (XP_AWK_IO_CLOSE, 0,
|
|
run->txtio_arg, XP_NULL, 0);
|
|
return -1;
|
|
}
|
|
|
|
if (x == 0) break; /* end of input */
|
|
|
|
if (__run_pattern_block_chain (run, run->awk->tree.chain) == -1)
|
|
{
|
|
/* don't care about the result of input close */
|
|
run->txtio (XP_AWK_IO_CLOSE, 0,
|
|
run->txtio_arg, XP_NULL, 0);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
n = run->txtio (XP_AWK_IO_CLOSE, 0, run->txtio_arg, XP_NULL, 0);
|
|
if (n == -1) PANIC_I (run, XP_AWK_ETXTINCLOSE);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int __run_pattern_block_chain (xp_awk_run_t* run, xp_awk_chain_t* chain)
|
|
{
|
|
xp_awk_nde_t* ptn;
|
|
|
|
while (run->exit_level != EXIT_GLOBAL &&
|
|
run->exit_level != EXIT_ABORT && chain != XP_NULL)
|
|
{
|
|
|
|
if (run->exit_level == EXIT_NEXT)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
break;
|
|
}
|
|
|
|
ptn = chain->pattern;
|
|
|
|
if (ptn == XP_NULL)
|
|
{
|
|
/* just execute the block */
|
|
xp_awk_nde_blk_t* blk;
|
|
blk = (xp_awk_nde_blk_t*)chain->action;
|
|
if (__run_block (run, blk) == -1) return -1;
|
|
}
|
|
else
|
|
{
|
|
xp_awk_val_t* v1;
|
|
|
|
v1 = __eval_expression (run, ptn);
|
|
if (v1 == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (v1);
|
|
|
|
if (ptn->next == XP_NULL)
|
|
{
|
|
if (xp_awk_valtobool(v1))
|
|
{
|
|
xp_awk_nde_blk_t* blk;
|
|
blk = (xp_awk_nde_blk_t*)chain->action;
|
|
if (__run_block (run, blk) == -1)
|
|
{
|
|
xp_awk_refdownval (run, v1);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
xp_awk_refdownval (run, v1);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (ptn->next->next == XP_NULL);
|
|
/* TODO: implement this */
|
|
xp_awk_refdownval (run, v1);
|
|
xp_printf (XP_T("ERROR: pattern, pattern NOT OMPLEMENTED\n"));
|
|
PANIC_I (run, XP_AWK_EINTERNAL);
|
|
}
|
|
}
|
|
|
|
chain = chain->next;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int __run_block (xp_awk_run_t* run, xp_awk_nde_blk_t* nde)
|
|
{
|
|
xp_awk_nde_t* p;
|
|
xp_size_t nlocals;
|
|
xp_size_t saved_stack_top;
|
|
int n = 0;
|
|
|
|
xp_assert (nde->type == XP_AWK_NDE_BLK);
|
|
|
|
p = nde->body;
|
|
nlocals = nde->nlocals;
|
|
|
|
/*xp_printf (XP_T("securing space for local variables nlocals = %d\n"), nlocals);*/
|
|
saved_stack_top = run->stack_top;
|
|
|
|
/* secure space for local variables */
|
|
while (nlocals > 0)
|
|
{
|
|
--nlocals;
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
/* restore stack top */
|
|
run->stack_top = saved_stack_top;
|
|
return -1;
|
|
}
|
|
|
|
/* refupval is not required for xp_awk_val_nil */
|
|
}
|
|
|
|
/*xp_printf (XP_T("executing block statements\n"));*/
|
|
while (p != XP_NULL && run->exit_level == EXIT_NONE)
|
|
{
|
|
/*xp_printf (XP_T("running a statement\n"));*/
|
|
if (__run_statement(run,p) == -1)
|
|
{
|
|
n = -1;
|
|
break;
|
|
}
|
|
p = p->next;
|
|
}
|
|
|
|
/*xp_printf (XP_T("popping off local variables\n"));*/
|
|
/* pop off local variables */
|
|
nlocals = nde->nlocals;
|
|
while (nlocals > 0)
|
|
{
|
|
--nlocals;
|
|
xp_awk_refdownval (run, STACK_LOCAL(run,nlocals));
|
|
__raw_pop (run);
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
static int __run_statement (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
switch (nde->type)
|
|
{
|
|
case XP_AWK_NDE_NULL:
|
|
{
|
|
/* do nothing */
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_BLK:
|
|
{
|
|
if (__run_block (run,
|
|
(xp_awk_nde_blk_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_IF:
|
|
{
|
|
if (__run_if (run,
|
|
(xp_awk_nde_if_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_WHILE:
|
|
case XP_AWK_NDE_DOWHILE:
|
|
{
|
|
if (__run_while (run,
|
|
(xp_awk_nde_while_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_FOR:
|
|
{
|
|
if (__run_for (run,
|
|
(xp_awk_nde_for_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_FOREACH:
|
|
{
|
|
if (__run_foreach (run,
|
|
(xp_awk_nde_foreach_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_BREAK:
|
|
{
|
|
if (__run_break (run,
|
|
(xp_awk_nde_break_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_CONTINUE:
|
|
{
|
|
if (__run_continue (run,
|
|
(xp_awk_nde_continue_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_RETURN:
|
|
{
|
|
if (__run_return (run,
|
|
(xp_awk_nde_return_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_EXIT:
|
|
{
|
|
if (__run_exit (run,
|
|
(xp_awk_nde_exit_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_NEXT:
|
|
{
|
|
if (__run_next (run,
|
|
(xp_awk_nde_next_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_NEXTFILE:
|
|
{
|
|
if (__run_nextfile (run,
|
|
(xp_awk_nde_nextfile_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_DELETE:
|
|
{
|
|
if (__run_delete (run,
|
|
(xp_awk_nde_delete_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
case XP_AWK_NDE_PRINT:
|
|
{
|
|
if (__run_print (run,
|
|
(xp_awk_nde_print_t*)nde) == -1) return -1;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
{
|
|
xp_awk_val_t* v;
|
|
v = __eval_expression(run,nde);
|
|
if (v == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (v);
|
|
xp_awk_refdownval (run, v);
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int __run_if (xp_awk_run_t* run, xp_awk_nde_if_t* nde)
|
|
{
|
|
xp_awk_val_t* test;
|
|
int n = 0;
|
|
|
|
/* the test expression for the if statement cannot have
|
|
* chained expressions. this should not be allowed by the
|
|
* parser first of all */
|
|
xp_assert (nde->test->next == XP_NULL);
|
|
|
|
test = __eval_expression (run, nde->test);
|
|
if (test == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (test);
|
|
if (xp_awk_valtobool(test))
|
|
{
|
|
n = __run_statement (run, nde->then_part);
|
|
}
|
|
else if (nde->else_part != XP_NULL)
|
|
{
|
|
n = __run_statement (run, nde->else_part);
|
|
}
|
|
|
|
xp_awk_refdownval (run, test); /* TODO: is this correct?*/
|
|
return n;
|
|
}
|
|
|
|
static int __run_while (xp_awk_run_t* run, xp_awk_nde_while_t* nde)
|
|
{
|
|
xp_awk_val_t* test;
|
|
|
|
if (nde->type == XP_AWK_NDE_WHILE)
|
|
{
|
|
/* no chained expressions are allowed for the test
|
|
* expression of the while statement */
|
|
xp_assert (nde->test->next == XP_NULL);
|
|
|
|
/* TODO: handle run-time abortion... */
|
|
while (1)
|
|
{
|
|
test = __eval_expression (run, nde->test);
|
|
if (test == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (test);
|
|
|
|
if (xp_awk_valtobool(test))
|
|
{
|
|
if (__run_statement(run,nde->body) == -1)
|
|
{
|
|
xp_awk_refdownval (run, test);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, test);
|
|
break;
|
|
}
|
|
|
|
xp_awk_refdownval (run, test);
|
|
|
|
if (run->exit_level == EXIT_BREAK)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
break;
|
|
}
|
|
else if (run->exit_level == EXIT_CONTINUE)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
}
|
|
else if (run->exit_level != EXIT_NONE) break;
|
|
}
|
|
}
|
|
else if (nde->type == XP_AWK_NDE_DOWHILE)
|
|
{
|
|
/* no chained expressions are allowed for the test
|
|
* expression of the while statement */
|
|
xp_assert (nde->test->next == XP_NULL);
|
|
|
|
/* TODO: handle run-time abortion... */
|
|
do
|
|
{
|
|
if (__run_statement(run,nde->body) == -1) return -1;
|
|
|
|
if (run->exit_level == EXIT_BREAK)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
break;
|
|
}
|
|
else if (run->exit_level == EXIT_CONTINUE)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
}
|
|
else if (run->exit_level != EXIT_NONE) break;
|
|
|
|
test = __eval_expression (run, nde->test);
|
|
if (test == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (test);
|
|
|
|
if (!xp_awk_valtobool(test))
|
|
{
|
|
xp_awk_refdownval (run, test);
|
|
break;
|
|
}
|
|
|
|
xp_awk_refdownval (run, test);
|
|
}
|
|
while (1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int __run_for (xp_awk_run_t* run, xp_awk_nde_for_t* nde)
|
|
{
|
|
xp_awk_val_t* val;
|
|
|
|
if (nde->init != XP_NULL)
|
|
{
|
|
xp_assert (nde->init->next == XP_NULL);
|
|
val = __eval_expression(run,nde->init);
|
|
if (val == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (val);
|
|
xp_awk_refdownval (run, val);
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
if (nde->test != XP_NULL)
|
|
{
|
|
xp_awk_val_t* test;
|
|
|
|
/* no chained expressions for the test expression of
|
|
* the for statement are allowed */
|
|
xp_assert (nde->test->next == XP_NULL);
|
|
|
|
test = __eval_expression (run, nde->test);
|
|
if (test == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (test);
|
|
if (xp_awk_valtobool(test))
|
|
{
|
|
if (__run_statement(run,nde->body) == -1)
|
|
{
|
|
xp_awk_refdownval (run, test);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, test);
|
|
break;
|
|
}
|
|
|
|
xp_awk_refdownval (run, test);
|
|
}
|
|
else
|
|
{
|
|
if (__run_statement(run,nde->body) == -1)
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (run->exit_level == EXIT_BREAK)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
break;
|
|
}
|
|
else if (run->exit_level == EXIT_CONTINUE)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
}
|
|
else if (run->exit_level != EXIT_NONE) break;
|
|
|
|
if (nde->incr != XP_NULL)
|
|
{
|
|
xp_assert (nde->incr->next == XP_NULL);
|
|
val = __eval_expression(run,nde->incr);
|
|
if (val == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (val);
|
|
xp_awk_refdownval (run, val);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct __foreach_walker_t
|
|
{
|
|
xp_awk_run_t* run;
|
|
xp_awk_nde_t* var;
|
|
xp_awk_nde_t* body;
|
|
};
|
|
|
|
static int __walk_foreach (xp_awk_pair_t* pair, void* arg)
|
|
{
|
|
struct __foreach_walker_t* w = (struct __foreach_walker_t*)arg;
|
|
xp_awk_val_t* str;
|
|
|
|
str = (xp_awk_val_t*)xp_awk_makestrval(pair->key,xp_strlen(pair->key));
|
|
if (str == XP_NULL) PANIC_I (w->run, XP_AWK_ENOMEM);
|
|
|
|
xp_awk_refupval (str);
|
|
if (__do_assignment (w->run, w->var, str) == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (w->run, str);
|
|
return -1;
|
|
}
|
|
|
|
if (__run_statement (w->run, w->body) == -1)
|
|
{
|
|
xp_awk_refdownval (w->run, str);
|
|
return -1;
|
|
}
|
|
|
|
xp_awk_refdownval (w->run, str);
|
|
return 0;
|
|
}
|
|
|
|
static int __run_foreach (xp_awk_run_t* run, xp_awk_nde_foreach_t* nde)
|
|
{
|
|
int n;
|
|
xp_awk_nde_exp_t* test;
|
|
xp_awk_val_t* rv;
|
|
xp_awk_map_t* map;
|
|
struct __foreach_walker_t walker;
|
|
|
|
test = (xp_awk_nde_exp_t*)nde->test;
|
|
xp_assert (test->type == XP_AWK_NDE_EXP_BIN &&
|
|
test->opcode == XP_AWK_BINOP_IN);
|
|
|
|
/* chained expressions should not be allowed
|
|
* by the parser first of all */
|
|
xp_assert (test->right->next == XP_NULL);
|
|
|
|
rv = __eval_expression (run, test->right);
|
|
if (rv == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (rv);
|
|
if (rv->type != XP_AWK_VAL_MAP)
|
|
{
|
|
xp_awk_refdownval (run, rv);
|
|
PANIC_I (run, XP_AWK_ENOTINDEXABLE);
|
|
}
|
|
map = ((xp_awk_val_map_t*)rv)->map;
|
|
|
|
walker.run = run;
|
|
walker.var = test->left;
|
|
walker.body = nde->body;
|
|
n = xp_awk_map_walk (map, __walk_foreach, &walker);
|
|
|
|
xp_awk_refdownval (run, rv);
|
|
return n;
|
|
}
|
|
|
|
static int __run_break (xp_awk_run_t* run, xp_awk_nde_break_t* nde)
|
|
{
|
|
run->exit_level = EXIT_BREAK;
|
|
return 0;
|
|
}
|
|
|
|
static int __run_continue (xp_awk_run_t* run, xp_awk_nde_continue_t* nde)
|
|
{
|
|
run->exit_level = EXIT_CONTINUE;
|
|
return 0;
|
|
}
|
|
|
|
static int __run_return (xp_awk_run_t* run, xp_awk_nde_return_t* nde)
|
|
{
|
|
|
|
if (nde->val != XP_NULL)
|
|
{
|
|
xp_awk_val_t* val;
|
|
|
|
/* chained expressions should not be allowed
|
|
* by the parser first of all */
|
|
xp_assert (nde->val->next == XP_NULL);
|
|
|
|
/*xp_printf (XP_T("returning....\n"));*/
|
|
val = __eval_expression(run, nde->val);
|
|
if (val == XP_NULL) return -1;
|
|
|
|
xp_awk_refdownval (run, STACK_RETVAL(run));
|
|
STACK_RETVAL(run) = val;
|
|
xp_awk_refupval (val); /* see __eval_call for the trick */
|
|
/*xp_printf (XP_T("set return value....\n"));*/
|
|
}
|
|
|
|
run->exit_level = EXIT_FUNCTION;
|
|
return 0;
|
|
}
|
|
|
|
static int __run_exit (xp_awk_run_t* run, xp_awk_nde_exit_t* nde)
|
|
{
|
|
if (nde->val != XP_NULL)
|
|
{
|
|
xp_awk_val_t* val;
|
|
|
|
/* chained expressions should not be allowed
|
|
* by the parser first of all */
|
|
xp_assert (nde->val->next == XP_NULL);
|
|
|
|
val = __eval_expression(run, nde->val);
|
|
if (val == XP_NULL) return -1;
|
|
|
|
xp_awk_refdownval (run, STACK_RETVAL_GLOBAL(run));
|
|
STACK_RETVAL_GLOBAL(run) = val; /* global return value */
|
|
|
|
xp_awk_refupval (val);
|
|
}
|
|
|
|
run->exit_level = EXIT_GLOBAL;
|
|
return 0;
|
|
}
|
|
|
|
static int __run_next (xp_awk_run_t* run, xp_awk_nde_next_t* nde)
|
|
{
|
|
/* TODO: trigger an error if "next" is called from BEGIN or END */
|
|
run->exit_level = EXIT_NEXT;
|
|
return 0;
|
|
}
|
|
|
|
static int __run_nextfile (xp_awk_run_t* run, xp_awk_nde_nextfile_t* nde)
|
|
{
|
|
xp_ssize_t n;
|
|
|
|
/* TODO: implement this properly */
|
|
/* TODO: how to pass opt properly for IO_NEXT??? -> READ? WRITE? */
|
|
n = run->txtio (XP_AWK_IO_NEXT, 0, run->txtio_arg, XP_NULL, 0);
|
|
if (n == -1) PANIC_I (run, XP_AWK_ETXTINNEXT);
|
|
return (n == -1)? -1: 0;
|
|
}
|
|
|
|
static int __run_delete (xp_awk_run_t* run, xp_awk_nde_delete_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* var;
|
|
|
|
var = (xp_awk_nde_var_t*) nde->var;
|
|
|
|
if (var->type == XP_AWK_NDE_NAMED ||
|
|
var->type == XP_AWK_NDE_NAMEDIDX)
|
|
{
|
|
xp_awk_pair_t* pair;
|
|
|
|
xp_assert ((var->type == XP_AWK_NDE_NAMED &&
|
|
var->idx == XP_NULL) ||
|
|
(var->type == XP_AWK_NDE_NAMEDIDX &&
|
|
var->idx != XP_NULL));
|
|
|
|
pair = xp_awk_map_get (&run->named, var->id.name);
|
|
if (pair == XP_NULL)
|
|
{
|
|
xp_awk_val_t* tmp;
|
|
|
|
/* value not set for the named variable.
|
|
* create a map and assign it to the variable */
|
|
|
|
tmp = xp_awk_makemapval (run);
|
|
if (tmp == XP_NULL) PANIC_I (run, XP_AWK_ENOMEM);
|
|
|
|
if (xp_awk_map_put (&run->named, var->id.name, tmp) == XP_NULL)
|
|
{
|
|
xp_awk_refupval (tmp);
|
|
xp_awk_refdownval (run, tmp);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_awk_refupval (tmp);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_val_t* val;
|
|
xp_awk_map_t* map;
|
|
|
|
val = (xp_awk_val_t*)pair->val;
|
|
xp_assert (val != XP_NULL);
|
|
|
|
if (val->type != XP_AWK_VAL_MAP)
|
|
PANIC_I (run, XP_AWK_ENOTDELETABLE);
|
|
|
|
map = ((xp_awk_val_map_t*)val)->map;
|
|
if (var->type == XP_AWK_NDE_NAMEDIDX)
|
|
{
|
|
xp_char_t* key;
|
|
xp_awk_val_t* idx;
|
|
int errnum;
|
|
|
|
xp_assert (var->idx != XP_NULL);
|
|
|
|
idx = __eval_expression (run, var->idx);
|
|
if (idx == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (idx);
|
|
key = xp_awk_valtostr (
|
|
idx, &errnum, XP_NULL, XP_NULL);
|
|
xp_awk_refdownval (run, idx);
|
|
|
|
if (key == XP_NULL) PANIC_I (run, errnum);
|
|
|
|
xp_awk_map_remove (map, key);
|
|
xp_free (key);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_map_clear (map);
|
|
}
|
|
}
|
|
}
|
|
else if (var->type == XP_AWK_NDE_GLOBAL ||
|
|
var->type == XP_AWK_NDE_LOCAL ||
|
|
var->type == XP_AWK_NDE_ARG ||
|
|
var->type == XP_AWK_NDE_GLOBALIDX ||
|
|
var->type == XP_AWK_NDE_LOCALIDX ||
|
|
var->type == XP_AWK_NDE_ARGIDX)
|
|
{
|
|
xp_awk_val_t* val;
|
|
|
|
if (var->type == XP_AWK_NDE_GLOBAL ||
|
|
var->type == XP_AWK_NDE_GLOBALIDX)
|
|
val = STACK_GLOBAL (run,var->id.idxa);
|
|
else if (var->type == XP_AWK_NDE_LOCAL ||
|
|
var->type == XP_AWK_NDE_LOCALIDX)
|
|
val = STACK_LOCAL (run,var->id.idxa);
|
|
else val = STACK_ARG (run,var->id.idxa);
|
|
|
|
xp_assert (val != XP_NULL);
|
|
|
|
if (val->type == XP_AWK_VAL_NIL)
|
|
{
|
|
xp_awk_val_t* tmp;
|
|
|
|
/* value not set for the named variable.
|
|
* create a map and assign it to the variable */
|
|
|
|
tmp = xp_awk_makemapval (run);
|
|
if (tmp == XP_NULL) PANIC_I (run, XP_AWK_ENOMEM);
|
|
|
|
/* no need to reduce the reference count of
|
|
* the previous value because it was nil. */
|
|
if (var->type == XP_AWK_NDE_GLOBAL ||
|
|
var->type == XP_AWK_NDE_GLOBALIDX)
|
|
STACK_GLOBAL(run,var->id.idxa) = tmp;
|
|
else if (var->type == XP_AWK_NDE_LOCAL ||
|
|
var->type == XP_AWK_NDE_LOCALIDX)
|
|
STACK_LOCAL(run,var->id.idxa) = tmp;
|
|
else STACK_ARG(run,var->id.idxa) = tmp;
|
|
|
|
xp_awk_refupval (tmp);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_map_t* map;
|
|
|
|
if (val->type != XP_AWK_VAL_MAP)
|
|
PANIC_I (run, XP_AWK_ENOTDELETABLE);
|
|
|
|
map = ((xp_awk_val_map_t*)val)->map;
|
|
if (var->type == XP_AWK_NDE_GLOBALIDX ||
|
|
var->type == XP_AWK_NDE_LOCALIDX ||
|
|
var->type == XP_AWK_NDE_ARGIDX)
|
|
{
|
|
xp_char_t* key;
|
|
xp_awk_val_t* idx;
|
|
int errnum;
|
|
|
|
xp_assert (var->idx != XP_NULL);
|
|
|
|
idx = __eval_expression (run, var->idx);
|
|
if (idx == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (idx);
|
|
key = xp_awk_valtostr (
|
|
idx, &errnum, XP_NULL, XP_NULL);
|
|
xp_awk_refdownval (run, idx);
|
|
|
|
if (key == XP_NULL) PANIC_I (run, errnum);
|
|
|
|
xp_awk_map_remove (map, key);
|
|
xp_free (key);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_map_clear (map);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_assert (!"should never happen - wrong variable type for delete");
|
|
PANIC_I (run, XP_AWK_EINTERNAL);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int __run_print (xp_awk_run_t* run, xp_awk_nde_print_t* nde)
|
|
{
|
|
xp_awk_nde_print_t* p = (xp_awk_nde_print_t*)nde;
|
|
xp_char_t* out = XP_NULL;
|
|
const xp_char_t* dst;
|
|
xp_awk_val_t* v;
|
|
xp_awk_nde_t* np;
|
|
int errnum, n;
|
|
|
|
xp_assert (
|
|
(p->out_type == XP_AWK_OUT_PIPE && p->out != XP_NULL) ||
|
|
(p->out_type == XP_AWK_OUT_COPROC && p->out != XP_NULL) ||
|
|
(p->out_type == XP_AWK_OUT_FILE && p->out != XP_NULL) ||
|
|
(p->out_type == XP_AWK_OUT_FILE_APPEND && p->out != XP_NULL) ||
|
|
(p->out_type == XP_AWK_OUT_CONSOLE && p->out == XP_NULL));
|
|
|
|
if (p->out != XP_NULL)
|
|
{
|
|
v = __eval_expression (run, p->out);
|
|
if (v == XP_NULL) return -1;
|
|
|
|
xp_awk_refupval (v);
|
|
out = xp_awk_valtostr (v, &errnum, XP_NULL, XP_NULL);
|
|
if (out == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, v);
|
|
PANIC_I (run, errnum);
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
}
|
|
|
|
dst = (out == XP_NULL)? XP_T(""): out;
|
|
|
|
if (p->args == XP_NULL)
|
|
{
|
|
/* TODO: get $0 ans use it for v */
|
|
v = xp_awk_makestrval0 (
|
|
XP_T("<TODO: PRINT $0 WITH A TRAILING NEWLINE>\n"));
|
|
if (v == XP_NULL)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_awk_refupval (v);
|
|
n = xp_awk_writeextio (run, p->out_type, dst, v, &errnum);
|
|
if (n < 0 && errnum != XP_AWK_ENOERR)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
xp_awk_refdownval (run, v);
|
|
PANIC_I (run, errnum);
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
/* TODO: how to handle n == -1 && errnum == XP_AWK_ENOERR. that is the user handler returned an error... */
|
|
}
|
|
else
|
|
{
|
|
for (np = p->args; np != XP_NULL; np = np->next)
|
|
{
|
|
v = __eval_expression (run, np);
|
|
if (v == XP_NULL)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
return -1;
|
|
}
|
|
xp_awk_refupval (v);
|
|
|
|
n = xp_awk_writeextio (
|
|
run, p->out_type, dst, v, &errnum);
|
|
if (n < 0 && errnum != XP_AWK_ENOERR)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
xp_awk_refdownval (run, v);
|
|
PANIC_I (run, errnum);
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
|
|
/* TODO: how to handle n == -1 && errnum == XP_AWK_ENOERR. that is the user handler returned an error... */
|
|
|
|
/* TODO: print proper field separator */
|
|
}
|
|
|
|
/* TODO: predefine the new line string
|
|
* for performance improvement*/
|
|
v = xp_awk_makestrval (XP_T("\n"), 1);
|
|
if (v == XP_NULL)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
xp_awk_refupval (v);
|
|
|
|
n = xp_awk_writeextio (
|
|
run, p->out_type, dst, v, &errnum);
|
|
if (n < 0 && errnum != XP_AWK_ENOERR)
|
|
{
|
|
if (out != XP_NULL) xp_free (out);
|
|
xp_awk_refdownval (run, v);
|
|
PANIC_I (run, errnum);
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
|
|
/* TODO: how to handle n == -1 && errnum == XP_AWK_ENOERR. that is the user handler returned an error... */
|
|
}
|
|
|
|
if (out != XP_NULL) xp_free (out);
|
|
return 0;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_expression (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
static eval_expr_t __eval_func[] =
|
|
{
|
|
/* the order of functions here should match the order
|
|
* of node types declared in tree.h */
|
|
__eval_group,
|
|
__eval_assignment,
|
|
__eval_binary,
|
|
__eval_unary,
|
|
__eval_incpre,
|
|
__eval_incpst,
|
|
__eval_cnd,
|
|
__eval_bfn,
|
|
__eval_afn,
|
|
__eval_int,
|
|
__eval_real,
|
|
__eval_str,
|
|
__eval_rex,
|
|
__eval_named,
|
|
__eval_global,
|
|
__eval_local,
|
|
__eval_arg,
|
|
__eval_namedidx,
|
|
__eval_globalidx,
|
|
__eval_localidx,
|
|
__eval_argidx,
|
|
__eval_pos,
|
|
__eval_getline
|
|
};
|
|
|
|
xp_assert (nde->type >= XP_AWK_NDE_GRP &&
|
|
(nde->type - XP_AWK_NDE_GRP) < xp_countof(__eval_func));
|
|
return __eval_func[nde->type-XP_AWK_NDE_GRP] (run, nde);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_group (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
/* __eval_binop_in evaluates the XP_AWK_NDE_GRP specially.
|
|
* so this function should never be reached. */
|
|
xp_assert (!"should never happen - NDE_GRP only for in");
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
return XP_NULL;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_assignment (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* val, * ret;
|
|
xp_awk_nde_ass_t* ass = (xp_awk_nde_ass_t*)nde;
|
|
|
|
xp_assert (ass->left != XP_NULL && ass->right != XP_NULL);
|
|
|
|
xp_assert (ass->right->next == XP_NULL);
|
|
val = __eval_expression(run, ass->right);
|
|
if (val == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (val);
|
|
|
|
if (ass->opcode != XP_AWK_ASSOP_NONE)
|
|
{
|
|
xp_awk_val_t* val2, * tmp;
|
|
|
|
xp_assert (ass->left->next == XP_NULL);
|
|
val2 = __eval_expression (run, ass->left);
|
|
if (val2 == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, val);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refupval (val2);
|
|
|
|
if (ass->opcode == XP_AWK_ASSOP_PLUS)
|
|
{
|
|
tmp = __eval_binop_plus (run, val2, val);
|
|
}
|
|
else if (ass->opcode == XP_AWK_ASSOP_MINUS)
|
|
{
|
|
tmp = __eval_binop_minus (run, val2, val);
|
|
}
|
|
else if (ass->opcode == XP_AWK_ASSOP_MUL)
|
|
{
|
|
tmp = __eval_binop_mul (run, val2, val);
|
|
}
|
|
else if (ass->opcode == XP_AWK_ASSOP_DIV)
|
|
{
|
|
tmp = __eval_binop_div (run, val2, val);
|
|
}
|
|
else if (ass->opcode == XP_AWK_ASSOP_MOD)
|
|
{
|
|
tmp = __eval_binop_mod (run, val2, val);
|
|
}
|
|
else if (ass->opcode == XP_AWK_ASSOP_EXP)
|
|
{
|
|
tmp = __eval_binop_exp (run, val2, val);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (!"should never happen - invalid assignment opcode");
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
}
|
|
|
|
if (tmp == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, val);
|
|
xp_awk_refdownval (run, val2);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refdownval (run, val);
|
|
val = tmp;
|
|
xp_awk_refupval (val);
|
|
}
|
|
|
|
ret = __do_assignment (run, ass->left, val);
|
|
xp_awk_refdownval (run, val);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static xp_awk_val_t* __do_assignment (
|
|
xp_awk_run_t* run, xp_awk_nde_t* var, xp_awk_val_t* val)
|
|
{
|
|
xp_awk_val_t* ret;
|
|
|
|
if (val->type == XP_AWK_VAL_MAP)
|
|
{
|
|
/* a map cannot be assigned to a variable */
|
|
PANIC (run, XP_AWK_ENOTASSIGNABLE);
|
|
}
|
|
|
|
if (var->type == XP_AWK_NDE_NAMED ||
|
|
var->type == XP_AWK_NDE_GLOBAL ||
|
|
var->type == XP_AWK_NDE_LOCAL ||
|
|
var->type == XP_AWK_NDE_ARG)
|
|
{
|
|
ret = __do_assignment_scalar (run, (xp_awk_nde_var_t*)var, val);
|
|
}
|
|
else if (var->type == XP_AWK_NDE_NAMEDIDX ||
|
|
var->type == XP_AWK_NDE_GLOBALIDX ||
|
|
var->type == XP_AWK_NDE_LOCALIDX ||
|
|
var->type == XP_AWK_NDE_ARGIDX)
|
|
{
|
|
ret = __do_assignment_map (run, (xp_awk_nde_var_t*)var, val);
|
|
}
|
|
else if (var->type == XP_AWK_NDE_POS)
|
|
{
|
|
ret = __do_assignment_pos (run, (xp_awk_nde_pos_t*)var, val);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (!"should never happen - invalid variable type");
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static xp_awk_val_t* __do_assignment_scalar (
|
|
xp_awk_run_t* run, xp_awk_nde_var_t* var, xp_awk_val_t* val)
|
|
{
|
|
xp_assert (
|
|
(var->type == XP_AWK_NDE_NAMED ||
|
|
var->type == XP_AWK_NDE_GLOBAL ||
|
|
var->type == XP_AWK_NDE_LOCAL ||
|
|
var->type == XP_AWK_NDE_ARG) && var->idx == XP_NULL);
|
|
|
|
xp_assert (val->type != XP_AWK_VAL_MAP);
|
|
|
|
if (var->type == XP_AWK_NDE_NAMED)
|
|
{
|
|
xp_awk_pair_t* pair;
|
|
int n;
|
|
|
|
pair = xp_awk_map_get (&run->named, var->id.name);
|
|
if (pair != XP_NULL &&
|
|
((xp_awk_val_t*)pair->val)->type == XP_AWK_VAL_MAP)
|
|
{
|
|
/* once a variable becomes an array,
|
|
* it cannot be changed to a scalar variable */
|
|
PANIC (run, XP_AWK_ENOTSCALARIZABLE);
|
|
}
|
|
|
|
n = xp_awk_map_putx (
|
|
&run->named, var->id.name, val, XP_NULL);
|
|
if (n < 0) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
xp_awk_refupval (val);
|
|
}
|
|
else if (var->type == XP_AWK_NDE_GLOBAL)
|
|
{
|
|
xp_awk_val_t* old = STACK_GLOBAL(run,var->id.idxa);
|
|
if (old->type == XP_AWK_VAL_MAP)
|
|
{
|
|
/* once a variable becomes an array,
|
|
* it cannot be changed to a scalar variable */
|
|
PANIC (run, XP_AWK_ENOTSCALARIZABLE);
|
|
}
|
|
|
|
xp_awk_refdownval (run, old);
|
|
STACK_GLOBAL(run,var->id.idxa) = val;
|
|
xp_awk_refupval (val);
|
|
}
|
|
else if (var->type == XP_AWK_NDE_LOCAL)
|
|
{
|
|
xp_awk_val_t* old = STACK_LOCAL(run,var->id.idxa);
|
|
if (old->type == XP_AWK_VAL_MAP)
|
|
{
|
|
/* once the variable becomes an array,
|
|
* it cannot be changed to a scalar variable */
|
|
PANIC (run, XP_AWK_ENOTSCALARIZABLE);
|
|
}
|
|
|
|
xp_awk_refdownval (run, old);
|
|
STACK_LOCAL(run,var->id.idxa) = val;
|
|
xp_awk_refupval (val);
|
|
}
|
|
else /* if (var->type == XP_AWK_NDE_ARG) */
|
|
{
|
|
xp_awk_val_t* old = STACK_ARG(run,var->id.idxa);
|
|
if (old->type == XP_AWK_VAL_MAP)
|
|
{
|
|
/* once the variable becomes an array,
|
|
* it cannot be changed to a scalar variable */
|
|
PANIC (run, XP_AWK_ENOTSCALARIZABLE);
|
|
}
|
|
|
|
xp_awk_refdownval (run, old);
|
|
STACK_ARG(run,var->id.idxa) = val;
|
|
xp_awk_refupval (val);
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __do_assignment_map (
|
|
xp_awk_run_t* run, xp_awk_nde_var_t* var, xp_awk_val_t* val)
|
|
{
|
|
xp_awk_val_map_t* map;
|
|
xp_char_t* str;
|
|
int n;
|
|
|
|
xp_assert (
|
|
(var->type == XP_AWK_NDE_NAMEDIDX ||
|
|
var->type == XP_AWK_NDE_GLOBALIDX ||
|
|
var->type == XP_AWK_NDE_LOCALIDX ||
|
|
var->type == XP_AWK_NDE_ARGIDX) && var->idx != XP_NULL);
|
|
xp_assert (val->type != XP_AWK_VAL_MAP);
|
|
|
|
if (var->type == XP_AWK_NDE_NAMEDIDX)
|
|
{
|
|
xp_awk_pair_t* pair;
|
|
pair = xp_awk_map_get (&run->named, var->id.name);
|
|
map = (pair == XP_NULL)?
|
|
(xp_awk_val_map_t*)xp_awk_val_nil:
|
|
(xp_awk_val_map_t*)pair->val;
|
|
}
|
|
else
|
|
{
|
|
map = (var->type == XP_AWK_NDE_GLOBALIDX)?
|
|
(xp_awk_val_map_t*)STACK_GLOBAL(run,var->id.idxa):
|
|
(var->type == XP_AWK_NDE_LOCALIDX)?
|
|
(xp_awk_val_map_t*)STACK_LOCAL(run,var->id.idxa):
|
|
(xp_awk_val_map_t*)STACK_ARG(run,var->id.idxa);
|
|
}
|
|
|
|
if (map->type == XP_AWK_VAL_NIL)
|
|
{
|
|
/* the map is not initialized yet */
|
|
xp_awk_val_t* tmp;
|
|
|
|
tmp = xp_awk_makemapval (run);
|
|
if (tmp == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
if (var->type == XP_AWK_NDE_NAMEDIDX)
|
|
{
|
|
/* doesn't have to decrease the reference count
|
|
* of the previous value here as it is done by
|
|
* xp_awk_map_put */
|
|
if (xp_awk_map_put (
|
|
&run->named, var->id.name, tmp) == XP_NULL)
|
|
{
|
|
xp_awk_refupval (tmp);
|
|
xp_awk_refdownval (run, tmp);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
else if (var->type == XP_AWK_NDE_GLOBALIDX)
|
|
{
|
|
/* decrease the reference count of the previous value.
|
|
* in fact, this is not necessary as map is always
|
|
* xp_awk_val_nil here. */
|
|
xp_awk_refdownval (run, (xp_awk_val_t*)map);
|
|
STACK_GLOBAL(run,var->id.idxa) = tmp;
|
|
}
|
|
else if (var->type == XP_AWK_NDE_LOCALIDX)
|
|
{
|
|
xp_awk_refdownval (run, (xp_awk_val_t*)map);
|
|
STACK_LOCAL(run,var->id.idxa) = tmp;
|
|
}
|
|
else /* if (var->type == XP_AWK_NDE_ARGIDX) */
|
|
{
|
|
xp_awk_refdownval (run, (xp_awk_val_t*)map);
|
|
STACK_ARG(run,var->id.idxa) = tmp;
|
|
}
|
|
|
|
xp_awk_refupval (tmp);
|
|
map = (xp_awk_val_map_t*) tmp;
|
|
}
|
|
else if (map->type != XP_AWK_VAL_MAP)
|
|
{
|
|
PANIC (run, XP_AWK_ENOTINDEXABLE);
|
|
}
|
|
|
|
str = __idxnde_to_str (run, var->idx);
|
|
if (str == XP_NULL) return XP_NULL;
|
|
|
|
/*
|
|
xp_printf (XP_T("**** index str=>%s, map->ref=%d, map->type=%d\n"), str, map->ref, map->type);
|
|
*/
|
|
n = xp_awk_map_putx (map->map, str, val, XP_NULL);
|
|
if (n < 0)
|
|
{
|
|
xp_free (str);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_free (str);
|
|
xp_awk_refupval (val);
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __do_assignment_pos (
|
|
xp_awk_run_t* run, xp_awk_nde_pos_t* pos, xp_awk_val_t* val)
|
|
{
|
|
xp_awk_val_t* v;
|
|
xp_long_t lv;
|
|
xp_real_t rv;
|
|
xp_char_t* str;
|
|
xp_size_t len;
|
|
int n, errnum;
|
|
|
|
/* get the position number */
|
|
v = __eval_expression (run, pos->val);
|
|
if (v == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (v);
|
|
n = xp_awk_valtonum (v, &lv, &rv);
|
|
xp_awk_refdownval (run, v);
|
|
|
|
if (n == -1) PANIC (run, XP_AWK_EPOSIDX);
|
|
if (n == 1) lv = (xp_long_t)rv;
|
|
if (lv < 0) PANIC (run, XP_AWK_EPOSIDX);
|
|
|
|
/* convert the value to the string */
|
|
str = xp_awk_valtostr (val, &errnum, XP_NULL, &len);
|
|
if (str == XP_NULL) PANIC (run, errnum);
|
|
|
|
if (lv == 0)
|
|
{
|
|
__clear_record (run, xp_false);
|
|
|
|
if (xp_str_ncpy (&run->inrec.line, str, len) == (xp_size_t)-1)
|
|
{
|
|
xp_free (str);
|
|
PANIC (run, errnum);
|
|
}
|
|
xp_free (str);
|
|
|
|
/* TODO: consider if "val" of a non-string type should be
|
|
* converted into a string value */
|
|
xp_awk_refdownval (run, run->inrec.d0);
|
|
run->inrec.d0 = val;
|
|
xp_awk_refupval (val);
|
|
|
|
if (__split_record (run) == -1)
|
|
{
|
|
__clear_record (run, xp_false);
|
|
return XP_NULL;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (__recomp_record_fields (
|
|
run, (xp_size_t)lv, str, len, &errnum) == -1)
|
|
{
|
|
xp_free (str);
|
|
__clear_record (run, xp_false);
|
|
PANIC (run, errnum);
|
|
}
|
|
xp_free (str);
|
|
|
|
/* recompose $0 */
|
|
v = xp_awk_makestrval (
|
|
XP_STR_BUF(&run->inrec.line),
|
|
XP_STR_LEN(&run->inrec.line));
|
|
if (v == XP_NULL)
|
|
{
|
|
__clear_record (run, xp_false);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_awk_refdownval (run, run->inrec.d0);
|
|
run->inrec.d0 = v;
|
|
xp_awk_refupval (v);
|
|
|
|
val = v;
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binary (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
static binop_func_t __binop_func[] =
|
|
{
|
|
/* the order of the functions should be inline with
|
|
* the operator declaration in run.h */
|
|
|
|
XP_NULL, /* __eval_binop_lor */
|
|
XP_NULL, /* __eval_binop_land */
|
|
XP_NULL, /* __eval_binop_in */
|
|
|
|
__eval_binop_bor,
|
|
__eval_binop_bxor,
|
|
__eval_binop_band,
|
|
|
|
__eval_binop_eq,
|
|
__eval_binop_ne,
|
|
__eval_binop_gt,
|
|
__eval_binop_ge,
|
|
__eval_binop_lt,
|
|
__eval_binop_le,
|
|
|
|
__eval_binop_lshift,
|
|
__eval_binop_rshift,
|
|
|
|
__eval_binop_plus,
|
|
__eval_binop_minus,
|
|
__eval_binop_mul,
|
|
__eval_binop_div,
|
|
__eval_binop_mod,
|
|
__eval_binop_exp,
|
|
|
|
__eval_binop_concat,
|
|
__eval_binop_ma,
|
|
__eval_binop_nm
|
|
};
|
|
|
|
xp_awk_nde_exp_t* exp = (xp_awk_nde_exp_t*)nde;
|
|
xp_awk_val_t* left, * right, * res;
|
|
|
|
xp_assert (exp->type == XP_AWK_NDE_EXP_BIN);
|
|
|
|
if (exp->opcode == XP_AWK_BINOP_LAND)
|
|
{
|
|
res = __eval_binop_land (run, exp->left, exp->right);
|
|
}
|
|
else if (exp->opcode == XP_AWK_BINOP_LOR)
|
|
{
|
|
res = __eval_binop_lor (run, exp->left, exp->right);
|
|
}
|
|
else if (exp->opcode == XP_AWK_BINOP_IN)
|
|
{
|
|
/* treat the in operator specially */
|
|
res = __eval_binop_in (run, exp->left, exp->right);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (exp->left->next == XP_NULL);
|
|
left = __eval_expression (run, exp->left);
|
|
if (left == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (left);
|
|
|
|
xp_assert (exp->right->next == XP_NULL);
|
|
right = __eval_expression (run, exp->right);
|
|
if (right == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refupval (right);
|
|
|
|
xp_assert (exp->opcode >= 0 &&
|
|
exp->opcode < xp_countof(__binop_func));
|
|
xp_assert (__binop_func[exp->opcode] != XP_NULL);
|
|
|
|
res = __binop_func[exp->opcode] (run, left, right);
|
|
|
|
xp_awk_refdownval (run, left);
|
|
xp_awk_refdownval (run, right);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_lor (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right)
|
|
{
|
|
/*
|
|
xp_awk_val_t* res = XP_NULL;
|
|
|
|
res = xp_awk_makeintval (run,
|
|
xp_awk_valtobool(left) || xp_awk_valtobool(right));
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
return res;
|
|
*/
|
|
|
|
/* short-circuit evaluation required special treatment */
|
|
xp_awk_val_t* lv, * rv, * res;
|
|
|
|
xp_assert (left->next == XP_NULL);
|
|
lv = __eval_expression (run, left);
|
|
if (lv == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (lv);
|
|
if (xp_awk_valtobool(lv))
|
|
{
|
|
res = xp_awk_makeintval (run, 1);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (right->next == XP_NULL);
|
|
rv = __eval_expression (run, right);
|
|
if (rv == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, lv);
|
|
return XP_NULL;
|
|
}
|
|
xp_awk_refupval (rv);
|
|
|
|
res = xp_awk_makeintval (run, (xp_awk_valtobool(rv)? 1: 0));
|
|
xp_awk_refdownval (run, rv);
|
|
}
|
|
|
|
xp_awk_refdownval (run, lv);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_land (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right)
|
|
{
|
|
/*
|
|
xp_awk_val_t* res = XP_NULL;
|
|
|
|
res = xp_awk_makeintval (run,
|
|
xp_awk_valtobool(left) && xp_awk_valtobool(right));
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
return res;
|
|
*/
|
|
|
|
/* short-circuit evaluation required special treatment */
|
|
xp_awk_val_t* lv, * rv, * res;
|
|
|
|
xp_assert (left->next == XP_NULL);
|
|
lv = __eval_expression (run, left);
|
|
if (lv == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (lv);
|
|
if (!xp_awk_valtobool(lv))
|
|
{
|
|
res = xp_awk_makeintval (run, 0);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (right->next == XP_NULL);
|
|
rv = __eval_expression (run, right);
|
|
if (rv == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, lv);
|
|
return XP_NULL;
|
|
}
|
|
xp_awk_refupval (rv);
|
|
|
|
res = xp_awk_makeintval (run, (xp_awk_valtobool(rv)? 1: 0));
|
|
xp_awk_refdownval (run, rv);
|
|
}
|
|
|
|
xp_awk_refdownval (run, lv);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_in (
|
|
xp_awk_run_t* run, xp_awk_nde_t* left, xp_awk_nde_t* right)
|
|
{
|
|
xp_awk_val_t* rv, * res;
|
|
xp_char_t* str;
|
|
|
|
if (right->type != XP_AWK_NDE_GLOBAL &&
|
|
right->type != XP_AWK_NDE_LOCAL &&
|
|
right->type != XP_AWK_NDE_ARG &&
|
|
right->type != XP_AWK_NDE_NAMED)
|
|
{
|
|
/* the compiler should have handled this case */
|
|
xp_assert (!"should never happen - in needs a plain variable");
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
return XP_NULL;
|
|
}
|
|
|
|
/* evaluate the left-hand side of the operator */
|
|
str = (left->type == XP_AWK_NDE_GRP)?
|
|
__idxnde_to_str (run, ((xp_awk_nde_grp_t*)left)->body):
|
|
__idxnde_to_str (run, left);
|
|
if (str == XP_NULL) return XP_NULL;
|
|
|
|
/* evaluate the right-hand side of the operator */
|
|
xp_assert (right->next == XP_NULL);
|
|
rv = __eval_expression (run, right);
|
|
if (rv == XP_NULL)
|
|
{
|
|
xp_free (str);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refupval (rv);
|
|
|
|
if (rv->type == XP_AWK_VAL_NIL)
|
|
{
|
|
res = xp_awk_makeintval (run, 0);
|
|
if (res == XP_NULL)
|
|
{
|
|
xp_free (str);
|
|
xp_awk_refdownval (run, rv);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_free (str);
|
|
xp_awk_refdownval (run, rv);
|
|
return res;
|
|
}
|
|
else if (rv->type == XP_AWK_VAL_MAP)
|
|
{
|
|
xp_long_t r;
|
|
|
|
r = xp_awk_map_get(((xp_awk_val_map_t*)rv)->map,str) != XP_NULL;
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL)
|
|
{
|
|
xp_free (str);
|
|
xp_awk_refdownval (run, rv);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_free (str);
|
|
xp_awk_refdownval (run, rv);
|
|
return res;
|
|
}
|
|
|
|
/* need an array */
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
return XP_NULL;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_bor (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r =
|
|
((xp_awk_val_int_t*)left)->val |
|
|
((xp_awk_val_int_t*)right)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_bxor (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r =
|
|
((xp_awk_val_int_t*)left)->val ^
|
|
((xp_awk_val_int_t*)right)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_band (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r =
|
|
((xp_awk_val_int_t*)left)->val &
|
|
((xp_awk_val_int_t*)right)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_eq (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_NIL ||
|
|
right->type == XP_AWK_VAL_NIL)
|
|
{
|
|
r = (left->type == right->type);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val ==
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val ==
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val ==
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val ==
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) == 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_ne (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_NIL ||
|
|
right->type == XP_AWK_VAL_NIL)
|
|
{
|
|
r = (left->type != right->type);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val !=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val !=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val !=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val !=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) != 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_gt (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val >
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val >
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val >
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val >
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) > 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_ge (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val >=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val >=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val >=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val >=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) >= 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_lt (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val <
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val <
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val <
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val <
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) < 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_le (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res = XP_NULL;
|
|
xp_long_t r = 0;
|
|
|
|
if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val <=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val <=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_INT &&
|
|
right->type == XP_AWK_VAL_REAL)
|
|
{
|
|
r = ((xp_awk_val_int_t*)left)->val <=
|
|
((xp_awk_val_real_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL &&
|
|
right->type == XP_AWK_VAL_INT)
|
|
{
|
|
r = ((xp_awk_val_real_t*)left)->val <=
|
|
((xp_awk_val_int_t*)right)->val;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
r = xp_strxncmp (
|
|
((xp_awk_val_str_t*)left)->buf,
|
|
((xp_awk_val_str_t*)left)->len,
|
|
((xp_awk_val_str_t*)right)->buf,
|
|
((xp_awk_val_str_t*)right)->len) <= 0;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_lshift (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
if (n3 == 0)
|
|
{
|
|
if (l2 == 0) PANIC (run, XP_AWK_EDIVBYZERO);
|
|
res = xp_awk_makeintval (run, (xp_long_t)l1 << (xp_long_t)l2);
|
|
}
|
|
else PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_rshift (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
if (n3 == 0)
|
|
{
|
|
if (l2 == 0) PANIC (run, XP_AWK_EDIVBYZERO);
|
|
res = xp_awk_makeintval (run, (xp_long_t)l1 >> (xp_long_t)l2);
|
|
}
|
|
else PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_plus (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
/*
|
|
n1 n2 n3
|
|
0 0 = 0
|
|
1 0 = 1
|
|
0 1 = 2
|
|
1 1 = 3
|
|
*/
|
|
n3 = n1 + (n2 << 1);
|
|
xp_assert (n3 >= 0 && n3 <= 3);
|
|
res = (n3 == 0)? xp_awk_makeintval(run,(xp_long_t)l1+(xp_long_t)l2):
|
|
(n3 == 1)? xp_awk_makerealval(run,(xp_real_t)r1+(xp_real_t)l2):
|
|
(n3 == 2)? xp_awk_makerealval(run,(xp_real_t)l1+(xp_real_t)r2):
|
|
xp_awk_makerealval(run,(xp_real_t)r1+(xp_real_t)r2);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_minus (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
xp_assert (n3 >= 0 && n3 <= 3);
|
|
res = (n3 == 0)? xp_awk_makeintval(run,(xp_long_t)l1-(xp_long_t)l2):
|
|
(n3 == 1)? xp_awk_makerealval(run,(xp_real_t)r1-(xp_real_t)l2):
|
|
(n3 == 2)? xp_awk_makerealval(run,(xp_real_t)l1-(xp_real_t)r2):
|
|
xp_awk_makerealval(run,(xp_real_t)r1-(xp_real_t)r2);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_mul (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
xp_assert (n3 >= 0 && n3 <= 3);
|
|
res = (n3 == 0)? xp_awk_makeintval(run,(xp_long_t)l1*(xp_long_t)l2):
|
|
(n3 == 1)? xp_awk_makerealval(run,(xp_real_t)r1*(xp_real_t)l2):
|
|
(n3 == 2)? xp_awk_makerealval(run,(xp_real_t)l1*(xp_real_t)r2):
|
|
xp_awk_makerealval(run,(xp_real_t)r1*(xp_real_t)r2);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_div (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
if (n3 == 0)
|
|
{
|
|
if (l2 == 0) PANIC (run, XP_AWK_EDIVBYZERO);
|
|
res = xp_awk_makeintval (run, (xp_long_t)l1 / (xp_long_t)l2);
|
|
}
|
|
else if (n3 == 1)
|
|
{
|
|
res = xp_awk_makerealval (run, (xp_real_t)r1 / (xp_real_t)l2);
|
|
}
|
|
else if (n3 == 2)
|
|
{
|
|
res = xp_awk_makerealval (run, (xp_real_t)l1 / (xp_real_t)r2);
|
|
}
|
|
else
|
|
{
|
|
xp_assert (n3 == 3);
|
|
res = xp_awk_makerealval (run, (xp_real_t)r1 / (xp_real_t)r2);
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_mod (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
if (n3 == 0)
|
|
{
|
|
if (l2 == 0) PANIC (run, XP_AWK_EDIVBYZERO);
|
|
res = xp_awk_makeintval (run, (xp_long_t)l1 % (xp_long_t)l2);
|
|
}
|
|
else PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_exp (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
int n1, n2, n3;
|
|
xp_long_t l1, l2;
|
|
xp_real_t r1, r2;
|
|
xp_awk_val_t* res;
|
|
|
|
n1 = xp_awk_valtonum (left, &l1, &r1);
|
|
n2 = xp_awk_valtonum (right, &l2, &r2);
|
|
|
|
if (n1 == -1 || n2 == -1) PANIC (run, XP_AWK_EOPERAND);
|
|
|
|
n3 = n1 + (n2 << 1);
|
|
if (n3 == 0)
|
|
{
|
|
xp_long_t v = 1;
|
|
while (l2-- > 0) v *= l1;
|
|
res = xp_awk_makeintval (run, v);
|
|
}
|
|
else if (n3 == 1)
|
|
{
|
|
/*res = xp_awk_makerealval (
|
|
run, pow((xp_real_t)r1,(xp_real_t)l2));*/
|
|
xp_real_t v = 1.0;
|
|
while (l2-- > 0) v *= r1;
|
|
res = xp_awk_makerealval (run, v);
|
|
}
|
|
else if (n3 == 2)
|
|
{
|
|
res = xp_awk_makerealval (
|
|
run, pow((xp_real_t)l1,(xp_real_t)r2));
|
|
}
|
|
else
|
|
{
|
|
xp_assert (n3 == 3);
|
|
res = xp_awk_makerealval (
|
|
run, pow((xp_real_t)r1,(xp_real_t)r2));
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_concat (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_str_t lv, rv;
|
|
xp_awk_val_t* res;
|
|
int errnum;
|
|
|
|
if (xp_str_open (&lv, DEF_BUF_CAPA) == XP_NULL)
|
|
{
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
if (xp_str_open (&rv, DEF_BUF_CAPA) == XP_NULL)
|
|
{
|
|
xp_str_close (&lv);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
if (xp_awk_valtostr (left, &errnum, &lv, XP_NULL) == XP_NULL)
|
|
{
|
|
xp_str_close (&lv);
|
|
xp_str_close (&rv);
|
|
PANIC (run, errnum);
|
|
}
|
|
|
|
if (xp_awk_valtostr (right, &errnum, &rv, XP_NULL) == XP_NULL)
|
|
{
|
|
xp_str_close (&lv);
|
|
xp_str_close (&rv);
|
|
PANIC (run, errnum);
|
|
}
|
|
|
|
res = xp_awk_makestrval2 (
|
|
XP_STR_BUF(&lv), XP_STR_LEN(&lv),
|
|
XP_STR_BUF(&rv), XP_STR_LEN(&rv));
|
|
if (res == XP_NULL)
|
|
{
|
|
xp_str_close (&lv);
|
|
xp_str_close (&rv);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_str_close (&lv);
|
|
xp_str_close (&rv);
|
|
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_ma (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
/* TODO: ... */
|
|
xp_printf (XP_T("eval_binop_ma not implemented yet...\n"));
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
return XP_NULL;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_binop_nm (
|
|
xp_awk_run_t* run, xp_awk_val_t* left, xp_awk_val_t* right)
|
|
{
|
|
xp_awk_val_t* res;
|
|
|
|
/* TODO: implement nm operator... */
|
|
if (left->type == XP_AWK_VAL_REX &&
|
|
right->type == XP_AWK_VAL_STR)
|
|
{
|
|
res = xp_awk_val_nil;
|
|
}
|
|
else if (left->type == XP_AWK_VAL_STR &&
|
|
right->type == XP_AWK_VAL_REX)
|
|
{
|
|
res = xp_awk_val_nil;
|
|
}
|
|
else
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_unary (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* left, * res = XP_NULL;
|
|
xp_awk_nde_exp_t* exp = (xp_awk_nde_exp_t*)nde;
|
|
|
|
xp_assert (exp->type == XP_AWK_NDE_EXP_UNR);
|
|
xp_assert (exp->left != XP_NULL && exp->right == XP_NULL);
|
|
|
|
xp_assert (exp->left->next == XP_NULL);
|
|
left = __eval_expression (run, exp->left);
|
|
if (left == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (left);
|
|
|
|
if (exp->opcode == XP_AWK_UNROP_PLUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, r);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else if (exp->opcode == XP_AWK_UNROP_MINUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, -r);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, -r);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else if (exp->opcode == XP_AWK_UNROP_NOT)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, !r);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, !r);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else if (exp->opcode == XP_AWK_UNROP_BNOT)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, ~r);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
|
|
if (res == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_awk_refdownval (run, left);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_incpre (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* left, * res;
|
|
xp_awk_nde_exp_t* exp = (xp_awk_nde_exp_t*)nde;
|
|
|
|
xp_assert (exp->type == XP_AWK_NDE_EXP_INCPRE);
|
|
xp_assert (exp->left != XP_NULL && exp->right == XP_NULL);
|
|
|
|
/* ugly. but let's keep going this way for the time being */
|
|
/*if (exp->left->type != XP_AWK_NDE_NAMED &&
|
|
exp->left->type != XP_AWK_NDE_GLOBAL &&
|
|
exp->left->type != XP_AWK_NDE_LOCAL &&
|
|
exp->left->type != XP_AWK_NDE_ARG &&
|
|
exp->left->type != XP_AWK_NDE_NAMEDIDX &&
|
|
exp->left->type != XP_AWK_NDE_GLOBALIDX &&
|
|
exp->left->type != XP_AWK_NDE_LOCALIDX &&
|
|
exp->left->type != XP_AWK_NDE_ARGIDX) */
|
|
if (exp->left->type < XP_AWK_NDE_NAMED ||
|
|
exp->left->type > XP_AWK_NDE_ARGIDX)
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
xp_assert (exp->left->next == XP_NULL);
|
|
left = __eval_expression (run, exp->left);
|
|
if (left == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (left);
|
|
|
|
if (exp->opcode == XP_AWK_INCOP_PLUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, r + 1);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, r + 1.0);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else if (exp->opcode == XP_AWK_INCOP_MINUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, r - 1);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, r - 1.0);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_assert (!"should never happen - invalid opcode");
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
}
|
|
|
|
if (__do_assignment (run, exp->left, res) == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refdownval (run, left);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_incpst (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* left, * res, * res2;
|
|
xp_awk_nde_exp_t* exp = (xp_awk_nde_exp_t*)nde;
|
|
|
|
xp_assert (exp->type == XP_AWK_NDE_EXP_INCPST);
|
|
xp_assert (exp->left != XP_NULL && exp->right == XP_NULL);
|
|
|
|
/* ugly. but let's keep going this way for the time being */
|
|
/*if (exp->left->type != XP_AWK_NDE_NAMED &&
|
|
exp->left->type != XP_AWK_NDE_GLOBAL &&
|
|
exp->left->type != XP_AWK_NDE_LOCAL &&
|
|
exp->left->type != XP_AWK_NDE_ARG &&
|
|
exp->left->type != XP_AWK_NDE_NAMEDIDX &&
|
|
exp->left->type != XP_AWK_NDE_GLOBALIDX &&
|
|
exp->left->type != XP_AWK_NDE_LOCALIDX &&
|
|
exp->left->type != XP_AWK_NDE_ARGIDX) */
|
|
if (exp->left->type < XP_AWK_NDE_NAMED ||
|
|
exp->left->type > XP_AWK_NDE_ARGIDX)
|
|
{
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
|
|
xp_assert (exp->left->next == XP_NULL);
|
|
left = __eval_expression (run, exp->left);
|
|
if (left == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (left);
|
|
|
|
if (exp->opcode == XP_AWK_INCOP_PLUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
res2 = xp_awk_makeintval (run, r + 1);
|
|
if (res2 == XP_NULL)
|
|
{
|
|
xp_awk_freeval (run, res, xp_true);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
res2 = xp_awk_makerealval (run, r + 1.0);
|
|
if (res2 == XP_NULL)
|
|
{
|
|
xp_awk_freeval (run, res, xp_true);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else if (exp->opcode == XP_AWK_INCOP_MINUS)
|
|
{
|
|
if (left->type == XP_AWK_VAL_INT)
|
|
{
|
|
xp_long_t r = ((xp_awk_val_int_t*)left)->val;
|
|
res = xp_awk_makeintval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
res2 = xp_awk_makeintval (run, r - 1);
|
|
if (res2 == XP_NULL)
|
|
{
|
|
xp_awk_freeval (run, res, xp_true);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
else if (left->type == XP_AWK_VAL_REAL)
|
|
{
|
|
xp_real_t r = ((xp_awk_val_real_t*)left)->val;
|
|
res = xp_awk_makerealval (run, r);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
res2 = xp_awk_makerealval (run, r - 1.0);
|
|
if (res2 == XP_NULL)
|
|
{
|
|
xp_awk_freeval (run, res, xp_true);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EOPERAND);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
xp_assert (!"should never happen - invalid opcode");
|
|
xp_awk_refdownval (run, left);
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
}
|
|
|
|
if (__do_assignment (run, exp->left, res2) == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, left);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refdownval (run, left);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_cnd (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* tv, * v;
|
|
xp_awk_nde_cnd_t* cnd = (xp_awk_nde_cnd_t*)nde;
|
|
|
|
xp_assert (cnd->test->next == XP_NULL);
|
|
tv = __eval_expression (run, cnd->test);
|
|
if (tv == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (tv);
|
|
|
|
xp_assert (cnd->left->next == XP_NULL &&
|
|
cnd->right->next == XP_NULL);
|
|
v = (xp_awk_valtobool(tv))?
|
|
__eval_expression (run, cnd->left):
|
|
__eval_expression (run, cnd->right);
|
|
|
|
xp_awk_refdownval (run, tv);
|
|
return v;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_bfn (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_call_t* call = (xp_awk_nde_call_t*)nde;
|
|
|
|
/* built-in function */
|
|
if (call->nargs < call->what.bfn.min_args)
|
|
{
|
|
PANIC (run, XP_AWK_ETOOFEWARGS);
|
|
}
|
|
|
|
if (call->nargs > call->what.bfn.max_args)
|
|
{
|
|
PANIC (run, XP_AWK_ETOOMANYARGS);
|
|
}
|
|
|
|
return __eval_call (run, nde, XP_NULL);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_afn (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_call_t* call = (xp_awk_nde_call_t*)nde;
|
|
xp_awk_afn_t* afn;
|
|
xp_awk_pair_t* pair;
|
|
|
|
pair = xp_awk_map_get (&run->awk->tree.afns, call->what.name);
|
|
if (pair == XP_NULL) PANIC (run, XP_AWK_ENOSUCHFUNC);
|
|
|
|
afn = (xp_awk_afn_t*)pair->val;
|
|
xp_assert (afn != XP_NULL);
|
|
|
|
if (call->nargs > afn->nargs)
|
|
{
|
|
/* TODO: is this correct? what if i want to allow arbitarary numbers of arguments? */
|
|
PANIC (run, XP_AWK_ETOOMANYARGS);
|
|
}
|
|
|
|
return __eval_call (run, nde, afn);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_call (
|
|
xp_awk_run_t* run, xp_awk_nde_t* nde, xp_awk_afn_t* afn)
|
|
{
|
|
xp_awk_nde_call_t* call = (xp_awk_nde_call_t*)nde;
|
|
xp_size_t saved_stack_top;
|
|
xp_size_t nargs, i;
|
|
xp_awk_nde_t* p;
|
|
xp_awk_val_t* v;
|
|
int n;
|
|
|
|
/*
|
|
* ---------------------
|
|
* localn <- stack top
|
|
* ---------------------
|
|
* ....
|
|
* ---------------------
|
|
* local0 local variables are pushed by __run_block
|
|
* =====================
|
|
* argn
|
|
* ---------------------
|
|
* ....
|
|
* ---------------------
|
|
* arg1
|
|
* ---------------------
|
|
* arg0
|
|
* ---------------------
|
|
* nargs
|
|
* ---------------------
|
|
* return value
|
|
* ---------------------
|
|
* previous stack top
|
|
* ---------------------
|
|
* previous stack base <- stack base
|
|
* =====================
|
|
* 0 (nargs) <- stack top
|
|
* ---------------------
|
|
* return value
|
|
* ---------------------
|
|
* previous stack top
|
|
* ---------------------
|
|
* previous stack base <- stack base
|
|
* =====================
|
|
* globaln
|
|
* ---------------------
|
|
* ....
|
|
* ---------------------
|
|
* global0
|
|
* ---------------------
|
|
*/
|
|
|
|
xp_assert (xp_sizeof(void*) >= xp_sizeof(run->stack_top));
|
|
xp_assert (xp_sizeof(void*) >= xp_sizeof(run->stack_base));
|
|
|
|
saved_stack_top = run->stack_top;
|
|
|
|
/*xp_printf (XP_T("setting up function stack frame stack_top = %ld stack_base = %ld\n"), run->stack_top, run->stack_base); */
|
|
if (__raw_push(run,(void*)run->stack_base) == -1)
|
|
{
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
if (__raw_push(run,(void*)saved_stack_top) == -1)
|
|
{
|
|
__raw_pop (run);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
/* secure space for a return value. */
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
/* secure space for nargs */
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
nargs = 0;
|
|
p = call->args;
|
|
while (p != XP_NULL)
|
|
{
|
|
v = __eval_expression(run,p);
|
|
if (v == XP_NULL)
|
|
{
|
|
while (nargs > 0)
|
|
{
|
|
/* TODO: test this portion. */
|
|
--nargs;
|
|
xp_awk_refdownval (run, STACK_ARG(run,nargs));
|
|
__raw_pop (run);
|
|
}
|
|
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
return XP_NULL;
|
|
}
|
|
|
|
if (__raw_push(run,v) == -1)
|
|
{
|
|
/* ugly - v needs to be freed if it doesn't have
|
|
* any reference. but its reference has not been
|
|
* updated for the successful stack push. so it adds
|
|
* up a reference and dereferences it*/
|
|
xp_awk_refupval (v);
|
|
xp_awk_refdownval (run, v);
|
|
|
|
while (nargs > 0)
|
|
{
|
|
/* TODO: test this portion. */
|
|
--nargs;
|
|
xp_awk_refdownval (run, STACK_ARG(run,nargs));
|
|
__raw_pop (run);
|
|
}
|
|
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_awk_refupval (v);
|
|
nargs++;
|
|
p = p->next;
|
|
}
|
|
|
|
xp_assert (nargs == call->nargs);
|
|
|
|
if (afn != XP_NULL)
|
|
{
|
|
/* extra step for normal awk functions */
|
|
|
|
while (nargs < afn->nargs)
|
|
{
|
|
/* push as many nils as the number of missing actual arguments */
|
|
if (__raw_push(run,xp_awk_val_nil) == -1)
|
|
{
|
|
while (nargs > 0)
|
|
{
|
|
/* TODO: test this portion. */
|
|
--nargs;
|
|
xp_awk_refdownval (run, STACK_ARG(run,nargs));
|
|
__raw_pop (run);
|
|
}
|
|
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
__raw_pop (run);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
nargs++;
|
|
}
|
|
}
|
|
|
|
run->stack_base = saved_stack_top;
|
|
STACK_NARGS(run) = (void*)nargs;
|
|
|
|
/*xp_printf (XP_T("running function body\n")); */
|
|
|
|
if (afn != XP_NULL)
|
|
{
|
|
/* normal awk function */
|
|
xp_assert (afn->body->type == XP_AWK_NDE_BLK);
|
|
n = __run_block(run,(xp_awk_nde_blk_t*)afn->body);
|
|
}
|
|
else
|
|
{
|
|
n = 0;
|
|
|
|
/* built-in function */
|
|
xp_assert (call->nargs >= call->what.bfn.min_args &&
|
|
call->nargs <= call->what.bfn.max_args);
|
|
|
|
if (call->what.bfn.handler != XP_NULL)
|
|
{
|
|
n = call->what.bfn.handler (run);
|
|
}
|
|
}
|
|
|
|
/*xp_printf (XP_T("block run complete\n")); */
|
|
|
|
/* refdown args in the run.stack */
|
|
nargs = (xp_size_t)STACK_NARGS(run);
|
|
/*xp_printf (XP_T("block run complete nargs = %d\n"), nargs); */
|
|
for (i = 0; i < nargs; i++)
|
|
{
|
|
xp_awk_refdownval (run, STACK_ARG(run,i));
|
|
}
|
|
/*xp_printf (XP_T("got return value\n")); */
|
|
|
|
/* this is the trick mentioned in __run_return.
|
|
* adjust the reference count of the return value.
|
|
* the value must not be freed even if the reference count
|
|
* is decremented to zero because its reference has been incremented
|
|
* in __run_return regardless of its reference count. */
|
|
v = STACK_RETVAL(run);
|
|
xp_awk_refdownval_nofree (run, v);
|
|
|
|
run->stack_top = (xp_size_t)run->stack[run->stack_base+1];
|
|
run->stack_base = (xp_size_t)run->stack[run->stack_base+0];
|
|
|
|
if (run->exit_level == EXIT_FUNCTION)
|
|
{
|
|
run->exit_level = EXIT_NONE;
|
|
}
|
|
|
|
/*xp_printf (XP_T("returning from function stack_top=%ld, stack_base=%ld\n"), run->stack_top, run->stack_base); */
|
|
return (n == -1)? XP_NULL: v;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_int (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* val;
|
|
val = xp_awk_makeintval (
|
|
run, ((xp_awk_nde_int_t*)nde)->val);
|
|
if (val == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_real (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* val;
|
|
val = xp_awk_makerealval (
|
|
run, ((xp_awk_nde_real_t*)nde)->val);
|
|
if (val == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_str (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_val_t* val;
|
|
val = xp_awk_makestrval (
|
|
((xp_awk_nde_str_t*)nde)->buf,
|
|
((xp_awk_nde_str_t*)nde)->len);
|
|
if (val == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
return val;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_rex (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
/* TODO */
|
|
xp_printf (XP_T("eval_rex not implemented yet...\n"));
|
|
PANIC (run, XP_AWK_EINTERNAL);
|
|
return XP_NULL;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_named (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_pair_t* pair;
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
|
|
pair = xp_awk_map_get(&run->named,tgt->id.name);
|
|
return (pair == XP_NULL)? xp_awk_val_nil: pair->val;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_global (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return STACK_GLOBAL(run,tgt->id.idxa);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_local (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return STACK_LOCAL(run,tgt->id.idxa);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_arg (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return STACK_ARG(run,tgt->id.idxa);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_indexed (
|
|
xp_awk_run_t* run, xp_awk_nde_var_t* nde, xp_awk_val_map_t* map)
|
|
{
|
|
xp_awk_val_t* res;
|
|
xp_awk_pair_t* pair;
|
|
xp_char_t* str;
|
|
|
|
/* TODO: should it be an error? should it return nil? */
|
|
if (map->type != XP_AWK_VAL_MAP)
|
|
{
|
|
PANIC (run, XP_AWK_ENOTINDEXABLE);
|
|
}
|
|
|
|
xp_assert (nde->idx != XP_NULL);
|
|
|
|
str = __idxnde_to_str (run, nde->idx);
|
|
if (str == XP_NULL) return XP_NULL;
|
|
|
|
/* TODO: check this out........ */
|
|
pair = xp_awk_map_get (((xp_awk_val_map_t*)map)->map, str);
|
|
res = (pair == XP_NULL)? xp_awk_val_nil: (xp_awk_val_t*)pair->val;
|
|
|
|
xp_free (str);
|
|
return res;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_namedidx (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
xp_awk_pair_t* pair;
|
|
|
|
pair = xp_awk_map_get (&run->named, tgt->id.name);
|
|
return __eval_indexed (run, tgt,
|
|
(pair == XP_NULL)? xp_awk_val_nil: pair->val);
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_globalidx (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return __eval_indexed (run, tgt, STACK_GLOBAL(run,tgt->id.idxa));
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_localidx (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return __eval_indexed (run, tgt, STACK_LOCAL(run,tgt->id.idxa));
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_argidx (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_var_t* tgt = (xp_awk_nde_var_t*)nde;
|
|
return __eval_indexed (run, tgt, STACK_ARG(run,tgt->id.idxa));
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_pos (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_pos_t* pos = (xp_awk_nde_pos_t*)nde;
|
|
xp_awk_val_t* v;
|
|
xp_long_t lv;
|
|
xp_real_t rv;
|
|
int n;
|
|
|
|
v = __eval_expression (run, pos->val);
|
|
if (v == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (v);
|
|
n = xp_awk_valtonum (v, &lv, &rv);
|
|
xp_awk_refdownval (run, v);
|
|
|
|
if (n == -1) PANIC (run, XP_AWK_EPOSIDX);
|
|
if (n == 1) lv = (xp_long_t)rv;
|
|
|
|
if (lv < 0) PANIC (run, XP_AWK_EPOSIDX);
|
|
if (lv == 0) v = run->inrec.d0;
|
|
else if (lv > 0 && lv <= run->inrec.nflds)
|
|
v = run->inrec.flds[lv-1].val;
|
|
else v = xp_awk_val_zls; /*xp_awk_val_nil;*/
|
|
|
|
return v;
|
|
}
|
|
|
|
static xp_awk_val_t* __eval_getline (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_awk_nde_getline_t* p;
|
|
xp_awk_val_t* v, * res;
|
|
xp_char_t* in = XP_NULL;
|
|
const xp_char_t* dst;
|
|
xp_str_t buf;
|
|
int errnum, n;
|
|
|
|
p = (xp_awk_nde_getline_t*)nde;
|
|
|
|
xp_assert ((p->in_type == XP_AWK_IN_PIPE && p->in != XP_NULL) ||
|
|
(p->in_type == XP_AWK_IN_COPROC && p->in != XP_NULL) ||
|
|
(p->in_type == XP_AWK_IN_FILE && p->in != XP_NULL) ||
|
|
(p->in_type == XP_AWK_IN_CONSOLE && p->in == XP_NULL));
|
|
|
|
if (p->in != XP_NULL)
|
|
{
|
|
v = __eval_expression (run, p->in);
|
|
if (v == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (v);
|
|
in = xp_awk_valtostr (v, &errnum, XP_NULL, XP_NULL);
|
|
if (in == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, v);
|
|
PANIC (run, errnum);
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
}
|
|
|
|
dst = (in == XP_NULL)? XP_T(""): in;
|
|
|
|
/* TODO: optimize the line buffer management */
|
|
if (xp_str_open (&buf, DEF_BUF_CAPA) == XP_NULL)
|
|
{
|
|
if (in != XP_NULL) xp_free (in);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
n = xp_awk_readextio (run, p->in_type, dst, &buf, &errnum);
|
|
if (in != XP_NULL) xp_free (in);
|
|
|
|
if (n < 0 && errnum != XP_AWK_ENOERR)
|
|
{
|
|
xp_str_close (&buf);
|
|
PANIC (run, errnum);
|
|
}
|
|
|
|
if (n > 0)
|
|
{
|
|
if (p->var == XP_NULL)
|
|
{
|
|
/* TODO: set $0 if var is null */
|
|
xp_printf (XP_T("set %s to $0\n"), XP_STR_BUF(&buf));
|
|
xp_str_close (&buf);
|
|
}
|
|
else
|
|
{
|
|
xp_awk_val_t* v;
|
|
|
|
v = xp_awk_makestrval (
|
|
XP_STR_BUF(&buf), XP_STR_LEN(&buf));
|
|
|
|
xp_str_close (&buf);
|
|
|
|
if (v == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
xp_awk_refupval (v);
|
|
|
|
if (__do_assignment(run, p->var, v) == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, v);
|
|
return XP_NULL;
|
|
}
|
|
xp_awk_refdownval (run, v);
|
|
}
|
|
}
|
|
|
|
res = xp_awk_makeintval (run, n);
|
|
if (res == XP_NULL) PANIC (run, XP_AWK_ENOMEM);
|
|
|
|
return res;
|
|
}
|
|
|
|
static int __raw_push (xp_awk_run_t* run, void* val)
|
|
{
|
|
if (run->stack_top >= run->stack_limit)
|
|
{
|
|
void* tmp;
|
|
xp_size_t n;
|
|
|
|
n = run->stack_limit + STACK_INCREMENT;
|
|
tmp = (void**) xp_realloc (run->stack, n * xp_sizeof(void*));
|
|
if (tmp == XP_NULL) return -1;
|
|
|
|
run->stack = tmp;
|
|
run->stack_limit = n;
|
|
}
|
|
|
|
run->stack[run->stack_top++] = val;
|
|
return 0;
|
|
}
|
|
|
|
static void __raw_pop (xp_awk_run_t* run)
|
|
{
|
|
xp_assert (run->stack_top > run->stack_base);
|
|
run->stack_top--;
|
|
}
|
|
|
|
static void __raw_pop_times (xp_awk_run_t* run, xp_size_t times)
|
|
{
|
|
while (times > 0)
|
|
{
|
|
--times;
|
|
__raw_pop (run);
|
|
}
|
|
}
|
|
|
|
static int __read_record (xp_awk_run_t* run)
|
|
{
|
|
xp_ssize_t n;
|
|
xp_char_t c;
|
|
xp_awk_val_t* v;
|
|
|
|
__clear_record (run, xp_false);
|
|
|
|
while (1)
|
|
{
|
|
if (run->inrec.buf_pos >= run->inrec.buf_len)
|
|
{
|
|
if (run->inrec.eof)
|
|
{
|
|
if (XP_STR_LEN(&run->inrec.line) == 0) return 0;
|
|
break;
|
|
}
|
|
|
|
n = run->txtio (XP_AWK_IO_READ, 0, run->txtio_arg,
|
|
run->inrec.buf, xp_countof(run->inrec.buf));
|
|
if (n == -1) PANIC_I (run, XP_AWK_ETXTINDATA);
|
|
if (n == 0)
|
|
{
|
|
run->inrec.eof = xp_true;
|
|
if (XP_STR_LEN(&run->inrec.line) == 0) return 0;
|
|
break;
|
|
}
|
|
|
|
run->inrec.buf_len = n;
|
|
run->inrec.buf_pos = 0;
|
|
}
|
|
|
|
c = run->inrec.buf[run->inrec.buf_pos++];
|
|
|
|
/* TODO: use RS instead of this hard coded value */
|
|
/* any better way to tell line terminating with newlines?
|
|
* line with new line characters removed or retained? */
|
|
if (c == XP_T('\n')) break;
|
|
|
|
if (xp_str_ccat (&run->inrec.line, c) == (xp_size_t)-1)
|
|
{
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
}
|
|
|
|
v = xp_awk_makestrval (
|
|
XP_STR_BUF(&run->inrec.line),
|
|
XP_STR_LEN(&run->inrec.line));
|
|
if (v == XP_NULL)
|
|
{
|
|
__clear_record (run, xp_false);
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
xp_assert (run->inrec.d0 == xp_awk_val_nil);
|
|
/* the record has been cleared in the beginning of the function.
|
|
* so xp_awk_refdownval is not needed over run->inrec.d0 */
|
|
run->inrec.d0 = v;
|
|
xp_awk_refupval (v);
|
|
|
|
if (__split_record (run) == -1)
|
|
{
|
|
__clear_record (run, xp_false);
|
|
return -1;
|
|
}
|
|
|
|
return 1; /* has read a record */
|
|
}
|
|
|
|
static int __split_record (xp_awk_run_t* run)
|
|
{
|
|
/* TODO: support FS and regular expression */
|
|
/* TODO: set NR after split */
|
|
|
|
xp_char_t* p, * tok;
|
|
xp_size_t len, tok_len, nflds;
|
|
xp_awk_val_t* v;
|
|
|
|
/* inrec should be cleared before __split_record is called */
|
|
xp_assert (run->inrec.nflds == 0);
|
|
|
|
/* scan the input record to count the fields */
|
|
p = XP_STR_BUF(&run->inrec.line);
|
|
len = XP_STR_LEN(&run->inrec.line);
|
|
|
|
nflds = 0;
|
|
while (p != XP_NULL)
|
|
{
|
|
p = xp_strxtok (p, len, XP_T(" \t"), &tok, &tok_len);
|
|
|
|
if (nflds == 0 && p == XP_NULL && tok_len == 0)
|
|
{
|
|
/* no fields */
|
|
return 0;
|
|
}
|
|
|
|
xp_assert ((tok != XP_NULL && tok_len > 0) || tok_len == 0);
|
|
|
|
nflds++;
|
|
len = XP_STR_LEN(&run->inrec.line) -
|
|
(p - XP_STR_BUF(&run->inrec.line));
|
|
}
|
|
|
|
/* allocate space */
|
|
if (nflds > run->inrec.maxflds)
|
|
{
|
|
void* tmp = xp_malloc (
|
|
xp_sizeof(*run->inrec.flds) * nflds);
|
|
if (tmp == XP_NULL) PANIC_I (run, XP_AWK_ENOMEM);
|
|
|
|
if (run->inrec.flds != NULL)
|
|
xp_free (run->inrec.flds);
|
|
run->inrec.flds = tmp;
|
|
run->inrec.maxflds = nflds;
|
|
}
|
|
|
|
/* scan again and split it */
|
|
p = XP_STR_BUF(&run->inrec.line);
|
|
len = XP_STR_LEN(&run->inrec.line);
|
|
|
|
while (p != XP_NULL)
|
|
{
|
|
p = xp_strxtok (p, len, XP_T(" \t"), &tok, &tok_len);
|
|
|
|
xp_assert ((tok != XP_NULL && tok_len > 0) || tok_len == 0);
|
|
|
|
run->inrec.flds[run->inrec.nflds].ptr = tok;
|
|
run->inrec.flds[run->inrec.nflds].len = tok_len;
|
|
run->inrec.flds[run->inrec.nflds].val =
|
|
xp_awk_makestrval (tok, tok_len);
|
|
|
|
if (run->inrec.flds[run->inrec.nflds].val == XP_NULL)
|
|
PANIC_I (run, XP_AWK_ENOMEM);
|
|
|
|
xp_awk_refupval (run->inrec.flds[run->inrec.nflds].val);
|
|
run->inrec.nflds++;
|
|
|
|
len = XP_STR_LEN(&run->inrec.line) -
|
|
(p - XP_STR_BUF(&run->inrec.line));
|
|
}
|
|
|
|
/* set the number of fields */
|
|
v = xp_awk_makeintval (run, (xp_long_t)nflds);
|
|
if (v == XP_NULL) PANIC_I (run, XP_AWK_ENOMEM);
|
|
xp_awk_setglobal (run, XP_AWK_GLOBAL_NF, v);
|
|
|
|
xp_assert (nflds == run->inrec.nflds);
|
|
return 0;
|
|
}
|
|
|
|
static void __clear_record (xp_awk_run_t* run, xp_bool_t noline)
|
|
{
|
|
xp_size_t i;
|
|
|
|
xp_awk_refdownval (run, run->inrec.d0);
|
|
run->inrec.d0 = xp_awk_val_nil;
|
|
|
|
if (run->inrec.nflds > 0)
|
|
{
|
|
xp_assert (run->inrec.flds != XP_NULL);
|
|
|
|
for (i = 0; i < run->inrec.nflds; i++)
|
|
{
|
|
xp_assert (run->inrec.flds[i].val != XP_NULL);
|
|
xp_awk_refdownval (run, run->inrec.flds[i].val);
|
|
}
|
|
run->inrec.nflds = 0;
|
|
xp_awk_setglobal (run, XP_AWK_GLOBAL_NF, xp_awk_val_zero);
|
|
}
|
|
|
|
xp_assert (run->inrec.nflds == 0);
|
|
if (!noline) xp_str_clear (&run->inrec.line);
|
|
}
|
|
|
|
static int __recomp_record_fields (xp_awk_run_t* run,
|
|
xp_size_t lv, xp_char_t* str, xp_size_t len, int* errnum)
|
|
{
|
|
xp_awk_val_t* v;
|
|
xp_char_t* ofsp = XP_NULL, * ofs;
|
|
xp_size_t ofs_len;
|
|
xp_size_t max, i, nflds;
|
|
|
|
xp_assert (lv > 0);
|
|
max = (lv > run->inrec.nflds)? lv: run->inrec.nflds;
|
|
|
|
nflds = run->inrec.nflds;
|
|
if (max > run->inrec.maxflds)
|
|
{
|
|
void* tmp = xp_realloc (
|
|
run->inrec.flds, xp_sizeof(*run->inrec.flds) * max);
|
|
if (tmp == XP_NULL)
|
|
{
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
run->inrec.flds = tmp;
|
|
run->inrec.maxflds = max;
|
|
}
|
|
|
|
lv = lv - 1; /* adjust the value to 0-based index */
|
|
|
|
xp_str_clear (&run->inrec.line);
|
|
|
|
if (max > 1)
|
|
{
|
|
v = STACK_GLOBAL(run, XP_AWK_GLOBAL_OFS);
|
|
if (v != xp_awk_val_nil)
|
|
{
|
|
ofsp = xp_awk_valtostr (v, errnum, XP_NULL, &ofs_len);
|
|
if (ofsp == XP_NULL) return -1;
|
|
|
|
ofs = ofsp;
|
|
}
|
|
else
|
|
{
|
|
/* OFS not set */
|
|
ofs = XP_T(" ");
|
|
ofs_len = 1;
|
|
}
|
|
|
|
}
|
|
|
|
for (i = 0; i < max; i++)
|
|
{
|
|
if (i > 0)
|
|
{
|
|
if (xp_str_ncat (
|
|
&run->inrec.line, ofs, ofs_len) == (xp_size_t)-1)
|
|
{
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (i == lv)
|
|
{
|
|
xp_awk_val_t* tmp;
|
|
|
|
run->inrec.flds[i].ptr =
|
|
XP_STR_BUF(&run->inrec.line) +
|
|
XP_STR_LEN(&run->inrec.line);
|
|
run->inrec.flds[i].len = len;
|
|
|
|
if (xp_str_ncat (
|
|
&run->inrec.line, str, len) == (xp_size_t)-1)
|
|
{
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
tmp = xp_awk_makestrval (str,len);
|
|
if (tmp == XP_NULL)
|
|
{
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
if (i < nflds)
|
|
xp_awk_refdownval (run, run->inrec.flds[i].val);
|
|
else run->inrec.nflds++;
|
|
run->inrec.flds[i].val = tmp;
|
|
xp_awk_refupval (tmp);
|
|
}
|
|
else if (i >= nflds)
|
|
{
|
|
run->inrec.flds[i].ptr =
|
|
XP_STR_BUF(&run->inrec.line) +
|
|
XP_STR_LEN(&run->inrec.line);
|
|
run->inrec.flds[i].len = 0;
|
|
|
|
if (xp_str_cat (
|
|
&run->inrec.line, XP_T("")) == (xp_size_t)-1)
|
|
{
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
/* xp_awk_refdownval should not be called over
|
|
* run->inrec.flds[i].val as it is not initialized
|
|
* to any valid values */
|
|
/*xp_awk_refdownval (run, run->inrec.flds[i].val);*/
|
|
run->inrec.flds[i].val = xp_awk_val_zls;
|
|
xp_awk_refupval (xp_awk_val_zls);
|
|
run->inrec.nflds++;
|
|
}
|
|
else
|
|
{
|
|
xp_awk_val_str_t* tmp;
|
|
|
|
tmp = (xp_awk_val_str_t*)run->inrec.flds[i].val;
|
|
|
|
run->inrec.flds[i].ptr =
|
|
XP_STR_BUF(&run->inrec.line) +
|
|
XP_STR_LEN(&run->inrec.line);
|
|
run->inrec.flds[i].len = tmp->len;
|
|
|
|
if (xp_str_ncat (&run->inrec.line,
|
|
tmp->buf, tmp->len) == (xp_size_t)-1)
|
|
{
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ofsp != XP_NULL) xp_free (ofsp);
|
|
|
|
v = STACK_GLOBAL(run, XP_AWK_GLOBAL_NF);
|
|
xp_assert (v->type == XP_AWK_VAL_INT);
|
|
if (((xp_awk_val_int_t*)v)->val != max)
|
|
{
|
|
v = xp_awk_makeintval (run, (xp_long_t)max);
|
|
if (v == XP_NULL)
|
|
{
|
|
*errnum = XP_AWK_ENOMEM;
|
|
return -1;
|
|
}
|
|
|
|
xp_awk_setglobal (run, XP_AWK_GLOBAL_NF, v);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static xp_char_t* __idxnde_to_str (xp_awk_run_t* run, xp_awk_nde_t* nde)
|
|
{
|
|
xp_char_t* str;
|
|
xp_awk_val_t* idx;
|
|
int errnum;
|
|
|
|
xp_assert (nde != XP_NULL);
|
|
|
|
if (nde->next == XP_NULL)
|
|
{
|
|
/* single node index */
|
|
idx = __eval_expression (run, nde);
|
|
if (idx == XP_NULL) return XP_NULL;
|
|
|
|
xp_awk_refupval (idx);
|
|
|
|
str = xp_awk_valtostr (idx, &errnum, XP_NULL, XP_NULL);
|
|
if (str == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, idx);
|
|
PANIC (run, errnum);
|
|
}
|
|
|
|
xp_awk_refdownval (run, idx);
|
|
}
|
|
else
|
|
{
|
|
/* multidimensional index */
|
|
xp_str_t idxstr;
|
|
|
|
if (xp_str_open (&idxstr, DEF_BUF_CAPA) == XP_NULL)
|
|
{
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
while (nde != XP_NULL)
|
|
{
|
|
idx = __eval_expression (run, nde);
|
|
if (idx == XP_NULL)
|
|
{
|
|
xp_str_close (&idxstr);
|
|
return XP_NULL;
|
|
}
|
|
|
|
xp_awk_refupval (idx);
|
|
|
|
/* TODO: configurable index seperator */
|
|
if (XP_STR_LEN(&idxstr) > 0 &&
|
|
xp_str_cat (&idxstr, XP_T(",")) == (xp_size_t)-1)
|
|
{
|
|
xp_awk_refdownval (run, idx);
|
|
xp_str_close (&idxstr);
|
|
PANIC (run, XP_AWK_ENOMEM);
|
|
}
|
|
|
|
if (xp_awk_valtostr (idx,
|
|
&errnum, &idxstr, XP_NULL) == XP_NULL)
|
|
{
|
|
xp_awk_refdownval (run, idx);
|
|
xp_str_close (&idxstr);
|
|
PANIC (run, errnum);
|
|
}
|
|
|
|
xp_awk_refdownval (run, idx);
|
|
nde = nde->next;
|
|
}
|
|
|
|
str = XP_STR_BUF(&idxstr);
|
|
xp_str_forfeit (&idxstr);
|
|
}
|
|
|
|
return str;
|
|
}
|
|
|