fixed a null-termination issue in hawk_querymodulewithname()

This commit is contained in:
hyung-hwan 2022-06-03 05:20:32 +00:00
parent 23bd2fdd31
commit 13f314fba3
2 changed files with 28 additions and 7 deletions

View File

@ -7457,7 +7457,8 @@ static hawk_mod_t* query_module (hawk_t* hawk, const hawk_oocs_t segs[], int nse
HAWK_MEMSET (&spec, 0, HAWK_SIZEOF(spec));
spec.prefix = (hawk->opt.mod[1].len > 0)? hawk->opt.mod[1].ptr: HAWK_T(HAWK_DEFAULT_MODPREFIX);
spec.postfix = (hawk->opt.mod[2].len > 0)? hawk->opt.mod[2].ptr: HAWK_T(HAWK_DEFAULT_MODPOSTFIX);
spec.name = segs[0].ptr;
spec.name = segs[0].ptr; /* the caller must ensure that this segment is null-terminated */
if (!hawk->prm.modopen || !hawk->prm.modgetsym || !hawk->prm.modclose)
{
hawk_seterrfmt (hawk, HAWK_NULL, HAWK_EINVAL, HAWK_T("module callbacks not set properly"));
@ -7492,7 +7493,8 @@ static hawk_mod_t* query_module (hawk_t* hawk, const hawk_oocs_t segs[], int nse
open_fail:
bem = hawk_backuperrmsg(hawk);
hawk_seterrfmt (hawk, HAWK_NULL, HAWK_ENOENT, HAWK_T("'%js%js%js' for module '%js' not found - %js"),
(spec.prefix? spec.prefix: HAWK_T("")), spec.name, (spec.postfix? spec.postfix: HAWK_T("")), spec.name, bem);
(spec.prefix? spec.prefix: HAWK_T("")), spec.name, (spec.postfix? spec.postfix: HAWK_T("")),
spec.name, bem);
return HAWK_NULL;
}
@ -7557,6 +7559,9 @@ hawk_mod_t* hawk_querymodulewithname (hawk_t* hawk, const hawk_oocs_t* name, haw
{
const hawk_ooch_t* dc;
hawk_oocs_t segs[2];
hawk_mod_t* mod;
hawk_ooch_t tmp;
/*TOOD: non-module builtin function? fnc? */
dc = hawk_find_oochars_in_oochars(name->ptr, name->len, HAWK_T("::"), 2, 0);
@ -7566,11 +7571,27 @@ hawk_mod_t* hawk_querymodulewithname (hawk_t* hawk, const hawk_oocs_t* name, haw
return HAWK_NULL;
}
segs[0].ptr = name->ptr;
#if 1
segs[0].len = dc - name->ptr;
segs[0].ptr = hawk_dupoochars(hawk, name->ptr, segs[0].len);
if (!segs[0].ptr) return HAWK_NULL;
#else
segs[0].len = dc - name->ptr;
segs[0].ptr = name->ptr;
tmp = name->ptr[segs[0].len];
name->ptr[segs[0].len] = '\0';
#endif
segs[1].ptr = segs[0].ptr + segs[0].len + 2;
segs[1].len = name->len - segs[0].len - 2;
segs[1].ptr = name->ptr + segs[0].len + 2;
return query_module(hawk, segs, 2, sym);
mod = query_module(hawk, segs, 2, sym);
#if 1
hawk_freemem (hawk, segs[0].ptr);
#else
name->ptr[segs[0].len] = tmp;
#endif
return mod;
}

View File

@ -24,7 +24,7 @@ function run_test_001 ()
}
else
{
tap_skip (sprintf("sed::str_top_str() is unavailable - %s[%d]", @SCRIPTNAME, @SCRIPTLINE));
tap_skip (sprintf("sed::str_to_str() is unavailable - %s[%d]", @SCRIPTNAME, @SCRIPTLINE));
}
}