fixed strcat null possibility in main()

[slstatus.git] / slstatus.c

diff --git a/slstatus.c b/slstatus.c
index d993eca..b74faa6 100644 (file)
--- a/slstatus.c
+++ b/slstatus.c
@@ -689,6 +689,10 @@ main(void)
             argument = args[i];
             char *res = argument.func(argument.args);
             char *element = smprintf(argument.format, res);
+            if (element == NULL) {
+                element = smprintf("n/a");
+                fprintf(stderr, "Failed to format output.");
+            }
             strcat(status_string, element);
             free(res);
             free(element);