Use regexp pattern instaed of string

Signed-off-by: gabor-mezei-arm <gabor.mezei@arm.com>
This commit is contained in:
gabor-mezei-arm 2021-06-29 11:17:54 +02:00
parent 7e0d724d40
commit 5071a2e30e
No known key found for this signature in database
GPG Key ID: 106F5A41ECC305BD

View File

@ -19,7 +19,7 @@ This module is entirely based on the PSA API.
# limitations under the License. # limitations under the License.
import re import re
from typing import Dict, Iterable, Optional, Tuple from typing import Dict, Iterable, Optional, Pattern, Tuple
from mbedtls_dev.asymmetric_key_data import ASYMMETRIC_KEY_DATA from mbedtls_dev.asymmetric_key_data import ASYMMETRIC_KEY_DATA
@ -138,9 +138,9 @@ class KeyType:
[self.DATA_BLOCK[:length % len(self.DATA_BLOCK)]]) [self.DATA_BLOCK[:length % len(self.DATA_BLOCK)]])
KEY_TYPE_FOR_SIGNATURE = { KEY_TYPE_FOR_SIGNATURE = {
'PSA_KEY_USAGE_SIGN_HASH': '.*KEY_PAIR', 'PSA_KEY_USAGE_SIGN_HASH': re.compile('.*KEY_PAIR'),
'PSA_KEY_USAGE_VERIFY_HASH': '.*KEY.*' 'PSA_KEY_USAGE_VERIFY_HASH': re.compile('.*KEY.*')
} #type: Dict[str, str] } #type: Dict[str, Pattern]
"""Use a regexp to determine key types for which signature is possible """Use a regexp to determine key types for which signature is possible
when using the actual usage flag. when using the actual usage flag.
""" """