xref: /illumos-gate/usr/src/common/crypto/modes/modes.c (revision ef2504f26d1ea5859db9838255bb63f488f1b050)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 #ifndef _KERNEL
27 #include <stdlib.h>
28 #endif
29 
30 #include <sys/strsun.h>
31 #include <sys/types.h>
32 #include <modes/modes.h>
33 #include <sys/crypto/common.h>
34 #include <sys/crypto/impl.h>
35 
36 /*
37  * Initialize by setting iov_or_mp to point to the current iovec or mp,
38  * and by setting current_offset to an offset within the current iovec or mp.
39  */
40 void
41 crypto_init_ptrs(crypto_data_t *out, void **iov_or_mp, offset_t *current_offset)
42 {
43 	offset_t offset;
44 
45 	switch (out->cd_format) {
46 	case CRYPTO_DATA_RAW:
47 		*current_offset = out->cd_offset;
48 		break;
49 
50 	case CRYPTO_DATA_UIO: {
51 		uio_t *uiop = out->cd_uio;
52 		uintptr_t vec_idx;
53 
54 		offset = out->cd_offset;
55 		for (vec_idx = 0; vec_idx < uiop->uio_iovcnt &&
56 		    offset >= uiop->uio_iov[vec_idx].iov_len;
57 		    offset -= uiop->uio_iov[vec_idx++].iov_len)
58 			;
59 
60 		*current_offset = offset;
61 		*iov_or_mp = (void *)vec_idx;
62 		break;
63 	}
64 
65 	case CRYPTO_DATA_MBLK: {
66 		mblk_t *mp;
67 
68 		offset = out->cd_offset;
69 		for (mp = out->cd_mp; mp != NULL && offset >= MBLKL(mp);
70 		    offset -= MBLKL(mp), mp = mp->b_cont)
71 			;
72 
73 		*current_offset = offset;
74 		*iov_or_mp = mp;
75 		break;
76 
77 	}
78 	} /* end switch */
79 }
80 
81 /*
82  * Get pointers for where in the output to copy a block of encrypted or
83  * decrypted data.  The iov_or_mp argument stores a pointer to the current
84  * iovec or mp, and offset stores an offset into the current iovec or mp.
85  */
86 void
87 crypto_get_ptrs(crypto_data_t *out, void **iov_or_mp, offset_t *current_offset,
88     uint8_t **out_data_1, size_t *out_data_1_len, uint8_t **out_data_2,
89     size_t amt)
90 {
91 	offset_t offset;
92 
93 	switch (out->cd_format) {
94 	case CRYPTO_DATA_RAW: {
95 		iovec_t *iov;
96 
97 		offset = *current_offset;
98 		iov = &out->cd_raw;
99 		if ((offset + amt) <= iov->iov_len) {
100 			/* one block fits */
101 			*out_data_1 = (uint8_t *)iov->iov_base + offset;
102 			*out_data_1_len = amt;
103 			*out_data_2 = NULL;
104 			*current_offset = offset + amt;
105 		}
106 		break;
107 	}
108 
109 	case CRYPTO_DATA_UIO: {
110 		uio_t *uio = out->cd_uio;
111 		iovec_t *iov;
112 		offset_t offset;
113 		uintptr_t vec_idx;
114 		uint8_t *p;
115 
116 		offset = *current_offset;
117 		vec_idx = (uintptr_t)(*iov_or_mp);
118 		iov = &uio->uio_iov[vec_idx];
119 		p = (uint8_t *)iov->iov_base + offset;
120 		*out_data_1 = p;
121 
122 		if (offset + amt <= iov->iov_len) {
123 			/* can fit one block into this iov */
124 			*out_data_1_len = amt;
125 			*out_data_2 = NULL;
126 			*current_offset = offset + amt;
127 		} else {
128 			/* one block spans two iovecs */
129 			*out_data_1_len = iov->iov_len - offset;
130 			if (vec_idx == uio->uio_iovcnt)
131 				return;
132 			vec_idx++;
133 			iov = &uio->uio_iov[vec_idx];
134 			*out_data_2 = (uint8_t *)iov->iov_base;
135 			*current_offset = amt - *out_data_1_len;
136 		}
137 		*iov_or_mp = (void *)vec_idx;
138 		break;
139 	}
140 
141 	case CRYPTO_DATA_MBLK: {
142 		mblk_t *mp;
143 		uint8_t *p;
144 
145 		offset = *current_offset;
146 		mp = (mblk_t *)*iov_or_mp;
147 		p = mp->b_rptr + offset;
148 		*out_data_1 = p;
149 		if ((p + amt) <= mp->b_wptr) {
150 			/* can fit one block into this mblk */
151 			*out_data_1_len = amt;
152 			*out_data_2 = NULL;
153 			*current_offset = offset + amt;
154 		} else {
155 			/* one block spans two mblks */
156 			*out_data_1_len = mp->b_wptr - p;
157 			if ((mp = mp->b_cont) == NULL)
158 				return;
159 			*out_data_2 = mp->b_rptr;
160 			*current_offset = (amt - *out_data_1_len);
161 		}
162 		*iov_or_mp = mp;
163 		break;
164 	}
165 	} /* end switch */
166 }
167 
168 void
169 crypto_free_mode_ctx(void *ctx)
170 {
171 	common_ctx_t *common_ctx = (common_ctx_t *)ctx;
172 
173 	switch (common_ctx->cc_flags &
174 	    (ECB_MODE|CBC_MODE|CTR_MODE|CCM_MODE|GCM_MODE|GMAC_MODE)) {
175 	case ECB_MODE:
176 #ifdef _KERNEL
177 		kmem_free(common_ctx, sizeof (ecb_ctx_t));
178 #else
179 		free(common_ctx);
180 #endif
181 		break;
182 
183 	case CBC_MODE:
184 #ifdef _KERNEL
185 		kmem_free(common_ctx, sizeof (cbc_ctx_t));
186 #else
187 		free(common_ctx);
188 #endif
189 		break;
190 
191 	case CTR_MODE:
192 #ifdef _KERNEL
193 		kmem_free(common_ctx, sizeof (ctr_ctx_t));
194 #else
195 		free(common_ctx);
196 #endif
197 		break;
198 
199 	case CCM_MODE:
200 #ifdef _KERNEL
201 		if (((ccm_ctx_t *)ctx)->ccm_pt_buf != NULL)
202 			kmem_free(((ccm_ctx_t *)ctx)->ccm_pt_buf,
203 			    ((ccm_ctx_t *)ctx)->ccm_data_len);
204 
205 		kmem_free(ctx, sizeof (ccm_ctx_t));
206 #else
207 		if (((ccm_ctx_t *)ctx)->ccm_pt_buf != NULL)
208 			free(((ccm_ctx_t *)ctx)->ccm_pt_buf);
209 		free(ctx);
210 #endif
211 		break;
212 
213 	case GCM_MODE:
214 	case GMAC_MODE:
215 #ifdef _KERNEL
216 		if (((gcm_ctx_t *)ctx)->gcm_pt_buf != NULL)
217 			kmem_free(((gcm_ctx_t *)ctx)->gcm_pt_buf,
218 			    ((gcm_ctx_t *)ctx)->gcm_pt_buf_len);
219 
220 		kmem_free(ctx, sizeof (gcm_ctx_t));
221 #else
222 		if (((gcm_ctx_t *)ctx)->gcm_pt_buf != NULL)
223 			free(((gcm_ctx_t *)ctx)->gcm_pt_buf);
224 		free(ctx);
225 #endif
226 	}
227 }
228