gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2025-11-27 10:00:59 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Add command-line options to accept paths of headers as a list of multiple file paths

Browse Source
This commit is contained in:
liang.he@intel.com 2025-10-28 07:24:44 +00:00
parent 7a9d37a20e
commit 1a1112f3d9
1 changed files with 84 additions and 72 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

156
ci/generate_checked_functions.py
View File

@ -1,5 +1,6 @@
from pycparser import c_parser, c_ast, parse_file from pycparser import c_parser, c_ast, parse_file
import os import os
import argparse
def collect_typedefs(ast): def collect_typedefs(ast):
@ -172,10 +173,23 @@ def generate_checked_function(func, typedefs):
return "\n".join(new_func) return "\n".join(new_func)
def parse_arguments():
parser = argparse.ArgumentParser(
description="Generate checked functions from header files."
)
parser.add_argument(
"--headers",
nargs="+",
required=True,
help="List of header file paths to process.",
)
return parser.parse_args()
# Updated process_header to scan all return types and create a proper Result type # Updated process_header to scan all return types and create a proper Result type
def process_header(): def process_headers(header_paths):
# Define the Result struct as a string # Define the Result struct as a string
RESULT_STRUCT = """ RESULT_STRUCT = """
typedef struct { typedef struct {
@ -186,84 +200,82 @@ def process_header():
} Result; } Result;
""" """
# Based on current file location, adjust the path to the header file for input_header in header_paths:
input_header = os.path.join( output_header = input_header.replace(".h", "_checked.h")
os.path.dirname(__file__), "../core/iwasm/include/wasm_export.h"
)
output_header = input_header.replace("wasm_export.h", "wasm_export_checked.h")
# Parse the header file with preprocessing # Parse the header file with preprocessing
ast = parse_file( ast = parse_file(
input_header, input_header,
use_cpp=True, use_cpp=True,
cpp_path="gcc", cpp_path="gcc",
cpp_args=[ cpp_args=[
"-E", "-E",
"-D__attribute__(x)=", "-D__attribute__(x)=",
"-D__asm__(x)=", "-D__asm__(x)=",
"-D__asm(x)=", "-D__asm(x)=",
"-D__builtin_va_list=int", "-D__builtin_va_list=int",
"-D__extension__=", "-D__extension__=",
"-D__inline__=", "-D__inline__=",
"-D__restrict=", "-D__restrict=",
"-D__restrict__=", "-D__restrict__=",
"-D_Static_assert(x, y)=", "-D_Static_assert(x, y)=",
"-D__signed=", "-D__signed=",
"-D__volatile__(x)=", "-D__volatile__(x)=",
"-Dstatic_assert(x, y)=", "-Dstatic_assert(x, y)=",
], ],
)
# Collect all typedefs
typedefs = collect_typedefs(ast)
# Collect all function declarations
functions = [
node
for node in ast.ext
if isinstance(node, c_ast.Decl) and isinstance(node.type, c_ast.FuncDecl)
]
# Scan all return types and update Result struct
return_types = set()
for func in functions:
if isinstance(func.type.type, c_ast.TypeDecl):
return_type = " ".join(func.type.type.type.names)
# resolved_type = resolve_typedef(typedefs, return_type)
return_types.add(return_type)
# Update the Result struct with all return types
for return_type in return_types:
if return_type == "void":
continue # No need to add void type
RESULT_STRUCT = RESULT_STRUCT.replace(
"// Add other types as needed",
f" {return_type} {return_type}_value;\n // Add other types as needed",
) )
# Generate the new header file # Collect all typedefs
with open(output_header, "w") as f: typedefs = collect_typedefs(ast)
f.write("#ifndef WASM_EXPORT_CHECKED_H\n#define WASM_EXPORT_CHECKED_H\n\n")
# necessary headers # Collect all function declarations
f.write("#include <stdbool.h>\n") functions = [
f.write("#include <stdint.h>\n") node
f.write("#include <stdlib.h>\n") for node in ast.ext
f.write("\n") if isinstance(node, c_ast.Decl) and isinstance(node.type, c_ast.FuncDecl)
f.write('#include "wasm_export.h"\n') ]
f.write('#include "lib_export.h"\n')
f.write("\n")
# Write the updated Result struct
f.write(RESULT_STRUCT + "\n")
# Scan all return types and update Result struct
return_types = set()
for func in functions: for func in functions:
new_func = generate_checked_function(func, typedefs) if isinstance(func.type.type, c_ast.TypeDecl):
f.write(new_func + "\n\n") return_type = " ".join(func.type.type.type.names)
# resolved_type = resolve_typedef(typedefs, return_type)
return_types.add(return_type)
f.write("#endif // WASM_EXPORT_CHECKED_H\n") # Update the Result struct with all return types
for return_type in return_types:
if return_type == "void":
continue # No need to add void type
RESULT_STRUCT = RESULT_STRUCT.replace(
"// Add other types as needed",
f" {return_type} {return_type}_value;\n // Add other types as needed",
)
# Generate the new header file
with open(output_header, "w") as f:
f.write("#ifndef WASM_EXPORT_CHECKED_H\n#define WASM_EXPORT_CHECKED_H\n\n")
# necessary headers
f.write("#include <stdbool.h>\n")
f.write("#include <stdint.h>\n")
f.write("#include <stdlib.h>\n")
f.write("\n")
f.write('#include "wasm_export.h"\n')
f.write('#include "lib_export.h"\n')
f.write("\n")
# Write the updated Result struct
f.write(RESULT_STRUCT + "\n")
for func in functions:
new_func = generate_checked_function(func, typedefs)
f.write(new_func + "\n\n")
f.write("#endif // WASM_EXPORT_CHECKED_H\n")
if __name__ == "__main__": if __name__ == "__main__":
process_header() args = parse_arguments()
process_headers(args.headers)