buildtools: allow multiple input files in pmdinfogen

Process any number of input object files and write a unified output.

Signed-off-by: Dmitry Kozlyuk <dmitry.kozliuk@gmail.com>
This commit is contained in:
Dmitry Kozlyuk 2021-01-08 05:47:21 +03:00 committed by Thomas Monjalon
parent 5031436f45
commit 0fe5c4e5ad

View File

@ -197,7 +197,9 @@ def dump_drivers(drivers, file):
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("format", help="object file format, 'elf' or 'coff'")
parser.add_argument("input", help="input object file path or '-' for stdin")
parser.add_argument(
"input", nargs='+', help="input object file path or '-' for stdin"
)
parser.add_argument("output", help="output C file path or '-' for stdout")
return parser.parse_args()
@ -233,13 +235,16 @@ def open_output(path):
def main():
args = parse_args()
if args.input.count('-') > 1:
raise Exception("'-' input cannot be used multiple times")
if args.format == "elf" and "ELFFile" not in globals():
raise Exception("elftools module not found")
image = load_image(args.format, args.input)
drivers = load_drivers(image)
output = open_output(args.output)
dump_drivers(drivers, output)
for path in args.input:
image = load_image(args.format, path)
drivers = load_drivers(image)
dump_drivers(drivers, output)
if __name__ == "__main__":