blob: baee9c98ecd7cecbab437c1843077ee658d81aa3 [file] [log] [blame]
Adam Lesinski1ab598f2015-08-14 14:26:04 -07001/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Adam Lesinskice5e56e2016-10-21 17:56:45 -070017#include <sys/stat.h>
18
19#include <fstream>
20#include <queue>
21#include <unordered_map>
22#include <vector>
23
24#include "android-base/errors.h"
25#include "android-base/file.h"
Adam Lesinskif34b6f42017-03-03 16:33:26 -080026#include "android-base/stringprintf.h"
Adam Lesinskid5083f62017-01-16 15:07:21 -080027#include "androidfw/StringPiece.h"
Adam Lesinskice5e56e2016-10-21 17:56:45 -070028
Adam Lesinski1ab598f2015-08-14 14:26:04 -070029#include "AppInfo.h"
30#include "Debug.h"
31#include "Flags.h"
Adam Lesinski6a008172016-02-02 17:02:58 -080032#include "Locale.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070033#include "NameMangler.h"
Adam Lesinski59e04c62016-02-04 15:59:23 -080034#include "ResourceUtils.h"
Adam Lesinskid3ffa8442017-09-28 13:34:35 -070035#include "ResourceValues.h"
36#include "ValueVisitor.h"
Adam Lesinskid0f492d2017-04-03 18:12:45 -070037#include "cmd/Util.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070038#include "compile/IdAssigner.h"
Adam Lesinski6a008172016-02-02 17:02:58 -080039#include "filter/ConfigFilter.h"
Adam Lesinski46708052017-09-29 14:49:15 -070040#include "format/Archive.h"
Adam Lesinski00451162017-10-03 07:44:08 -070041#include "format/Container.h"
Adam Lesinski46708052017-09-29 14:49:15 -070042#include "format/binary/BinaryResourceParser.h"
43#include "format/binary/TableFlattener.h"
44#include "format/binary/XmlFlattener.h"
45#include "format/proto/ProtoDeserialize.h"
46#include "format/proto/ProtoSerialize.h"
Adam Lesinski00451162017-10-03 07:44:08 -070047#include "io/BigBufferStream.h"
48#include "io/FileStream.h"
Adam Lesinskia40e9722015-11-24 19:11:46 -080049#include "io/FileSystem.h"
Adam Lesinskid0f492d2017-04-03 18:12:45 -070050#include "io/Util.h"
Adam Lesinskia40e9722015-11-24 19:11:46 -080051#include "io/ZipArchive.h"
Adam Lesinskica5638f2015-10-21 14:42:43 -070052#include "java/JavaClassGenerator.h"
53#include "java/ManifestClassGenerator.h"
54#include "java/ProguardRules.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070055#include "link/Linkers.h"
Adam Lesinskicacb28f2016-10-19 12:18:14 -070056#include "link/ManifestFixer.h"
Adam Lesinski467f1712015-11-16 17:35:44 -080057#include "link/ReferenceLinker.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070058#include "link/TableMerger.h"
Adam Lesinskic744ae82017-05-17 19:28:38 -070059#include "link/XmlCompatVersioner.h"
Adam Lesinskid48944a2017-02-21 14:22:30 -080060#include "optimize/ResourceDeduper.h"
61#include "optimize/VersionCollapser.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070062#include "process/IResourceTableConsumer.h"
63#include "process/SymbolTable.h"
Adam Lesinski355f2852016-02-13 20:26:45 -080064#include "split/TableSplitter.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070065#include "util/Files.h"
Adam Lesinski467f1712015-11-16 17:35:44 -080066#include "xml/XmlDom.h"
Adam Lesinski1ab598f2015-08-14 14:26:04 -070067
Adam Lesinskiefeb7af2017-08-02 14:57:43 -070068using ::aapt::io::FileInputStream;
69using ::android::StringPiece;
70using ::android::base::StringPrintf;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -070071
Adam Lesinski1ab598f2015-08-14 14:26:04 -070072namespace aapt {
73
74struct LinkOptions {
Adam Lesinskice5e56e2016-10-21 17:56:45 -070075 std::string output_path;
76 std::string manifest_path;
77 std::vector<std::string> include_paths;
78 std::vector<std::string> overlay_files;
Adam Lesinskib39ad7c2017-03-13 11:40:48 -070079 std::vector<std::string> assets_dirs;
Adam Lesinskice5e56e2016-10-21 17:56:45 -070080 bool output_to_directory = false;
81 bool auto_add_overlay = false;
Adam Lesinski36c73a52016-08-11 13:39:24 -070082
Adam Lesinskicacb28f2016-10-19 12:18:14 -070083 // Java/Proguard options.
Adam Lesinskice5e56e2016-10-21 17:56:45 -070084 Maybe<std::string> generate_java_class_path;
85 Maybe<std::string> custom_java_package;
86 std::set<std::string> extra_java_packages;
Adam Lesinski418763f2017-04-11 17:36:53 -070087 Maybe<std::string> generate_text_symbols_path;
Adam Lesinskice5e56e2016-10-21 17:56:45 -070088 Maybe<std::string> generate_proguard_rules_path;
89 Maybe<std::string> generate_main_dex_proguard_rules_path;
90 bool generate_non_final_ids = false;
91 std::vector<std::string> javadoc_annotations;
92 Maybe<std::string> private_symbols;
Adam Lesinski36c73a52016-08-11 13:39:24 -070093
Adam Lesinskice5e56e2016-10-21 17:56:45 -070094 // Optimizations/features.
95 bool no_auto_version = false;
96 bool no_version_vectors = false;
Yuichi Araki4d35cca2017-01-18 20:42:17 +090097 bool no_version_transitions = false;
Adam Lesinskice5e56e2016-10-21 17:56:45 -070098 bool no_resource_deduping = false;
99 bool no_xml_namespaces = false;
100 bool do_not_compress_anything = false;
101 std::unordered_set<std::string> extensions_to_not_compress;
102
103 // Static lib options.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700104 bool no_static_lib_packages = false;
105
106 // AndroidManifest.xml massaging options.
107 ManifestFixerOptions manifest_fixer_options;
108
109 // Products to use/filter on.
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700110 std::unordered_set<std::string> products;
Adam Lesinski36c73a52016-08-11 13:39:24 -0700111
Adam Lesinskic8f71aa2017-02-08 07:03:50 -0800112 // Flattening options.
113 TableFlattenerOptions table_flattener_options;
114
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700115 // Split APK options.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700116 TableSplitterOptions table_splitter_options;
117 std::vector<SplitConstraints> split_constraints;
118 std::vector<std::string> split_paths;
Adam Lesinski36c73a52016-08-11 13:39:24 -0700119
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700120 // Stable ID options.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700121 std::unordered_map<ResourceName, ResourceId> stable_id_map;
122 Maybe<std::string> resource_id_map_path;
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700123};
124
Adam Lesinski64587af2016-02-18 18:33:06 -0800125class LinkContext : public IAaptContext {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700126 public:
Chris Warrington820d72a2017-04-27 15:27:01 +0100127 LinkContext(IDiagnostics* diagnostics)
128 : diagnostics_(diagnostics), name_mangler_({}), symbols_(&name_mangler_) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700129 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700130
Adam Lesinskib522f042017-04-21 16:57:59 -0700131 PackageType GetPackageType() override {
132 return package_type_;
133 }
134
135 void SetPackageType(PackageType type) {
136 package_type_ = type;
137 }
138
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700139 IDiagnostics* GetDiagnostics() override {
Chris Warrington820d72a2017-04-27 15:27:01 +0100140 return diagnostics_;
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700141 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700142
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700143 NameMangler* GetNameMangler() override {
144 return &name_mangler_;
145 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700146
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700147 void SetNameManglerPolicy(const NameManglerPolicy& policy) {
148 name_mangler_ = NameMangler(policy);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700149 }
Adam Lesinski64587af2016-02-18 18:33:06 -0800150
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700151 const std::string& GetCompilationPackage() override {
152 return compilation_package_;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700153 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700154
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700155 void SetCompilationPackage(const StringPiece& package_name) {
Adam Lesinskid5083f62017-01-16 15:07:21 -0800156 compilation_package_ = package_name.to_string();
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700157 }
Adam Lesinski64587af2016-02-18 18:33:06 -0800158
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700159 uint8_t GetPackageId() override {
160 return package_id_;
161 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700162
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700163 void SetPackageId(uint8_t id) {
164 package_id_ = id;
165 }
Adam Lesinski64587af2016-02-18 18:33:06 -0800166
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700167 SymbolTable* GetExternalSymbols() override {
168 return &symbols_;
169 }
Adam Lesinski355f2852016-02-13 20:26:45 -0800170
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700171 bool IsVerbose() override {
172 return verbose_;
173 }
Adam Lesinski64587af2016-02-18 18:33:06 -0800174
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700175 void SetVerbose(bool val) {
176 verbose_ = val;
177 }
Adam Lesinski64587af2016-02-18 18:33:06 -0800178
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700179 int GetMinSdkVersion() override {
180 return min_sdk_version_;
181 }
Adam Lesinskifb6312f2016-06-28 14:40:32 -0700182
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700183 void SetMinSdkVersion(int minSdk) {
184 min_sdk_version_ = minSdk;
185 }
Adam Lesinskifb6312f2016-06-28 14:40:32 -0700186
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700187 private:
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700188 DISALLOW_COPY_AND_ASSIGN(LinkContext);
189
Adam Lesinskib522f042017-04-21 16:57:59 -0700190 PackageType package_type_ = PackageType::kApp;
Chris Warrington820d72a2017-04-27 15:27:01 +0100191 IDiagnostics* diagnostics_;
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700192 NameMangler name_mangler_;
193 std::string compilation_package_;
194 uint8_t package_id_ = 0x0;
195 SymbolTable symbols_;
196 bool verbose_ = false;
197 int min_sdk_version_ = 0;
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700198};
199
Adam Lesinski1e4b0e52017-04-27 15:01:10 -0700200// A custom delegate that generates compatible pre-O IDs for use with feature splits.
201// Feature splits use package IDs > 7f, which in Java (since Java doesn't have unsigned ints)
202// is interpreted as a negative number. Some verification was wrongly assuming negative values
203// were invalid.
204//
205// This delegate will attempt to masquerade any '@id/' references with ID 0xPPTTEEEE,
206// where PP > 7f, as 0x7fPPEEEE. Any potential overlapping is verified and an error occurs if such
207// an overlap exists.
208class FeatureSplitSymbolTableDelegate : public DefaultSymbolTableDelegate {
209 public:
210 FeatureSplitSymbolTableDelegate(IAaptContext* context) : context_(context) {
211 }
212
213 virtual ~FeatureSplitSymbolTableDelegate() = default;
214
215 virtual std::unique_ptr<SymbolTable::Symbol> FindByName(
216 const ResourceName& name,
217 const std::vector<std::unique_ptr<ISymbolSource>>& sources) override {
218 std::unique_ptr<SymbolTable::Symbol> symbol =
219 DefaultSymbolTableDelegate::FindByName(name, sources);
220 if (symbol == nullptr) {
221 return {};
222 }
223
224 // Check to see if this is an 'id' with the target package.
225 if (name.type == ResourceType::kId && symbol->id) {
226 ResourceId* id = &symbol->id.value();
227 if (id->package_id() > kAppPackageId) {
228 // Rewrite the resource ID to be compatible pre-O.
229 ResourceId rewritten_id(kAppPackageId, id->package_id(), id->entry_id());
230
231 // Check that this doesn't overlap another resource.
232 if (DefaultSymbolTableDelegate::FindById(rewritten_id, sources) != nullptr) {
233 // The ID overlaps, so log a message (since this is a weird failure) and fail.
234 context_->GetDiagnostics()->Error(DiagMessage() << "Failed to rewrite " << name
235 << " for pre-O feature split support");
236 return {};
237 }
238
239 if (context_->IsVerbose()) {
240 context_->GetDiagnostics()->Note(DiagMessage() << "rewriting " << name << " (" << *id
241 << ") -> (" << rewritten_id << ")");
242 }
243
244 *id = rewritten_id;
245 }
246 }
247 return symbol;
248 }
249
250 private:
251 DISALLOW_COPY_AND_ASSIGN(FeatureSplitSymbolTableDelegate);
252
253 IAaptContext* context_;
254};
255
Adam Lesinskic744ae82017-05-17 19:28:38 -0700256static bool FlattenXml(IAaptContext* context, xml::XmlResource* xml_res, const StringPiece& path,
Adam Lesinskiea13c1a2017-10-13 12:40:37 -0700257 bool keep_raw_values, bool utf16, IArchiveWriter* writer) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700258 BigBuffer buffer(1024);
259 XmlFlattenerOptions options = {};
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700260 options.keep_raw_values = keep_raw_values;
Adam Lesinskiea13c1a2017-10-13 12:40:37 -0700261 options.use_utf16 = utf16;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700262 XmlFlattener flattener(&buffer, options);
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700263 if (!flattener.Consume(context, xml_res)) {
Adam Lesinski355f2852016-02-13 20:26:45 -0800264 return false;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700265 }
266
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700267 if (context->IsVerbose()) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700268 context->GetDiagnostics()->Note(DiagMessage(path) << "writing to archive (keep_raw_values="
269 << (keep_raw_values ? "true" : "false")
270 << ")");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700271 }
272
Adam Lesinski06460ef2017-03-14 18:52:13 -0700273 io::BigBufferInputStream input_stream(&buffer);
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700274 return io::CopyInputStreamToArchive(context, &input_stream, path.to_string(),
275 ArchiveEntry::kCompress, writer);
Adam Lesinski355f2852016-02-13 20:26:45 -0800276}
277
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700278static std::unique_ptr<ResourceTable> LoadTableFromPb(const Source& source, const void* data,
279 size_t len, IDiagnostics* diag) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700280 pb::ResourceTable pb_table;
281 if (!pb_table.ParseFromArray(data, len)) {
282 diag->Error(DiagMessage(source) << "invalid compiled table");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700283 return {};
284 }
Adam Lesinski355f2852016-02-13 20:26:45 -0800285
Adam Lesinski8cdca1b2017-09-28 15:50:03 -0700286 std::unique_ptr<ResourceTable> table = util::make_unique<ResourceTable>();
287 std::string error;
288 if (!DeserializeTableFromPb(pb_table, table.get(), &error)) {
289 diag->Error(DiagMessage(source) << "invalid compiled table: " << error);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700290 return {};
291 }
292 return table;
Adam Lesinski355f2852016-02-13 20:26:45 -0800293}
294
Adam Lesinskiefeb7af2017-08-02 14:57:43 -0700295// Inflates an XML file from the source path.
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700296static std::unique_ptr<xml::XmlResource> LoadXml(const std::string& path, IDiagnostics* diag) {
Adam Lesinskiefeb7af2017-08-02 14:57:43 -0700297 FileInputStream fin(path);
298 if (fin.HadError()) {
299 diag->Error(DiagMessage(path) << "failed to load XML file: " << fin.GetError());
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700300 return {};
301 }
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700302 return xml::Inflate(&fin, diag, Source(path));
Adam Lesinski355f2852016-02-13 20:26:45 -0800303}
304
Adam Lesinski355f2852016-02-13 20:26:45 -0800305struct ResourceFileFlattenerOptions {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700306 bool no_auto_version = false;
307 bool no_version_vectors = false;
Yuichi Araki4d35cca2017-01-18 20:42:17 +0900308 bool no_version_transitions = false;
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700309 bool no_xml_namespaces = false;
310 bool keep_raw_values = false;
311 bool do_not_compress_anything = false;
312 bool update_proguard_spec = false;
313 std::unordered_set<std::string> extensions_to_not_compress;
Adam Lesinski355f2852016-02-13 20:26:45 -0800314};
315
Adam Lesinskic744ae82017-05-17 19:28:38 -0700316// A sampling of public framework resource IDs.
317struct R {
318 struct attr {
319 enum : uint32_t {
320 paddingLeft = 0x010100d6u,
321 paddingRight = 0x010100d8u,
322 paddingHorizontal = 0x0101053du,
323
324 paddingTop = 0x010100d7u,
325 paddingBottom = 0x010100d9u,
326 paddingVertical = 0x0101053eu,
327
328 layout_marginLeft = 0x010100f7u,
329 layout_marginRight = 0x010100f9u,
330 layout_marginHorizontal = 0x0101053bu,
331
332 layout_marginTop = 0x010100f8u,
333 layout_marginBottom = 0x010100fau,
334 layout_marginVertical = 0x0101053cu,
335 };
336 };
337};
338
Adam Lesinski355f2852016-02-13 20:26:45 -0800339class ResourceFileFlattener {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700340 public:
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700341 ResourceFileFlattener(const ResourceFileFlattenerOptions& options, IAaptContext* context,
Adam Lesinskic744ae82017-05-17 19:28:38 -0700342 proguard::KeepSet* keep_set);
Adam Lesinski355f2852016-02-13 20:26:45 -0800343
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700344 bool Flatten(ResourceTable* table, IArchiveWriter* archive_writer);
Adam Lesinski355f2852016-02-13 20:26:45 -0800345
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700346 private:
347 struct FileOperation {
348 ConfigDescription config;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -0700349
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700350 // The entry this file came from.
Adam Lesinskibb94f322017-07-12 07:41:55 -0700351 ResourceEntry* entry;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -0700352
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700353 // The file to copy as-is.
Adam Lesinskibb94f322017-07-12 07:41:55 -0700354 io::IFile* file_to_copy;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -0700355
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700356 // The XML to process and flatten.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700357 std::unique_ptr<xml::XmlResource> xml_to_flatten;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -0700358
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700359 // The destination to write this file to.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700360 std::string dst_path;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700361 };
Adam Lesinski355f2852016-02-13 20:26:45 -0800362
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700363 uint32_t GetCompressionFlags(const StringPiece& str);
Adam Lesinski355f2852016-02-13 20:26:45 -0800364
Adam Lesinskic744ae82017-05-17 19:28:38 -0700365 std::vector<std::unique_ptr<xml::XmlResource>> LinkAndVersionXmlFile(ResourceTable* table,
366 FileOperation* file_op);
Adam Lesinski355f2852016-02-13 20:26:45 -0800367
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700368 ResourceFileFlattenerOptions options_;
369 IAaptContext* context_;
370 proguard::KeepSet* keep_set_;
Adam Lesinskic744ae82017-05-17 19:28:38 -0700371 XmlCompatVersioner::Rules rules_;
Adam Lesinski355f2852016-02-13 20:26:45 -0800372};
373
Adam Lesinskic744ae82017-05-17 19:28:38 -0700374ResourceFileFlattener::ResourceFileFlattener(const ResourceFileFlattenerOptions& options,
375 IAaptContext* context, proguard::KeepSet* keep_set)
376 : options_(options), context_(context), keep_set_(keep_set) {
377 SymbolTable* symm = context_->GetExternalSymbols();
378
379 // Build up the rules for degrading newer attributes to older ones.
380 // NOTE(adamlesinski): These rules are hardcoded right now, but they should be
381 // generated from the attribute definitions themselves (b/62028956).
382 if (const SymbolTable::Symbol* s = symm->FindById(R::attr::paddingHorizontal)) {
383 std::vector<ReplacementAttr> replacements{
384 {"paddingLeft", R::attr::paddingLeft,
385 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
386 {"paddingRight", R::attr::paddingRight,
387 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
388 };
389 rules_[R::attr::paddingHorizontal] =
390 util::make_unique<DegradeToManyRule>(std::move(replacements));
391 }
392
393 if (const SymbolTable::Symbol* s = symm->FindById(R::attr::paddingVertical)) {
394 std::vector<ReplacementAttr> replacements{
395 {"paddingTop", R::attr::paddingTop,
396 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
397 {"paddingBottom", R::attr::paddingBottom,
398 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
399 };
400 rules_[R::attr::paddingVertical] =
401 util::make_unique<DegradeToManyRule>(std::move(replacements));
402 }
403
404 if (const SymbolTable::Symbol* s = symm->FindById(R::attr::layout_marginHorizontal)) {
405 std::vector<ReplacementAttr> replacements{
406 {"layout_marginLeft", R::attr::layout_marginLeft,
407 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
408 {"layout_marginRight", R::attr::layout_marginRight,
409 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
410 };
411 rules_[R::attr::layout_marginHorizontal] =
412 util::make_unique<DegradeToManyRule>(std::move(replacements));
413 }
414
415 if (const SymbolTable::Symbol* s = symm->FindById(R::attr::layout_marginVertical)) {
416 std::vector<ReplacementAttr> replacements{
417 {"layout_marginTop", R::attr::layout_marginTop,
418 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
419 {"layout_marginBottom", R::attr::layout_marginBottom,
420 Attribute(false, android::ResTable_map::TYPE_DIMENSION)},
421 };
422 rules_[R::attr::layout_marginVertical] =
423 util::make_unique<DegradeToManyRule>(std::move(replacements));
424 }
425}
426
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700427uint32_t ResourceFileFlattener::GetCompressionFlags(const StringPiece& str) {
428 if (options_.do_not_compress_anything) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700429 return 0;
430 }
Adam Lesinski355f2852016-02-13 20:26:45 -0800431
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700432 for (const std::string& extension : options_.extensions_to_not_compress) {
433 if (util::EndsWith(str, extension)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700434 return 0;
Adam Lesinski355f2852016-02-13 20:26:45 -0800435 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700436 }
437 return ArchiveEntry::kCompress;
Adam Lesinski355f2852016-02-13 20:26:45 -0800438}
439
Adam Lesinskibb94f322017-07-12 07:41:55 -0700440static bool IsTransitionElement(const std::string& name) {
441 return name == "fade" || name == "changeBounds" || name == "slide" || name == "explode" ||
442 name == "changeImageTransform" || name == "changeTransform" ||
443 name == "changeClipBounds" || name == "autoTransition" || name == "recolor" ||
444 name == "changeScroll" || name == "transitionSet" || name == "transition" ||
445 name == "transitionManager";
446}
447
448static bool IsVectorElement(const std::string& name) {
449 return name == "vector" || name == "animated-vector" || name == "pathInterpolator" ||
ztenghuiab2a38c2017-10-13 15:56:08 -0700450 name == "objectAnimator" || name == "gradient";
Adam Lesinskibb94f322017-07-12 07:41:55 -0700451}
452
Adam Lesinskic744ae82017-05-17 19:28:38 -0700453template <typename T>
454std::vector<T> make_singleton_vec(T&& val) {
455 std::vector<T> vec;
456 vec.emplace_back(std::forward<T>(val));
457 return vec;
458}
459
460std::vector<std::unique_ptr<xml::XmlResource>> ResourceFileFlattener::LinkAndVersionXmlFile(
461 ResourceTable* table, FileOperation* file_op) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700462 xml::XmlResource* doc = file_op->xml_to_flatten.get();
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700463 const Source& src = doc->file.source;
Adam Lesinski5eeaadd2016-08-25 12:26:56 -0700464
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700465 if (context_->IsVerbose()) {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -0700466 context_->GetDiagnostics()->Note(DiagMessage()
467 << "linking " << src.path << " (" << doc->file.name << ")");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700468 }
469
Adam Lesinski00451162017-10-03 07:44:08 -0700470 // First, strip out any tools namespace attributes. AAPT stripped them out early, which means
471 // that existing projects have out-of-date references which pass compilation.
472 xml::StripAndroidStudioAttributes(doc->root.get());
473
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700474 XmlReferenceLinker xml_linker;
475 if (!xml_linker.Consume(context_, doc)) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700476 return {};
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700477 }
478
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700479 if (options_.update_proguard_spec && !proguard::CollectProguardRules(src, doc, keep_set_)) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700480 return {};
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700481 }
482
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700483 if (options_.no_xml_namespaces) {
484 XmlNamespaceRemover namespace_remover;
485 if (!namespace_remover.Consume(context_, doc)) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700486 return {};
Adam Lesinski355f2852016-02-13 20:26:45 -0800487 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700488 }
Adam Lesinski355f2852016-02-13 20:26:45 -0800489
Adam Lesinskibb94f322017-07-12 07:41:55 -0700490 if (options_.no_auto_version) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700491 return make_singleton_vec(std::move(file_op->xml_to_flatten));
492 }
Alexandria Cornwalla7cc3f12016-08-16 13:33:32 -0700493
Adam Lesinskibb94f322017-07-12 07:41:55 -0700494 if (options_.no_version_vectors || options_.no_version_transitions) {
495 // Skip this if it is a vector or animated-vector.
Adam Lesinski6b372992017-08-09 10:54:23 -0700496 xml::Element* el = doc->root.get();
Adam Lesinskibb94f322017-07-12 07:41:55 -0700497 if (el && el->namespace_uri.empty()) {
498 if ((options_.no_version_vectors && IsVectorElement(el->name)) ||
499 (options_.no_version_transitions && IsTransitionElement(el->name))) {
500 return make_singleton_vec(std::move(file_op->xml_to_flatten));
501 }
502 }
503 }
504
Adam Lesinskic744ae82017-05-17 19:28:38 -0700505 const ConfigDescription& config = file_op->config;
506 ResourceEntry* entry = file_op->entry;
507
508 XmlCompatVersioner xml_compat_versioner(&rules_);
509 const util::Range<ApiVersion> api_range{config.sdkVersion,
510 FindNextApiVersionForConfig(entry, config)};
511 return xml_compat_versioner.Process(context_, doc, api_range);
Adam Lesinski355f2852016-02-13 20:26:45 -0800512}
513
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700514bool ResourceFileFlattener::Flatten(ResourceTable* table, IArchiveWriter* archive_writer) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700515 bool error = false;
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700516 std::map<std::pair<ConfigDescription, StringPiece>, FileOperation> config_sorted_files;
Adam Lesinski355f2852016-02-13 20:26:45 -0800517
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700518 for (auto& pkg : table->packages) {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -0700519 CHECK(!pkg->name.empty()) << "Packages must have names when being linked";
520
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700521 for (auto& type : pkg->types) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700522 // Sort by config and name, so that we get better locality in the zip file.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700523 config_sorted_files.clear();
Adam Lesinskibb94f322017-07-12 07:41:55 -0700524 std::queue<FileOperation> file_operations;
Adam Lesinski355f2852016-02-13 20:26:45 -0800525
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700526 // Populate the queue with all files in the ResourceTable.
527 for (auto& entry : type->entries) {
Adam Lesinskibb94f322017-07-12 07:41:55 -0700528 for (auto& config_value : entry->values) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700529 // WARNING! Do not insert or remove any resources while executing in this scope. It will
530 // corrupt the iteration order.
531
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700532 FileReference* file_ref = ValueCast<FileReference>(config_value->value.get());
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700533 if (!file_ref) {
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700534 continue;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700535 }
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700536
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700537 io::IFile* file = file_ref->file;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700538 if (!file) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700539 context_->GetDiagnostics()->Error(DiagMessage(file_ref->GetSource())
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700540 << "file not found");
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700541 return false;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700542 }
543
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700544 FileOperation file_op;
545 file_op.entry = entry.get();
546 file_op.dst_path = *file_ref->path;
547 file_op.config = config_value->config;
Adam Lesinskic744ae82017-05-17 19:28:38 -0700548 file_op.file_to_copy = file;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700549
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700550 if (type->type != ResourceType::kRaw &&
Adam Lesinski00451162017-10-03 07:44:08 -0700551 (file_ref->type == ResourceFile::Type::kBinaryXml ||
552 file_ref->type == ResourceFile::Type::kProtoXml)) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700553 std::unique_ptr<io::IData> data = file->OpenAsData();
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700554 if (!data) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700555 context_->GetDiagnostics()->Error(DiagMessage(file->GetSource())
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700556 << "failed to open file");
557 return false;
558 }
559
Adam Lesinski00451162017-10-03 07:44:08 -0700560 if (file_ref->type == ResourceFile::Type::kProtoXml) {
561 pb::XmlNode pb_xml_node;
562 if (!pb_xml_node.ParseFromArray(data->data(), static_cast<int>(data->size()))) {
563 context_->GetDiagnostics()->Error(DiagMessage(file->GetSource())
564 << "failed to parse proto xml");
565 return false;
566 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700567
Adam Lesinski00451162017-10-03 07:44:08 -0700568 std::string error;
569 file_op.xml_to_flatten = DeserializeXmlResourceFromPb(pb_xml_node, &error);
570 if (file_op.xml_to_flatten == nullptr) {
571 context_->GetDiagnostics()->Error(DiagMessage(file->GetSource())
572 << "failed to deserialize proto xml: " << error);
573 return false;
574 }
575 } else {
576 file_op.xml_to_flatten = xml::Inflate(data->data(), data->size(),
577 context_->GetDiagnostics(), file->GetSource());
578 if (file_op.xml_to_flatten == nullptr) {
579 return false;
580 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700581 }
582
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700583 file_op.xml_to_flatten->file.config = config_value->config;
584 file_op.xml_to_flatten->file.source = file_ref->GetSource();
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700585 file_op.xml_to_flatten->file.name = ResourceName(pkg->name, type->type, entry->name);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700586 }
Adam Lesinskic744ae82017-05-17 19:28:38 -0700587
588 // NOTE(adamlesinski): Explicitly construct a StringPiece here, or
589 // else we end up copying the string in the std::make_pair() method,
590 // then creating a StringPiece from the copy, which would cause us
591 // to end up referencing garbage in the map.
592 const StringPiece entry_name(entry->name);
593 config_sorted_files[std::make_pair(config_value->config, entry_name)] =
594 std::move(file_op);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700595 }
596 }
597
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700598 // Now flatten the sorted values.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700599 for (auto& map_entry : config_sorted_files) {
600 const ConfigDescription& config = map_entry.first.first;
Adam Lesinskic744ae82017-05-17 19:28:38 -0700601 FileOperation& file_op = map_entry.second;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700602
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700603 if (file_op.xml_to_flatten) {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700604 std::vector<std::unique_ptr<xml::XmlResource>> versioned_docs =
605 LinkAndVersionXmlFile(table, &file_op);
Adam Lesinskic0a5e1e2017-08-07 11:56:32 -0700606 if (versioned_docs.empty()) {
607 error = true;
608 continue;
609 }
610
Adam Lesinskic744ae82017-05-17 19:28:38 -0700611 for (std::unique_ptr<xml::XmlResource>& doc : versioned_docs) {
612 std::string dst_path = file_op.dst_path;
613 if (doc->file.config != file_op.config) {
614 // Only add the new versioned configurations.
615 if (context_->IsVerbose()) {
616 context_->GetDiagnostics()->Note(DiagMessage(doc->file.source)
617 << "auto-versioning resource from config '"
618 << config << "' -> '" << doc->file.config << "'");
619 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700620
Adam Lesinskic744ae82017-05-17 19:28:38 -0700621 dst_path =
622 ResourceUtils::BuildResourceFileName(doc->file, context_->GetNameMangler());
623 bool result = table->AddFileReferenceAllowMangled(doc->file.name, doc->file.config,
624 doc->file.source, dst_path, nullptr,
625 context_->GetDiagnostics());
626 if (!result) {
627 return false;
628 }
629 }
630 error |= !FlattenXml(context_, doc.get(), dst_path, options_.keep_raw_values,
Adam Lesinskiea13c1a2017-10-13 12:40:37 -0700631 false /*utf16*/, archive_writer);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700632 }
633 } else {
Adam Lesinskic744ae82017-05-17 19:28:38 -0700634 error |= !io::CopyFileToArchive(context_, file_op.file_to_copy, file_op.dst_path,
635 GetCompressionFlags(file_op.dst_path), archive_writer);
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700636 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700637 }
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700638 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700639 }
640 return !error;
641}
642
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700643static bool WriteStableIdMapToPath(IDiagnostics* diag,
644 const std::unordered_map<ResourceName, ResourceId>& id_map,
645 const std::string& id_map_path) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700646 std::ofstream fout(id_map_path, std::ofstream::binary);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700647 if (!fout) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700648 diag->Error(DiagMessage(id_map_path) << strerror(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700649 return false;
650 }
651
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700652 for (const auto& entry : id_map) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700653 const ResourceName& name = entry.first;
654 const ResourceId& id = entry.second;
655 fout << name << " = " << id << "\n";
656 }
657
658 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700659 diag->Error(DiagMessage(id_map_path) << "failed writing to file: "
660 << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700661 return false;
662 }
663
664 return true;
665}
666
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700667static bool LoadStableIdMap(IDiagnostics* diag, const std::string& path,
668 std::unordered_map<ResourceName, ResourceId>* out_id_map) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700669 std::string content;
Adam Lesinski2354b562017-05-26 16:31:38 -0700670 if (!android::base::ReadFileToString(path, &content, true /*follow_symlinks*/)) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700671 diag->Error(DiagMessage(path) << "failed reading stable ID file");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700672 return false;
673 }
674
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700675 out_id_map->clear();
676 size_t line_no = 0;
677 for (StringPiece line : util::Tokenize(content, '\n')) {
678 line_no++;
679 line = util::TrimWhitespace(line);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700680 if (line.empty()) {
681 continue;
682 }
683
684 auto iter = std::find(line.begin(), line.end(), '=');
685 if (iter == line.end()) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700686 diag->Error(DiagMessage(Source(path, line_no)) << "missing '='");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700687 return false;
688 }
689
690 ResourceNameRef name;
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700691 StringPiece res_name_str =
692 util::TrimWhitespace(line.substr(0, std::distance(line.begin(), iter)));
693 if (!ResourceUtils::ParseResourceName(res_name_str, &name)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700694 diag->Error(DiagMessage(Source(path, line_no)) << "invalid resource name '" << res_name_str
695 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700696 return false;
697 }
698
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700699 const size_t res_id_start_idx = std::distance(line.begin(), iter) + 1;
700 const size_t res_id_str_len = line.size() - res_id_start_idx;
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700701 StringPiece res_id_str = util::TrimWhitespace(line.substr(res_id_start_idx, res_id_str_len));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700702
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700703 Maybe<ResourceId> maybe_id = ResourceUtils::ParseResourceId(res_id_str);
704 if (!maybe_id) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700705 diag->Error(DiagMessage(Source(path, line_no)) << "invalid resource ID '" << res_id_str
706 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700707 return false;
708 }
709
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700710 (*out_id_map)[name.ToResourceName()] = maybe_id.value();
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700711 }
712 return true;
Adam Lesinskibf0bd0f2016-06-01 15:31:50 -0700713}
714
Adam Lesinskifb48d292015-11-07 15:52:13 -0800715class LinkCommand {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700716 public:
717 LinkCommand(LinkContext* context, const LinkOptions& options)
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700718 : options_(options),
719 context_(context),
720 final_table_(),
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700721 file_collection_(util::make_unique<io::FileCollection>()) {
722 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700723
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700724 /**
725 * Creates a SymbolTable that loads symbols from the various APKs and caches
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700726 * the results for faster lookup.
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700727 */
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700728 bool LoadSymbolsFromIncludePaths() {
729 std::unique_ptr<AssetManagerSymbolSource> asset_source =
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700730 util::make_unique<AssetManagerSymbolSource>();
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700731 for (const std::string& path : options_.include_paths) {
732 if (context_->IsVerbose()) {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -0700733 context_->GetDiagnostics()->Note(DiagMessage() << "including " << path);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700734 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700735
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700736 // First try to load the file as a static lib.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700737 std::string error_str;
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800738 std::unique_ptr<ResourceTable> include_static = LoadStaticLibrary(path, &error_str);
739 if (include_static) {
Adam Lesinskib522f042017-04-21 16:57:59 -0700740 if (context_->GetPackageType() != PackageType::kStaticLib) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800741 // Can't include static libraries when not building a static library (they have no IDs
742 // assigned).
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700743 context_->GetDiagnostics()->Error(
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800744 DiagMessage(path) << "can't include static library when not building a static lib");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700745 return false;
746 }
747
748 // If we are using --no-static-lib-packages, we need to rename the
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800749 // package of this table to our compilation package.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700750 if (options_.no_static_lib_packages) {
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800751 // Since package names can differ, and multiple packages can exist in a ResourceTable,
752 // we place the requirement that all static libraries are built with the package
753 // ID 0x7f. So if one is not found, this is an error.
754 if (ResourceTablePackage* pkg = include_static->FindPackageById(kAppPackageId)) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700755 pkg->name = context_->GetCompilationPackage();
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800756 } else {
757 context_->GetDiagnostics()->Error(DiagMessage(path)
758 << "no package with ID 0x7f found in static library");
759 return false;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700760 }
761 }
762
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700763 context_->GetExternalSymbols()->AppendSource(
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800764 util::make_unique<ResourceTableSymbolSource>(include_static.get()));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700765
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800766 static_table_includes_.push_back(std::move(include_static));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700767
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700768 } else if (!error_str.empty()) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700769 // We had an error with reading, so fail.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700770 context_->GetDiagnostics()->Error(DiagMessage(path) << error_str);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700771 return false;
772 }
773
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700774 if (!asset_source->AddAssetPath(path)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800775 context_->GetDiagnostics()->Error(DiagMessage(path) << "failed to load include path");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700776 return false;
777 }
778 }
779
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800780 // Capture the shared libraries so that the final resource table can be properly flattened
781 // with support for shared libraries.
782 for (auto& entry : asset_source->GetAssignedPackageIds()) {
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800783 if (entry.first > kFrameworkPackageId && entry.first < kAppPackageId) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -0800784 final_table_.included_packages_[entry.first] = entry.second;
785 }
786 }
787
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700788 context_->GetExternalSymbols()->AppendSource(std::move(asset_source));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700789 return true;
790 }
791
Adam Lesinskib0c47ef2017-03-06 20:05:57 -0800792 Maybe<AppInfo> ExtractAppInfoFromManifest(xml::XmlResource* xml_res, IDiagnostics* diag) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700793 // Make sure the first element is <manifest> with package attribute.
Adam Lesinskib0c47ef2017-03-06 20:05:57 -0800794 xml::Element* manifest_el = xml::FindRootElement(xml_res->root.get());
795 if (manifest_el == nullptr) {
796 return {};
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700797 }
Adam Lesinskib0c47ef2017-03-06 20:05:57 -0800798
799 AppInfo app_info;
800
801 if (!manifest_el->namespace_uri.empty() || manifest_el->name != "manifest") {
802 diag->Error(DiagMessage(xml_res->file.source) << "root tag must be <manifest>");
803 return {};
804 }
805
806 xml::Attribute* package_attr = manifest_el->FindAttribute({}, "package");
807 if (!package_attr) {
808 diag->Error(DiagMessage(xml_res->file.source)
809 << "<manifest> must have a 'package' attribute");
810 return {};
811 }
812 app_info.package = package_attr->value;
813
814 if (xml::Attribute* version_code_attr =
815 manifest_el->FindAttribute(xml::kSchemaAndroid, "versionCode")) {
816 Maybe<uint32_t> maybe_code = ResourceUtils::ParseInt(version_code_attr->value);
817 if (!maybe_code) {
818 diag->Error(DiagMessage(xml_res->file.source.WithLine(manifest_el->line_number))
819 << "invalid android:versionCode '" << version_code_attr->value << "'");
820 return {};
821 }
822 app_info.version_code = maybe_code.value();
823 }
824
825 if (xml::Attribute* revision_code_attr =
826 manifest_el->FindAttribute(xml::kSchemaAndroid, "revisionCode")) {
827 Maybe<uint32_t> maybe_code = ResourceUtils::ParseInt(revision_code_attr->value);
828 if (!maybe_code) {
829 diag->Error(DiagMessage(xml_res->file.source.WithLine(manifest_el->line_number))
830 << "invalid android:revisionCode '" << revision_code_attr->value << "'");
831 return {};
832 }
833 app_info.revision_code = maybe_code.value();
834 }
835
836 if (xml::Attribute* split_name_attr = manifest_el->FindAttribute({}, "split")) {
837 if (!split_name_attr->value.empty()) {
838 app_info.split_name = split_name_attr->value;
839 }
840 }
841
842 if (xml::Element* uses_sdk_el = manifest_el->FindChild({}, "uses-sdk")) {
843 if (xml::Attribute* min_sdk =
844 uses_sdk_el->FindAttribute(xml::kSchemaAndroid, "minSdkVersion")) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700845 app_info.min_sdk_version = ResourceUtils::ParseSdkVersion(min_sdk->value);
Adam Lesinskib0c47ef2017-03-06 20:05:57 -0800846 }
847 }
848 return app_info;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700849 }
850
Adam Lesinski1ef0fa92017-08-15 21:32:49 -0700851 // Precondition: ResourceTable doesn't have any IDs assigned yet, nor is it linked.
852 // Postcondition: ResourceTable has only one package left. All others are
853 // stripped, or there is an error and false is returned.
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700854 bool VerifyNoExternalPackages() {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700855 auto is_ext_package_func = [&](const std::unique_ptr<ResourceTablePackage>& pkg) -> bool {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700856 return context_->GetCompilationPackage() != pkg->name || !pkg->id ||
857 pkg->id.value() != context_->GetPackageId();
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700858 };
859
860 bool error = false;
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700861 for (const auto& package : final_table_.packages) {
862 if (is_ext_package_func(package)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700863 // We have a package that is not related to the one we're building!
864 for (const auto& type : package->types) {
865 for (const auto& entry : type->entries) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700866 ResourceNameRef res_name(package->name, type->type, entry->name);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700867
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700868 for (const auto& config_value : entry->values) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700869 // Special case the occurrence of an ID that is being generated
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700870 // for the 'android' package. This is due to legacy reasons.
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700871 if (ValueCast<Id>(config_value->value.get()) && package->name == "android") {
872 context_->GetDiagnostics()->Warn(DiagMessage(config_value->value->GetSource())
873 << "generated id '" << res_name
874 << "' for external package '" << package->name
875 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700876 } else {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700877 context_->GetDiagnostics()->Error(DiagMessage(config_value->value->GetSource())
878 << "defined resource '" << res_name
879 << "' for external package '" << package->name
880 << "'");
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700881 error = true;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700882 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700883 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700884 }
885 }
886 }
887 }
888
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700889 auto new_end_iter = std::remove_if(final_table_.packages.begin(), final_table_.packages.end(),
890 is_ext_package_func);
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700891 final_table_.packages.erase(new_end_iter, final_table_.packages.end());
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700892 return !error;
893 }
894
895 /**
896 * Returns true if no IDs have been set, false otherwise.
897 */
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700898 bool VerifyNoIdsSet() {
899 for (const auto& package : final_table_.packages) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700900 for (const auto& type : package->types) {
901 if (type->id) {
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800902 context_->GetDiagnostics()->Error(DiagMessage() << "type " << type->type << " has ID "
903 << StringPrintf("%02x", type->id.value())
904 << " assigned");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700905 return false;
Adam Lesinski1ab598f2015-08-14 14:26:04 -0700906 }
907
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700908 for (const auto& entry : type->entries) {
909 if (entry->id) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700910 ResourceNameRef res_name(package->name, type->type, entry->name);
911 context_->GetDiagnostics()->Error(
Adam Lesinskif34b6f42017-03-03 16:33:26 -0800912 DiagMessage() << "entry " << res_name << " has ID "
913 << StringPrintf("%02x", entry->id.value()) << " assigned");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700914 return false;
915 }
916 }
917 }
918 }
919 return true;
920 }
921
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700922 std::unique_ptr<IArchiveWriter> MakeArchiveWriter(const StringPiece& out) {
923 if (options_.output_to_directory) {
924 return CreateDirectoryArchiveWriter(context_->GetDiagnostics(), out);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700925 } else {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700926 return CreateZipFileArchiveWriter(context_->GetDiagnostics(), out);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700927 }
928 }
929
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700930 bool FlattenTable(ResourceTable* table, IArchiveWriter* writer) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700931 BigBuffer buffer(1024);
Adam Lesinskic8f71aa2017-02-08 07:03:50 -0800932 TableFlattener flattener(options_.table_flattener_options, &buffer);
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700933 if (!flattener.Consume(context_, table)) {
Adam Lesinski06460ef2017-03-14 18:52:13 -0700934 context_->GetDiagnostics()->Error(DiagMessage() << "failed to flatten resource table");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700935 return false;
936 }
937
Adam Lesinski06460ef2017-03-14 18:52:13 -0700938 io::BigBufferInputStream input_stream(&buffer);
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700939 return io::CopyInputStreamToArchive(context_, &input_stream, "resources.arsc",
940 ArchiveEntry::kAlign, writer);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700941 }
942
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700943 bool FlattenTableToPb(ResourceTable* table, IArchiveWriter* writer) {
Adam Lesinski8cdca1b2017-09-28 15:50:03 -0700944 pb::ResourceTable pb_table;
945 SerializeTableToPb(*table, &pb_table);
946 return io::CopyProtoToArchive(context_, &pb_table, "resources.arsc.flat", 0, writer);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700947 }
948
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700949 bool WriteJavaFile(ResourceTable* table, const StringPiece& package_name_to_generate,
Adam Lesinski418763f2017-04-11 17:36:53 -0700950 const StringPiece& out_package, const JavaClassGeneratorOptions& java_options,
Chih-Hung Hsieh4dc58122017-08-03 16:28:10 -0700951 const Maybe<std::string>& out_text_symbols_path = {}) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700952 if (!options_.generate_java_class_path) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700953 return true;
954 }
955
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700956 std::string out_path = options_.generate_java_class_path.value();
957 file::AppendPath(&out_path, file::PackageToPath(out_package));
958 if (!file::mkdirs(out_path)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700959 context_->GetDiagnostics()->Error(DiagMessage() << "failed to create directory '" << out_path
960 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700961 return false;
962 }
963
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700964 file::AppendPath(&out_path, "R.java");
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700965
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700966 std::ofstream fout(out_path, std::ofstream::binary);
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700967 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700968 context_->GetDiagnostics()->Error(DiagMessage()
969 << "failed writing to '" << out_path
970 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700971 return false;
972 }
973
Adam Lesinski418763f2017-04-11 17:36:53 -0700974 std::unique_ptr<std::ofstream> fout_text;
975 if (out_text_symbols_path) {
976 fout_text =
977 util::make_unique<std::ofstream>(out_text_symbols_path.value(), std::ofstream::binary);
978 if (!*fout_text) {
979 context_->GetDiagnostics()->Error(
980 DiagMessage() << "failed writing to '" << out_text_symbols_path.value()
981 << "': " << android::base::SystemErrorCodeToString(errno));
982 return false;
983 }
984 }
985
Adam Lesinskice5e56e2016-10-21 17:56:45 -0700986 JavaClassGenerator generator(context_, table, java_options);
Adam Lesinski418763f2017-04-11 17:36:53 -0700987 if (!generator.Generate(package_name_to_generate, out_package, &fout, fout_text.get())) {
Adam Lesinski06460ef2017-03-14 18:52:13 -0700988 context_->GetDiagnostics()->Error(DiagMessage(out_path) << generator.getError());
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700989 return false;
990 }
991
992 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -0700993 context_->GetDiagnostics()->Error(DiagMessage()
994 << "failed writing to '" << out_path
995 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinski1ef0fa92017-08-15 21:32:49 -0700996 return false;
Adam Lesinskicacb28f2016-10-19 12:18:14 -0700997 }
998 return true;
999 }
1000
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001001 bool GenerateJavaClasses() {
1002 // The set of packages whose R class to call in the main classes onResourcesLoaded callback.
1003 std::vector<std::string> packages_to_callback;
1004
1005 JavaClassGeneratorOptions template_options;
1006 template_options.types = JavaClassGeneratorOptions::SymbolTypes::kAll;
1007 template_options.javadoc_annotations = options_.javadoc_annotations;
1008
1009 if (context_->GetPackageType() == PackageType::kStaticLib || options_.generate_non_final_ids) {
1010 template_options.use_final = false;
1011 }
1012
1013 if (context_->GetPackageType() == PackageType::kSharedLib) {
1014 template_options.use_final = false;
1015 template_options.rewrite_callback_options = OnResourcesLoadedCallbackOptions{};
1016 }
1017
1018 const StringPiece actual_package = context_->GetCompilationPackage();
1019 StringPiece output_package = context_->GetCompilationPackage();
1020 if (options_.custom_java_package) {
1021 // Override the output java package to the custom one.
1022 output_package = options_.custom_java_package.value();
1023 }
1024
1025 // Generate the private symbols if required.
1026 if (options_.private_symbols) {
1027 packages_to_callback.push_back(options_.private_symbols.value());
1028
1029 // If we defined a private symbols package, we only emit Public symbols
1030 // to the original package, and private and public symbols to the private package.
1031 JavaClassGeneratorOptions options = template_options;
1032 options.types = JavaClassGeneratorOptions::SymbolTypes::kPublicPrivate;
1033 if (!WriteJavaFile(&final_table_, actual_package, options_.private_symbols.value(),
1034 options)) {
1035 return false;
1036 }
1037 }
1038
1039 // Generate copies of the original package R class but with different package names.
1040 // This is to support non-namespaced builds.
1041 for (const std::string& extra_package : options_.extra_java_packages) {
1042 packages_to_callback.push_back(extra_package);
1043
1044 JavaClassGeneratorOptions options = template_options;
1045 options.types = JavaClassGeneratorOptions::SymbolTypes::kAll;
1046 if (!WriteJavaFile(&final_table_, actual_package, extra_package, options)) {
1047 return false;
1048 }
1049 }
1050
1051 // Generate R classes for each package that was merged (static library).
1052 // Use the actual package's resources only.
1053 for (const std::string& package : table_merger_->merged_packages()) {
1054 packages_to_callback.push_back(package);
1055
1056 JavaClassGeneratorOptions options = template_options;
1057 options.types = JavaClassGeneratorOptions::SymbolTypes::kAll;
1058 if (!WriteJavaFile(&final_table_, package, package, options)) {
1059 return false;
1060 }
1061 }
1062
1063 // Generate the main public R class.
1064 JavaClassGeneratorOptions options = template_options;
1065
1066 // Only generate public symbols if we have a private package.
1067 if (options_.private_symbols) {
1068 options.types = JavaClassGeneratorOptions::SymbolTypes::kPublic;
1069 }
1070
1071 if (options.rewrite_callback_options) {
1072 options.rewrite_callback_options.value().packages_to_callback =
1073 std::move(packages_to_callback);
1074 }
1075
1076 if (!WriteJavaFile(&final_table_, actual_package, output_package, options,
1077 options_.generate_text_symbols_path)) {
1078 return false;
1079 }
1080
1081 return true;
1082 }
1083
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001084 bool WriteManifestJavaFile(xml::XmlResource* manifest_xml) {
1085 if (!options_.generate_java_class_path) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001086 return true;
1087 }
1088
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001089 std::unique_ptr<ClassDefinition> manifest_class =
1090 GenerateManifestClass(context_->GetDiagnostics(), manifest_xml);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001091
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001092 if (!manifest_class) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001093 // Something bad happened, but we already logged it, so exit.
1094 return false;
1095 }
1096
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001097 if (manifest_class->empty()) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001098 // Empty Manifest class, no need to generate it.
1099 return true;
1100 }
1101
1102 // Add any JavaDoc annotations to the generated class.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001103 for (const std::string& annotation : options_.javadoc_annotations) {
1104 std::string proper_annotation = "@";
1105 proper_annotation += annotation;
1106 manifest_class->GetCommentBuilder()->AppendComment(proper_annotation);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001107 }
1108
Adam Lesinski0d81f702017-06-27 15:51:09 -07001109 const std::string package_utf8 =
1110 options_.custom_java_package.value_or_default(context_->GetCompilationPackage());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001111
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001112 std::string out_path = options_.generate_java_class_path.value();
1113 file::AppendPath(&out_path, file::PackageToPath(package_utf8));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001114
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001115 if (!file::mkdirs(out_path)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001116 context_->GetDiagnostics()->Error(DiagMessage() << "failed to create directory '" << out_path
1117 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001118 return false;
1119 }
1120
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001121 file::AppendPath(&out_path, "Manifest.java");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001122
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001123 std::ofstream fout(out_path, std::ofstream::binary);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001124 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001125 context_->GetDiagnostics()->Error(DiagMessage()
1126 << "failed writing to '" << out_path
1127 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001128 return false;
1129 }
1130
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001131 if (!ClassDefinition::WriteJavaFile(manifest_class.get(), package_utf8, true, &fout)) {
1132 context_->GetDiagnostics()->Error(DiagMessage()
1133 << "failed writing to '" << out_path
1134 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001135 return false;
1136 }
1137 return true;
1138 }
1139
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001140 bool WriteProguardFile(const Maybe<std::string>& out, const proguard::KeepSet& keep_set) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001141 if (!out) {
1142 return true;
1143 }
1144
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001145 const std::string& out_path = out.value();
1146 std::ofstream fout(out_path, std::ofstream::binary);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001147 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001148 context_->GetDiagnostics()->Error(DiagMessage()
1149 << "failed to open '" << out_path
1150 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001151 return false;
1152 }
1153
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001154 proguard::WriteKeepSet(&fout, keep_set);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001155 if (!fout) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001156 context_->GetDiagnostics()->Error(DiagMessage()
1157 << "failed writing to '" << out_path
1158 << "': " << android::base::SystemErrorCodeToString(errno));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001159 return false;
1160 }
1161 return true;
1162 }
1163
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001164 std::unique_ptr<ResourceTable> LoadStaticLibrary(const std::string& input,
1165 std::string* out_error) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001166 std::unique_ptr<io::ZipFileCollection> collection =
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001167 io::ZipFileCollection::Create(input, out_error);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001168 if (!collection) {
1169 return {};
1170 }
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001171 return LoadTablePbFromCollection(collection.get());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001172 }
1173
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001174 std::unique_ptr<ResourceTable> LoadTablePbFromCollection(io::IFileCollection* collection) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001175 io::IFile* file = collection->FindFile("resources.arsc.flat");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001176 if (!file) {
1177 return {};
1178 }
1179
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001180 std::unique_ptr<io::IData> data = file->OpenAsData();
1181 return LoadTableFromPb(file->GetSource(), data->data(), data->size(),
1182 context_->GetDiagnostics());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001183 }
1184
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001185 bool MergeStaticLibrary(const std::string& input, bool override) {
1186 if (context_->IsVerbose()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001187 context_->GetDiagnostics()->Note(DiagMessage() << "merging static library " << input);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001188 }
1189
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001190 std::string error_str;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001191 std::unique_ptr<io::ZipFileCollection> collection =
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001192 io::ZipFileCollection::Create(input, &error_str);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001193 if (!collection) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001194 context_->GetDiagnostics()->Error(DiagMessage(input) << error_str);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001195 return false;
1196 }
1197
Adam Lesinskib5dc4bd2017-02-22 19:29:29 -08001198 std::unique_ptr<ResourceTable> table = LoadTablePbFromCollection(collection.get());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001199 if (!table) {
Adam Lesinskib5dc4bd2017-02-22 19:29:29 -08001200 context_->GetDiagnostics()->Error(DiagMessage(input) << "invalid static library");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001201 return false;
1202 }
1203
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001204 ResourceTablePackage* pkg = table->FindPackageById(kAppPackageId);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001205 if (!pkg) {
Adam Lesinskib5dc4bd2017-02-22 19:29:29 -08001206 context_->GetDiagnostics()->Error(DiagMessage(input) << "static library has no package");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001207 return false;
1208 }
1209
1210 bool result;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001211 if (options_.no_static_lib_packages) {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001212 // Merge all resources as if they were in the compilation package. This is the old behavior
1213 // of aapt.
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001214
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001215 // Add the package to the set of --extra-packages so we emit an R.java for each library
1216 // package.
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001217 if (!pkg->name.empty()) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001218 options_.extra_java_packages.insert(pkg->name);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001219 }
1220
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001221 // Clear the package name, so as to make the resources look like they are coming from the
1222 // local package.
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001223 pkg->name = "";
Adam Lesinski00451162017-10-03 07:44:08 -07001224 result = table_merger_->Merge(Source(input), table.get(), override, collection.get());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001225
1226 } else {
1227 // This is the proper way to merge libraries, where the package name is
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001228 // preserved and resource names are mangled.
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001229 result =
1230 table_merger_->MergeAndMangle(Source(input), pkg->name, table.get(), collection.get());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001231 }
1232
1233 if (!result) {
1234 return false;
1235 }
1236
1237 // Make sure to move the collection into the set of IFileCollections.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001238 collections_.push_back(std::move(collection));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001239 return true;
1240 }
1241
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001242 bool MergeResourceTable(io::IFile* file, bool override) {
1243 if (context_->IsVerbose()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001244 context_->GetDiagnostics()->Note(DiagMessage() << "merging resource table "
1245 << file->GetSource());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001246 }
1247
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001248 std::unique_ptr<io::IData> data = file->OpenAsData();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001249 if (!data) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001250 context_->GetDiagnostics()->Error(DiagMessage(file->GetSource()) << "failed to open file");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001251 return false;
1252 }
1253
1254 std::unique_ptr<ResourceTable> table =
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001255 LoadTableFromPb(file->GetSource(), data->data(), data->size(), context_->GetDiagnostics());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001256 if (!table) {
1257 return false;
1258 }
1259
Adam Lesinski00451162017-10-03 07:44:08 -07001260 return table_merger_->Merge(file->GetSource(), table.get(), override);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001261 }
1262
Adam Lesinski00451162017-10-03 07:44:08 -07001263 bool MergeCompiledFile(const ResourceFile& compiled_file, io::IFile* file, bool override) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001264 if (context_->IsVerbose()) {
Adam Lesinski00451162017-10-03 07:44:08 -07001265 context_->GetDiagnostics()->Note(DiagMessage()
1266 << "merging '" << compiled_file.name
1267 << "' from compiled file " << compiled_file.source);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001268 }
1269
Adam Lesinski00451162017-10-03 07:44:08 -07001270 if (!table_merger_->MergeFile(compiled_file, override, file)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001271 return false;
1272 }
1273
1274 // Add the exports of this file to the table.
Adam Lesinski00451162017-10-03 07:44:08 -07001275 for (const SourcedResourceName& exported_symbol : compiled_file.exported_symbols) {
1276 ResourceName res_name = exported_symbol.name;
1277 if (res_name.package.empty()) {
1278 res_name.package = context_->GetCompilationPackage();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001279 }
1280
Adam Lesinski00451162017-10-03 07:44:08 -07001281 Maybe<ResourceName> mangled_name = context_->GetNameMangler()->MangleName(res_name);
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001282 if (mangled_name) {
1283 res_name = mangled_name.value();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001284 }
1285
1286 std::unique_ptr<Id> id = util::make_unique<Id>();
Adam Lesinski00451162017-10-03 07:44:08 -07001287 id->SetSource(compiled_file.source.WithLine(exported_symbol.line));
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001288 bool result = final_table_.AddResourceAllowMangled(
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001289 res_name, ConfigDescription::DefaultConfig(), std::string(), std::move(id),
1290 context_->GetDiagnostics());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001291 if (!result) {
1292 return false;
1293 }
1294 }
1295 return true;
1296 }
1297
Adam Lesinski00451162017-10-03 07:44:08 -07001298 // Takes a path to load as a ZIP file and merges the files within into the master ResourceTable.
1299 // If override is true, conflicting resources are allowed to override each other, in order of last
1300 // seen.
1301 // An io::IFileCollection is created from the ZIP file and added to the set of
1302 // io::IFileCollections that are open.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001303 bool MergeArchive(const std::string& input, bool override) {
1304 if (context_->IsVerbose()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001305 context_->GetDiagnostics()->Note(DiagMessage() << "merging archive " << input);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001306 }
1307
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001308 std::string error_str;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001309 std::unique_ptr<io::ZipFileCollection> collection =
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001310 io::ZipFileCollection::Create(input, &error_str);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001311 if (!collection) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001312 context_->GetDiagnostics()->Error(DiagMessage(input) << error_str);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001313 return false;
1314 }
1315
1316 bool error = false;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001317 for (auto iter = collection->Iterator(); iter->HasNext();) {
1318 if (!MergeFile(iter->Next(), override)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001319 error = true;
1320 }
1321 }
1322
1323 // Make sure to move the collection into the set of IFileCollections.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001324 collections_.push_back(std::move(collection));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001325 return !error;
1326 }
1327
Adam Lesinski00451162017-10-03 07:44:08 -07001328 // Takes a path to load and merge into the master ResourceTable. If override is true,
1329 // conflicting resources are allowed to override each other, in order of last seen.
1330 // If the file path ends with .flata, .jar, .jack, or .zip the file is treated
1331 // as ZIP archive and the files within are merged individually.
1332 // Otherwise the file is processed on its own.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001333 bool MergePath(const std::string& path, bool override) {
1334 if (util::EndsWith(path, ".flata") || util::EndsWith(path, ".jar") ||
1335 util::EndsWith(path, ".jack") || util::EndsWith(path, ".zip")) {
1336 return MergeArchive(path, override);
1337 } else if (util::EndsWith(path, ".apk")) {
1338 return MergeStaticLibrary(path, override);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001339 }
1340
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001341 io::IFile* file = file_collection_->InsertFile(path);
1342 return MergeFile(file, override);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001343 }
1344
Adam Lesinski00451162017-10-03 07:44:08 -07001345 // Takes an AAPT Container file (.apc/.flat) to load and merge into the master ResourceTable.
1346 // If override is true, conflicting resources are allowed to override each other, in order of last
1347 // seen.
1348 // All other file types are ignored. This is because these files could be coming from a zip,
1349 // where we could have other files like classes.dex.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001350 bool MergeFile(io::IFile* file, bool override) {
1351 const Source& src = file->GetSource();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001352
Adam Lesinski00451162017-10-03 07:44:08 -07001353 if (util::EndsWith(src.path, ".xml") || util::EndsWith(src.path, ".png")) {
Adam Lesinski6a396c12016-10-20 14:38:23 -07001354 // Since AAPT compiles these file types and appends .flat to them, seeing
1355 // their raw extensions is a sign that they weren't compiled.
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001356 const StringPiece file_type = util::EndsWith(src.path, ".xml") ? "XML" : "PNG";
1357 context_->GetDiagnostics()->Error(DiagMessage(src) << "uncompiled " << file_type
1358 << " file passed as argument. Must be "
1359 "compiled first into .flat file.");
Adam Lesinski6a396c12016-10-20 14:38:23 -07001360 return false;
Adam Lesinski00451162017-10-03 07:44:08 -07001361 } else if (!util::EndsWith(src.path, ".apc") && !util::EndsWith(src.path, ".flat")) {
1362 if (context_->IsVerbose()) {
1363 context_->GetDiagnostics()->Warn(DiagMessage(src) << "ignoring unrecognized file");
1364 return true;
1365 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001366 }
1367
Adam Lesinski00451162017-10-03 07:44:08 -07001368 std::unique_ptr<io::InputStream> input_stream = file->OpenInputStream();
1369 if (input_stream == nullptr) {
1370 context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to open file");
1371 return false;
1372 }
1373
1374 if (input_stream->HadError()) {
1375 context_->GetDiagnostics()->Error(DiagMessage(src)
1376 << "failed to open file: " << input_stream->GetError());
1377 return false;
1378 }
1379
1380 ContainerReaderEntry* entry;
1381 ContainerReader reader(input_stream.get());
1382 while ((entry = reader.Next()) != nullptr) {
1383 if (entry->Type() == ContainerEntryType::kResTable) {
1384 pb::ResourceTable pb_table;
1385 if (!entry->GetResTable(&pb_table)) {
1386 context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to read resource table: "
1387 << entry->GetError());
1388 return false;
1389 }
1390
1391 ResourceTable table;
1392 std::string error;
1393 if (!DeserializeTableFromPb(pb_table, &table, &error)) {
1394 context_->GetDiagnostics()->Error(DiagMessage(src)
1395 << "failed to deserialize resource table: " << error);
1396 return false;
1397 }
1398
1399 if (!table_merger_->Merge(src, &table, override)) {
1400 context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to merge resource table");
1401 return false;
1402 }
1403 } else if (entry->Type() == ContainerEntryType::kResFile) {
1404 pb::internal::CompiledFile pb_compiled_file;
1405 off64_t offset;
1406 size_t len;
1407 if (!entry->GetResFileOffsets(&pb_compiled_file, &offset, &len)) {
1408 context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to get resource file: "
1409 << entry->GetError());
1410 return false;
1411 }
1412
1413 ResourceFile resource_file;
1414 std::string error;
1415 if (!DeserializeCompiledFileFromPb(pb_compiled_file, &resource_file, &error)) {
1416 context_->GetDiagnostics()->Error(DiagMessage(src)
1417 << "failed to read compiled header: " << error);
1418 return false;
1419 }
1420
1421 if (!MergeCompiledFile(resource_file, file->CreateFileSegment(offset, len), override)) {
1422 return false;
1423 }
1424 }
1425 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001426 return true;
1427 }
1428
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001429 bool CopyAssetsDirsToApk(IArchiveWriter* writer) {
1430 std::map<std::string, std::unique_ptr<io::RegularFile>> merged_assets;
1431 for (const std::string& assets_dir : options_.assets_dirs) {
1432 Maybe<std::vector<std::string>> files =
1433 file::FindFiles(assets_dir, context_->GetDiagnostics(), nullptr);
1434 if (!files) {
1435 return false;
1436 }
1437
1438 for (const std::string& file : files.value()) {
1439 std::string full_key = "assets/" + file;
1440 std::string full_path = assets_dir;
1441 file::AppendPath(&full_path, file);
1442
1443 auto iter = merged_assets.find(full_key);
1444 if (iter == merged_assets.end()) {
1445 merged_assets.emplace(std::move(full_key),
1446 util::make_unique<io::RegularFile>(Source(std::move(full_path))));
1447 } else if (context_->IsVerbose()) {
1448 context_->GetDiagnostics()->Warn(DiagMessage(iter->second->GetSource())
1449 << "asset file overrides '" << full_path << "'");
1450 }
1451 }
1452 }
1453
1454 for (auto& entry : merged_assets) {
1455 uint32_t compression_flags = ArchiveEntry::kCompress;
1456 std::string extension = file::GetExtension(entry.first).to_string();
1457 if (options_.extensions_to_not_compress.count(extension) > 0) {
1458 compression_flags = 0u;
1459 }
1460
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001461 if (!io::CopyFileToArchive(context_, entry.second.get(), entry.first, compression_flags,
1462 writer)) {
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001463 return false;
1464 }
1465 }
1466 return true;
1467 }
1468
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001469 /**
1470 * Writes the AndroidManifest, ResourceTable, and all XML files referenced by
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001471 * the ResourceTable to the IArchiveWriter.
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001472 */
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001473 bool WriteApk(IArchiveWriter* writer, proguard::KeepSet* keep_set, xml::XmlResource* manifest,
1474 ResourceTable* table) {
Adam Lesinskib522f042017-04-21 16:57:59 -07001475 const bool keep_raw_values = context_->GetPackageType() == PackageType::kStaticLib;
Adam Lesinskiea13c1a2017-10-13 12:40:37 -07001476 bool result = FlattenXml(context_, manifest, "AndroidManifest.xml", keep_raw_values,
1477 true /*utf16*/, writer);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001478 if (!result) {
1479 return false;
1480 }
1481
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001482 ResourceFileFlattenerOptions file_flattener_options;
1483 file_flattener_options.keep_raw_values = keep_raw_values;
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001484 file_flattener_options.do_not_compress_anything = options_.do_not_compress_anything;
1485 file_flattener_options.extensions_to_not_compress = options_.extensions_to_not_compress;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001486 file_flattener_options.no_auto_version = options_.no_auto_version;
1487 file_flattener_options.no_version_vectors = options_.no_version_vectors;
Yuichi Araki4d35cca2017-01-18 20:42:17 +09001488 file_flattener_options.no_version_transitions = options_.no_version_transitions;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001489 file_flattener_options.no_xml_namespaces = options_.no_xml_namespaces;
1490 file_flattener_options.update_proguard_spec =
1491 static_cast<bool>(options_.generate_proguard_rules_path);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001492
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001493 ResourceFileFlattener file_flattener(file_flattener_options, context_, keep_set);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001494
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001495 if (!file_flattener.Flatten(table, writer)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001496 context_->GetDiagnostics()->Error(DiagMessage() << "failed linking file resources");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001497 return false;
1498 }
1499
Adam Lesinskib522f042017-04-21 16:57:59 -07001500 if (context_->GetPackageType() == PackageType::kStaticLib) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001501 if (!FlattenTableToPb(table, writer)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001502 return false;
1503 }
1504 } else {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001505 if (!FlattenTable(table, writer)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001506 context_->GetDiagnostics()->Error(DiagMessage() << "failed to write resources.arsc");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001507 return false;
1508 }
1509 }
1510 return true;
1511 }
1512
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001513 int Run(const std::vector<std::string>& input_files) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001514 // Load the AndroidManifest.xml
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001515 std::unique_ptr<xml::XmlResource> manifest_xml =
1516 LoadXml(options_.manifest_path, context_->GetDiagnostics());
1517 if (!manifest_xml) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001518 return 1;
1519 }
1520
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001521 // First extract the Package name without modifying it (via --rename-manifest-package).
1522 if (Maybe<AppInfo> maybe_app_info =
1523 ExtractAppInfoFromManifest(manifest_xml.get(), context_->GetDiagnostics())) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001524 const AppInfo& app_info = maybe_app_info.value();
1525 context_->SetCompilationPackage(app_info.package);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001526 }
1527
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001528 ManifestFixer manifest_fixer(options_.manifest_fixer_options);
1529 if (!manifest_fixer.Consume(context_, manifest_xml.get())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001530 return 1;
1531 }
1532
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001533 Maybe<AppInfo> maybe_app_info =
1534 ExtractAppInfoFromManifest(manifest_xml.get(), context_->GetDiagnostics());
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001535 if (!maybe_app_info) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001536 return 1;
1537 }
1538
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001539 const AppInfo& app_info = maybe_app_info.value();
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001540 context_->SetMinSdkVersion(app_info.min_sdk_version.value_or_default(0));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001541
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001542 context_->SetNameManglerPolicy(NameManglerPolicy{context_->GetCompilationPackage()});
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001543
1544 // Override the package ID when it is "android".
1545 if (context_->GetCompilationPackage() == "android") {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001546 context_->SetPackageId(0x01);
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001547
1548 // Verify we're building a regular app.
Adam Lesinskib522f042017-04-21 16:57:59 -07001549 if (context_->GetPackageType() != PackageType::kApp) {
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001550 context_->GetDiagnostics()->Error(
1551 DiagMessage() << "package 'android' can only be built as a regular app");
1552 return 1;
1553 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001554 }
1555
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001556 if (!LoadSymbolsFromIncludePaths()) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001557 return 1;
1558 }
1559
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001560 TableMergerOptions table_merger_options;
1561 table_merger_options.auto_add_overlay = options_.auto_add_overlay;
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001562 table_merger_ = util::make_unique<TableMerger>(context_, &final_table_, table_merger_options);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001563
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001564 if (context_->IsVerbose()) {
1565 context_->GetDiagnostics()->Note(DiagMessage()
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001566 << StringPrintf("linking package '%s' using package ID %02x",
1567 context_->GetCompilationPackage().data(),
1568 context_->GetPackageId()));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001569 }
1570
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001571 for (const std::string& input : input_files) {
1572 if (!MergePath(input, false)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001573 context_->GetDiagnostics()->Error(DiagMessage() << "failed parsing input");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001574 return 1;
1575 }
1576 }
1577
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001578 for (const std::string& input : options_.overlay_files) {
1579 if (!MergePath(input, true)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001580 context_->GetDiagnostics()->Error(DiagMessage() << "failed parsing overlays");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001581 return 1;
1582 }
1583 }
1584
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001585 if (!VerifyNoExternalPackages()) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001586 return 1;
1587 }
1588
Adam Lesinskib522f042017-04-21 16:57:59 -07001589 if (context_->GetPackageType() != PackageType::kStaticLib) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001590 PrivateAttributeMover mover;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001591 if (!mover.Consume(context_, &final_table_)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001592 context_->GetDiagnostics()->Error(DiagMessage() << "failed moving private attributes");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001593 return 1;
1594 }
1595
1596 // Assign IDs if we are building a regular app.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001597 IdAssigner id_assigner(&options_.stable_id_map);
1598 if (!id_assigner.Consume(context_, &final_table_)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001599 context_->GetDiagnostics()->Error(DiagMessage() << "failed assigning IDs");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001600 return 1;
1601 }
1602
1603 // Now grab each ID and emit it as a file.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001604 if (options_.resource_id_map_path) {
1605 for (auto& package : final_table_.packages) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001606 for (auto& type : package->types) {
1607 for (auto& entry : type->entries) {
1608 ResourceName name(package->name, type->type, entry->name);
1609 // The IDs are guaranteed to exist.
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001610 options_.stable_id_map[std::move(name)] =
1611 ResourceId(package->id.value(), type->id.value(), entry->id.value());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001612 }
1613 }
1614 }
1615
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001616 if (!WriteStableIdMapToPath(context_->GetDiagnostics(), options_.stable_id_map,
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001617 options_.resource_id_map_path.value())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001618 return 1;
1619 }
1620 }
1621 } else {
1622 // Static libs are merged with other apps, and ID collisions are bad, so
1623 // verify that
1624 // no IDs have been set.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001625 if (!VerifyNoIdsSet()) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001626 return 1;
1627 }
1628 }
1629
1630 // Add the names to mangle based on our source merge earlier.
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001631 context_->SetNameManglerPolicy(
1632 NameManglerPolicy{context_->GetCompilationPackage(), table_merger_->merged_packages()});
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001633
1634 // Add our table to the symbol table.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001635 context_->GetExternalSymbols()->PrependSource(
1636 util::make_unique<ResourceTableSymbolSource>(&final_table_));
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001637
Adam Lesinski1e4b0e52017-04-27 15:01:10 -07001638 // Workaround for pre-O runtime that would treat negative resource IDs
1639 // (any ID with a package ID > 7f) as invalid. Intercept any ID (PPTTEEEE) with PP > 0x7f
1640 // and type == 'id', and return the ID 0x7fPPEEEE. IDs don't need to be real resources, they
1641 // are just identifiers.
1642 if (context_->GetMinSdkVersion() < SDK_O && context_->GetPackageType() == PackageType::kApp) {
1643 if (context_->IsVerbose()) {
1644 context_->GetDiagnostics()->Note(DiagMessage()
1645 << "enabling pre-O feature split ID rewriting");
1646 }
1647 context_->GetExternalSymbols()->SetDelegate(
1648 util::make_unique<FeatureSplitSymbolTableDelegate>(context_));
1649 }
1650
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001651 ReferenceLinker linker;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001652 if (!linker.Consume(context_, &final_table_)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001653 context_->GetDiagnostics()->Error(DiagMessage() << "failed linking references");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001654 return 1;
1655 }
1656
Adam Lesinskib522f042017-04-21 16:57:59 -07001657 if (context_->GetPackageType() == PackageType::kStaticLib) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001658 if (!options_.products.empty()) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001659 context_->GetDiagnostics()->Warn(DiagMessage()
1660 << "can't select products when building static library");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001661 }
1662 } else {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001663 ProductFilter product_filter(options_.products);
1664 if (!product_filter.Consume(context_, &final_table_)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001665 context_->GetDiagnostics()->Error(DiagMessage() << "failed stripping products");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001666 return 1;
1667 }
1668 }
1669
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001670 if (!options_.no_auto_version) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001671 AutoVersioner versioner;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001672 if (!versioner.Consume(context_, &final_table_)) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001673 context_->GetDiagnostics()->Error(DiagMessage() << "failed versioning styles");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001674 return 1;
1675 }
1676 }
1677
Adam Lesinskib522f042017-04-21 16:57:59 -07001678 if (context_->GetPackageType() != PackageType::kStaticLib && context_->GetMinSdkVersion() > 0) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001679 if (context_->IsVerbose()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001680 context_->GetDiagnostics()->Note(DiagMessage()
1681 << "collapsing resource versions for minimum SDK "
1682 << context_->GetMinSdkVersion());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001683 }
1684
1685 VersionCollapser collapser;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001686 if (!collapser.Consume(context_, &final_table_)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001687 return 1;
1688 }
1689 }
1690
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001691 if (!options_.no_resource_deduping) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001692 ResourceDeduper deduper;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001693 if (!deduper.Consume(context_, &final_table_)) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001694 context_->GetDiagnostics()->Error(DiagMessage() << "failed deduping resources");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001695 return 1;
1696 }
1697 }
1698
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001699 proguard::KeepSet proguard_keep_set;
1700 proguard::KeepSet proguard_main_dex_keep_set;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001701
Adam Lesinskib522f042017-04-21 16:57:59 -07001702 if (context_->GetPackageType() == PackageType::kStaticLib) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001703 if (options_.table_splitter_options.config_filter != nullptr ||
Pierre Lecesne672384b2017-02-06 10:29:02 +00001704 !options_.table_splitter_options.preferred_densities.empty()) {
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001705 context_->GetDiagnostics()->Warn(DiagMessage()
1706 << "can't strip resources when building static library");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001707 }
1708 } else {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001709 // Adjust the SplitConstraints so that their SDK version is stripped if it is less than or
1710 // equal to the minSdk.
1711 options_.split_constraints =
1712 AdjustSplitConstraintsForMinSdk(context_->GetMinSdkVersion(), options_.split_constraints);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001713
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001714 TableSplitter table_splitter(options_.split_constraints, options_.table_splitter_options);
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001715 if (!table_splitter.VerifySplitConstraints(context_)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001716 return 1;
1717 }
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001718 table_splitter.SplitTable(&final_table_);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001719
1720 // Now we need to write out the Split APKs.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001721 auto path_iter = options_.split_paths.begin();
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001722 auto split_constraints_iter = options_.split_constraints.begin();
1723 for (std::unique_ptr<ResourceTable>& split_table : table_splitter.splits()) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001724 if (context_->IsVerbose()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001725 context_->GetDiagnostics()->Note(DiagMessage(*path_iter)
1726 << "generating split with configurations '"
1727 << util::Joiner(split_constraints_iter->configs, ", ")
1728 << "'");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001729 }
1730
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001731 std::unique_ptr<IArchiveWriter> archive_writer = MakeArchiveWriter(*path_iter);
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001732 if (!archive_writer) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001733 context_->GetDiagnostics()->Error(DiagMessage() << "failed to create archive");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001734 return 1;
1735 }
1736
1737 // Generate an AndroidManifest.xml for each split.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001738 std::unique_ptr<xml::XmlResource> split_manifest =
1739 GenerateSplitManifest(app_info, *split_constraints_iter);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001740
1741 XmlReferenceLinker linker;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001742 if (!linker.Consume(context_, split_manifest.get())) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001743 context_->GetDiagnostics()->Error(DiagMessage()
1744 << "failed to create Split AndroidManifest.xml");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001745 return 1;
1746 }
1747
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001748 if (!WriteApk(archive_writer.get(), &proguard_keep_set, split_manifest.get(),
1749 split_table.get())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001750 return 1;
1751 }
1752
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001753 ++path_iter;
1754 ++split_constraints_iter;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001755 }
1756 }
1757
1758 // Start writing the base APK.
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001759 std::unique_ptr<IArchiveWriter> archive_writer = MakeArchiveWriter(options_.output_path);
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001760 if (!archive_writer) {
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001761 context_->GetDiagnostics()->Error(DiagMessage() << "failed to create archive");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001762 return 1;
1763 }
1764
1765 bool error = false;
1766 {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001767 // AndroidManifest.xml has no resource name, but the CallSite is built from the name
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001768 // (aka, which package the AndroidManifest.xml is coming from).
1769 // So we give it a package name so it can see local resources.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001770 manifest_xml->file.name.package = context_->GetCompilationPackage();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001771
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001772 XmlReferenceLinker manifest_linker;
1773 if (manifest_linker.Consume(context_, manifest_xml.get())) {
1774 if (options_.generate_proguard_rules_path &&
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001775 !proguard::CollectProguardRulesForManifest(Source(options_.manifest_path),
1776 manifest_xml.get(), &proguard_keep_set)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001777 error = true;
1778 }
1779
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001780 if (options_.generate_main_dex_proguard_rules_path &&
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001781 !proguard::CollectProguardRulesForManifest(Source(options_.manifest_path),
1782 manifest_xml.get(),
1783 &proguard_main_dex_keep_set, true)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001784 error = true;
Alexandria Cornwall637b4822016-08-11 09:53:16 -07001785 }
1786
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001787 if (options_.generate_java_class_path) {
1788 if (!WriteManifestJavaFile(manifest_xml.get())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001789 error = true;
1790 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -07001791 }
1792
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001793 if (options_.no_xml_namespaces) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001794 // PackageParser will fail if URIs are removed from
1795 // AndroidManifest.xml.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001796 XmlNamespaceRemover namespace_remover(true /* keepUris */);
1797 if (!namespace_remover.Consume(context_, manifest_xml.get())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001798 error = true;
1799 }
Rohit Agrawale49bb302016-04-22 12:27:55 -07001800 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001801 } else {
1802 error = true;
1803 }
Adam Lesinski1ab598f2015-08-14 14:26:04 -07001804 }
Adam Lesinskifb48d292015-11-07 15:52:13 -08001805
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001806 if (error) {
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001807 context_->GetDiagnostics()->Error(DiagMessage() << "failed processing manifest");
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001808 return 1;
1809 }
Adam Lesinskia6fe3452015-12-09 15:20:52 -08001810
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001811 if (!WriteApk(archive_writer.get(), &proguard_keep_set, manifest_xml.get(), &final_table_)) {
1812 return 1;
1813 }
1814
1815 if (!CopyAssetsDirsToApk(archive_writer.get())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001816 return 1;
1817 }
Adam Lesinskifb48d292015-11-07 15:52:13 -08001818
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001819 if (options_.generate_java_class_path) {
Adam Lesinski1ef0fa92017-08-15 21:32:49 -07001820 if (!GenerateJavaClasses()) {
Adam Lesinskib5dc4bd2017-02-22 19:29:29 -08001821 return 1;
1822 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001823 }
1824
Adam Lesinskib5dc4bd2017-02-22 19:29:29 -08001825 if (!WriteProguardFile(options_.generate_proguard_rules_path, proguard_keep_set)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001826 return 1;
1827 }
1828
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001829 if (!WriteProguardFile(options_.generate_main_dex_proguard_rules_path,
1830 proguard_main_dex_keep_set)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001831 return 1;
1832 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001833 return 0;
1834 }
1835
1836 private:
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001837 LinkOptions options_;
1838 LinkContext* context_;
1839 ResourceTable final_table_;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001840
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001841 std::unique_ptr<TableMerger> table_merger_;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001842
1843 // A pointer to the FileCollection representing the filesystem (not archives).
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001844 std::unique_ptr<io::FileCollection> file_collection_;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001845
1846 // A vector of IFileCollections. This is mainly here to keep ownership of the
1847 // collections.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001848 std::vector<std::unique_ptr<io::IFileCollection>> collections_;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001849
1850 // A vector of ResourceTables. This is here to retain ownership, so that the
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001851 // SymbolTable can use these.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001852 std::vector<std::unique_ptr<ResourceTable>> static_table_includes_;
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001853
1854 // The set of shared libraries being used, mapping their assigned package ID to package name.
1855 std::map<size_t, std::string> shared_libs_;
Adam Lesinski1ab598f2015-08-14 14:26:04 -07001856};
1857
Chris Warrington820d72a2017-04-27 15:27:01 +01001858int Link(const std::vector<StringPiece>& args, IDiagnostics* diagnostics) {
1859 LinkContext context(diagnostics);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001860 LinkOptions options;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001861 std::vector<std::string> overlay_arg_list;
1862 std::vector<std::string> extra_java_packages;
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001863 Maybe<std::string> package_id;
Adam Lesinski113ee092017-04-03 19:38:25 -07001864 std::vector<std::string> configs;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001865 Maybe<std::string> preferred_density;
1866 Maybe<std::string> product_list;
1867 bool legacy_x_flag = false;
1868 bool require_localization = false;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001869 bool verbose = false;
Adam Lesinskiceb9b2f2017-02-16 12:05:42 -08001870 bool shared_lib = false;
1871 bool static_lib = false;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001872 Maybe<std::string> stable_id_file_path;
1873 std::vector<std::string> split_args;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001874 Flags flags =
1875 Flags()
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001876 .RequiredFlag("-o", "Output path.", &options.output_path)
1877 .RequiredFlag("--manifest", "Path to the Android manifest to build.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001878 &options.manifest_path)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001879 .OptionalFlagList("-I", "Adds an Android APK to link against.", &options.include_paths)
Adam Lesinskib39ad7c2017-03-13 11:40:48 -07001880 .OptionalFlagList("-A",
1881 "An assets directory to include in the APK. These are unprocessed.",
1882 &options.assets_dirs)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001883 .OptionalFlagList("-R",
1884 "Compilation unit to link, using `overlay` semantics.\n"
1885 "The last conflicting resource given takes precedence.",
1886 &overlay_arg_list)
Adam Lesinskif34b6f42017-03-03 16:33:26 -08001887 .OptionalFlag("--package-id",
1888 "Specify the package ID to use for this app. Must be greater or equal to\n"
1889 "0x7f and can't be used with --static-lib or --shared-lib.",
1890 &package_id)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001891 .OptionalFlag("--java", "Directory in which to generate R.java.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001892 &options.generate_java_class_path)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001893 .OptionalFlag("--proguard", "Output file for generated Proguard rules.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001894 &options.generate_proguard_rules_path)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001895 .OptionalFlag("--proguard-main-dex",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001896 "Output file for generated Proguard rules for the main dex.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001897 &options.generate_main_dex_proguard_rules_path)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001898 .OptionalSwitch("--no-auto-version",
1899 "Disables automatic style and layout SDK versioning.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001900 &options.no_auto_version)
1901 .OptionalSwitch("--no-version-vectors",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001902 "Disables automatic versioning of vector drawables. Use this only\n"
1903 "when building with vector drawable support library.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001904 &options.no_version_vectors)
Yuichi Araki4d35cca2017-01-18 20:42:17 +09001905 .OptionalSwitch("--no-version-transitions",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001906 "Disables automatic versioning of transition resources. Use this only\n"
1907 "when building with transition support library.",
Yuichi Araki4d35cca2017-01-18 20:42:17 +09001908 &options.no_version_transitions)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001909 .OptionalSwitch("--no-resource-deduping",
Adam Lesinskicacb28f2016-10-19 12:18:14 -07001910 "Disables automatic deduping of resources with\n"
1911 "identical values across compatible configurations.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001912 &options.no_resource_deduping)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001913 .OptionalSwitch("--enable-sparse-encoding",
1914 "Enables encoding sparse entries using a binary search tree.\n"
1915 "This decreases APK size at the cost of resource retrieval performance.",
1916 &options.table_flattener_options.use_sparse_entries)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001917 .OptionalSwitch("-x", "Legacy flag that specifies to use the package identifier 0x01.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001918 &legacy_x_flag)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001919 .OptionalSwitch("-z", "Require localization of strings marked 'suggested'.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001920 &require_localization)
Adam Lesinski113ee092017-04-03 19:38:25 -07001921 .OptionalFlagList("-c",
Adam Lesinski418763f2017-04-11 17:36:53 -07001922 "Comma separated list of configurations to include. The default\n"
1923 "is all configurations.",
1924 &configs)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001925 .OptionalFlag("--preferred-density",
1926 "Selects the closest matching density and strips out all others.",
1927 &preferred_density)
1928 .OptionalFlag("--product", "Comma separated list of product names to keep", &product_list)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001929 .OptionalSwitch("--output-to-dir",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001930 "Outputs the APK contents to a directory specified by -o.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001931 &options.output_to_directory)
1932 .OptionalSwitch("--no-xml-namespaces",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001933 "Removes XML namespace prefix and URI information from\n"
1934 "AndroidManifest.xml and XML binaries in res/*.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001935 &options.no_xml_namespaces)
1936 .OptionalFlag("--min-sdk-version",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001937 "Default minimum SDK version to use for AndroidManifest.xml.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001938 &options.manifest_fixer_options.min_sdk_version_default)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001939 .OptionalFlag("--target-sdk-version",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001940 "Default target SDK version to use for AndroidManifest.xml.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001941 &options.manifest_fixer_options.target_sdk_version_default)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001942 .OptionalFlag("--version-code",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001943 "Version code (integer) to inject into the AndroidManifest.xml if none is\n"
1944 "present.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001945 &options.manifest_fixer_options.version_code_default)
1946 .OptionalFlag("--version-name",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001947 "Version name to inject into the AndroidManifest.xml if none is present.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001948 &options.manifest_fixer_options.version_name_default)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001949 .OptionalSwitch("--shared-lib", "Generates a shared Android runtime library.",
1950 &shared_lib)
1951 .OptionalSwitch("--static-lib", "Generate a static Android library.", &static_lib)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001952 .OptionalSwitch("--no-static-lib-packages",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001953 "Merge all library resources under the app's package.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001954 &options.no_static_lib_packages)
1955 .OptionalSwitch("--non-final-ids",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001956 "Generates R.java without the final modifier. This is implied when\n"
1957 "--static-lib is specified.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001958 &options.generate_non_final_ids)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001959 .OptionalFlag("--stable-ids", "File containing a list of name to ID mapping.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001960 &stable_id_file_path)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001961 .OptionalFlag("--emit-ids",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001962 "Emit a file at the given path with a list of name to ID mappings,\n"
1963 "suitable for use with --stable-ids.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001964 &options.resource_id_map_path)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001965 .OptionalFlag("--private-symbols",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001966 "Package name to use when generating R.java for private symbols.\n"
1967 "If not specified, public and private symbols will use the application's\n"
1968 "package name.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001969 &options.private_symbols)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001970 .OptionalFlag("--custom-package", "Custom Java package under which to generate R.java.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001971 &options.custom_java_package)
1972 .OptionalFlagList("--extra-packages",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001973 "Generate the same R.java but with different package names.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001974 &extra_java_packages)
1975 .OptionalFlagList("--add-javadoc-annotation",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001976 "Adds a JavaDoc annotation to all generated Java classes.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001977 &options.javadoc_annotations)
Adam Lesinski418763f2017-04-11 17:36:53 -07001978 .OptionalFlag("--output-text-symbols",
1979 "Generates a text file containing the resource symbols of the R class in\n"
1980 "the specified folder.",
1981 &options.generate_text_symbols_path)
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001982 .OptionalSwitch("--auto-add-overlay",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001983 "Allows the addition of new resources in overlays without\n"
1984 "<add-resource> tags.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001985 &options.auto_add_overlay)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001986 .OptionalFlag("--rename-manifest-package", "Renames the package in AndroidManifest.xml.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001987 &options.manifest_fixer_options.rename_manifest_package)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001988 .OptionalFlag("--rename-instrumentation-target-package",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001989 "Changes the name of the target package for instrumentation. Most useful\n"
1990 "when used in conjunction with --rename-manifest-package.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001991 &options.manifest_fixer_options.rename_instrumentation_target_package)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001992 .OptionalFlagList("-0", "File extensions not to compress.",
Adam Lesinskice5e56e2016-10-21 17:56:45 -07001993 &options.extensions_to_not_compress)
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001994 .OptionalFlagList("--split",
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001995 "Split resources matching a set of configs out to a Split APK.\n"
Adam Lesinskidb091572017-04-13 12:48:56 -07001996 "Syntax: path/to/output.apk:<config>[,<config>[...]].\n"
1997 "On Windows, use a semicolon ';' separator instead.",
Adam Lesinskic8f71aa2017-02-08 07:03:50 -08001998 &split_args)
Adam Lesinskid0f492d2017-04-03 18:12:45 -07001999 .OptionalSwitch("-v", "Enables verbose logging.", &verbose);
Adam Lesinski1ab598f2015-08-14 14:26:04 -07002000
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002001 if (!flags.Parse("aapt2 link", args, &std::cerr)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002002 return 1;
2003 }
2004
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002005 // Expand all argument-files passed into the command line. These start with '@'.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002006 std::vector<std::string> arg_list;
2007 for (const std::string& arg : flags.GetArgs()) {
2008 if (util::StartsWith(arg, "@")) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002009 const std::string path = arg.substr(1, arg.size() - 1);
2010 std::string error;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002011 if (!file::AppendArgsFromFile(path, &arg_list, &error)) {
2012 context.GetDiagnostics()->Error(DiagMessage(path) << error);
Adam Lesinski1ab598f2015-08-14 14:26:04 -07002013 return 1;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002014 }
2015 } else {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002016 arg_list.push_back(arg);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002017 }
2018 }
2019
2020 // Expand all argument-files passed to -R.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002021 for (const std::string& arg : overlay_arg_list) {
2022 if (util::StartsWith(arg, "@")) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002023 const std::string path = arg.substr(1, arg.size() - 1);
2024 std::string error;
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002025 if (!file::AppendArgsFromFile(path, &options.overlay_files, &error)) {
2026 context.GetDiagnostics()->Error(DiagMessage(path) << error);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002027 return 1;
2028 }
2029 } else {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002030 options.overlay_files.push_back(arg);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002031 }
2032 }
2033
2034 if (verbose) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002035 context.SetVerbose(verbose);
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002036 }
2037
Adam Lesinskif34b6f42017-03-03 16:33:26 -08002038 if (shared_lib && static_lib) {
2039 context.GetDiagnostics()->Error(DiagMessage()
2040 << "only one of --shared-lib and --static-lib can be defined");
2041 return 1;
2042 }
2043
2044 if (shared_lib) {
Adam Lesinskib522f042017-04-21 16:57:59 -07002045 context.SetPackageType(PackageType::kSharedLib);
Adam Lesinskif34b6f42017-03-03 16:33:26 -08002046 context.SetPackageId(0x00);
2047 } else if (static_lib) {
Adam Lesinskib522f042017-04-21 16:57:59 -07002048 context.SetPackageType(PackageType::kStaticLib);
Adam Lesinskif34b6f42017-03-03 16:33:26 -08002049 context.SetPackageId(kAppPackageId);
2050 } else {
Adam Lesinskib522f042017-04-21 16:57:59 -07002051 context.SetPackageType(PackageType::kApp);
Adam Lesinskif34b6f42017-03-03 16:33:26 -08002052 context.SetPackageId(kAppPackageId);
2053 }
2054
2055 if (package_id) {
Adam Lesinskib522f042017-04-21 16:57:59 -07002056 if (context.GetPackageType() != PackageType::kApp) {
Adam Lesinskif34b6f42017-03-03 16:33:26 -08002057 context.GetDiagnostics()->Error(
2058 DiagMessage() << "can't specify --package-id when not building a regular app");
2059 return 1;
2060 }
2061
2062 const Maybe<uint32_t> maybe_package_id_int = ResourceUtils::ParseInt(package_id.value());
2063 if (!maybe_package_id_int) {
2064 context.GetDiagnostics()->Error(DiagMessage() << "package ID '" << package_id.value()
2065 << "' is not a valid integer");
2066 return 1;
2067 }
2068
2069 const uint32_t package_id_int = maybe_package_id_int.value();
2070 if (package_id_int < kAppPackageId || package_id_int > std::numeric_limits<uint8_t>::max()) {
2071 context.GetDiagnostics()->Error(
2072 DiagMessage() << StringPrintf(
2073 "invalid package ID 0x%02x. Must be in the range 0x7f-0xff.", package_id_int));
2074 return 1;
2075 }
2076 context.SetPackageId(static_cast<uint8_t>(package_id_int));
2077 }
2078
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002079 // Populate the set of extra packages for which to generate R.java.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002080 for (std::string& extra_package : extra_java_packages) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002081 // A given package can actually be a colon separated list of packages.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002082 for (StringPiece package : util::Split(extra_package, ':')) {
Adam Lesinskid5083f62017-01-16 15:07:21 -08002083 options.extra_java_packages.insert(package.to_string());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002084 }
2085 }
2086
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002087 if (product_list) {
2088 for (StringPiece product : util::Tokenize(product_list.value(), ',')) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002089 if (product != "" && product != "default") {
Adam Lesinskid5083f62017-01-16 15:07:21 -08002090 options.products.insert(product.to_string());
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002091 }
2092 }
2093 }
2094
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002095 std::unique_ptr<IConfigFilter> filter;
Mihai Nitaf4dacf22017-04-07 08:25:06 -07002096 if (!configs.empty()) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002097 filter = ParseConfigFilterParameters(configs, context.GetDiagnostics());
2098 if (filter == nullptr) {
2099 return 1;
Adam Lesinski1ab598f2015-08-14 14:26:04 -07002100 }
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002101 options.table_splitter_options.config_filter = filter.get();
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002102 }
2103
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002104 if (preferred_density) {
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002105 Maybe<uint16_t> density =
2106 ParseTargetDensityParameter(preferred_density.value(), context.GetDiagnostics());
2107 if (!density) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002108 return 1;
Adam Lesinskic51562c2016-04-28 11:12:38 -07002109 }
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002110 options.table_splitter_options.preferred_densities.push_back(density.value());
2111 }
Adam Lesinskic51562c2016-04-28 11:12:38 -07002112
Adam Lesinskid0f492d2017-04-03 18:12:45 -07002113 // Parse the split parameters.
2114 for (const std::string& split_arg : split_args) {
2115 options.split_paths.push_back({});
2116 options.split_constraints.push_back({});
2117 if (!ParseSplitParameter(split_arg, context.GetDiagnostics(), &options.split_paths.back(),
2118 &options.split_constraints.back())) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002119 return 1;
Adam Lesinski1e21ff02016-06-24 14:57:58 -07002120 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002121 }
Adam Lesinski1e21ff02016-06-24 14:57:58 -07002122
Adam Lesinskib522f042017-04-21 16:57:59 -07002123 if (context.GetPackageType() != PackageType::kStaticLib && stable_id_file_path) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002124 if (!LoadStableIdMap(context.GetDiagnostics(), stable_id_file_path.value(),
2125 &options.stable_id_map)) {
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002126 return 1;
Adam Lesinski64587af2016-02-18 18:33:06 -08002127 }
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002128 }
Adam Lesinski64587af2016-02-18 18:33:06 -08002129
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002130 // Populate some default no-compress extensions that are already compressed.
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002131 options.extensions_to_not_compress.insert(
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002132 {".jpg", ".jpeg", ".png", ".gif", ".wav", ".mp2", ".mp3", ".ogg",
2133 ".aac", ".mpg", ".mpeg", ".mid", ".midi", ".smf", ".jet", ".rtttl",
2134 ".imy", ".xmf", ".mp4", ".m4a", ".m4v", ".3gp", ".3gpp", ".3g2",
2135 ".3gpp2", ".amr", ".awb", ".wma", ".wmv", ".webm", ".mkv"});
2136
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002137 // Turn off auto versioning for static-libs.
Adam Lesinskib522f042017-04-21 16:57:59 -07002138 if (context.GetPackageType() == PackageType::kStaticLib) {
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002139 options.no_auto_version = true;
2140 options.no_version_vectors = true;
Yuichi Araki4d35cca2017-01-18 20:42:17 +09002141 options.no_version_transitions = true;
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002142 }
Adam Lesinskie4bb9eb2016-02-12 22:18:51 -08002143
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002144 LinkCommand cmd(&context, options);
Adam Lesinskice5e56e2016-10-21 17:56:45 -07002145 return cmd.Run(arg_list);
Adam Lesinski1ab598f2015-08-14 14:26:04 -07002146}
2147
Adam Lesinskicacb28f2016-10-19 12:18:14 -07002148} // namespace aapt