summaryrefslogtreecommitdiffstats
path: root/src/basic/memstream-util.c
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-10 20:49:52 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-10 20:49:52 +0000
commit55944e5e40b1be2afc4855d8d2baf4b73d1876b5 (patch)
tree33f869f55a1b149e9b7c2b7e201867ca5dd52992 /src/basic/memstream-util.c
parentInitial commit. (diff)
downloadsystemd-55944e5e40b1be2afc4855d8d2baf4b73d1876b5.tar.xz
systemd-55944e5e40b1be2afc4855d8d2baf4b73d1876b5.zip
Adding upstream version 255.4.upstream/255.4
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/basic/memstream-util.c')
-rw-r--r--src/basic/memstream-util.c75
1 files changed, 75 insertions, 0 deletions
diff --git a/src/basic/memstream-util.c b/src/basic/memstream-util.c
new file mode 100644
index 0000000..4e147fd
--- /dev/null
+++ b/src/basic/memstream-util.c
@@ -0,0 +1,75 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+
+#include "alloc-util.h"
+#include "fd-util.h"
+#include "fileio.h"
+#include "log.h"
+#include "memstream-util.h"
+
+void memstream_done(MemStream *m) {
+ assert(m);
+
+ /* First, close file stream, as the buffer may be reallocated on close. */
+ safe_fclose(m->f);
+
+ /* Then, free buffer. */
+ free(m->buf);
+}
+
+FILE* memstream_init(MemStream *m) {
+ assert(m);
+ assert(!m->f);
+
+ m->f = open_memstream_unlocked(&m->buf, &m->sz);
+ return m->f;
+}
+
+int memstream_finalize(MemStream *m, char **ret_buf, size_t *ret_size) {
+ int r;
+
+ assert(m);
+ assert(m->f);
+ assert(ret_buf);
+
+ /* Add terminating NUL, so that the output buffer is a valid string. */
+ fputc('\0', m->f);
+
+ r = fflush_and_check(m->f);
+ if (r < 0)
+ return r;
+
+ m->f = safe_fclose(m->f);
+
+ /* On fclose(), the buffer may be reallocated, and may trigger OOM. */
+ if (!m->buf)
+ return -ENOMEM;
+
+ assert(m->sz > 0);
+
+ *ret_buf = TAKE_PTR(m->buf);
+ if (ret_size)
+ *ret_size = m->sz - 1;
+
+ m->sz = 0; /* For safety when the MemStream object will be reused later. */
+ return 0;
+}
+
+int memstream_dump_internal(
+ int level,
+ int error,
+ const char *file,
+ int line,
+ const char *func,
+ MemStream *m) {
+
+ _cleanup_free_ char *buf = NULL;
+ int r;
+
+ assert(m);
+
+ r = memstream_finalize(m, &buf, NULL);
+ if (r < 0)
+ return log_full_errno(level, r, "Failed to flush memstream: %m: %m");
+
+ return log_dump_internal(level, error, file, line, func, buf);
+}