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;