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 : 15455 : spdk_iov_memset(struct iovec *iovs, int iovcnt, int c)
11 : : {
12 : 15455 : int iov_idx = 0;
13 : : struct iovec *iov;
14 : :
15 [ + + ]: 30930 : while (iov_idx < iovcnt) {
16 : 15475 : iov = &iovs[iov_idx];
17 [ - + ]: 15475 : memset(iov->iov_base, c, iov->iov_len);
18 : 15475 : iov_idx++;
19 : : }
20 : 15455 : }
21 : :
22 : : size_t
23 : 246814463 : 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 : 72980393 : struct iovec *iovs[2];
29 : 72980393 : size_t iovcnts[2];
30 : 72980393 : void *out[2];
31 : : size_t len;
32 : :
33 : 246814463 : iovs[0] = siov;
34 : 246814463 : iovcnts[0] = siovcnt;
35 : :
36 : 246814463 : iovs[1] = diov;
37 : 246814463 : iovcnts[1] = diovcnt;
38 : :
39 : 246814463 : len = spdk_ioviter_firstv(iter, 2, iovs, iovcnts, out);
40 : :
41 [ + - ]: 246814463 : if (len > 0) {
42 : 246814463 : *src = out[0];
43 : 246814463 : *dst = out[1];
44 : : }
45 : :
46 : 246814463 : return len;
47 : : }
48 : :
49 : : size_t
50 : 247450378 : 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 : 247450378 : iter->count = count;
60 : :
61 [ + + ]: 743035959 : for (i = 0; i < count; i++) {
62 : 495585581 : it = &iter->iters[i];
63 : 495585581 : it->iov = iov[i];
64 : 495585581 : it->iovcnt = iovcnt[i];
65 : 495585581 : it->idx = 0;
66 : 495585581 : it->iov_len = iov[i][0].iov_len;
67 : 495585581 : it->iov_base = iov[i][0].iov_base;
68 : : }
69 : :
70 : 247450378 : return spdk_ioviter_nextv(iter, out);
71 : : }
72 : :
73 : : size_t
74 : 281382355 : spdk_ioviter_next(struct spdk_ioviter *iter, void **src, void **dst)
75 : : {
76 : 75938011 : void *out[2];
77 : : size_t len;
78 : :
79 : 281382355 : len = spdk_ioviter_nextv(iter, out);
80 : :
81 [ + + ]: 281382355 : if (len > 0) {
82 : 34567871 : *src = out[0];
83 : 34567871 : *dst = out[1];
84 : : }
85 : :
86 : 281382355 : return len;
87 : : }
88 : :
89 : : size_t
90 : 529037155 : 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 : 529037155 : len = UINT32_MAX;
98 [ + + ]: 1192193369 : for (i = 0; i < iter->count; i++) {
99 : 909970777 : it = &iter->iters[i];
100 [ + + + + ]: 909970777 : if (it->idx == it->iovcnt || it->iov_len == 0) {
101 : : /* This element has 0 bytes remaining, so we're done. */
102 : 246814463 : return 0;
103 : : }
104 : :
105 : 663156214 : len = spdk_min(len, it->iov_len);
106 : : }
107 : :
108 [ + + ]: 847823461 : for (i = 0; i < iter->count; i++) {
109 : 565600869 : it = &iter->iters[i];
110 : :
111 : 565600869 : out[i] = it->iov_base;
112 : :
113 [ + + ]: 565600869 : if (it->iov_len == len) {
114 : : /* Advance to next element */
115 : 429833699 : it->idx++;
116 [ + + ]: 429833699 : if (it->idx != it->iovcnt) {
117 : : /* Set up for next element */
118 : 58071359 : it->iov_len = it->iov[it->idx].iov_len;
119 : 58071359 : it->iov_base = it->iov[it->idx].iov_base;
120 : : }
121 : : } else {
122 : : /* Partial buffer */
123 : 135767128 : it->iov_base += len;
124 : 135767128 : it->iov_len -= len;
125 : : }
126 : : }
127 : :
128 : 282222692 : return len;
129 : : }
130 : :
131 : : size_t
132 : 121585844 : spdk_iovcpy(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
133 : : {
134 : 1428552 : struct spdk_ioviter iter;
135 : : size_t len, total_sz;
136 : 1428552 : void *src, *dst;
137 : :
138 : 121585844 : total_sz = 0;
139 [ + + ]: 121713960 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
140 [ + + ]: 246035048 : len != 0;
141 : 124614407 : len = spdk_ioviter_next(&iter, &src, &dst)) {
142 [ - + - + ]: 124614407 : memcpy(dst, src, len);
143 : 124614407 : total_sz += len;
144 : : }
145 : :
146 : 121585844 : 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 : 5046408 : spdk_iov_xfer_init(struct spdk_iov_xfer *ix, struct iovec *iovs, int iovcnt)
169 : : {
170 : 5046408 : ix->iovs = iovs;
171 : 5046408 : ix->iovcnt = iovcnt;
172 : 5046408 : ix->cur_iov_idx = 0;
173 : 5046408 : ix->cur_iov_offset = 0;
174 : 5046408 : }
175 : :
176 : : static size_t
177 : 5046870 : iov_xfer(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len, bool to_buf)
178 : : {
179 : 5046870 : size_t len, iov_remain_len, copied_len = 0;
180 : : struct iovec *iov;
181 : :
182 [ - + ]: 5046870 : if (buf_len == 0) {
183 : 0 : return 0;
184 : : }
185 : :
186 [ + + ]: 14935294 : while (ix->cur_iov_idx < ix->iovcnt) {
187 : 9991276 : iov = &ix->iovs[ix->cur_iov_idx];
188 : 9991276 : iov_remain_len = iov->iov_len - ix->cur_iov_offset;
189 [ + + ]: 9991276 : if (iov_remain_len == 0) {
190 : 4944215 : ix->cur_iov_idx++;
191 : 4944215 : ix->cur_iov_offset = 0;
192 : 4944215 : continue;
193 : : }
194 : :
195 : 5047061 : len = spdk_min(iov_remain_len, buf_len - copied_len);
196 : :
197 [ + + ]: 5047061 : if (to_buf) {
198 [ - + - + ]: 85953 : memcpy((char *)buf + copied_len,
199 : 85953 : (char *)iov->iov_base + ix->cur_iov_offset, len);
200 : : } else {
201 [ - + - + ]: 4961108 : memcpy((char *)iov->iov_base + ix->cur_iov_offset,
202 : : (const char *)buf + copied_len, len);
203 : : }
204 : 5047061 : copied_len += len;
205 : 5047061 : ix->cur_iov_offset += len;
206 : :
207 [ + + ]: 5047061 : if (buf_len == copied_len) {
208 : 102852 : return copied_len;
209 : : }
210 : : }
211 : :
212 : 4944018 : return copied_len;
213 : : }
214 : :
215 : : size_t
216 : 4960995 : spdk_iov_xfer_from_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
217 : : {
218 : 4960995 : return iov_xfer(ix, buf, buf_len, false);
219 : : }
220 : :
221 : : size_t
222 : 85875 : spdk_iov_xfer_to_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
223 : : {
224 : 85875 : return iov_xfer(ix, buf, buf_len, true);
225 : : }
226 : :
227 : : void
228 : 727 : spdk_copy_iovs_to_buf(void *buf, size_t buf_len, struct iovec *iovs, int iovcnt)
229 : : {
230 : 693 : struct spdk_iov_xfer ix;
231 : :
232 : 727 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
233 : 727 : spdk_iov_xfer_to_buf(&ix, buf, buf_len);
234 : 727 : }
235 : :
236 : : void
237 : 4949135 : spdk_copy_buf_to_iovs(struct iovec *iovs, int iovcnt, void *buf, size_t buf_len)
238 : : {
239 : 4948362 : struct spdk_iov_xfer ix;
240 : :
241 : 4949135 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
242 : 4949135 : spdk_iov_xfer_from_buf(&ix, buf, buf_len);
243 : 4949135 : }
|