chiark / gitweb /
journal: fix memleak, call set_free before return
[elogind.git] / src / journal / coredumpctl.c
index 5c442ffe9b1d4ec5cadb4b5ba04698bce9319f83..311c18b6ab8d362e48868796d7ed29ab466ae3ef 100644 (file)
@@ -22,6 +22,8 @@
 #include <stdio.h>
 #include <string.h>
 #include <getopt.h>
+#include <fcntl.h>
+#include <unistd.h>
 
 #include <systemd/sd-journal.h>
 
@@ -36,12 +38,13 @@ static enum {
         ACTION_NONE,
         ACTION_LIST,
         ACTION_DUMP,
+        ACTION_GDB,
 } arg_action = ACTION_LIST;
 
-Set *matches;
-FILE* output;
+static Set *matches = NULL;
+static FILE* output = NULL;
 
-int arg_no_pager;
+static int arg_no_pager = false;
 
 static Set *new_matches(void) {
         Set *set;
@@ -57,7 +60,7 @@ static Set *new_matches(void) {
         tmp = strdup("MESSAGE_ID=fc2e22bc6ee647b6b90729ab34a250b1");
         if (!tmp) {
                 log_oom();
-                set_clear_free(set);
+                set_free(set);
                 return NULL;
         }
 
@@ -65,7 +68,7 @@ static Set *new_matches(void) {
         if (r < 0) {
                 log_error("failed to add to set: %s", strerror(-r));
                 free(tmp);
-                set_clear_free(set);
+                set_free(set);
                 return NULL;
         }
 
@@ -191,6 +194,8 @@ static int parse_argv(int argc, char *argv[]) {
                         arg_action = ACTION_LIST;
                 else if (streq(cmd, "dump"))
                         arg_action = ACTION_DUMP;
+                else if (streq(cmd, "gdb"))
+                        arg_action = ACTION_GDB;
                 else {
                         log_error("Unknown action '%s'", cmd);
                         return -EINVAL;
@@ -207,19 +212,23 @@ static int parse_argv(int argc, char *argv[]) {
         return 0;
 }
 
-static int retrieve(sd_journal *j, const char *name, const char **var) {
-        const void *data;
-        size_t len, field;
-        int r;
+static int retrieve(const void *data,
+                    size_t len,
+                    const char *name,
+                    const char **var) {
 
-        r = sd_journal_get_data(j, name, &data, &len);
-        if (r < 0) {
-                log_warning("Failed to retrieve %s", name);
-                return r;
-        }
+        size_t field;
+
+        field = strlen(name) + 1; /* name + "=" */
 
-        field = strlen(name) + 1; // name + "="
-        assert(len >= field);
+        if (len < field)
+                return 0;
+
+        if (memcmp(data, name, field - 1) != 0)
+                return 0;
+
+        if (((const char*) data)[field - 1] != '=')
+                return 0;
 
         *var = strndup((const char*)data + field, len - field);
         if (!var)
@@ -228,40 +237,60 @@ static int retrieve(sd_journal *j, const char *name, const char **var) {
         return 0;
 }
 
-static void print_entry(FILE* file, sd_journal *j, int had_header) {
+static int print_entry(FILE* file, sd_journal *j, int had_header) {
         const char _cleanup_free_
                 *pid = NULL, *uid = NULL, *gid = NULL,
                 *sgnl = NULL, *exe = NULL;
+        const void *d;
+        size_t l;
+        usec_t t;
+        char buf[FORMAT_TIMESTAMP_MAX];
+        int r;
 
-        retrieve(j, "COREDUMP_PID", &pid);
-        retrieve(j, "COREDUMP_UID", &uid);
-        retrieve(j, "COREDUMP_GID", &gid);
-        retrieve(j, "COREDUMP_SIGNAL", &sgnl);
-        retrieve(j, "COREDUMP_EXE", &exe);
-        if (!exe)
-                retrieve(j, "COREDUMP_COMM", &exe);
-        if (!exe)
-                retrieve(j, "COREDUMP_CMDLINE", &exe);
+        SD_JOURNAL_FOREACH_DATA(j, d, l) {
+                retrieve(d, l, "COREDUMP_PID", &pid);
+                retrieve(d, l, "COREDUMP_PID", &pid);
+                retrieve(d, l, "COREDUMP_UID", &uid);
+                retrieve(d, l, "COREDUMP_GID", &gid);
+                retrieve(d, l, "COREDUMP_SIGNAL", &sgnl);
+                retrieve(d, l, "COREDUMP_EXE", &exe);
+                if (!exe)
+                        retrieve(d, l, "COREDUMP_COMM", &exe);
+                if (!exe)
+                        retrieve(d, l, "COREDUMP_CMDLINE", &exe);
+        }
 
         if (!pid && !uid && !gid && !sgnl && !exe) {
-                log_warning("empty coredump log entry");
-                return;
+                log_warning("Empty coredump log entry");
+                return -EINVAL;
+        }
+
+        r = sd_journal_get_realtime_usec(j, &t);
+        if (r < 0) {
+                log_error("Failed to get realtime timestamp: %s", strerror(-r));
+                return r;
         }
 
+        format_timestamp(buf, sizeof(buf), t);
+
         if (!had_header)
-                fprintf(file, "%*s %*s %*s %*s %s\n",
+                fprintf(file, "%-*s %*s %*s %*s %*s %s\n",
+                        FORMAT_TIMESTAMP_MAX-1, "TIME",
                         6, "PID",
                         5, "UID",
                         5, "GID",
-                        3, "sig",
-                        "exe");
+                        3, "SIG",
+                           "EXE");
 
-        fprintf(file, "%*s %*s %*s %*s %s\n",
+        fprintf(file, "%*s %*s %*s %*s %*s %s\n",
+                FORMAT_TIMESTAMP_MAX-1, buf,
                 6, pid,
                 5, uid,
                 5, gid,
                 3, sgnl,
                 exe);
+
+        return 0;
 }
 
 static int dump_list(sd_journal *j) {
@@ -273,20 +302,16 @@ static int dump_list(sd_journal *j) {
                 print_entry(stdout, j, found++);
 
         if (!found) {
-                log_error("no coredumps found");
+                log_notice("No coredumps found");
                 return -ESRCH;
         }
 
         return 0;
 }
 
-static int dump_core(sd_journal* j) {
-        const char *data;
-        size_t len, ret;
+static int focus(sd_journal *j) {
         int r;
 
-        assert(j);
-
         r = sd_journal_seek_tail(j);
         if (r == 0)
                 r = sd_journal_previous(j);
@@ -294,17 +319,23 @@ static int dump_core(sd_journal* j) {
                 log_error("Failed to search journal: %s", strerror(-r));
                 return r;
         }
+        if (r == 0) {
+                log_error("No match found");
+                return -ESRCH;
+        }
+        return r;
+}
 
-       if (r == 0) {
-               log_error("No match found");
-               return -ESRCH;
-       }
+static int dump_core(sd_journal* j) {
+        const void *data;
+        size_t len, ret;
+        int r;
 
-        r = sd_journal_get_data(j, "COREDUMP", (const void**) &data, &len);
-        if (r != 0) {
-                log_error("retrieve COREDUMP field: %s", strerror(-r));
+        assert(j);
+
+        r = focus(j);
+        if (r < 0)
                 return r;
-        }
 
         print_entry(output ? stdout : stderr, j, false);
 
@@ -313,9 +344,17 @@ static int dump_core(sd_journal* j) {
                 return -ENOTTY;
         }
 
+        r = sd_journal_get_data(j, "COREDUMP", (const void**) &data, &len);
+        if (r < 0) {
+                log_error("Failed to retrieve COREDUMP field: %s", strerror(-r));
+                return r;
+        }
+
         assert(len >= 9);
+        data = (const uint8_t*) data + 9;
+        len -= 9;
 
-        ret = fwrite(data+9, len-9, 1, output ? output : stdout);
+        ret = fwrite(data, len, 1, output ? output : stdout);
         if (ret != 1) {
                 log_error("dumping coredump: %m (%zu)", ret);
                 return -errno;
@@ -328,6 +367,100 @@ static int dump_core(sd_journal* j) {
         return 0;
 }
 
+static int run_gdb(sd_journal *j) {
+        char path[] = "/var/tmp/coredump-XXXXXX";
+        const void *data;
+        size_t len;
+        ssize_t sz;
+        pid_t pid;
+        _cleanup_free_ char *exe = NULL;
+        int r;
+        _cleanup_close_ int fd = -1;
+        siginfo_t st;
+
+        assert(j);
+
+        r = focus(j);
+        if (r < 0)
+                return r;
+
+        print_entry(stdout, j, false);
+
+        r = sd_journal_get_data(j, "COREDUMP_EXE", (const void**) &data, &len);
+        if (r < 0) {
+                log_error("Failed to retrieve COREDUMP_EXE field: %s", strerror(-r));
+                return r;
+        }
+
+        assert(len >= 13);
+        data = (const uint8_t*) data + 13;
+        len -= 13;
+
+        exe = strndup(data, len);
+        if (!exe)
+                return log_oom();
+
+        if (endswith(exe, " (deleted)")) {
+                log_error("Binary already deleted.");
+                return -ENOENT;
+        }
+
+        r = sd_journal_get_data(j, "COREDUMP", (const void**) &data, &len);
+        if (r < 0) {
+                log_error("Failed to retrieve COREDUMP field: %s", strerror(-r));
+                return r;
+        }
+
+        assert(len >= 9);
+        data = (const uint8_t*) data + 9;
+        len -= 9;
+
+        fd = mkostemp(path, O_WRONLY);
+        if (fd < 0) {
+                log_error("Failed to create temporary file: %m");
+                return -errno;
+        }
+
+        sz = write(fd, data, len);
+        if (sz < 0) {
+                log_error("Failed to write temporary file: %s", strerror(errno));
+                r = -errno;
+                goto finish;
+        }
+        if (sz != (ssize_t) len) {
+                log_error("Short write to temporary file.");
+                r = -EIO;
+                goto finish;
+        }
+
+        close_nointr_nofail(fd);
+        fd = -1;
+
+        pid = fork();
+        if (pid < 0) {
+                log_error("Failed to fork(): %m");
+                r = -errno;
+                goto finish;
+        }
+        if (pid == 0) {
+                execlp("gdb", "gdb", exe, path, NULL);
+                log_error("Failed to invoke gdb: %m");
+                _exit(1);
+        }
+
+        r = wait_for_terminate(pid, &st);
+        if (r < 0) {
+                log_error("Failed to wait for gdb: %m");
+                goto finish;
+        }
+
+        r = st.si_code == CLD_EXITED ? st.si_status : 255;
+
+finish:
+        unlink(path);
+        return r;
+}
+
 int main(int argc, char *argv[]) {
         sd_journal *j = NULL;
         const char* match;
@@ -354,8 +487,6 @@ int main(int argc, char *argv[]) {
         }
 
         SET_FOREACH(match, matches, it) {
-                log_info("Matching: %s", match);
-
                 r = sd_journal_add_match(j, match, strlen(match));
                 if (r != 0) {
                         log_error("Failed to add match '%s': %s",
@@ -365,16 +496,23 @@ int main(int argc, char *argv[]) {
         }
 
         switch(arg_action) {
+
         case ACTION_LIST:
                 if (!arg_no_pager)
                         pager_open();
 
                 r = dump_list(j);
                 break;
+
         case ACTION_DUMP:
                 r = dump_core(j);
                 break;
-        case ACTION_NONE:
+
+        case  ACTION_GDB:
+                r = run_gdb(j);
+                break;
+
+        default:
                 assert_not_reached("Shouldn't be here");
         }
 
@@ -389,5 +527,5 @@ end:
         if (output)
                 fclose(output);
 
-        return r == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
+        return r >= 0 ? r : EXIT_FAILURE;
 }