qemu

FORK: QEMU emulator
git clone https://git.neptards.moe/neptards/qemu.git
Log | Files | Refs | Submodules | LICENSE

nsis.py (3880B)

      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.startswith("\tDLL 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         rdeps = find_deps(dll, search_path, analyzed_deps)
     41         deps.extend(rdeps)
     42 
     43     return 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         for exe in glob.glob(os.path.join(destdir + prefix, "*.exe")):
     96             signcode(exe)
     97 
     98             # find all dll dependencies
     99             deps = set(find_deps(exe, search_path, set()))
    100             deps.remove(exe)
    101 
    102             # copy all dlls to the DLLDIR
    103             for dep in deps:
    104                 dllfile = os.path.join(dlldir, os.path.basename(dep))
    105                 if (os.path.exists(dllfile)):
    106                     continue
    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 == "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()