Magisk/build.py

458 lines
14 KiB
Python
Raw Normal View History

2017-06-04 01:39:52 +08:00
#!/usr/bin/env python3
2017-06-03 20:19:01 +08:00
import sys
import os
import subprocess
2017-11-12 04:17:56 +08:00
if os.name == 'nt':
2018-05-13 03:04:40 +08:00
import colorama
colorama.init()
2017-11-12 04:17:56 +08:00
2017-06-03 20:19:01 +08:00
def error(str):
print('\n' + '\033[41m' + str + '\033[0m' + '\n')
sys.exit(1)
def header(str):
print('\n' + '\033[44m' + str + '\033[0m' + '\n')
2018-08-06 02:01:04 +08:00
def vprint(str):
if args.verbose:
print(str)
2017-06-03 20:19:01 +08:00
# Environment checks
if not sys.version_info >= (3, 6):
error('Requires Python 3.6+')
2017-06-03 20:19:01 +08:00
if 'ANDROID_HOME' not in os.environ:
error('Please add Android SDK path to ANDROID_HOME environment variable!')
try:
subprocess.run(['java', '-version'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
except FileNotFoundError:
2018-05-13 03:04:40 +08:00
error('Please install JDK and make sure \'java\' is available in PATH')
2017-06-03 20:19:01 +08:00
import argparse
import multiprocessing
import zipfile
import datetime
import errno
import shutil
import lzma
import base64
2017-11-15 05:25:19 +08:00
import tempfile
2017-06-03 20:19:01 +08:00
if 'ANDROID_NDK_HOME' in os.environ:
ndk_build = os.path.join(os.environ['ANDROID_NDK_HOME'], 'ndk-build')
2017-12-04 15:16:41 +08:00
else:
ndk_build = os.path.join(os.environ['ANDROID_HOME'], 'ndk-bundle', 'ndk-build')
2018-05-13 03:04:40 +08:00
cpu_count = multiprocessing.cpu_count()
2018-07-16 06:52:18 +08:00
gradlew = os.path.join('.', 'gradlew.bat' if os.name == 'nt' else 'gradlew')
2018-06-10 16:55:00 +08:00
archs = ['armeabi-v7a', 'x86']
2018-08-06 02:01:04 +08:00
keystore = 'release-key.jks'
2018-08-06 19:32:37 +08:00
config = {}
2018-05-13 03:04:40 +08:00
def mv(source, target):
2018-05-13 03:04:40 +08:00
try:
shutil.move(source, target)
except:
pass
def cp(source, target):
2018-05-13 03:04:40 +08:00
try:
shutil.copyfile(source, target)
vprint(f'cp: {source} -> {target}')
2018-05-13 03:04:40 +08:00
except:
pass
def rm(file):
2017-06-03 20:19:01 +08:00
try:
2017-12-04 18:05:07 +08:00
os.remove(file)
2017-06-03 20:19:01 +08:00
except OSError as e:
if e.errno != errno.ENOENT:
raise
def mkdir(path, mode=0o777):
try:
os.mkdir(path, mode)
except:
pass
def mkdir_p(path, mode=0o777):
os.makedirs(path, mode, exist_ok=True)
2017-06-03 20:19:01 +08:00
def zip_with_msg(zipfile, source, target):
if not os.path.exists(source):
error(f'{source} does not exist! Try build \'binary\' and \'apk\' before zipping!')
2017-06-03 20:19:01 +08:00
zipfile.write(source, target)
vprint(f'zip: {source} -> {target}')
2017-06-03 20:19:01 +08:00
2018-05-13 03:04:40 +08:00
def collect_binary():
2018-06-10 16:55:00 +08:00
for arch in archs:
2018-05-13 03:04:40 +08:00
mkdir_p(os.path.join('native', 'out', arch))
2019-01-30 03:10:12 -05:00
for bin in ['magisk', 'magiskinit', 'magiskboot', 'busybox']:
2018-05-13 03:04:40 +08:00
source = os.path.join('native', 'libs', arch, bin)
target = os.path.join('native', 'out', arch, bin)
mv(source, target)
2017-06-03 20:19:01 +08:00
2018-08-06 02:01:04 +08:00
def execv(cmd, redirect=None):
return subprocess.run(cmd, stdout=redirect if redirect != None else STDOUT)
def system(cmd, redirect=None):
return subprocess.run(cmd, shell=True, stdout=redirect if redirect != None else STDOUT)
def xz(data):
return lzma.compress(data, preset=9, check=lzma.CHECK_NONE)
2018-08-10 04:41:21 +08:00
def sign_zip(unsigned, output, release):
signer_name = 'zipsigner-3.0.jar'
2019-01-30 03:10:12 -05:00
zipsigner = os.path.join('signing', 'build', 'libs', signer_name)
2018-08-10 04:41:21 +08:00
if not os.path.exists(zipsigner):
header('* Building ' + signer_name)
2019-01-30 03:10:12 -05:00
proc = execv([gradlew, 'signing:shadowJar'])
2018-08-10 04:41:21 +08:00
if proc.returncode != 0:
error(f'Build {signer_name} failed!')
2018-08-10 04:41:21 +08:00
header('* Signing Zip')
if release:
proc = execv(['java', '-jar', zipsigner, keystore, config['keyStorePass'], config['keyAlias'], config['keyPass'], unsigned, output])
else:
proc = execv(['java', '-jar', zipsigner, unsigned, output])
if proc.returncode != 0:
error('Signing zip failed!')
def binary_dump(src, out, var_name):
out.write(f'const static unsigned char {var_name}[] = {{')
for i, c in enumerate(xz(src.read())):
2018-08-10 04:41:21 +08:00
if i % 16 == 0:
out.write('\n')
out.write(f'0x{c:02X},')
2018-08-10 04:41:21 +08:00
out.write('\n};\n')
out.flush()
def gen_update_binary():
bs = 1024
update_bin = bytearray(bs)
file = os.path.join('native', 'out', 'x86', 'busybox')
with open(file, 'rb') as f:
x86_bb = f.read()
file = os.path.join('native', 'out', 'armeabi-v7a', 'busybox')
with open(file, 'rb') as f:
arm_bb = f.read()
file = os.path.join('scripts', 'update_binary.sh')
with open(file, 'rb') as f:
script = f.read()
# Align x86 busybox to bs
blkCnt = (len(x86_bb) - 1) // bs + 1
script = script.replace(b'__X86_CNT__', b'%d' % blkCnt)
update_bin[:len(script)] = script
update_bin.extend(x86_bb)
# Padding for alignment
update_bin.extend(b'\0' * (blkCnt * bs - len(x86_bb)))
update_bin.extend(arm_bb)
return update_bin
2018-08-10 04:41:21 +08:00
2018-05-13 03:04:40 +08:00
def build_binary(args):
support_targets = {'magisk', 'magiskinit', 'magiskboot', 'busybox'}
2018-08-10 05:54:26 +08:00
if len(args.target) == 0:
# If nothing specified, build everything
args.target = support_targets
else:
args.target = set(args.target) & support_targets
2018-05-19 16:53:00 +08:00
2018-08-10 05:54:26 +08:00
if len(args.target) == 0:
return
2018-08-10 05:54:26 +08:00
header('* Building binaries: ' + ' '.join(args.target))
2017-11-23 23:55:33 +08:00
2018-09-27 03:56:56 -04:00
os.utime(os.path.join('native', 'jni', 'include', 'flags.h'))
2018-05-13 03:04:40 +08:00
# Basic flags
base_flags = f'MAGISK_VERSION="{config["version"]}" MAGISK_VER_CODE={config["versionCode"]}'
2018-09-28 02:05:55 -04:00
if not args.release:
base_flags += ' MAGISK_DEBUG=1'
2017-06-03 20:19:01 +08:00
# Magisk is special case as it is a dependency of magiskinit
2018-08-10 05:54:26 +08:00
if 'magisk' in args.target:
proc = system(f'{ndk_build} -C native {base_flags} B_MAGISK=1 -j{cpu_count}')
2018-05-13 03:04:40 +08:00
if proc.returncode != 0:
error('Build Magisk binary failed!')
collect_binary()
# Dump the binary to header
for arch in archs:
bin_file = os.path.join('native', 'out', arch, 'magisk')
2018-08-09 18:13:07 +08:00
with open(os.path.join('native', 'out', arch, 'binaries_arch.h'), 'w') as out:
with open(bin_file, 'rb') as src:
binary_dump(src, out, 'magisk_xz')
2018-05-13 03:04:40 +08:00
# BusyBox is special case as it needs special flags to build
if 'busybox' in args.target:
proc = system(f'{ndk_build} -C native {base_flags} B_BB=1 -j{cpu_count}')
if proc.returncode != 0:
error('Build binaries failed!')
collect_binary()
build = False
flags = base_flags
2018-08-10 05:54:26 +08:00
if 'magiskinit' in args.target:
2018-08-09 18:13:07 +08:00
if not os.path.exists(os.path.join('native', 'out', 'x86', 'binaries_arch.h')):
error('Build "magisk" before building "magiskinit"')
2018-08-09 18:13:07 +08:00
if not os.path.exists(os.path.join('native', 'out', 'binaries.h')):
2018-08-10 04:41:21 +08:00
error('Build stub APK before building "magiskinit"')
flags += ' B_INIT=1'
build = True
2018-05-13 03:04:40 +08:00
2018-08-10 05:54:26 +08:00
if 'magiskboot' in args.target:
flags += ' B_BOOT=1'
build = True
if build:
proc = system(f'{ndk_build} -C native {flags} -j{cpu_count}')
2018-05-13 03:04:40 +08:00
if proc.returncode != 0:
error('Build binaries failed!')
collect_binary()
2018-08-20 12:02:38 +08:00
def build_apk(args, flavor):
header('* Building {} Magisk Manager'.format(flavor))
2017-06-03 20:19:01 +08:00
2018-08-20 12:02:38 +08:00
buildType = 'Release' if args.release else 'Debug'
proc = execv([gradlew, f'app:assemble{flavor}{buildType}', '-PconfigPath=' + os.path.abspath(args.config)])
2018-08-20 12:02:38 +08:00
if proc.returncode != 0:
error('Build Magisk Manager failed!')
2017-06-03 22:04:22 +08:00
2018-08-20 12:02:38 +08:00
flavor = flavor.lower()
buildType = buildType.lower()
apk = f'app-{flavor}-{buildType}.apk'
2018-08-10 05:54:26 +08:00
2018-08-20 12:02:38 +08:00
source = os.path.join('app', 'build', 'outputs', 'apk', flavor, buildType, apk)
target = os.path.join(config['outdir'], apk)
mv(source, target)
2018-08-10 05:54:26 +08:00
header('Output: ' + target)
2018-08-20 12:02:38 +08:00
return target
2018-08-20 12:02:38 +08:00
def build_app(args):
source = os.path.join('scripts', 'util_functions.sh')
2019-01-30 03:10:12 -05:00
target = os.path.join('app-core', 'src', 'main', 'res', 'raw', 'util_functions.sh')
2018-08-20 12:02:38 +08:00
cp(source, target)
build_apk(args, 'Full')
2018-05-27 14:55:24 +08:00
2018-08-20 12:02:38 +08:00
def build_stub(args):
stub = build_apk(args, 'Stub')
2018-08-10 05:15:39 +08:00
# Dump the stub APK to header
mkdir(os.path.join('native', 'out'))
with open(os.path.join('native', 'out', 'binaries.h'), 'w') as out:
2018-08-20 12:02:38 +08:00
with open(stub, 'rb') as src:
2018-08-10 05:15:39 +08:00
binary_dump(src, out, 'manager_xz');
2017-10-07 22:48:16 +08:00
def build_snet(args):
2018-08-06 02:01:04 +08:00
proc = execv([gradlew, 'snet:assembleRelease'])
2017-10-07 22:48:16 +08:00
if proc.returncode != 0:
error('Build snet extention failed!')
source = os.path.join('snet', 'build', 'outputs', 'apk', 'release', 'snet-release-unsigned.apk')
2018-05-13 03:04:40 +08:00
target = os.path.join(config['outdir'], 'snet.apk')
# Re-compress the whole APK for smaller size
with zipfile.ZipFile(target, 'w', compression=zipfile.ZIP_DEFLATED, allowZip64=False) as zout:
with zipfile.ZipFile(source) as zin:
for item in zin.infolist():
zout.writestr(item.filename, zin.read(item))
2018-06-10 16:55:00 +08:00
rm(source)
2018-05-13 03:04:40 +08:00
header('Output: ' + target)
2017-06-03 20:19:01 +08:00
def zip_main(args):
header('* Packing Flashable Zip')
2017-11-15 05:25:19 +08:00
unsigned = tempfile.mkstemp()[1]
with zipfile.ZipFile(unsigned, 'w', compression=zipfile.ZIP_DEFLATED, allowZip64=False) as zipf:
# META-INF
# update-binary
target = os.path.join('META-INF', 'com', 'google', 'android', 'update-binary')
2018-08-06 02:01:04 +08:00
vprint('zip: ' + target)
zipf.writestr(target, gen_update_binary())
# updater-script
source = os.path.join('scripts', 'flash_script.sh')
target = os.path.join('META-INF', 'com', 'google', 'android', 'updater-script')
zip_with_msg(zipf, source, target)
# Binaries
2018-04-22 14:13:27 +08:00
for lib_dir, zip_dir in [('armeabi-v7a', 'arm'), ('x86', 'x86')]:
2017-11-10 00:54:54 +08:00
for binary in ['magiskinit', 'magiskboot']:
2018-05-13 03:04:40 +08:00
source = os.path.join('native', 'out', lib_dir, binary)
2017-06-03 20:19:01 +08:00
target = os.path.join(zip_dir, binary)
zip_with_msg(zipf, source, target)
# APK
2018-08-21 00:31:41 +08:00
source = os.path.join(config['outdir'], 'app-full-release.apk' if args.release else 'app-full-debug.apk')
2017-06-03 20:19:01 +08:00
target = os.path.join('common', 'magisk.apk')
zip_with_msg(zipf, source, target)
# Scripts
2017-06-19 00:15:44 +08:00
# boot_patch.sh
2017-06-03 20:19:01 +08:00
source = os.path.join('scripts', 'boot_patch.sh')
target = os.path.join('common', 'boot_patch.sh')
zip_with_msg(zipf, source, target)
2017-06-19 00:15:44 +08:00
# util_functions.sh
source = os.path.join('scripts', 'util_functions.sh')
2017-07-11 01:54:11 +08:00
with open(source, 'r') as script:
# Add version info util_functions.sh
2018-05-13 03:04:40 +08:00
util_func = script.read().replace('#MAGISK_VERSION_STUB',
f'MAGISK_VER="{config["version"]}"\nMAGISK_VER_CODE={config["versionCode"]}')
2017-07-11 01:54:11 +08:00
target = os.path.join('common', 'util_functions.sh')
vprint(f'zip: {source} -> {target}')
2017-07-11 01:54:11 +08:00
zipf.writestr(target, util_func)
# addon.d.sh
source = os.path.join('scripts', 'addon.d.sh')
2018-06-21 11:54:21 +08:00
target = os.path.join('common', 'addon.d.sh')
zip_with_msg(zipf, source, target)
2017-06-03 20:19:01 +08:00
# Prebuilts
for chromeos in ['futility', 'kernel_data_key.vbprivk', 'kernel.keyblock']:
source = os.path.join('chromeos', chromeos)
zip_with_msg(zipf, source, source)
# End of zipping
output = os.path.join(config['outdir'], f'Magisk-v{config["version"]}.zip' if config['prettyName'] else
2018-05-13 03:04:40 +08:00
'magisk-release.zip' if args.release else 'magisk-debug.zip')
sign_zip(unsigned, output, args.release)
2018-05-13 03:04:40 +08:00
header('Output: ' + output)
2017-06-03 20:19:01 +08:00
def zip_uninstaller(args):
header('* Packing Uninstaller Zip')
2017-11-15 05:25:19 +08:00
unsigned = tempfile.mkstemp()[1]
with zipfile.ZipFile(unsigned, 'w', compression=zipfile.ZIP_DEFLATED, allowZip64=False) as zipf:
# META-INF
# update-binary
target = os.path.join('META-INF', 'com', 'google', 'android', 'update-binary')
2018-08-06 02:01:04 +08:00
vprint('zip: ' + target)
zipf.writestr(target, gen_update_binary())
# updater-script
2018-06-27 06:00:01 +08:00
source = os.path.join('scripts', 'magisk_uninstaller.sh')
target = os.path.join('META-INF', 'com', 'google', 'android', 'updater-script')
zip_with_msg(zipf, source, target)
# Binaries
2018-04-22 14:13:27 +08:00
for lib_dir, zip_dir in [('armeabi-v7a', 'arm'), ('x86', 'x86')]:
2018-06-27 06:00:01 +08:00
for bin in ['magisk', 'magiskboot']:
source = os.path.join('native', 'out', lib_dir, bin)
target = os.path.join(zip_dir, bin)
zip_with_msg(zipf, source, target)
2017-06-04 00:03:36 +08:00
# Scripts
2017-07-11 01:54:11 +08:00
# util_functions.sh
2017-07-10 00:17:34 +08:00
source = os.path.join('scripts', 'util_functions.sh')
2017-07-11 01:54:11 +08:00
with open(source, 'r') as script:
target = os.path.join('util_functions.sh')
vprint(f'zip: {source} -> {target}')
2017-12-21 03:36:18 +08:00
zipf.writestr(target, script.read())
2017-07-10 00:17:34 +08:00
# Prebuilts
for chromeos in ['futility', 'kernel_data_key.vbprivk', 'kernel.keyblock']:
source = os.path.join('chromeos', chromeos)
zip_with_msg(zipf, source, source)
# End of zipping
output = os.path.join(config['outdir'], f'Magisk-uninstaller-{datetime.datetime.now().strftime("%Y%m%d")}.zip'
2018-05-13 03:04:40 +08:00
if config['prettyName'] else 'magisk-uninstaller.zip')
sign_zip(unsigned, output, args.release)
2018-05-13 03:04:40 +08:00
header('Output: ' + output)
2017-06-03 20:19:01 +08:00
def cleanup(args):
2018-08-10 05:54:26 +08:00
support_targets = {'native', 'java'}
2017-06-03 20:19:01 +08:00
if len(args.target) == 0:
2018-08-10 05:54:26 +08:00
# If nothing specified, clean everything
args.target = support_targets
else:
args.target = set(args.target) & support_targets
2017-06-03 20:19:01 +08:00
2018-07-08 00:02:18 +08:00
if 'native' in args.target:
header('* Cleaning native')
2019-01-30 03:35:07 -05:00
system(ndk_build + ' -C native B_MAGISK=1 B_INIT=1 B_BOOT=1 B_BB=1 clean')
2018-05-13 03:04:40 +08:00
shutil.rmtree(os.path.join('native', 'out'), ignore_errors=True)
2017-06-03 20:19:01 +08:00
2017-10-07 22:48:16 +08:00
if 'java' in args.target:
header('* Cleaning java')
2019-01-30 03:10:12 -05:00
execv([gradlew, 'app:clean', 'app-core:clean', 'snet:clean', 'signing:clean'])
2018-05-13 03:04:40 +08:00
2018-08-10 05:54:26 +08:00
def build_all(args):
vars(args)['target'] = []
build_stub(args)
2018-08-21 00:31:41 +08:00
build_app(args)
2018-08-10 05:54:26 +08:00
build_binary(args)
zip_main(args)
zip_uninstaller(args)
build_snet(args)
2017-06-03 20:19:01 +08:00
parser = argparse.ArgumentParser(description='Magisk build script')
parser.add_argument('-r', '--release', action='store_true', help='compile Magisk for release')
parser.add_argument('-v', '--verbose', action='store_true', help='verbose output')
2018-09-05 14:24:28 -04:00
parser.add_argument('-c', '--config', default='config.prop', help='config file location')
2017-06-03 20:19:01 +08:00
subparsers = parser.add_subparsers(title='actions')
2018-05-13 03:04:40 +08:00
all_parser = subparsers.add_parser('all', help='build everything (binaries/apks/zips)')
2017-06-03 20:19:01 +08:00
all_parser.set_defaults(func=build_all)
2018-08-10 05:54:26 +08:00
binary_parser = subparsers.add_parser('binary', help='build binaries')
binary_parser.add_argument('target', nargs='*', help='Support: magisk, magiskinit, magiskboot, busybox. Leave empty to build all.')
2017-06-03 20:19:01 +08:00
binary_parser.set_defaults(func=build_binary)
apk_parser = subparsers.add_parser('apk', help='build Magisk Manager APK')
2018-08-20 12:02:38 +08:00
apk_parser.set_defaults(func=build_app)
2017-06-03 20:19:01 +08:00
2018-08-10 05:15:39 +08:00
stub_parser = subparsers.add_parser('stub', help='build stub Magisk Manager APK')
stub_parser.set_defaults(func=build_stub)
2017-10-07 22:48:16 +08:00
snet_parser = subparsers.add_parser('snet', help='build snet extention for Magisk Manager')
snet_parser.set_defaults(func=build_snet)
zip_parser = subparsers.add_parser('zip', help='zip Magisk into a flashable zip')
2017-06-03 20:19:01 +08:00
zip_parser.set_defaults(func=zip_main)
2018-08-10 05:54:26 +08:00
un_parser = subparsers.add_parser('uninstaller', help='create flashable uninstaller')
un_parser.set_defaults(func=zip_uninstaller)
2017-06-03 20:19:01 +08:00
2018-08-10 05:54:26 +08:00
clean_parser = subparsers.add_parser('clean', help='cleanup.')
clean_parser.add_argument('target', nargs='*', help='Support: native, java. Leave empty to clean all.')
2017-06-03 20:19:01 +08:00
clean_parser.set_defaults(func=cleanup)
if len(sys.argv) == 1:
2017-12-04 18:05:07 +08:00
parser.print_help()
sys.exit(1)
2017-06-03 20:19:01 +08:00
args = parser.parse_args()
2018-09-05 14:24:28 -04:00
# Some default values
config['outdir'] = 'out'
config['prettyName'] = 'false'
with open(args.config, 'r') as f:
for line in [l.strip(' \t\r\n') for l in f]:
if line.startswith('#') or len(line) == 0:
continue
prop = line.split('=')
config[prop[0].strip(' \t\r\n')] = prop[1].strip(' \t\r\n')
if 'version' not in config or 'versionCode' not in config:
error('"version" and "versionCode" is required in "config.prop"')
try:
config['versionCode'] = int(config['versionCode'])
except ValueError:
error('"versionCode" is required to be an integer')
config['prettyName'] = config['prettyName'].lower() == 'true'
mkdir_p(config['outdir'])
2018-08-06 02:01:04 +08:00
if args.release and not os.path.exists(keystore):
error(f'Please generate a java keystore and place it in "{keystore}"')
STDOUT = None if args.verbose else subprocess.DEVNULL
2017-06-03 20:19:01 +08:00
args.func(args)