1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# 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/.
import os
import shutil
import subprocess
import sys
import tempfile
import buildconfig
import mozpack.path as mozpath
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")
upx = buildconfig.substs.get("UPX", "upx")
wine = buildconfig.substs.get("WINE")
if wine and upx.lower().endswith(".exe"):
cmd = [wine, upx]
else:
cmd = [upx]
subprocess.check_call(
cmd
+ [
"--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:]))
|