| #!/usr/bin/env python3 |
| # Copyright (c) 2018, the R8 project authors. Please see the AUTHORS file |
| # for details. All rights reserved. Use of this source code is governed by a |
| # BSD-style license that can be found in the LICENSE file. |
| |
| import optparse |
| import os |
| import shutil |
| import subprocess |
| import sys |
| import time |
| |
| import utils |
| import zip_utils |
| |
| USAGE = 'usage: %prog [options] <apk>' |
| |
| def parse_options(): |
| parser = optparse.OptionParser(usage=USAGE) |
| parser.add_option('--keystore', |
| help='keystore file (default ~/.android/app.keystore)', |
| default='~/.android/app.keystore') |
| parser.add_option('--sign', |
| help='Sign the passed in apk.', |
| default=False, |
| action='store_true') |
| parser.add_option('--use_apksigner', |
| help='Use apksigner to sign.', |
| default=False, |
| action='store_true') |
| parser.add_option('--output', |
| help='Where to put the signed apk.)', |
| default=None) |
| |
| (options, args) = parser.parse_args() |
| if len(args) != 1: |
| parser.error('Expected <apk> argument, got: ' + ' '.join(args)) |
| apk = args[0] |
| return (options, apk) |
| |
| def add_baseline_profile_to_apk( |
| apk, baseline_profile, baseline_profile_metadata, tmp_dir): |
| if baseline_profile is None: |
| return apk |
| ts = time.time_ns() |
| dest_apk = os.path.join(tmp_dir, 'app-%s.apk' % ts) |
| dest_apk_aligned = os.path.join(tmp_dir, 'app-aligned-%s.apk' % ts) |
| dest_apk_signed = os.path.join(tmp_dir, 'app-signed-%s.apk' % ts) |
| shutil.copy2(apk, dest_apk) |
| zip_utils.remove_files_from_zip( |
| ['assets/dexopt/baseline.prof', 'assets/dexopt/baseline.profm'], dest_apk) |
| zip_utils.add_file_to_zip( |
| baseline_profile, 'assets/dexopt/baseline.prof', dest_apk) |
| if baseline_profile_metadata is not None: |
| zip_utils.add_file_to_zip( |
| baseline_profile_metadata, 'assets/dexopt/baseline.profm', dest_apk) |
| align(dest_apk, dest_apk_aligned) |
| sign_with_apksigner(dest_apk_aligned, dest_apk_signed) |
| return dest_apk_signed |
| |
| def align(apk, aligned_apk): |
| zipalign_path = ( |
| 'zipalign' if 'build_tools' in os.environ.get('PATH') |
| else os.path.join(utils.getAndroidBuildTools(), 'zipalign')) |
| cmd = [zipalign_path, '-f', '-p', '4', apk, aligned_apk] |
| utils.RunCmd(cmd, quiet=True, logging=False) |
| return aligned_apk |
| |
| def default_keystore(): |
| return os.path.join(os.getenv('HOME'), '.android', 'app.keystore') |
| |
| def get_min_api(apk): |
| aapt = os.path.join(utils.getAndroidBuildTools(), 'aapt') |
| cmd = [aapt, 'dump', 'badging', apk] |
| stdout = subprocess.check_output(cmd).decode('utf-8').strip() |
| for line in stdout.splitlines(): |
| if line.startswith('sdkVersion:\''): |
| return int(line[len('sdkVersion:\''): -1]) |
| raise ValueError('Unexpected stdout: %s' % stdout) |
| |
| def sign(unsigned_apk, signed_apk, keystore, quiet=False, logging=True): |
| utils.Print('Signing (ignore the warnings)', quiet=quiet) |
| cmd = ['zip', '-d', unsigned_apk, 'META-INF/*'] |
| utils.RunCmd(cmd, quiet=quiet, logging=logging, fail=False) |
| cmd = [ |
| 'jarsigner', |
| '-sigalg', 'SHA1withRSA', |
| '-digestalg', 'SHA1', |
| '-keystore', keystore, |
| '-storepass', 'android', |
| '-signedjar', signed_apk, |
| unsigned_apk, |
| 'androiddebugkey' |
| ] |
| utils.RunCmd(cmd, quiet=quiet) |
| |
| def sign_with_apksigner( |
| unsigned_apk, signed_apk, keystore=None, password='android', quiet=False, |
| logging=True): |
| cmd = [ |
| os.path.join(utils.getAndroidBuildTools(), 'apksigner'), |
| 'sign', |
| '-v', |
| '--ks', keystore or default_keystore(), |
| '--ks-pass', 'pass:' + password, |
| '--min-sdk-version', '19', |
| '--out', signed_apk, |
| '--v2-signing-enabled', |
| unsigned_apk |
| ] |
| utils.RunCmd(cmd, quiet=quiet, logging=logging) |
| return signed_apk |
| |
| def main(): |
| (options, apk) = parse_options() |
| if options.sign: |
| if not options.output: |
| print('When signing you must specify an output apk') |
| return 1 |
| if not options.keystore: |
| print('When signing you must specify a keystore') |
| return 1 |
| if options.use_apksigner: |
| sign_with_apksigner(apk, options.output, options.keystore) |
| else: |
| sign(apk, options.output, options.keystore) |
| return 0 |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |