lib/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-07-07 12:21:11 +03:00
Code Activity

parse_function_arguments: extract per-argument function

Browse Source 
Internal refactoring only, no behavior change.

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine
2022-12-04 14:10:39 +01:00
parent 3a37f19e31
commit f153c5676b
1 changed files with 38 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
tests/scripts/generate_test_code.py
View File

@ -444,6 +444,40 @@ def parse_function_dependencies(line):
return dependencies return dependencies
def parse_function_argument(arg, arg_idx, args, local_vars, args_dispatch):
"""
Parses one test function's argument declaration.
:param arg: argument declaration.
:param arg_idx: current wrapper argument index.
:param args: accumulator of arguments' internal types.
:param local_vars: accumulator of internal variable declarations.
:param args_dispatch: accumulator of argument usage expressions.
:return: the number of new wrapper arguments,
or None if the argument declaration is invalid.
"""
arg = arg.strip()
if arg == '':
return 0
if re.search(INT_CHECK_REGEX, arg.strip()):
args.append('int')
args_dispatch.append('((mbedtls_test_argument_t*)params[%d])->s32' % arg_idx)
return 1
if re.search(CHAR_CHECK_REGEX, arg.strip()):
args.append('char*')
args_dispatch.append('(char *) params[%d]' % arg_idx)
return 1
if re.search(DATA_T_CHECK_REGEX, arg.strip()):
args.append('hex')
# create a structure
pointer_initializer = '(uint8_t *) params[%d]' % arg_idx
len_initializer = '((mbedtls_test_argument_t*)params[%d])->len' % (arg_idx+1)
local_vars.append(' data_t data%d = {%s, %s};\n' %
(arg_idx, pointer_initializer, len_initializer))
args_dispatch.append('&data%d' % arg_idx)
return 2
return None
def parse_function_arguments(line): def parse_function_arguments(line):
""" """
Parses test function signature for validation and generates Parses test function signature for validation and generates
@ -466,28 +500,12 @@ def parse_function_arguments(line):
# i.e. the test functions will not have a function pointer # i.e. the test functions will not have a function pointer
# argument. # argument.
for arg in line[:line.find(')')].split(','): for arg in line[:line.find(')')].split(','):
arg = arg.strip() indexes = parse_function_argument(arg, arg_idx,
if arg == '': args, local_vars, args_dispatch)
continue if indexes is None:
if re.search(INT_CHECK_REGEX, arg.strip()):
args.append('int')
args_dispatch.append('((mbedtls_test_argument_t*)params[%d])->s32' % arg_idx)
elif re.search(CHAR_CHECK_REGEX, arg.strip()):
args.append('char*')
args_dispatch.append('(char *) params[%d]' % arg_idx)
elif re.search(DATA_T_CHECK_REGEX, arg.strip()):
args.append('hex')
# create a structure
pointer_initializer = '(uint8_t *) params[%d]' % arg_idx
len_initializer = '((mbedtls_test_argument_t*)params[%d])->len' % (arg_idx+1)
local_vars.append(' data_t data%d = {%s, %s};\n' %
(arg_idx, pointer_initializer, len_initializer))
args_dispatch.append('&data%d' % arg_idx)
arg_idx += 1
else:
raise ValueError("Test function arguments can only be 'int', " raise ValueError("Test function arguments can only be 'int', "
"'char *' or 'data_t'\n%s" % line) "'char *' or 'data_t'\n%s" % line)
arg_idx += 1 arg_idx += indexes
return args, ''.join(local_vars), args_dispatch return args, ''.join(local_vars), args_dispatch