blob: 5b6f45ebb38d2b397fa9c132d36bbfa2f76185a9 [file] [log] [blame]
Pierre Lecesneff759e62017-02-01 00:29:25 +00001/*
2 * Copyright (C) 2016 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
17#ifndef AAPT_LOADEDAPK_H
18#define AAPT_LOADEDAPK_H
19
20#include "androidfw/StringPiece.h"
21
Pierre Lecesneff759e62017-02-01 00:29:25 +000022#include "ResourceTable.h"
Shane Farmer57669432017-06-19 12:52:04 -070023#include "filter/Filter.h"
Adam Lesinski46708052017-09-29 14:49:15 -070024#include "format/Archive.h"
25#include "format/binary/BinaryResourceParser.h"
26#include "format/binary/TableFlattener.h"
Pierre Lecesne2599aa42017-02-01 22:47:03 +000027#include "io/ZipArchive.h"
Shane Farmer3edd4722017-09-01 14:34:22 -070028#include "xml/XmlDom.h"
Pierre Lecesneff759e62017-02-01 00:29:25 +000029
Pierre Lecesneff759e62017-02-01 00:29:25 +000030namespace aapt {
31
Adam Lesinski8780eb62017-10-31 17:44:39 -070032constexpr static const char kApkResourceTablePath[] = "resources.arsc";
33constexpr static const char kProtoResourceTablePath[] = "resources.pb";
34constexpr static const char kAndroidManifestPath[] = "AndroidManifest.xml";
35
Pierre Lecesnef267a402017-12-01 11:39:01 +000036enum ApkFormat {
37 kUnknown,
38 kBinary,
39 kProto,
40};
41
Adam Lesinski46708052017-09-29 14:49:15 -070042// Info about an APK loaded in memory.
Pierre Lecesneff759e62017-02-01 00:29:25 +000043class LoadedApk {
44 public:
Yi Kong0556f332018-01-09 22:31:45 -080045 virtual ~LoadedApk() = default;
46
Adam Lesinski8780eb62017-10-31 17:44:39 -070047 // Loads both binary and proto APKs from disk.
48 static std::unique_ptr<LoadedApk> LoadApkFromPath(const ::android::StringPiece& path,
49 IDiagnostics* diag);
50
51 // Loads a proto APK from the given file collection.
52 static std::unique_ptr<LoadedApk> LoadProtoApkFromFileCollection(
53 const Source& source, std::unique_ptr<io::IFileCollection> collection, IDiagnostics* diag);
54
55 // Loads a binary APK from the given file collection.
56 static std::unique_ptr<LoadedApk> LoadBinaryApkFromFileCollection(
57 const Source& source, std::unique_ptr<io::IFileCollection> collection, IDiagnostics* diag);
58
59 LoadedApk(const Source& source, std::unique_ptr<io::IFileCollection> apk,
corysmith@google.comf7db43e2018-03-13 12:04:10 -040060 std::unique_ptr<ResourceTable> table, std::unique_ptr<xml::XmlResource> manifest,
61 const ApkFormat& format)
Adam Lesinski8780eb62017-10-31 17:44:39 -070062 : source_(source),
63 apk_(std::move(apk)),
64 table_(std::move(table)),
corysmith@google.comf7db43e2018-03-13 12:04:10 -040065 manifest_(std::move(manifest)),
66 format_(format) {
Adam Lesinski8780eb62017-10-31 17:44:39 -070067 }
Adam Lesinski46708052017-09-29 14:49:15 -070068
69 io::IFileCollection* GetFileCollection() {
70 return apk_.get();
Shane Farmer3edd4722017-09-01 14:34:22 -070071 }
Pierre Lecesneff759e62017-02-01 00:29:25 +000072
Ryan Mitchell5d275512018-07-19 14:29:00 -070073 ApkFormat GetApkFormat() {
74 return format_;
75 }
76
Adam Lesinski8780eb62017-10-31 17:44:39 -070077 const ResourceTable* GetResourceTable() const {
78 return table_.get();
79 }
80
Adam Lesinski46708052017-09-29 14:49:15 -070081 ResourceTable* GetResourceTable() {
82 return table_.get();
83 }
Pierre Lecesneff759e62017-02-01 00:29:25 +000084
Adam Lesinski46708052017-09-29 14:49:15 -070085 const Source& GetSource() {
86 return source_;
87 }
Pierre Lecesneff759e62017-02-01 00:29:25 +000088
Adam Lesinski8780eb62017-10-31 17:44:39 -070089 const xml::XmlResource* GetManifest() const {
90 return manifest_.get();
91 }
92
Pierre Lecesne2599aa42017-02-01 22:47:03 +000093 /**
94 * Writes the APK on disk at the given path, while also removing the resource
95 * files that are not referenced in the resource table.
96 */
Shane Farmer0a5b2012017-06-22 12:24:12 -070097 virtual bool WriteToArchive(IAaptContext* context, const TableFlattenerOptions& options,
98 IArchiveWriter* writer);
Pierre Lecesne2599aa42017-02-01 22:47:03 +000099
Shane Farmer57669432017-06-19 12:52:04 -0700100 /**
Shane Farmer3edd4722017-09-01 14:34:22 -0700101 * Writes the APK on disk at the given path, while also removing the resource files that are not
102 * referenced in the resource table. The provided filter chain is applied to each entry in the APK
103 * file.
104 *
105 * If the manifest is also provided, it will be written to the new APK file, otherwise the
106 * original manifest will be written. The manifest is only required if the contents of the new APK
107 * have been modified in a way that require the AndroidManifest.xml to also be modified.
Shane Farmer57669432017-06-19 12:52:04 -0700108 */
Shane Farmer0a5b2012017-06-22 12:24:12 -0700109 virtual bool WriteToArchive(IAaptContext* context, ResourceTable* split_table,
110 const TableFlattenerOptions& options, FilterChain* filters,
Shane Farmer3edd4722017-09-01 14:34:22 -0700111 IArchiveWriter* writer, xml::XmlResource* manifest = nullptr);
112
Ryan Mitchell5d275512018-07-19 14:29:00 -0700113 /** Loads the file as an xml document. */
Ryan Mitchellfc225b22018-08-21 14:52:51 -0700114 std::unique_ptr<xml::XmlResource> LoadXml(const std::string& file_path, IDiagnostics* diag) const;
Pierre Lecesneff759e62017-02-01 00:29:25 +0000115
116 private:
Adam Lesinski46708052017-09-29 14:49:15 -0700117 DISALLOW_COPY_AND_ASSIGN(LoadedApk);
118
Pierre Lecesneff759e62017-02-01 00:29:25 +0000119 Source source_;
120 std::unique_ptr<io::IFileCollection> apk_;
121 std::unique_ptr<ResourceTable> table_;
Adam Lesinski8780eb62017-10-31 17:44:39 -0700122 std::unique_ptr<xml::XmlResource> manifest_;
corysmith@google.comf7db43e2018-03-13 12:04:10 -0400123 ApkFormat format_;
Pierre Lecesneff759e62017-02-01 00:29:25 +0000124};
125
126} // namespace aapt
127
128#endif /* AAPT_LOADEDAPK_H */