14c87aefeSPatrick Mooney /*-
2*32640292SAndy Fiddaman * SPDX-License-Identifier: BSD-2-Clause
34c87aefeSPatrick Mooney *
44c87aefeSPatrick Mooney * Copyright (c) 2016 Jakub Klama <jceel@FreeBSD.org>.
54c87aefeSPatrick Mooney * Copyright (c) 2018 Alexander Motin <mav@FreeBSD.org>
64c87aefeSPatrick Mooney * All rights reserved.
74c87aefeSPatrick Mooney *
84c87aefeSPatrick Mooney * Redistribution and use in source and binary forms, with or without
94c87aefeSPatrick Mooney * modification, are permitted provided that the following conditions
104c87aefeSPatrick Mooney * are met:
114c87aefeSPatrick Mooney * 1. Redistributions of source code must retain the above copyright
124c87aefeSPatrick Mooney * notice, this list of conditions and the following disclaimer
134c87aefeSPatrick Mooney * in this position and unchanged.
144c87aefeSPatrick Mooney * 2. Redistributions in binary form must reproduce the above copyright
154c87aefeSPatrick Mooney * notice, this list of conditions and the following disclaimer in the
164c87aefeSPatrick Mooney * documentation and/or other materials provided with the distribution.
174c87aefeSPatrick Mooney *
184c87aefeSPatrick Mooney * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
194c87aefeSPatrick Mooney * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
204c87aefeSPatrick Mooney * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
214c87aefeSPatrick Mooney * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
224c87aefeSPatrick Mooney * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
234c87aefeSPatrick Mooney * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
244c87aefeSPatrick Mooney * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
254c87aefeSPatrick Mooney * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
264c87aefeSPatrick Mooney * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
274c87aefeSPatrick Mooney * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
284c87aefeSPatrick Mooney * SUCH DAMAGE.
294c87aefeSPatrick Mooney */
304c87aefeSPatrick Mooney
314c87aefeSPatrick Mooney #include <sys/cdefs.h>
324c87aefeSPatrick Mooney
334c87aefeSPatrick Mooney #include <sys/param.h>
344c87aefeSPatrick Mooney #include <sys/types.h>
354c87aefeSPatrick Mooney #include <sys/uio.h>
364c87aefeSPatrick Mooney
374c87aefeSPatrick Mooney #include <stdlib.h>
384c87aefeSPatrick Mooney #include <string.h>
394c87aefeSPatrick Mooney #include "iov.h"
404c87aefeSPatrick Mooney
414c87aefeSPatrick Mooney void
seek_iov(const struct iovec * iov1,int niov1,struct iovec * iov2,int * niov2,size_t seek)424c87aefeSPatrick Mooney seek_iov(const struct iovec *iov1, int niov1, struct iovec *iov2, int *niov2,
434c87aefeSPatrick Mooney size_t seek)
444c87aefeSPatrick Mooney {
454c87aefeSPatrick Mooney size_t remainder = 0;
464c87aefeSPatrick Mooney size_t left = seek;
474c87aefeSPatrick Mooney int i, j;
484c87aefeSPatrick Mooney
494c87aefeSPatrick Mooney for (i = 0; i < niov1; i++) {
504c87aefeSPatrick Mooney size_t toseek = MIN(left, iov1[i].iov_len);
514c87aefeSPatrick Mooney left -= toseek;
524c87aefeSPatrick Mooney
534c87aefeSPatrick Mooney if (toseek == iov1[i].iov_len)
544c87aefeSPatrick Mooney continue;
554c87aefeSPatrick Mooney
564c87aefeSPatrick Mooney if (left == 0) {
574c87aefeSPatrick Mooney remainder = toseek;
584c87aefeSPatrick Mooney break;
594c87aefeSPatrick Mooney }
604c87aefeSPatrick Mooney }
614c87aefeSPatrick Mooney
624c87aefeSPatrick Mooney for (j = i; j < niov1; j++) {
634c87aefeSPatrick Mooney iov2[j - i].iov_base = (char *)iov1[j].iov_base + remainder;
644c87aefeSPatrick Mooney iov2[j - i].iov_len = iov1[j].iov_len - remainder;
654c87aefeSPatrick Mooney remainder = 0;
664c87aefeSPatrick Mooney }
674c87aefeSPatrick Mooney
684c87aefeSPatrick Mooney *niov2 = j - i;
694c87aefeSPatrick Mooney }
704c87aefeSPatrick Mooney
714c87aefeSPatrick Mooney size_t
count_iov(const struct iovec * iov,int niov)724c87aefeSPatrick Mooney count_iov(const struct iovec *iov, int niov)
734c87aefeSPatrick Mooney {
744c87aefeSPatrick Mooney size_t total = 0;
754c87aefeSPatrick Mooney int i;
764c87aefeSPatrick Mooney
774c87aefeSPatrick Mooney for (i = 0; i < niov; i++)
784c87aefeSPatrick Mooney total += iov[i].iov_len;
794c87aefeSPatrick Mooney
804c87aefeSPatrick Mooney return (total);
814c87aefeSPatrick Mooney }
824c87aefeSPatrick Mooney
834c87aefeSPatrick Mooney void
truncate_iov(struct iovec * iov,int * niov,size_t length)844c87aefeSPatrick Mooney truncate_iov(struct iovec *iov, int *niov, size_t length)
854c87aefeSPatrick Mooney {
864c87aefeSPatrick Mooney size_t done = 0;
874c87aefeSPatrick Mooney int i;
884c87aefeSPatrick Mooney
894c87aefeSPatrick Mooney for (i = 0; i < *niov; i++) {
904c87aefeSPatrick Mooney size_t toseek = MIN(length - done, iov[i].iov_len);
914c87aefeSPatrick Mooney done += toseek;
924c87aefeSPatrick Mooney
934c87aefeSPatrick Mooney if (toseek <= iov[i].iov_len) {
944c87aefeSPatrick Mooney iov[i].iov_len = toseek;
954c87aefeSPatrick Mooney *niov = i + 1;
964c87aefeSPatrick Mooney return;
974c87aefeSPatrick Mooney }
984c87aefeSPatrick Mooney }
994c87aefeSPatrick Mooney }
1004c87aefeSPatrick Mooney
1014c87aefeSPatrick Mooney ssize_t
iov_to_buf(const struct iovec * iov,int niov,void ** buf)1024c87aefeSPatrick Mooney iov_to_buf(const struct iovec *iov, int niov, void **buf)
1034c87aefeSPatrick Mooney {
1044c87aefeSPatrick Mooney size_t ptr, total;
1054c87aefeSPatrick Mooney int i;
1064c87aefeSPatrick Mooney
1074c87aefeSPatrick Mooney total = count_iov(iov, niov);
1084c87aefeSPatrick Mooney *buf = realloc(*buf, total);
1094c87aefeSPatrick Mooney if (*buf == NULL)
1104c87aefeSPatrick Mooney return (-1);
1114c87aefeSPatrick Mooney
1124c87aefeSPatrick Mooney for (i = 0, ptr = 0; i < niov; i++) {
11359d65d31SAndy Fiddaman memcpy((uint8_t *)*buf + ptr, iov[i].iov_base, iov[i].iov_len);
1144c87aefeSPatrick Mooney ptr += iov[i].iov_len;
1154c87aefeSPatrick Mooney }
1164c87aefeSPatrick Mooney
1174c87aefeSPatrick Mooney return (total);
1184c87aefeSPatrick Mooney }
1194c87aefeSPatrick Mooney
1204c87aefeSPatrick Mooney ssize_t
buf_to_iov(const void * buf,size_t buflen,const struct iovec * iov,int niov,size_t seek)121154972afSPatrick Mooney buf_to_iov(const void *buf, size_t buflen, const struct iovec *iov, int niov,
1224c87aefeSPatrick Mooney size_t seek)
1234c87aefeSPatrick Mooney {
1244c87aefeSPatrick Mooney struct iovec *diov;
1254c87aefeSPatrick Mooney size_t off = 0, len;
126154972afSPatrick Mooney int i;
127154972afSPatrick Mooney
128154972afSPatrick Mooney #ifndef __FreeBSD__
129154972afSPatrick Mooney diov = NULL;
130154972afSPatrick Mooney #endif
1314c87aefeSPatrick Mooney
1324c87aefeSPatrick Mooney if (seek > 0) {
133154972afSPatrick Mooney int ndiov;
134154972afSPatrick Mooney
1354c87aefeSPatrick Mooney diov = malloc(sizeof(struct iovec) * niov);
1364c87aefeSPatrick Mooney seek_iov(iov, niov, diov, &ndiov, seek);
137154972afSPatrick Mooney iov = diov;
138154972afSPatrick Mooney niov = ndiov;
1394c87aefeSPatrick Mooney }
1404c87aefeSPatrick Mooney
141154972afSPatrick Mooney for (i = 0; i < niov && off < buflen; i++) {
142154972afSPatrick Mooney len = MIN(iov[i].iov_len, buflen - off);
14359d65d31SAndy Fiddaman memcpy(iov[i].iov_base, (const uint8_t *)buf + off, len);
1444c87aefeSPatrick Mooney off += len;
1454c87aefeSPatrick Mooney }
1464c87aefeSPatrick Mooney
1474c87aefeSPatrick Mooney if (seek > 0)
1484c87aefeSPatrick Mooney free(diov);
1494c87aefeSPatrick Mooney
1504c87aefeSPatrick Mooney return ((ssize_t)off);
1514c87aefeSPatrick Mooney }
1524c87aefeSPatrick Mooney
153