24b8fb17dfd4bc145f28948620fdfd96012246f5
[openssl.git] / providers / implementations / ciphers / cipher_aes_xts_hw.c
1 /*
2  * Copyright 2019-2023 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the Apache License 2.0 (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 /*
11  * This file uses the low level AES functions (which are deprecated for
12  * non-internal use) in order to implement provider AES ciphers.
13  */
14 #include "internal/deprecated.h"
15
16 #include "cipher_aes_xts.h"
17
18 #define XTS_SET_KEY_FN(fn_set_enc_key, fn_set_dec_key,                         \
19                        fn_block_enc, fn_block_dec,                             \
20                        fn_stream_enc, fn_stream_dec) {                         \
21     size_t bytes = keylen / 2;                                                 \
22     size_t bits = bytes * 8;                                                   \
23                                                                                \
24     if (ctx->enc) {                                                            \
25         fn_set_enc_key(key, bits, &xctx->ks1.ks);                              \
26         xctx->xts.block1 = (block128_f)fn_block_enc;                           \
27     } else {                                                                   \
28         fn_set_dec_key(key, bits, &xctx->ks1.ks);                              \
29         xctx->xts.block1 = (block128_f)fn_block_dec;                           \
30     }                                                                          \
31     fn_set_enc_key(key + bytes, bits, &xctx->ks2.ks);                          \
32     xctx->xts.block2 = (block128_f)fn_block_enc;                               \
33     xctx->xts.key1 = &xctx->ks1;                                               \
34     xctx->xts.key2 = &xctx->ks2;                                               \
35     xctx->stream = ctx->enc ? fn_stream_enc : fn_stream_dec;                   \
36 }
37
38 static int cipher_hw_aes_xts_generic_initkey(PROV_CIPHER_CTX *ctx,
39                                              const unsigned char *key,
40                                              size_t keylen)
41 {
42     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
43     OSSL_xts_stream_fn stream_enc = NULL;
44     OSSL_xts_stream_fn stream_dec = NULL;
45
46 #ifdef AES_XTS_ASM
47     stream_enc = AES_xts_encrypt;
48     stream_dec = AES_xts_decrypt;
49 #endif /* AES_XTS_ASM */
50
51 #ifdef HWAES_CAPABLE
52     if (HWAES_CAPABLE) {
53 # ifdef HWAES_xts_encrypt
54         stream_enc = HWAES_xts_encrypt;
55 # endif /* HWAES_xts_encrypt */
56 # ifdef HWAES_xts_decrypt
57         stream_dec = HWAES_xts_decrypt;
58 # endif /* HWAES_xts_decrypt */
59         XTS_SET_KEY_FN(HWAES_set_encrypt_key, HWAES_set_decrypt_key,
60                        HWAES_encrypt, HWAES_decrypt,
61                        stream_enc, stream_dec);
62         return 1;
63     } else
64 #endif /* HWAES_CAPABLE */
65
66 #ifdef BSAES_CAPABLE
67     if (BSAES_CAPABLE) {
68         stream_enc = ossl_bsaes_xts_encrypt;
69         stream_dec = ossl_bsaes_xts_decrypt;
70     } else
71 #endif /* BSAES_CAPABLE */
72 #ifdef VPAES_CAPABLE
73     if (VPAES_CAPABLE) {
74         XTS_SET_KEY_FN(vpaes_set_encrypt_key, vpaes_set_decrypt_key,
75                        vpaes_encrypt, vpaes_decrypt, stream_enc, stream_dec);
76         return 1;
77     } else
78 #endif /* VPAES_CAPABLE */
79     {
80         (void)0;
81     }
82     {
83         XTS_SET_KEY_FN(AES_set_encrypt_key, AES_set_decrypt_key,
84                        AES_encrypt, AES_decrypt, stream_enc, stream_dec);
85     }
86     return 1;
87 }
88
89 static void cipher_hw_aes_xts_copyctx(PROV_CIPHER_CTX *dst,
90                                       const PROV_CIPHER_CTX *src)
91 {
92     PROV_AES_XTS_CTX *sctx = (PROV_AES_XTS_CTX *)src;
93     PROV_AES_XTS_CTX *dctx = (PROV_AES_XTS_CTX *)dst;
94
95     *dctx = *sctx;
96     dctx->xts.key1 = &dctx->ks1.ks;
97     dctx->xts.key2 = &dctx->ks2.ks;
98 }
99
100 #if defined(AESNI_CAPABLE)
101
102 static int cipher_hw_aesni_xts_initkey(PROV_CIPHER_CTX *ctx,
103                                        const unsigned char *key, size_t keylen)
104 {
105     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
106
107     XTS_SET_KEY_FN(aesni_set_encrypt_key, aesni_set_decrypt_key,
108                    aesni_encrypt, aesni_decrypt,
109                    aesni_xts_encrypt, aesni_xts_decrypt);
110     return 1;
111 }
112
113 # define PROV_CIPHER_HW_declare_xts()                                          \
114 static const PROV_CIPHER_HW aesni_xts = {                                      \
115     cipher_hw_aesni_xts_initkey,                                               \
116     NULL,                                                                      \
117     cipher_hw_aes_xts_copyctx                                                  \
118 };
119 # define PROV_CIPHER_HW_select_xts()                                           \
120 if (AESNI_CAPABLE)                                                             \
121     return &aesni_xts;
122
123 # elif defined(SPARC_AES_CAPABLE)
124
125 static int cipher_hw_aes_xts_t4_initkey(PROV_CIPHER_CTX *ctx,
126                                         const unsigned char *key, size_t keylen)
127 {
128     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
129     OSSL_xts_stream_fn stream_enc = NULL;
130     OSSL_xts_stream_fn stream_dec = NULL;
131
132     /* Note: keylen is the size of 2 keys */
133     switch (keylen) {
134     case 32:
135         stream_enc = aes128_t4_xts_encrypt;
136         stream_dec = aes128_t4_xts_decrypt;
137         break;
138     case 64:
139         stream_enc = aes256_t4_xts_encrypt;
140         stream_dec = aes256_t4_xts_decrypt;
141         break;
142     default:
143         return 0;
144     }
145
146     XTS_SET_KEY_FN(aes_t4_set_encrypt_key, aes_t4_set_decrypt_key,
147                    aes_t4_encrypt, aes_t4_decrypt,
148                    stream_enc, stream_dec);
149     return 1;
150 }
151
152 # define PROV_CIPHER_HW_declare_xts()                                          \
153 static const PROV_CIPHER_HW aes_xts_t4 = {                                     \
154     cipher_hw_aes_xts_t4_initkey,                                              \
155     NULL,                                                                      \
156     cipher_hw_aes_xts_copyctx                                                  \
157 };
158 # define PROV_CIPHER_HW_select_xts()                                           \
159 if (SPARC_AES_CAPABLE)                                                         \
160     return &aes_xts_t4;
161
162 #elif defined(__riscv) && __riscv_xlen == 64
163
164 static int cipher_hw_aes_xts_rv64i_zknd_zkne_initkey(PROV_CIPHER_CTX *ctx,
165                                                      const unsigned char *key,
166                                                      size_t keylen)
167 {
168     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
169     OSSL_xts_stream_fn stream_enc = NULL;
170     OSSL_xts_stream_fn stream_dec = NULL;
171
172     XTS_SET_KEY_FN(rv64i_zkne_set_encrypt_key, rv64i_zknd_set_decrypt_key,
173                    rv64i_zkne_encrypt, rv64i_zknd_decrypt,
174                    stream_enc, stream_dec);
175     return 1;
176 }
177
178 static int cipher_hw_aes_xts_rv64i_zvkned_initkey(PROV_CIPHER_CTX *ctx,
179                                                      const unsigned char *key,
180                                                      size_t keylen)
181 {
182     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
183     OSSL_xts_stream_fn stream_enc = NULL;
184     OSSL_xts_stream_fn stream_dec = NULL;
185
186     /* Zvkned only supports 128 and 256 bit keys. */
187     if (keylen * 8 == 128 * 2 || keylen * 8 == 256 * 2) {
188         XTS_SET_KEY_FN(rv64i_zvkned_set_encrypt_key,
189                        rv64i_zvkned_set_decrypt_key,
190                        rv64i_zvkned_encrypt, rv64i_zvkned_decrypt,
191                        stream_enc, stream_dec);
192     } else {
193         XTS_SET_KEY_FN(AES_set_encrypt_key, AES_set_encrypt_key,
194                        rv64i_zvkned_encrypt, rv64i_zvkned_decrypt,
195                        stream_enc, stream_dec);
196     }
197     return 1;
198 }
199
200 # define PROV_CIPHER_HW_declare_xts()                                          \
201 static const PROV_CIPHER_HW aes_xts_rv64i_zknd_zkne = {                        \
202     cipher_hw_aes_xts_rv64i_zknd_zkne_initkey,                                 \
203     NULL,                                                                      \
204     cipher_hw_aes_xts_copyctx                                                  \
205 };                                                                             \
206 static const PROV_CIPHER_HW aes_xts_rv64i_zvkned = {                           \
207     cipher_hw_aes_xts_rv64i_zvkned_initkey,                                    \
208     NULL,                                                                      \
209     cipher_hw_aes_xts_copyctx                                                  \
210 };
211
212 # define PROV_CIPHER_HW_select_xts()                                           \
213 if (RISCV_HAS_ZVKNED() && riscv_vlen() >= 128)                                 \
214     return &aes_xts_rv64i_zvkned;                                              \
215 else if (RISCV_HAS_ZKND_AND_ZKNE())                                            \
216     return &aes_xts_rv64i_zknd_zkne;                                           \
217
218 #elif defined(__riscv) && __riscv_xlen == 32
219
220 static int cipher_hw_aes_xts_rv32i_zknd_zkne_initkey(PROV_CIPHER_CTX *ctx,
221                                                      const unsigned char *key,
222                                                      size_t keylen)
223 {
224     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
225
226     XTS_SET_KEY_FN(rv32i_zkne_set_encrypt_key, rv32i_zknd_zkne_set_decrypt_key,
227                    rv32i_zkne_encrypt, rv32i_zknd_decrypt,
228                    NULL, NULL);
229     return 1;
230 }
231
232 static int cipher_hw_aes_xts_rv32i_zbkb_zknd_zkne_initkey(PROV_CIPHER_CTX *ctx,
233                                                          const unsigned char *key,
234                                                          size_t keylen)
235 {
236     PROV_AES_XTS_CTX *xctx = (PROV_AES_XTS_CTX *)ctx;
237
238     XTS_SET_KEY_FN(rv32i_zbkb_zkne_set_encrypt_key, rv32i_zbkb_zknd_zkne_set_decrypt_key,
239                    rv32i_zkne_encrypt, rv32i_zknd_decrypt,
240                    NULL, NULL);
241     return 1;
242 }
243
244 # define PROV_CIPHER_HW_declare_xts()                                          \
245 static const PROV_CIPHER_HW aes_xts_rv32i_zknd_zkne = {                        \
246     cipher_hw_aes_xts_rv32i_zknd_zkne_initkey,                                 \
247     NULL,                                                                      \
248     cipher_hw_aes_xts_copyctx                                                  \
249 };                                                                             \
250 static const PROV_CIPHER_HW aes_xts_rv32i_zbkb_zknd_zkne = {                   \
251     cipher_hw_aes_xts_rv32i_zbkb_zknd_zkne_initkey,                            \
252     NULL,                                                                      \
253     cipher_hw_aes_xts_copyctx                                                  \
254 };
255 # define PROV_CIPHER_HW_select_xts()                                           \
256 if (RISCV_HAS_ZBKB_AND_ZKND_AND_ZKNE())                                        \
257     return &aes_xts_rv32i_zbkb_zknd_zkne;                                      \
258 if (RISCV_HAS_ZKND_ZKNE())                                                     \
259     return &aes_xts_rv32i_zknd_zkne;
260 # else
261 /* The generic case */
262 # define PROV_CIPHER_HW_declare_xts()
263 # define PROV_CIPHER_HW_select_xts()
264 #endif
265
266 static const PROV_CIPHER_HW aes_generic_xts = {
267     cipher_hw_aes_xts_generic_initkey,
268     NULL,
269     cipher_hw_aes_xts_copyctx
270 };
271 PROV_CIPHER_HW_declare_xts()
272 const PROV_CIPHER_HW *ossl_prov_cipher_hw_aes_xts(size_t keybits)
273 {
274     PROV_CIPHER_HW_select_xts()
275     return &aes_generic_xts;
276 }