xref: /freebsd/sys/crypto/openssl/ossl_aes.c (revision af23369a6deaaeb612ab266eb88b8bb8d560c322)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2021 Stormshield.
5  * Copyright (c) 2021 Semihalf.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
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  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
30 
31 #include <sys/param.h>
32 #include <sys/malloc.h>
33 
34 #include <opencrypto/cryptodev.h>
35 
36 #include <crypto/openssl/ossl.h>
37 #include <crypto/openssl/ossl_cipher.h>
38 
39 #if defined(__amd64__) || defined(__i386__)
40 #include <crypto/openssl/ossl_x86.h>
41 #elif defined (__aarch64__)
42 #include <crypto/openssl/ossl_aarch64.h>
43 #endif
44 
45 static ossl_cipher_process_t ossl_aes_cbc;
46 
47 struct ossl_cipher ossl_cipher_aes_cbc = {
48 	.type = CRYPTO_AES_CBC,
49 	.blocksize = AES_BLOCK_LEN,
50 	.ivsize = AES_BLOCK_LEN,
51 
52 	/* Filled during initialization based on CPU caps. */
53 	.set_encrypt_key = NULL,
54 	.set_decrypt_key = NULL,
55 	.process = ossl_aes_cbc
56 };
57 
58 static int
59 ossl_aes_cbc(struct ossl_session_cipher *s, struct cryptop *crp,
60     const struct crypto_session_params *csp)
61 {
62 	struct crypto_buffer_cursor cc_in, cc_out;
63 	unsigned char block[EALG_MAX_BLOCK_LEN];
64 	unsigned char iv[EALG_MAX_BLOCK_LEN];
65 	const unsigned char *in, *inseg;
66 	unsigned char *out, *outseg;
67 	size_t plen, seglen, inlen, outlen;
68 	struct ossl_cipher_context key;
69 	struct ossl_cipher *cipher;
70 	int blocklen, error;
71 	bool encrypt;
72 
73 	cipher = s->cipher;
74 	encrypt = CRYPTO_OP_IS_ENCRYPT(crp->crp_op);
75 	plen = crp->crp_payload_length;
76 	blocklen = cipher->blocksize;
77 
78 	if (plen % blocklen)
79 		return (EINVAL);
80 
81 	if (crp->crp_cipher_key != NULL) {
82 		if (encrypt)
83 			error = cipher->set_encrypt_key(crp->crp_cipher_key,
84 			    8 * csp->csp_cipher_klen, &key);
85 		else
86 			error = cipher->set_decrypt_key(crp->crp_cipher_key,
87 			    8 * csp->csp_cipher_klen, &key);
88 		if (error)
89 			return (error);
90 	} else {
91 		if (encrypt)
92 			key = s->enc_ctx;
93 		else
94 			key = s->dec_ctx;
95 	}
96 
97 	crypto_read_iv(crp, iv);
98 
99 	/* Derived from ossl_chacha20.c */
100 	crypto_cursor_init(&cc_in, &crp->crp_buf);
101 	crypto_cursor_advance(&cc_in, crp->crp_payload_start);
102 	inseg = crypto_cursor_segment(&cc_in, &inlen);
103 	if (CRYPTO_HAS_OUTPUT_BUFFER(crp)) {
104 		crypto_cursor_init(&cc_out, &crp->crp_obuf);
105 		crypto_cursor_advance(&cc_out, crp->crp_payload_output_start);
106 	} else {
107 		cc_out = cc_in;
108 	}
109 	outseg = crypto_cursor_segment(&cc_out, &outlen);
110 
111 	while (plen >= blocklen) {
112 		if (inlen < blocklen) {
113 			crypto_cursor_copydata(&cc_in, blocklen, block);
114 			in = block;
115 			inlen = blocklen;
116 		} else {
117 			in = inseg;
118 		}
119 		if (outlen < blocklen) {
120 			out = block;
121 			outlen = blocklen;
122 		} else {
123 			out = outseg;
124 		}
125 
126 		/* Figure out how many blocks we can encrypt/decrypt at once. */
127 		seglen = rounddown(MIN(plen, MIN(inlen, outlen)), blocklen);
128 
129 		AES_CBC_ENCRYPT(in, out, seglen, &key, iv, encrypt);
130 
131 		if (out == block) {
132 			crypto_cursor_copyback(&cc_out, blocklen, block);
133 			outseg = crypto_cursor_segment(&cc_out, &outlen);
134 		} else {
135 			crypto_cursor_advance(&cc_out, seglen);
136 			outseg += seglen;
137 			outlen -= seglen;
138 		}
139 		if (in == block) {
140 			inseg = crypto_cursor_segment(&cc_in, &inlen);
141 		} else {
142 			crypto_cursor_advance(&cc_in, seglen);
143 			inseg += seglen;
144 			inlen -= seglen;
145 		}
146 		plen -= seglen;
147 	}
148 
149 	explicit_bzero(block, sizeof(block));
150 	explicit_bzero(iv, sizeof(iv));
151 	explicit_bzero(&key, sizeof(key));
152 	return (0);
153 }
154