emo/busybox
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

awk: unbreak "printf('%c') can output NUL" testcase

Browse Source 
function                                             old     new   delta
awk_printf                                           546     593     +47

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
This commit is contained in:
Denys Vlasenko 2021-07-11 12:25:33 +02:00
parent 3d57a84907
commit 4ef8841b21
1 changed files with 26 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
editors/awk.c
View File

@ -2309,11 +2309,11 @@ static int awk_getline(rstream *rsm, var *v)
#if !ENABLE_FEATURE_AWK_GNU_EXTENSIONS
# define awk_printf(a, b) awk_printf(a)
#endif
static char *awk_printf(node *n, int *len)
static char *awk_printf(node *n, size_t *len)
{
char *b;
char *fmt, *f;
int i;
size_t i;
//tmpvar = nvalloc(1);
#define TMPVAR (&G.awk_printf__tmpvar)
@ -2333,6 +2333,7 @@ static char *awk_printf(node *n, int *len)
char c;
char sv;
var *arg;
size_t slen;
s = f;
while (*f && (*f != '%' || *++f == '%'))
@ -2347,6 +2348,7 @@ static char *awk_printf(node *n, int *len)
/* Tail of fmt with no percent chars,
* or "....%" (percent seen, but no format specifier char found)
*/
slen = strlen(s);
goto tail;
}
sv = *++f;
@ -2357,9 +2359,13 @@ static char *awk_printf(node *n, int *len)
* printf "%99999s", "BOOM"
*/
if (c == 'c') {
s = xasprintf(s, is_numeric(arg) ?
(char)getvar_i(arg) : *getvar_s(arg));
} else if (c == 's') {
c = is_numeric(arg) ? getvar_i(arg) : *getvar_s(arg);
s = xasprintf(s, c);
/* + 1 if c == NUL: handle printf "%c" 0 case
* (and printf "%22c" 0 etc, but still fails for e.g. printf "%-22c" 0) */
slen = strlen(s) + (c == '\0');
} else {
if (c == 's') {
s = xasprintf(s, getvar_s(arg));
} else {
double d = getvar_i(arg);
@ -2372,16 +2378,19 @@ static char *awk_printf(node *n, int *len)
syntax_error(EMSG_INV_FMT);
}
}
slen = strlen(s);
}
*f = sv;
if (i == 0) {
b = s;
i = strlen(b);
i = slen;
continue;
}
tail:
b = xrealloc(b, i + strlen(s) + 1);
i = stpcpy(b + i, s) - b;
b = xrealloc(b, i + slen + 1);
strcpy(b + i, s);
i += slen;
if (!c) /* tail? */
break;
free(s);
@ -2926,7 +2935,6 @@ static var *evaluate(node *op, var *res)
debug_printf_eval("PRINTF\n");
{
FILE *F = stdout;
IF_FEATURE_AWK_GNU_EXTENSIONS(int len;)
if (op->r.n) {
rstream *rsm = newfile(R.s);
@ -2966,6 +2974,7 @@ static var *evaluate(node *op, var *res)
}
fputs(getvar_s(intvar[ORS]), F);
} else { /* PRINTF */
IF_FEATURE_AWK_GNU_EXTENSIONS(size_t len;)
char *s = awk_printf(op1, &len);
#if ENABLE_FEATURE_AWK_GNU_EXTENSIONS
fwrite(s, len, 1, F);