libbb: prevent resursion on malloc failure
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
This commit is contained in:
parent
6d48d3ee2a
commit
e2944af43e
@ -15,7 +15,7 @@ const char *msg_eol = "\n";
|
||||
|
||||
void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr)
|
||||
{
|
||||
char *msg;
|
||||
char *msg, *msg1;
|
||||
int applet_len, strerr_len, msgeol_len, used;
|
||||
|
||||
if (!logmode)
|
||||
@ -36,27 +36,36 @@ void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr)
|
||||
applet_len = strlen(applet_name) + 2; /* "applet: " */
|
||||
strerr_len = strerr ? strlen(strerr) : 0;
|
||||
msgeol_len = strlen(msg_eol);
|
||||
/* can't use xrealloc: it calls error_msg on failure,
|
||||
* that may result in a recursion */
|
||||
/* +3 is for ": " before strerr and for terminating NUL */
|
||||
msg = xrealloc(msg, applet_len + used + strerr_len + msgeol_len + 3);
|
||||
/* TODO: maybe use writev instead of memmoving? Need full_writev? */
|
||||
memmove(msg + applet_len, msg, used);
|
||||
used += applet_len;
|
||||
strcpy(msg, applet_name);
|
||||
msg[applet_len - 2] = ':';
|
||||
msg[applet_len - 1] = ' ';
|
||||
if (strerr) {
|
||||
if (s[0]) { /* not perror_nomsg? */
|
||||
msg[used++] = ':';
|
||||
msg[used++] = ' ';
|
||||
msg1 = realloc(msg, applet_len + used + strerr_len + msgeol_len + 3);
|
||||
if (!msg1) {
|
||||
msg[used++] = '\n'; /* overwrites NUL */
|
||||
applet_len = 0;
|
||||
} else {
|
||||
msg = msg1;
|
||||
/* TODO: maybe use writev instead of memmoving? Need full_writev? */
|
||||
memmove(msg + applet_len, msg, used);
|
||||
used += applet_len;
|
||||
strcpy(msg, applet_name);
|
||||
msg[applet_len - 2] = ':';
|
||||
msg[applet_len - 1] = ' ';
|
||||
if (strerr) {
|
||||
if (s[0]) { /* not perror_nomsg? */
|
||||
msg[used++] = ':';
|
||||
msg[used++] = ' ';
|
||||
}
|
||||
strcpy(&msg[used], strerr);
|
||||
used += strerr_len;
|
||||
}
|
||||
strcpy(&msg[used], strerr);
|
||||
used += strerr_len;
|
||||
strcpy(&msg[used], msg_eol);
|
||||
used += msgeol_len;
|
||||
}
|
||||
strcpy(&msg[used], msg_eol);
|
||||
|
||||
if (logmode & LOGMODE_STDIO) {
|
||||
fflush(stdout);
|
||||
full_write(STDERR_FILENO, msg, used + msgeol_len);
|
||||
full_write(STDERR_FILENO, msg, used);
|
||||
}
|
||||
if (logmode & LOGMODE_SYSLOG) {
|
||||
syslog(LOG_ERR, "%s", msg + applet_len);
|
||||
|
Loading…
Reference in New Issue
Block a user