| 1 | #!/usr/bin/env python3 |
| 2 | # |
| 3 | # Copyright (C) 2020 Red Hat, Inc. |
| 4 | # |
| 5 | # SPDX-License-Identifier: GPL-2.0-or-later |
| 6 | |
| 7 | import argparse |
| 8 | import glob |
| 9 | import os |
| 10 | import shutil |
| 11 | import subprocess |
| 12 | import tempfile |
| 13 | |
| 14 | |
| 15 | def signcode(path): |
| 16 | cmd = os.environ.get("SIGNCODE") |
| 17 | if not cmd: |
| 18 | return |
| 19 | subprocess.run([cmd, path]) |
| 20 | |
| 21 | def find_deps(exe_or_dll, search_path, analyzed_deps): |
| 22 | deps = [exe_or_dll] |
| 23 | output = subprocess.check_output(["objdump", "-p", exe_or_dll], text=True) |
| 24 | output = output.split("\n") |
| 25 | for line in output: |
| 26 | if not line.lstrip().startswith("DLL Name: "): |
| 27 | continue |
| 28 | |
| 29 | dep = line.split("DLL Name: ")[1].strip() |
| 30 | if dep in analyzed_deps: |
| 31 | continue |
| 32 | |
| 33 | dll = os.path.join(search_path, dep) |
| 34 | if not os.path.exists(dll): |
| 35 | # assume it's a Windows provided dll, skip it |
| 36 | continue |
| 37 | |
| 38 | analyzed_deps.add(dep) |
| 39 | # locate the dll dependencies recursively |
| 40 | analyzed_deps, rdeps = find_deps(dll, search_path, analyzed_deps) |
| 41 | deps.extend(rdeps) |
| 42 | |
| 43 | return analyzed_deps, deps |
| 44 | |
| 45 | def main(): |
| 46 | parser = argparse.ArgumentParser(description="QEMU NSIS build helper.") |
| 47 | parser.add_argument("outfile") |
| 48 | parser.add_argument("prefix") |
| 49 | parser.add_argument("srcdir") |
| 50 | parser.add_argument("dlldir") |
| 51 | parser.add_argument("cpu") |
| 52 | parser.add_argument("nsisargs", nargs="*") |
| 53 | args = parser.parse_args() |
| 54 | |
| 55 | # canonicalize the Windows native prefix path |
| 56 | prefix = os.path.splitdrive(args.prefix)[1] |
| 57 | destdir = tempfile.mkdtemp() |
| 58 | try: |
| 59 | subprocess.run(["make", "install", "DESTDIR=" + destdir]) |
| 60 | with open( |
| 61 | os.path.join(destdir + prefix, "system-emulations.nsh"), "w" |
| 62 | ) as nsh, open( |
| 63 | os.path.join(destdir + prefix, "system-mui-text.nsh"), "w" |
| 64 | ) as muinsh: |
| 65 | for exe in sorted(glob.glob( |
| 66 | os.path.join(destdir + prefix, "qemu-system-*.exe") |
| 67 | )): |
| 68 | exe = os.path.basename(exe) |
| 69 | arch = exe[12:-4] |
| 70 | nsh.write( |
| 71 | """ |
| 72 | Section "{0}" Section_{0} |
| 73 | SetOutPath "$INSTDIR" |
| 74 | File "${{BINDIR}}\\{1}" |
| 75 | SectionEnd |
| 76 | """.format( |
| 77 | arch, exe |
| 78 | ) |
| 79 | ) |
| 80 | if arch.endswith('w'): |
| 81 | desc = arch[:-1] + " emulation (GUI)." |
| 82 | else: |
| 83 | desc = arch + " emulation." |
| 84 | |
| 85 | muinsh.write( |
| 86 | """ |
| 87 | !insertmacro MUI_DESCRIPTION_TEXT ${{Section_{0}}} "{1}" |
| 88 | """.format(arch, desc)) |
| 89 | |
| 90 | search_path = args.dlldir |
| 91 | print("Searching '%s' for the dependent dlls ..." % search_path) |
| 92 | dlldir = os.path.join(destdir + prefix, "dll") |
| 93 | os.mkdir(dlldir) |
| 94 | |
| 95 | analyzed_deps = set() |
| 96 | for exe in glob.glob(os.path.join(destdir + prefix, "*.exe")): |
| 97 | signcode(exe) |
| 98 | |
| 99 | # find all dll dependencies |
| 100 | analyzed_deps, deps = find_deps(exe, search_path, analyzed_deps) |
| 101 | deps = set(deps) |
| 102 | deps.remove(exe) |
| 103 | |
| 104 | # copy all dlls to the DLLDIR |
| 105 | for dep in deps: |
| 106 | dllfile = os.path.join(dlldir, os.path.basename(dep)) |
| 107 | print("Copying '%s' to '%s'" % (dep, dllfile)) |
| 108 | shutil.copy(dep, dllfile) |
| 109 | |
| 110 | makensis = [ |
| 111 | "makensis", |
| 112 | "-V2", |
| 113 | "-NOCD", |
| 114 | "-DSRCDIR=" + args.srcdir, |
| 115 | "-DBINDIR=" + destdir + prefix, |
| 116 | ] |
| 117 | if args.cpu == "aarch64" or args.cpu == "x86_64": |
| 118 | makensis += ["-DW64"] |
| 119 | makensis += ["-DDLLDIR=" + dlldir] |
| 120 | |
| 121 | makensis += ["-DOUTFILE=" + args.outfile] + args.nsisargs |
| 122 | subprocess.run(makensis) |
| 123 | signcode(args.outfile) |
| 124 | finally: |
| 125 | shutil.rmtree(destdir) |
| 126 | |
| 127 | |
| 128 | if __name__ == "__main__": |
| 129 | main() |