1 /*
2 * Test driver for cipher functions
3 */
4 /* Copyright The Mbed TLS Contributors
5 * SPDX-License-Identifier: Apache-2.0
6 *
7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
8 * not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 */
19
20 #ifndef PSA_CRYPTO_TEST_DRIVERS_CIPHER_H
21 #define PSA_CRYPTO_TEST_DRIVERS_CIPHER_H
22
23 #include "mbedtls/build_info.h"
24
25 #if defined(PSA_CRYPTO_DRIVER_TEST)
26 #include <psa/crypto_driver_common.h>
27 #include <psa/crypto.h>
28
29 #include "mbedtls/cipher.h"
30
31 typedef struct {
32 /* If non-null, on success, copy this to the output. */
33 void *forced_output;
34 size_t forced_output_length;
35 /* If not PSA_SUCCESS, return this error code instead of processing the
36 * function call. */
37 psa_status_t forced_status;
38 /* Count the amount of times one of the cipher driver functions is called. */
39 unsigned long hits;
40 } mbedtls_test_driver_cipher_hooks_t;
41
42 #define MBEDTLS_TEST_DRIVER_CIPHER_INIT { NULL, 0, PSA_SUCCESS, 0 }
43 static inline mbedtls_test_driver_cipher_hooks_t
mbedtls_test_driver_cipher_hooks_init(void)44 mbedtls_test_driver_cipher_hooks_init(void)
45 {
46 const mbedtls_test_driver_cipher_hooks_t v = MBEDTLS_TEST_DRIVER_CIPHER_INIT;
47 return v;
48 }
49
50 extern mbedtls_test_driver_cipher_hooks_t mbedtls_test_driver_cipher_hooks;
51
52 psa_status_t mbedtls_test_transparent_cipher_encrypt(
53 const psa_key_attributes_t *attributes,
54 const uint8_t *key, size_t key_length,
55 psa_algorithm_t alg,
56 const uint8_t *iv, size_t iv_length,
57 const uint8_t *input, size_t input_length,
58 uint8_t *output, size_t output_size, size_t *output_length);
59
60 psa_status_t mbedtls_test_transparent_cipher_decrypt(
61 const psa_key_attributes_t *attributes,
62 const uint8_t *key, size_t key_length,
63 psa_algorithm_t alg,
64 const uint8_t *input, size_t input_length,
65 uint8_t *output, size_t output_size, size_t *output_length);
66
67 psa_status_t mbedtls_test_transparent_cipher_encrypt_setup(
68 mbedtls_transparent_test_driver_cipher_operation_t *operation,
69 const psa_key_attributes_t *attributes,
70 const uint8_t *key, size_t key_length,
71 psa_algorithm_t alg);
72
73 psa_status_t mbedtls_test_transparent_cipher_decrypt_setup(
74 mbedtls_transparent_test_driver_cipher_operation_t *operation,
75 const psa_key_attributes_t *attributes,
76 const uint8_t *key, size_t key_length,
77 psa_algorithm_t alg);
78
79 psa_status_t mbedtls_test_transparent_cipher_abort(
80 mbedtls_transparent_test_driver_cipher_operation_t *operation);
81
82 psa_status_t mbedtls_test_transparent_cipher_set_iv(
83 mbedtls_transparent_test_driver_cipher_operation_t *operation,
84 const uint8_t *iv, size_t iv_length);
85
86 psa_status_t mbedtls_test_transparent_cipher_update(
87 mbedtls_transparent_test_driver_cipher_operation_t *operation,
88 const uint8_t *input, size_t input_length,
89 uint8_t *output, size_t output_size, size_t *output_length);
90
91 psa_status_t mbedtls_test_transparent_cipher_finish(
92 mbedtls_transparent_test_driver_cipher_operation_t *operation,
93 uint8_t *output, size_t output_size, size_t *output_length);
94
95 /*
96 * opaque versions
97 */
98 psa_status_t mbedtls_test_opaque_cipher_encrypt(
99 const psa_key_attributes_t *attributes,
100 const uint8_t *key, size_t key_length,
101 psa_algorithm_t alg,
102 const uint8_t *iv, size_t iv_length,
103 const uint8_t *input, size_t input_length,
104 uint8_t *output, size_t output_size, size_t *output_length);
105
106 psa_status_t mbedtls_test_opaque_cipher_decrypt(
107 const psa_key_attributes_t *attributes,
108 const uint8_t *key, size_t key_length,
109 psa_algorithm_t alg,
110 const uint8_t *input, size_t input_length,
111 uint8_t *output, size_t output_size, size_t *output_length);
112
113 psa_status_t mbedtls_test_opaque_cipher_encrypt_setup(
114 mbedtls_opaque_test_driver_cipher_operation_t *operation,
115 const psa_key_attributes_t *attributes,
116 const uint8_t *key, size_t key_length,
117 psa_algorithm_t alg);
118
119 psa_status_t mbedtls_test_opaque_cipher_decrypt_setup(
120 mbedtls_opaque_test_driver_cipher_operation_t *operation,
121 const psa_key_attributes_t *attributes,
122 const uint8_t *key, size_t key_length,
123 psa_algorithm_t alg);
124
125 psa_status_t mbedtls_test_opaque_cipher_abort(
126 mbedtls_opaque_test_driver_cipher_operation_t *operation);
127
128 psa_status_t mbedtls_test_opaque_cipher_set_iv(
129 mbedtls_opaque_test_driver_cipher_operation_t *operation,
130 const uint8_t *iv, size_t iv_length);
131
132 psa_status_t mbedtls_test_opaque_cipher_update(
133 mbedtls_opaque_test_driver_cipher_operation_t *operation,
134 const uint8_t *input, size_t input_length,
135 uint8_t *output, size_t output_size, size_t *output_length);
136
137 psa_status_t mbedtls_test_opaque_cipher_finish(
138 mbedtls_opaque_test_driver_cipher_operation_t *operation,
139 uint8_t *output, size_t output_size, size_t *output_length);
140
141 #endif /* PSA_CRYPTO_DRIVER_TEST */
142 #endif /* PSA_CRYPTO_TEST_DRIVERS_CIPHER_H */
143