xref: /optee_os/ta/pkcs11/src/token_capabilities.c (revision 0ef6b1440ff5d6f15c5f6933453cc8c47240f7eb)
18849c126SEtienne Carriere // SPDX-License-Identifier: BSD-2-Clause
28849c126SEtienne Carriere /*
38849c126SEtienne Carriere  * Copyright (c) 2017-2020, Linaro Limited
48849c126SEtienne Carriere  */
58849c126SEtienne Carriere 
68849c126SEtienne Carriere #include <assert.h>
78849c126SEtienne Carriere #include <pkcs11_ta.h>
88849c126SEtienne Carriere #include <string.h>
98849c126SEtienne Carriere #include <util.h>
108849c126SEtienne Carriere #include <tee_api.h>
118849c126SEtienne Carriere #include <tee_internal_api_extensions.h>
128849c126SEtienne Carriere 
138849c126SEtienne Carriere #include "pkcs11_helpers.h"
148849c126SEtienne Carriere #include "token_capabilities.h"
158849c126SEtienne Carriere 
168849c126SEtienne Carriere #define ALLOWED_PKCS11_CKFM	\
178849c126SEtienne Carriere 	(PKCS11_CKFM_ENCRYPT | PKCS11_CKFM_DECRYPT |		\
188849c126SEtienne Carriere 	 PKCS11_CKFM_DERIVE | PKCS11_CKFM_DIGEST |		\
198849c126SEtienne Carriere 	 PKCS11_CKFM_SIGN | PKCS11_CKFM_SIGN_RECOVER |		\
208849c126SEtienne Carriere 	 PKCS11_CKFM_VERIFY | PKCS11_CKFM_VERIFY_RECOVER |	\
218849c126SEtienne Carriere 	 PKCS11_CKFM_GENERATE |	PKCS11_CKFM_GENERATE_KEY_PAIR |	\
228849c126SEtienne Carriere 	 PKCS11_CKFM_WRAP | PKCS11_CKFM_UNWRAP)
238849c126SEtienne Carriere 
248849c126SEtienne Carriere /*
258849c126SEtienne Carriere  * Definition of supported processings for a PKCS#11 mechanisms
268849c126SEtienne Carriere  * @id: Mechanism ID
278849c126SEtienne Carriere  * @flags: Valid PKCS11_CKFM_* for a mechanism as per PKCS#11
288849c126SEtienne Carriere  * @one_shot: true of mechanism can be used for a one-short processing
298849c126SEtienne Carriere  * @string: Helper string of the mechanism ID for debug purpose
308849c126SEtienne Carriere  */
318849c126SEtienne Carriere struct pkcs11_mechachism_modes {
328849c126SEtienne Carriere 	uint32_t id;
338849c126SEtienne Carriere 	uint32_t flags;
348849c126SEtienne Carriere 	bool one_shot;
358849c126SEtienne Carriere #if CFG_TEE_TA_LOG_LEVEL > 0
368849c126SEtienne Carriere 	const char *string;
378849c126SEtienne Carriere #endif
388849c126SEtienne Carriere };
398849c126SEtienne Carriere 
408849c126SEtienne Carriere #if CFG_TEE_TA_LOG_LEVEL > 0
418849c126SEtienne Carriere #define MECHANISM(_label, _flags, _single_part)	\
428849c126SEtienne Carriere 	{					\
438849c126SEtienne Carriere 		.id = _label,			\
448849c126SEtienne Carriere 		.one_shot = (_single_part),	\
458849c126SEtienne Carriere 		.flags = (_flags),		\
468849c126SEtienne Carriere 		.string = #_label,		\
478849c126SEtienne Carriere 	}
488849c126SEtienne Carriere #else
498849c126SEtienne Carriere #define MECHANISM(_label, _flags, _single_part)	\
508849c126SEtienne Carriere 	{					\
518849c126SEtienne Carriere 		.id = _label,			\
528849c126SEtienne Carriere 		.one_shot = (_single_part),	\
538849c126SEtienne Carriere 		.flags = (_flags),		\
548849c126SEtienne Carriere 	}
558849c126SEtienne Carriere #endif
568849c126SEtienne Carriere 
578849c126SEtienne Carriere #define SINGLE_PART_ONLY	true
588849c126SEtienne Carriere #define ANY_PART		false
598849c126SEtienne Carriere 
60512cbf1dSJens Wiklander #define CKFM_CIPHER		(PKCS11_CKFM_ENCRYPT | PKCS11_CKFM_DECRYPT)
61512cbf1dSJens Wiklander #define CKFM_WRAP_UNWRAP	(PKCS11_CKFM_WRAP | PKCS11_CKFM_UNWRAP)
62512cbf1dSJens Wiklander #define CKFM_CIPHER_WRAP	(CKFM_CIPHER | CKFM_WRAP_UNWRAP)
63512cbf1dSJens Wiklander #define CKFM_CIPHER_WRAP_DERIVE	(CKFM_CIPHER_WRAP | PKCS11_CKFM_DERIVE)
64512cbf1dSJens Wiklander #define CKFM_AUTH_NO_RECOVER	(PKCS11_CKFM_SIGN | PKCS11_CKFM_VERIFY)
65512cbf1dSJens Wiklander #define CKFM_AUTH_WITH_RECOVER	(PKCS11_CKFM_SIGN_RECOVER | \
66512cbf1dSJens Wiklander 				 PKCS11_CKFM_VERIFY_RECOVER)
67512cbf1dSJens Wiklander 
688849c126SEtienne Carriere /* PKCS#11 specificies permitted operation for each mechanism  */
698849c126SEtienne Carriere static const struct pkcs11_mechachism_modes pkcs11_modes[] = {
70512cbf1dSJens Wiklander 	/* AES */
71512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_ECB, CKFM_CIPHER_WRAP, ANY_PART),
72512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_CBC, CKFM_CIPHER_WRAP, ANY_PART),
73512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_CBC_PAD, CKFM_CIPHER_WRAP, ANY_PART),
74512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_CTS, CKFM_CIPHER_WRAP, ANY_PART),
75512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_CTR, CKFM_CIPHER_WRAP, ANY_PART),
76*0ef6b144SVictor Chong 	MECHANISM(PKCS11_CKM_AES_CMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
77*0ef6b144SVictor Chong 	MECHANISM(PKCS11_CKM_AES_CMAC_GENERAL, CKFM_AUTH_NO_RECOVER, ANY_PART),
78512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_ECB_ENCRYPT_DATA, PKCS11_CKFM_DERIVE,
798849c126SEtienne Carriere 		  ANY_PART),
80512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_CBC_ENCRYPT_DATA, PKCS11_CKFM_DERIVE,
81512cbf1dSJens Wiklander 		  ANY_PART),
82512cbf1dSJens Wiklander 	MECHANISM(PKCS11_CKM_AES_KEY_GEN, PKCS11_CKFM_GENERATE, ANY_PART),
83fa247a2aSRuchika Gupta 	MECHANISM(PKCS11_CKM_GENERIC_SECRET_KEY_GEN, PKCS11_CKFM_GENERATE,
84fa247a2aSRuchika Gupta 		  ANY_PART),
859e91a619SVesa Jääskeläinen 	/* Digest */
869e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_MD5, PKCS11_CKFM_DIGEST, ANY_PART),
879e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA_1, PKCS11_CKFM_DIGEST, ANY_PART),
889e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA224, PKCS11_CKFM_DIGEST, ANY_PART),
899e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA256, PKCS11_CKFM_DIGEST, ANY_PART),
909e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA384, PKCS11_CKFM_DIGEST, ANY_PART),
919e91a619SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA512, PKCS11_CKFM_DIGEST, ANY_PART),
92689f4e5bSRuchika Gupta 	/* HMAC */
93689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_MD5_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
94689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_SHA_1_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
95689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_SHA224_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
96689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_SHA256_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
97689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_SHA384_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
98689f4e5bSRuchika Gupta 	MECHANISM(PKCS11_CKM_SHA512_HMAC, CKFM_AUTH_NO_RECOVER, ANY_PART),
9970b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_MD5_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER, ANY_PART),
10070b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_SHA_1_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER,
10170b6683bSVictor Chong 		  ANY_PART),
10270b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_SHA224_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER,
10370b6683bSVictor Chong 		  ANY_PART),
10470b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_SHA256_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER,
10570b6683bSVictor Chong 		  ANY_PART),
10670b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_SHA384_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER,
10770b6683bSVictor Chong 		  ANY_PART),
10870b6683bSVictor Chong 	MECHANISM(PKCS11_CKM_SHA512_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER,
10970b6683bSVictor Chong 		  ANY_PART),
11002b16804SVesa Jääskeläinen 	/* EC */
11102b16804SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_EC_KEY_PAIR_GEN, PKCS11_CKFM_GENERATE_KEY_PAIR,
11202b16804SVesa Jääskeläinen 		  ANY_PART),
113fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA, CKFM_AUTH_NO_RECOVER, SINGLE_PART_ONLY),
114fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA_SHA1, CKFM_AUTH_NO_RECOVER, ANY_PART),
115fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA_SHA224, CKFM_AUTH_NO_RECOVER, ANY_PART),
116fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA_SHA256, CKFM_AUTH_NO_RECOVER, ANY_PART),
117fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA_SHA384, CKFM_AUTH_NO_RECOVER, ANY_PART),
118fb279d8bSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_ECDSA_SHA512, CKFM_AUTH_NO_RECOVER, ANY_PART),
11986922832SVesa Jääskeläinen 	/* RSA */
12086922832SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_RSA_PKCS_KEY_PAIR_GEN,
12186922832SVesa Jääskeläinen 		  PKCS11_CKFM_GENERATE_KEY_PAIR, ANY_PART),
1220442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_RSA_PKCS, CKFM_CIPHER_WRAP | CKFM_AUTH_NO_RECOVER |
1230442c956SVesa Jääskeläinen 		  CKFM_AUTH_WITH_RECOVER, SINGLE_PART_ONLY),
124dc8c77fcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_RSA_PKCS_OAEP, CKFM_CIPHER_WRAP,
125dc8c77fcSVesa Jääskeläinen 		  SINGLE_PART_ONLY),
126d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER,
127d9af50bcSVesa Jääskeläinen 		  SINGLE_PART_ONLY),
1280442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_MD5_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
1290442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA1_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
130d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA1_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER, ANY_PART),
1310442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA256_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
1320442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA384_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
1330442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA512_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
134d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA256_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER,
135d9af50bcSVesa Jääskeläinen 		  ANY_PART),
136d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA384_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER,
137d9af50bcSVesa Jääskeläinen 		  ANY_PART),
138d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA512_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER,
139d9af50bcSVesa Jääskeläinen 		  ANY_PART),
1400442c956SVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA224_RSA_PKCS, CKFM_AUTH_NO_RECOVER, ANY_PART),
141d9af50bcSVesa Jääskeläinen 	MECHANISM(PKCS11_CKM_SHA224_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER,
142d9af50bcSVesa Jääskeläinen 		  ANY_PART),
1438849c126SEtienne Carriere };
1448849c126SEtienne Carriere 
1458849c126SEtienne Carriere #if CFG_TEE_TA_LOG_LEVEL > 0
1468849c126SEtienne Carriere const char *mechanism_string_id(enum pkcs11_mechanism_id id)
1478849c126SEtienne Carriere {
1488849c126SEtienne Carriere 	const size_t offset = sizeof("PKCS11_CKM_") - 1;
1498849c126SEtienne Carriere 	size_t n = 0;
1508849c126SEtienne Carriere 
1518849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(pkcs11_modes); n++)
1528849c126SEtienne Carriere 		if (pkcs11_modes[n].id == id)
1538849c126SEtienne Carriere 			return pkcs11_modes[n].string + offset;
1548849c126SEtienne Carriere 
1558849c126SEtienne Carriere 	return "Unknown ID";
1568849c126SEtienne Carriere }
1578849c126SEtienne Carriere #endif /*CFG_TEE_TA_LOG_LEVEL*/
1588849c126SEtienne Carriere 
1598849c126SEtienne Carriere /*
1608849c126SEtienne Carriere  * Return true if @id is a valid mechanism ID
1618849c126SEtienne Carriere  */
1628849c126SEtienne Carriere bool mechanism_is_valid(enum pkcs11_mechanism_id id)
1638849c126SEtienne Carriere {
1648849c126SEtienne Carriere 	size_t n = 0;
1658849c126SEtienne Carriere 
1668849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(pkcs11_modes); n++)
1678849c126SEtienne Carriere 		if (id == pkcs11_modes[n].id)
1688849c126SEtienne Carriere 			return true;
1698849c126SEtienne Carriere 
1708849c126SEtienne Carriere 	return false;
1718849c126SEtienne Carriere }
1728849c126SEtienne Carriere 
1738849c126SEtienne Carriere /*
1748849c126SEtienne Carriere  * Return true if mechanism ID is valid and flags matches PKCS#11 compliancy
1758849c126SEtienne Carriere  */
1768849c126SEtienne Carriere bool __maybe_unused mechanism_flags_complies_pkcs11(uint32_t mechanism_type,
1778849c126SEtienne Carriere 						    uint32_t flags)
1788849c126SEtienne Carriere {
1798849c126SEtienne Carriere 	size_t n = 0;
1808849c126SEtienne Carriere 
1818849c126SEtienne Carriere 	assert((flags & ~ALLOWED_PKCS11_CKFM) == 0);
1828849c126SEtienne Carriere 
1838849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(pkcs11_modes); n++) {
1848849c126SEtienne Carriere 		if (pkcs11_modes[n].id == mechanism_type) {
1858849c126SEtienne Carriere 			if (flags & ~pkcs11_modes[n].flags)
1868849c126SEtienne Carriere 				EMSG("%s flags: 0x%"PRIx32" vs 0x%"PRIx32,
1878849c126SEtienne Carriere 				     id2str_mechanism(mechanism_type),
1888849c126SEtienne Carriere 				     flags, pkcs11_modes[n].flags);
1898849c126SEtienne Carriere 
1908849c126SEtienne Carriere 			return (flags & ~pkcs11_modes[n].flags) == 0;
1918849c126SEtienne Carriere 		}
1928849c126SEtienne Carriere 	}
1938849c126SEtienne Carriere 
1948849c126SEtienne Carriere 	/* Mechanism ID unexpectedly not found */
1958849c126SEtienne Carriere 	return false;
1968849c126SEtienne Carriere }
1978849c126SEtienne Carriere 
198512cbf1dSJens Wiklander bool mechanism_is_one_shot_only(uint32_t mechanism_type)
199512cbf1dSJens Wiklander {
200512cbf1dSJens Wiklander 	size_t n = 0;
201512cbf1dSJens Wiklander 
202512cbf1dSJens Wiklander 	for (n = 0; n < ARRAY_SIZE(pkcs11_modes); n++)
203512cbf1dSJens Wiklander 		if (pkcs11_modes[n].id == mechanism_type)
204512cbf1dSJens Wiklander 			return pkcs11_modes[n].one_shot;
205512cbf1dSJens Wiklander 
206512cbf1dSJens Wiklander 	/* Mechanism ID unexpectedly not found */
207512cbf1dSJens Wiklander 	TEE_Panic(PKCS11_RV_NOT_FOUND);
208512cbf1dSJens Wiklander 	/* Dummy return to keep compiler happy */
209512cbf1dSJens Wiklander 	return false;
210512cbf1dSJens Wiklander }
211512cbf1dSJens Wiklander 
212512cbf1dSJens Wiklander /*
213512cbf1dSJens Wiklander  * Field single_part_only is unused from array token_mechanism[], hence
214512cbf1dSJens Wiklander  * simply use ANY_PART for all mechanism there.
215512cbf1dSJens Wiklander  */
216512cbf1dSJens Wiklander #define TA_MECHANISM(_label, _flags)	MECHANISM((_label), (_flags), ANY_PART)
217512cbf1dSJens Wiklander 
2188849c126SEtienne Carriere /*
2198849c126SEtienne Carriere  * Arrays that centralizes the IDs and processing flags for mechanisms
220c95980b1SEtienne Carriere  * supported by each embedded token.
2218849c126SEtienne Carriere  */
2228849c126SEtienne Carriere const struct pkcs11_mechachism_modes token_mechanism[] = {
2235f80f270SRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_AES_ECB, CKFM_CIPHER_WRAP),
2245f80f270SRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_AES_CBC, CKFM_CIPHER_WRAP),
225512cbf1dSJens Wiklander 	TA_MECHANISM(PKCS11_CKM_AES_CBC_PAD, CKFM_CIPHER),
226512cbf1dSJens Wiklander 	TA_MECHANISM(PKCS11_CKM_AES_CTR, CKFM_CIPHER),
227512cbf1dSJens Wiklander 	TA_MECHANISM(PKCS11_CKM_AES_CTS, CKFM_CIPHER),
228*0ef6b144SVictor Chong 	TA_MECHANISM(PKCS11_CKM_AES_CMAC, CKFM_AUTH_NO_RECOVER),
229*0ef6b144SVictor Chong 	TA_MECHANISM(PKCS11_CKM_AES_CMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
230512cbf1dSJens Wiklander 	TA_MECHANISM(PKCS11_CKM_AES_ECB_ENCRYPT_DATA, PKCS11_CKFM_DERIVE),
231512cbf1dSJens Wiklander 	TA_MECHANISM(PKCS11_CKM_AES_CBC_ENCRYPT_DATA, PKCS11_CKFM_DERIVE),
232fa247a2aSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_AES_KEY_GEN, PKCS11_CKFM_GENERATE),
233fa247a2aSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_GENERIC_SECRET_KEY_GEN, PKCS11_CKFM_GENERATE),
2349e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_MD5, PKCS11_CKFM_DIGEST),
2359e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA_1, PKCS11_CKFM_DIGEST),
2369e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA224, PKCS11_CKFM_DIGEST),
2379e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA256, PKCS11_CKFM_DIGEST),
2389e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA384, PKCS11_CKFM_DIGEST),
2399e91a619SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA512, PKCS11_CKFM_DIGEST),
240689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_MD5_HMAC, CKFM_AUTH_NO_RECOVER),
241689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_SHA_1_HMAC, CKFM_AUTH_NO_RECOVER),
242689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_SHA224_HMAC, CKFM_AUTH_NO_RECOVER),
243689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_SHA256_HMAC, CKFM_AUTH_NO_RECOVER),
244689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_SHA384_HMAC, CKFM_AUTH_NO_RECOVER),
245689f4e5bSRuchika Gupta 	TA_MECHANISM(PKCS11_CKM_SHA512_HMAC, CKFM_AUTH_NO_RECOVER),
24670b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_MD5_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
24770b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_SHA_1_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
24870b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_SHA224_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
24970b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_SHA256_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
25070b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_SHA384_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
25170b6683bSVictor Chong 	TA_MECHANISM(PKCS11_CKM_SHA512_HMAC_GENERAL, CKFM_AUTH_NO_RECOVER),
25202b16804SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_EC_KEY_PAIR_GEN,
25302b16804SVesa Jääskeläinen 		     PKCS11_CKFM_GENERATE_KEY_PAIR),
254fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA, CKFM_AUTH_NO_RECOVER),
255fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA_SHA1, CKFM_AUTH_NO_RECOVER),
256fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA_SHA224, CKFM_AUTH_NO_RECOVER),
257fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA_SHA256, CKFM_AUTH_NO_RECOVER),
258fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA_SHA384, CKFM_AUTH_NO_RECOVER),
259fb279d8bSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_ECDSA_SHA512, CKFM_AUTH_NO_RECOVER),
26086922832SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_RSA_PKCS_KEY_PAIR_GEN,
26186922832SVesa Jääskeläinen 		     PKCS11_CKFM_GENERATE_KEY_PAIR),
2620442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_RSA_PKCS, CKFM_CIPHER | CKFM_AUTH_NO_RECOVER),
263d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
2640442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_MD5_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
2650442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA1_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
266dc8c77fcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_RSA_PKCS_OAEP, CKFM_CIPHER),
267d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA1_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
2680442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA256_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
2690442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA384_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
2700442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA512_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
271d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA256_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
272d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA384_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
273d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA512_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
2740442c956SVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA224_RSA_PKCS, CKFM_AUTH_NO_RECOVER),
275d9af50bcSVesa Jääskeläinen 	TA_MECHANISM(PKCS11_CKM_SHA224_RSA_PKCS_PSS, CKFM_AUTH_NO_RECOVER),
2768849c126SEtienne Carriere };
2778849c126SEtienne Carriere 
2788849c126SEtienne Carriere /*
2798849c126SEtienne Carriere  * tee_malloc_mechanism_array - Allocate and fill array of supported mechanisms
2808849c126SEtienne Carriere  * @count: [in] [out] Pointer to number of mechanism IDs in client resource
2818849c126SEtienne Carriere  * Return allocated array of the supported mechanism IDs
2828849c126SEtienne Carriere  *
2838849c126SEtienne Carriere  * Allocates array with 32bit cells mechanism IDs for the supported ones only
2848849c126SEtienne Carriere  * if *@count covers number mechanism IDs exposed.
2858849c126SEtienne Carriere  */
2868849c126SEtienne Carriere uint32_t *tee_malloc_mechanism_list(size_t *out_count)
2878849c126SEtienne Carriere {
2888849c126SEtienne Carriere 	size_t n = 0;
2898849c126SEtienne Carriere 	size_t count = 0;
2908849c126SEtienne Carriere 	uint32_t *array = NULL;
2918849c126SEtienne Carriere 
2928849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(token_mechanism); n++)
2938849c126SEtienne Carriere 		if (token_mechanism[n].flags)
2948849c126SEtienne Carriere 			count++;
2958849c126SEtienne Carriere 
2968849c126SEtienne Carriere 	if (*out_count >= count)
2978849c126SEtienne Carriere 		array = TEE_Malloc(count * sizeof(*array),
2988849c126SEtienne Carriere 				   TEE_USER_MEM_HINT_NO_FILL_ZERO);
2998849c126SEtienne Carriere 
3008849c126SEtienne Carriere 	*out_count = count;
3018849c126SEtienne Carriere 
3028849c126SEtienne Carriere 	if (!array)
3038849c126SEtienne Carriere 		return NULL;
3048849c126SEtienne Carriere 
3058849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(token_mechanism); n++) {
3068849c126SEtienne Carriere 		if (token_mechanism[n].flags) {
3078849c126SEtienne Carriere 			count--;
3088849c126SEtienne Carriere 			array[count] = token_mechanism[n].id;
3098849c126SEtienne Carriere 		}
3108849c126SEtienne Carriere 	}
3118849c126SEtienne Carriere 	assert(!count);
3128849c126SEtienne Carriere 
3138849c126SEtienne Carriere 	return array;
3148849c126SEtienne Carriere }
3158849c126SEtienne Carriere 
3168849c126SEtienne Carriere uint32_t mechanism_supported_flags(enum pkcs11_mechanism_id id)
3178849c126SEtienne Carriere {
3188849c126SEtienne Carriere 	size_t n = 0;
3198849c126SEtienne Carriere 
3208849c126SEtienne Carriere 	for (n = 0; n < ARRAY_SIZE(token_mechanism); n++) {
3218849c126SEtienne Carriere 		if (id == token_mechanism[n].id) {
3228849c126SEtienne Carriere 			uint32_t flags = token_mechanism[n].flags;
3238849c126SEtienne Carriere 
3248849c126SEtienne Carriere 			assert(mechanism_flags_complies_pkcs11(id, flags));
3258849c126SEtienne Carriere 			return flags;
3268849c126SEtienne Carriere 		}
3278849c126SEtienne Carriere 	}
3288849c126SEtienne Carriere 
3298849c126SEtienne Carriere 	return 0;
3308849c126SEtienne Carriere }
331512cbf1dSJens Wiklander 
3322d0cd829SRuchika Gupta void pkcs11_mechanism_supported_key_sizes(uint32_t proc_id,
3332d0cd829SRuchika Gupta 					  uint32_t *min_key_size,
334512cbf1dSJens Wiklander 					  uint32_t *max_key_size)
335512cbf1dSJens Wiklander {
336512cbf1dSJens Wiklander 	switch (proc_id) {
337fa247a2aSRuchika Gupta 	case PKCS11_CKM_GENERIC_SECRET_KEY_GEN:
338fa247a2aSRuchika Gupta 		/* This mechanism expects the keysize to be returned in bits */
339fa247a2aSRuchika Gupta 		*min_key_size = 1;		/* in bits */
340fa247a2aSRuchika Gupta 		*max_key_size = 4096;		/* in bits */
341fa247a2aSRuchika Gupta 		break;
342689f4e5bSRuchika Gupta 	case PKCS11_CKM_MD5_HMAC:
34370b6683bSVictor Chong 	case PKCS11_CKM_MD5_HMAC_GENERAL:
344d92c3cddSRuchika Gupta 		*min_key_size = 8;
345d92c3cddSRuchika Gupta 		*max_key_size = 64;
346689f4e5bSRuchika Gupta 		break;
347689f4e5bSRuchika Gupta 	case PKCS11_CKM_SHA_1_HMAC:
34870b6683bSVictor Chong 	case PKCS11_CKM_SHA_1_HMAC_GENERAL:
349d92c3cddSRuchika Gupta 		*min_key_size = 10;
350d92c3cddSRuchika Gupta 		*max_key_size = 64;
351689f4e5bSRuchika Gupta 		break;
352689f4e5bSRuchika Gupta 	case PKCS11_CKM_SHA224_HMAC:
35370b6683bSVictor Chong 	case PKCS11_CKM_SHA224_HMAC_GENERAL:
354d92c3cddSRuchika Gupta 		*min_key_size = 14;
355d92c3cddSRuchika Gupta 		*max_key_size = 64;
356689f4e5bSRuchika Gupta 		break;
357689f4e5bSRuchika Gupta 	case PKCS11_CKM_SHA256_HMAC:
35870b6683bSVictor Chong 	case PKCS11_CKM_SHA256_HMAC_GENERAL:
359d92c3cddSRuchika Gupta 		*min_key_size = 24;
360d92c3cddSRuchika Gupta 		*max_key_size = 128;
361689f4e5bSRuchika Gupta 		break;
362689f4e5bSRuchika Gupta 	case PKCS11_CKM_SHA384_HMAC:
36370b6683bSVictor Chong 	case PKCS11_CKM_SHA384_HMAC_GENERAL:
364d92c3cddSRuchika Gupta 		*min_key_size = 32;
365d92c3cddSRuchika Gupta 		*max_key_size = 128;
366689f4e5bSRuchika Gupta 		break;
367689f4e5bSRuchika Gupta 	case PKCS11_CKM_SHA512_HMAC:
36870b6683bSVictor Chong 	case PKCS11_CKM_SHA512_HMAC_GENERAL:
369d92c3cddSRuchika Gupta 		*min_key_size = 32;
370d92c3cddSRuchika Gupta 		*max_key_size = 128;
371689f4e5bSRuchika Gupta 		break;
372512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_KEY_GEN:
373512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_ECB:
374512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_CBC:
375512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_CBC_PAD:
376512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_CTR:
377512cbf1dSJens Wiklander 	case PKCS11_CKM_AES_CTS:
378*0ef6b144SVictor Chong 	case PKCS11_CKM_AES_CMAC:
379*0ef6b144SVictor Chong 	case PKCS11_CKM_AES_CMAC_GENERAL:
380512cbf1dSJens Wiklander 		*min_key_size = 16;
381512cbf1dSJens Wiklander 		*max_key_size = 32;
382512cbf1dSJens Wiklander 		break;
38302b16804SVesa Jääskeläinen 	case PKCS11_CKM_EC_KEY_PAIR_GEN:
384fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA:
385fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA1:
386fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA224:
387fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA256:
388fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA384:
389fb279d8bSVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA512:
39002b16804SVesa Jääskeläinen 		*min_key_size = 160;	/* in bits */
39102b16804SVesa Jääskeläinen 		*max_key_size = 521;	/* in bits */
39202b16804SVesa Jääskeläinen 		break;
39386922832SVesa Jääskeläinen 	case PKCS11_CKM_RSA_PKCS_KEY_PAIR_GEN:
3940442c956SVesa Jääskeläinen 	case PKCS11_CKM_RSA_PKCS:
3950442c956SVesa Jääskeläinen 	case PKCS11_CKM_MD5_RSA_PKCS:
3960442c956SVesa Jääskeläinen 	case PKCS11_CKM_SHA1_RSA_PKCS:
397dc8c77fcSVesa Jääskeläinen 	case PKCS11_CKM_RSA_PKCS_OAEP:
398d9af50bcSVesa Jääskeläinen 	case PKCS11_CKM_SHA1_RSA_PKCS_PSS:
3990442c956SVesa Jääskeläinen 	case PKCS11_CKM_SHA256_RSA_PKCS:
4000442c956SVesa Jääskeläinen 	case PKCS11_CKM_SHA384_RSA_PKCS:
4010442c956SVesa Jääskeläinen 	case PKCS11_CKM_SHA512_RSA_PKCS:
402d9af50bcSVesa Jääskeläinen 	case PKCS11_CKM_SHA256_RSA_PKCS_PSS:
403d9af50bcSVesa Jääskeläinen 	case PKCS11_CKM_SHA384_RSA_PKCS_PSS:
404d9af50bcSVesa Jääskeläinen 	case PKCS11_CKM_SHA512_RSA_PKCS_PSS:
4050442c956SVesa Jääskeläinen 	case PKCS11_CKM_SHA224_RSA_PKCS:
406d9af50bcSVesa Jääskeläinen 	case PKCS11_CKM_SHA224_RSA_PKCS_PSS:
40786922832SVesa Jääskeläinen 		*min_key_size = 256;	/* in bits */
40886922832SVesa Jääskeläinen 		*max_key_size = 4096;	/* in bits */
40986922832SVesa Jääskeläinen 		break;
410512cbf1dSJens Wiklander 	default:
411512cbf1dSJens Wiklander 		*min_key_size = 0;
412512cbf1dSJens Wiklander 		*max_key_size = 0;
413512cbf1dSJens Wiklander 		break;
414512cbf1dSJens Wiklander 	}
415512cbf1dSJens Wiklander }
4162d0cd829SRuchika Gupta 
4172d0cd829SRuchika Gupta void mechanism_supported_key_sizes_bytes(uint32_t proc_id,
4182d0cd829SRuchika Gupta 					 uint32_t *min_key_size,
4192d0cd829SRuchika Gupta 					 uint32_t *max_key_size)
4202d0cd829SRuchika Gupta {
4212d0cd829SRuchika Gupta 	pkcs11_mechanism_supported_key_sizes(proc_id, min_key_size,
4222d0cd829SRuchika Gupta 					     max_key_size);
4232d0cd829SRuchika Gupta 
424db28c542SVesa Jääskeläinen 	switch (proc_id) {
425db28c542SVesa Jääskeläinen 	case PKCS11_CKM_GENERIC_SECRET_KEY_GEN:
426db28c542SVesa Jääskeläinen 	case PKCS11_CKM_EC_KEY_PAIR_GEN:
427db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA:
428db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA1:
429db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA224:
430db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA256:
431db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA384:
432db28c542SVesa Jääskeläinen 	case PKCS11_CKM_ECDSA_SHA512:
43386922832SVesa Jääskeläinen 	case PKCS11_CKM_RSA_PKCS_KEY_PAIR_GEN:
434db28c542SVesa Jääskeläinen 		/* Size is in bits -> convert to bytes and ceil */
435db28c542SVesa Jääskeläinen 		*min_key_size = ROUNDUP(*min_key_size, 8) / 8;
436db28c542SVesa Jääskeläinen 		*max_key_size = ROUNDUP(*max_key_size, 8) / 8;
437db28c542SVesa Jääskeläinen 		break;
438db28c542SVesa Jääskeläinen 	default:
439db28c542SVesa Jääskeläinen 		/* Size is already in bytes */
440db28c542SVesa Jääskeläinen 		break;
4412d0cd829SRuchika Gupta 	}
4422d0cd829SRuchika Gupta }
443