Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1 | /* |
| 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 Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 17 | #include <sys/stat.h> |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 18 | #include <cinttypes> |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 19 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 20 | #include <queue> |
| 21 | #include <unordered_map> |
| 22 | #include <vector> |
| 23 | |
| 24 | #include "android-base/errors.h" |
| 25 | #include "android-base/file.h" |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 26 | #include "android-base/stringprintf.h" |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 27 | #include "androidfw/StringPiece.h" |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 28 | |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 29 | #include "AppInfo.h" |
| 30 | #include "Debug.h" |
| 31 | #include "Flags.h" |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 32 | #include "LoadedApk.h" |
Adam Lesinski | 6a00817 | 2016-02-02 17:02:58 -0800 | [diff] [blame] | 33 | #include "Locale.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 34 | #include "NameMangler.h" |
Adam Lesinski | 59e04c6 | 2016-02-04 15:59:23 -0800 | [diff] [blame] | 35 | #include "ResourceUtils.h" |
Adam Lesinski | d3ffa844 | 2017-09-28 13:34:35 -0700 | [diff] [blame] | 36 | #include "ResourceValues.h" |
| 37 | #include "ValueVisitor.h" |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 38 | #include "cmd/Util.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 39 | #include "compile/IdAssigner.h" |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 40 | #include "compile/XmlIdCollector.h" |
Adam Lesinski | 6a00817 | 2016-02-02 17:02:58 -0800 | [diff] [blame] | 41 | #include "filter/ConfigFilter.h" |
Adam Lesinski | 4670805 | 2017-09-29 14:49:15 -0700 | [diff] [blame] | 42 | #include "format/Archive.h" |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 43 | #include "format/Container.h" |
Adam Lesinski | 4670805 | 2017-09-29 14:49:15 -0700 | [diff] [blame] | 44 | #include "format/binary/TableFlattener.h" |
| 45 | #include "format/binary/XmlFlattener.h" |
| 46 | #include "format/proto/ProtoDeserialize.h" |
| 47 | #include "format/proto/ProtoSerialize.h" |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 48 | #include "io/BigBufferStream.h" |
| 49 | #include "io/FileStream.h" |
Adam Lesinski | a40e972 | 2015-11-24 19:11:46 -0800 | [diff] [blame] | 50 | #include "io/FileSystem.h" |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 51 | #include "io/Util.h" |
Adam Lesinski | a40e972 | 2015-11-24 19:11:46 -0800 | [diff] [blame] | 52 | #include "io/ZipArchive.h" |
Adam Lesinski | ca5638f | 2015-10-21 14:42:43 -0700 | [diff] [blame] | 53 | #include "java/JavaClassGenerator.h" |
| 54 | #include "java/ManifestClassGenerator.h" |
| 55 | #include "java/ProguardRules.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 56 | #include "link/Linkers.h" |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 57 | #include "link/ManifestFixer.h" |
Adam Lesinski | 34a1687 | 2018-02-23 16:18:10 -0800 | [diff] [blame] | 58 | #include "link/NoDefaultResourceRemover.h" |
Adam Lesinski | 467f171 | 2015-11-16 17:35:44 -0800 | [diff] [blame] | 59 | #include "link/ReferenceLinker.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 60 | #include "link/TableMerger.h" |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 61 | #include "link/XmlCompatVersioner.h" |
Adam Lesinski | d48944a | 2017-02-21 14:22:30 -0800 | [diff] [blame] | 62 | #include "optimize/ResourceDeduper.h" |
| 63 | #include "optimize/VersionCollapser.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 64 | #include "process/IResourceTableConsumer.h" |
| 65 | #include "process/SymbolTable.h" |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 66 | #include "split/TableSplitter.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 67 | #include "util/Files.h" |
Adam Lesinski | 467f171 | 2015-11-16 17:35:44 -0800 | [diff] [blame] | 68 | #include "xml/XmlDom.h" |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 69 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 70 | using ::aapt::io::FileInputStream; |
| 71 | using ::android::StringPiece; |
| 72 | using ::android::base::StringPrintf; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 73 | |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 74 | namespace aapt { |
| 75 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 76 | enum class OutputFormat { |
| 77 | kApk, |
| 78 | kProto, |
| 79 | }; |
| 80 | |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 81 | struct LinkOptions { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 82 | std::string output_path; |
| 83 | std::string manifest_path; |
| 84 | std::vector<std::string> include_paths; |
| 85 | std::vector<std::string> overlay_files; |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 86 | std::vector<std::string> assets_dirs; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 87 | bool output_to_directory = false; |
| 88 | bool auto_add_overlay = false; |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 89 | OutputFormat output_format = OutputFormat::kApk; |
Adam Lesinski | 36c73a5 | 2016-08-11 13:39:24 -0700 | [diff] [blame] | 90 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 91 | // Java/Proguard options. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 92 | Maybe<std::string> generate_java_class_path; |
| 93 | Maybe<std::string> custom_java_package; |
| 94 | std::set<std::string> extra_java_packages; |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 95 | Maybe<std::string> generate_text_symbols_path; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 96 | Maybe<std::string> generate_proguard_rules_path; |
| 97 | Maybe<std::string> generate_main_dex_proguard_rules_path; |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 98 | bool generate_conditional_proguard_rules = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 99 | bool generate_non_final_ids = false; |
| 100 | std::vector<std::string> javadoc_annotations; |
| 101 | Maybe<std::string> private_symbols; |
Adam Lesinski | 36c73a5 | 2016-08-11 13:39:24 -0700 | [diff] [blame] | 102 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 103 | // Optimizations/features. |
| 104 | bool no_auto_version = false; |
| 105 | bool no_version_vectors = false; |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 106 | bool no_version_transitions = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 107 | bool no_resource_deduping = false; |
| 108 | bool no_xml_namespaces = false; |
| 109 | bool do_not_compress_anything = false; |
| 110 | std::unordered_set<std::string> extensions_to_not_compress; |
| 111 | |
| 112 | // Static lib options. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 113 | bool no_static_lib_packages = false; |
| 114 | |
| 115 | // AndroidManifest.xml massaging options. |
| 116 | ManifestFixerOptions manifest_fixer_options; |
| 117 | |
| 118 | // Products to use/filter on. |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 119 | std::unordered_set<std::string> products; |
Adam Lesinski | 36c73a5 | 2016-08-11 13:39:24 -0700 | [diff] [blame] | 120 | |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 121 | // Flattening options. |
| 122 | TableFlattenerOptions table_flattener_options; |
| 123 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 124 | // Split APK options. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 125 | TableSplitterOptions table_splitter_options; |
| 126 | std::vector<SplitConstraints> split_constraints; |
| 127 | std::vector<std::string> split_paths; |
Adam Lesinski | 36c73a5 | 2016-08-11 13:39:24 -0700 | [diff] [blame] | 128 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 129 | // Stable ID options. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 130 | std::unordered_map<ResourceName, ResourceId> stable_id_map; |
| 131 | Maybe<std::string> resource_id_map_path; |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 132 | }; |
| 133 | |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 134 | class LinkContext : public IAaptContext { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 135 | public: |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 136 | LinkContext(IDiagnostics* diagnostics) |
| 137 | : diagnostics_(diagnostics), name_mangler_({}), symbols_(&name_mangler_) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 138 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 139 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 140 | PackageType GetPackageType() override { |
| 141 | return package_type_; |
| 142 | } |
| 143 | |
| 144 | void SetPackageType(PackageType type) { |
| 145 | package_type_ = type; |
| 146 | } |
| 147 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 148 | IDiagnostics* GetDiagnostics() override { |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 149 | return diagnostics_; |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 150 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 151 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 152 | NameMangler* GetNameMangler() override { |
| 153 | return &name_mangler_; |
| 154 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 155 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 156 | void SetNameManglerPolicy(const NameManglerPolicy& policy) { |
| 157 | name_mangler_ = NameMangler(policy); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 158 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 159 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 160 | const std::string& GetCompilationPackage() override { |
| 161 | return compilation_package_; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 162 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 163 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 164 | void SetCompilationPackage(const StringPiece& package_name) { |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 165 | compilation_package_ = package_name.to_string(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 166 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 167 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 168 | uint8_t GetPackageId() override { |
| 169 | return package_id_; |
| 170 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 171 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 172 | void SetPackageId(uint8_t id) { |
| 173 | package_id_ = id; |
| 174 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 175 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 176 | SymbolTable* GetExternalSymbols() override { |
| 177 | return &symbols_; |
| 178 | } |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 179 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 180 | bool IsVerbose() override { |
| 181 | return verbose_; |
| 182 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 183 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 184 | void SetVerbose(bool val) { |
| 185 | verbose_ = val; |
| 186 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 187 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 188 | int GetMinSdkVersion() override { |
| 189 | return min_sdk_version_; |
| 190 | } |
Adam Lesinski | fb6312f | 2016-06-28 14:40:32 -0700 | [diff] [blame] | 191 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 192 | void SetMinSdkVersion(int minSdk) { |
| 193 | min_sdk_version_ = minSdk; |
| 194 | } |
Adam Lesinski | fb6312f | 2016-06-28 14:40:32 -0700 | [diff] [blame] | 195 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 196 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 197 | DISALLOW_COPY_AND_ASSIGN(LinkContext); |
| 198 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 199 | PackageType package_type_ = PackageType::kApp; |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 200 | IDiagnostics* diagnostics_; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 201 | NameMangler name_mangler_; |
| 202 | std::string compilation_package_; |
| 203 | uint8_t package_id_ = 0x0; |
| 204 | SymbolTable symbols_; |
| 205 | bool verbose_ = false; |
| 206 | int min_sdk_version_ = 0; |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 207 | }; |
| 208 | |
Adam Lesinski | 1e4b0e5 | 2017-04-27 15:01:10 -0700 | [diff] [blame] | 209 | // A custom delegate that generates compatible pre-O IDs for use with feature splits. |
| 210 | // Feature splits use package IDs > 7f, which in Java (since Java doesn't have unsigned ints) |
| 211 | // is interpreted as a negative number. Some verification was wrongly assuming negative values |
| 212 | // were invalid. |
| 213 | // |
| 214 | // This delegate will attempt to masquerade any '@id/' references with ID 0xPPTTEEEE, |
| 215 | // where PP > 7f, as 0x7fPPEEEE. Any potential overlapping is verified and an error occurs if such |
| 216 | // an overlap exists. |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 217 | // |
| 218 | // See b/37498913. |
Adam Lesinski | 1e4b0e5 | 2017-04-27 15:01:10 -0700 | [diff] [blame] | 219 | class FeatureSplitSymbolTableDelegate : public DefaultSymbolTableDelegate { |
| 220 | public: |
| 221 | FeatureSplitSymbolTableDelegate(IAaptContext* context) : context_(context) { |
| 222 | } |
| 223 | |
| 224 | virtual ~FeatureSplitSymbolTableDelegate() = default; |
| 225 | |
| 226 | virtual std::unique_ptr<SymbolTable::Symbol> FindByName( |
| 227 | const ResourceName& name, |
| 228 | const std::vector<std::unique_ptr<ISymbolSource>>& sources) override { |
| 229 | std::unique_ptr<SymbolTable::Symbol> symbol = |
| 230 | DefaultSymbolTableDelegate::FindByName(name, sources); |
| 231 | if (symbol == nullptr) { |
| 232 | return {}; |
| 233 | } |
| 234 | |
| 235 | // Check to see if this is an 'id' with the target package. |
| 236 | if (name.type == ResourceType::kId && symbol->id) { |
| 237 | ResourceId* id = &symbol->id.value(); |
| 238 | if (id->package_id() > kAppPackageId) { |
| 239 | // Rewrite the resource ID to be compatible pre-O. |
| 240 | ResourceId rewritten_id(kAppPackageId, id->package_id(), id->entry_id()); |
| 241 | |
| 242 | // Check that this doesn't overlap another resource. |
| 243 | if (DefaultSymbolTableDelegate::FindById(rewritten_id, sources) != nullptr) { |
| 244 | // The ID overlaps, so log a message (since this is a weird failure) and fail. |
| 245 | context_->GetDiagnostics()->Error(DiagMessage() << "Failed to rewrite " << name |
| 246 | << " for pre-O feature split support"); |
| 247 | return {}; |
| 248 | } |
| 249 | |
| 250 | if (context_->IsVerbose()) { |
| 251 | context_->GetDiagnostics()->Note(DiagMessage() << "rewriting " << name << " (" << *id |
| 252 | << ") -> (" << rewritten_id << ")"); |
| 253 | } |
| 254 | |
| 255 | *id = rewritten_id; |
| 256 | } |
| 257 | } |
| 258 | return symbol; |
| 259 | } |
| 260 | |
| 261 | private: |
| 262 | DISALLOW_COPY_AND_ASSIGN(FeatureSplitSymbolTableDelegate); |
| 263 | |
| 264 | IAaptContext* context_; |
| 265 | }; |
| 266 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 267 | static bool FlattenXml(IAaptContext* context, const xml::XmlResource& xml_res, |
| 268 | const StringPiece& path, bool keep_raw_values, bool utf16, |
| 269 | OutputFormat format, IArchiveWriter* writer) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 270 | if (context->IsVerbose()) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 271 | context->GetDiagnostics()->Note(DiagMessage(path) << "writing to archive (keep_raw_values=" |
| 272 | << (keep_raw_values ? "true" : "false") |
| 273 | << ")"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 274 | } |
| 275 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 276 | switch (format) { |
| 277 | case OutputFormat::kApk: { |
| 278 | BigBuffer buffer(1024); |
| 279 | XmlFlattenerOptions options = {}; |
| 280 | options.keep_raw_values = keep_raw_values; |
| 281 | options.use_utf16 = utf16; |
| 282 | XmlFlattener flattener(&buffer, options); |
| 283 | if (!flattener.Consume(context, &xml_res)) { |
| 284 | return false; |
| 285 | } |
| 286 | |
| 287 | io::BigBufferInputStream input_stream(&buffer); |
| 288 | return io::CopyInputStreamToArchive(context, &input_stream, path.to_string(), |
| 289 | ArchiveEntry::kCompress, writer); |
| 290 | } break; |
| 291 | |
| 292 | case OutputFormat::kProto: { |
| 293 | pb::XmlNode pb_node; |
| 294 | SerializeXmlResourceToPb(xml_res, &pb_node); |
| 295 | return io::CopyProtoToArchive(context, &pb_node, path.to_string(), ArchiveEntry::kCompress, |
| 296 | writer); |
| 297 | } break; |
| 298 | } |
| 299 | return false; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 300 | } |
| 301 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 302 | // Inflates an XML file from the source path. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 303 | static std::unique_ptr<xml::XmlResource> LoadXml(const std::string& path, IDiagnostics* diag) { |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 304 | FileInputStream fin(path); |
| 305 | if (fin.HadError()) { |
| 306 | diag->Error(DiagMessage(path) << "failed to load XML file: " << fin.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 307 | return {}; |
| 308 | } |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 309 | return xml::Inflate(&fin, diag, Source(path)); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 310 | } |
| 311 | |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 312 | struct ResourceFileFlattenerOptions { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 313 | bool no_auto_version = false; |
| 314 | bool no_version_vectors = false; |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 315 | bool no_version_transitions = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 316 | bool no_xml_namespaces = false; |
| 317 | bool keep_raw_values = false; |
| 318 | bool do_not_compress_anything = false; |
| 319 | bool update_proguard_spec = false; |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 320 | OutputFormat output_format = OutputFormat::kApk; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 321 | std::unordered_set<std::string> extensions_to_not_compress; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 322 | }; |
| 323 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 324 | // A sampling of public framework resource IDs. |
| 325 | struct R { |
| 326 | struct attr { |
| 327 | enum : uint32_t { |
| 328 | paddingLeft = 0x010100d6u, |
| 329 | paddingRight = 0x010100d8u, |
| 330 | paddingHorizontal = 0x0101053du, |
| 331 | |
| 332 | paddingTop = 0x010100d7u, |
| 333 | paddingBottom = 0x010100d9u, |
| 334 | paddingVertical = 0x0101053eu, |
| 335 | |
| 336 | layout_marginLeft = 0x010100f7u, |
| 337 | layout_marginRight = 0x010100f9u, |
| 338 | layout_marginHorizontal = 0x0101053bu, |
| 339 | |
| 340 | layout_marginTop = 0x010100f8u, |
| 341 | layout_marginBottom = 0x010100fau, |
| 342 | layout_marginVertical = 0x0101053cu, |
| 343 | }; |
| 344 | }; |
| 345 | }; |
| 346 | |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 347 | class ResourceFileFlattener { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 348 | public: |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 349 | ResourceFileFlattener(const ResourceFileFlattenerOptions& options, IAaptContext* context, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 350 | proguard::KeepSet* keep_set); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 351 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 352 | bool Flatten(ResourceTable* table, IArchiveWriter* archive_writer); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 353 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 354 | private: |
| 355 | struct FileOperation { |
| 356 | ConfigDescription config; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 357 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 358 | // The entry this file came from. |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 359 | ResourceEntry* entry; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 360 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 361 | // The file to copy as-is. |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 362 | io::IFile* file_to_copy; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 363 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 364 | // The XML to process and flatten. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 365 | std::unique_ptr<xml::XmlResource> xml_to_flatten; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 366 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 367 | // The destination to write this file to. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 368 | std::string dst_path; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 369 | }; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 370 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 371 | uint32_t GetCompressionFlags(const StringPiece& str); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 372 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 373 | std::vector<std::unique_ptr<xml::XmlResource>> LinkAndVersionXmlFile(ResourceTable* table, |
| 374 | FileOperation* file_op); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 375 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 376 | ResourceFileFlattenerOptions options_; |
| 377 | IAaptContext* context_; |
| 378 | proguard::KeepSet* keep_set_; |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 379 | XmlCompatVersioner::Rules rules_; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 380 | }; |
| 381 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 382 | ResourceFileFlattener::ResourceFileFlattener(const ResourceFileFlattenerOptions& options, |
| 383 | IAaptContext* context, proguard::KeepSet* keep_set) |
| 384 | : options_(options), context_(context), keep_set_(keep_set) { |
| 385 | SymbolTable* symm = context_->GetExternalSymbols(); |
| 386 | |
| 387 | // Build up the rules for degrading newer attributes to older ones. |
| 388 | // NOTE(adamlesinski): These rules are hardcoded right now, but they should be |
| 389 | // generated from the attribute definitions themselves (b/62028956). |
| 390 | if (const SymbolTable::Symbol* s = symm->FindById(R::attr::paddingHorizontal)) { |
| 391 | std::vector<ReplacementAttr> replacements{ |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 392 | {"paddingLeft", R::attr::paddingLeft, Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
| 393 | {"paddingRight", R::attr::paddingRight, Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 394 | }; |
| 395 | rules_[R::attr::paddingHorizontal] = |
| 396 | util::make_unique<DegradeToManyRule>(std::move(replacements)); |
| 397 | } |
| 398 | |
| 399 | if (const SymbolTable::Symbol* s = symm->FindById(R::attr::paddingVertical)) { |
| 400 | std::vector<ReplacementAttr> replacements{ |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 401 | {"paddingTop", R::attr::paddingTop, Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
| 402 | {"paddingBottom", R::attr::paddingBottom, Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 403 | }; |
| 404 | rules_[R::attr::paddingVertical] = |
| 405 | util::make_unique<DegradeToManyRule>(std::move(replacements)); |
| 406 | } |
| 407 | |
| 408 | if (const SymbolTable::Symbol* s = symm->FindById(R::attr::layout_marginHorizontal)) { |
| 409 | std::vector<ReplacementAttr> replacements{ |
| 410 | {"layout_marginLeft", R::attr::layout_marginLeft, |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 411 | Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 412 | {"layout_marginRight", R::attr::layout_marginRight, |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 413 | Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 414 | }; |
| 415 | rules_[R::attr::layout_marginHorizontal] = |
| 416 | util::make_unique<DegradeToManyRule>(std::move(replacements)); |
| 417 | } |
| 418 | |
| 419 | if (const SymbolTable::Symbol* s = symm->FindById(R::attr::layout_marginVertical)) { |
| 420 | std::vector<ReplacementAttr> replacements{ |
| 421 | {"layout_marginTop", R::attr::layout_marginTop, |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 422 | Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 423 | {"layout_marginBottom", R::attr::layout_marginBottom, |
Adam Lesinski | 73bff1e | 2017-12-08 16:06:10 -0800 | [diff] [blame] | 424 | Attribute(android::ResTable_map::TYPE_DIMENSION)}, |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 425 | }; |
| 426 | rules_[R::attr::layout_marginVertical] = |
| 427 | util::make_unique<DegradeToManyRule>(std::move(replacements)); |
| 428 | } |
| 429 | } |
| 430 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 431 | uint32_t ResourceFileFlattener::GetCompressionFlags(const StringPiece& str) { |
| 432 | if (options_.do_not_compress_anything) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 433 | return 0; |
| 434 | } |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 435 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 436 | for (const std::string& extension : options_.extensions_to_not_compress) { |
| 437 | if (util::EndsWith(str, extension)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 438 | return 0; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 439 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 440 | } |
| 441 | return ArchiveEntry::kCompress; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 442 | } |
| 443 | |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 444 | static bool IsTransitionElement(const std::string& name) { |
| 445 | return name == "fade" || name == "changeBounds" || name == "slide" || name == "explode" || |
| 446 | name == "changeImageTransform" || name == "changeTransform" || |
| 447 | name == "changeClipBounds" || name == "autoTransition" || name == "recolor" || |
| 448 | name == "changeScroll" || name == "transitionSet" || name == "transition" || |
| 449 | name == "transitionManager"; |
| 450 | } |
| 451 | |
| 452 | static bool IsVectorElement(const std::string& name) { |
| 453 | return name == "vector" || name == "animated-vector" || name == "pathInterpolator" || |
Nick Butcher | e78a816 | 2018-01-09 15:24:21 +0000 | [diff] [blame] | 454 | name == "objectAnimator" || name == "gradient" || name == "animated-selector"; |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 455 | } |
| 456 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 457 | template <typename T> |
| 458 | std::vector<T> make_singleton_vec(T&& val) { |
| 459 | std::vector<T> vec; |
| 460 | vec.emplace_back(std::forward<T>(val)); |
| 461 | return vec; |
| 462 | } |
| 463 | |
| 464 | std::vector<std::unique_ptr<xml::XmlResource>> ResourceFileFlattener::LinkAndVersionXmlFile( |
| 465 | ResourceTable* table, FileOperation* file_op) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 466 | xml::XmlResource* doc = file_op->xml_to_flatten.get(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 467 | const Source& src = doc->file.source; |
Adam Lesinski | 5eeaadd | 2016-08-25 12:26:56 -0700 | [diff] [blame] | 468 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 469 | if (context_->IsVerbose()) { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 470 | context_->GetDiagnostics()->Note(DiagMessage() |
| 471 | << "linking " << src.path << " (" << doc->file.name << ")"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 472 | } |
| 473 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 474 | // First, strip out any tools namespace attributes. AAPT stripped them out early, which means |
| 475 | // that existing projects have out-of-date references which pass compilation. |
| 476 | xml::StripAndroidStudioAttributes(doc->root.get()); |
| 477 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 478 | XmlReferenceLinker xml_linker; |
| 479 | if (!xml_linker.Consume(context_, doc)) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 480 | return {}; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 481 | } |
| 482 | |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 483 | if (options_.update_proguard_spec && !proguard::CollectProguardRules(doc, keep_set_)) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 484 | return {}; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 485 | } |
| 486 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 487 | if (options_.no_xml_namespaces) { |
| 488 | XmlNamespaceRemover namespace_remover; |
| 489 | if (!namespace_remover.Consume(context_, doc)) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 490 | return {}; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 491 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 492 | } |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 493 | |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 494 | if (options_.no_auto_version) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 495 | return make_singleton_vec(std::move(file_op->xml_to_flatten)); |
| 496 | } |
Alexandria Cornwall | a7cc3f1 | 2016-08-16 13:33:32 -0700 | [diff] [blame] | 497 | |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 498 | if (options_.no_version_vectors || options_.no_version_transitions) { |
| 499 | // Skip this if it is a vector or animated-vector. |
Adam Lesinski | 6b37299 | 2017-08-09 10:54:23 -0700 | [diff] [blame] | 500 | xml::Element* el = doc->root.get(); |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 501 | if (el && el->namespace_uri.empty()) { |
| 502 | if ((options_.no_version_vectors && IsVectorElement(el->name)) || |
| 503 | (options_.no_version_transitions && IsTransitionElement(el->name))) { |
| 504 | return make_singleton_vec(std::move(file_op->xml_to_flatten)); |
| 505 | } |
| 506 | } |
| 507 | } |
| 508 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 509 | const ConfigDescription& config = file_op->config; |
| 510 | ResourceEntry* entry = file_op->entry; |
| 511 | |
| 512 | XmlCompatVersioner xml_compat_versioner(&rules_); |
| 513 | const util::Range<ApiVersion> api_range{config.sdkVersion, |
| 514 | FindNextApiVersionForConfig(entry, config)}; |
| 515 | return xml_compat_versioner.Process(context_, doc, api_range); |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 516 | } |
| 517 | |
Adam Lesinski | a65bbdf | 2018-02-15 12:39:44 -0800 | [diff] [blame] | 518 | ResourceFile::Type XmlFileTypeForOutputFormat(OutputFormat format) { |
| 519 | switch (format) { |
| 520 | case OutputFormat::kApk: |
| 521 | return ResourceFile::Type::kBinaryXml; |
| 522 | case OutputFormat::kProto: |
| 523 | return ResourceFile::Type::kProtoXml; |
| 524 | } |
| 525 | LOG_ALWAYS_FATAL("unreachable"); |
| 526 | return ResourceFile::Type::kUnknown; |
| 527 | } |
| 528 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 529 | bool ResourceFileFlattener::Flatten(ResourceTable* table, IArchiveWriter* archive_writer) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 530 | bool error = false; |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 531 | std::map<std::pair<ConfigDescription, StringPiece>, FileOperation> config_sorted_files; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 532 | |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 533 | proguard::CollectResourceReferences(context_, table, keep_set_); |
| 534 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 535 | for (auto& pkg : table->packages) { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 536 | CHECK(!pkg->name.empty()) << "Packages must have names when being linked"; |
| 537 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 538 | for (auto& type : pkg->types) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 539 | // Sort by config and name, so that we get better locality in the zip file. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 540 | config_sorted_files.clear(); |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 541 | std::queue<FileOperation> file_operations; |
Adam Lesinski | 355f285 | 2016-02-13 20:26:45 -0800 | [diff] [blame] | 542 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 543 | // Populate the queue with all files in the ResourceTable. |
| 544 | for (auto& entry : type->entries) { |
Adam Lesinski | bb94f32 | 2017-07-12 07:41:55 -0700 | [diff] [blame] | 545 | for (auto& config_value : entry->values) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 546 | // WARNING! Do not insert or remove any resources while executing in this scope. It will |
| 547 | // corrupt the iteration order. |
| 548 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 549 | FileReference* file_ref = ValueCast<FileReference>(config_value->value.get()); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 550 | if (!file_ref) { |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 551 | continue; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 552 | } |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 553 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 554 | io::IFile* file = file_ref->file; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 555 | if (!file) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 556 | context_->GetDiagnostics()->Error(DiagMessage(file_ref->GetSource()) |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 557 | << "file not found"); |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 558 | return false; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 559 | } |
| 560 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 561 | FileOperation file_op; |
| 562 | file_op.entry = entry.get(); |
| 563 | file_op.dst_path = *file_ref->path; |
| 564 | file_op.config = config_value->config; |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 565 | file_op.file_to_copy = file; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 566 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 567 | if (type->type != ResourceType::kRaw && |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 568 | (file_ref->type == ResourceFile::Type::kBinaryXml || |
| 569 | file_ref->type == ResourceFile::Type::kProtoXml)) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 570 | std::unique_ptr<io::IData> data = file->OpenAsData(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 571 | if (!data) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 572 | context_->GetDiagnostics()->Error(DiagMessage(file->GetSource()) |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 573 | << "failed to open file"); |
| 574 | return false; |
| 575 | } |
| 576 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 577 | if (file_ref->type == ResourceFile::Type::kProtoXml) { |
| 578 | pb::XmlNode pb_xml_node; |
| 579 | if (!pb_xml_node.ParseFromArray(data->data(), static_cast<int>(data->size()))) { |
| 580 | context_->GetDiagnostics()->Error(DiagMessage(file->GetSource()) |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 581 | << "failed to parse proto XML"); |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 582 | return false; |
| 583 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 584 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 585 | std::string error; |
| 586 | file_op.xml_to_flatten = DeserializeXmlResourceFromPb(pb_xml_node, &error); |
| 587 | if (file_op.xml_to_flatten == nullptr) { |
| 588 | context_->GetDiagnostics()->Error(DiagMessage(file->GetSource()) |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 589 | << "failed to deserialize proto XML: " << error); |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 590 | return false; |
| 591 | } |
| 592 | } else { |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 593 | std::string error_str; |
| 594 | file_op.xml_to_flatten = xml::Inflate(data->data(), data->size(), &error_str); |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 595 | if (file_op.xml_to_flatten == nullptr) { |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 596 | context_->GetDiagnostics()->Error(DiagMessage(file->GetSource()) |
| 597 | << "failed to parse binary XML: " << error_str); |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 598 | return false; |
| 599 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 600 | } |
| 601 | |
Adam Lesinski | a65bbdf | 2018-02-15 12:39:44 -0800 | [diff] [blame] | 602 | // Update the type that this file will be written as. |
| 603 | file_ref->type = XmlFileTypeForOutputFormat(options_.output_format); |
| 604 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 605 | file_op.xml_to_flatten->file.config = config_value->config; |
| 606 | file_op.xml_to_flatten->file.source = file_ref->GetSource(); |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 607 | file_op.xml_to_flatten->file.name = ResourceName(pkg->name, type->type, entry->name); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 608 | } |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 609 | |
| 610 | // NOTE(adamlesinski): Explicitly construct a StringPiece here, or |
| 611 | // else we end up copying the string in the std::make_pair() method, |
| 612 | // then creating a StringPiece from the copy, which would cause us |
| 613 | // to end up referencing garbage in the map. |
| 614 | const StringPiece entry_name(entry->name); |
| 615 | config_sorted_files[std::make_pair(config_value->config, entry_name)] = |
| 616 | std::move(file_op); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 617 | } |
| 618 | } |
| 619 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 620 | // Now flatten the sorted values. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 621 | for (auto& map_entry : config_sorted_files) { |
| 622 | const ConfigDescription& config = map_entry.first.first; |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 623 | FileOperation& file_op = map_entry.second; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 624 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 625 | if (file_op.xml_to_flatten) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 626 | std::vector<std::unique_ptr<xml::XmlResource>> versioned_docs = |
| 627 | LinkAndVersionXmlFile(table, &file_op); |
Adam Lesinski | c0a5e1e | 2017-08-07 11:56:32 -0700 | [diff] [blame] | 628 | if (versioned_docs.empty()) { |
| 629 | error = true; |
| 630 | continue; |
| 631 | } |
| 632 | |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 633 | for (std::unique_ptr<xml::XmlResource>& doc : versioned_docs) { |
| 634 | std::string dst_path = file_op.dst_path; |
| 635 | if (doc->file.config != file_op.config) { |
| 636 | // Only add the new versioned configurations. |
| 637 | if (context_->IsVerbose()) { |
| 638 | context_->GetDiagnostics()->Note(DiagMessage(doc->file.source) |
| 639 | << "auto-versioning resource from config '" |
| 640 | << config << "' -> '" << doc->file.config << "'"); |
| 641 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 642 | |
Adam Lesinski | a65bbdf | 2018-02-15 12:39:44 -0800 | [diff] [blame] | 643 | const ResourceFile& file = doc->file; |
| 644 | dst_path = ResourceUtils::BuildResourceFileName(file, context_->GetNameMangler()); |
| 645 | |
| 646 | std::unique_ptr<FileReference> file_ref = |
| 647 | util::make_unique<FileReference>(table->string_pool.MakeRef(dst_path)); |
| 648 | file_ref->SetSource(doc->file.source); |
| 649 | // Update the output format of this XML file. |
| 650 | file_ref->type = XmlFileTypeForOutputFormat(options_.output_format); |
| 651 | if (!table->AddResourceMangled(file.name, file.config, {}, std::move(file_ref), |
| 652 | context_->GetDiagnostics())) { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 653 | return false; |
| 654 | } |
| 655 | } |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 656 | |
| 657 | error |= !FlattenXml(context_, *doc, dst_path, options_.keep_raw_values, |
| 658 | false /*utf16*/, options_.output_format, archive_writer); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 659 | } |
| 660 | } else { |
Adam Lesinski | c744ae8 | 2017-05-17 19:28:38 -0700 | [diff] [blame] | 661 | error |= !io::CopyFileToArchive(context_, file_op.file_to_copy, file_op.dst_path, |
| 662 | GetCompressionFlags(file_op.dst_path), archive_writer); |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 663 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 664 | } |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 665 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 666 | } |
| 667 | return !error; |
| 668 | } |
| 669 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 670 | static bool WriteStableIdMapToPath(IDiagnostics* diag, |
| 671 | const std::unordered_map<ResourceName, ResourceId>& id_map, |
| 672 | const std::string& id_map_path) { |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 673 | io::FileOutputStream fout(id_map_path); |
| 674 | if (fout.HadError()) { |
| 675 | diag->Error(DiagMessage(id_map_path) << "failed to open: " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 676 | return false; |
| 677 | } |
| 678 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 679 | text::Printer printer(&fout); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 680 | for (const auto& entry : id_map) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 681 | const ResourceName& name = entry.first; |
| 682 | const ResourceId& id = entry.second; |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 683 | printer.Print(name.to_string()); |
| 684 | printer.Print(" = "); |
| 685 | printer.Println(id.to_string()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 686 | } |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 687 | fout.Flush(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 688 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 689 | if (fout.HadError()) { |
| 690 | diag->Error(DiagMessage(id_map_path) << "failed writing to file: " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 691 | return false; |
| 692 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 693 | return true; |
| 694 | } |
| 695 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 696 | static bool LoadStableIdMap(IDiagnostics* diag, const std::string& path, |
| 697 | std::unordered_map<ResourceName, ResourceId>* out_id_map) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 698 | std::string content; |
Adam Lesinski | 2354b56 | 2017-05-26 16:31:38 -0700 | [diff] [blame] | 699 | if (!android::base::ReadFileToString(path, &content, true /*follow_symlinks*/)) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 700 | diag->Error(DiagMessage(path) << "failed reading stable ID file"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 701 | return false; |
| 702 | } |
| 703 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 704 | out_id_map->clear(); |
| 705 | size_t line_no = 0; |
| 706 | for (StringPiece line : util::Tokenize(content, '\n')) { |
| 707 | line_no++; |
| 708 | line = util::TrimWhitespace(line); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 709 | if (line.empty()) { |
| 710 | continue; |
| 711 | } |
| 712 | |
| 713 | auto iter = std::find(line.begin(), line.end(), '='); |
| 714 | if (iter == line.end()) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 715 | diag->Error(DiagMessage(Source(path, line_no)) << "missing '='"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 716 | return false; |
| 717 | } |
| 718 | |
| 719 | ResourceNameRef name; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 720 | StringPiece res_name_str = |
| 721 | util::TrimWhitespace(line.substr(0, std::distance(line.begin(), iter))); |
| 722 | if (!ResourceUtils::ParseResourceName(res_name_str, &name)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 723 | diag->Error(DiagMessage(Source(path, line_no)) << "invalid resource name '" << res_name_str |
| 724 | << "'"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 725 | return false; |
| 726 | } |
| 727 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 728 | const size_t res_id_start_idx = std::distance(line.begin(), iter) + 1; |
| 729 | const size_t res_id_str_len = line.size() - res_id_start_idx; |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 730 | StringPiece res_id_str = util::TrimWhitespace(line.substr(res_id_start_idx, res_id_str_len)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 731 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 732 | Maybe<ResourceId> maybe_id = ResourceUtils::ParseResourceId(res_id_str); |
| 733 | if (!maybe_id) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 734 | diag->Error(DiagMessage(Source(path, line_no)) << "invalid resource ID '" << res_id_str |
| 735 | << "'"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 736 | return false; |
| 737 | } |
| 738 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 739 | (*out_id_map)[name.ToResourceName()] = maybe_id.value(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 740 | } |
| 741 | return true; |
Adam Lesinski | bf0bd0f | 2016-06-01 15:31:50 -0700 | [diff] [blame] | 742 | } |
| 743 | |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 744 | static int32_t FindFrameworkAssetManagerCookie(const android::AssetManager& assets) { |
| 745 | using namespace android; |
| 746 | |
| 747 | // Find the system package (0x01). AAPT always generates attributes with the type 0x01, so |
| 748 | // we're looking for the first attribute resource in the system package. |
| 749 | const ResTable& table = assets.getResources(true); |
| 750 | Res_value val; |
| 751 | ssize_t idx = table.getResource(0x01010000, &val, true); |
| 752 | if (idx != NO_ERROR) { |
| 753 | // Try as a bag. |
| 754 | const ResTable::bag_entry* entry; |
| 755 | ssize_t cnt = table.lockBag(0x01010000, &entry); |
| 756 | if (cnt >= 0) { |
| 757 | idx = entry->stringBlock; |
| 758 | } |
| 759 | table.unlockBag(entry); |
| 760 | } |
| 761 | |
| 762 | if (idx < 0) { |
| 763 | return 0; |
| 764 | } |
| 765 | return table.getTableCookie(idx); |
| 766 | } |
| 767 | |
Adam Lesinski | fb48d29 | 2015-11-07 15:52:13 -0800 | [diff] [blame] | 768 | class LinkCommand { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 769 | public: |
| 770 | LinkCommand(LinkContext* context, const LinkOptions& options) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 771 | : options_(options), |
| 772 | context_(context), |
| 773 | final_table_(), |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 774 | file_collection_(util::make_unique<io::FileCollection>()) { |
| 775 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 776 | |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 777 | void ExtractCompileSdkVersions(android::AssetManager* assets) { |
| 778 | using namespace android; |
| 779 | |
| 780 | int32_t cookie = FindFrameworkAssetManagerCookie(*assets); |
| 781 | if (cookie == 0) { |
| 782 | // No Framework assets loaded. Not a failure. |
| 783 | return; |
| 784 | } |
| 785 | |
| 786 | std::unique_ptr<Asset> manifest( |
| 787 | assets->openNonAsset(cookie, kAndroidManifestPath, Asset::AccessMode::ACCESS_BUFFER)); |
| 788 | if (manifest == nullptr) { |
| 789 | // No errors. |
| 790 | return; |
| 791 | } |
| 792 | |
| 793 | std::string error; |
| 794 | std::unique_ptr<xml::XmlResource> manifest_xml = |
| 795 | xml::Inflate(manifest->getBuffer(true /*wordAligned*/), manifest->getLength(), &error); |
| 796 | if (manifest_xml == nullptr) { |
| 797 | // No errors. |
| 798 | return; |
| 799 | } |
| 800 | |
Todd Kennedy | 9f6dec1 | 2018-04-20 12:29:29 -0700 | [diff] [blame^] | 801 | if (!options_.manifest_fixer_options.compile_sdk_version) { |
| 802 | xml::Attribute* attr = manifest_xml->root->FindAttribute(xml::kSchemaAndroid, "versionCode"); |
| 803 | if (attr != nullptr) { |
| 804 | Maybe<std::string>& compile_sdk_version = options_.manifest_fixer_options.compile_sdk_version; |
| 805 | if (BinaryPrimitive* prim = ValueCast<BinaryPrimitive>(attr->compiled_value.get())) { |
| 806 | switch (prim->value.dataType) { |
| 807 | case Res_value::TYPE_INT_DEC: |
| 808 | compile_sdk_version = StringPrintf("%" PRId32, static_cast<int32_t>(prim->value.data)); |
| 809 | break; |
| 810 | case Res_value::TYPE_INT_HEX: |
| 811 | compile_sdk_version = StringPrintf("%" PRIx32, prim->value.data); |
| 812 | break; |
| 813 | default: |
| 814 | break; |
| 815 | } |
| 816 | } else if (String* str = ValueCast<String>(attr->compiled_value.get())) { |
| 817 | compile_sdk_version = *str->value; |
| 818 | } else { |
| 819 | compile_sdk_version = attr->value; |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 820 | } |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 821 | } |
| 822 | } |
| 823 | |
Todd Kennedy | 9f6dec1 | 2018-04-20 12:29:29 -0700 | [diff] [blame^] | 824 | if (!options_.manifest_fixer_options.compile_sdk_version_codename) { |
| 825 | xml::Attribute* attr = manifest_xml->root->FindAttribute(xml::kSchemaAndroid, "versionName"); |
| 826 | if (attr != nullptr) { |
| 827 | Maybe<std::string>& compile_sdk_version_codename = |
| 828 | options_.manifest_fixer_options.compile_sdk_version_codename; |
| 829 | if (String* str = ValueCast<String>(attr->compiled_value.get())) { |
| 830 | compile_sdk_version_codename = *str->value; |
| 831 | } else { |
| 832 | compile_sdk_version_codename = attr->value; |
| 833 | } |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 834 | } |
| 835 | } |
| 836 | } |
| 837 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 838 | // Creates a SymbolTable that loads symbols from the various APKs. |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 839 | // Pre-condition: context_->GetCompilationPackage() needs to be set. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 840 | bool LoadSymbolsFromIncludePaths() { |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 841 | auto asset_source = util::make_unique<AssetManagerSymbolSource>(); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 842 | for (const std::string& path : options_.include_paths) { |
| 843 | if (context_->IsVerbose()) { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 844 | context_->GetDiagnostics()->Note(DiagMessage() << "including " << path); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 845 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 846 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 847 | std::string error; |
| 848 | auto zip_collection = io::ZipFileCollection::Create(path, &error); |
| 849 | if (zip_collection == nullptr) { |
| 850 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to open APK: " << error); |
| 851 | return false; |
| 852 | } |
| 853 | |
| 854 | if (zip_collection->FindFile(kProtoResourceTablePath) != nullptr) { |
| 855 | // Load this as a static library include. |
| 856 | std::unique_ptr<LoadedApk> static_apk = LoadedApk::LoadProtoApkFromFileCollection( |
| 857 | Source(path), std::move(zip_collection), context_->GetDiagnostics()); |
| 858 | if (static_apk == nullptr) { |
| 859 | return false; |
| 860 | } |
| 861 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 862 | if (context_->GetPackageType() != PackageType::kStaticLib) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 863 | // Can't include static libraries when not building a static library (they have no IDs |
| 864 | // assigned). |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 865 | context_->GetDiagnostics()->Error( |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 866 | DiagMessage(path) << "can't include static library when not building a static lib"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 867 | return false; |
| 868 | } |
| 869 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 870 | ResourceTable* table = static_apk->GetResourceTable(); |
| 871 | |
| 872 | // If we are using --no-static-lib-packages, we need to rename the package of this table to |
| 873 | // our compilation package. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 874 | if (options_.no_static_lib_packages) { |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 875 | // Since package names can differ, and multiple packages can exist in a ResourceTable, |
| 876 | // we place the requirement that all static libraries are built with the package |
| 877 | // ID 0x7f. So if one is not found, this is an error. |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 878 | if (ResourceTablePackage* pkg = table->FindPackageById(kAppPackageId)) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 879 | pkg->name = context_->GetCompilationPackage(); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 880 | } else { |
| 881 | context_->GetDiagnostics()->Error(DiagMessage(path) |
| 882 | << "no package with ID 0x7f found in static library"); |
| 883 | return false; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 884 | } |
| 885 | } |
| 886 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 887 | context_->GetExternalSymbols()->AppendSource( |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 888 | util::make_unique<ResourceTableSymbolSource>(table)); |
| 889 | static_library_includes_.push_back(std::move(static_apk)); |
| 890 | } else { |
| 891 | if (!asset_source->AddAssetPath(path)) { |
| 892 | context_->GetDiagnostics()->Error(DiagMessage() |
| 893 | << "failed to load include path " << path); |
| 894 | return false; |
| 895 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 896 | } |
| 897 | } |
| 898 | |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 899 | // Capture the shared libraries so that the final resource table can be properly flattened |
| 900 | // with support for shared libraries. |
| 901 | for (auto& entry : asset_source->GetAssignedPackageIds()) { |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 902 | if (entry.first > kFrameworkPackageId && entry.first < kAppPackageId) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 903 | final_table_.included_packages_[entry.first] = entry.second; |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 904 | } else if (entry.first == kAppPackageId) { |
| 905 | // Capture the included base feature package. |
| 906 | included_feature_base_ = entry.second; |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 907 | } else if (entry.first == kFrameworkPackageId) { |
| 908 | // Try to embed which version of the framework we're compiling against. |
| 909 | // First check if we should use compileSdkVersion at all. Otherwise compilation may fail |
| 910 | // when linking our synthesized 'android:compileSdkVersion' attribute. |
| 911 | std::unique_ptr<SymbolTable::Symbol> symbol = asset_source->FindByName( |
| 912 | ResourceName("android", ResourceType::kAttr, "compileSdkVersion")); |
| 913 | if (symbol != nullptr && symbol->is_public) { |
| 914 | // The symbol is present and public, extract the android:versionName and |
| 915 | // android:versionCode from the framework AndroidManifest.xml. |
| 916 | ExtractCompileSdkVersions(asset_source->GetAssetManager()); |
| 917 | } |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 918 | } |
| 919 | } |
| 920 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 921 | context_->GetExternalSymbols()->AppendSource(std::move(asset_source)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 922 | return true; |
| 923 | } |
| 924 | |
Adam Lesinski | b0c47ef | 2017-03-06 20:05:57 -0800 | [diff] [blame] | 925 | Maybe<AppInfo> ExtractAppInfoFromManifest(xml::XmlResource* xml_res, IDiagnostics* diag) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 926 | // Make sure the first element is <manifest> with package attribute. |
Adam Lesinski | b0c47ef | 2017-03-06 20:05:57 -0800 | [diff] [blame] | 927 | xml::Element* manifest_el = xml::FindRootElement(xml_res->root.get()); |
| 928 | if (manifest_el == nullptr) { |
| 929 | return {}; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 930 | } |
Adam Lesinski | b0c47ef | 2017-03-06 20:05:57 -0800 | [diff] [blame] | 931 | |
| 932 | AppInfo app_info; |
| 933 | |
| 934 | if (!manifest_el->namespace_uri.empty() || manifest_el->name != "manifest") { |
| 935 | diag->Error(DiagMessage(xml_res->file.source) << "root tag must be <manifest>"); |
| 936 | return {}; |
| 937 | } |
| 938 | |
| 939 | xml::Attribute* package_attr = manifest_el->FindAttribute({}, "package"); |
| 940 | if (!package_attr) { |
| 941 | diag->Error(DiagMessage(xml_res->file.source) |
| 942 | << "<manifest> must have a 'package' attribute"); |
| 943 | return {}; |
| 944 | } |
| 945 | app_info.package = package_attr->value; |
| 946 | |
| 947 | if (xml::Attribute* version_code_attr = |
| 948 | manifest_el->FindAttribute(xml::kSchemaAndroid, "versionCode")) { |
| 949 | Maybe<uint32_t> maybe_code = ResourceUtils::ParseInt(version_code_attr->value); |
| 950 | if (!maybe_code) { |
| 951 | diag->Error(DiagMessage(xml_res->file.source.WithLine(manifest_el->line_number)) |
| 952 | << "invalid android:versionCode '" << version_code_attr->value << "'"); |
| 953 | return {}; |
| 954 | } |
| 955 | app_info.version_code = maybe_code.value(); |
| 956 | } |
| 957 | |
| 958 | if (xml::Attribute* revision_code_attr = |
| 959 | manifest_el->FindAttribute(xml::kSchemaAndroid, "revisionCode")) { |
| 960 | Maybe<uint32_t> maybe_code = ResourceUtils::ParseInt(revision_code_attr->value); |
| 961 | if (!maybe_code) { |
| 962 | diag->Error(DiagMessage(xml_res->file.source.WithLine(manifest_el->line_number)) |
| 963 | << "invalid android:revisionCode '" << revision_code_attr->value << "'"); |
| 964 | return {}; |
| 965 | } |
| 966 | app_info.revision_code = maybe_code.value(); |
| 967 | } |
| 968 | |
| 969 | if (xml::Attribute* split_name_attr = manifest_el->FindAttribute({}, "split")) { |
| 970 | if (!split_name_attr->value.empty()) { |
| 971 | app_info.split_name = split_name_attr->value; |
| 972 | } |
| 973 | } |
| 974 | |
| 975 | if (xml::Element* uses_sdk_el = manifest_el->FindChild({}, "uses-sdk")) { |
| 976 | if (xml::Attribute* min_sdk = |
| 977 | uses_sdk_el->FindAttribute(xml::kSchemaAndroid, "minSdkVersion")) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 978 | app_info.min_sdk_version = ResourceUtils::ParseSdkVersion(min_sdk->value); |
Adam Lesinski | b0c47ef | 2017-03-06 20:05:57 -0800 | [diff] [blame] | 979 | } |
| 980 | } |
| 981 | return app_info; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 982 | } |
| 983 | |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 984 | // Precondition: ResourceTable doesn't have any IDs assigned yet, nor is it linked. |
| 985 | // Postcondition: ResourceTable has only one package left. All others are |
| 986 | // stripped, or there is an error and false is returned. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 987 | bool VerifyNoExternalPackages() { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 988 | auto is_ext_package_func = [&](const std::unique_ptr<ResourceTablePackage>& pkg) -> bool { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 989 | return context_->GetCompilationPackage() != pkg->name || !pkg->id || |
| 990 | pkg->id.value() != context_->GetPackageId(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 991 | }; |
| 992 | |
| 993 | bool error = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 994 | for (const auto& package : final_table_.packages) { |
| 995 | if (is_ext_package_func(package)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 996 | // We have a package that is not related to the one we're building! |
| 997 | for (const auto& type : package->types) { |
| 998 | for (const auto& entry : type->entries) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 999 | ResourceNameRef res_name(package->name, type->type, entry->name); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1000 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1001 | for (const auto& config_value : entry->values) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1002 | // Special case the occurrence of an ID that is being generated |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1003 | // for the 'android' package. This is due to legacy reasons. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1004 | if (ValueCast<Id>(config_value->value.get()) && package->name == "android") { |
| 1005 | context_->GetDiagnostics()->Warn(DiagMessage(config_value->value->GetSource()) |
| 1006 | << "generated id '" << res_name |
| 1007 | << "' for external package '" << package->name |
| 1008 | << "'"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1009 | } else { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1010 | context_->GetDiagnostics()->Error(DiagMessage(config_value->value->GetSource()) |
| 1011 | << "defined resource '" << res_name |
| 1012 | << "' for external package '" << package->name |
| 1013 | << "'"); |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1014 | error = true; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1015 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1016 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1017 | } |
| 1018 | } |
| 1019 | } |
| 1020 | } |
| 1021 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1022 | auto new_end_iter = std::remove_if(final_table_.packages.begin(), final_table_.packages.end(), |
| 1023 | is_ext_package_func); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1024 | final_table_.packages.erase(new_end_iter, final_table_.packages.end()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1025 | return !error; |
| 1026 | } |
| 1027 | |
| 1028 | /** |
| 1029 | * Returns true if no IDs have been set, false otherwise. |
| 1030 | */ |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1031 | bool VerifyNoIdsSet() { |
| 1032 | for (const auto& package : final_table_.packages) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1033 | for (const auto& type : package->types) { |
| 1034 | if (type->id) { |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1035 | context_->GetDiagnostics()->Error(DiagMessage() << "type " << type->type << " has ID " |
| 1036 | << StringPrintf("%02x", type->id.value()) |
| 1037 | << " assigned"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1038 | return false; |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1039 | } |
| 1040 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1041 | for (const auto& entry : type->entries) { |
| 1042 | if (entry->id) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1043 | ResourceNameRef res_name(package->name, type->type, entry->name); |
| 1044 | context_->GetDiagnostics()->Error( |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1045 | DiagMessage() << "entry " << res_name << " has ID " |
| 1046 | << StringPrintf("%02x", entry->id.value()) << " assigned"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1047 | return false; |
| 1048 | } |
| 1049 | } |
| 1050 | } |
| 1051 | } |
| 1052 | return true; |
| 1053 | } |
| 1054 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1055 | std::unique_ptr<IArchiveWriter> MakeArchiveWriter(const StringPiece& out) { |
| 1056 | if (options_.output_to_directory) { |
| 1057 | return CreateDirectoryArchiveWriter(context_->GetDiagnostics(), out); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1058 | } else { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1059 | return CreateZipFileArchiveWriter(context_->GetDiagnostics(), out); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1060 | } |
| 1061 | } |
| 1062 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1063 | bool FlattenTable(ResourceTable* table, OutputFormat format, IArchiveWriter* writer) { |
| 1064 | switch (format) { |
| 1065 | case OutputFormat::kApk: { |
| 1066 | BigBuffer buffer(1024); |
| 1067 | TableFlattener flattener(options_.table_flattener_options, &buffer); |
| 1068 | if (!flattener.Consume(context_, table)) { |
| 1069 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to flatten resource table"); |
| 1070 | return false; |
| 1071 | } |
| 1072 | |
| 1073 | io::BigBufferInputStream input_stream(&buffer); |
| 1074 | return io::CopyInputStreamToArchive(context_, &input_stream, kApkResourceTablePath, |
| 1075 | ArchiveEntry::kAlign, writer); |
| 1076 | } break; |
| 1077 | |
| 1078 | case OutputFormat::kProto: { |
| 1079 | pb::ResourceTable pb_table; |
Ryan Mitchell | 70414f2 | 2018-03-26 11:05:31 -0700 | [diff] [blame] | 1080 | SerializeTableToPb(*table, &pb_table, context_->GetDiagnostics()); |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1081 | return io::CopyProtoToArchive(context_, &pb_table, kProtoResourceTablePath, |
| 1082 | ArchiveEntry::kCompress, writer); |
| 1083 | } break; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1084 | } |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1085 | return false; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1086 | } |
| 1087 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1088 | bool WriteJavaFile(ResourceTable* table, const StringPiece& package_name_to_generate, |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 1089 | const StringPiece& out_package, const JavaClassGeneratorOptions& java_options, |
Chih-Hung Hsieh | 4dc5812 | 2017-08-03 16:28:10 -0700 | [diff] [blame] | 1090 | const Maybe<std::string>& out_text_symbols_path = {}) { |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1091 | if (!options_.generate_java_class_path && !out_text_symbols_path) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1092 | return true; |
| 1093 | } |
| 1094 | |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1095 | std::string out_path; |
| 1096 | std::unique_ptr<io::FileOutputStream> fout; |
| 1097 | if (options_.generate_java_class_path) { |
| 1098 | out_path = options_.generate_java_class_path.value(); |
| 1099 | file::AppendPath(&out_path, file::PackageToPath(out_package)); |
| 1100 | if (!file::mkdirs(out_path)) { |
| 1101 | context_->GetDiagnostics()->Error(DiagMessage() |
| 1102 | << "failed to create directory '" << out_path << "'"); |
| 1103 | return false; |
| 1104 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1105 | |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1106 | file::AppendPath(&out_path, "R.java"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1107 | |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1108 | fout = util::make_unique<io::FileOutputStream>(out_path); |
| 1109 | if (fout->HadError()) { |
| 1110 | context_->GetDiagnostics()->Error(DiagMessage() << "failed writing to '" << out_path |
| 1111 | << "': " << fout->GetError()); |
| 1112 | return false; |
| 1113 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1114 | } |
| 1115 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1116 | std::unique_ptr<io::FileOutputStream> fout_text; |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 1117 | if (out_text_symbols_path) { |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1118 | fout_text = util::make_unique<io::FileOutputStream>(out_text_symbols_path.value()); |
| 1119 | if (fout_text->HadError()) { |
| 1120 | context_->GetDiagnostics()->Error(DiagMessage() |
| 1121 | << "failed writing to '" << out_text_symbols_path.value() |
| 1122 | << "': " << fout_text->GetError()); |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 1123 | return false; |
| 1124 | } |
| 1125 | } |
| 1126 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1127 | JavaClassGenerator generator(context_, table, java_options); |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1128 | if (!generator.Generate(package_name_to_generate, out_package, fout.get(), fout_text.get())) { |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1129 | context_->GetDiagnostics()->Error(DiagMessage(out_path) << generator.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1130 | return false; |
| 1131 | } |
| 1132 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1133 | return true; |
| 1134 | } |
| 1135 | |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1136 | bool GenerateJavaClasses() { |
| 1137 | // The set of packages whose R class to call in the main classes onResourcesLoaded callback. |
| 1138 | std::vector<std::string> packages_to_callback; |
| 1139 | |
| 1140 | JavaClassGeneratorOptions template_options; |
| 1141 | template_options.types = JavaClassGeneratorOptions::SymbolTypes::kAll; |
| 1142 | template_options.javadoc_annotations = options_.javadoc_annotations; |
| 1143 | |
| 1144 | if (context_->GetPackageType() == PackageType::kStaticLib || options_.generate_non_final_ids) { |
| 1145 | template_options.use_final = false; |
| 1146 | } |
| 1147 | |
| 1148 | if (context_->GetPackageType() == PackageType::kSharedLib) { |
| 1149 | template_options.use_final = false; |
| 1150 | template_options.rewrite_callback_options = OnResourcesLoadedCallbackOptions{}; |
| 1151 | } |
| 1152 | |
| 1153 | const StringPiece actual_package = context_->GetCompilationPackage(); |
| 1154 | StringPiece output_package = context_->GetCompilationPackage(); |
| 1155 | if (options_.custom_java_package) { |
| 1156 | // Override the output java package to the custom one. |
| 1157 | output_package = options_.custom_java_package.value(); |
| 1158 | } |
| 1159 | |
| 1160 | // Generate the private symbols if required. |
| 1161 | if (options_.private_symbols) { |
| 1162 | packages_to_callback.push_back(options_.private_symbols.value()); |
| 1163 | |
| 1164 | // If we defined a private symbols package, we only emit Public symbols |
| 1165 | // to the original package, and private and public symbols to the private package. |
| 1166 | JavaClassGeneratorOptions options = template_options; |
| 1167 | options.types = JavaClassGeneratorOptions::SymbolTypes::kPublicPrivate; |
| 1168 | if (!WriteJavaFile(&final_table_, actual_package, options_.private_symbols.value(), |
| 1169 | options)) { |
| 1170 | return false; |
| 1171 | } |
| 1172 | } |
| 1173 | |
| 1174 | // Generate copies of the original package R class but with different package names. |
| 1175 | // This is to support non-namespaced builds. |
| 1176 | for (const std::string& extra_package : options_.extra_java_packages) { |
| 1177 | packages_to_callback.push_back(extra_package); |
| 1178 | |
| 1179 | JavaClassGeneratorOptions options = template_options; |
| 1180 | options.types = JavaClassGeneratorOptions::SymbolTypes::kAll; |
| 1181 | if (!WriteJavaFile(&final_table_, actual_package, extra_package, options)) { |
| 1182 | return false; |
| 1183 | } |
| 1184 | } |
| 1185 | |
| 1186 | // Generate R classes for each package that was merged (static library). |
| 1187 | // Use the actual package's resources only. |
| 1188 | for (const std::string& package : table_merger_->merged_packages()) { |
| 1189 | packages_to_callback.push_back(package); |
| 1190 | |
| 1191 | JavaClassGeneratorOptions options = template_options; |
| 1192 | options.types = JavaClassGeneratorOptions::SymbolTypes::kAll; |
| 1193 | if (!WriteJavaFile(&final_table_, package, package, options)) { |
| 1194 | return false; |
| 1195 | } |
| 1196 | } |
| 1197 | |
| 1198 | // Generate the main public R class. |
| 1199 | JavaClassGeneratorOptions options = template_options; |
| 1200 | |
| 1201 | // Only generate public symbols if we have a private package. |
| 1202 | if (options_.private_symbols) { |
| 1203 | options.types = JavaClassGeneratorOptions::SymbolTypes::kPublic; |
| 1204 | } |
| 1205 | |
| 1206 | if (options.rewrite_callback_options) { |
| 1207 | options.rewrite_callback_options.value().packages_to_callback = |
| 1208 | std::move(packages_to_callback); |
| 1209 | } |
| 1210 | |
| 1211 | if (!WriteJavaFile(&final_table_, actual_package, output_package, options, |
| 1212 | options_.generate_text_symbols_path)) { |
| 1213 | return false; |
| 1214 | } |
| 1215 | |
| 1216 | return true; |
| 1217 | } |
| 1218 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1219 | bool WriteManifestJavaFile(xml::XmlResource* manifest_xml) { |
| 1220 | if (!options_.generate_java_class_path) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1221 | return true; |
| 1222 | } |
| 1223 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1224 | std::unique_ptr<ClassDefinition> manifest_class = |
| 1225 | GenerateManifestClass(context_->GetDiagnostics(), manifest_xml); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1226 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1227 | if (!manifest_class) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1228 | // Something bad happened, but we already logged it, so exit. |
| 1229 | return false; |
| 1230 | } |
| 1231 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1232 | if (manifest_class->empty()) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1233 | // Empty Manifest class, no need to generate it. |
| 1234 | return true; |
| 1235 | } |
| 1236 | |
| 1237 | // Add any JavaDoc annotations to the generated class. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1238 | for (const std::string& annotation : options_.javadoc_annotations) { |
| 1239 | std::string proper_annotation = "@"; |
| 1240 | proper_annotation += annotation; |
| 1241 | manifest_class->GetCommentBuilder()->AppendComment(proper_annotation); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1242 | } |
| 1243 | |
Adam Lesinski | 0d81f70 | 2017-06-27 15:51:09 -0700 | [diff] [blame] | 1244 | const std::string package_utf8 = |
| 1245 | options_.custom_java_package.value_or_default(context_->GetCompilationPackage()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1246 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1247 | std::string out_path = options_.generate_java_class_path.value(); |
| 1248 | file::AppendPath(&out_path, file::PackageToPath(package_utf8)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1249 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1250 | if (!file::mkdirs(out_path)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1251 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to create directory '" << out_path |
| 1252 | << "'"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1253 | return false; |
| 1254 | } |
| 1255 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1256 | file::AppendPath(&out_path, "Manifest.java"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1257 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1258 | io::FileOutputStream fout(out_path); |
| 1259 | if (fout.HadError()) { |
| 1260 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to open '" << out_path |
| 1261 | << "': " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1262 | return false; |
| 1263 | } |
| 1264 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1265 | ClassDefinition::WriteJavaFile(manifest_class.get(), package_utf8, true, &fout); |
| 1266 | fout.Flush(); |
| 1267 | |
| 1268 | if (fout.HadError()) { |
| 1269 | context_->GetDiagnostics()->Error(DiagMessage() << "failed writing to '" << out_path |
| 1270 | << "': " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1271 | return false; |
| 1272 | } |
| 1273 | return true; |
| 1274 | } |
| 1275 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1276 | bool WriteProguardFile(const Maybe<std::string>& out, const proguard::KeepSet& keep_set) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1277 | if (!out) { |
| 1278 | return true; |
| 1279 | } |
| 1280 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1281 | const std::string& out_path = out.value(); |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1282 | io::FileOutputStream fout(out_path); |
| 1283 | if (fout.HadError()) { |
| 1284 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to open '" << out_path |
| 1285 | << "': " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1286 | return false; |
| 1287 | } |
| 1288 | |
Adam Lesinski | a693c4a | 2017-11-09 11:29:39 -0800 | [diff] [blame] | 1289 | proguard::WriteKeepSet(keep_set, &fout); |
| 1290 | fout.Flush(); |
| 1291 | |
| 1292 | if (fout.HadError()) { |
| 1293 | context_->GetDiagnostics()->Error(DiagMessage() << "failed writing to '" << out_path |
| 1294 | << "': " << fout.GetError()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1295 | return false; |
| 1296 | } |
| 1297 | return true; |
| 1298 | } |
| 1299 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1300 | bool MergeStaticLibrary(const std::string& input, bool override) { |
| 1301 | if (context_->IsVerbose()) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1302 | context_->GetDiagnostics()->Note(DiagMessage() << "merging static library " << input); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1303 | } |
| 1304 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1305 | std::unique_ptr<LoadedApk> apk = LoadedApk::LoadApkFromPath(input, context_->GetDiagnostics()); |
| 1306 | if (apk == nullptr) { |
Adam Lesinski | b5dc4bd | 2017-02-22 19:29:29 -0800 | [diff] [blame] | 1307 | context_->GetDiagnostics()->Error(DiagMessage(input) << "invalid static library"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1308 | return false; |
| 1309 | } |
| 1310 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1311 | ResourceTable* table = apk->GetResourceTable(); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1312 | ResourceTablePackage* pkg = table->FindPackageById(kAppPackageId); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1313 | if (!pkg) { |
Adam Lesinski | b5dc4bd | 2017-02-22 19:29:29 -0800 | [diff] [blame] | 1314 | context_->GetDiagnostics()->Error(DiagMessage(input) << "static library has no package"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1315 | return false; |
| 1316 | } |
| 1317 | |
| 1318 | bool result; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1319 | if (options_.no_static_lib_packages) { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1320 | // Merge all resources as if they were in the compilation package. This is the old behavior |
| 1321 | // of aapt. |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1322 | |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1323 | // Add the package to the set of --extra-packages so we emit an R.java for each library |
| 1324 | // package. |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1325 | if (!pkg->name.empty()) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1326 | options_.extra_java_packages.insert(pkg->name); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1327 | } |
| 1328 | |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1329 | // Clear the package name, so as to make the resources look like they are coming from the |
| 1330 | // local package. |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1331 | pkg->name = ""; |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1332 | result = table_merger_->Merge(Source(input), table, override); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1333 | |
| 1334 | } else { |
| 1335 | // This is the proper way to merge libraries, where the package name is |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1336 | // preserved and resource names are mangled. |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1337 | result = table_merger_->MergeAndMangle(Source(input), pkg->name, table); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1338 | } |
| 1339 | |
| 1340 | if (!result) { |
| 1341 | return false; |
| 1342 | } |
| 1343 | |
| 1344 | // Make sure to move the collection into the set of IFileCollections. |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1345 | merged_apks_.push_back(std::move(apk)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1346 | return true; |
| 1347 | } |
| 1348 | |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 1349 | bool MergeExportedSymbols(const Source& source, |
| 1350 | const std::vector<SourcedResourceName>& exported_symbols) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1351 | // Add the exports of this file to the table. |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 1352 | for (const SourcedResourceName& exported_symbol : exported_symbols) { |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1353 | ResourceName res_name = exported_symbol.name; |
| 1354 | if (res_name.package.empty()) { |
| 1355 | res_name.package = context_->GetCompilationPackage(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1356 | } |
| 1357 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1358 | Maybe<ResourceName> mangled_name = context_->GetNameMangler()->MangleName(res_name); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1359 | if (mangled_name) { |
| 1360 | res_name = mangled_name.value(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1361 | } |
| 1362 | |
| 1363 | std::unique_ptr<Id> id = util::make_unique<Id>(); |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 1364 | id->SetSource(source.WithLine(exported_symbol.line)); |
Adam Lesinski | 71be705 | 2017-12-12 16:48:07 -0800 | [diff] [blame] | 1365 | bool result = |
| 1366 | final_table_.AddResourceMangled(res_name, ConfigDescription::DefaultConfig(), |
| 1367 | std::string(), std::move(id), context_->GetDiagnostics()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1368 | if (!result) { |
| 1369 | return false; |
| 1370 | } |
| 1371 | } |
| 1372 | return true; |
| 1373 | } |
| 1374 | |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 1375 | bool MergeCompiledFile(const ResourceFile& compiled_file, io::IFile* file, bool override) { |
| 1376 | if (context_->IsVerbose()) { |
| 1377 | context_->GetDiagnostics()->Note(DiagMessage() |
| 1378 | << "merging '" << compiled_file.name |
| 1379 | << "' from compiled file " << compiled_file.source); |
| 1380 | } |
| 1381 | |
| 1382 | if (!table_merger_->MergeFile(compiled_file, override, file)) { |
| 1383 | return false; |
| 1384 | } |
| 1385 | return MergeExportedSymbols(compiled_file.source, compiled_file.exported_symbols); |
| 1386 | } |
| 1387 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1388 | // Takes a path to load as a ZIP file and merges the files within into the master ResourceTable. |
| 1389 | // If override is true, conflicting resources are allowed to override each other, in order of last |
| 1390 | // seen. |
| 1391 | // An io::IFileCollection is created from the ZIP file and added to the set of |
| 1392 | // io::IFileCollections that are open. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1393 | bool MergeArchive(const std::string& input, bool override) { |
| 1394 | if (context_->IsVerbose()) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1395 | context_->GetDiagnostics()->Note(DiagMessage() << "merging archive " << input); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1396 | } |
| 1397 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1398 | std::string error_str; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1399 | std::unique_ptr<io::ZipFileCollection> collection = |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1400 | io::ZipFileCollection::Create(input, &error_str); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1401 | if (!collection) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1402 | context_->GetDiagnostics()->Error(DiagMessage(input) << error_str); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1403 | return false; |
| 1404 | } |
| 1405 | |
| 1406 | bool error = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1407 | for (auto iter = collection->Iterator(); iter->HasNext();) { |
| 1408 | if (!MergeFile(iter->Next(), override)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1409 | error = true; |
| 1410 | } |
| 1411 | } |
| 1412 | |
| 1413 | // Make sure to move the collection into the set of IFileCollections. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1414 | collections_.push_back(std::move(collection)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1415 | return !error; |
| 1416 | } |
| 1417 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1418 | // Takes a path to load and merge into the master ResourceTable. If override is true, |
| 1419 | // conflicting resources are allowed to override each other, in order of last seen. |
| 1420 | // If the file path ends with .flata, .jar, .jack, or .zip the file is treated |
| 1421 | // as ZIP archive and the files within are merged individually. |
| 1422 | // Otherwise the file is processed on its own. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1423 | bool MergePath(const std::string& path, bool override) { |
| 1424 | if (util::EndsWith(path, ".flata") || util::EndsWith(path, ".jar") || |
| 1425 | util::EndsWith(path, ".jack") || util::EndsWith(path, ".zip")) { |
| 1426 | return MergeArchive(path, override); |
| 1427 | } else if (util::EndsWith(path, ".apk")) { |
| 1428 | return MergeStaticLibrary(path, override); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1429 | } |
| 1430 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1431 | io::IFile* file = file_collection_->InsertFile(path); |
| 1432 | return MergeFile(file, override); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1433 | } |
| 1434 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1435 | // Takes an AAPT Container file (.apc/.flat) to load and merge into the master ResourceTable. |
| 1436 | // If override is true, conflicting resources are allowed to override each other, in order of last |
| 1437 | // seen. |
| 1438 | // All other file types are ignored. This is because these files could be coming from a zip, |
| 1439 | // where we could have other files like classes.dex. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1440 | bool MergeFile(io::IFile* file, bool override) { |
| 1441 | const Source& src = file->GetSource(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1442 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1443 | if (util::EndsWith(src.path, ".xml") || util::EndsWith(src.path, ".png")) { |
Adam Lesinski | 6a396c1 | 2016-10-20 14:38:23 -0700 | [diff] [blame] | 1444 | // Since AAPT compiles these file types and appends .flat to them, seeing |
| 1445 | // their raw extensions is a sign that they weren't compiled. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1446 | const StringPiece file_type = util::EndsWith(src.path, ".xml") ? "XML" : "PNG"; |
| 1447 | context_->GetDiagnostics()->Error(DiagMessage(src) << "uncompiled " << file_type |
| 1448 | << " file passed as argument. Must be " |
| 1449 | "compiled first into .flat file."); |
Adam Lesinski | 6a396c1 | 2016-10-20 14:38:23 -0700 | [diff] [blame] | 1450 | return false; |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1451 | } else if (!util::EndsWith(src.path, ".apc") && !util::EndsWith(src.path, ".flat")) { |
| 1452 | if (context_->IsVerbose()) { |
| 1453 | context_->GetDiagnostics()->Warn(DiagMessage(src) << "ignoring unrecognized file"); |
| 1454 | return true; |
| 1455 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1456 | } |
| 1457 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1458 | std::unique_ptr<io::InputStream> input_stream = file->OpenInputStream(); |
| 1459 | if (input_stream == nullptr) { |
| 1460 | context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to open file"); |
| 1461 | return false; |
| 1462 | } |
| 1463 | |
| 1464 | if (input_stream->HadError()) { |
| 1465 | context_->GetDiagnostics()->Error(DiagMessage(src) |
| 1466 | << "failed to open file: " << input_stream->GetError()); |
| 1467 | return false; |
| 1468 | } |
| 1469 | |
| 1470 | ContainerReaderEntry* entry; |
| 1471 | ContainerReader reader(input_stream.get()); |
Mohamed Heikal | 1084432 | 2018-02-07 15:17:38 -0500 | [diff] [blame] | 1472 | |
| 1473 | if (reader.HadError()) { |
| 1474 | context_->GetDiagnostics()->Error(DiagMessage(src) |
| 1475 | << "failed to read file: " << reader.GetError()); |
| 1476 | return false; |
| 1477 | } |
| 1478 | |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1479 | while ((entry = reader.Next()) != nullptr) { |
| 1480 | if (entry->Type() == ContainerEntryType::kResTable) { |
| 1481 | pb::ResourceTable pb_table; |
| 1482 | if (!entry->GetResTable(&pb_table)) { |
| 1483 | context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to read resource table: " |
| 1484 | << entry->GetError()); |
| 1485 | return false; |
| 1486 | } |
| 1487 | |
| 1488 | ResourceTable table; |
| 1489 | std::string error; |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1490 | if (!DeserializeTableFromPb(pb_table, nullptr /*files*/, &table, &error)) { |
Adam Lesinski | 0045116 | 2017-10-03 07:44:08 -0700 | [diff] [blame] | 1491 | context_->GetDiagnostics()->Error(DiagMessage(src) |
| 1492 | << "failed to deserialize resource table: " << error); |
| 1493 | return false; |
| 1494 | } |
| 1495 | |
| 1496 | if (!table_merger_->Merge(src, &table, override)) { |
| 1497 | context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to merge resource table"); |
| 1498 | return false; |
| 1499 | } |
| 1500 | } else if (entry->Type() == ContainerEntryType::kResFile) { |
| 1501 | pb::internal::CompiledFile pb_compiled_file; |
| 1502 | off64_t offset; |
| 1503 | size_t len; |
| 1504 | if (!entry->GetResFileOffsets(&pb_compiled_file, &offset, &len)) { |
| 1505 | context_->GetDiagnostics()->Error(DiagMessage(src) << "failed to get resource file: " |
| 1506 | << entry->GetError()); |
| 1507 | return false; |
| 1508 | } |
| 1509 | |
| 1510 | ResourceFile resource_file; |
| 1511 | std::string error; |
| 1512 | if (!DeserializeCompiledFileFromPb(pb_compiled_file, &resource_file, &error)) { |
| 1513 | context_->GetDiagnostics()->Error(DiagMessage(src) |
| 1514 | << "failed to read compiled header: " << error); |
| 1515 | return false; |
| 1516 | } |
| 1517 | |
| 1518 | if (!MergeCompiledFile(resource_file, file->CreateFileSegment(offset, len), override)) { |
| 1519 | return false; |
| 1520 | } |
| 1521 | } |
| 1522 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1523 | return true; |
| 1524 | } |
| 1525 | |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1526 | bool CopyAssetsDirsToApk(IArchiveWriter* writer) { |
| 1527 | std::map<std::string, std::unique_ptr<io::RegularFile>> merged_assets; |
| 1528 | for (const std::string& assets_dir : options_.assets_dirs) { |
| 1529 | Maybe<std::vector<std::string>> files = |
| 1530 | file::FindFiles(assets_dir, context_->GetDiagnostics(), nullptr); |
| 1531 | if (!files) { |
| 1532 | return false; |
| 1533 | } |
| 1534 | |
| 1535 | for (const std::string& file : files.value()) { |
| 1536 | std::string full_key = "assets/" + file; |
| 1537 | std::string full_path = assets_dir; |
| 1538 | file::AppendPath(&full_path, file); |
| 1539 | |
| 1540 | auto iter = merged_assets.find(full_key); |
| 1541 | if (iter == merged_assets.end()) { |
| 1542 | merged_assets.emplace(std::move(full_key), |
| 1543 | util::make_unique<io::RegularFile>(Source(std::move(full_path)))); |
| 1544 | } else if (context_->IsVerbose()) { |
| 1545 | context_->GetDiagnostics()->Warn(DiagMessage(iter->second->GetSource()) |
| 1546 | << "asset file overrides '" << full_path << "'"); |
| 1547 | } |
| 1548 | } |
| 1549 | } |
| 1550 | |
| 1551 | for (auto& entry : merged_assets) { |
| 1552 | uint32_t compression_flags = ArchiveEntry::kCompress; |
| 1553 | std::string extension = file::GetExtension(entry.first).to_string(); |
| 1554 | if (options_.extensions_to_not_compress.count(extension) > 0) { |
| 1555 | compression_flags = 0u; |
| 1556 | } |
| 1557 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1558 | if (!io::CopyFileToArchive(context_, entry.second.get(), entry.first, compression_flags, |
| 1559 | writer)) { |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1560 | return false; |
| 1561 | } |
| 1562 | } |
| 1563 | return true; |
| 1564 | } |
| 1565 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1566 | // Writes the AndroidManifest, ResourceTable, and all XML files referenced by the ResourceTable |
| 1567 | // to the IArchiveWriter. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1568 | bool WriteApk(IArchiveWriter* writer, proguard::KeepSet* keep_set, xml::XmlResource* manifest, |
| 1569 | ResourceTable* table) { |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1570 | const bool keep_raw_values = context_->GetPackageType() == PackageType::kStaticLib; |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1571 | bool result = FlattenXml(context_, *manifest, "AndroidManifest.xml", keep_raw_values, |
| 1572 | true /*utf16*/, options_.output_format, writer); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1573 | if (!result) { |
| 1574 | return false; |
| 1575 | } |
| 1576 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1577 | ResourceFileFlattenerOptions file_flattener_options; |
| 1578 | file_flattener_options.keep_raw_values = keep_raw_values; |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1579 | file_flattener_options.do_not_compress_anything = options_.do_not_compress_anything; |
| 1580 | file_flattener_options.extensions_to_not_compress = options_.extensions_to_not_compress; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1581 | file_flattener_options.no_auto_version = options_.no_auto_version; |
| 1582 | file_flattener_options.no_version_vectors = options_.no_version_vectors; |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 1583 | file_flattener_options.no_version_transitions = options_.no_version_transitions; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1584 | file_flattener_options.no_xml_namespaces = options_.no_xml_namespaces; |
| 1585 | file_flattener_options.update_proguard_spec = |
| 1586 | static_cast<bool>(options_.generate_proguard_rules_path); |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 1587 | file_flattener_options.output_format = options_.output_format; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1588 | |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1589 | ResourceFileFlattener file_flattener(file_flattener_options, context_, keep_set); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1590 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1591 | if (!file_flattener.Flatten(table, writer)) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1592 | context_->GetDiagnostics()->Error(DiagMessage() << "failed linking file resources"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1593 | return false; |
| 1594 | } |
| 1595 | |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 1596 | // Hack to fix b/68820737. |
| 1597 | // We need to modify the ResourceTable's package name, but that should NOT affect |
| 1598 | // anything else being generated, which includes the Java classes. |
| 1599 | // If required, the package name is modifed before flattening, and then modified back |
| 1600 | // to its original name. |
| 1601 | ResourceTablePackage* package_to_rewrite = nullptr; |
| 1602 | if (context_->GetPackageId() > kAppPackageId && |
| 1603 | included_feature_base_ == make_value(context_->GetCompilationPackage())) { |
| 1604 | // The base APK is included, and this is a feature split. If the base package is |
| 1605 | // the same as this package, then we are building an old style Android Instant Apps feature |
| 1606 | // split and must apply this workaround to avoid requiring namespaces support. |
| 1607 | package_to_rewrite = table->FindPackage(context_->GetCompilationPackage()); |
| 1608 | if (package_to_rewrite != nullptr) { |
| 1609 | CHECK_EQ(1u, table->packages.size()) << "can't change name of package when > 1 package"; |
| 1610 | |
| 1611 | std::string new_package_name = |
| 1612 | StringPrintf("%s.%s", package_to_rewrite->name.c_str(), |
| 1613 | app_info_.split_name.value_or_default("feature").c_str()); |
| 1614 | |
| 1615 | if (context_->IsVerbose()) { |
| 1616 | context_->GetDiagnostics()->Note( |
| 1617 | DiagMessage() << "rewriting resource package name for feature split to '" |
| 1618 | << new_package_name << "'"); |
| 1619 | } |
| 1620 | package_to_rewrite->name = new_package_name; |
| 1621 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1622 | } |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 1623 | |
| 1624 | bool success = FlattenTable(table, options_.output_format, writer); |
| 1625 | |
| 1626 | if (package_to_rewrite != nullptr) { |
| 1627 | // Change the name back. |
| 1628 | package_to_rewrite->name = context_->GetCompilationPackage(); |
| 1629 | if (package_to_rewrite->id) { |
| 1630 | table->included_packages_.erase(package_to_rewrite->id.value()); |
| 1631 | } |
| 1632 | } |
| 1633 | |
| 1634 | if (!success) { |
| 1635 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to write resource table"); |
| 1636 | } |
| 1637 | return success; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1638 | } |
| 1639 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1640 | int Run(const std::vector<std::string>& input_files) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1641 | // Load the AndroidManifest.xml |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1642 | std::unique_ptr<xml::XmlResource> manifest_xml = |
| 1643 | LoadXml(options_.manifest_path, context_->GetDiagnostics()); |
| 1644 | if (!manifest_xml) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1645 | return 1; |
| 1646 | } |
| 1647 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1648 | // First extract the Package name without modifying it (via --rename-manifest-package). |
| 1649 | if (Maybe<AppInfo> maybe_app_info = |
| 1650 | ExtractAppInfoFromManifest(manifest_xml.get(), context_->GetDiagnostics())) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1651 | const AppInfo& app_info = maybe_app_info.value(); |
| 1652 | context_->SetCompilationPackage(app_info.package); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1653 | } |
| 1654 | |
Adam Lesinski | c628437 | 2017-12-04 13:46:23 -0800 | [diff] [blame] | 1655 | // Now that the compilation package is set, load the dependencies. This will also extract |
| 1656 | // the Android framework's versionCode and versionName, if they exist. |
| 1657 | if (!LoadSymbolsFromIncludePaths()) { |
| 1658 | return 1; |
| 1659 | } |
| 1660 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1661 | ManifestFixer manifest_fixer(options_.manifest_fixer_options); |
| 1662 | if (!manifest_fixer.Consume(context_, manifest_xml.get())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1663 | return 1; |
| 1664 | } |
| 1665 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1666 | Maybe<AppInfo> maybe_app_info = |
| 1667 | ExtractAppInfoFromManifest(manifest_xml.get(), context_->GetDiagnostics()); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1668 | if (!maybe_app_info) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1669 | return 1; |
| 1670 | } |
| 1671 | |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 1672 | app_info_ = maybe_app_info.value(); |
| 1673 | context_->SetMinSdkVersion(app_info_.min_sdk_version.value_or_default(0)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1674 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1675 | context_->SetNameManglerPolicy(NameManglerPolicy{context_->GetCompilationPackage()}); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1676 | |
| 1677 | // Override the package ID when it is "android". |
| 1678 | if (context_->GetCompilationPackage() == "android") { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1679 | context_->SetPackageId(0x01); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1680 | |
| 1681 | // Verify we're building a regular app. |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1682 | if (context_->GetPackageType() != PackageType::kApp) { |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1683 | context_->GetDiagnostics()->Error( |
| 1684 | DiagMessage() << "package 'android' can only be built as a regular app"); |
| 1685 | return 1; |
| 1686 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1687 | } |
| 1688 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1689 | TableMergerOptions table_merger_options; |
| 1690 | table_merger_options.auto_add_overlay = options_.auto_add_overlay; |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1691 | table_merger_ = util::make_unique<TableMerger>(context_, &final_table_, table_merger_options); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1692 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1693 | if (context_->IsVerbose()) { |
| 1694 | context_->GetDiagnostics()->Note(DiagMessage() |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 1695 | << StringPrintf("linking package '%s' using package ID %02x", |
| 1696 | context_->GetCompilationPackage().data(), |
| 1697 | context_->GetPackageId())); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1698 | } |
| 1699 | |
Adam Lesinski | 2427dce | 2017-11-30 15:10:28 -0800 | [diff] [blame] | 1700 | // Extract symbols from AndroidManifest.xml, since this isn't merged like the other XML files |
| 1701 | // in res/**/*. |
| 1702 | { |
| 1703 | XmlIdCollector collector; |
| 1704 | if (!collector.Consume(context_, manifest_xml.get())) { |
| 1705 | return false; |
| 1706 | } |
| 1707 | |
| 1708 | if (!MergeExportedSymbols(manifest_xml->file.source, manifest_xml->file.exported_symbols)) { |
| 1709 | return false; |
| 1710 | } |
| 1711 | } |
| 1712 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1713 | for (const std::string& input : input_files) { |
| 1714 | if (!MergePath(input, false)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1715 | context_->GetDiagnostics()->Error(DiagMessage() << "failed parsing input"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1716 | return 1; |
| 1717 | } |
| 1718 | } |
| 1719 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1720 | for (const std::string& input : options_.overlay_files) { |
| 1721 | if (!MergePath(input, true)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1722 | context_->GetDiagnostics()->Error(DiagMessage() << "failed parsing overlays"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1723 | return 1; |
| 1724 | } |
| 1725 | } |
| 1726 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1727 | if (!VerifyNoExternalPackages()) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1728 | return 1; |
| 1729 | } |
| 1730 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1731 | if (context_->GetPackageType() != PackageType::kStaticLib) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1732 | PrivateAttributeMover mover; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1733 | if (!mover.Consume(context_, &final_table_)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1734 | context_->GetDiagnostics()->Error(DiagMessage() << "failed moving private attributes"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1735 | return 1; |
| 1736 | } |
| 1737 | |
| 1738 | // Assign IDs if we are building a regular app. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1739 | IdAssigner id_assigner(&options_.stable_id_map); |
| 1740 | if (!id_assigner.Consume(context_, &final_table_)) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1741 | context_->GetDiagnostics()->Error(DiagMessage() << "failed assigning IDs"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1742 | return 1; |
| 1743 | } |
| 1744 | |
| 1745 | // Now grab each ID and emit it as a file. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1746 | if (options_.resource_id_map_path) { |
| 1747 | for (auto& package : final_table_.packages) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1748 | for (auto& type : package->types) { |
| 1749 | for (auto& entry : type->entries) { |
| 1750 | ResourceName name(package->name, type->type, entry->name); |
| 1751 | // The IDs are guaranteed to exist. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1752 | options_.stable_id_map[std::move(name)] = |
| 1753 | ResourceId(package->id.value(), type->id.value(), entry->id.value()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1754 | } |
| 1755 | } |
| 1756 | } |
| 1757 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1758 | if (!WriteStableIdMapToPath(context_->GetDiagnostics(), options_.stable_id_map, |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1759 | options_.resource_id_map_path.value())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1760 | return 1; |
| 1761 | } |
| 1762 | } |
| 1763 | } else { |
| 1764 | // Static libs are merged with other apps, and ID collisions are bad, so |
| 1765 | // verify that |
| 1766 | // no IDs have been set. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1767 | if (!VerifyNoIdsSet()) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1768 | return 1; |
| 1769 | } |
| 1770 | } |
| 1771 | |
| 1772 | // Add the names to mangle based on our source merge earlier. |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1773 | context_->SetNameManglerPolicy( |
| 1774 | NameManglerPolicy{context_->GetCompilationPackage(), table_merger_->merged_packages()}); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1775 | |
| 1776 | // Add our table to the symbol table. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1777 | context_->GetExternalSymbols()->PrependSource( |
| 1778 | util::make_unique<ResourceTableSymbolSource>(&final_table_)); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1779 | |
Adam Lesinski | 1e4b0e5 | 2017-04-27 15:01:10 -0700 | [diff] [blame] | 1780 | // Workaround for pre-O runtime that would treat negative resource IDs |
| 1781 | // (any ID with a package ID > 7f) as invalid. Intercept any ID (PPTTEEEE) with PP > 0x7f |
| 1782 | // and type == 'id', and return the ID 0x7fPPEEEE. IDs don't need to be real resources, they |
| 1783 | // are just identifiers. |
| 1784 | if (context_->GetMinSdkVersion() < SDK_O && context_->GetPackageType() == PackageType::kApp) { |
| 1785 | if (context_->IsVerbose()) { |
| 1786 | context_->GetDiagnostics()->Note(DiagMessage() |
| 1787 | << "enabling pre-O feature split ID rewriting"); |
| 1788 | } |
| 1789 | context_->GetExternalSymbols()->SetDelegate( |
| 1790 | util::make_unique<FeatureSplitSymbolTableDelegate>(context_)); |
| 1791 | } |
| 1792 | |
Adam Lesinski | 34a1687 | 2018-02-23 16:18:10 -0800 | [diff] [blame] | 1793 | // Before we process anything, remove the resources whose default values don't exist. |
| 1794 | // We want to force any references to these to fail the build. |
| 1795 | if (!NoDefaultResourceRemover{}.Consume(context_, &final_table_)) { |
| 1796 | context_->GetDiagnostics()->Error(DiagMessage() |
| 1797 | << "failed removing resources with no defaults"); |
| 1798 | return 1; |
| 1799 | } |
| 1800 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1801 | ReferenceLinker linker; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1802 | if (!linker.Consume(context_, &final_table_)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1803 | context_->GetDiagnostics()->Error(DiagMessage() << "failed linking references"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1804 | return 1; |
| 1805 | } |
| 1806 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1807 | if (context_->GetPackageType() == PackageType::kStaticLib) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1808 | if (!options_.products.empty()) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1809 | context_->GetDiagnostics()->Warn(DiagMessage() |
| 1810 | << "can't select products when building static library"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1811 | } |
| 1812 | } else { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1813 | ProductFilter product_filter(options_.products); |
| 1814 | if (!product_filter.Consume(context_, &final_table_)) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1815 | context_->GetDiagnostics()->Error(DiagMessage() << "failed stripping products"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1816 | return 1; |
| 1817 | } |
| 1818 | } |
| 1819 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1820 | if (!options_.no_auto_version) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1821 | AutoVersioner versioner; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1822 | if (!versioner.Consume(context_, &final_table_)) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1823 | context_->GetDiagnostics()->Error(DiagMessage() << "failed versioning styles"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1824 | return 1; |
| 1825 | } |
| 1826 | } |
| 1827 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1828 | if (context_->GetPackageType() != PackageType::kStaticLib && context_->GetMinSdkVersion() > 0) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1829 | if (context_->IsVerbose()) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1830 | context_->GetDiagnostics()->Note(DiagMessage() |
| 1831 | << "collapsing resource versions for minimum SDK " |
| 1832 | << context_->GetMinSdkVersion()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1833 | } |
| 1834 | |
| 1835 | VersionCollapser collapser; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1836 | if (!collapser.Consume(context_, &final_table_)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1837 | return 1; |
| 1838 | } |
| 1839 | } |
| 1840 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1841 | if (!options_.no_resource_deduping) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1842 | ResourceDeduper deduper; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1843 | if (!deduper.Consume(context_, &final_table_)) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1844 | context_->GetDiagnostics()->Error(DiagMessage() << "failed deduping resources"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1845 | return 1; |
| 1846 | } |
| 1847 | } |
| 1848 | |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 1849 | proguard::KeepSet proguard_keep_set = |
| 1850 | proguard::KeepSet(options_.generate_conditional_proguard_rules); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1851 | proguard::KeepSet proguard_main_dex_keep_set; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1852 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 1853 | if (context_->GetPackageType() == PackageType::kStaticLib) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1854 | if (options_.table_splitter_options.config_filter != nullptr || |
Pierre Lecesne | 672384b | 2017-02-06 10:29:02 +0000 | [diff] [blame] | 1855 | !options_.table_splitter_options.preferred_densities.empty()) { |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 1856 | context_->GetDiagnostics()->Warn(DiagMessage() |
| 1857 | << "can't strip resources when building static library"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1858 | } |
| 1859 | } else { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1860 | // Adjust the SplitConstraints so that their SDK version is stripped if it is less than or |
| 1861 | // equal to the minSdk. |
| 1862 | options_.split_constraints = |
| 1863 | AdjustSplitConstraintsForMinSdk(context_->GetMinSdkVersion(), options_.split_constraints); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1864 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1865 | TableSplitter table_splitter(options_.split_constraints, options_.table_splitter_options); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1866 | if (!table_splitter.VerifySplitConstraints(context_)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1867 | return 1; |
| 1868 | } |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1869 | table_splitter.SplitTable(&final_table_); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1870 | |
| 1871 | // Now we need to write out the Split APKs. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1872 | auto path_iter = options_.split_paths.begin(); |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1873 | auto split_constraints_iter = options_.split_constraints.begin(); |
| 1874 | for (std::unique_ptr<ResourceTable>& split_table : table_splitter.splits()) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1875 | if (context_->IsVerbose()) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1876 | context_->GetDiagnostics()->Note(DiagMessage(*path_iter) |
| 1877 | << "generating split with configurations '" |
| 1878 | << util::Joiner(split_constraints_iter->configs, ", ") |
| 1879 | << "'"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1880 | } |
| 1881 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1882 | std::unique_ptr<IArchiveWriter> archive_writer = MakeArchiveWriter(*path_iter); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1883 | if (!archive_writer) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1884 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to create archive"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1885 | return 1; |
| 1886 | } |
| 1887 | |
| 1888 | // Generate an AndroidManifest.xml for each split. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1889 | std::unique_ptr<xml::XmlResource> split_manifest = |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 1890 | GenerateSplitManifest(app_info_, *split_constraints_iter); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1891 | |
| 1892 | XmlReferenceLinker linker; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1893 | if (!linker.Consume(context_, split_manifest.get())) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1894 | context_->GetDiagnostics()->Error(DiagMessage() |
| 1895 | << "failed to create Split AndroidManifest.xml"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1896 | return 1; |
| 1897 | } |
| 1898 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 1899 | if (!WriteApk(archive_writer.get(), &proguard_keep_set, split_manifest.get(), |
| 1900 | split_table.get())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1901 | return 1; |
| 1902 | } |
| 1903 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1904 | ++path_iter; |
| 1905 | ++split_constraints_iter; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1906 | } |
| 1907 | } |
| 1908 | |
| 1909 | // Start writing the base APK. |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1910 | std::unique_ptr<IArchiveWriter> archive_writer = MakeArchiveWriter(options_.output_path); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1911 | if (!archive_writer) { |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1912 | context_->GetDiagnostics()->Error(DiagMessage() << "failed to create archive"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1913 | return 1; |
| 1914 | } |
| 1915 | |
| 1916 | bool error = false; |
| 1917 | { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1918 | // AndroidManifest.xml has no resource name, but the CallSite is built from the name |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1919 | // (aka, which package the AndroidManifest.xml is coming from). |
| 1920 | // So we give it a package name so it can see local resources. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1921 | manifest_xml->file.name.package = context_->GetCompilationPackage(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1922 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1923 | XmlReferenceLinker manifest_linker; |
| 1924 | if (manifest_linker.Consume(context_, manifest_xml.get())) { |
| 1925 | if (options_.generate_proguard_rules_path && |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 1926 | !proguard::CollectProguardRulesForManifest(manifest_xml.get(), &proguard_keep_set)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1927 | error = true; |
| 1928 | } |
| 1929 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1930 | if (options_.generate_main_dex_proguard_rules_path && |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 1931 | !proguard::CollectProguardRulesForManifest(manifest_xml.get(), |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1932 | &proguard_main_dex_keep_set, true)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1933 | error = true; |
Alexandria Cornwall | 637b482 | 2016-08-11 09:53:16 -0700 | [diff] [blame] | 1934 | } |
| 1935 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1936 | if (options_.generate_java_class_path) { |
| 1937 | if (!WriteManifestJavaFile(manifest_xml.get())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1938 | error = true; |
| 1939 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1940 | } |
| 1941 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1942 | if (options_.no_xml_namespaces) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1943 | // PackageParser will fail if URIs are removed from |
| 1944 | // AndroidManifest.xml. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1945 | XmlNamespaceRemover namespace_remover(true /* keepUris */); |
| 1946 | if (!namespace_remover.Consume(context_, manifest_xml.get())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1947 | error = true; |
| 1948 | } |
Rohit Agrawal | e49bb30 | 2016-04-22 12:27:55 -0700 | [diff] [blame] | 1949 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1950 | } else { |
| 1951 | error = true; |
| 1952 | } |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 1953 | } |
Adam Lesinski | fb48d29 | 2015-11-07 15:52:13 -0800 | [diff] [blame] | 1954 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1955 | if (error) { |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1956 | context_->GetDiagnostics()->Error(DiagMessage() << "failed processing manifest"); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1957 | return 1; |
| 1958 | } |
Adam Lesinski | a6fe345 | 2015-12-09 15:20:52 -0800 | [diff] [blame] | 1959 | |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 1960 | if (!WriteApk(archive_writer.get(), &proguard_keep_set, manifest_xml.get(), &final_table_)) { |
| 1961 | return 1; |
| 1962 | } |
| 1963 | |
| 1964 | if (!CopyAssetsDirsToApk(archive_writer.get())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1965 | return 1; |
| 1966 | } |
Adam Lesinski | fb48d29 | 2015-11-07 15:52:13 -0800 | [diff] [blame] | 1967 | |
Izabela Orlowska | 23a6e1e | 2017-12-05 14:52:07 +0000 | [diff] [blame] | 1968 | if (options_.generate_java_class_path || options_.generate_text_symbols_path) { |
Adam Lesinski | 1ef0fa9 | 2017-08-15 21:32:49 -0700 | [diff] [blame] | 1969 | if (!GenerateJavaClasses()) { |
Adam Lesinski | b5dc4bd | 2017-02-22 19:29:29 -0800 | [diff] [blame] | 1970 | return 1; |
| 1971 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1972 | } |
| 1973 | |
Adam Lesinski | b5dc4bd | 2017-02-22 19:29:29 -0800 | [diff] [blame] | 1974 | if (!WriteProguardFile(options_.generate_proguard_rules_path, proguard_keep_set)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1975 | return 1; |
| 1976 | } |
| 1977 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1978 | if (!WriteProguardFile(options_.generate_main_dex_proguard_rules_path, |
| 1979 | proguard_main_dex_keep_set)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1980 | return 1; |
| 1981 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1982 | return 0; |
| 1983 | } |
| 1984 | |
| 1985 | private: |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1986 | LinkOptions options_; |
| 1987 | LinkContext* context_; |
| 1988 | ResourceTable final_table_; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1989 | |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 1990 | AppInfo app_info_; |
| 1991 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1992 | std::unique_ptr<TableMerger> table_merger_; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1993 | |
| 1994 | // A pointer to the FileCollection representing the filesystem (not archives). |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1995 | std::unique_ptr<io::FileCollection> file_collection_; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1996 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 1997 | // A vector of IFileCollections. This is mainly here to retain ownership of the |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 1998 | // collections. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 1999 | std::vector<std::unique_ptr<io::IFileCollection>> collections_; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2000 | |
Adam Lesinski | 8780eb6 | 2017-10-31 17:44:39 -0700 | [diff] [blame] | 2001 | // The set of merged APKs. This is mainly here to retain ownership of the APKs. |
| 2002 | std::vector<std::unique_ptr<LoadedApk>> merged_apks_; |
| 2003 | |
| 2004 | // The set of included APKs (not merged). This is mainly here to retain ownership of the APKs. |
| 2005 | std::vector<std::unique_ptr<LoadedApk>> static_library_includes_; |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 2006 | |
| 2007 | // The set of shared libraries being used, mapping their assigned package ID to package name. |
| 2008 | std::map<size_t, std::string> shared_libs_; |
Adam Lesinski | 490595a | 2017-11-07 17:08:07 -0800 | [diff] [blame] | 2009 | |
| 2010 | // The package name of the base application, if it is included. |
| 2011 | Maybe<std::string> included_feature_base_; |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 2012 | }; |
| 2013 | |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 2014 | int Link(const std::vector<StringPiece>& args, IDiagnostics* diagnostics) { |
| 2015 | LinkContext context(diagnostics); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2016 | LinkOptions options; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2017 | std::vector<std::string> overlay_arg_list; |
| 2018 | std::vector<std::string> extra_java_packages; |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2019 | Maybe<std::string> package_id; |
Adam Lesinski | 113ee09 | 2017-04-03 19:38:25 -0700 | [diff] [blame] | 2020 | std::vector<std::string> configs; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2021 | Maybe<std::string> preferred_density; |
| 2022 | Maybe<std::string> product_list; |
| 2023 | bool legacy_x_flag = false; |
| 2024 | bool require_localization = false; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2025 | bool verbose = false; |
Adam Lesinski | ceb9b2f | 2017-02-16 12:05:42 -0800 | [diff] [blame] | 2026 | bool shared_lib = false; |
| 2027 | bool static_lib = false; |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 2028 | bool proto_format = false; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2029 | Maybe<std::string> stable_id_file_path; |
| 2030 | std::vector<std::string> split_args; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2031 | Flags flags = |
| 2032 | Flags() |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2033 | .RequiredFlag("-o", "Output path.", &options.output_path) |
| 2034 | .RequiredFlag("--manifest", "Path to the Android manifest to build.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2035 | &options.manifest_path) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2036 | .OptionalFlagList("-I", "Adds an Android APK to link against.", &options.include_paths) |
Adam Lesinski | b39ad7c | 2017-03-13 11:40:48 -0700 | [diff] [blame] | 2037 | .OptionalFlagList("-A", |
| 2038 | "An assets directory to include in the APK. These are unprocessed.", |
| 2039 | &options.assets_dirs) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2040 | .OptionalFlagList("-R", |
| 2041 | "Compilation unit to link, using `overlay` semantics.\n" |
| 2042 | "The last conflicting resource given takes precedence.", |
| 2043 | &overlay_arg_list) |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2044 | .OptionalFlag("--package-id", |
| 2045 | "Specify the package ID to use for this app. Must be greater or equal to\n" |
| 2046 | "0x7f and can't be used with --static-lib or --shared-lib.", |
| 2047 | &package_id) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2048 | .OptionalFlag("--java", "Directory in which to generate R.java.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2049 | &options.generate_java_class_path) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2050 | .OptionalFlag("--proguard", "Output file for generated Proguard rules.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2051 | &options.generate_proguard_rules_path) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2052 | .OptionalFlag("--proguard-main-dex", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2053 | "Output file for generated Proguard rules for the main dex.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2054 | &options.generate_main_dex_proguard_rules_path) |
Adam Koski | dc21dea | 2017-07-21 10:55:27 -0700 | [diff] [blame] | 2055 | .OptionalSwitch("--proguard-conditional-keep-rules", |
| 2056 | "Generate conditional Proguard keep rules.", |
| 2057 | &options.generate_conditional_proguard_rules) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2058 | .OptionalSwitch("--no-auto-version", |
| 2059 | "Disables automatic style and layout SDK versioning.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2060 | &options.no_auto_version) |
| 2061 | .OptionalSwitch("--no-version-vectors", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2062 | "Disables automatic versioning of vector drawables. Use this only\n" |
| 2063 | "when building with vector drawable support library.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2064 | &options.no_version_vectors) |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 2065 | .OptionalSwitch("--no-version-transitions", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2066 | "Disables automatic versioning of transition resources. Use this only\n" |
| 2067 | "when building with transition support library.", |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 2068 | &options.no_version_transitions) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2069 | .OptionalSwitch("--no-resource-deduping", |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2070 | "Disables automatic deduping of resources with\n" |
| 2071 | "identical values across compatible configurations.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2072 | &options.no_resource_deduping) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2073 | .OptionalSwitch("--enable-sparse-encoding", |
| 2074 | "Enables encoding sparse entries using a binary search tree.\n" |
| 2075 | "This decreases APK size at the cost of resource retrieval performance.", |
| 2076 | &options.table_flattener_options.use_sparse_entries) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2077 | .OptionalSwitch("-x", "Legacy flag that specifies to use the package identifier 0x01.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2078 | &legacy_x_flag) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2079 | .OptionalSwitch("-z", "Require localization of strings marked 'suggested'.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2080 | &require_localization) |
Adam Lesinski | 113ee09 | 2017-04-03 19:38:25 -0700 | [diff] [blame] | 2081 | .OptionalFlagList("-c", |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 2082 | "Comma separated list of configurations to include. The default\n" |
| 2083 | "is all configurations.", |
| 2084 | &configs) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2085 | .OptionalFlag("--preferred-density", |
| 2086 | "Selects the closest matching density and strips out all others.", |
| 2087 | &preferred_density) |
| 2088 | .OptionalFlag("--product", "Comma separated list of product names to keep", &product_list) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2089 | .OptionalSwitch("--output-to-dir", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2090 | "Outputs the APK contents to a directory specified by -o.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2091 | &options.output_to_directory) |
| 2092 | .OptionalSwitch("--no-xml-namespaces", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2093 | "Removes XML namespace prefix and URI information from\n" |
| 2094 | "AndroidManifest.xml and XML binaries in res/*.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2095 | &options.no_xml_namespaces) |
| 2096 | .OptionalFlag("--min-sdk-version", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2097 | "Default minimum SDK version to use for AndroidManifest.xml.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2098 | &options.manifest_fixer_options.min_sdk_version_default) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2099 | .OptionalFlag("--target-sdk-version", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2100 | "Default target SDK version to use for AndroidManifest.xml.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2101 | &options.manifest_fixer_options.target_sdk_version_default) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2102 | .OptionalFlag("--version-code", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2103 | "Version code (integer) to inject into the AndroidManifest.xml if none is\n" |
| 2104 | "present.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2105 | &options.manifest_fixer_options.version_code_default) |
| 2106 | .OptionalFlag("--version-name", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2107 | "Version name to inject into the AndroidManifest.xml if none is present.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2108 | &options.manifest_fixer_options.version_name_default) |
Todd Kennedy | 9f6dec1 | 2018-04-20 12:29:29 -0700 | [diff] [blame^] | 2109 | .OptionalFlag("--compile-sdk-version-code", |
| 2110 | "Version code (integer) to inject into the AndroidManifest.xml if none is\n" |
| 2111 | "present.", |
| 2112 | &options.manifest_fixer_options.compile_sdk_version) |
| 2113 | .OptionalFlag("--compile-sdk-version-name", |
| 2114 | "Version name to inject into the AndroidManifest.xml if none is present.", |
| 2115 | &options.manifest_fixer_options.compile_sdk_version_codename) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2116 | .OptionalSwitch("--shared-lib", "Generates a shared Android runtime library.", |
| 2117 | &shared_lib) |
| 2118 | .OptionalSwitch("--static-lib", "Generate a static Android library.", &static_lib) |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 2119 | .OptionalSwitch("--proto-format", |
| 2120 | "Generates compiled resources in Protobuf format.\n" |
| 2121 | "Suitable as input to the bundle tool for generating an App Bundle.", |
| 2122 | &proto_format) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2123 | .OptionalSwitch("--no-static-lib-packages", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2124 | "Merge all library resources under the app's package.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2125 | &options.no_static_lib_packages) |
| 2126 | .OptionalSwitch("--non-final-ids", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2127 | "Generates R.java without the final modifier. This is implied when\n" |
| 2128 | "--static-lib is specified.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2129 | &options.generate_non_final_ids) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2130 | .OptionalFlag("--stable-ids", "File containing a list of name to ID mapping.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2131 | &stable_id_file_path) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2132 | .OptionalFlag("--emit-ids", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2133 | "Emit a file at the given path with a list of name to ID mappings,\n" |
| 2134 | "suitable for use with --stable-ids.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2135 | &options.resource_id_map_path) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2136 | .OptionalFlag("--private-symbols", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2137 | "Package name to use when generating R.java for private symbols.\n" |
| 2138 | "If not specified, public and private symbols will use the application's\n" |
| 2139 | "package name.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2140 | &options.private_symbols) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2141 | .OptionalFlag("--custom-package", "Custom Java package under which to generate R.java.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2142 | &options.custom_java_package) |
| 2143 | .OptionalFlagList("--extra-packages", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2144 | "Generate the same R.java but with different package names.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2145 | &extra_java_packages) |
| 2146 | .OptionalFlagList("--add-javadoc-annotation", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2147 | "Adds a JavaDoc annotation to all generated Java classes.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2148 | &options.javadoc_annotations) |
Adam Lesinski | 418763f | 2017-04-11 17:36:53 -0700 | [diff] [blame] | 2149 | .OptionalFlag("--output-text-symbols", |
| 2150 | "Generates a text file containing the resource symbols of the R class in\n" |
| 2151 | "the specified folder.", |
| 2152 | &options.generate_text_symbols_path) |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2153 | .OptionalSwitch("--auto-add-overlay", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2154 | "Allows the addition of new resources in overlays without\n" |
| 2155 | "<add-resource> tags.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2156 | &options.auto_add_overlay) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2157 | .OptionalFlag("--rename-manifest-package", "Renames the package in AndroidManifest.xml.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2158 | &options.manifest_fixer_options.rename_manifest_package) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2159 | .OptionalFlag("--rename-instrumentation-target-package", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2160 | "Changes the name of the target package for instrumentation. Most useful\n" |
| 2161 | "when used in conjunction with --rename-manifest-package.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2162 | &options.manifest_fixer_options.rename_instrumentation_target_package) |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2163 | .OptionalFlagList("-0", "File extensions not to compress.", |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2164 | &options.extensions_to_not_compress) |
Izabela Orlowska | ad9e132 | 2017-12-19 16:22:42 +0000 | [diff] [blame] | 2165 | .OptionalSwitch("--warn-manifest-validation", |
| 2166 | "Treat manifest validation errors as warnings.", |
| 2167 | &options.manifest_fixer_options.warn_validation) |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2168 | .OptionalFlagList("--split", |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2169 | "Split resources matching a set of configs out to a Split APK.\n" |
Adam Lesinski | db09157 | 2017-04-13 12:48:56 -0700 | [diff] [blame] | 2170 | "Syntax: path/to/output.apk:<config>[,<config>[...]].\n" |
| 2171 | "On Windows, use a semicolon ';' separator instead.", |
Adam Lesinski | c8f71aa | 2017-02-08 07:03:50 -0800 | [diff] [blame] | 2172 | &split_args) |
Ryan Mitchell | e5b38a6 | 2018-03-23 13:35:00 -0700 | [diff] [blame] | 2173 | .OptionalSwitch("-v", "Enables verbose logging.", &verbose) |
| 2174 | .OptionalSwitch("--debug-mode", |
| 2175 | "Inserts android:debuggable=\"true\" in to the application node of the\n" |
| 2176 | "manifest, making the application debuggable even on production devices.", |
| 2177 | &options.manifest_fixer_options.debug_mode); |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 2178 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2179 | if (!flags.Parse("aapt2 link", args, &std::cerr)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2180 | return 1; |
| 2181 | } |
| 2182 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2183 | // Expand all argument-files passed into the command line. These start with '@'. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2184 | std::vector<std::string> arg_list; |
| 2185 | for (const std::string& arg : flags.GetArgs()) { |
| 2186 | if (util::StartsWith(arg, "@")) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2187 | const std::string path = arg.substr(1, arg.size() - 1); |
| 2188 | std::string error; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2189 | if (!file::AppendArgsFromFile(path, &arg_list, &error)) { |
| 2190 | context.GetDiagnostics()->Error(DiagMessage(path) << error); |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 2191 | return 1; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2192 | } |
| 2193 | } else { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2194 | arg_list.push_back(arg); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2195 | } |
| 2196 | } |
| 2197 | |
| 2198 | // Expand all argument-files passed to -R. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2199 | for (const std::string& arg : overlay_arg_list) { |
| 2200 | if (util::StartsWith(arg, "@")) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2201 | const std::string path = arg.substr(1, arg.size() - 1); |
| 2202 | std::string error; |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2203 | if (!file::AppendArgsFromFile(path, &options.overlay_files, &error)) { |
| 2204 | context.GetDiagnostics()->Error(DiagMessage(path) << error); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2205 | return 1; |
| 2206 | } |
| 2207 | } else { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2208 | options.overlay_files.push_back(arg); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2209 | } |
| 2210 | } |
| 2211 | |
| 2212 | if (verbose) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2213 | context.SetVerbose(verbose); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2214 | } |
| 2215 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 2216 | if (int{shared_lib} + int{static_lib} + int{proto_format} > 1) { |
| 2217 | context.GetDiagnostics()->Error( |
| 2218 | DiagMessage() |
| 2219 | << "only one of --shared-lib, --static-lib, or --proto_format can be defined"); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2220 | return 1; |
| 2221 | } |
| 2222 | |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 2223 | // The default build type. |
| 2224 | context.SetPackageType(PackageType::kApp); |
| 2225 | context.SetPackageId(kAppPackageId); |
| 2226 | |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2227 | if (shared_lib) { |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 2228 | context.SetPackageType(PackageType::kSharedLib); |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2229 | context.SetPackageId(0x00); |
| 2230 | } else if (static_lib) { |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 2231 | context.SetPackageType(PackageType::kStaticLib); |
Adam Lesinski | e59f0d8 | 2017-10-13 09:36:53 -0700 | [diff] [blame] | 2232 | options.output_format = OutputFormat::kProto; |
| 2233 | } else if (proto_format) { |
| 2234 | options.output_format = OutputFormat::kProto; |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2235 | } |
| 2236 | |
| 2237 | if (package_id) { |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 2238 | if (context.GetPackageType() != PackageType::kApp) { |
Adam Lesinski | f34b6f4 | 2017-03-03 16:33:26 -0800 | [diff] [blame] | 2239 | context.GetDiagnostics()->Error( |
| 2240 | DiagMessage() << "can't specify --package-id when not building a regular app"); |
| 2241 | return 1; |
| 2242 | } |
| 2243 | |
| 2244 | const Maybe<uint32_t> maybe_package_id_int = ResourceUtils::ParseInt(package_id.value()); |
| 2245 | if (!maybe_package_id_int) { |
| 2246 | context.GetDiagnostics()->Error(DiagMessage() << "package ID '" << package_id.value() |
| 2247 | << "' is not a valid integer"); |
| 2248 | return 1; |
| 2249 | } |
| 2250 | |
| 2251 | const uint32_t package_id_int = maybe_package_id_int.value(); |
| 2252 | if (package_id_int < kAppPackageId || package_id_int > std::numeric_limits<uint8_t>::max()) { |
| 2253 | context.GetDiagnostics()->Error( |
| 2254 | DiagMessage() << StringPrintf( |
| 2255 | "invalid package ID 0x%02x. Must be in the range 0x7f-0xff.", package_id_int)); |
| 2256 | return 1; |
| 2257 | } |
| 2258 | context.SetPackageId(static_cast<uint8_t>(package_id_int)); |
| 2259 | } |
| 2260 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2261 | // Populate the set of extra packages for which to generate R.java. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2262 | for (std::string& extra_package : extra_java_packages) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2263 | // A given package can actually be a colon separated list of packages. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2264 | for (StringPiece package : util::Split(extra_package, ':')) { |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 2265 | options.extra_java_packages.insert(package.to_string()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2266 | } |
| 2267 | } |
| 2268 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2269 | if (product_list) { |
| 2270 | for (StringPiece product : util::Tokenize(product_list.value(), ',')) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2271 | if (product != "" && product != "default") { |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 2272 | options.products.insert(product.to_string()); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2273 | } |
| 2274 | } |
| 2275 | } |
| 2276 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2277 | std::unique_ptr<IConfigFilter> filter; |
Mihai Nita | f4dacf2 | 2017-04-07 08:25:06 -0700 | [diff] [blame] | 2278 | if (!configs.empty()) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2279 | filter = ParseConfigFilterParameters(configs, context.GetDiagnostics()); |
| 2280 | if (filter == nullptr) { |
| 2281 | return 1; |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 2282 | } |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2283 | options.table_splitter_options.config_filter = filter.get(); |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2284 | } |
| 2285 | |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2286 | if (preferred_density) { |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2287 | Maybe<uint16_t> density = |
| 2288 | ParseTargetDensityParameter(preferred_density.value(), context.GetDiagnostics()); |
| 2289 | if (!density) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2290 | return 1; |
Adam Lesinski | c51562c | 2016-04-28 11:12:38 -0700 | [diff] [blame] | 2291 | } |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2292 | options.table_splitter_options.preferred_densities.push_back(density.value()); |
| 2293 | } |
Adam Lesinski | c51562c | 2016-04-28 11:12:38 -0700 | [diff] [blame] | 2294 | |
Adam Lesinski | d0f492d | 2017-04-03 18:12:45 -0700 | [diff] [blame] | 2295 | // Parse the split parameters. |
| 2296 | for (const std::string& split_arg : split_args) { |
| 2297 | options.split_paths.push_back({}); |
| 2298 | options.split_constraints.push_back({}); |
| 2299 | if (!ParseSplitParameter(split_arg, context.GetDiagnostics(), &options.split_paths.back(), |
| 2300 | &options.split_constraints.back())) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2301 | return 1; |
Adam Lesinski | 1e21ff0 | 2016-06-24 14:57:58 -0700 | [diff] [blame] | 2302 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2303 | } |
Adam Lesinski | 1e21ff0 | 2016-06-24 14:57:58 -0700 | [diff] [blame] | 2304 | |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 2305 | if (context.GetPackageType() != PackageType::kStaticLib && stable_id_file_path) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2306 | if (!LoadStableIdMap(context.GetDiagnostics(), stable_id_file_path.value(), |
| 2307 | &options.stable_id_map)) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2308 | return 1; |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 2309 | } |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2310 | } |
Adam Lesinski | 64587af | 2016-02-18 18:33:06 -0800 | [diff] [blame] | 2311 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2312 | // Populate some default no-compress extensions that are already compressed. |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2313 | options.extensions_to_not_compress.insert( |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2314 | {".jpg", ".jpeg", ".png", ".gif", ".wav", ".mp2", ".mp3", ".ogg", |
| 2315 | ".aac", ".mpg", ".mpeg", ".mid", ".midi", ".smf", ".jet", ".rtttl", |
| 2316 | ".imy", ".xmf", ".mp4", ".m4a", ".m4v", ".3gp", ".3gpp", ".3g2", |
| 2317 | ".3gpp2", ".amr", ".awb", ".wma", ".wmv", ".webm", ".mkv"}); |
| 2318 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2319 | // Turn off auto versioning for static-libs. |
Adam Lesinski | b522f04 | 2017-04-21 16:57:59 -0700 | [diff] [blame] | 2320 | if (context.GetPackageType() == PackageType::kStaticLib) { |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2321 | options.no_auto_version = true; |
| 2322 | options.no_version_vectors = true; |
Yuichi Araki | 4d35cca | 2017-01-18 20:42:17 +0900 | [diff] [blame] | 2323 | options.no_version_transitions = true; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2324 | } |
Adam Lesinski | e4bb9eb | 2016-02-12 22:18:51 -0800 | [diff] [blame] | 2325 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2326 | LinkCommand cmd(&context, options); |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 2327 | return cmd.Run(arg_list); |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 2328 | } |
| 2329 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 2330 | } // namespace aapt |