xref: /openbmc/qemu/scripts/nsis.py (revision b14df228) 
1#!/usr/bin/env python3
2#
3# Copyright (C) 2020 Red Hat, Inc.
4#
5# SPDX-License-Identifier: GPL-2.0-or-later
6
7import argparse
8import glob
9import os
10import shutil
11import subprocess
12import tempfile
13
14
15def signcode(path):
16    cmd = os.environ.get("SIGNCODE")
17    if not cmd:
18        return
19    subprocess.run([cmd, path])
20
21
22def main():
23    parser = argparse.ArgumentParser(description="QEMU NSIS build helper.")
24    parser.add_argument("outfile")
25    parser.add_argument("prefix")
26    parser.add_argument("srcdir")
27    parser.add_argument("cpu")
28    parser.add_argument("nsisargs", nargs="*")
29    args = parser.parse_args()
30
31    destdir = tempfile.mkdtemp()
32    try:
33        subprocess.run(["make", "install", "DESTDIR=" + destdir + os.path.sep])
34        with open(
35            os.path.join(destdir + args.prefix, "system-emulations.nsh"), "w"
36        ) as nsh, open(
37            os.path.join(destdir + args.prefix, "system-mui-text.nsh"), "w"
38        ) as muinsh:
39            for exe in sorted(glob.glob(
40                os.path.join(destdir + args.prefix, "qemu-system-*.exe")
41            )):
42                exe = os.path.basename(exe)
43                arch = exe[12:-4]
44                nsh.write(
45                    """
46                Section "{0}" Section_{0}
47                SetOutPath "$INSTDIR"
48                File "${{BINDIR}}\\{1}"
49                SectionEnd
50                """.format(
51                        arch, exe
52                    )
53                )
54                if arch.endswith('w'):
55                    desc = arch[:-1] + " emulation (GUI)."
56                else:
57                    desc = arch + " emulation."
58
59                muinsh.write(
60                    """
61                !insertmacro MUI_DESCRIPTION_TEXT ${{Section_{0}}} "{1}"
62                """.format(arch, desc))
63
64        for exe in glob.glob(os.path.join(destdir + args.prefix, "*.exe")):
65            signcode(exe)
66
67        makensis = [
68            "makensis",
69            "-V2",
70            "-NOCD",
71            "-DSRCDIR=" + args.srcdir,
72            "-DBINDIR=" + destdir + args.prefix,
73        ]
74        dlldir = "w32"
75        if args.cpu == "x86_64":
76            dlldir = "w64"
77            makensis += ["-DW64"]
78        if os.path.exists(os.path.join(args.srcdir, "dll")):
79            makensis += ["-DDLLDIR={0}/dll/{1}".format(args.srcdir, dlldir)]
80
81        makensis += ["-DOUTFILE=" + args.outfile] + args.nsisargs
82        subprocess.run(makensis)
83        signcode(args.outfile)
84    finally:
85        shutil.rmtree(destdir)
86
87
88if __name__ == "__main__":
89    main()
90