177 	aes_ctx_t aes_ctx;  in aes_encrypt_atomic()  local
183 	memset(&aes_ctx, 0, sizeof (aes_ctx_t));  in aes_encrypt_atomic()
190 	ret = aes_common_init_ctx(&aes_ctx, template, mechanism, key,  in aes_encrypt_atomic()
197 		length_needed = plaintext->cd_length + aes_ctx.ac_mac_len;  in aes_encrypt_atomic()
200 		length_needed = plaintext->cd_length + aes_ctx.ac_tag_len;  in aes_encrypt_atomic()
221 		ret = crypto_update_iov(&aes_ctx, plaintext, ciphertext,  in aes_encrypt_atomic()
225 		ret = crypto_update_uio(&aes_ctx, plaintext, ciphertext,  in aes_encrypt_atomic()
234 			ret = ccm_encrypt_final((ccm_ctx_t *)&aes_ctx,  in aes_encrypt_atomic()
239 			ASSERT(aes_ctx.ac_remainder_len == 0);  in aes_encrypt_atomic()
241 			ret = gcm_encrypt_final((gcm_ctx_t *)&aes_ctx,  in aes_encrypt_atomic()
246 			ASSERT(aes_ctx.ac_remainder_len == 0);  in aes_encrypt_atomic()
248 			ASSERT(aes_ctx.ac_remainder_len == 0);  in aes_encrypt_atomic()
261 	if (aes_ctx.ac_flags & PROVIDER_OWNS_KEY_SCHEDULE) {  in aes_encrypt_atomic()
262 		memset(aes_ctx.ac_keysched, 0, aes_ctx.ac_keysched_len);  in aes_encrypt_atomic()
263 		kmem_free(aes_ctx.ac_keysched, aes_ctx.ac_keysched_len);  in aes_encrypt_atomic()
265 	if (aes_ctx.ac_flags & GCM_MODE) {  in aes_encrypt_atomic()
266 		gcm_clear_ctx((gcm_ctx_t *)&aes_ctx);  in aes_encrypt_atomic()
276 	aes_ctx_t aes_ctx;  in aes_decrypt_atomic()  local
282 	memset(&aes_ctx, 0, sizeof (aes_ctx_t));  in aes_decrypt_atomic()
289 	ret = aes_common_init_ctx(&aes_ctx, template, mechanism, key,  in aes_decrypt_atomic()
296 		length_needed = aes_ctx.ac_data_len;  in aes_decrypt_atomic()
299 		length_needed = ciphertext->cd_length - aes_ctx.ac_tag_len;  in aes_decrypt_atomic()
320 		ret = crypto_update_iov(&aes_ctx, ciphertext, plaintext,  in aes_decrypt_atomic()
324 		ret = crypto_update_uio(&aes_ctx, ciphertext, plaintext,  in aes_decrypt_atomic()
333 			ASSERT(aes_ctx.ac_processed_data_len  in aes_decrypt_atomic()
334 			    == aes_ctx.ac_data_len);  in aes_decrypt_atomic()
335 			ASSERT(aes_ctx.ac_processed_mac_len  in aes_decrypt_atomic()
336 			    == aes_ctx.ac_mac_len);  in aes_decrypt_atomic()
337 			ret = ccm_decrypt_final((ccm_ctx_t *)&aes_ctx,  in aes_decrypt_atomic()
340 			ASSERT(aes_ctx.ac_remainder_len == 0);  in aes_decrypt_atomic()
349 			ret = gcm_decrypt_final((gcm_ctx_t *)&aes_ctx,  in aes_decrypt_atomic()
352 			ASSERT(aes_ctx.ac_remainder_len == 0);  in aes_decrypt_atomic()
368 	if (aes_ctx.ac_flags & PROVIDER_OWNS_KEY_SCHEDULE) {  in aes_decrypt_atomic()
369 		memset(aes_ctx.ac_keysched, 0, aes_ctx.ac_keysched_len);  in aes_decrypt_atomic()
370 		kmem_free(aes_ctx.ac_keysched, aes_ctx.ac_keysched_len);  in aes_decrypt_atomic()
373 	if (aes_ctx.ac_flags & CCM_MODE) {  in aes_decrypt_atomic()
374 		if (aes_ctx.ac_pt_buf != NULL) {  in aes_decrypt_atomic()
375 			vmem_free(aes_ctx.ac_pt_buf, aes_ctx.ac_data_len);  in aes_decrypt_atomic()
377 	} else if (aes_ctx.ac_flags & GCM_MODE) {  in aes_decrypt_atomic()
378 		gcm_clear_ctx((gcm_ctx_t *)&aes_ctx);  in aes_decrypt_atomic()
423 	aes_ctx_t *aes_ctx = ctx->cc_provider_private;  in aes_free_context()  local
425 	if (aes_ctx != NULL) {  in aes_free_context()
426 		if (aes_ctx->ac_flags & PROVIDER_OWNS_KEY_SCHEDULE) {  in aes_free_context()
427 			ASSERT(aes_ctx->ac_keysched_len != 0);  in aes_free_context()
428 			memset(aes_ctx->ac_keysched, 0,  in aes_free_context()
429 			    aes_ctx->ac_keysched_len);  in aes_free_context()
430 			kmem_free(aes_ctx->ac_keysched,  in aes_free_context()
431 			    aes_ctx->ac_keysched_len);  in aes_free_context()
433 		crypto_free_mode_ctx(aes_ctx);  in aes_free_context()
442 aes_common_init_ctx(aes_ctx_t *aes_ctx, crypto_spi_ctx_template_t *template,  in aes_common_init_ctx()  argument
462 		aes_ctx->ac_flags |= PROVIDER_OWNS_KEY_SCHEDULE;  in aes_common_init_ctx()
463 		aes_ctx->ac_keysched_len = size;  in aes_common_init_ctx()
467 	aes_ctx->ac_keysched = keysched;  in aes_common_init_ctx()
475 		rv = ccm_init_ctx((ccm_ctx_t *)aes_ctx, mechanism->cm_param,  in aes_common_init_ctx()
484 		rv = gcm_init_ctx((gcm_ctx_t *)aes_ctx, mechanism->cm_param,  in aes_common_init_ctx()
491 		if (aes_ctx->ac_flags & PROVIDER_OWNS_KEY_SCHEDULE) {  in aes_common_init_ctx()