diff options
| -rw-r--r-- | iov.c | 31 | ||||
| -rw-r--r-- | iov.h | 11 |
2 files changed, 42 insertions, 0 deletions
@@ -360,6 +360,37 @@ void *iov_peek_header_(struct iov_tail *tail, void *v, size_t len, size_t align) } /** + * iov_push_header_() - Write a new header to an IOV tail + * @tail: IOV tail to write header to + * @v: Pointer to header data to write + * @len: Length of header to write, in bytes + * + * Return: number of bytes written + */ +/* cppcheck-suppress unusedFunction */ +size_t iov_push_header_(struct iov_tail *tail, const void *v, size_t len) +{ + size_t l; + + if (!iov_tail_prune(tail)) + return 0; /* No space */ + + if ((char *)tail->iov[0].iov_base + tail->off == v) /* already in place */ + l = len; + else + l = iov_from_buf(tail->iov, tail->cnt, tail->off, v, len); + + if (l != len) { + warn("iov_push_header_(): Not enough space to store header" + " (%zu != %zu)", l, len); + } + + tail->off = tail->off + l; + + return l; +} + +/** * iov_remove_header_() - Remove a header from an IOV tail * @tail: IOV tail to remove header from (modified) * @v: Temporary memory to use if the memory in @tail @@ -93,6 +93,7 @@ bool iov_tail_prune(struct iov_tail *tail); size_t iov_tail_size(struct iov_tail *tail); bool iov_drop_header(struct iov_tail *tail, size_t len); void *iov_peek_header_(struct iov_tail *tail, void *v, size_t len, size_t align); +size_t iov_push_header_(struct iov_tail *tail, const void *v, size_t len); void *iov_remove_header_(struct iov_tail *tail, void *v, size_t len, size_t align); ssize_t iov_tail_clone(struct iovec *dst_iov, size_t dst_iov_cnt, struct iov_tail *tail); @@ -116,6 +117,16 @@ ssize_t iov_tail_clone(struct iovec *dst_iov, size_t dst_iov_cnt, __alignof__(var_)))) /** + * IOV_PUSH_HEADER() - Write a new header to an IOV tail + * @tail_: IOV tail to write header to + * @var_: A variable containing the header data to write + * + * Return: number of bytes written + */ +#define IOV_PUSH_HEADER(tail_, var_) \ + (iov_push_header_((tail_), &(var_), sizeof(var_))) + +/** * IOV_REMOVE_HEADER() - Remove and return typed header from an IOV tail * @tail_: IOV tail to remove header from (modified) * @var_: Temporary buffer of the type of the header to use if |