Filipe Rodrigues
e47b2e8639
Elf building now requires the start address.
Rust libraries now also extract to an object file ("temporarily" to fix a linker bug).
80 lines
1.9 KiB
Python
Executable File
80 lines
1.9 KiB
Python
Executable File
#!/bin/env python3
|
|
|
|
# Import
|
|
import yaml
|
|
import argparse
|
|
from pathlib import Path
|
|
import subprocess
|
|
|
|
|
|
def process_path(path: str | Path, input_dir: Path) -> Path:
|
|
path = Path(path)
|
|
if path.is_absolute():
|
|
# TODO: Make this work on windows?
|
|
return path.relative_to("/")
|
|
else:
|
|
return input_dir.joinpath(path)
|
|
|
|
|
|
def main(args):
|
|
config = yaml.safe_load(open(args.input_yaml))
|
|
input_dir = Path(args.input_yaml).parent
|
|
|
|
entry = config.get("entry")
|
|
start_addr = config["start_addr"]
|
|
objs = config["objs"]
|
|
objs = list(map(lambda obj_path: str(process_path(obj_path, input_dir)), objs))
|
|
|
|
sections = config["sections"]
|
|
sections = map(lambda section: f"KEEP(*({section}));", sections)
|
|
sections = '\n\t\t'.join(sections)
|
|
|
|
link_with = config.get("link_with") or []
|
|
link_with = list(map(lambda file: f"--just-symbols={file}", link_with))
|
|
|
|
# Create the linker script
|
|
with open(args.linker_script_output, "w") as linker_script_file:
|
|
|
|
linker_script_file.write(f"""\
|
|
SECTIONS {{
|
|
_.text = {hex(start_addr)};
|
|
.text _.text : {{
|
|
{sections}
|
|
}}
|
|
}}
|
|
|
|
SECTIONS {{
|
|
/DISCARD/ : {{ *(*) }}
|
|
}}""")
|
|
|
|
if entry != None:
|
|
linker_script_file.write(f"""\
|
|
|
|
|
|
ENTRY({entry})""")
|
|
|
|
args = [
|
|
args.ld_bin, #
|
|
"-o",
|
|
args.output,
|
|
"-EL",
|
|
"--omagic",
|
|
"--script",
|
|
args.linker_script_output,
|
|
"--warn-section-align",
|
|
"--no-warn-mismatch", # TODO: Might be worth considering some mismatches?
|
|
"--warn-common",
|
|
] + link_with + objs
|
|
subprocess.run(args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="`ld` wrapper")
|
|
parser.add_argument("input_yaml", type=str)
|
|
parser.add_argument("-o", dest="output", type=str, required=True)
|
|
parser.add_argument("--linker-script-output", dest="linker_script_output", type=str, required=True)
|
|
parser.add_argument("--ld-bin", dest="ld_bin", type=str, required=True)
|
|
|
|
args = parser.parse_args()
|
|
main(args)
|