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 : 15541 : spdk_iov_memset(struct iovec *iovs, int iovcnt, int c)
11 : : {
12 : 15541 : int iov_idx = 0;
13 : : struct iovec *iov;
14 : :
15 [ + + ]: 31102 : while (iov_idx < iovcnt) {
16 : 15561 : iov = &iovs[iov_idx];
17 [ - + ]: 15561 : memset(iov->iov_base, c, iov->iov_len);
18 : 15561 : iov_idx++;
19 : : }
20 : 15541 : }
21 : :
22 : : size_t
23 : 250051335 : 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 : 66921564 : struct iovec *iovs[2];
29 : 66921564 : size_t iovcnts[2];
30 : 66921564 : void *out[2];
31 : : size_t len;
32 : :
33 : 250051335 : iovs[0] = siov;
34 : 250051335 : iovcnts[0] = siovcnt;
35 : :
36 : 250051335 : iovs[1] = diov;
37 : 250051335 : iovcnts[1] = diovcnt;
38 : :
39 : 250051335 : len = spdk_ioviter_firstv(iter, 2, iovs, iovcnts, out);
40 : :
41 [ + - ]: 250051335 : if (len > 0) {
42 : 250051335 : *src = out[0];
43 : 250051335 : *dst = out[1];
44 : : }
45 : :
46 : 250051335 : return len;
47 : : }
48 : :
49 : : size_t
50 : 250604110 : 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 : 250604110 : iter->count = count;
60 : :
61 [ + + ]: 752414150 : for (i = 0; i < count; i++) {
62 : 501810040 : it = &iter->iters[i];
63 : 501810040 : it->iov = iov[i];
64 : 501810040 : it->iovcnt = iovcnt[i];
65 : 501810040 : it->idx = 0;
66 : 501810040 : it->iov_len = iov[i][0].iov_len;
67 : 501810040 : it->iov_base = iov[i][0].iov_base;
68 : : }
69 : :
70 : 250604110 : return spdk_ioviter_nextv(iter, out);
71 : : }
72 : :
73 : : size_t
74 : 285359304 : spdk_ioviter_next(struct spdk_ioviter *iter, void **src, void **dst)
75 : : {
76 : 69861080 : void *out[2];
77 : : size_t len;
78 : :
79 : 285359304 : len = spdk_ioviter_nextv(iter, out);
80 : :
81 [ + + ]: 285359304 : if (len > 0) {
82 : 35307968 : *src = out[0];
83 : 35307968 : *dst = out[1];
84 : : }
85 : :
86 : 285359304 : return len;
87 : : }
88 : :
89 : : size_t
90 : 536167886 : 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 : 536167886 : len = UINT32_MAX;
98 [ + + ]: 1211213478 : for (i = 0; i < iter->count; i++) {
99 : 925096917 : it = &iter->iters[i];
100 [ + + + + ]: 925096917 : if (it->idx == it->iovcnt || it->iov_len == 0) {
101 : : /* This element has 0 bytes remaining, so we're done. */
102 : 250051335 : return 0;
103 : : }
104 : :
105 : 675045592 : len = spdk_min(len, it->iov_len);
106 : : }
107 : :
108 [ + + ]: 859422063 : for (i = 0; i < iter->count; i++) {
109 : 573305502 : it = &iter->iters[i];
110 : :
111 : 573305502 : out[i] = it->iov_base;
112 : :
113 [ + + ]: 573305502 : if (it->iov_len == len) {
114 : : /* Advance to next element */
115 : 432479894 : it->idx++;
116 [ + + ]: 432479894 : if (it->idx != it->iovcnt) {
117 : : /* Set up for next element */
118 : 61567074 : it->iov_len = it->iov[it->idx].iov_len;
119 : 61567074 : it->iov_base = it->iov[it->idx].iov_base;
120 : : }
121 : : } else {
122 : : /* Partial buffer */
123 : 140825640 : it->iov_base += len;
124 : 140825640 : it->iov_len -= len;
125 : : }
126 : : }
127 : :
128 : 286116561 : return len;
129 : : }
130 : :
131 : : size_t
132 : 128528656 : spdk_iovcpy(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
133 : : {
134 : 1500968 : struct spdk_ioviter iter;
135 : : size_t len, total_sz;
136 : 1500968 : void *src, *dst;
137 : :
138 : 128528656 : total_sz = 0;
139 [ + + ]: 128656772 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
140 [ + + ]: 259925814 : len != 0;
141 : 131562361 : len = spdk_ioviter_next(&iter, &src, &dst)) {
142 [ - + - + ]: 131562361 : memcpy(dst, src, len);
143 : 131562361 : total_sz += len;
144 : : }
145 : :
146 : 128528656 : 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 : 4627602 : spdk_iov_xfer_init(struct spdk_iov_xfer *ix, struct iovec *iovs, int iovcnt)
169 : : {
170 : 4627602 : ix->iovs = iovs;
171 : 4627602 : ix->iovcnt = iovcnt;
172 : 4627602 : ix->cur_iov_idx = 0;
173 : 4627602 : ix->cur_iov_offset = 0;
174 : 4627602 : }
175 : :
176 : : static size_t
177 : 4628062 : iov_xfer(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len, bool to_buf)
178 : : {
179 : 4628062 : size_t len, iov_remain_len, copied_len = 0;
180 : : struct iovec *iov;
181 : :
182 [ - + ]: 4628062 : if (buf_len == 0) {
183 : 0 : return 0;
184 : : }
185 : :
186 [ + + ]: 13682786 : while (ix->cur_iov_idx < ix->iovcnt) {
187 : 9155618 : iov = &ix->iovs[ix->cur_iov_idx];
188 : 9155618 : iov_remain_len = iov->iov_len - ix->cur_iov_offset;
189 [ + + ]: 9155618 : if (iov_remain_len == 0) {
190 : 4527365 : ix->cur_iov_idx++;
191 : 4527365 : ix->cur_iov_offset = 0;
192 : 4527365 : continue;
193 : : }
194 : :
195 : 4628253 : len = spdk_min(iov_remain_len, buf_len - copied_len);
196 : :
197 [ + + ]: 4628253 : if (to_buf) {
198 [ - + - + ]: 106943 : memcpy((char *)buf + copied_len,
199 : 106943 : (char *)iov->iov_base + ix->cur_iov_offset, len);
200 : : } else {
201 [ - + - + ]: 4521310 : memcpy((char *)iov->iov_base + ix->cur_iov_offset,
202 : : (const char *)buf + copied_len, len);
203 : : }
204 : 4628253 : copied_len += len;
205 : 4628253 : ix->cur_iov_offset += len;
206 : :
207 [ + + ]: 4628253 : if (buf_len == copied_len) {
208 : 100894 : return copied_len;
209 : : }
210 : : }
211 : :
212 : 4527168 : return copied_len;
213 : : }
214 : :
215 : : size_t
216 : 4521197 : spdk_iov_xfer_from_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
217 : : {
218 : 4521197 : return iov_xfer(ix, buf, buf_len, false);
219 : : }
220 : :
221 : : size_t
222 : 106865 : spdk_iov_xfer_to_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
223 : : {
224 : 106865 : return iov_xfer(ix, buf, buf_len, true);
225 : : }
226 : :
227 : : void
228 : 23486 : spdk_copy_iovs_to_buf(void *buf, size_t buf_len, struct iovec *iovs, int iovcnt)
229 : : {
230 : 23452 : struct spdk_iov_xfer ix;
231 : :
232 : 23486 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
233 : 23486 : spdk_iov_xfer_to_buf(&ix, buf, buf_len);
234 : 23486 : }
235 : :
236 : : void
237 : 4509219 : spdk_copy_buf_to_iovs(struct iovec *iovs, int iovcnt, void *buf, size_t buf_len)
238 : : {
239 : 4508446 : struct spdk_iov_xfer ix;
240 : :
241 : 4509219 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
242 : 4509219 : spdk_iov_xfer_from_buf(&ix, buf, buf_len);
243 : 4509219 : }
|