python/mozbuild/mozbuild/action/exe_7z_archive.py
author Matt Howell <mhowell@mozilla.com>
Mon, 14 Jan 2019 21:59:06 +0000
changeset 513929 63debb1d02d3ca285def538c88c1e47295923613
parent 449984 988a2837b8dc9c7e423ef7c0c8d91dbb6b1769df
child 513937 113f3db802c4ed0408d8cca513c595959d1be8a7
permissions -rw-r--r--
Bug 1514806 Part 6 - Disable running UPX in the installer package for targets it doesn't support. r=firefox-build-system-reviewers,mshal#firefox-build-system-reviewers Differential Revision: https://phabricator.services.mozilla.com/D15984

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from __future__ import print_function

import os
import shutil
import sys
import subprocess
import tempfile
import mozpack.path as mozpath
import buildconfig
from mozbuild.base import BuildEnvironmentNotFoundException

def archive_exe(pkg_dir, tagfile, sfx_package, package, use_upx):
    tmpdir = tempfile.mkdtemp(prefix='tmp')
    try:
        if pkg_dir:
            shutil.move(pkg_dir, 'core')

        if use_upx:
            final_sfx = mozpath.join(tmpdir, '7zSD.sfx')
            subprocess.check_call(['upx', '--best', '-o', final_sfx, sfx_package])
        else:
            final_sfx = sfx_package

        try:
            sevenz = buildconfig.config.substs['7Z']
        except BuildEnvironmentNotFoundException:
            # configure hasn't been run, just use the default
            sevenz = '7z'
        subprocess.check_call([sevenz, 'a', '-r', '-t7z', mozpath.join(tmpdir, 'app.7z'), '-mx', '-m0=BCJ2', '-m1=LZMA:d25', '-m2=LZMA:d19', '-m3=LZMA:d19', '-mb0:1', '-mb0s1:2', '-mb0s2:3'])

        with open(package, 'wb') as o:
            for i in [final_sfx, tagfile, mozpath.join(tmpdir, 'app.7z')]:
                shutil.copyfileobj(open(i, 'rb'), o)
        os.chmod(package, 0o0755)
    finally:
        if pkg_dir:
            shutil.move('core', pkg_dir)
        shutil.rmtree(tmpdir)

def main(args):
    if len(args) != 4:
        print('Usage: exe_7z_archive.py <pkg_dir> <tagfile> <sfx_package> <package> <use_upx>',
              file=sys.stderr)
        return 1
    else:
        archive_exe(args[0], args[1], args[2], args[3], args[4])
        return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))