mirror of
https://github.com/espressif/esp-idf.git
synced 2025-08-07 20:00:53 +00:00
c4eec756f3
Also: - Split the secure service table into two parts: one DRAM-resident and the other DROM-resident. The former holds the services invoked when the cache is disabled or suspended while the latter holds rest of the services.
114 lines
4.0 KiB
Python
114 lines
4.0 KiB
Python
# SPDX-FileCopyrightText: 2025 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import argparse
|
|
import re
|
|
from typing import List
|
|
from typing import Tuple
|
|
|
|
SEC_SRV_TABLE_SPLIT_ID = 48
|
|
|
|
|
|
def parse_services(secure_service_tbl: str) -> List[Tuple[int, str, int]]:
|
|
services, service_ids = [], set()
|
|
pattern = re.compile(r'^([0-9A-Fa-fXx]+)\s+\S+\s+(\S+)\s+(\d+)')
|
|
|
|
with open(secure_service_tbl, 'r') as f:
|
|
for line in f:
|
|
if match := pattern.match(line):
|
|
service_id = int(match.group(1), 0)
|
|
if service_id in service_ids:
|
|
raise ValueError(f'Duplicate service call ID found: 0x{service_id:X}')
|
|
service_ids.add(service_id)
|
|
services.append((service_id, match.group(2), int(match.group(3))))
|
|
|
|
return sorted(services, key=lambda x: x[0])
|
|
|
|
|
|
def generate_num_header(services: List[Tuple[int, str, int]], output_file: str) -> None:
|
|
header = '''/**
|
|
* THIS FILE WAS AUTOMATICALLY GENERATED. DO NOT EDIT!
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
'''
|
|
body = '\n'.join(f'#define SS_{name.upper()}\t{nr}' for nr, name, _ in services)
|
|
footer = f'\n#define MAX_SECURE_SERVICES_ID\t{services[-1][0] + 1 if services else 0}\n'
|
|
footer += f'#define SECURE_SERVICES_NUM\t{len(services)}\n\n'
|
|
footer += f'#define SECURE_SERVICES_SPLIT_ID\t{SEC_SRV_TABLE_SPLIT_ID}\n\n'
|
|
footer += '''typedef void (*secure_service_t)(void);
|
|
typedef struct { secure_service_t func; int nargs; } secure_service_entry_t;
|
|
'''
|
|
footer += '\n#ifdef __cplusplus\n}\n#endif\n'
|
|
with open(output_file, 'w') as f:
|
|
f.write(header + body + footer)
|
|
|
|
|
|
def generate_dec_header(services: List[Tuple[int, str, int]], output_file: str) -> None:
|
|
header = '''/**
|
|
* THIS FILE WAS AUTOMATICALLY GENERATED. DO NOT EDIT!
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
'''
|
|
body = '\n'.join(f'void _ss_{name}(void);' for _, name, _ in services)
|
|
footer = '\n#ifdef __cplusplus\n}\n#endif\n'
|
|
with open(output_file, 'w') as f:
|
|
f.write(header + body + footer)
|
|
|
|
|
|
def generate_table_split(services: List[Tuple[int, str, int]], output_file_1: str, output_file_2: str) -> None:
|
|
header = '''/**
|
|
* THIS FILE WAS AUTOMATICALLY GENERATED. DO NOT EDIT!
|
|
*/
|
|
|
|
#pragma once
|
|
'''
|
|
services_1 = [service for service in services if service[0] < SEC_SRV_TABLE_SPLIT_ID]
|
|
services_2 = [service for service in services if service[0] >= SEC_SRV_TABLE_SPLIT_ID]
|
|
|
|
body_1 = '\n'.join(f'__SECURE_SERVICE({nr}, {name}, {nargs})' for nr, name, nargs in services_1)
|
|
body_2 = '\n'.join(f'__SECURE_SERVICE({nr}, {name}, {nargs})' for nr, name, nargs in services_2)
|
|
|
|
with open(output_file_1, 'w') as f1, open(output_file_2, 'w') as f2:
|
|
f1.write(header + body_1)
|
|
f2.write(header + body_2)
|
|
|
|
|
|
def generate_wrap_list(secure_service_tbl: str) -> None:
|
|
pattern = re.compile(r'^[0-9A-Fa-fXx]+\s+IDF\s+(\S+)\s+\d+')
|
|
with open(secure_service_tbl, 'r') as f:
|
|
wrap_list = [f'-Wl,--wrap={match.group(1)}' for line in f if (match := pattern.match(line))]
|
|
print(' '.join(wrap_list), end='')
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description='Generate secure service outputs')
|
|
parser.add_argument('--wrap', action='store_true', help='Generate linker wrap options')
|
|
parser.add_argument('secure_service_tbl', type=str, help='Path to secure service table file')
|
|
parser.add_argument('output_files', nargs='*', help='Output files: [secure_service_num.h, secure_service_dec.h, secure_service_1.h, secure_service_2.h]')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.wrap:
|
|
generate_wrap_list(args.secure_service_tbl)
|
|
else:
|
|
if len(args.output_files) != 4:
|
|
parser.error('Missing output header files!')
|
|
services = parse_services(args.secure_service_tbl)
|
|
generate_num_header(services, args.output_files[0])
|
|
generate_dec_header(services, args.output_files[1])
|
|
generate_table_split(services, args.output_files[2], args.output_files[3])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|