X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/83e84bebb6accdf0971bfa09a519d8cee37b0237..046e273a7eb270f4b275c02546e4547d8cfc2a82:/slstatus.c diff --git a/slstatus.c b/slstatus.c index 42e5751..214f667 100644 --- a/slstatus.c +++ b/slstatus.c @@ -40,21 +40,10 @@ char * smprintf(char *fmt, ...) { va_list fmtargs; - char *ret; - int len; - - va_start(fmtargs, fmt); - len = vsnprintf(NULL, 0, fmt, fmtargs); - va_end(fmtargs); - - ret = malloc(++len); - if (ret == NULL) { - fprintf(stderr, "Malloc error."); - exit(1); - } - + char *ret = NULL; va_start(fmtargs, fmt); - vsnprintf(ret, len, fmt, fmtargs); + if (vasprintf(&ret, fmt, fmtargs) < 0) + return NULL; va_end(fmtargs); return ret;