Adding upstream version 255.4.upstream/255.4

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 70 insertions, 0 deletions

diff --git a/src/basic/iovec-util.c b/src/basic/iovec-util.c
new file mode 100644
index 0000000..991889a
--- /dev/null
+++ b/src/basic/iovec-util.c
@@ -0,0 +1,70 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+
+#include "iovec-util.h"
+#include "string-util.h"
+
+size_t iovec_total_size(const struct iovec *iovec, size_t n) {
+        size_t sum = 0;
+
+        assert(iovec || n == 0);
+
+        FOREACH_ARRAY(j, iovec, n)
+                sum += j->iov_len;
+
+        return sum;
+}
+
+bool iovec_increment(struct iovec *iovec, size_t n, size_t k) {
+        assert(iovec || n == 0);
+
+        /* Returns true if there is nothing else to send (bytes written cover all of the iovec),
+         * false if there's still work to do. */
+
+        FOREACH_ARRAY(j, iovec, n) {
+                size_t sub;
+
+                if (j->iov_len == 0)
+                        continue;
+                if (k == 0)
+                        return false;
+
+                sub = MIN(j->iov_len, k);
+                j->iov_len -= sub;
+                j->iov_base = (uint8_t*) j->iov_base + sub;
+                k -= sub;
+        }
+
+        assert(k == 0); /* Anything else would mean that we wrote more bytes than available,
+                         * or the kernel reported writing more bytes than sent. */
+        return true;
+}
+
+char* set_iovec_string_field(struct iovec *iovec, size_t *n_iovec, const char *field, const char *value) {
+        char *x;
+
+        assert(iovec);
+        assert(n_iovec);
+
+        x = strjoin(field, value);
+        if (x)
+                iovec[(*n_iovec)++] = IOVEC_MAKE_STRING(x);
+        return x;
+}
+
+char* set_iovec_string_field_free(struct iovec *iovec, size_t *n_iovec, const char *field, char *value) {
+        char *x;
+
+        assert(iovec);
+        assert(n_iovec);
+
+        x = set_iovec_string_field(iovec, n_iovec, field, value);
+        free(value);
+        return x;
+}
+
+void iovec_array_free(struct iovec *iovec, size_t n) {
+        FOREACH_ARRAY(i, iovec, n)
+                free(i->iov_base);
+
+        free(iovec);
+}