1 /* $OpenBSD: criov.c,v 1.9 2002/01/29 15:48:29 jason Exp $ */
2
3 /*-
4 * Copyright (c) 1999 Theo de Raadt
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #include <sys/cdefs.h>
31 __FBSDID("$FreeBSD$");
32
33 #include <sys/param.h>
34 #include <sys/systm.h>
35 #include <sys/proc.h>
36 #include <sys/errno.h>
37 #include <sys/malloc.h>
38 #include <sys/kernel.h>
39 #include <sys/mbuf.h>
40 #include <sys/uio.h>
41 #include <sys/limits.h>
42 #include <sys/lock.h>
43
44 #include <opencrypto/cryptodev.h>
45
46 /*
47 * This macro is only for avoiding code duplication, as we need to skip
48 * given number of bytes in the same way in three functions below.
49 */
50 #define CUIO_SKIP() do { \
51 KASSERT(off >= 0, ("%s: off %d < 0", __func__, off)); \
52 KASSERT(len >= 0, ("%s: len %d < 0", __func__, len)); \
53 while (off > 0) { \
54 KASSERT(iol >= 0, ("%s: empty in skip", __func__)); \
55 if (off < iov->iov_len) \
56 break; \
57 off -= iov->iov_len; \
58 iol--; \
59 iov++; \
60 } \
61 } while (0)
62
63 void
64 cuio_copydata(struct uio* uio, int off, int len, caddr_t cp)
65 {
66 struct iovec *iov = uio->uio_iov;
67 int iol __diagused = uio->uio_iovcnt;
68 unsigned count;
69
70 CUIO_SKIP();
71 while (len > 0) {
72 KASSERT(iol >= 0, ("%s: empty", __func__));
73 count = min(iov->iov_len - off, len);
74 bcopy(((caddr_t)iov->iov_base) + off, cp, count);
75 len -= count;
76 cp += count;
77 off = 0;
78 iol--;
79 iov++;
80 }
81 }
82
83 void
84 cuio_copyback(struct uio* uio, int off, int len, c_caddr_t cp)
85 {
86 struct iovec *iov = uio->uio_iov;
87 int iol __diagused = uio->uio_iovcnt;
88 unsigned count;
89
90 CUIO_SKIP();
91 while (len > 0) {
92 KASSERT(iol >= 0, ("%s: empty", __func__));
93 count = min(iov->iov_len - off, len);
94 bcopy(cp, ((caddr_t)iov->iov_base) + off, count);
95 len -= count;
96 cp += count;
97 off = 0;
98 iol--;
99 iov++;
100 }
101 }
102
103 /*
104 * Return the index and offset of location in iovec list.
105 */
106 int
107 cuio_getptr(struct uio *uio, int loc, int *off)
108 {
109 int ind, len;
110
111 ind = 0;
112 while (loc >= 0 && ind < uio->uio_iovcnt) {
113 len = uio->uio_iov[ind].iov_len;
114 if (len > loc) {
115 *off = loc;
116 return (ind);
117 }
118 loc -= len;
119 ind++;
120 }
121
122 if (ind > 0 && loc == 0) {
123 ind--;
124 *off = uio->uio_iov[ind].iov_len;
125 return (ind);
126 }
127
128 return (-1);
129 }
130
131 /*
132 * Apply function f to the data in an iovec list starting "off" bytes from
133 * the beginning, continuing for "len" bytes.
134 */
135 int
136 cuio_apply(struct uio *uio, int off, int len, int (*f)(void *, void *, u_int),
137 void *arg)
138 {
139 struct iovec *iov = uio->uio_iov;
140 int iol __diagused = uio->uio_iovcnt;
141 unsigned count;
142 int rval;
143
144 CUIO_SKIP();
145 while (len > 0) {
146 KASSERT(iol >= 0, ("%s: empty", __func__));
147 count = min(iov->iov_len - off, len);
148 rval = (*f)(arg, ((caddr_t)iov->iov_base) + off, count);
149 if (rval)
150 return (rval);
151 len -= count;
152 off = 0;
153 iol--;
154 iov++;
155 }
156 return (0);
157 }
158
159 void
160 crypto_copyback(int flags, caddr_t buf, int off, int size, c_caddr_t in)
161 {
162
163 if ((flags & CRYPTO_F_IMBUF) != 0)
164 m_copyback((struct mbuf *)buf, off, size, in);
165 else if ((flags & CRYPTO_F_IOV) != 0)
166 cuio_copyback((struct uio *)buf, off, size, in);
167 else
168 bcopy(in, buf + off, size);
169 }
170
171 void
172 crypto_copydata(int flags, caddr_t buf, int off, int size, caddr_t out)
173 {
174
175 if ((flags & CRYPTO_F_IMBUF) != 0)
176 m_copydata((struct mbuf *)buf, off, size, out);
177 else if ((flags & CRYPTO_F_IOV) != 0)
178 cuio_copydata((struct uio *)buf, off, size, out);
179 else
180 bcopy(buf + off, out, size);
181 }
182
183 int
184 crypto_apply(int flags, caddr_t buf, int off, int len,
185 int (*f)(void *, void *, u_int), void *arg)
186 {
187 int error;
188
189 if ((flags & CRYPTO_F_IMBUF) != 0)
190 error = m_apply((struct mbuf *)buf, off, len, f, arg);
191 else if ((flags & CRYPTO_F_IOV) != 0)
192 error = cuio_apply((struct uio *)buf, off, len, f, arg);
193 else
194 error = (*f)(arg, buf + off, len);
195 return (error);
196 }
197
198 int
199 crypto_mbuftoiov(struct mbuf *mbuf, struct iovec **iovptr, int *cnt,
200 int *allocated)
201 {
202 struct iovec *iov;
203 struct mbuf *m, *mtmp;
204 int i, j;
205
206 *allocated = 0;
207 iov = *iovptr;
208 if (iov == NULL)
209 *cnt = 0;
210
211 m = mbuf;
212 i = 0;
213 while (m != NULL) {
214 if (i == *cnt) {
215 /* we need to allocate a larger array */
216 j = 1;
217 mtmp = m;
218 while ((mtmp = mtmp->m_next) != NULL)
219 j++;
220 iov = malloc(sizeof *iov * (i + j), M_CRYPTO_DATA,
221 M_NOWAIT);
222 if (iov == NULL)
223 return ENOMEM;
224 *allocated = 1;
225 *cnt = i + j;
226 memcpy(iov, *iovptr, sizeof *iov * i);
227 }
228
229 iov[i].iov_base = m->m_data;
230 iov[i].iov_len = m->m_len;
231
232 i++;
233 m = m->m_next;
234 }
235
236 if (*allocated)
237 KASSERT(*cnt == i, ("did not allocate correct amount: %d != %d",
238 *cnt, i));
239
240 *iovptr = iov;
241 *cnt = i;
242 return 0;
243 }
244
245 static inline void *
246 m_contiguous_subsegment(struct mbuf *m, size_t skip, size_t len)
247 {
248 int rel_off;
249
250 MPASS(skip <= INT_MAX);
251
252 m = m_getptr(m, (int)skip, &rel_off);
253 if (m == NULL)
254 return (NULL);
255
256 MPASS(rel_off >= 0);
257 skip = rel_off;
258 if (skip + len > m->m_len)
259 return (NULL);
260
261 return (mtod(m, char*) + skip);
262 }
263
264 static inline void *
265 cuio_contiguous_segment(struct uio *uio, size_t skip, size_t len)
266 {
267 int rel_off, idx;
268
269 MPASS(skip <= INT_MAX);
270 idx = cuio_getptr(uio, (int)skip, &rel_off);
271 if (idx < 0)
272 return (NULL);
273
274 MPASS(rel_off >= 0);
275 skip = rel_off;
276 if (skip + len > uio->uio_iov[idx].iov_len)
277 return (NULL);
278 return ((char *)uio->uio_iov[idx].iov_base + skip);
279 }
280
281 void *
282 crypto_contiguous_subsegment(int crp_flags, void *crpbuf,
283 size_t skip, size_t len)
284 {
285 if ((crp_flags & CRYPTO_F_IMBUF) != 0)
286 return (m_contiguous_subsegment(crpbuf, skip, len));
287 else if ((crp_flags & CRYPTO_F_IOV) != 0)
288 return (cuio_contiguous_segment(crpbuf, skip, len));
289 else {
290 MPASS((crp_flags & (CRYPTO_F_IMBUF | CRYPTO_F_IOV)) !=
291 (CRYPTO_F_IMBUF | CRYPTO_F_IOV));
292 return ((char*)crpbuf + skip);
293 }
294 }
295
Cache object: 9da922586d9c3fef6414894caa7ee09a
|