| # SPDX-License-Identifier: GPL-2.0+ |
| # Copyright (c) 2022 Google, Inc |
| """Script for generating C array for binary files |
| |
| Usage: <script> output file1 file2 file3 ... |
| """ |
| |
| import subprocess |
| import sys |
| from pathlib import Path |
| |
| COPYRIGHT_HEADER = """/* |
| * Copyright (c) 2022 The Fuchsia Authors |
| * |
| * SPDX-License-Identifier: BSD-3-Clause |
| */ |
| |
| #include <stddef.h> |
| |
| """ |
| |
| def byte_array_declaration(data: bytes, name: str) -> str: |
| """Generates a byte C array declaration for a byte array""" |
| type_name = 'const uint8_t' |
| byte_str = ''.join([f'0x{b:02x},' for b in data]) |
| array_body = f'{{{byte_str}}}' |
| return f'{type_name} {name}[] = {array_body};' |
| |
| def main() -> int: |
| if len(sys.argv) < 3: |
| print("not enough arguments") |
| return |
| |
| with open(sys.argv[1], "w") as output_file: |
| output_file.write(COPYRIGHT_HEADER) |
| for file in sys.argv[2:]: |
| with open(file, "rb") as input_file: |
| # Best effort conversion to valid C variable name. |
| var_name = "arr_" + Path(file).name.replace(".", "_") |
| output_file.write(byte_array_declaration(input_file.read(), var_name)) |
| output_file.write("\n") |
| |
| subprocess.run([ |
| "/usr/bin/clang-format", |
| "--style=file", |
| "-i", |
| sys.argv[1], |
| ], check=True) |
| |
| return 0 |
| |
| if __name__ == "__main__": |
| sys.exit(main()) |
