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 : 19599 : spdk_iov_memset(struct iovec *iovs, int iovcnt, int c)
11 : : {
12 : 19599 : int iov_idx = 0;
13 : : struct iovec *iov;
14 : :
15 [ + + ]: 39212 : while (iov_idx < iovcnt) {
16 : 19613 : iov = &iovs[iov_idx];
17 [ - + ]: 19613 : memset(iov->iov_base, c, iov->iov_len);
18 : 19613 : iov_idx++;
19 : : }
20 : 19599 : }
21 : :
22 : : size_t
23 : 207636054 : 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 : 207636054 : struct iovec *iovs[2];
29 : 207636054 : size_t iovcnts[2];
30 : 207636054 : void *out[2];
31 : : size_t len;
32 : :
33 : 207636054 : iovs[0] = siov;
34 : 207636054 : iovcnts[0] = siovcnt;
35 : :
36 : 207636054 : iovs[1] = diov;
37 : 207636054 : iovcnts[1] = diovcnt;
38 : :
39 : 207636054 : len = spdk_ioviter_firstv(iter, 2, iovs, iovcnts, out);
40 : :
41 [ + - ]: 207636054 : if (len > 0) {
42 : 207636054 : *src = out[0];
43 : 207636054 : *dst = out[1];
44 : : }
45 : :
46 : 207636054 : return len;
47 : : }
48 : :
49 : : size_t
50 : 208227237 : 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 : 208227237 : iter->count = count;
60 : :
61 [ + + ]: 625299527 : for (i = 0; i < count; i++) {
62 : 417072390 : it = &iter->iters[i];
63 : 417072390 : it->iov = iov[i];
64 : 417072390 : it->iovcnt = iovcnt[i];
65 : 417072390 : it->idx = 0;
66 : 417072390 : it->iov_len = iov[i][0].iov_len;
67 : 417072390 : it->iov_base = iov[i][0].iov_base;
68 : : }
69 : :
70 : 208227237 : return spdk_ioviter_nextv(iter, out);
71 : : }
72 : :
73 : : size_t
74 : 241860746 : spdk_ioviter_next(struct spdk_ioviter *iter, void **src, void **dst)
75 : : {
76 : 241860746 : void *out[2];
77 : : size_t len;
78 : :
79 : 241860746 : len = spdk_ioviter_nextv(iter, out);
80 : :
81 [ + + ]: 241860746 : if (len > 0) {
82 : 34224711 : *src = out[0];
83 : 34224711 : *dst = out[1];
84 : : }
85 : :
86 : 241860746 : return len;
87 : : }
88 : :
89 : : size_t
90 : 450190201 : 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 : 450190201 : len = UINT32_MAX;
98 [ + + ]: 1022630862 : for (i = 0; i < iter->count; i++) {
99 : 780076715 : it = &iter->iters[i];
100 [ + + + + ]: 780076715 : if (it->idx == it->iovcnt || it->iov_len == 0) {
101 : : /* This element has 0 bytes remaining, so we're done. */
102 : 207636054 : return 0;
103 : : }
104 : :
105 : 572440661 : len = spdk_min(len, it->iov_len);
106 : : }
107 : :
108 [ + + ]: 728515721 : for (i = 0; i < iter->count; i++) {
109 : 485961574 : it = &iter->iters[i];
110 : :
111 : 485961574 : out[i] = it->iov_base;
112 : :
113 [ + + ]: 485961574 : if (it->iov_len == len) {
114 : : /* Advance to next element */
115 : 362269473 : it->idx++;
116 [ + + ]: 362269473 : if (it->idx != it->iovcnt) {
117 : : /* Set up for next element */
118 : 53060761 : it->iov_len = it->iov[it->idx].iov_len;
119 : 53060761 : it->iov_base = it->iov[it->idx].iov_base;
120 : : }
121 : : } else {
122 : : /* Partial buffer */
123 : 123692077 : it->iov_base += len;
124 : 123692077 : it->iov_len -= len;
125 : : }
126 : : }
127 : :
128 : 242554147 : return len;
129 : : }
130 : :
131 : : size_t
132 : 106294328 : spdk_iovcpy(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
133 : : {
134 : 106294328 : struct spdk_ioviter iter;
135 : : size_t len, total_sz;
136 : 106294328 : void *src, *dst;
137 : :
138 : 106294328 : total_sz = 0;
139 [ + + ]: 106422444 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
140 [ + + ]: 214988872 : len != 0;
141 : 108859747 : len = spdk_ioviter_next(&iter, &src, &dst)) {
142 [ - + - + ]: 108859747 : memcpy(dst, src, len);
143 : 108859747 : total_sz += len;
144 : : }
145 : :
146 : 106294328 : return total_sz;
147 : : }
148 : :
149 : : size_t
150 : 140 : spdk_iovmove(struct iovec *siov, size_t siovcnt, struct iovec *diov, size_t diovcnt)
151 : : {
152 : 140 : struct spdk_ioviter iter;
153 : : size_t len, total_sz;
154 : 140 : void *src, *dst;
155 : :
156 : 140 : total_sz = 0;
157 [ + + ]: 210 : for (len = spdk_ioviter_first(&iter, siov, siovcnt, diov, diovcnt, &src, &dst);
158 [ + + ]: 140 : len != 0;
159 : 140 : len = spdk_ioviter_next(&iter, &src, &dst)) {
160 [ - + - + ]: 140 : memmove(dst, src, len);
161 : 140 : total_sz += len;
162 : : }
163 : :
164 : 140 : return total_sz;
165 : : }
166 : :
167 : : void
168 : 3586195 : spdk_iov_xfer_init(struct spdk_iov_xfer *ix, struct iovec *iovs, int iovcnt)
169 : : {
170 : 3586195 : ix->iovs = iovs;
171 : 3586195 : ix->iovcnt = iovcnt;
172 : 3586195 : ix->cur_iov_idx = 0;
173 : 3586195 : ix->cur_iov_offset = 0;
174 : 3586195 : }
175 : :
176 : : static size_t
177 : 3586649 : iov_xfer(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len, bool to_buf)
178 : : {
179 : 3586649 : size_t len, iov_remain_len, copied_len = 0;
180 : : struct iovec *iov;
181 : :
182 [ - + ]: 3586649 : if (buf_len == 0) {
183 : 0 : return 0;
184 : : }
185 : :
186 [ + + ]: 10471217 : while (ix->cur_iov_idx < ix->iovcnt) {
187 : 7029078 : iov = &ix->iovs[ix->cur_iov_idx];
188 : 7029078 : iov_remain_len = iov->iov_len - ix->cur_iov_offset;
189 [ + + ]: 7029078 : if (iov_remain_len == 0) {
190 : 3442286 : ix->cur_iov_idx++;
191 : 3442286 : ix->cur_iov_offset = 0;
192 : 3442286 : continue;
193 : : }
194 : :
195 : 3586792 : len = spdk_min(iov_remain_len, buf_len - copied_len);
196 : :
197 [ + + ]: 3586792 : if (to_buf) {
198 [ - + - + ]: 152687 : memcpy((char *)buf + copied_len,
199 : 152687 : (char *)iov->iov_base + ix->cur_iov_offset, len);
200 : : } else {
201 [ - + - + ]: 3434105 : memcpy((char *)iov->iov_base + ix->cur_iov_offset,
202 : : (const char *)buf + copied_len, len);
203 : : }
204 : 3586792 : copied_len += len;
205 : 3586792 : ix->cur_iov_offset += len;
206 : :
207 [ + + ]: 3586792 : if (buf_len == copied_len) {
208 : 144510 : return copied_len;
209 : : }
210 : : }
211 : :
212 : 3442139 : return copied_len;
213 : : }
214 : :
215 : : size_t
216 : 3434014 : spdk_iov_xfer_from_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
217 : : {
218 : 3434014 : return iov_xfer(ix, buf, buf_len, false);
219 : : }
220 : :
221 : : size_t
222 : 152635 : spdk_iov_xfer_to_buf(struct spdk_iov_xfer *ix, const void *buf, size_t buf_len)
223 : : {
224 : 152635 : return iov_xfer(ix, buf, buf_len, true);
225 : : }
226 : :
227 : : void
228 : 27663 : spdk_copy_iovs_to_buf(void *buf, size_t buf_len, struct iovec *iovs, int iovcnt)
229 : : {
230 : 27663 : struct spdk_iov_xfer ix;
231 : :
232 : 27663 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
233 : 27663 : spdk_iov_xfer_to_buf(&ix, buf, buf_len);
234 : 27663 : }
235 : :
236 : : void
237 : 3419294 : spdk_copy_buf_to_iovs(struct iovec *iovs, int iovcnt, void *buf, size_t buf_len)
238 : : {
239 : 3419294 : struct spdk_iov_xfer ix;
240 : :
241 : 3419294 : spdk_iov_xfer_init(&ix, iovs, iovcnt);
242 : 3419294 : spdk_iov_xfer_from_buf(&ix, buf, buf_len);
243 : 3419294 : }
|