yuzu-emu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2024-12-02 00:44:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Separate the parsing and generation parts of MacroCollector

Browse Source 
Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine 2021-01-25 21:36:53 +01:00
parent 58fd20c464
commit 7bb85c2640
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/generate_psa_constants.py
View File

@ -205,11 +205,8 @@ BIT_TEST_TEMPLATE = '''\
}\ }\
''' '''
class MacroCollector: class PSAMacroCollector:
"""Collect PSA crypto macro definitions from C header files. """Collect PSA crypto macro definitions from C header files.
1. Call `read_file` on the input header file(s).
2. Call `write_file` to write ``psa_constant_names_generated.c``.
""" """
def __init__(self): def __init__(self):
@ -303,6 +300,13 @@ class MacroCollector:
line = re.sub(self._nonascii_re, rb'', line).decode('ascii') line = re.sub(self._nonascii_re, rb'', line).decode('ascii')
self.read_line(line) self.read_line(line)
class CaseBuilder(PSAMacroCollector):
"""Collect PSA crypto macro definitions and write value recognition functions.
1. Call `read_file` on the input header file(s).
2. Call `write_file` to write ``psa_constant_names_generated.c``.
"""
@staticmethod @staticmethod
def _make_return_case(name): def _make_return_case(name):
return 'case %(name)s: return "%(name)s";' % {'name': name} return 'case %(name)s: return "%(name)s";' % {'name': name}
@ -404,7 +408,7 @@ class MacroCollector:
output_file.write(OUTPUT_TEMPLATE % data) output_file.write(OUTPUT_TEMPLATE % data)
def generate_psa_constants(header_file_names, output_file_name): def generate_psa_constants(header_file_names, output_file_name):
collector = MacroCollector() collector = CaseBuilder()
for header_file_name in header_file_names: for header_file_name in header_file_names:
with open(header_file_name, 'rb') as header_file: with open(header_file_name, 'rb') as header_file:
collector.read_file(header_file) collector.read_file(header_file)