xref: /optee_os/lib/libutee/tee_api_operations.c (revision f5c3d85a579c9594ee7592af0c0783891c21d9e0)
11bb92983SJerome Forissier // SPDX-License-Identifier: BSD-2-Clause
2b0104773SPascal Brand /*
3b0104773SPascal Brand  * Copyright (c) 2014, STMicroelectronics International N.V.
4eee637e7SAlexander Zakharov  * Copyright (c) 2021, SumUp Services GmbH
5b0104773SPascal Brand  */
624ea7613SJens Wiklander #include <assert.h>
779170ce0SJerome Forissier #include <config.h>
8b0104773SPascal Brand #include <stdlib.h>
9b0104773SPascal Brand #include <string.h>
10b796ebf3SJerome Forissier #include <string_ext.h>
11b0104773SPascal Brand #include <tee_api.h>
128854d3c6SJerome Forissier #include <tee_api_defines_extensions.h>
13b0104773SPascal Brand #include <tee_internal_api_extensions.h>
14b0104773SPascal Brand #include <utee_syscalls.h>
15b0104773SPascal Brand #include <utee_defines.h>
16fc26c92aSJens Wiklander #include <util.h>
17e86f1266SJens Wiklander #include "tee_api_private.h"
18b0104773SPascal Brand 
19b0104773SPascal Brand struct __TEE_OperationHandle {
20b0104773SPascal Brand 	TEE_OperationInfo info;
21b0104773SPascal Brand 	TEE_ObjectHandle key1;
22b0104773SPascal Brand 	TEE_ObjectHandle key2;
23642a1607SCedric Chaumont 	uint32_t operationState;/* Operation state : INITIAL or ACTIVE */
2424ea7613SJens Wiklander 
2524ea7613SJens Wiklander 	/*
2624ea7613SJens Wiklander 	 * buffer to collect complete blocks or to keep a complete digest
2724ea7613SJens Wiklander 	 * for TEE_DigestExtract().
2824ea7613SJens Wiklander 	 */
2924ea7613SJens Wiklander 	uint8_t *buffer;
30b0104773SPascal Brand 	bool buffer_two_blocks;	/* True if two blocks need to be buffered */
31b0104773SPascal Brand 	size_t block_size;	/* Block size of cipher */
32b0104773SPascal Brand 	size_t buffer_offs;	/* Offset in buffer */
33b0104773SPascal Brand 	uint32_t state;		/* Handle to state in TEE Core */
34b0104773SPascal Brand };
35b0104773SPascal Brand 
36b0104773SPascal Brand /* Cryptographic Operations API - Generic Operation Functions */
37b0104773SPascal Brand 
38b0104773SPascal Brand TEE_Result TEE_AllocateOperation(TEE_OperationHandle *operation,
39b0104773SPascal Brand 				 uint32_t algorithm, uint32_t mode,
40b0104773SPascal Brand 				 uint32_t maxKeySize)
41b0104773SPascal Brand {
42b0104773SPascal Brand 	TEE_Result res;
43b0104773SPascal Brand 	TEE_OperationHandle op = TEE_HANDLE_NULL;
44b0104773SPascal Brand 	uint32_t handle_state = 0;
45b0104773SPascal Brand 	size_t block_size = 1;
46b0104773SPascal Brand 	uint32_t req_key_usage;
47b0104773SPascal Brand 	bool with_private_key = false;
48b0104773SPascal Brand 	bool buffer_two_blocks = false;
49b0104773SPascal Brand 
509b52c538SCedric Chaumont 	if (!operation)
51b0104773SPascal Brand 		TEE_Panic(0);
52b0104773SPascal Brand 
53696f56acSPingan Xie 	if (algorithm == TEE_ALG_AES_XTS || algorithm == TEE_ALG_SM2_KEP ||
54696f56acSPingan Xie 	    algorithm == TEE_ALG_SM4_XTS)
55b0104773SPascal Brand 		handle_state = TEE_HANDLE_FLAG_EXPECT_TWO_KEYS;
56b0104773SPascal Brand 
57218d9055SCedric Chaumont 	/* Check algorithm max key size */
58218d9055SCedric Chaumont 	switch (algorithm) {
59218d9055SCedric Chaumont 	case TEE_ALG_DSA_SHA1:
60218d9055SCedric Chaumont 		if (maxKeySize < 512)
61218d9055SCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
62218d9055SCedric Chaumont 		if (maxKeySize > 1024)
63218d9055SCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
64218d9055SCedric Chaumont 		if (maxKeySize % 64 != 0)
65218d9055SCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
66218d9055SCedric Chaumont 		break;
67218d9055SCedric Chaumont 
68218d9055SCedric Chaumont 	case TEE_ALG_DSA_SHA224:
69218d9055SCedric Chaumont 		if (maxKeySize != 2048)
70218d9055SCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
71218d9055SCedric Chaumont 		break;
72218d9055SCedric Chaumont 
73218d9055SCedric Chaumont 	case TEE_ALG_DSA_SHA256:
74218d9055SCedric Chaumont 		if (maxKeySize != 2048 && maxKeySize != 3072)
75218d9055SCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
76218d9055SCedric Chaumont 		break;
77218d9055SCedric Chaumont 
78fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA1:
79fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P192:
80fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P192:
811220586eSCedric Chaumont 		if (maxKeySize != 192)
821220586eSCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
831220586eSCedric Chaumont 		break;
841220586eSCedric Chaumont 
85fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA224:
86fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P224:
87fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P224:
881220586eSCedric Chaumont 		if (maxKeySize != 224)
891220586eSCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
901220586eSCedric Chaumont 		break;
911220586eSCedric Chaumont 
92fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA256:
93fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P256:
94fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P256:
9591fc6bd8SJerome Forissier 	case TEE_ALG_SM2_PKE:
960f151943SJerome Forissier 	case TEE_ALG_SM2_DSA_SM3:
971220586eSCedric Chaumont 		if (maxKeySize != 256)
981220586eSCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
991220586eSCedric Chaumont 		break;
1001220586eSCedric Chaumont 
1015b385b3fSJerome Forissier 	case TEE_ALG_SM2_KEP:
1025b385b3fSJerome Forissier 		/* Two 256-bit keys */
1035b385b3fSJerome Forissier 		if (maxKeySize != 512)
1045b385b3fSJerome Forissier 			return TEE_ERROR_NOT_SUPPORTED;
1055b385b3fSJerome Forissier 		break;
1065b385b3fSJerome Forissier 
107fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA384:
108fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P384:
109fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P384:
1101220586eSCedric Chaumont 		if (maxKeySize != 384)
1111220586eSCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
1121220586eSCedric Chaumont 		break;
1131220586eSCedric Chaumont 
114fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA512:
115fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P521:
116fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P521:
1171220586eSCedric Chaumont 		if (maxKeySize != 521)
1181220586eSCedric Chaumont 			return TEE_ERROR_NOT_SUPPORTED;
1191220586eSCedric Chaumont 		break;
120fe2fd3ffSJens Wiklander 
121fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDH_DERIVE_SHARED_SECRET:
122fe2fd3ffSJens Wiklander 		if (maxKeySize > 521)
123fe2fd3ffSJens Wiklander 			return TEE_ERROR_NOT_SUPPORTED;
124fe2fd3ffSJens Wiklander 		break;
125fe2fd3ffSJens Wiklander 
126e1f9cee7SSergiy Kibrik 	case TEE_ALG_ED25519:
1273f61056dSSohaib ul Hassan 	case TEE_ALG_X25519:
1283f61056dSSohaib ul Hassan 		if (maxKeySize != 256)
1293f61056dSSohaib ul Hassan 			return TEE_ERROR_NOT_SUPPORTED;
1303f61056dSSohaib ul Hassan 		break;
131218d9055SCedric Chaumont 	default:
132218d9055SCedric Chaumont 		break;
133218d9055SCedric Chaumont 	}
134218d9055SCedric Chaumont 
135cf5c060cSJens Wiklander 	/* Check algorithm mode (and maxKeySize for digests) */
136b0104773SPascal Brand 	switch (algorithm) {
137b0104773SPascal Brand 	case TEE_ALG_AES_CTS:
138b0104773SPascal Brand 	case TEE_ALG_AES_XTS:
139696f56acSPingan Xie 	case TEE_ALG_SM4_XTS:
140b0104773SPascal Brand 		buffer_two_blocks = true;
141919a5a68SJerome Forissier 		fallthrough;
1424bd53c54SJerome Forissier 	case TEE_ALG_AES_ECB_NOPAD:
143b0104773SPascal Brand 	case TEE_ALG_AES_CBC_NOPAD:
144b0104773SPascal Brand 	case TEE_ALG_AES_CCM:
145b0104773SPascal Brand 	case TEE_ALG_DES_ECB_NOPAD:
146b0104773SPascal Brand 	case TEE_ALG_DES_CBC_NOPAD:
147b0104773SPascal Brand 	case TEE_ALG_DES3_ECB_NOPAD:
148b0104773SPascal Brand 	case TEE_ALG_DES3_CBC_NOPAD:
149ade6f848SJerome Forissier 	case TEE_ALG_SM4_ECB_NOPAD:
150ade6f848SJerome Forissier 	case TEE_ALG_SM4_CBC_NOPAD:
151ade6f848SJerome Forissier 	case TEE_ALG_SM4_CTR:
152b0104773SPascal Brand 		if (TEE_ALG_GET_MAIN_ALG(algorithm) == TEE_MAIN_ALGO_AES)
153b0104773SPascal Brand 			block_size = TEE_AES_BLOCK_SIZE;
154ade6f848SJerome Forissier 		else if (TEE_ALG_GET_MAIN_ALG(algorithm) == TEE_MAIN_ALGO_SM4)
155ade6f848SJerome Forissier 			block_size = TEE_SM4_BLOCK_SIZE;
156b0104773SPascal Brand 		else
157b0104773SPascal Brand 			block_size = TEE_DES_BLOCK_SIZE;
158919a5a68SJerome Forissier 		fallthrough;
15957aabac5SBogdan Liulko 	case TEE_ALG_AES_CTR:
160afc0c182SBogdan Liulko 	case TEE_ALG_AES_GCM:
161b0104773SPascal Brand 		if (mode == TEE_MODE_ENCRYPT)
162b0104773SPascal Brand 			req_key_usage = TEE_USAGE_ENCRYPT;
163b0104773SPascal Brand 		else if (mode == TEE_MODE_DECRYPT)
164b0104773SPascal Brand 			req_key_usage = TEE_USAGE_DECRYPT;
165b0104773SPascal Brand 		else
166b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
167b0104773SPascal Brand 		break;
168b0104773SPascal Brand 
1696a2e0a9fSGabor Szekely #if defined(CFG_CRYPTO_RSASSA_NA1)
1706a2e0a9fSGabor Szekely 	case TEE_ALG_RSASSA_PKCS1_V1_5:
1716a2e0a9fSGabor Szekely #endif
172b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_MD5:
173b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_SHA1:
174b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_SHA224:
175b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_SHA256:
176b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_SHA384:
177b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_V1_5_SHA512:
178*f5c3d85aSJulien Masson 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_MD5:
179b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA1:
180b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA224:
181b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA256:
182b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA384:
183b0104773SPascal Brand 	case TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA512:
184b0104773SPascal Brand 	case TEE_ALG_DSA_SHA1:
185218d9055SCedric Chaumont 	case TEE_ALG_DSA_SHA224:
186218d9055SCedric Chaumont 	case TEE_ALG_DSA_SHA256:
187fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA1:
188fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA224:
189fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA256:
190fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA384:
191fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDSA_SHA512:
192fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P192:
193fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P224:
194fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P256:
195fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P384:
196fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDSA_P521:
1970f151943SJerome Forissier 	case TEE_ALG_SM2_DSA_SM3:
198e1f9cee7SSergiy Kibrik 	case TEE_ALG_ED25519:
199b0104773SPascal Brand 		if (mode == TEE_MODE_SIGN) {
200b0104773SPascal Brand 			with_private_key = true;
201b0104773SPascal Brand 			req_key_usage = TEE_USAGE_SIGN;
202b0104773SPascal Brand 		} else if (mode == TEE_MODE_VERIFY) {
203b0104773SPascal Brand 			req_key_usage = TEE_USAGE_VERIFY;
204b0104773SPascal Brand 		} else {
205b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
206b0104773SPascal Brand 		}
207b0104773SPascal Brand 		break;
208b0104773SPascal Brand 
209b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_V1_5:
210*f5c3d85aSJulien Masson 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_MD5:
211b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA1:
212b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA224:
213b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA256:
214b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA384:
215b0104773SPascal Brand 	case TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA512:
21691fc6bd8SJerome Forissier 	case TEE_ALG_SM2_PKE:
217b0104773SPascal Brand 		if (mode == TEE_MODE_ENCRYPT) {
218b0104773SPascal Brand 			req_key_usage = TEE_USAGE_ENCRYPT;
219b0104773SPascal Brand 		} else if (mode == TEE_MODE_DECRYPT) {
220b0104773SPascal Brand 			with_private_key = true;
221b0104773SPascal Brand 			req_key_usage = TEE_USAGE_DECRYPT;
222b0104773SPascal Brand 		} else {
223b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
224b0104773SPascal Brand 		}
225b0104773SPascal Brand 		break;
226b0104773SPascal Brand 
227b0104773SPascal Brand 	case TEE_ALG_RSA_NOPAD:
228b0104773SPascal Brand 		if (mode == TEE_MODE_ENCRYPT) {
229b0104773SPascal Brand 			req_key_usage = TEE_USAGE_ENCRYPT | TEE_USAGE_VERIFY;
230b0104773SPascal Brand 		} else if (mode == TEE_MODE_DECRYPT) {
231b0104773SPascal Brand 			with_private_key = true;
232b0104773SPascal Brand 			req_key_usage = TEE_USAGE_DECRYPT | TEE_USAGE_SIGN;
233b0104773SPascal Brand 		} else {
234b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
235b0104773SPascal Brand 		}
236b0104773SPascal Brand 		break;
237b0104773SPascal Brand 
238b0104773SPascal Brand 	case TEE_ALG_DH_DERIVE_SHARED_SECRET:
239fe2fd3ffSJens Wiklander 	case TEE_ALG_ECDH_DERIVE_SHARED_SECRET:
240fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P192:
241fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P224:
242fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P256:
243fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P384:
244fe2fd3ffSJens Wiklander 	case __OPTEE_ALG_ECDH_P521:
245cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_MD5_DERIVE_KEY:
246cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_SHA1_DERIVE_KEY:
247cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_SHA224_DERIVE_KEY:
248cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_SHA256_DERIVE_KEY:
249cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_SHA384_DERIVE_KEY:
250cdb198a7SJerome Forissier 	case TEE_ALG_HKDF_SHA512_DERIVE_KEY:
2518854d3c6SJerome Forissier 	case TEE_ALG_CONCAT_KDF_SHA1_DERIVE_KEY:
2528854d3c6SJerome Forissier 	case TEE_ALG_CONCAT_KDF_SHA224_DERIVE_KEY:
2538854d3c6SJerome Forissier 	case TEE_ALG_CONCAT_KDF_SHA256_DERIVE_KEY:
2548854d3c6SJerome Forissier 	case TEE_ALG_CONCAT_KDF_SHA384_DERIVE_KEY:
2558854d3c6SJerome Forissier 	case TEE_ALG_CONCAT_KDF_SHA512_DERIVE_KEY:
2560f2293b7SJerome Forissier 	case TEE_ALG_PBKDF2_HMAC_SHA1_DERIVE_KEY:
2575b385b3fSJerome Forissier 	case TEE_ALG_SM2_KEP:
2583f61056dSSohaib ul Hassan 	case TEE_ALG_X25519:
259b0104773SPascal Brand 		if (mode != TEE_MODE_DERIVE)
260b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
261b0104773SPascal Brand 		with_private_key = true;
262b0104773SPascal Brand 		req_key_usage = TEE_USAGE_DERIVE;
263b0104773SPascal Brand 		break;
264b0104773SPascal Brand 
265b0104773SPascal Brand 	case TEE_ALG_MD5:
266b0104773SPascal Brand 	case TEE_ALG_SHA1:
267b0104773SPascal Brand 	case TEE_ALG_SHA224:
268b0104773SPascal Brand 	case TEE_ALG_SHA256:
269b0104773SPascal Brand 	case TEE_ALG_SHA384:
270b0104773SPascal Brand 	case TEE_ALG_SHA512:
27124ea7613SJens Wiklander 	case TEE_ALG_SHA3_224:
27224ea7613SJens Wiklander 	case TEE_ALG_SHA3_256:
27324ea7613SJens Wiklander 	case TEE_ALG_SHA3_384:
27424ea7613SJens Wiklander 	case TEE_ALG_SHA3_512:
27524ea7613SJens Wiklander 	case TEE_ALG_SHAKE128:
27624ea7613SJens Wiklander 	case TEE_ALG_SHAKE256:
27747645577SJerome Forissier 	case TEE_ALG_SM3:
278b0104773SPascal Brand 		if (mode != TEE_MODE_DIGEST)
279b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
280cf5c060cSJens Wiklander 		if (maxKeySize)
281cf5c060cSJens Wiklander 			return TEE_ERROR_NOT_SUPPORTED;
28205304565SCedric Chaumont 		/* v1.1: flags always set for digest operations */
283b0104773SPascal Brand 		handle_state |= TEE_HANDLE_FLAG_KEY_SET;
284b0104773SPascal Brand 		req_key_usage = 0;
285b0104773SPascal Brand 		break;
286b0104773SPascal Brand 
287b0104773SPascal Brand 	case TEE_ALG_DES_CBC_MAC_NOPAD:
288b0104773SPascal Brand 	case TEE_ALG_AES_CBC_MAC_NOPAD:
289b0104773SPascal Brand 	case TEE_ALG_AES_CBC_MAC_PKCS5:
290b0104773SPascal Brand 	case TEE_ALG_AES_CMAC:
291b0104773SPascal Brand 	case TEE_ALG_DES_CBC_MAC_PKCS5:
292b0104773SPascal Brand 	case TEE_ALG_DES3_CBC_MAC_NOPAD:
293b0104773SPascal Brand 	case TEE_ALG_DES3_CBC_MAC_PKCS5:
294eee637e7SAlexander Zakharov 	case TEE_ALG_DES3_CMAC:
295b0104773SPascal Brand 	case TEE_ALG_HMAC_MD5:
296b0104773SPascal Brand 	case TEE_ALG_HMAC_SHA1:
297b0104773SPascal Brand 	case TEE_ALG_HMAC_SHA224:
298b0104773SPascal Brand 	case TEE_ALG_HMAC_SHA256:
299b0104773SPascal Brand 	case TEE_ALG_HMAC_SHA384:
300b0104773SPascal Brand 	case TEE_ALG_HMAC_SHA512:
301260b4028SJens Wiklander 	case TEE_ALG_HMAC_SHA3_224:
302260b4028SJens Wiklander 	case TEE_ALG_HMAC_SHA3_256:
303260b4028SJens Wiklander 	case TEE_ALG_HMAC_SHA3_384:
304260b4028SJens Wiklander 	case TEE_ALG_HMAC_SHA3_512:
30547645577SJerome Forissier 	case TEE_ALG_HMAC_SM3:
306b0104773SPascal Brand 		if (mode != TEE_MODE_MAC)
307b0104773SPascal Brand 			return TEE_ERROR_NOT_SUPPORTED;
308b0104773SPascal Brand 		req_key_usage = TEE_USAGE_MAC;
309b0104773SPascal Brand 		break;
310b0104773SPascal Brand 
311b0104773SPascal Brand 	default:
312b0104773SPascal Brand 		return TEE_ERROR_NOT_SUPPORTED;
313b0104773SPascal Brand 	}
314b0104773SPascal Brand 
315b66f219bSJens Wiklander 	op = TEE_Malloc(sizeof(*op), TEE_MALLOC_FILL_ZERO);
3169b52c538SCedric Chaumont 	if (!op)
317b0104773SPascal Brand 		return TEE_ERROR_OUT_OF_MEMORY;
318b0104773SPascal Brand 
319b0104773SPascal Brand 	op->info.algorithm = algorithm;
320b0104773SPascal Brand 	op->info.operationClass = TEE_ALG_GET_CLASS(algorithm);
3216a2e0a9fSGabor Szekely #ifdef CFG_CRYPTO_RSASSA_NA1
3226a2e0a9fSGabor Szekely 	if (algorithm == TEE_ALG_RSASSA_PKCS1_V1_5)
3236a2e0a9fSGabor Szekely 		op->info.operationClass = TEE_OPERATION_ASYMMETRIC_SIGNATURE;
3246a2e0a9fSGabor Szekely #endif
325b0104773SPascal Brand 	op->info.mode = mode;
3262e5e6460SAlbert Schwarzkopf 	op->info.digestLength = TEE_ALG_GET_DIGEST_SIZE(algorithm);
327b0104773SPascal Brand 	op->info.maxKeySize = maxKeySize;
328b0104773SPascal Brand 	op->info.requiredKeyUsage = req_key_usage;
329b0104773SPascal Brand 	op->info.handleState = handle_state;
330b0104773SPascal Brand 
33124ea7613SJens Wiklander 	/*
33224ea7613SJens Wiklander 	 * Needed to buffer the digest if TEE_DigestExtract() doesn't
33324ea7613SJens Wiklander 	 * retrieve the entire digest in one go.
33424ea7613SJens Wiklander 	 */
33524ea7613SJens Wiklander 	if (op->info.operationClass == TEE_OPERATION_DIGEST)
33624ea7613SJens Wiklander 		block_size = op->info.digestLength;
33724ea7613SJens Wiklander 
338b0104773SPascal Brand 	if (block_size > 1) {
339b0104773SPascal Brand 		size_t buffer_size = block_size;
340b0104773SPascal Brand 
341b0104773SPascal Brand 		if (buffer_two_blocks)
342b0104773SPascal Brand 			buffer_size *= 2;
343b0104773SPascal Brand 
3449b52c538SCedric Chaumont 		op->buffer = TEE_Malloc(buffer_size,
3459b52c538SCedric Chaumont 					TEE_USER_MEM_HINT_NO_FILL_ZERO);
346b0104773SPascal Brand 		if (op->buffer == NULL) {
347b0104773SPascal Brand 			res = TEE_ERROR_OUT_OF_MEMORY;
348b66f219bSJens Wiklander 			goto out;
349b0104773SPascal Brand 		}
350b0104773SPascal Brand 	}
351b0104773SPascal Brand 	op->block_size = block_size;
352b0104773SPascal Brand 	op->buffer_two_blocks = buffer_two_blocks;
353b0104773SPascal Brand 
354b0104773SPascal Brand 	if (TEE_ALG_GET_CLASS(algorithm) != TEE_OPERATION_DIGEST) {
355b0104773SPascal Brand 		uint32_t mks = maxKeySize;
356b0104773SPascal Brand 		TEE_ObjectType key_type = TEE_ALG_GET_KEY_TYPE(algorithm,
357b0104773SPascal Brand 						       with_private_key);
358b0104773SPascal Brand 
359b0104773SPascal Brand 		/*
360b0104773SPascal Brand 		 * If two keys are expected the max key size is the sum of
361b0104773SPascal Brand 		 * the size of both keys.
362b0104773SPascal Brand 		 */
363b0104773SPascal Brand 		if (op->info.handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS)
364b0104773SPascal Brand 			mks /= 2;
365b0104773SPascal Brand 
366b0104773SPascal Brand 		res = TEE_AllocateTransientObject(key_type, mks, &op->key1);
367b0104773SPascal Brand 		if (res != TEE_SUCCESS)
368b66f219bSJens Wiklander 			goto out;
369b0104773SPascal Brand 
37005304565SCedric Chaumont 		if (op->info.handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS) {
3719b52c538SCedric Chaumont 			res = TEE_AllocateTransientObject(key_type, mks,
372b0104773SPascal Brand 							  &op->key2);
373b0104773SPascal Brand 			if (res != TEE_SUCCESS)
374b66f219bSJens Wiklander 				goto out;
375b0104773SPascal Brand 		}
376b0104773SPascal Brand 	}
377b0104773SPascal Brand 
3782c028fdeSJerome Forissier 	res = _utee_cryp_state_alloc(algorithm, mode, (unsigned long)op->key1,
379e86f1266SJens Wiklander 				     (unsigned long)op->key2, &op->state);
380b66f219bSJens Wiklander 	if (res != TEE_SUCCESS)
381b66f219bSJens Wiklander 		goto out;
382b0104773SPascal Brand 
38305304565SCedric Chaumont 	/*
38405304565SCedric Chaumont 	 * Initialize digest operations
38505304565SCedric Chaumont 	 * Other multi-stage operations initialized w/ TEE_xxxInit functions
38605304565SCedric Chaumont 	 * Non-applicable on asymmetric operations
38705304565SCedric Chaumont 	 */
38805304565SCedric Chaumont 	if (TEE_ALG_GET_CLASS(algorithm) == TEE_OPERATION_DIGEST) {
3892c028fdeSJerome Forissier 		res = _utee_hash_init(op->state, NULL, 0);
39005304565SCedric Chaumont 		if (res != TEE_SUCCESS)
391b66f219bSJens Wiklander 			goto out;
39205304565SCedric Chaumont 		/* v1.1: flags always set for digest operations */
39305304565SCedric Chaumont 		op->info.handleState |= TEE_HANDLE_FLAG_INITIALIZED;
39405304565SCedric Chaumont 	}
39505304565SCedric Chaumont 
396642a1607SCedric Chaumont 	op->operationState = TEE_OPERATION_STATE_INITIAL;
397642a1607SCedric Chaumont 
398b0104773SPascal Brand 	*operation = op;
399b0104773SPascal Brand 
400b66f219bSJens Wiklander out:
401b66f219bSJens Wiklander 	if (res != TEE_SUCCESS) {
402b66f219bSJens Wiklander 		if (res != TEE_ERROR_OUT_OF_MEMORY &&
4039b52c538SCedric Chaumont 		    res != TEE_ERROR_NOT_SUPPORTED)
404b36311adSJerome Forissier 			TEE_Panic(res);
405b66f219bSJens Wiklander 		if (op) {
406b66f219bSJens Wiklander 			if (op->state) {
407b66f219bSJens Wiklander 				TEE_FreeOperation(op);
408b66f219bSJens Wiklander 			} else {
409b66f219bSJens Wiklander 				TEE_Free(op->buffer);
410b66f219bSJens Wiklander 				TEE_FreeTransientObject(op->key1);
411b66f219bSJens Wiklander 				TEE_FreeTransientObject(op->key2);
412b66f219bSJens Wiklander 				TEE_Free(op);
413b66f219bSJens Wiklander 			}
414b66f219bSJens Wiklander 		}
415b66f219bSJens Wiklander 	}
416b66f219bSJens Wiklander 
417b0104773SPascal Brand 	return res;
418b0104773SPascal Brand }
419b0104773SPascal Brand 
420b0104773SPascal Brand void TEE_FreeOperation(TEE_OperationHandle operation)
421b0104773SPascal Brand {
422e889e80bSCedric Chaumont 	TEE_Result res;
423e889e80bSCedric Chaumont 
424e889e80bSCedric Chaumont 	if (operation == TEE_HANDLE_NULL)
425c036e912SJens Wiklander 		return;
426e889e80bSCedric Chaumont 
427b0104773SPascal Brand 	/*
428b0104773SPascal Brand 	 * Note that keys should not be freed here, since they are
429b0104773SPascal Brand 	 * claimed by the operation they will be freed by
430b0104773SPascal Brand 	 * utee_cryp_state_free().
431b0104773SPascal Brand 	 */
4322c028fdeSJerome Forissier 	res = _utee_cryp_state_free(operation->state);
433e889e80bSCedric Chaumont 	if (res != TEE_SUCCESS)
434b36311adSJerome Forissier 		TEE_Panic(res);
435e889e80bSCedric Chaumont 
436b0104773SPascal Brand 	TEE_Free(operation->buffer);
437b0104773SPascal Brand 	TEE_Free(operation);
438b0104773SPascal Brand }
439b0104773SPascal Brand 
440c036e912SJens Wiklander void __GP11_TEE_FreeOperation(TEE_OperationHandle operation)
441c036e912SJens Wiklander {
442c036e912SJens Wiklander 	if (operation == TEE_HANDLE_NULL)
443c036e912SJens Wiklander 		TEE_Panic(0);
444c036e912SJens Wiklander 	TEE_FreeOperation(operation);
445c036e912SJens Wiklander }
446c036e912SJens Wiklander 
447b0104773SPascal Brand void TEE_GetOperationInfo(TEE_OperationHandle operation,
448b0104773SPascal Brand 			  TEE_OperationInfo *operationInfo)
449b0104773SPascal Brand {
450b0104773SPascal Brand 	if (operation == TEE_HANDLE_NULL)
451b0104773SPascal Brand 		TEE_Panic(0);
452b0104773SPascal Brand 
4536915bbbbSJens Wiklander 	__utee_check_out_annotation(operationInfo, sizeof(*operationInfo));
454b0104773SPascal Brand 
455b0104773SPascal Brand 	*operationInfo = operation->info;
456bac3a8a7SJens Wiklander 	if (operationInfo->handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS) {
457bac3a8a7SJens Wiklander 		operationInfo->keySize = 0;
458bac3a8a7SJens Wiklander 		operationInfo->requiredKeyUsage = 0;
459bac3a8a7SJens Wiklander 	}
460b0104773SPascal Brand }
461b0104773SPascal Brand 
462ee2f75afSJens Wiklander TEE_Result TEE_GetOperationInfoMultiple(TEE_OperationHandle op,
463ee2f75afSJens Wiklander 					TEE_OperationInfoMultiple *op_info,
464cb98b7b2SJens Wiklander 					size_t *size)
46505304565SCedric Chaumont {
46605304565SCedric Chaumont 	TEE_Result res = TEE_SUCCESS;
467ee2f75afSJens Wiklander 	TEE_ObjectInfo kinfo = { };
468ee2f75afSJens Wiklander 	size_t max_key_count = 0;
469ee2f75afSJens Wiklander 	bool two_keys = false;
47005304565SCedric Chaumont 
471ee2f75afSJens Wiklander 	if (op == TEE_HANDLE_NULL) {
47205304565SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
47305304565SCedric Chaumont 		goto out;
47405304565SCedric Chaumont 	}
47505304565SCedric Chaumont 
476cb98b7b2SJens Wiklander 	__utee_check_outbuf_annotation(op_info, size);
47705304565SCedric Chaumont 
478ee2f75afSJens Wiklander 	if (*size < sizeof(*op_info)) {
479ee2f75afSJens Wiklander 		res = TEE_ERROR_BAD_PARAMETERS;
480ee2f75afSJens Wiklander 		goto out;
481ee2f75afSJens Wiklander 	}
482ee2f75afSJens Wiklander 	max_key_count = (*size - sizeof(*op_info)) /
48305304565SCedric Chaumont 			sizeof(TEE_OperationInfoKey);
48405304565SCedric Chaumont 
485ee2f75afSJens Wiklander 	TEE_MemFill(op_info, 0, *size);
48605304565SCedric Chaumont 
48705304565SCedric Chaumont 	/* Two keys flag (TEE_ALG_AES_XTS only) */
488ee2f75afSJens Wiklander 	two_keys = op->info.handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS;
489ee2f75afSJens Wiklander 
490ee2f75afSJens Wiklander 	if (op->info.mode == TEE_MODE_DIGEST) {
491ee2f75afSJens Wiklander 		op_info->numberOfKeys = 0;
492ee2f75afSJens Wiklander 	} else if (!two_keys) {
493ee2f75afSJens Wiklander 		if (max_key_count < 1) {
49405304565SCedric Chaumont 			res = TEE_ERROR_SHORT_BUFFER;
49505304565SCedric Chaumont 			goto out;
49605304565SCedric Chaumont 		}
49705304565SCedric Chaumont 
498ee2f75afSJens Wiklander 		res = TEE_GetObjectInfo1(op->key1, &kinfo);
499ee2f75afSJens Wiklander 		/* Key1 is not a valid handle, "can't happen". */
500ee2f75afSJens Wiklander 		if (res)
50105304565SCedric Chaumont 			goto out;
50205304565SCedric Chaumont 
503d372a47cSJens Wiklander 		op_info->keyInformation[0].keySize = kinfo.objectSize;
504ee2f75afSJens Wiklander 		op_info->keyInformation[0].requiredKeyUsage =
505ee2f75afSJens Wiklander 			op->info.requiredKeyUsage;
506ee2f75afSJens Wiklander 		op_info->numberOfKeys = 1;
507ee2f75afSJens Wiklander 	} else {
508ee2f75afSJens Wiklander 		if (max_key_count < 2) {
509ee2f75afSJens Wiklander 			res = TEE_ERROR_SHORT_BUFFER;
51005304565SCedric Chaumont 			goto out;
51105304565SCedric Chaumont 		}
51205304565SCedric Chaumont 
513ee2f75afSJens Wiklander 		res = TEE_GetObjectInfo1(op->key1, &kinfo);
514ee2f75afSJens Wiklander 		/* Key1 is not a valid handle, "can't happen". */
515ee2f75afSJens Wiklander 		if (res)
516ee2f75afSJens Wiklander 			goto out;
517ee2f75afSJens Wiklander 
518d372a47cSJens Wiklander 		op_info->keyInformation[0].keySize = kinfo.objectSize;
519ee2f75afSJens Wiklander 		op_info->keyInformation[0].requiredKeyUsage =
520ee2f75afSJens Wiklander 			op->info.requiredKeyUsage;
521ee2f75afSJens Wiklander 
522ee2f75afSJens Wiklander 		res = TEE_GetObjectInfo1(op->key2, &kinfo);
523ee2f75afSJens Wiklander 		/* Key2 is not a valid handle, "can't happen". */
524ee2f75afSJens Wiklander 		if (res)
525ee2f75afSJens Wiklander 			goto out;
526ee2f75afSJens Wiklander 
527d372a47cSJens Wiklander 		op_info->keyInformation[1].keySize = kinfo.objectSize;
528ee2f75afSJens Wiklander 		op_info->keyInformation[1].requiredKeyUsage =
529ee2f75afSJens Wiklander 			op->info.requiredKeyUsage;
530ee2f75afSJens Wiklander 
531ee2f75afSJens Wiklander 		op_info->numberOfKeys = 2;
53205304565SCedric Chaumont 	}
53305304565SCedric Chaumont 
534ee2f75afSJens Wiklander 	op_info->algorithm = op->info.algorithm;
535ee2f75afSJens Wiklander 	op_info->operationClass = op->info.operationClass;
536ee2f75afSJens Wiklander 	op_info->mode = op->info.mode;
537ee2f75afSJens Wiklander 	op_info->digestLength = op->info.digestLength;
538ee2f75afSJens Wiklander 	op_info->maxKeySize = op->info.maxKeySize;
539ee2f75afSJens Wiklander 	op_info->handleState = op->info.handleState;
540ee2f75afSJens Wiklander 	op_info->operationState = op->operationState;
54105304565SCedric Chaumont 
54205304565SCedric Chaumont out:
54305304565SCedric Chaumont 	if (res != TEE_SUCCESS &&
54405304565SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
545b36311adSJerome Forissier 		TEE_Panic(res);
54605304565SCedric Chaumont 
54705304565SCedric Chaumont 	return res;
54805304565SCedric Chaumont }
54905304565SCedric Chaumont 
550cb98b7b2SJens Wiklander TEE_Result
551cb98b7b2SJens Wiklander __GP11_TEE_GetOperationInfoMultiple(TEE_OperationHandle operation,
552cb98b7b2SJens Wiklander 				    TEE_OperationInfoMultiple *info,
553cb98b7b2SJens Wiklander 				    uint32_t *operationSize)
554cb98b7b2SJens Wiklander {
555cb98b7b2SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
556cb98b7b2SJens Wiklander 	size_t s = 0;
557cb98b7b2SJens Wiklander 
558cb98b7b2SJens Wiklander 	__utee_check_gp11_outbuf_annotation(info, operationSize);
559cb98b7b2SJens Wiklander 	s = *operationSize;
560cb98b7b2SJens Wiklander 	res = TEE_GetOperationInfoMultiple(operation, info, &s);
561cb98b7b2SJens Wiklander 	*operationSize = s;
562cb98b7b2SJens Wiklander 	return res;
563cb98b7b2SJens Wiklander }
564cb98b7b2SJens Wiklander 
565287a6182SJens Wiklander static void reset_operation_state(TEE_OperationHandle op)
566287a6182SJens Wiklander {
567287a6182SJens Wiklander 	op->operationState = TEE_OPERATION_STATE_INITIAL;
568287a6182SJens Wiklander 
569287a6182SJens Wiklander 	if (op->info.operationClass == TEE_OPERATION_DIGEST) {
570287a6182SJens Wiklander 		TEE_Result res = _utee_hash_init(op->state, NULL, 0);
571287a6182SJens Wiklander 
572287a6182SJens Wiklander 		if (res != TEE_SUCCESS)
573287a6182SJens Wiklander 			TEE_Panic(res);
574287a6182SJens Wiklander 		op->info.handleState |= TEE_HANDLE_FLAG_INITIALIZED;
575287a6182SJens Wiklander 	} else {
576287a6182SJens Wiklander 		op->info.handleState &= ~TEE_HANDLE_FLAG_INITIALIZED;
577287a6182SJens Wiklander 	}
578287a6182SJens Wiklander }
579287a6182SJens Wiklander 
580b0104773SPascal Brand void TEE_ResetOperation(TEE_OperationHandle operation)
581b0104773SPascal Brand {
582b0104773SPascal Brand 	if (operation == TEE_HANDLE_NULL)
583b0104773SPascal Brand 		TEE_Panic(0);
584bf80076aSCedric Chaumont 
585642a1607SCedric Chaumont 	if (!(operation->info.handleState & TEE_HANDLE_FLAG_KEY_SET))
586bf80076aSCedric Chaumont 		TEE_Panic(0);
587bf80076aSCedric Chaumont 
588287a6182SJens Wiklander 	reset_operation_state(operation);
58905304565SCedric Chaumont }
590b0104773SPascal Brand 
591b0104773SPascal Brand TEE_Result TEE_SetOperationKey(TEE_OperationHandle operation,
592b0104773SPascal Brand 			       TEE_ObjectHandle key)
593b0104773SPascal Brand {
5947583c59eSCedric Chaumont 	TEE_Result res;
595b0104773SPascal Brand 	uint32_t key_size = 0;
596b0104773SPascal Brand 	TEE_ObjectInfo key_info;
597b0104773SPascal Brand 
598a57c1e2eSCedric Chaumont 	if (operation == TEE_HANDLE_NULL) {
599a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
600a57c1e2eSCedric Chaumont 		goto out;
601a57c1e2eSCedric Chaumont 	}
602a57c1e2eSCedric Chaumont 
603a57c1e2eSCedric Chaumont 	if (key == TEE_HANDLE_NULL) {
604a57c1e2eSCedric Chaumont 		/* Operation key cleared */
605a57c1e2eSCedric Chaumont 		TEE_ResetTransientObject(operation->key1);
6066c4ea258SJens Wiklander 		operation->info.handleState &= ~TEE_HANDLE_FLAG_KEY_SET;
607287a6182SJens Wiklander 		if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
608287a6182SJens Wiklander 			reset_operation_state(operation);
6096c4ea258SJens Wiklander 		return TEE_SUCCESS;
610a57c1e2eSCedric Chaumont 	}
611a57c1e2eSCedric Chaumont 
612a57c1e2eSCedric Chaumont 	/* No key for digest operation */
613a57c1e2eSCedric Chaumont 	if (operation->info.operationClass == TEE_OPERATION_DIGEST) {
614a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
615a57c1e2eSCedric Chaumont 		goto out;
616a57c1e2eSCedric Chaumont 	}
617a57c1e2eSCedric Chaumont 
618a57c1e2eSCedric Chaumont 	/* Two keys flag not expected (TEE_ALG_AES_XTS excluded) */
619a57c1e2eSCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS) !=
620a57c1e2eSCedric Chaumont 	    0) {
621a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
622a57c1e2eSCedric Chaumont 		goto out;
623a57c1e2eSCedric Chaumont 	}
624a57c1e2eSCedric Chaumont 
6257583c59eSCedric Chaumont 	res = TEE_GetObjectInfo1(key, &key_info);
626a57c1e2eSCedric Chaumont 	/* Key is not a valid handle */
6277583c59eSCedric Chaumont 	if (res != TEE_SUCCESS)
628a57c1e2eSCedric Chaumont 		goto out;
6297583c59eSCedric Chaumont 
630b0104773SPascal Brand 	/* Supplied key has to meet required usage */
631b0104773SPascal Brand 	if ((key_info.objectUsage & operation->info.requiredKeyUsage) !=
632b0104773SPascal Brand 	    operation->info.requiredKeyUsage) {
633a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
634a57c1e2eSCedric Chaumont 		goto out;
635b0104773SPascal Brand 	}
636b0104773SPascal Brand 
637d372a47cSJens Wiklander 	if (operation->info.maxKeySize < key_info.objectSize) {
638a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
639a57c1e2eSCedric Chaumont 		goto out;
640a57c1e2eSCedric Chaumont 	}
641b0104773SPascal Brand 
642d372a47cSJens Wiklander 	key_size = key_info.objectSize;
643b0104773SPascal Brand 
644b0104773SPascal Brand 	TEE_ResetTransientObject(operation->key1);
645b0104773SPascal Brand 	operation->info.handleState &= ~TEE_HANDLE_FLAG_KEY_SET;
646b0104773SPascal Brand 
6477583c59eSCedric Chaumont 	res = TEE_CopyObjectAttributes1(operation->key1, key);
6487583c59eSCedric Chaumont 	if (res != TEE_SUCCESS)
649a57c1e2eSCedric Chaumont 		goto out;
6507583c59eSCedric Chaumont 
651b0104773SPascal Brand 	operation->info.handleState |= TEE_HANDLE_FLAG_KEY_SET;
652b0104773SPascal Brand 
653b0104773SPascal Brand 	operation->info.keySize = key_size;
654b0104773SPascal Brand 
655287a6182SJens Wiklander 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
656287a6182SJens Wiklander 		reset_operation_state(operation);
657287a6182SJens Wiklander 
6587583c59eSCedric Chaumont out:
659a57c1e2eSCedric Chaumont 	if (res != TEE_SUCCESS  &&
660a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_CORRUPT_OBJECT &&
661a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_STORAGE_NOT_AVAILABLE)
662b36311adSJerome Forissier 		TEE_Panic(res);
663a57c1e2eSCedric Chaumont 
664a57c1e2eSCedric Chaumont 	return res;
665b0104773SPascal Brand }
666b0104773SPascal Brand 
667287a6182SJens Wiklander TEE_Result __GP11_TEE_SetOperationKey(TEE_OperationHandle operation,
668287a6182SJens Wiklander 				      TEE_ObjectHandle key)
669287a6182SJens Wiklander {
670287a6182SJens Wiklander 	if (operation == TEE_HANDLE_NULL ||
671287a6182SJens Wiklander 	    operation->operationState != TEE_OPERATION_STATE_INITIAL)
672287a6182SJens Wiklander 		TEE_Panic(0);
673287a6182SJens Wiklander 
674287a6182SJens Wiklander 	return TEE_SetOperationKey(operation, key);
675287a6182SJens Wiklander }
676287a6182SJens Wiklander 
677287a6182SJens Wiklander static TEE_Result set_operation_key2(TEE_OperationHandle operation,
678287a6182SJens Wiklander 				     TEE_ObjectHandle key1,
679287a6182SJens Wiklander 				     TEE_ObjectHandle key2)
680b0104773SPascal Brand {
6817583c59eSCedric Chaumont 	TEE_Result res;
682b0104773SPascal Brand 	uint32_t key_size = 0;
683b0104773SPascal Brand 	TEE_ObjectInfo key_info1;
684b0104773SPascal Brand 	TEE_ObjectInfo key_info2;
685b0104773SPascal Brand 
686a57c1e2eSCedric Chaumont 	if (operation == TEE_HANDLE_NULL) {
687a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
688a57c1e2eSCedric Chaumont 		goto out;
689a57c1e2eSCedric Chaumont 	}
690a57c1e2eSCedric Chaumont 
691a57c1e2eSCedric Chaumont 	/*
692a57c1e2eSCedric Chaumont 	 * Key1/Key2 and/or are not initialized and
693a57c1e2eSCedric Chaumont 	 * Either both keys are NULL or both are not NULL
694a57c1e2eSCedric Chaumont 	 */
6956c4ea258SJens Wiklander 	if (!key1 && !key2) {
6966c4ea258SJens Wiklander 		/* Clear the keys */
697a57c1e2eSCedric Chaumont 		TEE_ResetTransientObject(operation->key1);
698a57c1e2eSCedric Chaumont 		TEE_ResetTransientObject(operation->key2);
6996c4ea258SJens Wiklander 		operation->info.handleState &= ~TEE_HANDLE_FLAG_KEY_SET;
700287a6182SJens Wiklander 		if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
701287a6182SJens Wiklander 			reset_operation_state(operation);
7026c4ea258SJens Wiklander 		return TEE_SUCCESS;
7036c4ea258SJens Wiklander 	} else if (!key1 || !key2) {
7046c4ea258SJens Wiklander 		/* Both keys are obviously not valid. */
705a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
706a57c1e2eSCedric Chaumont 		goto out;
707a57c1e2eSCedric Chaumont 	}
708a57c1e2eSCedric Chaumont 
709a57c1e2eSCedric Chaumont 	/* No key for digest operation */
710a57c1e2eSCedric Chaumont 	if (operation->info.operationClass == TEE_OPERATION_DIGEST) {
711a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
712a57c1e2eSCedric Chaumont 		goto out;
713a57c1e2eSCedric Chaumont 	}
714a57c1e2eSCedric Chaumont 
7155b385b3fSJerome Forissier 	/* Two keys flag expected (TEE_ALG_AES_XTS and TEE_ALG_SM2_KEP only) */
716a57c1e2eSCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_EXPECT_TWO_KEYS) ==
717a57c1e2eSCedric Chaumont 	    0) {
718a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
719a57c1e2eSCedric Chaumont 		goto out;
720a57c1e2eSCedric Chaumont 	}
721a57c1e2eSCedric Chaumont 
7227583c59eSCedric Chaumont 	res = TEE_GetObjectInfo1(key1, &key_info1);
723a57c1e2eSCedric Chaumont 	/* Key1 is not a valid handle */
7247583c59eSCedric Chaumont 	if (res != TEE_SUCCESS)
725a57c1e2eSCedric Chaumont 		goto out;
7267583c59eSCedric Chaumont 
727b0104773SPascal Brand 	/* Supplied key has to meet required usage */
728b0104773SPascal Brand 	if ((key_info1.objectUsage & operation->info.
729b0104773SPascal Brand 	     requiredKeyUsage) != operation->info.requiredKeyUsage) {
730a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
731a57c1e2eSCedric Chaumont 		goto out;
732b0104773SPascal Brand 	}
733b0104773SPascal Brand 
7347583c59eSCedric Chaumont 	res = TEE_GetObjectInfo1(key2, &key_info2);
735a57c1e2eSCedric Chaumont 	/* Key2 is not a valid handle */
7367583c59eSCedric Chaumont 	if (res != TEE_SUCCESS) {
7377583c59eSCedric Chaumont 		if (res == TEE_ERROR_CORRUPT_OBJECT)
7387583c59eSCedric Chaumont 			res = TEE_ERROR_CORRUPT_OBJECT_2;
739a57c1e2eSCedric Chaumont 		goto out;
7407583c59eSCedric Chaumont 	}
7417583c59eSCedric Chaumont 
742b0104773SPascal Brand 	/* Supplied key has to meet required usage */
743b0104773SPascal Brand 	if ((key_info2.objectUsage & operation->info.
744b0104773SPascal Brand 	     requiredKeyUsage) != operation->info.requiredKeyUsage) {
745a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
746a57c1e2eSCedric Chaumont 		goto out;
747b0104773SPascal Brand 	}
748b0104773SPascal Brand 
749b0104773SPascal Brand 	/*
7505b385b3fSJerome Forissier 	 * All the multi key algorithm currently supported requires the keys to
7515b385b3fSJerome Forissier 	 * be of equal size.
752b0104773SPascal Brand 	 */
753d372a47cSJens Wiklander 	if (key_info1.objectSize != key_info2.objectSize) {
754a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
755a57c1e2eSCedric Chaumont 		goto out;
756b0104773SPascal Brand 
757a57c1e2eSCedric Chaumont 	}
758a57c1e2eSCedric Chaumont 
759d372a47cSJens Wiklander 	if (operation->info.maxKeySize < key_info1.objectSize) {
760a57c1e2eSCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
761a57c1e2eSCedric Chaumont 		goto out;
762a57c1e2eSCedric Chaumont 	}
763b0104773SPascal Brand 
764b0104773SPascal Brand 	/*
765b0104773SPascal Brand 	 * Odd that only the size of one key should be reported while
766b0104773SPascal Brand 	 * size of two key are used when allocating the operation.
767b0104773SPascal Brand 	 */
768d372a47cSJens Wiklander 	key_size = key_info1.objectSize;
769b0104773SPascal Brand 
770b0104773SPascal Brand 	TEE_ResetTransientObject(operation->key1);
771b0104773SPascal Brand 	TEE_ResetTransientObject(operation->key2);
772b0104773SPascal Brand 	operation->info.handleState &= ~TEE_HANDLE_FLAG_KEY_SET;
773b0104773SPascal Brand 
7747583c59eSCedric Chaumont 	res = TEE_CopyObjectAttributes1(operation->key1, key1);
7757583c59eSCedric Chaumont 	if (res != TEE_SUCCESS)
776a57c1e2eSCedric Chaumont 		goto out;
7777583c59eSCedric Chaumont 	res = TEE_CopyObjectAttributes1(operation->key2, key2);
7787583c59eSCedric Chaumont 	if (res != TEE_SUCCESS) {
7797583c59eSCedric Chaumont 		if (res == TEE_ERROR_CORRUPT_OBJECT)
7807583c59eSCedric Chaumont 			res = TEE_ERROR_CORRUPT_OBJECT_2;
781a57c1e2eSCedric Chaumont 		goto out;
7827583c59eSCedric Chaumont 	}
7837583c59eSCedric Chaumont 
784b0104773SPascal Brand 	operation->info.handleState |= TEE_HANDLE_FLAG_KEY_SET;
785b0104773SPascal Brand 
786b0104773SPascal Brand 	operation->info.keySize = key_size;
787b0104773SPascal Brand 
788287a6182SJens Wiklander 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
789287a6182SJens Wiklander 		reset_operation_state(operation);
7907583c59eSCedric Chaumont out:
791a57c1e2eSCedric Chaumont 	if (res != TEE_SUCCESS  &&
792a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_CORRUPT_OBJECT &&
793a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_CORRUPT_OBJECT_2 &&
794a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_STORAGE_NOT_AVAILABLE &&
795a57c1e2eSCedric Chaumont 	    res != TEE_ERROR_STORAGE_NOT_AVAILABLE_2)
796b36311adSJerome Forissier 		TEE_Panic(res);
797a57c1e2eSCedric Chaumont 
798a57c1e2eSCedric Chaumont 	return res;
799b0104773SPascal Brand }
800b0104773SPascal Brand 
801287a6182SJens Wiklander TEE_Result TEE_SetOperationKey2(TEE_OperationHandle operation,
802287a6182SJens Wiklander 				TEE_ObjectHandle key1, TEE_ObjectHandle key2)
803287a6182SJens Wiklander {
804287a6182SJens Wiklander 	if (operation != TEE_HANDLE_NULL && key1 && key1 == key2)
805287a6182SJens Wiklander 		return TEE_ERROR_SECURITY;
806287a6182SJens Wiklander 
807287a6182SJens Wiklander 	return set_operation_key2(operation, key1, key2);
808287a6182SJens Wiklander }
809287a6182SJens Wiklander 
810287a6182SJens Wiklander TEE_Result __GP11_TEE_SetOperationKey2(TEE_OperationHandle operation,
811287a6182SJens Wiklander 				       TEE_ObjectHandle key1,
812287a6182SJens Wiklander 				       TEE_ObjectHandle key2)
813287a6182SJens Wiklander {
814287a6182SJens Wiklander 	if (operation == TEE_HANDLE_NULL ||
815287a6182SJens Wiklander 	    operation->operationState != TEE_OPERATION_STATE_INITIAL)
816287a6182SJens Wiklander 		TEE_Panic(0);
817287a6182SJens Wiklander 
818287a6182SJens Wiklander 	return set_operation_key2(operation, key1, key2);
819287a6182SJens Wiklander }
820287a6182SJens Wiklander 
821b0104773SPascal Brand void TEE_CopyOperation(TEE_OperationHandle dst_op, TEE_OperationHandle src_op)
822b0104773SPascal Brand {
823b0104773SPascal Brand 	TEE_Result res;
824b0104773SPascal Brand 
825b0104773SPascal Brand 	if (dst_op == TEE_HANDLE_NULL || src_op == TEE_HANDLE_NULL)
826b0104773SPascal Brand 		TEE_Panic(0);
827b0104773SPascal Brand 	if (dst_op->info.algorithm != src_op->info.algorithm)
828b0104773SPascal Brand 		TEE_Panic(0);
8298734de30SJens Wiklander 	if (dst_op->info.mode != src_op->info.mode)
8308734de30SJens Wiklander 		TEE_Panic(0);
831b0104773SPascal Brand 	if (src_op->info.operationClass != TEE_OPERATION_DIGEST) {
832b0104773SPascal Brand 		TEE_ObjectHandle key1 = TEE_HANDLE_NULL;
833b0104773SPascal Brand 		TEE_ObjectHandle key2 = TEE_HANDLE_NULL;
834b0104773SPascal Brand 
835b0104773SPascal Brand 		if (src_op->info.handleState & TEE_HANDLE_FLAG_KEY_SET) {
836b0104773SPascal Brand 			key1 = src_op->key1;
837b0104773SPascal Brand 			key2 = src_op->key2;
838b0104773SPascal Brand 		}
839b0104773SPascal Brand 
840b0104773SPascal Brand 		if ((src_op->info.handleState &
841b0104773SPascal Brand 		     TEE_HANDLE_FLAG_EXPECT_TWO_KEYS) == 0) {
842b0104773SPascal Brand 			TEE_SetOperationKey(dst_op, key1);
843b0104773SPascal Brand 		} else {
844b0104773SPascal Brand 			TEE_SetOperationKey2(dst_op, key1, key2);
845b0104773SPascal Brand 		}
846b0104773SPascal Brand 	}
847b0104773SPascal Brand 	dst_op->info.handleState = src_op->info.handleState;
848b0104773SPascal Brand 	dst_op->info.keySize = src_op->info.keySize;
8498e07702eSJens Wiklander 	dst_op->info.digestLength = src_op->info.digestLength;
850642a1607SCedric Chaumont 	dst_op->operationState = src_op->operationState;
851b0104773SPascal Brand 
852b0104773SPascal Brand 	if (dst_op->buffer_two_blocks != src_op->buffer_two_blocks ||
853b0104773SPascal Brand 	    dst_op->block_size != src_op->block_size)
854b0104773SPascal Brand 		TEE_Panic(0);
855b0104773SPascal Brand 
856b0104773SPascal Brand 	if (dst_op->buffer != NULL) {
85724ea7613SJens Wiklander 		size_t sz = src_op->block_size;
85824ea7613SJens Wiklander 
859b0104773SPascal Brand 		if (src_op->buffer == NULL)
860b0104773SPascal Brand 			TEE_Panic(0);
861b0104773SPascal Brand 
86224ea7613SJens Wiklander 		if (src_op->buffer_two_blocks)
86324ea7613SJens Wiklander 			sz *= 2;
86424ea7613SJens Wiklander 		memcpy(dst_op->buffer, src_op->buffer, sz);
865b0104773SPascal Brand 		dst_op->buffer_offs = src_op->buffer_offs;
866b0104773SPascal Brand 	} else if (src_op->buffer != NULL) {
867b0104773SPascal Brand 		TEE_Panic(0);
868b0104773SPascal Brand 	}
869b0104773SPascal Brand 
8702c028fdeSJerome Forissier 	res = _utee_cryp_state_copy(dst_op->state, src_op->state);
871b0104773SPascal Brand 	if (res != TEE_SUCCESS)
872b0104773SPascal Brand 		TEE_Panic(res);
873b0104773SPascal Brand }
874b0104773SPascal Brand 
875b0104773SPascal Brand /* Cryptographic Operations API - Message Digest Functions */
876b0104773SPascal Brand 
8778f07fe6fSJerome Forissier static void init_hash_operation(TEE_OperationHandle operation, const void *IV,
8786d15db08SJerome Forissier 				uint32_t IVLen)
8796d15db08SJerome Forissier {
8806d15db08SJerome Forissier 	TEE_Result res;
8816d15db08SJerome Forissier 
8826d15db08SJerome Forissier 	/*
8836d15db08SJerome Forissier 	 * Note : IV and IVLen are never used in current implementation
8846d15db08SJerome Forissier 	 * This is why coherent values of IV and IVLen are not checked
8856d15db08SJerome Forissier 	 */
8862c028fdeSJerome Forissier 	res = _utee_hash_init(operation->state, IV, IVLen);
8876d15db08SJerome Forissier 	if (res != TEE_SUCCESS)
8886d15db08SJerome Forissier 		TEE_Panic(res);
8896d15db08SJerome Forissier 	operation->buffer_offs = 0;
8906d15db08SJerome Forissier 	operation->info.handleState |= TEE_HANDLE_FLAG_INITIALIZED;
8916d15db08SJerome Forissier }
8926d15db08SJerome Forissier 
893b0104773SPascal Brand void TEE_DigestUpdate(TEE_OperationHandle operation,
894185bf58cSJens Wiklander 		      const void *chunk, size_t chunkSize)
895b0104773SPascal Brand {
89673d6c3baSJoakim Bech 	TEE_Result res = TEE_ERROR_GENERIC;
897b0104773SPascal Brand 
89873d6c3baSJoakim Bech 	if (operation == TEE_HANDLE_NULL ||
89973d6c3baSJoakim Bech 	    operation->info.operationClass != TEE_OPERATION_DIGEST)
900b0104773SPascal Brand 		TEE_Panic(0);
90173d6c3baSJoakim Bech 
902642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_ACTIVE;
903642a1607SCedric Chaumont 
9042c028fdeSJerome Forissier 	res = _utee_hash_update(operation->state, chunk, chunkSize);
905b0104773SPascal Brand 	if (res != TEE_SUCCESS)
906b0104773SPascal Brand 		TEE_Panic(res);
907b0104773SPascal Brand }
908b0104773SPascal Brand 
909185bf58cSJens Wiklander void __GP11_TEE_DigestUpdate(TEE_OperationHandle operation,
910185bf58cSJens Wiklander 			     const void *chunk, uint32_t chunkSize)
911185bf58cSJens Wiklander {
912185bf58cSJens Wiklander 	return TEE_DigestUpdate(operation, chunk, chunkSize);
913185bf58cSJens Wiklander }
914185bf58cSJens Wiklander 
9158f07fe6fSJerome Forissier TEE_Result TEE_DigestDoFinal(TEE_OperationHandle operation, const void *chunk,
916185bf58cSJens Wiklander 			     size_t chunkLen, void *hash, size_t *hashLen)
917b0104773SPascal Brand {
91824ea7613SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
91924ea7613SJens Wiklander 	uint64_t hl = 0;
92024ea7613SJens Wiklander 	size_t len = 0;
92187c2f6b6SCedric Chaumont 
92287c2f6b6SCedric Chaumont 	if ((operation == TEE_HANDLE_NULL) ||
92387c2f6b6SCedric Chaumont 	    (!chunk && chunkLen) ||
92487c2f6b6SCedric Chaumont 	    (operation->info.operationClass != TEE_OPERATION_DIGEST)) {
92587c2f6b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
92687c2f6b6SCedric Chaumont 		goto out;
92787c2f6b6SCedric Chaumont 	}
92824ea7613SJens Wiklander 	if (operation->operationState == TEE_OPERATION_STATE_EXTRACTING &&
92924ea7613SJens Wiklander 	    chunkLen) {
93024ea7613SJens Wiklander 		res = TEE_ERROR_BAD_PARAMETERS;
93124ea7613SJens Wiklander 		goto out;
93224ea7613SJens Wiklander 	}
9336915bbbbSJens Wiklander 	__utee_check_inout_annotation(hashLen, sizeof(*hashLen));
93487c2f6b6SCedric Chaumont 
93524ea7613SJens Wiklander 	if (operation->operationState == TEE_OPERATION_STATE_EXTRACTING &&
93624ea7613SJens Wiklander 	    operation->buffer) {
93724ea7613SJens Wiklander 		/*
93824ea7613SJens Wiklander 		 * This is not an Extendable-Output Function and we have
93924ea7613SJens Wiklander 		 * already started extracting
94024ea7613SJens Wiklander 		 */
94124ea7613SJens Wiklander 		len = MIN(operation->block_size - operation->buffer_offs,
94224ea7613SJens Wiklander 			  *hashLen);
94324ea7613SJens Wiklander 		memcpy(hash, operation->buffer + operation->buffer_offs, len);
94424ea7613SJens Wiklander 		*hashLen = len;
94524ea7613SJens Wiklander 	} else {
946e86f1266SJens Wiklander 		hl = *hashLen;
94724ea7613SJens Wiklander 		res = _utee_hash_final(operation->state, chunk, chunkLen, hash,
94824ea7613SJens Wiklander 				       &hl);
949e86f1266SJens Wiklander 		*hashLen = hl;
95024ea7613SJens Wiklander 		if (res)
9516d15db08SJerome Forissier 			goto out;
95224ea7613SJens Wiklander 	}
9536d15db08SJerome Forissier 
9546d15db08SJerome Forissier 	/* Reset operation state */
9556d15db08SJerome Forissier 	init_hash_operation(operation, NULL, 0);
95687c2f6b6SCedric Chaumont 
957642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
958642a1607SCedric Chaumont 
95987c2f6b6SCedric Chaumont out:
96087c2f6b6SCedric Chaumont 	if (res != TEE_SUCCESS &&
96187c2f6b6SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
962b36311adSJerome Forissier 		TEE_Panic(res);
96373d6c3baSJoakim Bech 
96487c2f6b6SCedric Chaumont 	return res;
965b0104773SPascal Brand }
966b0104773SPascal Brand 
967185bf58cSJens Wiklander TEE_Result __GP11_TEE_DigestDoFinal(TEE_OperationHandle operation,
968185bf58cSJens Wiklander 				    const void *chunk, uint32_t chunkLen,
969185bf58cSJens Wiklander 				    void *hash, uint32_t *hashLen)
970185bf58cSJens Wiklander {
971185bf58cSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
972185bf58cSJens Wiklander 	size_t l = 0;
973185bf58cSJens Wiklander 
974185bf58cSJens Wiklander 	__utee_check_inout_annotation(hashLen, sizeof(*hashLen));
975185bf58cSJens Wiklander 	l = *hashLen;
976185bf58cSJens Wiklander 	res = TEE_DigestDoFinal(operation, chunk, chunkLen, hash, &l);
977185bf58cSJens Wiklander 	*hashLen = l;
978185bf58cSJens Wiklander 	return res;
979185bf58cSJens Wiklander }
980185bf58cSJens Wiklander 
98124ea7613SJens Wiklander TEE_Result TEE_DigestExtract(TEE_OperationHandle operation, void *hash,
98224ea7613SJens Wiklander 			     size_t *hashLen)
98324ea7613SJens Wiklander {
98424ea7613SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
98524ea7613SJens Wiklander 	uint64_t hl = 0;
98624ea7613SJens Wiklander 	size_t len = 0;
98724ea7613SJens Wiklander 
98824ea7613SJens Wiklander 	if (operation == TEE_HANDLE_NULL ||
98924ea7613SJens Wiklander 	    operation->info.operationClass != TEE_OPERATION_DIGEST)
99024ea7613SJens Wiklander 		TEE_Panic(0);
99124ea7613SJens Wiklander 	__utee_check_inout_annotation(hashLen, sizeof(*hashLen));
99224ea7613SJens Wiklander 
99324ea7613SJens Wiklander 	if (!operation->buffer) {
99424ea7613SJens Wiklander 		/* This is an Extendable-Output Function */
99524ea7613SJens Wiklander 		operation->info.handleState |= TEE_HANDLE_FLAG_EXTRACTING;
99624ea7613SJens Wiklander 		operation->operationState = TEE_OPERATION_STATE_EXTRACTING;
99724ea7613SJens Wiklander 		hl = *hashLen;
99824ea7613SJens Wiklander 		res = _utee_hash_final(operation->state, NULL, 0, hash, &hl);
99924ea7613SJens Wiklander 		if (res)
100024ea7613SJens Wiklander 			TEE_Panic(0);
100124ea7613SJens Wiklander 		*hashLen = hl;
100224ea7613SJens Wiklander 
100324ea7613SJens Wiklander 		return TEE_SUCCESS;
100424ea7613SJens Wiklander 	}
100524ea7613SJens Wiklander 
100624ea7613SJens Wiklander 	if (operation->operationState != TEE_OPERATION_STATE_EXTRACTING) {
100724ea7613SJens Wiklander 		hl = operation->block_size;
100824ea7613SJens Wiklander 		res = _utee_hash_final(operation->state, NULL, 0,
100924ea7613SJens Wiklander 				       operation->buffer, &hl);
101024ea7613SJens Wiklander 		if (res)
101124ea7613SJens Wiklander 			TEE_Panic(0);
101224ea7613SJens Wiklander 		if (hl != operation->block_size)
101324ea7613SJens Wiklander 			TEE_Panic(0);
101424ea7613SJens Wiklander 		assert(!operation->buffer_offs);
101524ea7613SJens Wiklander 		operation->info.handleState |= TEE_HANDLE_FLAG_EXTRACTING;
101624ea7613SJens Wiklander 		operation->operationState = TEE_OPERATION_STATE_EXTRACTING;
101724ea7613SJens Wiklander 	}
101824ea7613SJens Wiklander 
101924ea7613SJens Wiklander 	len = MIN(operation->block_size - operation->buffer_offs, *hashLen);
102024ea7613SJens Wiklander 	memcpy(hash, operation->buffer + operation->buffer_offs, len);
102124ea7613SJens Wiklander 	*hashLen = len;
102224ea7613SJens Wiklander 	operation->buffer_offs += len;
102324ea7613SJens Wiklander 
102424ea7613SJens Wiklander 	return TEE_SUCCESS;
102524ea7613SJens Wiklander }
102624ea7613SJens Wiklander 
1027b0104773SPascal Brand /* Cryptographic Operations API - Symmetric Cipher Functions */
1028b0104773SPascal Brand 
10298f07fe6fSJerome Forissier void TEE_CipherInit(TEE_OperationHandle operation, const void *IV,
1030185bf58cSJens Wiklander 		    size_t IVLen)
1031b0104773SPascal Brand {
1032b0104773SPascal Brand 	TEE_Result res;
1033b0104773SPascal Brand 
1034b0104773SPascal Brand 	if (operation == TEE_HANDLE_NULL)
1035b0104773SPascal Brand 		TEE_Panic(0);
1036642a1607SCedric Chaumont 
1037b0104773SPascal Brand 	if (operation->info.operationClass != TEE_OPERATION_CIPHER)
1038b0104773SPascal Brand 		TEE_Panic(0);
1039642a1607SCedric Chaumont 
1040642a1607SCedric Chaumont 	if (!(operation->info.handleState & TEE_HANDLE_FLAG_KEY_SET) ||
1041642a1607SCedric Chaumont 	    !(operation->key1))
1042642a1607SCedric Chaumont 		TEE_Panic(0);
1043642a1607SCedric Chaumont 
1044642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
1045642a1607SCedric Chaumont 		TEE_ResetOperation(operation);
1046642a1607SCedric Chaumont 
1047ad7aa2a5SSadiq Hussain 	if (IV && IVLen) {
1048ad7aa2a5SSadiq Hussain 		if (operation->info.algorithm == TEE_ALG_AES_ECB_NOPAD ||
1049ad7aa2a5SSadiq Hussain 		    operation->info.algorithm == TEE_ALG_DES_ECB_NOPAD ||
1050ad7aa2a5SSadiq Hussain 		    operation->info.algorithm == TEE_ALG_DES3_ECB_NOPAD ||
1051ad7aa2a5SSadiq Hussain 		    operation->info.algorithm == TEE_ALG_SM4_ECB_NOPAD)
1052ad7aa2a5SSadiq Hussain 			TEE_Panic(0);
1053ad7aa2a5SSadiq Hussain 	}
1054ad7aa2a5SSadiq Hussain 
1055642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_ACTIVE;
1056642a1607SCedric Chaumont 
10572c028fdeSJerome Forissier 	res = _utee_cipher_init(operation->state, IV, IVLen);
1058b0104773SPascal Brand 	if (res != TEE_SUCCESS)
1059b0104773SPascal Brand 		TEE_Panic(res);
1060642a1607SCedric Chaumont 
1061b0104773SPascal Brand 	operation->buffer_offs = 0;
1062b0104773SPascal Brand 	operation->info.handleState |= TEE_HANDLE_FLAG_INITIALIZED;
1063b0104773SPascal Brand }
1064b0104773SPascal Brand 
1065185bf58cSJens Wiklander void __GP11_TEE_CipherInit(TEE_OperationHandle operation, const void *IV,
1066185bf58cSJens Wiklander 			   uint32_t IVLen)
1067185bf58cSJens Wiklander {
1068185bf58cSJens Wiklander 	return TEE_CipherInit(operation, IV, IVLen);
1069185bf58cSJens Wiklander }
1070185bf58cSJens Wiklander 
1071b0104773SPascal Brand static TEE_Result tee_buffer_update(
1072b0104773SPascal Brand 		TEE_OperationHandle op,
1073e86f1266SJens Wiklander 		TEE_Result(*update_func)(unsigned long state, const void *src,
1074e86f1266SJens Wiklander 				size_t slen, void *dst, uint64_t *dlen),
1075b0104773SPascal Brand 		const void *src_data, size_t src_len,
1076e86f1266SJens Wiklander 		void *dest_data, uint64_t *dest_len)
1077b0104773SPascal Brand {
1078b0104773SPascal Brand 	TEE_Result res;
1079b0104773SPascal Brand 	const uint8_t *src = src_data;
1080b0104773SPascal Brand 	size_t slen = src_len;
1081b0104773SPascal Brand 	uint8_t *dst = dest_data;
1082b0104773SPascal Brand 	size_t dlen = *dest_len;
1083b0104773SPascal Brand 	size_t acc_dlen = 0;
1084e86f1266SJens Wiklander 	uint64_t tmp_dlen;
1085b0104773SPascal Brand 	size_t l;
1086b0104773SPascal Brand 	size_t buffer_size;
1087d3588802SPascal Brand 	size_t buffer_left;
1088b0104773SPascal Brand 
1089090268f5SJens Wiklander 	if (!src) {
1090090268f5SJens Wiklander 		if (slen)
1091090268f5SJens Wiklander 			TEE_Panic(0);
1092090268f5SJens Wiklander 		goto out;
1093090268f5SJens Wiklander 	}
1094090268f5SJens Wiklander 
1095d3588802SPascal Brand 	if (op->buffer_two_blocks) {
1096b0104773SPascal Brand 		buffer_size = op->block_size * 2;
1097d3588802SPascal Brand 		buffer_left = 1;
1098d3588802SPascal Brand 	} else {
1099b0104773SPascal Brand 		buffer_size = op->block_size;
1100d3588802SPascal Brand 		buffer_left = 0;
1101d3588802SPascal Brand 	}
1102b0104773SPascal Brand 
1103b0104773SPascal Brand 	if (op->buffer_offs > 0) {
1104b0104773SPascal Brand 		/* Fill up complete block */
1105b0104773SPascal Brand 		if (op->buffer_offs < op->block_size)
1106b0104773SPascal Brand 			l = MIN(slen, op->block_size - op->buffer_offs);
1107b0104773SPascal Brand 		else
1108b0104773SPascal Brand 			l = MIN(slen, buffer_size - op->buffer_offs);
1109b0104773SPascal Brand 		memcpy(op->buffer + op->buffer_offs, src, l);
1110b0104773SPascal Brand 		op->buffer_offs += l;
1111b0104773SPascal Brand 		src += l;
1112b0104773SPascal Brand 		slen -= l;
1113b0104773SPascal Brand 		if ((op->buffer_offs % op->block_size) != 0)
1114b0104773SPascal Brand 			goto out;	/* Nothing left to do */
1115b0104773SPascal Brand 	}
1116b0104773SPascal Brand 
1117b0104773SPascal Brand 	/* If we can feed from buffer */
1118d3588802SPascal Brand 	if ((op->buffer_offs > 0) &&
1119d3588802SPascal Brand 	    ((op->buffer_offs + slen) >= (buffer_size + buffer_left))) {
11202ff3fdbbSPascal Brand 		l = ROUNDUP(op->buffer_offs + slen - buffer_size,
1121b0104773SPascal Brand 				op->block_size);
1122b0104773SPascal Brand 		l = MIN(op->buffer_offs, l);
1123b0104773SPascal Brand 		tmp_dlen = dlen;
1124b0104773SPascal Brand 		res = update_func(op->state, op->buffer, l, dst, &tmp_dlen);
1125b0104773SPascal Brand 		if (res != TEE_SUCCESS)
1126b0104773SPascal Brand 			TEE_Panic(res);
1127b0104773SPascal Brand 		dst += tmp_dlen;
1128b0104773SPascal Brand 		dlen -= tmp_dlen;
1129b0104773SPascal Brand 		acc_dlen += tmp_dlen;
1130b0104773SPascal Brand 		op->buffer_offs -= l;
1131b0104773SPascal Brand 		if (op->buffer_offs > 0) {
1132b0104773SPascal Brand 			/*
1133b0104773SPascal Brand 			 * Slen is small enough to be contained in rest buffer.
1134b0104773SPascal Brand 			 */
1135b0104773SPascal Brand 			memcpy(op->buffer, op->buffer + l, buffer_size - l);
1136b0104773SPascal Brand 			memcpy(op->buffer + op->buffer_offs, src, slen);
1137b0104773SPascal Brand 			op->buffer_offs += slen;
1138b0104773SPascal Brand 			goto out;	/* Nothing left to do */
1139b0104773SPascal Brand 		}
1140b0104773SPascal Brand 	}
1141b0104773SPascal Brand 
1142d3588802SPascal Brand 	if (slen >= (buffer_size + buffer_left)) {
1143b0104773SPascal Brand 		/* Buffer is empty, feed as much as possible from src */
1144bf7a587fSJerome Forissier 		if (op->info.algorithm == TEE_ALG_AES_CTS)
1145b1ecda78SJerome Forissier 			l = ROUNDUP(slen - buffer_size, op->block_size);
1146bf7a587fSJerome Forissier 		else
1147bf7a587fSJerome Forissier 			l = ROUNDUP(slen - buffer_size + 1, op->block_size);
1148b0104773SPascal Brand 
1149b0104773SPascal Brand 		tmp_dlen = dlen;
1150b0104773SPascal Brand 		res = update_func(op->state, src, l, dst, &tmp_dlen);
1151b0104773SPascal Brand 		if (res != TEE_SUCCESS)
1152b0104773SPascal Brand 			TEE_Panic(res);
1153b0104773SPascal Brand 		src += l;
1154b0104773SPascal Brand 		slen -= l;
1155b0104773SPascal Brand 		dst += tmp_dlen;
1156b0104773SPascal Brand 		dlen -= tmp_dlen;
1157b0104773SPascal Brand 		acc_dlen += tmp_dlen;
1158b0104773SPascal Brand 	}
1159b0104773SPascal Brand 
1160b0104773SPascal Brand 	/* Slen is small enough to be contained in buffer. */
1161b0104773SPascal Brand 	memcpy(op->buffer + op->buffer_offs, src, slen);
1162b0104773SPascal Brand 	op->buffer_offs += slen;
1163b0104773SPascal Brand 
1164b0104773SPascal Brand out:
1165b0104773SPascal Brand 	*dest_len = acc_dlen;
1166b0104773SPascal Brand 	return TEE_SUCCESS;
1167b0104773SPascal Brand }
1168b0104773SPascal Brand 
11698f07fe6fSJerome Forissier TEE_Result TEE_CipherUpdate(TEE_OperationHandle operation, const void *srcData,
1170185bf58cSJens Wiklander 			    size_t srcLen, void *destData, size_t *destLen)
1171b0104773SPascal Brand {
1172dea1f2b6SCedric Chaumont 	TEE_Result res;
1173b0104773SPascal Brand 	size_t req_dlen;
1174e86f1266SJens Wiklander 	uint64_t dl;
1175b0104773SPascal Brand 
11766915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1177dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1178dea1f2b6SCedric Chaumont 		goto out;
1179dea1f2b6SCedric Chaumont 	}
11806915bbbbSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
1181dea1f2b6SCedric Chaumont 
1182642a1607SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_CIPHER) {
1183dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1184dea1f2b6SCedric Chaumont 		goto out;
1185dea1f2b6SCedric Chaumont 	}
1186dea1f2b6SCedric Chaumont 
1187642a1607SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1188642a1607SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1189642a1607SCedric Chaumont 		goto out;
1190642a1607SCedric Chaumont 	}
1191642a1607SCedric Chaumont 
1192642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_ACTIVE) {
1193dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1194dea1f2b6SCedric Chaumont 		goto out;
1195dea1f2b6SCedric Chaumont 	}
1196b0104773SPascal Brand 
1197e32c5ddfSJerome Forissier 	if (!srcData && !srcLen) {
1198090268f5SJens Wiklander 		*destLen = 0;
1199e32c5ddfSJerome Forissier 		res = TEE_SUCCESS;
1200e32c5ddfSJerome Forissier 		goto out;
1201e32c5ddfSJerome Forissier 	}
1202e32c5ddfSJerome Forissier 
1203b0104773SPascal Brand 	/* Calculate required dlen */
120457aabac5SBogdan Liulko 	if (operation->block_size > 1) {
120557aabac5SBogdan Liulko 		req_dlen = ((operation->buffer_offs + srcLen) /
120657aabac5SBogdan Liulko 			    operation->block_size) * operation->block_size;
120757aabac5SBogdan Liulko 	} else {
120857aabac5SBogdan Liulko 		req_dlen = srcLen;
120957aabac5SBogdan Liulko 	}
1210642a1607SCedric Chaumont 	if (operation->buffer_two_blocks) {
1211642a1607SCedric Chaumont 		if (req_dlen > operation->block_size * 2)
1212642a1607SCedric Chaumont 			req_dlen -= operation->block_size * 2;
1213b0104773SPascal Brand 		else
1214b0104773SPascal Brand 			req_dlen = 0;
1215b0104773SPascal Brand 	}
1216b0104773SPascal Brand 	/*
1217b0104773SPascal Brand 	 * Check that required destLen is big enough before starting to feed
1218b0104773SPascal Brand 	 * data to the algorithm. Errors during feeding of data are fatal as we
1219b0104773SPascal Brand 	 * can't restore sync with this API.
1220b0104773SPascal Brand 	 */
1221b0104773SPascal Brand 	if (*destLen < req_dlen) {
1222b0104773SPascal Brand 		*destLen = req_dlen;
1223dea1f2b6SCedric Chaumont 		res = TEE_ERROR_SHORT_BUFFER;
1224dea1f2b6SCedric Chaumont 		goto out;
1225b0104773SPascal Brand 	}
1226b0104773SPascal Brand 
1227e86f1266SJens Wiklander 	dl = *destLen;
122857aabac5SBogdan Liulko 	if (operation->block_size > 1) {
12292c028fdeSJerome Forissier 		res = tee_buffer_update(operation, _utee_cipher_update, srcData,
123057aabac5SBogdan Liulko 					srcLen, destData, &dl);
123157aabac5SBogdan Liulko 	} else {
123257aabac5SBogdan Liulko 		if (srcLen > 0) {
12332c028fdeSJerome Forissier 			res = _utee_cipher_update(operation->state, srcData,
123457aabac5SBogdan Liulko 						  srcLen, destData, &dl);
123557aabac5SBogdan Liulko 		} else {
123657aabac5SBogdan Liulko 			res = TEE_SUCCESS;
123757aabac5SBogdan Liulko 			dl = 0;
123857aabac5SBogdan Liulko 		}
123957aabac5SBogdan Liulko 	}
1240e86f1266SJens Wiklander 	*destLen = dl;
1241b0104773SPascal Brand 
1242dea1f2b6SCedric Chaumont out:
1243dea1f2b6SCedric Chaumont 	if (res != TEE_SUCCESS &&
1244dea1f2b6SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
1245b36311adSJerome Forissier 		TEE_Panic(res);
1246dea1f2b6SCedric Chaumont 
1247dea1f2b6SCedric Chaumont 	return res;
1248b0104773SPascal Brand }
1249b0104773SPascal Brand 
1250185bf58cSJens Wiklander TEE_Result __GP11_TEE_CipherUpdate(TEE_OperationHandle operation,
12518f07fe6fSJerome Forissier 				   const void *srcData, uint32_t srcLen,
12528f07fe6fSJerome Forissier 				   void *destData, uint32_t *destLen)
1253b0104773SPascal Brand {
12546915bbbbSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1255185bf58cSJens Wiklander 	size_t dl = 0;
1256185bf58cSJens Wiklander 
1257185bf58cSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
1258185bf58cSJens Wiklander 	dl = *destLen;
1259185bf58cSJens Wiklander 	res = TEE_CipherUpdate(operation, srcData, srcLen, destData, &dl);
1260185bf58cSJens Wiklander 	*destLen = dl;
1261185bf58cSJens Wiklander 	return res;
1262185bf58cSJens Wiklander }
1263185bf58cSJens Wiklander 
1264185bf58cSJens Wiklander TEE_Result TEE_CipherDoFinal(TEE_OperationHandle operation,
1265185bf58cSJens Wiklander 			     const void *srcData, size_t srcLen,
1266185bf58cSJens Wiklander 			     void *destData, size_t *destLen)
1267185bf58cSJens Wiklander {
1268185bf58cSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1269b0104773SPascal Brand 	uint8_t *dst = destData;
1270b0104773SPascal Brand 	size_t acc_dlen = 0;
12716915bbbbSJens Wiklander 	uint64_t tmp_dlen = 0;
12726915bbbbSJens Wiklander 	size_t req_dlen = 0;
1273b0104773SPascal Brand 
12746915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1275dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1276dea1f2b6SCedric Chaumont 		goto out;
1277dea1f2b6SCedric Chaumont 	}
12786915bbbbSJens Wiklander 	if (destLen)
12796915bbbbSJens Wiklander 		__utee_check_inout_annotation(destLen, sizeof(*destLen));
1280dea1f2b6SCedric Chaumont 
1281642a1607SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_CIPHER) {
1282dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1283dea1f2b6SCedric Chaumont 		goto out;
1284dea1f2b6SCedric Chaumont 	}
1285dea1f2b6SCedric Chaumont 
1286642a1607SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1287642a1607SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1288642a1607SCedric Chaumont 		goto out;
1289642a1607SCedric Chaumont 	}
1290642a1607SCedric Chaumont 
1291642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_ACTIVE) {
1292dea1f2b6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1293dea1f2b6SCedric Chaumont 		goto out;
1294dea1f2b6SCedric Chaumont 	}
1295b0104773SPascal Brand 
1296b0104773SPascal Brand 	/*
1297b0104773SPascal Brand 	 * Check that the final block doesn't require padding for those
1298b0104773SPascal Brand 	 * algorithms that requires client to supply padding.
1299b0104773SPascal Brand 	 */
1300642a1607SCedric Chaumont 	if (operation->info.algorithm == TEE_ALG_AES_ECB_NOPAD ||
1301642a1607SCedric Chaumont 	    operation->info.algorithm == TEE_ALG_AES_CBC_NOPAD ||
1302642a1607SCedric Chaumont 	    operation->info.algorithm == TEE_ALG_DES_ECB_NOPAD ||
1303642a1607SCedric Chaumont 	    operation->info.algorithm == TEE_ALG_DES_CBC_NOPAD ||
1304642a1607SCedric Chaumont 	    operation->info.algorithm == TEE_ALG_DES3_ECB_NOPAD ||
1305ade6f848SJerome Forissier 	    operation->info.algorithm == TEE_ALG_DES3_CBC_NOPAD ||
1306ade6f848SJerome Forissier 	    operation->info.algorithm == TEE_ALG_SM4_ECB_NOPAD ||
1307ade6f848SJerome Forissier 	    operation->info.algorithm == TEE_ALG_SM4_CBC_NOPAD) {
1308642a1607SCedric Chaumont 		if (((operation->buffer_offs + srcLen) % operation->block_size)
1309642a1607SCedric Chaumont 		    != 0) {
1310dea1f2b6SCedric Chaumont 			res = TEE_ERROR_BAD_PARAMETERS;
1311dea1f2b6SCedric Chaumont 			goto out;
1312dea1f2b6SCedric Chaumont 		}
1313b0104773SPascal Brand 	}
1314b0104773SPascal Brand 
1315b0104773SPascal Brand 	/*
1316b0104773SPascal Brand 	 * Check that required destLen is big enough before starting to feed
1317b0104773SPascal Brand 	 * data to the algorithm. Errors during feeding of data are fatal as we
1318b0104773SPascal Brand 	 * can't restore sync with this API.
1319b0104773SPascal Brand 	 */
132057aabac5SBogdan Liulko 	if (operation->block_size > 1) {
1321642a1607SCedric Chaumont 		req_dlen = operation->buffer_offs + srcLen;
132257aabac5SBogdan Liulko 	} else {
132357aabac5SBogdan Liulko 		req_dlen = srcLen;
132457aabac5SBogdan Liulko 	}
13256915bbbbSJens Wiklander 	if (destLen)
13266915bbbbSJens Wiklander 		tmp_dlen = *destLen;
13276915bbbbSJens Wiklander 	if (tmp_dlen < req_dlen) {
13286915bbbbSJens Wiklander 		if (destLen)
1329b0104773SPascal Brand 			*destLen = req_dlen;
1330dea1f2b6SCedric Chaumont 		res = TEE_ERROR_SHORT_BUFFER;
1331dea1f2b6SCedric Chaumont 		goto out;
1332b0104773SPascal Brand 	}
1333b0104773SPascal Brand 
133457aabac5SBogdan Liulko 	if (operation->block_size > 1) {
1335dea9063eSJens Wiklander 		if (srcLen) {
13362c028fdeSJerome Forissier 			res = tee_buffer_update(operation, _utee_cipher_update,
1337dea9063eSJens Wiklander 						srcData, srcLen, dst,
1338dea9063eSJens Wiklander 						&tmp_dlen);
1339dea1f2b6SCedric Chaumont 			if (res != TEE_SUCCESS)
1340dea1f2b6SCedric Chaumont 				goto out;
1341dea1f2b6SCedric Chaumont 
1342b0104773SPascal Brand 			dst += tmp_dlen;
1343b0104773SPascal Brand 			acc_dlen += tmp_dlen;
1344b0104773SPascal Brand 
1345b0104773SPascal Brand 			tmp_dlen = *destLen - acc_dlen;
1346dea9063eSJens Wiklander 		}
13472c028fdeSJerome Forissier 		res = _utee_cipher_final(operation->state, operation->buffer,
13482c028fdeSJerome Forissier 					 operation->buffer_offs, dst,
13492c028fdeSJerome Forissier 					 &tmp_dlen);
135057aabac5SBogdan Liulko 	} else {
13512c028fdeSJerome Forissier 		res = _utee_cipher_final(operation->state, srcData, srcLen, dst,
13522c028fdeSJerome Forissier 					 &tmp_dlen);
135357aabac5SBogdan Liulko 	}
1354b0104773SPascal Brand 	if (res != TEE_SUCCESS)
1355dea1f2b6SCedric Chaumont 		goto out;
1356dea1f2b6SCedric Chaumont 
1357b0104773SPascal Brand 	acc_dlen += tmp_dlen;
13586915bbbbSJens Wiklander 	if (destLen)
1359b0104773SPascal Brand 		*destLen = acc_dlen;
1360dea1f2b6SCedric Chaumont 
1361642a1607SCedric Chaumont 	operation->info.handleState &= ~TEE_HANDLE_FLAG_INITIALIZED;
1362642a1607SCedric Chaumont 
1363642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
1364642a1607SCedric Chaumont 
1365dea1f2b6SCedric Chaumont out:
1366dea1f2b6SCedric Chaumont 	if (res != TEE_SUCCESS &&
1367dea1f2b6SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
1368b36311adSJerome Forissier 		TEE_Panic(res);
1369dea1f2b6SCedric Chaumont 
1370dea1f2b6SCedric Chaumont 	return res;
1371b0104773SPascal Brand }
1372b0104773SPascal Brand 
1373185bf58cSJens Wiklander TEE_Result __GP11_TEE_CipherDoFinal(TEE_OperationHandle operation,
1374185bf58cSJens Wiklander 				    const void *srcData, uint32_t srcLen,
1375185bf58cSJens Wiklander 				    void *destData, uint32_t *destLen)
1376185bf58cSJens Wiklander {
1377185bf58cSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1378185bf58cSJens Wiklander 	size_t dl = 0;
1379185bf58cSJens Wiklander 
1380185bf58cSJens Wiklander 	if (destLen) {
1381185bf58cSJens Wiklander 		__utee_check_inout_annotation(destLen, sizeof(*destLen));
1382185bf58cSJens Wiklander 		dl = *destLen;
1383185bf58cSJens Wiklander 	}
1384185bf58cSJens Wiklander 	res = TEE_CipherDoFinal(operation, srcData, srcLen, destData, &dl);
1385185bf58cSJens Wiklander 	if (destLen)
1386185bf58cSJens Wiklander 		*destLen = dl;
1387185bf58cSJens Wiklander 	return res;
1388185bf58cSJens Wiklander }
1389185bf58cSJens Wiklander 
1390b0104773SPascal Brand /* Cryptographic Operations API - MAC Functions */
1391b0104773SPascal Brand 
1392185bf58cSJens Wiklander void TEE_MACInit(TEE_OperationHandle operation, const void *IV, size_t IVLen)
1393b0104773SPascal Brand {
1394b0104773SPascal Brand 	if (operation == TEE_HANDLE_NULL)
1395b0104773SPascal Brand 		TEE_Panic(0);
1396642a1607SCedric Chaumont 
1397b0104773SPascal Brand 	if (operation->info.operationClass != TEE_OPERATION_MAC)
1398b0104773SPascal Brand 		TEE_Panic(0);
1399642a1607SCedric Chaumont 
1400642a1607SCedric Chaumont 	if (!(operation->info.handleState & TEE_HANDLE_FLAG_KEY_SET) ||
1401642a1607SCedric Chaumont 	    !(operation->key1))
1402642a1607SCedric Chaumont 		TEE_Panic(0);
1403642a1607SCedric Chaumont 
1404642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
1405642a1607SCedric Chaumont 		TEE_ResetOperation(operation);
1406642a1607SCedric Chaumont 
1407642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_ACTIVE;
1408642a1607SCedric Chaumont 
14096d15db08SJerome Forissier 	init_hash_operation(operation, IV, IVLen);
1410b0104773SPascal Brand }
1411b0104773SPascal Brand 
1412185bf58cSJens Wiklander void __GP11_TEE_MACInit(TEE_OperationHandle operation, const void *IV,
1413185bf58cSJens Wiklander 			uint32_t IVLen)
1414185bf58cSJens Wiklander {
1415185bf58cSJens Wiklander 	return TEE_MACInit(operation, IV, IVLen);
1416185bf58cSJens Wiklander }
1417185bf58cSJens Wiklander 
14188f07fe6fSJerome Forissier void TEE_MACUpdate(TEE_OperationHandle operation, const void *chunk,
1419185bf58cSJens Wiklander 		   size_t chunkSize)
1420b0104773SPascal Brand {
1421b0104773SPascal Brand 	TEE_Result res;
1422b0104773SPascal Brand 
142328e0efc6SCedric Chaumont 	if (operation == TEE_HANDLE_NULL || (chunk == NULL && chunkSize != 0))
1424b0104773SPascal Brand 		TEE_Panic(0);
1425642a1607SCedric Chaumont 
142628e0efc6SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_MAC)
1427b0104773SPascal Brand 		TEE_Panic(0);
1428642a1607SCedric Chaumont 
142928e0efc6SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0)
1430b0104773SPascal Brand 		TEE_Panic(0);
1431b0104773SPascal Brand 
1432642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_ACTIVE)
1433642a1607SCedric Chaumont 		TEE_Panic(0);
1434642a1607SCedric Chaumont 
14352c028fdeSJerome Forissier 	res = _utee_hash_update(operation->state, chunk, chunkSize);
1436b0104773SPascal Brand 	if (res != TEE_SUCCESS)
1437b0104773SPascal Brand 		TEE_Panic(res);
1438b0104773SPascal Brand }
1439b0104773SPascal Brand 
1440185bf58cSJens Wiklander void __GP11_TEE_MACUpdate(TEE_OperationHandle operation, const void *chunk,
1441185bf58cSJens Wiklander 			  uint32_t chunkSize)
1442185bf58cSJens Wiklander {
1443185bf58cSJens Wiklander 	return TEE_MACUpdate(operation, chunk, chunkSize);
1444185bf58cSJens Wiklander }
1445185bf58cSJens Wiklander 
144628e0efc6SCedric Chaumont TEE_Result TEE_MACComputeFinal(TEE_OperationHandle operation,
1447185bf58cSJens Wiklander 			       const void *message, size_t messageLen,
1448185bf58cSJens Wiklander 			       void *mac, size_t *macLen)
1449b0104773SPascal Brand {
1450b0104773SPascal Brand 	TEE_Result res;
1451e86f1266SJens Wiklander 	uint64_t ml;
1452b0104773SPascal Brand 
14536915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!message && messageLen)) {
145428e0efc6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
145528e0efc6SCedric Chaumont 		goto out;
145628e0efc6SCedric Chaumont 	}
14576915bbbbSJens Wiklander 	__utee_check_inout_annotation(macLen, sizeof(*macLen));
1458b0104773SPascal Brand 
145928e0efc6SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_MAC) {
146028e0efc6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
146128e0efc6SCedric Chaumont 		goto out;
146228e0efc6SCedric Chaumont 	}
146328e0efc6SCedric Chaumont 
146428e0efc6SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
146528e0efc6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
146628e0efc6SCedric Chaumont 		goto out;
146728e0efc6SCedric Chaumont 	}
146828e0efc6SCedric Chaumont 
1469642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_ACTIVE) {
1470642a1607SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1471642a1607SCedric Chaumont 		goto out;
1472642a1607SCedric Chaumont 	}
1473642a1607SCedric Chaumont 
1474e86f1266SJens Wiklander 	ml = *macLen;
14752c028fdeSJerome Forissier 	res = _utee_hash_final(operation->state, message, messageLen, mac, &ml);
1476e86f1266SJens Wiklander 	*macLen = ml;
147728e0efc6SCedric Chaumont 	if (res != TEE_SUCCESS)
147828e0efc6SCedric Chaumont 		goto out;
147928e0efc6SCedric Chaumont 
148028e0efc6SCedric Chaumont 	operation->info.handleState &= ~TEE_HANDLE_FLAG_INITIALIZED;
148128e0efc6SCedric Chaumont 
1482642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
1483642a1607SCedric Chaumont 
148428e0efc6SCedric Chaumont out:
148528e0efc6SCedric Chaumont 	if (res != TEE_SUCCESS &&
148628e0efc6SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
148728e0efc6SCedric Chaumont 		TEE_Panic(res);
148828e0efc6SCedric Chaumont 
1489b0104773SPascal Brand 	return res;
1490b0104773SPascal Brand }
1491b0104773SPascal Brand 
1492185bf58cSJens Wiklander TEE_Result __GP11_TEE_MACComputeFinal(TEE_OperationHandle operation,
14938f07fe6fSJerome Forissier 				      const void *message, uint32_t messageLen,
1494185bf58cSJens Wiklander 				      void *mac, uint32_t *macLen)
1495185bf58cSJens Wiklander {
1496185bf58cSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1497185bf58cSJens Wiklander 	size_t ml = 0;
1498185bf58cSJens Wiklander 
1499185bf58cSJens Wiklander 	__utee_check_inout_annotation(macLen, sizeof(*macLen));
1500185bf58cSJens Wiklander 	ml = *macLen;
1501185bf58cSJens Wiklander 	res = TEE_MACComputeFinal(operation, message, messageLen, mac, &ml);
1502185bf58cSJens Wiklander 	*macLen = ml;
1503185bf58cSJens Wiklander 	return res;
1504185bf58cSJens Wiklander }
1505185bf58cSJens Wiklander 
1506185bf58cSJens Wiklander TEE_Result TEE_MACCompareFinal(TEE_OperationHandle operation,
1507185bf58cSJens Wiklander 			       const void *message, size_t messageLen,
1508185bf58cSJens Wiklander 			       const void *mac, size_t macLen)
1509b0104773SPascal Brand {
1510b0104773SPascal Brand 	TEE_Result res;
1511ee4ba3d1SVictor Chong 	uint8_t computed_mac[TEE_MAX_HASH_SIZE] = { 0 };
1512185bf58cSJens Wiklander 	size_t computed_mac_size = TEE_MAX_HASH_SIZE;
1513b0104773SPascal Brand 
151428e0efc6SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_MAC) {
151528e0efc6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
151628e0efc6SCedric Chaumont 		goto out;
151728e0efc6SCedric Chaumont 	}
151828e0efc6SCedric Chaumont 
151928e0efc6SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
152028e0efc6SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
152128e0efc6SCedric Chaumont 		goto out;
152228e0efc6SCedric Chaumont 	}
152328e0efc6SCedric Chaumont 
1524642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_ACTIVE) {
1525642a1607SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1526642a1607SCedric Chaumont 		goto out;
1527642a1607SCedric Chaumont 	}
1528642a1607SCedric Chaumont 
1529b0104773SPascal Brand 	res = TEE_MACComputeFinal(operation, message, messageLen, computed_mac,
1530b0104773SPascal Brand 				  &computed_mac_size);
1531b0104773SPascal Brand 	if (res != TEE_SUCCESS)
153228e0efc6SCedric Chaumont 		goto out;
153328e0efc6SCedric Chaumont 
153428e0efc6SCedric Chaumont 	if (computed_mac_size != macLen) {
153528e0efc6SCedric Chaumont 		res = TEE_ERROR_MAC_INVALID;
153628e0efc6SCedric Chaumont 		goto out;
153728e0efc6SCedric Chaumont 	}
153828e0efc6SCedric Chaumont 
153948e10604SJerome Forissier 	if (consttime_memcmp(mac, computed_mac, computed_mac_size) != 0) {
154028e0efc6SCedric Chaumont 		res = TEE_ERROR_MAC_INVALID;
154128e0efc6SCedric Chaumont 		goto out;
154228e0efc6SCedric Chaumont 	}
154328e0efc6SCedric Chaumont 
1544642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
1545642a1607SCedric Chaumont 
154628e0efc6SCedric Chaumont out:
154728e0efc6SCedric Chaumont 	if (res != TEE_SUCCESS &&
154828e0efc6SCedric Chaumont 	    res != TEE_ERROR_MAC_INVALID)
154928e0efc6SCedric Chaumont 		TEE_Panic(res);
155028e0efc6SCedric Chaumont 
1551b0104773SPascal Brand 	return res;
1552b0104773SPascal Brand }
1553b0104773SPascal Brand 
1554185bf58cSJens Wiklander TEE_Result __GP11_TEE_MACCompareFinal(TEE_OperationHandle operation,
1555185bf58cSJens Wiklander 				      const void *message, uint32_t messageLen,
1556185bf58cSJens Wiklander 				      const void *mac, uint32_t macLen)
1557185bf58cSJens Wiklander {
1558185bf58cSJens Wiklander 	return TEE_MACCompareFinal(operation, message, messageLen, mac, macLen);
1559185bf58cSJens Wiklander }
1560185bf58cSJens Wiklander 
1561b0104773SPascal Brand /* Cryptographic Operations API - Authenticated Encryption Functions */
1562b0104773SPascal Brand 
15638f07fe6fSJerome Forissier TEE_Result TEE_AEInit(TEE_OperationHandle operation, const void *nonce,
1564d9096215SJens Wiklander 		      size_t nonceLen, uint32_t tagLen, size_t AADLen,
1565d9096215SJens Wiklander 		      size_t payloadLen)
1566b0104773SPascal Brand {
1567b0104773SPascal Brand 	TEE_Result res;
1568b0104773SPascal Brand 
1569b5816c88SCedric Chaumont 	if (operation == TEE_HANDLE_NULL || nonce == NULL) {
1570b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1571b5816c88SCedric Chaumont 		goto out;
1572b5816c88SCedric Chaumont 	}
1573b5816c88SCedric Chaumont 
1574b5816c88SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_AE) {
1575b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1576b5816c88SCedric Chaumont 		goto out;
1577b5816c88SCedric Chaumont 	}
1578b0104773SPascal Brand 
1579642a1607SCedric Chaumont 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL) {
1580642a1607SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1581642a1607SCedric Chaumont 		goto out;
1582642a1607SCedric Chaumont 	}
1583642a1607SCedric Chaumont 
1584b0104773SPascal Brand 	/*
1585b0104773SPascal Brand 	 * AES-CCM tag len is specified by AES-CCM spec and handled in TEE Core
1586b0104773SPascal Brand 	 * in the implementation. But AES-GCM spec doesn't specify the tag len
1587b0104773SPascal Brand 	 * according to the same principle so we have to check here instead to
1588b0104773SPascal Brand 	 * be GP compliant.
1589b0104773SPascal Brand 	 */
1590b5816c88SCedric Chaumont 	if (operation->info.algorithm == TEE_ALG_AES_GCM) {
1591b0104773SPascal Brand 		/*
1592b0104773SPascal Brand 		 * From GP spec: For AES-GCM, can be 128, 120, 112, 104, or 96
1593b0104773SPascal Brand 		 */
1594b5816c88SCedric Chaumont 		if (tagLen < 96 || tagLen > 128 || (tagLen % 8 != 0)) {
1595b5816c88SCedric Chaumont 			res = TEE_ERROR_NOT_SUPPORTED;
1596b5816c88SCedric Chaumont 			goto out;
1597b5816c88SCedric Chaumont 		}
1598b0104773SPascal Brand 	}
1599b0104773SPascal Brand 
16002c028fdeSJerome Forissier 	res = _utee_authenc_init(operation->state, nonce, nonceLen, tagLen / 8,
16012c028fdeSJerome Forissier 				 AADLen, payloadLen);
1602b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS)
1603b5816c88SCedric Chaumont 		goto out;
1604b5816c88SCedric Chaumont 
16057acaf5adSAlbert Schwarzkopf 	operation->info.digestLength = tagLen / 8;
1606f2674567SSumit Garg 	operation->buffer_offs = 0;
1607b5816c88SCedric Chaumont 	operation->info.handleState |= TEE_HANDLE_FLAG_INITIALIZED;
1608b5816c88SCedric Chaumont 
1609b5816c88SCedric Chaumont out:
1610b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS &&
1611b5816c88SCedric Chaumont 	    res != TEE_ERROR_NOT_SUPPORTED)
1612b0104773SPascal Brand 			TEE_Panic(res);
1613b5816c88SCedric Chaumont 
1614b0104773SPascal Brand 	return res;
1615b0104773SPascal Brand }
1616b0104773SPascal Brand 
1617d9096215SJens Wiklander TEE_Result __GP11_TEE_AEInit(TEE_OperationHandle operation, const void *nonce,
1618d9096215SJens Wiklander 			     uint32_t nonceLen, uint32_t tagLen,
1619d9096215SJens Wiklander 			     uint32_t AADLen, uint32_t payloadLen)
1620d9096215SJens Wiklander {
1621d9096215SJens Wiklander 	return TEE_AEInit(operation, nonce, nonceLen, tagLen, AADLen,
1622d9096215SJens Wiklander 			  payloadLen);
1623d9096215SJens Wiklander }
1624d9096215SJens Wiklander 
16258f07fe6fSJerome Forissier void TEE_AEUpdateAAD(TEE_OperationHandle operation, const void *AADdata,
1626d9096215SJens Wiklander 		     size_t AADdataLen)
1627d9096215SJens Wiklander {
1628d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1629d9096215SJens Wiklander 
1630d9096215SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!AADdata && AADdataLen))
1631d9096215SJens Wiklander 		TEE_Panic(0);
1632d9096215SJens Wiklander 
1633d9096215SJens Wiklander 	if (operation->info.operationClass != TEE_OPERATION_AE)
1634d9096215SJens Wiklander 		TEE_Panic(0);
1635d9096215SJens Wiklander 
1636d9096215SJens Wiklander 	if (operation->operationState != TEE_OPERATION_STATE_INITIAL)
1637d9096215SJens Wiklander 		TEE_Panic(0);
1638d9096215SJens Wiklander 
1639d9096215SJens Wiklander 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0)
1640d9096215SJens Wiklander 		TEE_Panic(0);
1641d9096215SJens Wiklander 
1642d9096215SJens Wiklander 	res = _utee_authenc_update_aad(operation->state, AADdata, AADdataLen);
1643d9096215SJens Wiklander 	if (res != TEE_SUCCESS)
1644d9096215SJens Wiklander 		TEE_Panic(res);
1645d9096215SJens Wiklander }
1646d9096215SJens Wiklander 
1647d9096215SJens Wiklander void __GP11_TEE_AEUpdateAAD(TEE_OperationHandle operation, const void *AADdata,
164879a3c601SCedric Chaumont 			    uint32_t AADdataLen)
1649b0104773SPascal Brand {
1650d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1651b0104773SPascal Brand 
1652b5816c88SCedric Chaumont 	if (operation == TEE_HANDLE_NULL ||
1653b5816c88SCedric Chaumont 	    (AADdata == NULL && AADdataLen != 0))
1654b0104773SPascal Brand 		TEE_Panic(0);
1655642a1607SCedric Chaumont 
1656b5816c88SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_AE)
1657b0104773SPascal Brand 		TEE_Panic(0);
1658642a1607SCedric Chaumont 
1659b5816c88SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0)
1660b0104773SPascal Brand 		TEE_Panic(0);
1661b0104773SPascal Brand 
16622c028fdeSJerome Forissier 	res = _utee_authenc_update_aad(operation->state, AADdata, AADdataLen);
1663642a1607SCedric Chaumont 
1664642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_ACTIVE;
1665642a1607SCedric Chaumont 
1666b0104773SPascal Brand 	if (res != TEE_SUCCESS)
1667b0104773SPascal Brand 		TEE_Panic(res);
1668b0104773SPascal Brand }
1669b0104773SPascal Brand 
1670d9096215SJens Wiklander static TEE_Result ae_update_helper(TEE_OperationHandle operation,
1671d9096215SJens Wiklander 				   const void *src, size_t slen, void *dst,
1672d9096215SJens Wiklander 				   size_t *dlen)
1673b0104773SPascal Brand {
16746915bbbbSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
16756915bbbbSJens Wiklander 	size_t req_dlen = 0;
16766915bbbbSJens Wiklander 	uint64_t dl = 0;
1677b0104773SPascal Brand 
1678d9096215SJens Wiklander 	if (!src && !slen) {
1679d9096215SJens Wiklander 		*dlen = 0;
1680d9096215SJens Wiklander 		return TEE_SUCCESS;
1681d9096215SJens Wiklander 	}
1682d9096215SJens Wiklander 
1683d9096215SJens Wiklander 	/*
1684d9096215SJens Wiklander 	 * Check that required destLen is big enough before starting to feed
1685d9096215SJens Wiklander 	 * data to the algorithm. Errors during feeding of data are fatal as we
1686d9096215SJens Wiklander 	 * can't restore sync with this API.
1687d9096215SJens Wiklander 	 */
1688d9096215SJens Wiklander 	if (operation->block_size > 1) {
1689d9096215SJens Wiklander 		req_dlen = ROUNDDOWN(operation->buffer_offs + slen,
1690d9096215SJens Wiklander 				     operation->block_size);
1691d9096215SJens Wiklander 	} else {
1692d9096215SJens Wiklander 		req_dlen = slen;
1693d9096215SJens Wiklander 	}
1694d9096215SJens Wiklander 
1695d9096215SJens Wiklander 	dl = *dlen;
1696d9096215SJens Wiklander 	if (dl < req_dlen) {
1697d9096215SJens Wiklander 		*dlen = req_dlen;
1698d9096215SJens Wiklander 		return TEE_ERROR_SHORT_BUFFER;
1699d9096215SJens Wiklander 	}
1700d9096215SJens Wiklander 
1701d9096215SJens Wiklander 	if (operation->block_size > 1) {
1702d9096215SJens Wiklander 		res = tee_buffer_update(operation, _utee_authenc_update_payload,
1703d9096215SJens Wiklander 					src, slen, dst, &dl);
1704d9096215SJens Wiklander 	} else {
1705d9096215SJens Wiklander 		if (slen > 0) {
1706d9096215SJens Wiklander 			res = _utee_authenc_update_payload(operation->state,
1707d9096215SJens Wiklander 							   src, slen, dst, &dl);
1708d9096215SJens Wiklander 		} else {
1709d9096215SJens Wiklander 			dl = 0;
1710d9096215SJens Wiklander 			res = TEE_SUCCESS;
1711d9096215SJens Wiklander 		}
1712d9096215SJens Wiklander 	}
1713d9096215SJens Wiklander 
1714d9096215SJens Wiklander 	if (!res)
1715d9096215SJens Wiklander 		*dlen = dl;
1716d9096215SJens Wiklander 
1717d9096215SJens Wiklander 	return res;
1718d9096215SJens Wiklander }
1719d9096215SJens Wiklander 
1720d9096215SJens Wiklander TEE_Result TEE_AEUpdate(TEE_OperationHandle operation, const void *srcData,
1721d9096215SJens Wiklander 			size_t srcLen, void *destData, size_t *destLen)
1722d9096215SJens Wiklander {
1723d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1724d9096215SJens Wiklander 
17256915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1726b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1727b5816c88SCedric Chaumont 		goto out;
1728b5816c88SCedric Chaumont 	}
1729d9096215SJens Wiklander 	__utee_check_outbuf_annotation(destData, destLen);
1730b5816c88SCedric Chaumont 
1731b5816c88SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_AE) {
1732b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1733b5816c88SCedric Chaumont 		goto out;
1734b5816c88SCedric Chaumont 	}
1735b5816c88SCedric Chaumont 
1736b5816c88SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1737b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1738b5816c88SCedric Chaumont 		goto out;
1739b5816c88SCedric Chaumont 	}
1740b0104773SPascal Brand 
1741d9096215SJens Wiklander 	res = ae_update_helper(operation, srcData, srcLen, destData, destLen);
1742d9096215SJens Wiklander 	if (res != TEE_ERROR_SHORT_BUFFER && srcLen)
1743d9096215SJens Wiklander 		operation->operationState = TEE_OPERATION_STATE_ACTIVE;
1744d9096215SJens Wiklander 
1745d9096215SJens Wiklander out:
1746d9096215SJens Wiklander 	if (res != TEE_SUCCESS &&
1747d9096215SJens Wiklander 	    res != TEE_ERROR_SHORT_BUFFER)
1748d9096215SJens Wiklander 		TEE_Panic(res);
1749d9096215SJens Wiklander 
1750d9096215SJens Wiklander 	return res;
1751d9096215SJens Wiklander }
1752d9096215SJens Wiklander 
1753d9096215SJens Wiklander TEE_Result __GP11_TEE_AEUpdate(TEE_OperationHandle operation,
1754d9096215SJens Wiklander 			       const void *srcData, uint32_t srcLen,
1755d9096215SJens Wiklander 			       void *destData, uint32_t *destLen)
1756d9096215SJens Wiklander {
1757d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1758d9096215SJens Wiklander 	size_t dl = 0;
1759d9096215SJens Wiklander 
1760d9096215SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1761d9096215SJens Wiklander 		res = TEE_ERROR_BAD_PARAMETERS;
1762d9096215SJens Wiklander 		goto out;
1763d9096215SJens Wiklander 	}
1764d9096215SJens Wiklander 	__utee_check_gp11_outbuf_annotation(destData, destLen);
1765d9096215SJens Wiklander 
1766d9096215SJens Wiklander 	if (operation->info.operationClass != TEE_OPERATION_AE) {
1767d9096215SJens Wiklander 		res = TEE_ERROR_BAD_PARAMETERS;
1768827308b8SJerome Forissier 		goto out;
1769827308b8SJerome Forissier 	}
1770827308b8SJerome Forissier 
1771d9096215SJens Wiklander 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1772d9096215SJens Wiklander 		res = TEE_ERROR_BAD_PARAMETERS;
1773d9096215SJens Wiklander 		goto out;
1774afc0c182SBogdan Liulko 	}
1775afc0c182SBogdan Liulko 
17766915bbbbSJens Wiklander 	dl = *destLen;
1777d9096215SJens Wiklander 	res = ae_update_helper(operation, srcData, srcLen, destData, &dl);
1778d9096215SJens Wiklander 	*destLen = dl;
1779b0104773SPascal Brand 
1780afc0c182SBogdan Liulko 	if (res != TEE_SUCCESS)
1781afc0c182SBogdan Liulko 		goto out;
1782afc0c182SBogdan Liulko 
1783642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_ACTIVE;
1784642a1607SCedric Chaumont 
1785b5816c88SCedric Chaumont out:
1786b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS &&
1787b5816c88SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
1788b5816c88SCedric Chaumont 			TEE_Panic(res);
1789b5816c88SCedric Chaumont 
1790b5816c88SCedric Chaumont 	return res;
1791b0104773SPascal Brand }
1792b0104773SPascal Brand 
1793b5816c88SCedric Chaumont TEE_Result TEE_AEEncryptFinal(TEE_OperationHandle operation,
1794d9096215SJens Wiklander 			      const void *srcData, size_t srcLen,
1795d9096215SJens Wiklander 			      void *destData, size_t *destLen, void *tag,
1796d9096215SJens Wiklander 			      size_t *tagLen)
1797b0104773SPascal Brand {
1798d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1799b0104773SPascal Brand 	uint8_t *dst = destData;
1800b0104773SPascal Brand 	size_t acc_dlen = 0;
1801d9096215SJens Wiklander 	uint64_t tmp_dlen = 0;
1802d9096215SJens Wiklander 	size_t req_dlen = 0;
1803d9096215SJens Wiklander 	uint64_t tl = 0;
1804b0104773SPascal Brand 
18056915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1806b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1807b5816c88SCedric Chaumont 		goto out;
1808b5816c88SCedric Chaumont 	}
18096915bbbbSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
18106915bbbbSJens Wiklander 	__utee_check_inout_annotation(tagLen, sizeof(*tagLen));
1811b5816c88SCedric Chaumont 
1812b5816c88SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_AE) {
1813b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1814b5816c88SCedric Chaumont 		goto out;
1815b5816c88SCedric Chaumont 	}
1816b5816c88SCedric Chaumont 
1817b5816c88SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1818b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1819b5816c88SCedric Chaumont 		goto out;
1820b5816c88SCedric Chaumont 	}
1821b0104773SPascal Brand 
1822b0104773SPascal Brand 	/*
1823b0104773SPascal Brand 	 * Check that required destLen is big enough before starting to feed
1824b0104773SPascal Brand 	 * data to the algorithm. Errors during feeding of data are fatal as we
1825b0104773SPascal Brand 	 * can't restore sync with this API.
18262733280aSEtienne Carriere 	 *
18272733280aSEtienne Carriere 	 * Need to check this before update_payload since sync would be lost if
18282733280aSEtienne Carriere 	 * we return short buffer after that.
1829b0104773SPascal Brand 	 */
18302733280aSEtienne Carriere 	res = TEE_ERROR_GENERIC;
18312733280aSEtienne Carriere 
1832b5816c88SCedric Chaumont 	req_dlen = operation->buffer_offs + srcLen;
1833b0104773SPascal Brand 	if (*destLen < req_dlen) {
1834b0104773SPascal Brand 		*destLen = req_dlen;
1835b5816c88SCedric Chaumont 		res = TEE_ERROR_SHORT_BUFFER;
1836b0104773SPascal Brand 	}
1837b0104773SPascal Brand 
18387acaf5adSAlbert Schwarzkopf 	if (*tagLen < operation->info.digestLength) {
18397acaf5adSAlbert Schwarzkopf 		*tagLen = operation->info.digestLength;
1840b5816c88SCedric Chaumont 		res = TEE_ERROR_SHORT_BUFFER;
1841b0104773SPascal Brand 	}
1842b0104773SPascal Brand 
18432733280aSEtienne Carriere 	if (res == TEE_ERROR_SHORT_BUFFER)
18442733280aSEtienne Carriere 		goto out;
18452733280aSEtienne Carriere 
1846afc0c182SBogdan Liulko 	tl = *tagLen;
1847b0104773SPascal Brand 	tmp_dlen = *destLen - acc_dlen;
1848afc0c182SBogdan Liulko 	if (operation->block_size > 1) {
18492c028fdeSJerome Forissier 		res = tee_buffer_update(operation, _utee_authenc_update_payload,
1850afc0c182SBogdan Liulko 					srcData, srcLen, dst, &tmp_dlen);
1851b5816c88SCedric Chaumont 		if (res != TEE_SUCCESS)
1852b5816c88SCedric Chaumont 			goto out;
1853b5816c88SCedric Chaumont 
1854b0104773SPascal Brand 		dst += tmp_dlen;
1855b0104773SPascal Brand 		acc_dlen += tmp_dlen;
1856b0104773SPascal Brand 
1857b0104773SPascal Brand 		tmp_dlen = *destLen - acc_dlen;
18582c028fdeSJerome Forissier 		res = _utee_authenc_enc_final(operation->state,
1859afc0c182SBogdan Liulko 					      operation->buffer,
1860afc0c182SBogdan Liulko 					      operation->buffer_offs, dst,
1861afc0c182SBogdan Liulko 					      &tmp_dlen, tag, &tl);
1862afc0c182SBogdan Liulko 	} else {
18632c028fdeSJerome Forissier 		res = _utee_authenc_enc_final(operation->state, srcData,
1864afc0c182SBogdan Liulko 					      srcLen, dst, &tmp_dlen,
1865e86f1266SJens Wiklander 					      tag, &tl);
1866afc0c182SBogdan Liulko 	}
1867e86f1266SJens Wiklander 	*tagLen = tl;
1868b0104773SPascal Brand 	if (res != TEE_SUCCESS)
1869b5816c88SCedric Chaumont 		goto out;
1870b0104773SPascal Brand 
1871b5816c88SCedric Chaumont 	acc_dlen += tmp_dlen;
1872b0104773SPascal Brand 	*destLen = acc_dlen;
1873642a1607SCedric Chaumont 
1874b5816c88SCedric Chaumont 	operation->info.handleState &= ~TEE_HANDLE_FLAG_INITIALIZED;
1875b5816c88SCedric Chaumont 
1876642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
1877642a1607SCedric Chaumont 
1878b5816c88SCedric Chaumont out:
1879b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS &&
1880b5816c88SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER)
1881b5816c88SCedric Chaumont 			TEE_Panic(res);
1882b0104773SPascal Brand 
1883b0104773SPascal Brand 	return res;
1884b0104773SPascal Brand }
1885b0104773SPascal Brand 
1886d9096215SJens Wiklander TEE_Result __GP11_TEE_AEEncryptFinal(TEE_OperationHandle operation,
18878f07fe6fSJerome Forissier 				     const void *srcData, uint32_t srcLen,
1888d9096215SJens Wiklander 				     void *destData, uint32_t *destLen,
1889d9096215SJens Wiklander 				     void *tag, uint32_t *tagLen)
1890b0104773SPascal Brand {
1891d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1892d9096215SJens Wiklander 	size_t dl = 0;
1893d9096215SJens Wiklander 	size_t tl = 0;
1894d9096215SJens Wiklander 
1895d9096215SJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
1896d9096215SJens Wiklander 	__utee_check_inout_annotation(tagLen, sizeof(*tagLen));
1897d9096215SJens Wiklander 	dl = *destLen;
1898d9096215SJens Wiklander 	tl = *tagLen;
1899d9096215SJens Wiklander 	res = TEE_AEEncryptFinal(operation, srcData, srcLen, destData, &dl,
1900d9096215SJens Wiklander 				 tag, &tl);
1901d9096215SJens Wiklander 	*destLen = dl;
1902d9096215SJens Wiklander 	*tagLen = tl;
1903d9096215SJens Wiklander 	return res;
1904d9096215SJens Wiklander }
1905d9096215SJens Wiklander 
1906d9096215SJens Wiklander TEE_Result TEE_AEDecryptFinal(TEE_OperationHandle operation,
1907d9096215SJens Wiklander 			      const void *srcData, size_t srcLen,
1908d9096215SJens Wiklander 			      void *destData, size_t *destLen, void *tag,
1909d9096215SJens Wiklander 			      size_t tagLen)
1910d9096215SJens Wiklander {
1911d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1912b0104773SPascal Brand 	uint8_t *dst = destData;
1913b0104773SPascal Brand 	size_t acc_dlen = 0;
1914d9096215SJens Wiklander 	uint64_t tmp_dlen = 0;
1915d9096215SJens Wiklander 	size_t req_dlen = 0;
1916b0104773SPascal Brand 
19176915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen)) {
1918b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1919b5816c88SCedric Chaumont 		goto out;
1920b5816c88SCedric Chaumont 	}
19216915bbbbSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
1922b5816c88SCedric Chaumont 
1923b5816c88SCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_AE) {
1924b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1925b5816c88SCedric Chaumont 		goto out;
1926b5816c88SCedric Chaumont 	}
1927b5816c88SCedric Chaumont 
1928b5816c88SCedric Chaumont 	if ((operation->info.handleState & TEE_HANDLE_FLAG_INITIALIZED) == 0) {
1929b5816c88SCedric Chaumont 		res = TEE_ERROR_BAD_PARAMETERS;
1930b5816c88SCedric Chaumont 		goto out;
1931b5816c88SCedric Chaumont 	}
1932b0104773SPascal Brand 
1933b0104773SPascal Brand 	/*
1934b0104773SPascal Brand 	 * Check that required destLen is big enough before starting to feed
1935b0104773SPascal Brand 	 * data to the algorithm. Errors during feeding of data are fatal as we
1936b0104773SPascal Brand 	 * can't restore sync with this API.
1937b0104773SPascal Brand 	 */
1938b5816c88SCedric Chaumont 	req_dlen = operation->buffer_offs + srcLen;
1939b0104773SPascal Brand 	if (*destLen < req_dlen) {
1940b0104773SPascal Brand 		*destLen = req_dlen;
1941b5816c88SCedric Chaumont 		res = TEE_ERROR_SHORT_BUFFER;
1942b5816c88SCedric Chaumont 		goto out;
1943b0104773SPascal Brand 	}
1944b0104773SPascal Brand 
1945b0104773SPascal Brand 	tmp_dlen = *destLen - acc_dlen;
1946afc0c182SBogdan Liulko 	if (operation->block_size > 1) {
19472c028fdeSJerome Forissier 		res = tee_buffer_update(operation, _utee_authenc_update_payload,
1948afc0c182SBogdan Liulko 					srcData, srcLen, dst, &tmp_dlen);
1949b5816c88SCedric Chaumont 		if (res != TEE_SUCCESS)
1950b5816c88SCedric Chaumont 			goto out;
1951b5816c88SCedric Chaumont 
1952b0104773SPascal Brand 		dst += tmp_dlen;
1953b0104773SPascal Brand 		acc_dlen += tmp_dlen;
1954b0104773SPascal Brand 
1955b0104773SPascal Brand 		tmp_dlen = *destLen - acc_dlen;
19562c028fdeSJerome Forissier 		res = _utee_authenc_dec_final(operation->state,
1957afc0c182SBogdan Liulko 					      operation->buffer,
1958afc0c182SBogdan Liulko 					      operation->buffer_offs, dst,
1959afc0c182SBogdan Liulko 					      &tmp_dlen, tag, tagLen);
1960afc0c182SBogdan Liulko 	} else {
19612c028fdeSJerome Forissier 		res = _utee_authenc_dec_final(operation->state, srcData,
1962afc0c182SBogdan Liulko 					      srcLen, dst, &tmp_dlen,
1963b5816c88SCedric Chaumont 					      tag, tagLen);
1964afc0c182SBogdan Liulko 	}
1965b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS)
1966b5816c88SCedric Chaumont 		goto out;
1967b5816c88SCedric Chaumont 
1968b0104773SPascal Brand 	/* Supplied tagLen should match what we initiated with */
19697acaf5adSAlbert Schwarzkopf 	if (tagLen != operation->info.digestLength)
1970b0104773SPascal Brand 		res = TEE_ERROR_MAC_INVALID;
1971b0104773SPascal Brand 
1972b0104773SPascal Brand 	acc_dlen += tmp_dlen;
1973b0104773SPascal Brand 	*destLen = acc_dlen;
1974642a1607SCedric Chaumont 
1975b5816c88SCedric Chaumont 	operation->info.handleState &= ~TEE_HANDLE_FLAG_INITIALIZED;
1976b5816c88SCedric Chaumont 
1977642a1607SCedric Chaumont 	operation->operationState = TEE_OPERATION_STATE_INITIAL;
1978642a1607SCedric Chaumont 
1979b5816c88SCedric Chaumont out:
1980b5816c88SCedric Chaumont 	if (res != TEE_SUCCESS &&
1981b5816c88SCedric Chaumont 	    res != TEE_ERROR_SHORT_BUFFER &&
1982b5816c88SCedric Chaumont 	    res != TEE_ERROR_MAC_INVALID)
1983b5816c88SCedric Chaumont 			TEE_Panic(res);
1984b0104773SPascal Brand 
1985b0104773SPascal Brand 	return res;
1986b0104773SPascal Brand }
1987b0104773SPascal Brand 
1988d9096215SJens Wiklander TEE_Result __GP11_TEE_AEDecryptFinal(TEE_OperationHandle operation,
1989d9096215SJens Wiklander 				     const void *srcData, uint32_t srcLen,
1990d9096215SJens Wiklander 				     void *destData, uint32_t *destLen,
1991d9096215SJens Wiklander 				     void *tag, uint32_t tagLen)
1992d9096215SJens Wiklander {
1993d9096215SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
1994d9096215SJens Wiklander 	size_t dl = 0;
1995d9096215SJens Wiklander 
1996d9096215SJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
1997d9096215SJens Wiklander 	dl = *destLen;
1998d9096215SJens Wiklander 	res = TEE_AEDecryptFinal(operation, srcData, srcLen, destData, &dl,
1999d9096215SJens Wiklander 				 tag, tagLen);
2000d9096215SJens Wiklander 	*destLen = dl;
2001d9096215SJens Wiklander 	return res;
2002d9096215SJens Wiklander }
2003d9096215SJens Wiklander 
2004b0104773SPascal Brand /* Cryptographic Operations API - Asymmetric Functions */
2005b0104773SPascal Brand 
200612e66b6fSCedric Chaumont TEE_Result TEE_AsymmetricEncrypt(TEE_OperationHandle operation,
20078f07fe6fSJerome Forissier 				 const TEE_Attribute *params,
20088f07fe6fSJerome Forissier 				 uint32_t paramCount, const void *srcData,
2009999b69d0SJens Wiklander 				 size_t srcLen, void *destData,
2010999b69d0SJens Wiklander 				 size_t *destLen)
2011b0104773SPascal Brand {
20126915bbbbSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
2013e86f1266SJens Wiklander 	struct utee_attribute ua[paramCount];
20146915bbbbSJens Wiklander 	uint64_t dl = 0;
2015b0104773SPascal Brand 
20166915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen))
2017b0104773SPascal Brand 		TEE_Panic(0);
20186915bbbbSJens Wiklander 
20196915bbbbSJens Wiklander 	__utee_check_attr_in_annotation(params, paramCount);
20206915bbbbSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
20216915bbbbSJens Wiklander 
202212e66b6fSCedric Chaumont 	if (!operation->key1)
2023b0104773SPascal Brand 		TEE_Panic(0);
202412e66b6fSCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_ASYMMETRIC_CIPHER)
202512e66b6fSCedric Chaumont 		TEE_Panic(0);
202612e66b6fSCedric Chaumont 	if (operation->info.mode != TEE_MODE_ENCRYPT)
2027b0104773SPascal Brand 		TEE_Panic(0);
2028b0104773SPascal Brand 
2029e86f1266SJens Wiklander 	__utee_from_attr(ua, params, paramCount);
2030e86f1266SJens Wiklander 	dl = *destLen;
20312c028fdeSJerome Forissier 	res = _utee_asymm_operate(operation->state, ua, paramCount, srcData,
2032e86f1266SJens Wiklander 				  srcLen, destData, &dl);
2033e86f1266SJens Wiklander 	*destLen = dl;
203412e66b6fSCedric Chaumont 
20358844ebfcSPascal Brand 	if (res != TEE_SUCCESS &&
20368844ebfcSPascal Brand 	    res != TEE_ERROR_SHORT_BUFFER &&
20371bc176d8SJens Wiklander 	    res != TEE_ERROR_BAD_PARAMETERS &&
20381bc176d8SJens Wiklander 	    res != TEE_ERROR_CIPHERTEXT_INVALID &&
20391bc176d8SJens Wiklander 	    res != TEE_ERROR_NOT_SUPPORTED)
2040b0104773SPascal Brand 		TEE_Panic(res);
204112e66b6fSCedric Chaumont 
2042b0104773SPascal Brand 	return res;
2043b0104773SPascal Brand }
2044b0104773SPascal Brand 
20454f4374c8SJens Wiklander TEE_Result __GP11_TEE_AsymmetricEncrypt(TEE_OperationHandle operation,
20464f4374c8SJens Wiklander 					const __GP11_TEE_Attribute *params,
20474f4374c8SJens Wiklander 					uint32_t paramCount,
20484f4374c8SJens Wiklander 					const void *srcData, uint32_t srcLen,
20494f4374c8SJens Wiklander 					void *destData, uint32_t *destLen)
20504f4374c8SJens Wiklander {
20514f4374c8SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
20524f4374c8SJens Wiklander 	struct utee_attribute ua[paramCount];
20534f4374c8SJens Wiklander 	uint64_t dl = 0;
20544f4374c8SJens Wiklander 
20554f4374c8SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen))
20564f4374c8SJens Wiklander 		TEE_Panic(0);
20574f4374c8SJens Wiklander 
20584f4374c8SJens Wiklander 	__utee_check_gp11_attr_in_annotation(params, paramCount);
20594f4374c8SJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
20604f4374c8SJens Wiklander 
20614f4374c8SJens Wiklander 	if (!operation->key1)
20624f4374c8SJens Wiklander 		TEE_Panic(0);
20634f4374c8SJens Wiklander 	if (operation->info.operationClass != TEE_OPERATION_ASYMMETRIC_CIPHER)
20644f4374c8SJens Wiklander 		TEE_Panic(0);
20654f4374c8SJens Wiklander 	if (operation->info.mode != TEE_MODE_ENCRYPT)
20664f4374c8SJens Wiklander 		TEE_Panic(0);
20674f4374c8SJens Wiklander 
20684f4374c8SJens Wiklander 	__utee_from_gp11_attr(ua, params, paramCount);
20694f4374c8SJens Wiklander 	dl = *destLen;
20704f4374c8SJens Wiklander 	res = _utee_asymm_operate(operation->state, ua, paramCount, srcData,
20714f4374c8SJens Wiklander 				  srcLen, destData, &dl);
20724f4374c8SJens Wiklander 	*destLen = dl;
20734f4374c8SJens Wiklander 
20744f4374c8SJens Wiklander 	if (res != TEE_SUCCESS &&
20754f4374c8SJens Wiklander 	    res != TEE_ERROR_SHORT_BUFFER &&
20764f4374c8SJens Wiklander 	    res != TEE_ERROR_BAD_PARAMETERS)
20774f4374c8SJens Wiklander 		TEE_Panic(res);
20784f4374c8SJens Wiklander 
20794f4374c8SJens Wiklander 	return res;
20804f4374c8SJens Wiklander }
20814f4374c8SJens Wiklander 
208212e66b6fSCedric Chaumont TEE_Result TEE_AsymmetricDecrypt(TEE_OperationHandle operation,
20838f07fe6fSJerome Forissier 				 const TEE_Attribute *params,
20848f07fe6fSJerome Forissier 				 uint32_t paramCount, const void *srcData,
2085999b69d0SJens Wiklander 				 size_t srcLen, void *destData,
2086999b69d0SJens Wiklander 				 size_t *destLen)
2087b0104773SPascal Brand {
20886915bbbbSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
2089e86f1266SJens Wiklander 	struct utee_attribute ua[paramCount];
20906915bbbbSJens Wiklander 	uint64_t dl = 0;
2091b0104773SPascal Brand 
20926915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen))
2093b0104773SPascal Brand 		TEE_Panic(0);
20946915bbbbSJens Wiklander 
20956915bbbbSJens Wiklander 	__utee_check_attr_in_annotation(params, paramCount);
20966915bbbbSJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
20976915bbbbSJens Wiklander 
209812e66b6fSCedric Chaumont 	if (!operation->key1)
2099b0104773SPascal Brand 		TEE_Panic(0);
210012e66b6fSCedric Chaumont 	if (operation->info.operationClass != TEE_OPERATION_ASYMMETRIC_CIPHER)
210112e66b6fSCedric Chaumont 		TEE_Panic(0);
210212e66b6fSCedric Chaumont 	if (operation->info.mode != TEE_MODE_DECRYPT)
2103b0104773SPascal Brand 		TEE_Panic(0);
2104b0104773SPascal Brand 
2105e86f1266SJens Wiklander 	__utee_from_attr(ua, params, paramCount);
2106e86f1266SJens Wiklander 	dl = *destLen;
21072c028fdeSJerome Forissier 	res = _utee_asymm_operate(operation->state, ua, paramCount, srcData,
2108e86f1266SJens Wiklander 				  srcLen, destData, &dl);
2109e86f1266SJens Wiklander 	*destLen = dl;
211012e66b6fSCedric Chaumont 
21118844ebfcSPascal Brand 	if (res != TEE_SUCCESS &&
21128844ebfcSPascal Brand 	    res != TEE_ERROR_SHORT_BUFFER &&
21131bc176d8SJens Wiklander 	    res != TEE_ERROR_BAD_PARAMETERS &&
21141bc176d8SJens Wiklander 	    res != TEE_ERROR_CIPHERTEXT_INVALID &&
21151bc176d8SJens Wiklander 	    res != TEE_ERROR_NOT_SUPPORTED)
2116b0104773SPascal Brand 		TEE_Panic(res);
211712e66b6fSCedric Chaumont 
2118b0104773SPascal Brand 	return res;
2119b0104773SPascal Brand }
2120b0104773SPascal Brand 
21214f4374c8SJens Wiklander TEE_Result __GP11_TEE_AsymmetricDecrypt(TEE_OperationHandle operation,
21224f4374c8SJens Wiklander 					const __GP11_TEE_Attribute *params,
21234f4374c8SJens Wiklander 					uint32_t paramCount,
21244f4374c8SJens Wiklander 					const void *srcData, uint32_t srcLen,
21254f4374c8SJens Wiklander 					void *destData, uint32_t *destLen)
21264f4374c8SJens Wiklander {
21274f4374c8SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
21284f4374c8SJens Wiklander 	struct utee_attribute ua[paramCount];
21294f4374c8SJens Wiklander 	uint64_t dl = 0;
21304f4374c8SJens Wiklander 
21314f4374c8SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!srcData && srcLen))
21324f4374c8SJens Wiklander 		TEE_Panic(0);
21334f4374c8SJens Wiklander 
21344f4374c8SJens Wiklander 	__utee_check_gp11_attr_in_annotation(params, paramCount);
21354f4374c8SJens Wiklander 	__utee_check_inout_annotation(destLen, sizeof(*destLen));
21364f4374c8SJens Wiklander 
21374f4374c8SJens Wiklander 	if (!operation->key1)
21384f4374c8SJens Wiklander 		TEE_Panic(0);
21394f4374c8SJens Wiklander 	if (operation->info.operationClass != TEE_OPERATION_ASYMMETRIC_CIPHER)
21404f4374c8SJens Wiklander 		TEE_Panic(0);
21414f4374c8SJens Wiklander 	if (operation->info.mode != TEE_MODE_DECRYPT)
21424f4374c8SJens Wiklander 		TEE_Panic(0);
21434f4374c8SJens Wiklander 
21444f4374c8SJens Wiklander 	__utee_from_gp11_attr(ua, params, paramCount);
21454f4374c8SJens Wiklander 	dl = *destLen;
21464f4374c8SJens Wiklander 	res = _utee_asymm_operate(operation->state, ua, paramCount, srcData,
21474f4374c8SJens Wiklander 				  srcLen, destData, &dl);
21484f4374c8SJens Wiklander 	*destLen = dl;
21494f4374c8SJens Wiklander 
21504f4374c8SJens Wiklander 	if (res != TEE_SUCCESS &&
21514f4374c8SJens Wiklander 	    res != TEE_ERROR_SHORT_BUFFER &&
21524f4374c8SJens Wiklander 	    res != TEE_ERROR_BAD_PARAMETERS)
21534f4374c8SJens Wiklander 		TEE_Panic(res);
21544f4374c8SJens Wiklander 
21554f4374c8SJens Wiklander 	return res;
21564f4374c8SJens Wiklander }
21574f4374c8SJens Wiklander 
215812e66b6fSCedric Chaumont TEE_Result TEE_AsymmetricSignDigest(TEE_OperationHandle operation,
21598f07fe6fSJerome Forissier 				    const TEE_Attribute *params,
21608f07fe6fSJerome Forissier 				    uint32_t paramCount, const void *digest,
21610b354ec8SJens Wiklander 				    size_t digestLen, void *signature,
21620b354ec8SJens Wiklander 				    size_t *signatureLen)
2163b0104773SPascal Brand {
21646915bbbbSJens Wiklander 	TEE_Result res = TEE_SUCCESS;
2165e86f1266SJens Wiklander 	struct utee_attribute ua[paramCount];
21666915bbbbSJens Wiklander 	uint64_t sl = 0;
2167b0104773SPascal Brand 
21686915bbbbSJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!digest && digestLen))
2169b0104773SPascal Brand 		TEE_Panic(0);
21706915bbbbSJens Wiklander 
21716915bbbbSJens Wiklander 	__utee_check_attr_in_annotation(params, paramCount);
21726915bbbbSJens Wiklander 	__utee_check_inout_annotation(signatureLen, sizeof(*signatureLen));
21736915bbbbSJens Wiklander 
217412e66b6fSCedric Chaumont 	if (!operation->key1)
2175b0104773SPascal Brand 		TEE_Panic(0);
217612e66b6fSCedric Chaumont 	if (operation->info.operationClass !=
217712e66b6fSCedric Chaumont 	    TEE_OPERATION_ASYMMETRIC_SIGNATURE)
217812e66b6fSCedric Chaumont 		TEE_Panic(0);
217912e66b6fSCedric Chaumont 	if (operation->info.mode != TEE_MODE_SIGN)
2180b0104773SPascal Brand 		TEE_Panic(0);
2181b0104773SPascal Brand 
2182e86f1266SJens Wiklander 	__utee_from_attr(ua, params, paramCount);
2183e86f1266SJens Wiklander 	sl = *signatureLen;
21842c028fdeSJerome Forissier 	res = _utee_asymm_operate(operation->state, ua, paramCount, digest,
2185e86f1266SJens Wiklander 				  digestLen, signature, &sl);
2186e86f1266SJens Wiklander 	*signatureLen = sl;
218712e66b6fSCedric Chaumont 
2188b0104773SPascal Brand 	if (res != TEE_SUCCESS && res != TEE_ERROR_SHORT_BUFFER)
2189b0104773SPascal Brand 		TEE_Panic(res);
219012e66b6fSCedric Chaumont 
2191b0104773SPascal Brand 	return res;
2192b0104773SPascal Brand }
2193b0104773SPascal Brand 
21944f4374c8SJens Wiklander TEE_Result __GP11_TEE_AsymmetricSignDigest(TEE_OperationHandle operation,
21954f4374c8SJens Wiklander 					   const __GP11_TEE_Attribute *params,
21964f4374c8SJens Wiklander 					   uint32_t paramCount,
21974f4374c8SJens Wiklander 					   const void *digest,
21984f4374c8SJens Wiklander 					   uint32_t digestLen, void *signature,
21994f4374c8SJens Wiklander 					   uint32_t *signatureLen)
22004f4374c8SJens Wiklander {
22014f4374c8SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
22024f4374c8SJens Wiklander 	struct utee_attribute ua[paramCount];
22034f4374c8SJens Wiklander 	uint64_t sl = 0;
22044f4374c8SJens Wiklander 
22054f4374c8SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!digest && digestLen))
22064f4374c8SJens Wiklander 		TEE_Panic(0);
22074f4374c8SJens Wiklander 
22084f4374c8SJens Wiklander 	__utee_check_gp11_attr_in_annotation(params, paramCount);
22094f4374c8SJens Wiklander 	__utee_check_inout_annotation(signatureLen, sizeof(*signatureLen));
22104f4374c8SJens Wiklander 
22114f4374c8SJens Wiklander 	if (!operation->key1)
22124f4374c8SJens Wiklander 		TEE_Panic(0);
22134f4374c8SJens Wiklander 	if (operation->info.operationClass !=
22144f4374c8SJens Wiklander 	    TEE_OPERATION_ASYMMETRIC_SIGNATURE)
22154f4374c8SJens Wiklander 		TEE_Panic(0);
22164f4374c8SJens Wiklander 	if (operation->info.mode != TEE_MODE_SIGN)
22174f4374c8SJens Wiklander 		TEE_Panic(0);
22184f4374c8SJens Wiklander 
22194f4374c8SJens Wiklander 	__utee_from_gp11_attr(ua, params, paramCount);
22204f4374c8SJens Wiklander 	sl = *signatureLen;
22214f4374c8SJens Wiklander 	res = _utee_asymm_operate(operation->state, ua, paramCount, digest,
22224f4374c8SJens Wiklander 				  digestLen, signature, &sl);
22234f4374c8SJens Wiklander 	*signatureLen = sl;
22244f4374c8SJens Wiklander 
22254f4374c8SJens Wiklander 	if (res != TEE_SUCCESS && res != TEE_ERROR_SHORT_BUFFER)
22264f4374c8SJens Wiklander 		TEE_Panic(res);
22274f4374c8SJens Wiklander 
22284f4374c8SJens Wiklander 	return res;
22294f4374c8SJens Wiklander }
22304f4374c8SJens Wiklander 
223112e66b6fSCedric Chaumont TEE_Result TEE_AsymmetricVerifyDigest(TEE_OperationHandle operation,
22328f07fe6fSJerome Forissier 				      const TEE_Attribute *params,
22338f07fe6fSJerome Forissier 				      uint32_t paramCount, const void *digest,
22340b354ec8SJens Wiklander 				      size_t digestLen,
22358f07fe6fSJerome Forissier 				      const void *signature,
22360b354ec8SJens Wiklander 				      size_t signatureLen)
2237b0104773SPascal Brand {
2238b0104773SPascal Brand 	TEE_Result res;
2239e86f1266SJens Wiklander 	struct utee_attribute ua[paramCount];
2240b0104773SPascal Brand 
224112e66b6fSCedric Chaumont 	if (operation == TEE_HANDLE_NULL ||
224212e66b6fSCedric Chaumont 	    (digest == NULL && digestLen != 0) ||
2243b0104773SPascal Brand 	    (signature == NULL && signatureLen != 0))
2244b0104773SPascal Brand 		TEE_Panic(0);
22456915bbbbSJens Wiklander 
22466915bbbbSJens Wiklander 	__utee_check_attr_in_annotation(params, paramCount);
22476915bbbbSJens Wiklander 
224812e66b6fSCedric Chaumont 	if (!operation->key1)
2249b0104773SPascal Brand 		TEE_Panic(0);
225012e66b6fSCedric Chaumont 	if (operation->info.operationClass !=
225112e66b6fSCedric Chaumont 	    TEE_OPERATION_ASYMMETRIC_SIGNATURE)
225212e66b6fSCedric Chaumont 		TEE_Panic(0);
225312e66b6fSCedric Chaumont 	if (operation->info.mode != TEE_MODE_VERIFY)
2254b0104773SPascal Brand 		TEE_Panic(0);
2255b0104773SPascal Brand 
2256e86f1266SJens Wiklander 	__utee_from_attr(ua, params, paramCount);
22572c028fdeSJerome Forissier 	res = _utee_asymm_verify(operation->state, ua, paramCount, digest,
225812e66b6fSCedric Chaumont 				 digestLen, signature, signatureLen);
225912e66b6fSCedric Chaumont 
2260b0104773SPascal Brand 	if (res != TEE_SUCCESS && res != TEE_ERROR_SIGNATURE_INVALID)
2261b0104773SPascal Brand 		TEE_Panic(res);
226212e66b6fSCedric Chaumont 
2263b0104773SPascal Brand 	return res;
2264b0104773SPascal Brand }
2265b0104773SPascal Brand 
22664f4374c8SJens Wiklander TEE_Result __GP11_TEE_AsymmetricVerifyDigest(TEE_OperationHandle operation,
22674f4374c8SJens Wiklander 					     const __GP11_TEE_Attribute *params,
22684f4374c8SJens Wiklander 					     uint32_t paramCount,
22694f4374c8SJens Wiklander 					     const void *digest,
22704f4374c8SJens Wiklander 					     uint32_t digestLen,
22714f4374c8SJens Wiklander 					     const void *signature,
22724f4374c8SJens Wiklander 					     uint32_t signatureLen)
22734f4374c8SJens Wiklander {
22744f4374c8SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
22754f4374c8SJens Wiklander 	struct utee_attribute ua[paramCount];
22764f4374c8SJens Wiklander 
22774f4374c8SJens Wiklander 	if (operation == TEE_HANDLE_NULL || (!digest && digestLen) ||
22784f4374c8SJens Wiklander 	    (!signature && signatureLen))
22794f4374c8SJens Wiklander 		TEE_Panic(0);
22804f4374c8SJens Wiklander 
22814f4374c8SJens Wiklander 	__utee_check_gp11_attr_in_annotation(params, paramCount);
22824f4374c8SJens Wiklander 
22834f4374c8SJens Wiklander 	if (!operation->key1)
22844f4374c8SJens Wiklander 		TEE_Panic(0);
22854f4374c8SJens Wiklander 	if (operation->info.operationClass !=
22864f4374c8SJens Wiklander 	    TEE_OPERATION_ASYMMETRIC_SIGNATURE)
22874f4374c8SJens Wiklander 		TEE_Panic(0);
22884f4374c8SJens Wiklander 	if (operation->info.mode != TEE_MODE_VERIFY)
22894f4374c8SJens Wiklander 		TEE_Panic(0);
22904f4374c8SJens Wiklander 
22914f4374c8SJens Wiklander 	__utee_from_gp11_attr(ua, params, paramCount);
22924f4374c8SJens Wiklander 	res = _utee_asymm_verify(operation->state, ua, paramCount, digest,
22934f4374c8SJens Wiklander 				 digestLen, signature, signatureLen);
22944f4374c8SJens Wiklander 
22954f4374c8SJens Wiklander 	if (res != TEE_SUCCESS && res != TEE_ERROR_SIGNATURE_INVALID)
22964f4374c8SJens Wiklander 		TEE_Panic(res);
22974f4374c8SJens Wiklander 
22984f4374c8SJens Wiklander 	return res;
22994f4374c8SJens Wiklander }
23004f4374c8SJens Wiklander 
2301b0104773SPascal Brand /* Cryptographic Operations API - Key Derivation Functions */
2302b0104773SPascal Brand 
2303b0104773SPascal Brand void TEE_DeriveKey(TEE_OperationHandle operation,
2304b0104773SPascal Brand 		   const TEE_Attribute *params, uint32_t paramCount,
2305b0104773SPascal Brand 		   TEE_ObjectHandle derivedKey)
2306b0104773SPascal Brand {
2307e86f1266SJens Wiklander 	struct utee_attribute ua[paramCount];
230875d6a373SJens Wiklander 	struct utee_object_info key_info = { };
230975d6a373SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
2310b0104773SPascal Brand 
2311b0104773SPascal Brand 	if (operation == TEE_HANDLE_NULL || derivedKey == 0)
2312b0104773SPascal Brand 		TEE_Panic(0);
23136915bbbbSJens Wiklander 
23146915bbbbSJens Wiklander 	__utee_check_attr_in_annotation(params, paramCount);
23156915bbbbSJens Wiklander 
23168854d3c6SJerome Forissier 	if (TEE_ALG_GET_CLASS(operation->info.algorithm) !=
23178854d3c6SJerome Forissier 	    TEE_OPERATION_KEY_DERIVATION)
2318b0104773SPascal Brand 		TEE_Panic(0);
2319b0104773SPascal Brand 
2320b0104773SPascal Brand 	if (operation->info.operationClass != TEE_OPERATION_KEY_DERIVATION)
2321b0104773SPascal Brand 		TEE_Panic(0);
232284fa9467SCedric Chaumont 	if (!operation->key1)
232384fa9467SCedric Chaumont 		TEE_Panic(0);
2324b0104773SPascal Brand 	if (operation->info.mode != TEE_MODE_DERIVE)
2325b0104773SPascal Brand 		TEE_Panic(0);
2326b0104773SPascal Brand 	if ((operation->info.handleState & TEE_HANDLE_FLAG_KEY_SET) == 0)
2327b0104773SPascal Brand 		TEE_Panic(0);
2328b0104773SPascal Brand 
23292c028fdeSJerome Forissier 	res = _utee_cryp_obj_get_info((unsigned long)derivedKey, &key_info);
2330b0104773SPascal Brand 	if (res != TEE_SUCCESS)
2331b36311adSJerome Forissier 		TEE_Panic(res);
2332b0104773SPascal Brand 
233375d6a373SJens Wiklander 	if (key_info.obj_type != TEE_TYPE_GENERIC_SECRET)
2334b0104773SPascal Brand 		TEE_Panic(0);
233575d6a373SJens Wiklander 	if ((key_info.handle_flags & TEE_HANDLE_FLAG_INITIALIZED) != 0)
2336b0104773SPascal Brand 		TEE_Panic(0);
2337b0104773SPascal Brand 
2338e86f1266SJens Wiklander 	__utee_from_attr(ua, params, paramCount);
23392c028fdeSJerome Forissier 	res = _utee_cryp_derive_key(operation->state, ua, paramCount,
2340e86f1266SJens Wiklander 				    (unsigned long)derivedKey);
2341b0104773SPascal Brand 	if (res != TEE_SUCCESS)
2342b0104773SPascal Brand 		TEE_Panic(res);
2343b0104773SPascal Brand }
2344b0104773SPascal Brand 
23454f4374c8SJens Wiklander void __GP11_TEE_DeriveKey(TEE_OperationHandle operation,
23464f4374c8SJens Wiklander 			  const __GP11_TEE_Attribute *params,
23474f4374c8SJens Wiklander 			  uint32_t paramCount, TEE_ObjectHandle derivedKey)
23484f4374c8SJens Wiklander {
23494f4374c8SJens Wiklander 	struct utee_attribute ua[paramCount];
23504f4374c8SJens Wiklander 	struct utee_object_info key_info = { };
23514f4374c8SJens Wiklander 	TEE_Result res = TEE_SUCCESS;
23524f4374c8SJens Wiklander 
23534f4374c8SJens Wiklander 	if (operation == TEE_HANDLE_NULL || derivedKey == 0)
23544f4374c8SJens Wiklander 		TEE_Panic(0);
23554f4374c8SJens Wiklander 
23564f4374c8SJens Wiklander 	__utee_check_gp11_attr_in_annotation(params, paramCount);
23574f4374c8SJens Wiklander 
23584f4374c8SJens Wiklander 	if (TEE_ALG_GET_CLASS(operation->info.algorithm) !=
23594f4374c8SJens Wiklander 	    TEE_OPERATION_KEY_DERIVATION)
23604f4374c8SJens Wiklander 		TEE_Panic(0);
23614f4374c8SJens Wiklander 
23624f4374c8SJens Wiklander 	if (operation->info.operationClass != TEE_OPERATION_KEY_DERIVATION)
23634f4374c8SJens Wiklander 		TEE_Panic(0);
23644f4374c8SJens Wiklander 	if (!operation->key1)
23654f4374c8SJens Wiklander 		TEE_Panic(0);
23664f4374c8SJens Wiklander 	if (operation->info.mode != TEE_MODE_DERIVE)
23674f4374c8SJens Wiklander 		TEE_Panic(0);
23684f4374c8SJens Wiklander 	if ((operation->info.handleState & TEE_HANDLE_FLAG_KEY_SET) == 0)
23694f4374c8SJens Wiklander 		TEE_Panic(0);
23704f4374c8SJens Wiklander 
23714f4374c8SJens Wiklander 	res = _utee_cryp_obj_get_info((unsigned long)derivedKey, &key_info);
23724f4374c8SJens Wiklander 	if (res != TEE_SUCCESS)
23734f4374c8SJens Wiklander 		TEE_Panic(res);
23744f4374c8SJens Wiklander 
23754f4374c8SJens Wiklander 	if (key_info.obj_type != TEE_TYPE_GENERIC_SECRET)
23764f4374c8SJens Wiklander 		TEE_Panic(0);
23774f4374c8SJens Wiklander 	if ((key_info.handle_flags & TEE_HANDLE_FLAG_INITIALIZED) != 0)
23784f4374c8SJens Wiklander 		TEE_Panic(0);
23794f4374c8SJens Wiklander 
23804f4374c8SJens Wiklander 	__utee_from_gp11_attr(ua, params, paramCount);
23814f4374c8SJens Wiklander 	res = _utee_cryp_derive_key(operation->state, ua, paramCount,
23824f4374c8SJens Wiklander 				    (unsigned long)derivedKey);
23834f4374c8SJens Wiklander 	if (res != TEE_SUCCESS)
23844f4374c8SJens Wiklander 		TEE_Panic(res);
23854f4374c8SJens Wiklander }
23864f4374c8SJens Wiklander 
2387b0104773SPascal Brand /* Cryptographic Operations API - Random Number Generation Functions */
2388b0104773SPascal Brand 
2389411a488aSJens Wiklander void TEE_GenerateRandom(void *randomBuffer, size_t randomBufferLen)
2390b0104773SPascal Brand {
2391b0104773SPascal Brand 	TEE_Result res;
2392b0104773SPascal Brand 
23932c028fdeSJerome Forissier 	res = _utee_cryp_random_number_generate(randomBuffer, randomBufferLen);
2394b0104773SPascal Brand 	if (res != TEE_SUCCESS)
2395b0104773SPascal Brand 		TEE_Panic(res);
2396b0104773SPascal Brand }
2397433c4257SJens Wiklander 
2398411a488aSJens Wiklander void __GP11_TEE_GenerateRandom(void *randomBuffer, uint32_t randomBufferLen)
2399411a488aSJens Wiklander {
2400411a488aSJens Wiklander 	TEE_GenerateRandom(randomBuffer, randomBufferLen);
2401411a488aSJens Wiklander }
2402411a488aSJens Wiklander 
2403433c4257SJens Wiklander int rand(void)
2404433c4257SJens Wiklander {
2405433c4257SJens Wiklander 	int rc;
2406433c4257SJens Wiklander 
2407433c4257SJens Wiklander 	TEE_GenerateRandom(&rc, sizeof(rc));
2408433c4257SJens Wiklander 
2409433c4257SJens Wiklander 	/*
2410433c4257SJens Wiklander 	 * RAND_MAX is the larges int, INT_MAX which is all bits but the
2411433c4257SJens Wiklander 	 * highest bit set.
2412433c4257SJens Wiklander 	 */
2413433c4257SJens Wiklander 	return rc & RAND_MAX;
2414433c4257SJens Wiklander }
241579170ce0SJerome Forissier 
241679170ce0SJerome Forissier TEE_Result TEE_IsAlgorithmSupported(uint32_t alg, uint32_t element)
241779170ce0SJerome Forissier {
241879170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_AES)) {
241979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_ECB)) {
242079170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_ECB_NOPAD)
242179170ce0SJerome Forissier 				goto check_element_none;
242279170ce0SJerome Forissier 		}
242379170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CBC)) {
242479170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CBC_NOPAD)
242579170ce0SJerome Forissier 				goto check_element_none;
242679170ce0SJerome Forissier 		}
242779170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CTR)) {
242879170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CTR)
242979170ce0SJerome Forissier 				goto check_element_none;
243079170ce0SJerome Forissier 		}
243179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CTS)) {
243279170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CTS)
243379170ce0SJerome Forissier 				goto check_element_none;
243479170ce0SJerome Forissier 		}
243579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_XTS)) {
243679170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_XTS)
243779170ce0SJerome Forissier 				goto check_element_none;
243879170ce0SJerome Forissier 		}
243979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CBC_MAC)) {
244079170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CBC_MAC_NOPAD ||
244179170ce0SJerome Forissier 			    alg == TEE_ALG_AES_CBC_MAC_PKCS5)
244279170ce0SJerome Forissier 				goto check_element_none;
244379170ce0SJerome Forissier 		}
244479170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CMAC)) {
244579170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CMAC)
244679170ce0SJerome Forissier 				goto check_element_none;
244779170ce0SJerome Forissier 		}
244879170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CCM)) {
244979170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_CCM)
245079170ce0SJerome Forissier 				goto check_element_none;
245179170ce0SJerome Forissier 		}
245279170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_GCM)) {
245379170ce0SJerome Forissier 			if (alg == TEE_ALG_AES_GCM)
245479170ce0SJerome Forissier 				goto check_element_none;
245579170ce0SJerome Forissier 		}
245679170ce0SJerome Forissier 	}
245779170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_DES)) {
245879170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_ECB)) {
245979170ce0SJerome Forissier 			if (alg == TEE_ALG_DES_ECB_NOPAD ||
246079170ce0SJerome Forissier 			    alg == TEE_ALG_DES3_ECB_NOPAD)
246179170ce0SJerome Forissier 				goto check_element_none;
246279170ce0SJerome Forissier 		}
246379170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CBC)) {
246479170ce0SJerome Forissier 			if (alg == TEE_ALG_DES_CBC_NOPAD ||
246579170ce0SJerome Forissier 			    alg == TEE_ALG_DES3_CBC_NOPAD)
246679170ce0SJerome Forissier 				goto check_element_none;
246779170ce0SJerome Forissier 		}
246879170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CBC_MAC)) {
246979170ce0SJerome Forissier 			if (alg == TEE_ALG_DES_CBC_MAC_NOPAD ||
247079170ce0SJerome Forissier 			    alg == TEE_ALG_DES_CBC_MAC_PKCS5 ||
247179170ce0SJerome Forissier 			    alg == TEE_ALG_DES3_CBC_MAC_NOPAD ||
247279170ce0SJerome Forissier 			    alg == TEE_ALG_DES3_CBC_MAC_PKCS5)
247379170ce0SJerome Forissier 				goto check_element_none;
247479170ce0SJerome Forissier 		}
247579170ce0SJerome Forissier 	}
247679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_MD5)) {
247779170ce0SJerome Forissier 		if (alg == TEE_ALG_MD5)
247879170ce0SJerome Forissier 			goto check_element_none;
247979170ce0SJerome Forissier 	}
248079170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SHA1)) {
248179170ce0SJerome Forissier 		if (alg == TEE_ALG_SHA1)
248279170ce0SJerome Forissier 			goto check_element_none;
248379170ce0SJerome Forissier 	}
248479170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SHA224)) {
248579170ce0SJerome Forissier 		if (alg == TEE_ALG_SHA224)
248679170ce0SJerome Forissier 			goto check_element_none;
248779170ce0SJerome Forissier 	}
248879170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SHA256)) {
248979170ce0SJerome Forissier 		if (alg == TEE_ALG_SHA256)
249079170ce0SJerome Forissier 			goto check_element_none;
249179170ce0SJerome Forissier 	}
249279170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SHA384)) {
249379170ce0SJerome Forissier 		if (alg == TEE_ALG_SHA384)
249479170ce0SJerome Forissier 			goto check_element_none;
249579170ce0SJerome Forissier 	}
249679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SHA512)) {
249779170ce0SJerome Forissier 		if (alg == TEE_ALG_SHA512)
249879170ce0SJerome Forissier 			goto check_element_none;
249979170ce0SJerome Forissier 	}
2500260b4028SJens Wiklander 	if (IS_ENABLED(CFG_CRYPTO_SHA3_224)) {
2501260b4028SJens Wiklander 		if (alg == TEE_ALG_SHA3_224)
2502260b4028SJens Wiklander 			goto check_element_none;
2503260b4028SJens Wiklander 	}
2504260b4028SJens Wiklander 	if (IS_ENABLED(CFG_CRYPTO_SHA3_256)) {
2505260b4028SJens Wiklander 		if (alg == TEE_ALG_SHA3_256)
2506260b4028SJens Wiklander 			goto check_element_none;
2507260b4028SJens Wiklander 	}
2508260b4028SJens Wiklander 	if (IS_ENABLED(CFG_CRYPTO_SHA3_384)) {
2509260b4028SJens Wiklander 		if (alg == TEE_ALG_SHA3_384)
2510260b4028SJens Wiklander 			goto check_element_none;
2511260b4028SJens Wiklander 	}
2512260b4028SJens Wiklander 	if (IS_ENABLED(CFG_CRYPTO_SHA3_512)) {
2513260b4028SJens Wiklander 		if (alg == TEE_ALG_SHA3_512)
2514260b4028SJens Wiklander 			goto check_element_none;
2515260b4028SJens Wiklander 	}
251679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_MD5) && IS_ENABLED(CFG_CRYPTO_SHA1)) {
251779170ce0SJerome Forissier 		if (alg == TEE_ALG_MD5SHA1)
251879170ce0SJerome Forissier 			goto check_element_none;
251979170ce0SJerome Forissier 	}
252079170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_HMAC)) {
252179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_MD5)) {
252279170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_MD5)
252379170ce0SJerome Forissier 				goto check_element_none;
252479170ce0SJerome Forissier 		}
252579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA1)) {
252679170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SHA1)
252779170ce0SJerome Forissier 				goto check_element_none;
252879170ce0SJerome Forissier 		}
252979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA224)) {
253079170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SHA224)
253179170ce0SJerome Forissier 				goto check_element_none;
253279170ce0SJerome Forissier 		}
253379170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA256)) {
253479170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SHA256)
253579170ce0SJerome Forissier 				goto check_element_none;
253679170ce0SJerome Forissier 		}
253779170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA384)) {
253879170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SHA384)
253979170ce0SJerome Forissier 				goto check_element_none;
254079170ce0SJerome Forissier 		}
254179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA512)) {
254279170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SHA512)
254379170ce0SJerome Forissier 				goto check_element_none;
254479170ce0SJerome Forissier 		}
2545260b4028SJens Wiklander 		if (IS_ENABLED(CFG_CRYPTO_SHA3_224)) {
2546260b4028SJens Wiklander 			if (alg == TEE_ALG_HMAC_SHA3_224)
2547260b4028SJens Wiklander 				goto check_element_none;
2548260b4028SJens Wiklander 		}
2549260b4028SJens Wiklander 		if (IS_ENABLED(CFG_CRYPTO_SHA3_256)) {
2550260b4028SJens Wiklander 			if (alg == TEE_ALG_HMAC_SHA3_256)
2551260b4028SJens Wiklander 				goto check_element_none;
2552260b4028SJens Wiklander 		}
2553260b4028SJens Wiklander 		if (IS_ENABLED(CFG_CRYPTO_SHA3_384)) {
2554260b4028SJens Wiklander 			if (alg == TEE_ALG_HMAC_SHA3_384)
2555260b4028SJens Wiklander 				goto check_element_none;
2556260b4028SJens Wiklander 		}
2557260b4028SJens Wiklander 		if (IS_ENABLED(CFG_CRYPTO_SHA3_512)) {
2558260b4028SJens Wiklander 			if (alg == TEE_ALG_HMAC_SHA3_512)
2559260b4028SJens Wiklander 				goto check_element_none;
2560260b4028SJens Wiklander 		}
256179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SM3)) {
256279170ce0SJerome Forissier 			if (alg == TEE_ALG_HMAC_SM3)
256379170ce0SJerome Forissier 				goto check_element_none;
256479170ce0SJerome Forissier 		}
256579170ce0SJerome Forissier 	}
256679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SM3)) {
256779170ce0SJerome Forissier 		if (alg == TEE_ALG_SM3)
256879170ce0SJerome Forissier 			goto check_element_none;
256979170ce0SJerome Forissier 	}
257079170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SM4)) {
257179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_ECB)) {
257279170ce0SJerome Forissier 			if (alg == TEE_ALG_SM4_ECB_NOPAD)
257379170ce0SJerome Forissier 				goto check_element_none;
257479170ce0SJerome Forissier 		}
257579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CBC)) {
257679170ce0SJerome Forissier 			if (alg == TEE_ALG_SM4_CBC_NOPAD)
257779170ce0SJerome Forissier 				goto check_element_none;
257879170ce0SJerome Forissier 		}
257979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_CTR)) {
258079170ce0SJerome Forissier 			if (alg == TEE_ALG_SM4_CTR)
258179170ce0SJerome Forissier 				goto check_element_none;
258279170ce0SJerome Forissier 		}
2583696f56acSPingan Xie 		if (IS_ENABLED(CFG_CRYPTO_XTS)) {
2584696f56acSPingan Xie 			if (alg == TEE_ALG_SM4_XTS)
2585696f56acSPingan Xie 				goto check_element_none;
2586696f56acSPingan Xie 		}
258779170ce0SJerome Forissier 	}
258879170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_RSA)) {
258979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_MD5)) {
2590*f5c3d85aSJulien Masson 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_MD5 ||
2591*f5c3d85aSJulien Masson 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_MD5 ||
2592*f5c3d85aSJulien Masson 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_MD5)
259379170ce0SJerome Forissier 				goto check_element_none;
259479170ce0SJerome Forissier 		}
259579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA1)) {
259679170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_SHA1 ||
259779170ce0SJerome Forissier 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA1 ||
259879170ce0SJerome Forissier 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA1)
259979170ce0SJerome Forissier 				goto check_element_none;
260079170ce0SJerome Forissier 		}
260179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_MD5) && IS_ENABLED(CFG_CRYPTO_SHA1)) {
260279170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_MD5SHA1)
260379170ce0SJerome Forissier 				goto check_element_none;
260479170ce0SJerome Forissier 		}
260579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA224)) {
260679170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_SHA224 ||
260779170ce0SJerome Forissier 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA224 ||
260879170ce0SJerome Forissier 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA224)
260979170ce0SJerome Forissier 				goto check_element_none;
261079170ce0SJerome Forissier 		}
261179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA256)) {
261279170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_SHA256 ||
261379170ce0SJerome Forissier 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA256 ||
261479170ce0SJerome Forissier 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA256)
261579170ce0SJerome Forissier 				goto check_element_none;
261679170ce0SJerome Forissier 		}
261779170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA384)) {
261879170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_SHA384 ||
261979170ce0SJerome Forissier 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA384 ||
262079170ce0SJerome Forissier 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA384)
262179170ce0SJerome Forissier 				goto check_element_none;
262279170ce0SJerome Forissier 		}
262379170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA512)) {
262479170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5_SHA512 ||
262579170ce0SJerome Forissier 			    alg == TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA512 ||
262679170ce0SJerome Forissier 			    alg == TEE_ALG_RSAES_PKCS1_OAEP_MGF1_SHA512)
262779170ce0SJerome Forissier 				goto check_element_none;
262879170ce0SJerome Forissier 		}
262979170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_RSASSA_NA1)) {
263079170ce0SJerome Forissier 			if (alg == TEE_ALG_RSASSA_PKCS1_V1_5)
263179170ce0SJerome Forissier 				goto check_element_none;
263279170ce0SJerome Forissier 		}
263379170ce0SJerome Forissier 		if (alg == TEE_ALG_RSA_NOPAD)
263479170ce0SJerome Forissier 			goto check_element_none;
263579170ce0SJerome Forissier 	}
263679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_DSA)) {
263779170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA1)) {
263879170ce0SJerome Forissier 			if (alg == TEE_ALG_DSA_SHA1)
263979170ce0SJerome Forissier 				goto check_element_none;
264079170ce0SJerome Forissier 		}
264179170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA224)) {
264279170ce0SJerome Forissier 			if (alg == TEE_ALG_DSA_SHA224)
264379170ce0SJerome Forissier 				goto check_element_none;
264479170ce0SJerome Forissier 		}
264579170ce0SJerome Forissier 		if (IS_ENABLED(CFG_CRYPTO_SHA256)) {
264679170ce0SJerome Forissier 			if (alg == TEE_ALG_DSA_SHA256)
264779170ce0SJerome Forissier 				goto check_element_none;
264879170ce0SJerome Forissier 		}
264979170ce0SJerome Forissier 	}
265079170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_DH)) {
265179170ce0SJerome Forissier 		if (alg == TEE_ALG_DH_DERIVE_SHARED_SECRET)
265279170ce0SJerome Forissier 			goto check_element_none;
265379170ce0SJerome Forissier 	}
265479170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_ECC)) {
2655fe2fd3ffSJens Wiklander 		if ((alg == __OPTEE_ALG_ECDH_P192 ||
2656fe2fd3ffSJens Wiklander 		     alg == __OPTEE_ALG_ECDSA_P192 ||
2657fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDH_DERIVE_SHARED_SECRET ||
2658fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDSA_SHA1) &&
265979170ce0SJerome Forissier 		    element == TEE_ECC_CURVE_NIST_P192)
266079170ce0SJerome Forissier 			return TEE_SUCCESS;
2661fe2fd3ffSJens Wiklander 		if ((alg == __OPTEE_ALG_ECDH_P224 ||
2662fe2fd3ffSJens Wiklander 		     alg == __OPTEE_ALG_ECDSA_P224 ||
2663fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDH_DERIVE_SHARED_SECRET ||
2664fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDSA_SHA224) &&
266579170ce0SJerome Forissier 		    element == TEE_ECC_CURVE_NIST_P224)
266679170ce0SJerome Forissier 			return TEE_SUCCESS;
2667fe2fd3ffSJens Wiklander 		if ((alg == __OPTEE_ALG_ECDH_P256 ||
2668fe2fd3ffSJens Wiklander 		     alg == __OPTEE_ALG_ECDSA_P256 ||
2669fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDH_DERIVE_SHARED_SECRET ||
2670fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDSA_SHA256) &&
267179170ce0SJerome Forissier 		    element == TEE_ECC_CURVE_NIST_P256)
267279170ce0SJerome Forissier 			return TEE_SUCCESS;
2673fe2fd3ffSJens Wiklander 		if ((alg == __OPTEE_ALG_ECDH_P384 ||
2674fe2fd3ffSJens Wiklander 		     alg == __OPTEE_ALG_ECDSA_P384 ||
2675fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDH_DERIVE_SHARED_SECRET ||
2676fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDSA_SHA384) &&
267779170ce0SJerome Forissier 		    element == TEE_ECC_CURVE_NIST_P384)
267879170ce0SJerome Forissier 			return TEE_SUCCESS;
2679fe2fd3ffSJens Wiklander 		if ((alg == __OPTEE_ALG_ECDH_P521 ||
2680fe2fd3ffSJens Wiklander 		     alg == __OPTEE_ALG_ECDSA_P521 ||
2681fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDH_DERIVE_SHARED_SECRET ||
2682fe2fd3ffSJens Wiklander 		     alg == TEE_ALG_ECDSA_SHA512) &&
268379170ce0SJerome Forissier 		    element == TEE_ECC_CURVE_NIST_P521)
268479170ce0SJerome Forissier 			return TEE_SUCCESS;
268579170ce0SJerome Forissier 	}
268679170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SM2_DSA)) {
268779170ce0SJerome Forissier 		if (alg == TEE_ALG_SM2_DSA_SM3 && element == TEE_ECC_CURVE_SM2)
268879170ce0SJerome Forissier 			return TEE_SUCCESS;
268979170ce0SJerome Forissier 	}
269079170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SM2_KEP)) {
269179170ce0SJerome Forissier 		if (alg == TEE_ALG_SM2_KEP && element == TEE_ECC_CURVE_SM2)
269279170ce0SJerome Forissier 			return TEE_SUCCESS;
269379170ce0SJerome Forissier 	}
269479170ce0SJerome Forissier 	if (IS_ENABLED(CFG_CRYPTO_SM2_PKE)) {
269579170ce0SJerome Forissier 		if (alg == TEE_ALG_SM2_PKE && element == TEE_ECC_CURVE_SM2)
269679170ce0SJerome Forissier 			return TEE_SUCCESS;
269779170ce0SJerome Forissier 	}
26983f61056dSSohaib ul Hassan 	if (IS_ENABLED(CFG_CRYPTO_X25519)) {
26993f61056dSSohaib ul Hassan 		if (alg == TEE_ALG_X25519 && element == TEE_ECC_CURVE_25519)
27003f61056dSSohaib ul Hassan 			return TEE_SUCCESS;
27013f61056dSSohaib ul Hassan 	}
2702e1f9cee7SSergiy Kibrik 	if (IS_ENABLED(CFG_CRYPTO_ED25519)) {
2703e1f9cee7SSergiy Kibrik 		if (alg == TEE_ALG_ED25519 && element == TEE_ECC_CURVE_25519)
2704e1f9cee7SSergiy Kibrik 			return TEE_SUCCESS;
2705e1f9cee7SSergiy Kibrik 	}
270679170ce0SJerome Forissier 
270779170ce0SJerome Forissier 	return TEE_ERROR_NOT_SUPPORTED;
270879170ce0SJerome Forissier check_element_none:
270979170ce0SJerome Forissier 	if (element == TEE_CRYPTO_ELEMENT_NONE)
271079170ce0SJerome Forissier 		return TEE_SUCCESS;
271179170ce0SJerome Forissier 	return TEE_ERROR_NOT_SUPPORTED;
271279170ce0SJerome Forissier }
2713