Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2019 Intel Corporation.
3 : : * All rights reserved.
4 : : */
5 : :
6 : : #include "spdk/util.h"
7 : : #include "spdk/log.h"
8 : :
9 : : void
10 : 15491 : spdk_iov_memset(struct iovec *iovs, int iovcnt, int c)
11 : : {
12 : 15491 : int iov_idx = 0;
13 : : struct iovec *iov;
14 : :
15 [ + + ]: 31002 : while (iov_idx < iovcnt) {
16 : 15511 : iov = &iovs[iov_idx];
17 [ - + ]: 15511 : memset(iov->iov_base, c, iov->iov_len);
18 : 15511 : iov_idx++;
19 : : }
20 : 15491 : }
21 : :
22 : : size_t
23 : 268096361 : spdk_ioviter_first(struct spdk_ioviter *iter,
24 : : struct iovec *siov, size_t siovcnt,
25 : : struct iovec *diov, size_t diovcnt,
26 : : void **src, void **dst)
27 : : {
28 : 72504196 : struct iovec *iovs[2];
29 : 72504196 : size_t iovcnts[2];
30 : 72504196 : void *out[2];
31 : : size_t len;
32 : :
33 : 268096361 : iovs[0] = siov;
34 : 268096361 : iovcnts[0] = siovcnt;
35 : :
36 : 268096361 : iovs[1] = diov;
37 : 268096361 : iovcnts[1] = diovcnt;
38 : :
39 : 268096361 : len = spdk_ioviter_firstv(iter, 2, iovs, iovcnts, out);
40 : :
41 [ + - ]: 268096361 : if (len > 0) {
42 : 268096361 : *src = out[0];
43 : 268096361 : *dst = out[1];
44 : : }
45 : :
46 : 268096361 : return len;
47 : : }
48 : :
49 : : size_t
50 : 268715219 : spdk_ioviter_firstv(struct spdk_ioviter *iter,
51 : : uint32_t count,
52 : : struct iovec **iov,
53 : : size_t *iovcnt,
54 : : void **out)
55 : : {
56 : : struct spdk_single_ioviter *it;
57 : : uint32_t i;
58 : :
59 : 268715219 : iter->count = count;
60 : :
61 [ + + ]: 806813445 : for (i = 0; i < count; i++) {
62 : 538098326 : it = &iter->iters[i];
63 : 538098326 : it->iov = iov[i];
64 : 538098326 : it->iovcnt = iovcnt[i];
65 : 538098326 : it->idx = 0;
66 : 538098326 : it->iov_len = iov[i][0].iov_len;
67 : 538098326 : it->iov_base = iov[i][0].iov_base;
68 : : }
69 : :
70 : 268715219 : return spdk_ioviter_nextv(iter, out);
71 : : }
72 : :
73 : : size_t
74 : 303348878 : spdk_ioviter_next(struct spdk_ioviter *iter, void **src, void **dst)
75 : : {
76 : 75764789 : void *out[2];
77 : : size_t len;
78 : :
79 : 303348878 : len = spdk_ioviter_nextv(iter, out);
80 : :
81 [ + + ]: 303348878 : if (len > 0) {
82 : 35252488 : *src = out[0];
83 : 35252488 : *dst = out[1];
84 : : }
85 : :
86 : 303348878 : return len;
87 : : }
88 : :
89 : : size_t
90 : 572268519 : spdk_ioviter_nextv(struct spdk_ioviter *iter, void **out)
91 : : {
92 : : struct spdk_single_ioviter *it;
93 : : size_t len;
94 : : uint32_t i;
95 : :
96 : : /* Figure out the minimum size of each iovec's next segment */
97 : 572268519 : len = UINT32_MAX;
98 [ + + ]: 1293744417 : for (i = 0; i < iter->count; i++) {
99 : 989572259 : it = &iter->iters[i];
100 [ + + + + ]: 989572259 : if (it->idx == it->iovcnt || it->iov_len == 0) {
101 : : /* This element has 0 bytes remaining, so we're done. */
102 : 268096361 : return 0;
103 : : }
104 : :
105 : 721475898 : len = spdk_min(len, it->iov_len);
106 : : }
107 : :
108 [ + + ]: 913654922 : for (i = 0; i < iter->count; i++) {
109 : 609482764 : it = &iter->iters[i];
110 : :
111 : 609482764 : out[i] = it->iov_base;
112 : :
113 [ + + ]: 609482764 : if (it->iov_len == len) {
114 : : /* Advance to next element */
115 : 458338937 : it->idx++;
116 [ + + ]: 458338937 : if (it->idx != it->iovcnt) {
117 : : /* Set up for next element */
118 : 63123220 : it->iov_len = it->iov[it->idx].iov_len;
119 : 63123220 : it->iov_base = it->iov[it->idx].iov_base;
120 : : }
121 : : } else {
122 : : /* Partial buffer */
123 : 151143857 : it->iov_base += len;
124 : 151143857 : it->iov_len -= len;
125 : : }
126 : : }
127 : :
128 : 304172158 : return len;
129 : : }
130 : :
131 : : size_t
132 : 140376124 : spdk_iovcpy(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
133 : : {
134 : 2136016 : struct spdk_ioviter iter;
135 : : size_t len, total_sz;
136 : 2136016 : void *src, *dst;
137 : :
138 : 140376124 : total_sz = 0;
139 [ + + ]: 140504240 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
140 [ + + ]: 283967150 : len != 0;
141 : 143756229 : len = spdk_ioviter_next(&iter, &src, &dst)) {
142 [ - + - + ]: 143756229 : memcpy(dst, src, len);
143 : 143756229 : total_sz += len;
144 : : }
145 : :
146 : 140376124 : return total_sz;
147 : : }
148 : :
149 : : size_t
150 : 210 : spdk_iovmove(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
151 : : {
152 : 175 : struct spdk_ioviter iter;
153 : : size_t len, total_sz;
154 : 175 : void *src, *dst;
155 : :
156 : 210 : total_sz = 0;
157 [ + + ]: 280 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
158 [ + + ]: 280 : len != 0;
159 : 210 : len = spdk_ioviter_next(&iter, &src, &dst)) {
160 [ - + - + ]: 210 : memmove(dst, src, len);
161 : 210 : total_sz += len;
162 : : }
163 : :
164 : 210 : return total_sz;
165 : : }
166 : :
167 : : void
168 : 4398957 : spdk_iov_xfer_init(struct spdk_iov_xfer *ix, struct iovec *iovs, int iovcnt)
169 : : {
170 : 4398957 : ix->iovs = iovs;
171 : 4398957 : ix->iovcnt = iovcnt;
172 : 4398957 : ix->cur_iov_idx = 0;
173 : 4398957 : ix->cur_iov_offset = 0;
174 : 4398957 : }
175 : :
176 : : static size_t
177 : 4399415 : iov_xfer(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len, bool to_buf)
178 : : {
179 : 4399415 : size_t len, iov_remain_len, copied_len = 0;
180 : : struct iovec *iov;
181 : :
182 [ - + ]: 4399415 : if (buf_len == 0) {
183 : 0 : return 0;
184 : : }
185 : :
186 [ + + ]: 12967813 : while (ix->cur_iov_idx < ix->iovcnt) {
187 : 8683808 : iov = &ix->iovs[ix->cur_iov_idx];
188 : 8683808 : iov_remain_len = iov->iov_len - ix->cur_iov_offset;
189 [ + + ]: 8683808 : if (iov_remain_len == 0) {
190 : 4284202 : ix->cur_iov_idx++;
191 : 4284202 : ix->cur_iov_offset = 0;
192 : 4284202 : continue;
193 : : }
194 : :
195 : 4399606 : len = spdk_min(iov_remain_len, buf_len - copied_len);
196 : :
197 [ + + ]: 4399606 : if (to_buf) {
198 [ - + - + ]: 157896 : memcpy((char *)buf + copied_len,
199 : 157896 : (char *)iov->iov_base + ix->cur_iov_offset, len);
200 : : } else {
201 [ - + - + ]: 4241710 : memcpy((char *)iov->iov_base + ix->cur_iov_offset,
202 : : (const char *)buf + copied_len, len);
203 : : }
204 : 4399606 : copied_len += len;
205 : 4399606 : ix->cur_iov_offset += len;
206 : :
207 [ + + ]: 4399606 : if (buf_len == copied_len) {
208 : 115410 : return copied_len;
209 : : }
210 : : }
211 : :
212 : 4284005 : return copied_len;
213 : : }
214 : :
215 : : size_t
216 : 4241597 : spdk_iov_xfer_from_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
217 : : {
218 : 4241597 : return iov_xfer(ix, buf, buf_len, false);
219 : : }
220 : :
221 : : size_t
222 : 157818 : spdk_iov_xfer_to_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
223 : : {
224 : 157818 : return iov_xfer(ix, buf, buf_len, true);
225 : : }
226 : :
227 : : void
228 : 59586 : spdk_copy_iovs_to_buf(void *buf, size_t buf_len, struct iovec *iovs, int iovcnt)
229 : : {
230 : 59552 : struct spdk_iov_xfer ix;
231 : :
232 : 59586 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
233 : 59586 : spdk_iov_xfer_to_buf(&ix, buf, buf_len);
234 : 59586 : }
235 : :
236 : : void
237 : 4229701 : spdk_copy_buf_to_iovs(struct iovec *iovs, int iovcnt, void *buf, size_t buf_len)
238 : : {
239 : 4228928 : struct spdk_iov_xfer ix;
240 : :
241 : 4229701 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
242 : 4229701 : spdk_iov_xfer_from_buf(&ix, buf, buf_len);
243 : 4229701 : }
|