X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Fjournal%2Fjournal-send.c;h=db5dbc0711c7c9a06d0b8f158529064178292ea2;hb=e91af489a25e8bb65016a63f533ae28a505119ef;hp=238d64c13eb698614fc390204c41d14140b6c7db;hpb=d0bbc21caa6e68693a47db60c93e99422bf2a858;p=elogind.git diff --git a/src/journal/journal-send.c b/src/journal/journal-send.c index 238d64c13..db5dbc071 100644 --- a/src/journal/journal-send.c +++ b/src/journal/journal-send.c @@ -26,6 +26,7 @@ #include "sd-journal.h" #include "util.h" +#include "socket-util.h" /* We open a single fd, and we'll share it with the current process, * all its threads, and all its subprocesses. This means we need to @@ -52,7 +53,7 @@ retry: return fd; } -int sd_journal_print(int priority, const char *format, ...) { +_public_ int sd_journal_print(int priority, const char *format, ...) { int r; va_list ap; @@ -63,10 +64,16 @@ int sd_journal_print(int priority, const char *format, ...) { return r; } -int sd_journal_printv(int priority, const char *format, va_list ap) { +_public_ int sd_journal_printv(int priority, const char *format, va_list ap) { char buffer[8 + LINE_MAX], p[11]; struct iovec iov[2]; + if (priority < 0 || priority > 7) + return -EINVAL; + + if (!format) + return -EINVAL; + snprintf(p, sizeof(p), "PRIORITY=%i", priority & LOG_PRIMASK); char_array_0(p); @@ -81,7 +88,7 @@ int sd_journal_printv(int priority, const char *format, va_list ap) { return sd_journal_sendv(iov, 2); } -int sd_journal_send(const char *format, ...) { +_public_ int sd_journal_send(const char *format, ...) { int r, n = 0, i = 0, j; va_list ap; struct iovec *iov = NULL; @@ -124,7 +131,7 @@ fail: return r; } -int sd_journal_sendv(const struct iovec *iov, int n) { +_public_ int sd_journal_sendv(const struct iovec *iov, int n) { int fd; struct iovec *w; uint64_t *l; @@ -141,8 +148,12 @@ int sd_journal_sendv(const struct iovec *iov, int n) { for (i = 0; i < n; i++) { char *c, *nl; + if (!iov[i].iov_base || + iov[i].iov_len <= 1) + return -EINVAL; + c = memchr(iov[i].iov_base, '=', iov[i].iov_len); - if (!c) + if (!c || c == iov[i].iov_base) return -EINVAL; nl = memchr(iov[i].iov_base, '\n', iov[i].iov_len); @@ -197,3 +208,56 @@ int sd_journal_sendv(const struct iovec *iov, int n) { return 0; } + +_public_ int sd_journal_stream_fd(const char *tag, int priority, int priority_prefix) { + union sockaddr_union sa; + int fd; + char *header; + size_t l; + ssize_t r; + + if (priority < 0 || priority > 7) + return -EINVAL; + + fd = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC, 0); + if (fd < 0) + return -errno; + + zero(sa); + sa.un.sun_family = AF_UNIX; + strncpy(sa.un.sun_path, "/run/systemd/stdout", sizeof(sa.un.sun_path)); + + r = connect(fd, &sa.sa, offsetof(union sockaddr_union, un.sun_path) + strlen(sa.un.sun_path)); + if (r < 0) { + close_nointr_nofail(fd); + return -errno; + } + + if (!tag) + tag = ""; + + l = strlen(tag); + header = alloca(l + 1 + 2 + 2 + 2); + + memcpy(header, tag, l); + header[l++] = '\n'; + header[l++] = '0' + priority; + header[l++] = '\n'; + header[l++] = '0' + !!priority_prefix; + header[l++] = '\n'; + header[l++] = '0'; + header[l++] = '\n'; + + r = loop_write(fd, header, l, false); + if (r < 0) { + close_nointr_nofail(fd); + return (int) r; + } + + if ((size_t) r != l) { + close_nointr_nofail(fd); + return -errno; + } + + return fd; +}