1*d57664e9SAndroid Build Coastguard Worker /*
2*d57664e9SAndroid Build Coastguard Worker * Copyright (C) 2014 The Android Open Source Project
3*d57664e9SAndroid Build Coastguard Worker *
4*d57664e9SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*d57664e9SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*d57664e9SAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*d57664e9SAndroid Build Coastguard Worker *
8*d57664e9SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*d57664e9SAndroid Build Coastguard Worker *
10*d57664e9SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*d57664e9SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*d57664e9SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*d57664e9SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*d57664e9SAndroid Build Coastguard Worker * limitations under the License.
15*d57664e9SAndroid Build Coastguard Worker */
16*d57664e9SAndroid Build Coastguard Worker
17*d57664e9SAndroid Build Coastguard Worker #include "AaptAssets.h"
18*d57664e9SAndroid Build Coastguard Worker #include "ApkBuilder.h"
19*d57664e9SAndroid Build Coastguard Worker
20*d57664e9SAndroid Build Coastguard Worker using namespace android;
21*d57664e9SAndroid Build Coastguard Worker
ApkBuilder(const sp<WeakResourceFilter> & configFilter)22*d57664e9SAndroid Build Coastguard Worker ApkBuilder::ApkBuilder(const sp<WeakResourceFilter>& configFilter)
23*d57664e9SAndroid Build Coastguard Worker : mConfigFilter(configFilter)
24*d57664e9SAndroid Build Coastguard Worker , mDefaultFilter(new AndResourceFilter()) {
25*d57664e9SAndroid Build Coastguard Worker // Add the default split, which is present for all APKs.
26*d57664e9SAndroid Build Coastguard Worker mDefaultFilter->addFilter(mConfigFilter);
27*d57664e9SAndroid Build Coastguard Worker mSplits.add(new ApkSplit(std::set<ConfigDescription>(), mDefaultFilter, true));
28*d57664e9SAndroid Build Coastguard Worker }
29*d57664e9SAndroid Build Coastguard Worker
createSplitForConfigs(const std::set<ConfigDescription> & configs)30*d57664e9SAndroid Build Coastguard Worker status_t ApkBuilder::createSplitForConfigs(const std::set<ConfigDescription>& configs) {
31*d57664e9SAndroid Build Coastguard Worker const size_t N = mSplits.size();
32*d57664e9SAndroid Build Coastguard Worker for (size_t i = 0; i < N; i++) {
33*d57664e9SAndroid Build Coastguard Worker const std::set<ConfigDescription>& splitConfigs = mSplits[i]->getConfigs();
34*d57664e9SAndroid Build Coastguard Worker std::set<ConfigDescription>::const_iterator iter = configs.begin();
35*d57664e9SAndroid Build Coastguard Worker for (; iter != configs.end(); iter++) {
36*d57664e9SAndroid Build Coastguard Worker if (splitConfigs.count(*iter) > 0) {
37*d57664e9SAndroid Build Coastguard Worker // Can't have overlapping configurations.
38*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, "ERROR: Split configuration '%s' is already defined "
39*d57664e9SAndroid Build Coastguard Worker "in another split.\n", iter->toString().c_str());
40*d57664e9SAndroid Build Coastguard Worker return ALREADY_EXISTS;
41*d57664e9SAndroid Build Coastguard Worker }
42*d57664e9SAndroid Build Coastguard Worker }
43*d57664e9SAndroid Build Coastguard Worker }
44*d57664e9SAndroid Build Coastguard Worker
45*d57664e9SAndroid Build Coastguard Worker sp<StrongResourceFilter> splitFilter = new StrongResourceFilter(configs);
46*d57664e9SAndroid Build Coastguard Worker
47*d57664e9SAndroid Build Coastguard Worker // Add the inverse filter of this split filter to the base apk filter so it will
48*d57664e9SAndroid Build Coastguard Worker // omit resources that belong in this split.
49*d57664e9SAndroid Build Coastguard Worker mDefaultFilter->addFilter(new InverseResourceFilter(splitFilter));
50*d57664e9SAndroid Build Coastguard Worker
51*d57664e9SAndroid Build Coastguard Worker // Now add the apk-wide config filter to our split filter.
52*d57664e9SAndroid Build Coastguard Worker sp<AndResourceFilter> filter = new AndResourceFilter();
53*d57664e9SAndroid Build Coastguard Worker filter->addFilter(splitFilter);
54*d57664e9SAndroid Build Coastguard Worker filter->addFilter(mConfigFilter);
55*d57664e9SAndroid Build Coastguard Worker mSplits.add(new ApkSplit(configs, filter));
56*d57664e9SAndroid Build Coastguard Worker return NO_ERROR;
57*d57664e9SAndroid Build Coastguard Worker }
58*d57664e9SAndroid Build Coastguard Worker
addEntry(const String8 & path,const sp<AaptFile> & file)59*d57664e9SAndroid Build Coastguard Worker status_t ApkBuilder::addEntry(const String8& path, const sp<AaptFile>& file) {
60*d57664e9SAndroid Build Coastguard Worker const size_t N = mSplits.size();
61*d57664e9SAndroid Build Coastguard Worker for (size_t i = 0; i < N; i++) {
62*d57664e9SAndroid Build Coastguard Worker if (mSplits[i]->matches(file)) {
63*d57664e9SAndroid Build Coastguard Worker return mSplits.editItemAt(i)->addEntry(path, file);
64*d57664e9SAndroid Build Coastguard Worker }
65*d57664e9SAndroid Build Coastguard Worker }
66*d57664e9SAndroid Build Coastguard Worker // Entry can be dropped if it doesn't match any split. This will only happen
67*d57664e9SAndroid Build Coastguard Worker // if the enry doesn't mConfigFilter.
68*d57664e9SAndroid Build Coastguard Worker return NO_ERROR;
69*d57664e9SAndroid Build Coastguard Worker }
70*d57664e9SAndroid Build Coastguard Worker
print() const71*d57664e9SAndroid Build Coastguard Worker void ApkBuilder::print() const {
72*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, "APK Builder\n");
73*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, "-----------\n");
74*d57664e9SAndroid Build Coastguard Worker const size_t N = mSplits.size();
75*d57664e9SAndroid Build Coastguard Worker for (size_t i = 0; i < N; i++) {
76*d57664e9SAndroid Build Coastguard Worker mSplits[i]->print();
77*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, "\n");
78*d57664e9SAndroid Build Coastguard Worker }
79*d57664e9SAndroid Build Coastguard Worker }
80*d57664e9SAndroid Build Coastguard Worker
ApkSplit(const std::set<ConfigDescription> & configs,const sp<ResourceFilter> & filter,bool isBase)81*d57664e9SAndroid Build Coastguard Worker ApkSplit::ApkSplit(const std::set<ConfigDescription>& configs, const sp<ResourceFilter>& filter, bool isBase)
82*d57664e9SAndroid Build Coastguard Worker : mConfigs(configs), mFilter(filter), mIsBase(isBase) {
83*d57664e9SAndroid Build Coastguard Worker std::set<ConfigDescription>::const_iterator iter = configs.begin();
84*d57664e9SAndroid Build Coastguard Worker for (; iter != configs.end(); iter++) {
85*d57664e9SAndroid Build Coastguard Worker if (mName.size() > 0) {
86*d57664e9SAndroid Build Coastguard Worker mName.append(",");
87*d57664e9SAndroid Build Coastguard Worker mDirName.append("_");
88*d57664e9SAndroid Build Coastguard Worker mPackageSafeName.append(".");
89*d57664e9SAndroid Build Coastguard Worker }
90*d57664e9SAndroid Build Coastguard Worker
91*d57664e9SAndroid Build Coastguard Worker String8 configStr = iter->toString();
92*d57664e9SAndroid Build Coastguard Worker String8 packageConfigStr(configStr);
93*d57664e9SAndroid Build Coastguard Worker size_t len = packageConfigStr.length();
94*d57664e9SAndroid Build Coastguard Worker if (len > 0) {
95*d57664e9SAndroid Build Coastguard Worker char* buf = packageConfigStr.lockBuffer(len);
96*d57664e9SAndroid Build Coastguard Worker for (char* end = buf + len; buf < end; ++buf) {
97*d57664e9SAndroid Build Coastguard Worker if (*buf == '-') {
98*d57664e9SAndroid Build Coastguard Worker *buf = '_';
99*d57664e9SAndroid Build Coastguard Worker }
100*d57664e9SAndroid Build Coastguard Worker }
101*d57664e9SAndroid Build Coastguard Worker packageConfigStr.unlockBuffer(len);
102*d57664e9SAndroid Build Coastguard Worker }
103*d57664e9SAndroid Build Coastguard Worker mName.append(configStr);
104*d57664e9SAndroid Build Coastguard Worker mDirName.append(configStr);
105*d57664e9SAndroid Build Coastguard Worker mPackageSafeName.append(packageConfigStr);
106*d57664e9SAndroid Build Coastguard Worker }
107*d57664e9SAndroid Build Coastguard Worker }
108*d57664e9SAndroid Build Coastguard Worker
addEntry(const String8 & path,const sp<AaptFile> & file)109*d57664e9SAndroid Build Coastguard Worker status_t ApkSplit::addEntry(const String8& path, const sp<AaptFile>& file) {
110*d57664e9SAndroid Build Coastguard Worker if (!mFiles.insert(OutputEntry(path, file)).second) {
111*d57664e9SAndroid Build Coastguard Worker // Duplicate file.
112*d57664e9SAndroid Build Coastguard Worker return ALREADY_EXISTS;
113*d57664e9SAndroid Build Coastguard Worker }
114*d57664e9SAndroid Build Coastguard Worker return NO_ERROR;
115*d57664e9SAndroid Build Coastguard Worker }
116*d57664e9SAndroid Build Coastguard Worker
print() const117*d57664e9SAndroid Build Coastguard Worker void ApkSplit::print() const {
118*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, "APK Split '%s'\n", mName.c_str());
119*d57664e9SAndroid Build Coastguard Worker
120*d57664e9SAndroid Build Coastguard Worker std::set<OutputEntry>::const_iterator iter = mFiles.begin();
121*d57664e9SAndroid Build Coastguard Worker for (; iter != mFiles.end(); iter++) {
122*d57664e9SAndroid Build Coastguard Worker fprintf(stderr, " %s (%s)\n", iter->getPath().c_str(), iter->getFile()->getSourceFile().c_str());
123*d57664e9SAndroid Build Coastguard Worker }
124*d57664e9SAndroid Build Coastguard Worker }
125