diff options
Diffstat (limited to 'src/journal/coredump.c')
| -rw-r--r-- | src/journal/coredump.c | 30 |
1 files changed, 13 insertions, 17 deletions
diff --git a/src/journal/coredump.c b/src/journal/coredump.c index 29342de681..74b1dbd013 100644 --- a/src/journal/coredump.c +++ b/src/journal/coredump.c @@ -39,10 +39,10 @@ #include "journald-native.h" /* Few programs have less than 3MiB resident */ -#define COREDUMP_MIN_START (3*1024*1024) +#define COREDUMP_MIN_START (3*1024*1024u) /* Make sure to not make this larger than the maximum journal entry * size. See ENTRY_SIZE_MAX in journald-native.c. */ -#define COREDUMP_MAX (767*1024*1024) +#define COREDUMP_MAX (767*1024*1024u) assert_cc(COREDUMP_MAX <= ENTRY_SIZE_MAX); enum { @@ -107,7 +107,7 @@ int main(int argc, char* argv[]) { uid_t uid; gid_t gid; struct iovec iovec[14]; - size_t coredump_bufsize, coredump_size; + size_t coredump_bufsize = 0, coredump_size = 0; _cleanup_free_ char *core_pid = NULL, *core_uid = NULL, *core_gid = NULL, *core_signal = NULL, *core_timestamp = NULL, *core_comm = NULL, *core_exe = NULL, *core_unit = NULL, *core_session = NULL, *core_message = NULL, *core_cmdline = NULL, *coredump_data = NULL; @@ -239,17 +239,18 @@ int main(int argc, char* argv[]) { goto finish; } - coredump_bufsize = COREDUMP_MIN_START; - coredump_data = malloc(coredump_bufsize); - if (!coredump_data) { - log_warning("Failed to allocate memory for core, core will not be stored."); - goto finalize; - } + for (;;) { + if (!GREEDY_REALLOC(coredump_data, coredump_bufsize, + MAX(coredump_size + 1, COREDUMP_MIN_START/2))) { + log_warning("Failed to allocate memory for core, core will not be stored."); + goto finalize; + } - memcpy(coredump_data, "COREDUMP=", 9); - coredump_size = 9; + if (coredump_size == 0) { + memcpy(coredump_data, "COREDUMP=", 9); + coredump_size = 9; + } - for (;;) { n = loop_read(STDIN_FILENO, coredump_data + coredump_size, coredump_bufsize - coredump_size, false); if (n < 0) { @@ -265,11 +266,6 @@ int main(int argc, char* argv[]) { log_error("Core too large, core will not be stored."); goto finalize; } - - if (!GREEDY_REALLOC(coredump_data, coredump_bufsize, coredump_size + 1)) { - log_warning("Failed to allocate memory for core, core will not be stored."); - goto finalize; - } } iovec[j].iov_base = coredump_data; |