X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/914440b4fc3c5ecb7d9eedf4da6f2a1c7ec2e7a9..96f3a8a54eeb3b2294ed953dad8b15349f3e2703:/slstatus.c

diff --git a/slstatus.c b/slstatus.c
index 89a3268..f7c4503 100644
--- a/slstatus.c
+++ b/slstatus.c
@@ -1,6 +1,4 @@
 /* See LICENSE file for copyright and license details. */
-#include <err.h>
-#include <errno.h>
 #include <locale.h>
 #include <signal.h>
 #include <stdio.h>
@@ -87,8 +85,10 @@ main(int argc, char *argv[])
 
 		status[0] = '\0';
 		for (i = len = 0; i < LEN(args); i++) {
+			const char * res = args[i].func(args[i].args);
+			res = (res == NULL) ? unknown_str : res;
 			len += snprintf(status + len, sizeof(status) - len,
-			                args[i].fmt, args[i].func(args[i].args));
+			                args[i].fmt, res);
 
 			if (len >= sizeof(status)) {
 				status[sizeof(status) - 1] = '\0';