/src/samba/lib/util/iov_buf.h

Source
/*
 * Unix SMB/CIFS implementation.
 * Samba system utilities
 * Copyright (C) Volker Lendecke 2014
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef __LIB_IOV_BUF_H__
#define __LIB_IOV_BUF_H__

#include "replace.h"
#include <talloc.h>
#include "system/filesys.h"

static inline
ssize_t iov_buf(const struct iovec *iov, int iovcnt,
    uint8_t *buf, size_t buflen)
{
  size_t needed = 0;
  uint8_t *p = buf;
  int i;

  for (i=0; i<iovcnt; i++) {
    size_t thislen = iov[i].iov_len;
    size_t tmp;

    tmp = needed + thislen;

    if (tmp < needed) {
      /* wrap */
      return -1;
    }
    needed = tmp;

    if ((p != NULL) && needed <= buflen && thislen > 0) {
      memcpy(p, iov[i].iov_base, thislen);
      p += thislen;
    }
  }

  return needed;
}

static inline
ssize_t iov_buflen(const struct iovec *iov, int iovcnt)
{
  return iov_buf(iov, iovcnt, NULL, 0);
}

static inline
bool iov_advance(struct iovec **iov, int *iovcnt, size_t n)
{
  struct iovec *v = *iov;
  int cnt = *iovcnt;

  while (n > 0) {
    if (cnt == 0) {
      return false;
    }
    if (n < v->iov_len) {
      v->iov_base = (char *)v->iov_base + n;
      v->iov_len -= n;
      break;
    }
    n -= v->iov_len;
    v += 1;
    cnt -= 1;
  }

  /*
   * Skip 0-length iovec's
   *
   * There might be empty buffers at the end of iov. Next time we do a
   * readv/writev based on this iov would give 0 transferred bytes, also
   * known as EPIPE. So we need to be careful discarding them.
   */

  while ((cnt > 0) && (v->iov_len == 0)) {
    v += 1;
    cnt -= 1;
  }

  *iov = v;
  *iovcnt = cnt;
  return true;
}

uint8_t *iov_concat(TALLOC_CTX *mem_ctx, const struct iovec *iov, int count);

#endif

