Xinqi Bao's Git

check daemon() return value
[slstatus.git] / slstatus.c
index e7c1c3f..0a0f743 100644 (file)
@@ -400,17 +400,19 @@ static char *
 run_command(const char *cmd)
 {
        FILE *fp;
-       char buf[64] = "n/a";
+       char buf[1024] = "n/a";
 
        fp = popen(cmd, "r");
        if (fp == NULL) {
                warn("Failed to get command output for %s", cmd);
                return smprintf(UNKNOWN_STR);
        }
-       fgets(buf, sizeof(buf), fp);
-       buf[sizeof(buf)-1] = '\0';
+       fgets(buf, sizeof(buf)-1, fp);
        pclose(fp);
 
+       buf[strlen(buf)] = '\0';
+       strtok(buf, "\n");
+
        return smprintf("%s", buf);
 }
 
@@ -619,8 +621,8 @@ main(int argc, char *argv[])
        if (dflag && oflag) {
                usage();
        }
-       if (dflag) {
-               (void)daemon(1, 1);
+       if (dflag && daemon(1, 1) < 0) {
+               err(1, "daemon");
        }
 
        memset(&act, 0, sizeof(act));