From: Lennart Poettering <lennart@poettering.net>
Date: Thu, 19 Oct 2023 15:04:04 +0000 (+0200)
Subject: iovec-util: add some useful helpers for dealing with iovecs that refer to dynamic... 
X-Git-Tag: v255-rc1~181^2~2
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=3a856171c2dc78f040c560142d3b275de1c1bb0a;p=thirdparty%2Fsystemd.git

iovec-util: add some useful helpers for dealing with iovecs that refer to dynamic memory
---

diff --git a/src/basic/iovec-util.h b/src/basic/iovec-util.h
index 1f72898cba0..1f652a0ede6 100644
--- a/src/basic/iovec-util.h
+++ b/src/basic/iovec-util.h
@@ -5,6 +5,7 @@
 #include <sys/types.h>
 #include <sys/uio.h>
 
+#include "alloc-util.h"
 #include "macro.h"
 
 size_t iovec_total_size(const struct iovec *i, size_t n);
@@ -20,6 +21,27 @@ bool iovec_increment(struct iovec *i, size_t n, size_t k);
                 IOVEC_MAKE((char*) _s, strlen(_s));     \
         })
 
+#define TAKE_IOVEC(p) TAKE_GENERIC((p), struct iovec, IOVEC_NULL)
+
+static inline void iovec_done(struct iovec *iovec) {
+        /* A _cleanup_() helper that frees the iov_base in the iovec */
+        assert(iovec);
+
+        iovec->iov_base = mfree(iovec->iov_base);
+        iovec->iov_len = 0;
+}
+
+static inline void iovec_done_erase(struct iovec *iovec) {
+        assert(iovec);
+
+        iovec->iov_base = erase_and_free(iovec->iov_base);
+        iovec->iov_len = 0;
+}
+
+static inline bool iovec_is_set(const struct iovec *iov) {
+        return iov && iov->iov_len > 0 && iov->iov_base;
+}
+
 char* set_iovec_string_field(struct iovec *iovec, size_t *n_iovec, const char *field, const char *value);
 char* set_iovec_string_field_free(struct iovec *iovec, size_t *n_iovec, const char *field, char *value);