Line	Count	Source
1		/*
2		* Unix SMB/CIFS implementation.
3		* Samba system utilities
4		* Copyright (C) Volker Lendecke 2014
5		*
6		* This program is free software; you can redistribute it and/or modify
7		* it under the terms of the GNU General Public License as published by
8		* the Free Software Foundation; either version 3 of the License, or
9		* (at your option) any later version.
10		*
11		* This program is distributed in the hope that it will be useful,
12		* but WITHOUT ANY WARRANTY; without even the implied warranty of
13		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14		* GNU General Public License for more details.
15		*
16		* You should have received a copy of the GNU General Public License
17		* along with this program. If not, see <http://www.gnu.org/licenses/>.
18		*/
19
20		#ifndef __LIB_IOV_BUF_H__
21		#define __LIB_IOV_BUF_H__
22
23		#include "replace.h"
24		#include <talloc.h>
25		#include "system/filesys.h"
26
27		static inline
28		ssize_t iov_buf(const struct iovec *iov, int iovcnt,
29		uint8_t *buf, size_t buflen)
30	0	{
31	0	size_t needed = 0;
32	0	uint8_t *p = buf;
33	0	int i;
34
35	0	for (i=0; i<iovcnt; i++) {
36	0	size_t thislen = iov[i].iov_len;
37	0	size_t tmp;
38
39	0	tmp = needed + thislen;
40
41	0	if (tmp < needed) {
42		/* wrap */
43	0	return -1;
44	0	}
45	0	needed = tmp;
46
47	0	if ((p != NULL) && needed <= buflen && thislen > 0) {
48	0	memcpy(p, iov[i].iov_base, thislen);
49	0	p += thislen;
50	0	}
51	0	}
52
53	0	return needed;
54	0	} Unexecuted instantiation: sys_rw_data.c:iov_buf Unexecuted instantiation: iov_buf.c:iov_buf Unexecuted instantiation: tsocket.c:iov_buf Unexecuted instantiation: tsocket_bsd.c:iov_buf Unexecuted instantiation: async_sock.c:iov_buf Unexecuted instantiation: messages_dgm.c:iov_buf Unexecuted instantiation: msghdr.c:iov_buf Unexecuted instantiation: messages.c:iov_buf Unexecuted instantiation: smb2_server.c:iov_buf Unexecuted instantiation: smbXsrv_client.c:iov_buf Unexecuted instantiation: smb2_signing.c:iov_buf Unexecuted instantiation: smbXcli_base.c:iov_buf Unexecuted instantiation: tstream_smbXcli_np.c:iov_buf Unexecuted instantiation: reparse.c:iov_buf Unexecuted instantiation: notifyd.c:iov_buf
55
56		static inline
57		ssize_t iov_buflen(const struct iovec *iov, int iovcnt)
58	0	{
59	0	return iov_buf(iov, iovcnt, NULL, 0);
60	0	} Unexecuted instantiation: sys_rw_data.c:iov_buflen Unexecuted instantiation: iov_buf.c:iov_buflen Unexecuted instantiation: tsocket.c:iov_buflen Unexecuted instantiation: tsocket_bsd.c:iov_buflen Unexecuted instantiation: async_sock.c:iov_buflen Unexecuted instantiation: messages_dgm.c:iov_buflen Unexecuted instantiation: msghdr.c:iov_buflen Unexecuted instantiation: messages.c:iov_buflen Unexecuted instantiation: smb2_server.c:iov_buflen Unexecuted instantiation: smbXsrv_client.c:iov_buflen Unexecuted instantiation: smb2_signing.c:iov_buflen Unexecuted instantiation: smbXcli_base.c:iov_buflen Unexecuted instantiation: tstream_smbXcli_np.c:iov_buflen Unexecuted instantiation: reparse.c:iov_buflen Unexecuted instantiation: notifyd.c:iov_buflen
61
62		static inline
63		bool iov_advance(struct iovec *iov, int iovcnt, size_t n)
64	0	{
65	0	struct iovec v = iov;
66	0	int cnt = *iovcnt;
67
68	0	while (n > 0) {
69	0	if (cnt == 0) {
70	0	return false;
71	0	}
72	0	if (n < v->iov_len) {
73	0	v->iov_base = (char *)v->iov_base + n;
74	0	v->iov_len -= n;
75	0	break;
76	0	}
77	0	n -= v->iov_len;
78	0	v += 1;
79	0	cnt -= 1;
80	0	}
81
82		/*
83		* Skip 0-length iovec's
84		*
85		* There might be empty buffers at the end of iov. Next time we do a
86		* readv/writev based on this iov would give 0 transferred bytes, also
87		* known as EPIPE. So we need to be careful discarding them.
88		*/
89
90	0	while ((cnt > 0) && (v->iov_len == 0)) {
91	0	v += 1;
92	0	cnt -= 1;
93	0	}
94
95	0	*iov = v;
96	0	*iovcnt = cnt;
97		return true;
98	0	} Unexecuted instantiation: sys_rw_data.c:iov_advance Unexecuted instantiation: iov_buf.c:iov_advance Unexecuted instantiation: tsocket.c:iov_advance Unexecuted instantiation: tsocket_bsd.c:iov_advance Unexecuted instantiation: async_sock.c:iov_advance Unexecuted instantiation: messages_dgm.c:iov_advance Unexecuted instantiation: msghdr.c:iov_advance Unexecuted instantiation: messages.c:iov_advance Unexecuted instantiation: smb2_server.c:iov_advance Unexecuted instantiation: smbXsrv_client.c:iov_advance Unexecuted instantiation: smb2_signing.c:iov_advance Unexecuted instantiation: smbXcli_base.c:iov_advance Unexecuted instantiation: tstream_smbXcli_np.c:iov_advance Unexecuted instantiation: reparse.c:iov_advance Unexecuted instantiation: notifyd.c:iov_advance
99
100		uint8_t iov_concat(TALLOC_CTX mem_ctx, const struct iovec *iov, int count);
101
102		#endif

Coverage Report

Created: 2026-06-07 07:07