KeyType: do a sanity check on the key type expression
Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
diff --git a/scripts/mbedtls_dev/crypto_knowledge.py b/scripts/mbedtls_dev/crypto_knowledge.py
index fb96ec0..4ff4f16 100644
--- a/scripts/mbedtls_dev/crypto_knowledge.py
+++ b/scripts/mbedtls_dev/crypto_knowledge.py
@@ -36,6 +36,7 @@
passing an expression of the form 'PSA_KEY_TYPE_xxx(param1, param2)'
in `name` as a string.
"""
+
self.name = name.strip()
"""The key type macro name (``PSA_KEY_TYPE_xxx``).
@@ -54,10 +55,13 @@
None if the key type is a macro without arguments.
"""
+ assert re.match(r'PSA_KEY_TYPE_\w+\Z', self.name)
+
self.expression = self.name
"""A C expression whose value is the key type encoding."""
if self.params is not None:
self.expression += '(' + ', '.join(self.params) + ')'
+
self.private_type = re.sub(r'_PUBLIC_KEY\Z', r'_KEY_PAIR', self.name)
"""The key type macro name for the corresponding key pair type.
