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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
#!/usr/bin/env python
#
# Copyright (c) 2015 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Adds the code parts to a resource APK."""
import argparse
import itertools
import os
import shutil
import sys
import zipfile
from util import build_utils
# Taken from aapt's Package.cpp:
_NO_COMPRESS_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.gif', '.wav', '.mp2',
'.mp3', '.ogg', '.aac', '.mpg', '.mpeg', '.mid',
'.midi', '.smf', '.jet', '.rtttl', '.imy', '.xmf',
'.mp4', '.m4a', '.m4v', '.3gp', '.3gpp', '.3g2',
'.3gpp2', '.amr', '.awb', '.wma', '.wmv')
def _ParseArgs(args):
parser = argparse.ArgumentParser()
build_utils.AddDepfileOption(parser)
parser.add_argument('--assets',
help='GYP-list of files to add as assets in the form '
'"srcPath:zipPath", where ":zipPath" is optional.',
default='[]')
parser.add_argument('--uncompressed-assets',
help='Same as --assets, except disables compression.',
default='[]')
parser.add_argument('--resource-apk',
help='An .ap_ file built using aapt',
required=True)
parser.add_argument('--output-apk',
help='Path to the output file',
required=True)
parser.add_argument('--dex-file',
help='Path to the classes.dex to use')
# TODO(agrieve): Switch this to be a list of files rather than a directory.
parser.add_argument('--native-libs-dir',
help='Directory containing native libraries to include',
default=[])
parser.add_argument('--android-abi',
help='Android architecture to use for native libraries')
parser.add_argument('--create-placeholder-lib',
action='store_true',
help='Whether to add a dummy library file')
options = parser.parse_args(args)
if not options.android_abi and (options.native_libs_dir or
options.create_placeholder_lib):
raise Exception('Must specify --android-abi with --native-libs-dir')
options.assets = build_utils.ParseGypList(options.assets)
options.uncompressed_assets = build_utils.ParseGypList(
options.uncompressed_assets)
return options
def _ListSubPaths(path):
"""Returns a list of full paths to all files in the given path."""
return [os.path.join(path, name) for name in os.listdir(path)]
def _SplitAssetPath(path):
"""Returns (src, dest) given an asset path in the form src[:dest]."""
path_parts = path.split(':')
src_path = path_parts[0]
if len(path_parts) > 1:
dest_path = path_parts[1]
else:
dest_path = os.path.basename(src_path)
return src_path, dest_path
def _AddAssets(apk, paths, disable_compression=False):
"""Adds the given paths to the apk.
Args:
apk: ZipFile to write to.
paths: List of paths (with optional :zipPath suffix) to add.
disable_compression: Whether to disable compression.
"""
# Group all uncompressed assets together in the hope that it will increase
# locality of mmap'ed files.
for target_compress_type in (zipfile.ZIP_STORED, zipfile.ZIP_DEFLATED):
for path in paths:
src_path, dest_path = _SplitAssetPath(path)
compress_type = zipfile.ZIP_DEFLATED
if disable_compression or (
os.path.splitext(src_path)[1] in _NO_COMPRESS_EXTENSIONS):
compress_type = zipfile.ZIP_STORED
if target_compress_type == compress_type:
apk_path = 'assets/' + dest_path
try:
apk.getinfo(apk_path)
# Should never happen since write_build_config.py handles merging.
raise Exception('Multiple targets specified the asset path: %s' %
apk_path)
except KeyError:
apk.write(src_path, apk_path, compress_type)
def main(args):
args = build_utils.ExpandFileArgs(args)
options = _ParseArgs(args)
native_libs = []
if options.native_libs_dir:
native_libs = _ListSubPaths(options.native_libs_dir)
input_paths = [options.resource_apk, __file__] + native_libs
if options.dex_file:
input_paths.append(options.dex_file)
input_strings = [options.create_placeholder_lib, options.android_abi]
for path in itertools.chain(options.assets, options.uncompressed_assets):
src_path, dest_path = _SplitAssetPath(path)
input_paths.append(src_path)
input_strings.append(dest_path)
def on_stale_md5():
tmp_apk = options.output_apk + '.tmp'
try:
# Use a temp file to avoid creating an output if anything goes wrong.
shutil.copyfile(options.resource_apk, tmp_apk)
# TODO(agrieve): It would be more efficient to combine this step
# with finalize_apk(), which sometimes aligns and uncompresses the
# native libraries.
with zipfile.ZipFile(tmp_apk, 'a', zipfile.ZIP_DEFLATED) as apk:
_AddAssets(apk, options.assets, disable_compression=False)
_AddAssets(apk, options.uncompressed_assets, disable_compression=True)
for path in native_libs:
basename = os.path.basename(path)
apk.write(path, 'lib/%s/%s' % (options.android_abi, basename))
if options.create_placeholder_lib:
# Make it non-empty so that its checksum is non-zero and is not
# ignored by md5_check.
apk.writestr('lib/%s/libplaceholder.so' % options.android_abi, ':-)')
if options.dex_file:
apk.write(options.dex_file, 'classes.dex')
shutil.move(tmp_apk, options.output_apk)
finally:
if os.path.exists(tmp_apk):
os.unlink(tmp_apk)
build_utils.CallAndWriteDepfileIfStale(
on_stale_md5,
options,
input_paths=input_paths,
input_strings=input_strings,
output_paths=[options.output_apk])
if __name__ == '__main__':
main(sys.argv[1:])
|
