hyung-hwan/hawk
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed a potential invalid memory access issue related to function argument spec handling
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
hyung-hwan 2024-05-01 13:24:08 +09:00
parent 3105f52a71
commit 4fbbe049ba
3 changed files with 28 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/hawk.h
View File

@ -481,7 +481,8 @@ struct hawk_fun_t
{ {
hawk_oocs_t name; hawk_oocs_t name;
hawk_oow_t nargs; hawk_oow_t nargs;
hawk_ooch_t* argspec; /* similar to the argument spec of hawk_fnc_arg_t. supports v & r only. */ hawk_ooch_t* argspec; /* similar to the argument spec of hawk_fnc_arg_t. supports v & r only. neither R nor x is set internally. */
hawk_oow_t argspeclen; /* the length of argspec. it can be different from nargs if there are call-by-value parameters after the last call-by-reference parameter or variadic arguments are supported */
hawk_nde_t* body; hawk_nde_t* body;
}; };
typedef struct hawk_fun_t hawk_fun_t; typedef struct hawk_fun_t hawk_fun_t;

14
lib/parse.c
View File

@ -1302,6 +1302,7 @@ static hawk_nde_t* parse_function (hawk_t* hawk)
hawk_fun_t* fun = HAWK_NULL; hawk_fun_t* fun = HAWK_NULL;
hawk_ooch_t* argspec = HAWK_NULL; hawk_ooch_t* argspec = HAWK_NULL;
hawk_oow_t argspeccapa = 0; hawk_oow_t argspeccapa = 0;
hawk_oow_t argspeclen;
hawk_oow_t nargs, g; hawk_oow_t nargs, g;
hawk_htb_pair_t* pair; hawk_htb_pair_t* pair;
hawk_loc_t xloc; hawk_loc_t xloc;
@ -1384,7 +1385,7 @@ static hawk_nde_t* parse_function (hawk_t* hawk)
{ {
hawk_oow_t i, newcapa = HAWK_ALIGN_POW2(nargs + 2, 64); hawk_oow_t i, newcapa = HAWK_ALIGN_POW2(nargs + 2, 64);
argspec = hawk_reallocmem(hawk, argspec, newcapa * HAWK_SIZEOF(*argspec)); argspec = hawk_reallocmem(hawk, argspec, newcapa * HAWK_SIZEOF(*argspec));
if (!argspec) goto oops; if (HAWK_UNLIKELY(!argspec)) goto oops;
for (i = argspeccapa; i < newcapa; i++) argspec[i] = HAWK_T(' '); for (i = argspeccapa; i < newcapa; i++) argspec[i] = HAWK_T(' ');
argspeccapa = newcapa; argspeccapa = newcapa;
} }
@ -1449,7 +1450,13 @@ static hawk_nde_t* parse_function (hawk_t* hawk)
while (MATCH(hawk,TOK_NEWLINE)); while (MATCH(hawk,TOK_NEWLINE));
} }
if (argspec) argspec[nargs + 1] = '\0'; if (argspec)
{
/* nargs is the number taken before the current argument word was added to the parse.params array.
* so the actual number of arguments is nargs + 1 */
argspeclen = nargs + 1;
argspec[argspeclen] = '\0';
}
if (get_token(hawk) <= -1) goto oops; if (get_token(hawk) <= -1) goto oops;
} }
@ -1502,6 +1509,7 @@ static hawk_nde_t* parse_function (hawk_t* hawk)
fun->name.len = 0; fun->name.len = 0;
fun->nargs = nargs; fun->nargs = nargs;
fun->argspec = argspec; fun->argspec = argspec;
fun->argspeclen = argspeclen;
fun->body = body; fun->body = body;
pair = hawk_htb_insert(hawk->tree.funs, name.ptr, name.len, fun, 0); pair = hawk_htb_insert(hawk->tree.funs, name.ptr, name.len, fun, 0);
@ -7236,7 +7244,7 @@ static hawk_htb_walk_t deparse_func (hawk_htb_t* map, hawk_htb_pair_t* pair, voi
for (i = 0; i < fun->nargs; ) for (i = 0; i < fun->nargs; )
{ {
if (fun->argspec && fun->argspec[i] == 'r') PUT_S (df, HAWK_T("&")); if (fun->argspec && i < fun->argspeclen && fun->argspec[i] == 'r') PUT_S (df, HAWK_T("&"));
n = hawk_int_to_oocstr(i++, 10, HAWK_T("__p"), df->tmp, df->tmp_len); n = hawk_int_to_oocstr(i++, 10, HAWK_T("__p"), df->tmp, df->tmp_len);
HAWK_ASSERT (n != (hawk_oow_t)-1); HAWK_ASSERT (n != (hawk_oow_t)-1);
PUT_SX (df, df->tmp, n); PUT_SX (df, df->tmp, n);

20
lib/run.c
View File

@ -46,6 +46,7 @@ struct pafv_t
hawk_val_t** args; hawk_val_t** args;
hawk_oow_t nargs; hawk_oow_t nargs;
const hawk_ooch_t* argspec; const hawk_ooch_t* argspec;
hawk_oow_t argspeclen;
}; };
#define DEFAULT_CONVFMT HAWK_T("%.6g") #define DEFAULT_CONVFMT HAWK_T("%.6g")
@ -1758,6 +1759,7 @@ hawk_val_t* hawk_rtx_callfun (hawk_rtx_t* rtx, hawk_fun_t* fun, hawk_val_t* args
pafv.args = args; pafv.args = args;
pafv.nargs = nargs; pafv.nargs = nargs;
pafv.argspec = fun->argspec; pafv.argspec = fun->argspec;
pafv.argspeclen = fun->argspeclen;
if (HAWK_UNLIKELY(rtx->exit_level >= EXIT_GLOBAL)) if (HAWK_UNLIKELY(rtx->exit_level >= EXIT_GLOBAL))
{ {
@ -6633,6 +6635,7 @@ static hawk_val_t* eval_fncall_fnc (hawk_rtx_t* rtx, hawk_nde_t* nde)
static HAWK_INLINE hawk_val_t* eval_fncall_fun (hawk_rtx_t* rtx, hawk_nde_t* nde) static HAWK_INLINE hawk_val_t* eval_fncall_fun (hawk_rtx_t* rtx, hawk_nde_t* nde)
{ {
/* user-defined function */
hawk_nde_fncall_t* call = (hawk_nde_fncall_t*)nde; hawk_nde_fncall_t* call = (hawk_nde_fncall_t*)nde;
hawk_fun_t* fun; hawk_fun_t* fun;
hawk_htb_pair_t* pair; hawk_htb_pair_t* pair;
@ -6863,7 +6866,7 @@ hawk_val_t* hawk_rtx_evalcall (
/* even if fun->argspec exists, call->nargs may still be 0. so i test if call->nargs > 0. /* even if fun->argspec exists, call->nargs may still be 0. so i test if call->nargs > 0.
* function x(a1, &a2) {} * function x(a1, &a2) {}
* BEGIN { x(); } * BEGIN { x(); }
* all parameters are nils in this case. nargs and fun->nargs are 2 but call->nargs is 0. * all parameters are @nils in this case. nargs and fun->nargs are 2 but call->nargs is 0.
*/ */
if (call->args) if (call->args)
@ -6874,8 +6877,11 @@ hawk_val_t* hawk_rtx_evalcall (
for (; i < call->nargs; i++) for (; i < call->nargs; i++)
{ {
if (n >= 0 && (fun->argspec[i] == 'r' || fun->argspec[i] == 'R')) if (n >= 0 && i < fun->argspeclen && (fun->argspec[i] == 'r' || fun->argspec[i] == 'R'))
{ {
/* if the function call is successful, update the call-by-reference arguments
* with hawk_rtx_setrefval() */
hawk_val_t** ref; hawk_val_t** ref;
hawk_val_ref_t refv; hawk_val_ref_t refv;
hawk_val_t* av; hawk_val_t* av;
@ -6900,7 +6906,7 @@ hawk_val_t* hawk_rtx_evalcall (
* the argument node information stored is relative to the previous stack frame. * the argument node information stored is relative to the previous stack frame.
* i revert rtx->stack_base to the previous stack frame base before calling * i revert rtx->stack_base to the previous stack frame base before calling
* get_reference() and restors it back to the current base. this tactic * get_reference() and restors it back to the current base. this tactic
* is very ugly because the assumptions for this is depecallnt on get_reference() * is very ugly because the assumption for this is dependent on get_reference()
* implementation */ * implementation */
rtx->stack_base = prev_stack_base; /* UGLY */ rtx->stack_base = prev_stack_base; /* UGLY */
r = get_reference(rtx, p, &ref); r = get_reference(rtx, p, &ref);
@ -6983,7 +6989,7 @@ hawk_val_t* hawk_rtx_evalcall (
if (hawk_rtx_geterrnum(rtx) == HAWK_ENOERR && errhandler != HAWK_NULL) if (hawk_rtx_geterrnum(rtx) == HAWK_ENOERR && errhandler != HAWK_NULL)
{ {
/* errhandler is passed only when hawk_rtx_evalcall() is /* errhandler is passed only when hawk_rtx_evalcall() is
* invoked from hawk_rtx_call(). Ucallr this * invoked from hawk_rtx_call(). Under this
* circumstance, this stack frame is the first * circumstance, this stack frame is the first
* activated and the stack base is the first element * activated and the stack base is the first element
* after the global variables. so HAWK_RTX_STACK_RETVAL(rtx) * after the global variables. so HAWK_RTX_STACK_RETVAL(rtx)
@ -7115,7 +7121,9 @@ static hawk_oow_t push_arg_from_nde (hawk_rtx_t* rtx, hawk_nde_fncall_t* call, v
{ {
hawk_ooch_t spec; hawk_ooch_t spec;
/* if not sufficient number of spec characters given, take the last value and use it */ /* if not sufficient number of spec characters given, take the last value and use it.
* as said above, arg_spec is provided for the implicit functions only.
* so the logic taking the last value at run-time is elsewhere (e.g hawk_rtx_evalcall()). */
spec = (spec_len <= 0)? '\0': arg_spec[((nargs < spec_len)? nargs: spec_len - 1)]; spec = (spec_len <= 0)? '\0': arg_spec[((nargs < spec_len)? nargs: spec_len - 1)];
switch (spec) switch (spec)
@ -7947,7 +7955,7 @@ read_again:
} }
#if defined(DEBUG_RUN) #if defined(DEBUG_RUN)
hawk_logfbmt (hawk_rtx_gethawk(rtx), "record len = %d str=[%.*js]\n", (int)HAWK_OOECS_LEN(buf), HAWK_OOECS_LEN(buf), HAWK_OOECS_PTR(buf)); hawk_logbfmt (hawk_rtx_gethawk(rtx), "record len = %d str=[%.*js]\n", (int)HAWK_OOECS_LEN(buf), HAWK_OOECS_LEN(buf), HAWK_OOECS_PTR(buf));
#endif #endif
if (n == 0) if (n == 0)
{ {