xref: /aosp_15_r20/frameworks/base/libs/androidfw/AttributeResolution.cpp (revision d57664e9bc4670b3ecf6748a746a57c557b6bc9e)
1*d57664e9SAndroid Build Coastguard Worker /*
2*d57664e9SAndroid Build Coastguard Worker  * Copyright (C) 2016 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 "androidfw/AttributeResolution.h"
18*d57664e9SAndroid Build Coastguard Worker 
19*d57664e9SAndroid Build Coastguard Worker #include <cstdint>
20*d57664e9SAndroid Build Coastguard Worker 
21*d57664e9SAndroid Build Coastguard Worker #include <log/log.h>
22*d57664e9SAndroid Build Coastguard Worker 
23*d57664e9SAndroid Build Coastguard Worker #include "androidfw/AssetManager2.h"
24*d57664e9SAndroid Build Coastguard Worker #include "androidfw/AttributeFinder.h"
25*d57664e9SAndroid Build Coastguard Worker 
26*d57664e9SAndroid Build Coastguard Worker constexpr bool kDebugStyles = false;
27*d57664e9SAndroid Build Coastguard Worker #define DEBUG_LOG(...) do { if (kDebugStyles) { ALOGI(__VA_ARGS__); } } while(0)
28*d57664e9SAndroid Build Coastguard Worker 
29*d57664e9SAndroid Build Coastguard Worker namespace android {
30*d57664e9SAndroid Build Coastguard Worker 
31*d57664e9SAndroid Build Coastguard Worker namespace {
32*d57664e9SAndroid Build Coastguard Worker 
33*d57664e9SAndroid Build Coastguard Worker // Java asset cookies have 0 as an invalid cookie, but TypedArray expects < 0.
ApkAssetsCookieToJavaCookie(ApkAssetsCookie cookie)34*d57664e9SAndroid Build Coastguard Worker static uint32_t ApkAssetsCookieToJavaCookie(ApkAssetsCookie cookie) {
35*d57664e9SAndroid Build Coastguard Worker   return cookie != kInvalidCookie ? static_cast<uint32_t>(cookie + 1) : static_cast<uint32_t>(-1);
36*d57664e9SAndroid Build Coastguard Worker }
37*d57664e9SAndroid Build Coastguard Worker 
38*d57664e9SAndroid Build Coastguard Worker class XmlAttributeFinder
39*d57664e9SAndroid Build Coastguard Worker     : public BackTrackingAttributeFinder<XmlAttributeFinder, size_t> {
40*d57664e9SAndroid Build Coastguard Worker  public:
XmlAttributeFinder(const ResXMLParser * parser)41*d57664e9SAndroid Build Coastguard Worker   explicit XmlAttributeFinder(const ResXMLParser* parser)
42*d57664e9SAndroid Build Coastguard Worker       : BackTrackingAttributeFinder(0, parser != nullptr ? parser->getAttributeCount() : 0),
43*d57664e9SAndroid Build Coastguard Worker         parser_(parser) {}
44*d57664e9SAndroid Build Coastguard Worker 
GetAttribute(size_t index) const45*d57664e9SAndroid Build Coastguard Worker   inline uint32_t GetAttribute(size_t index) const {
46*d57664e9SAndroid Build Coastguard Worker     return parser_->getAttributeNameResID(index);
47*d57664e9SAndroid Build Coastguard Worker   }
48*d57664e9SAndroid Build Coastguard Worker 
49*d57664e9SAndroid Build Coastguard Worker  private:
50*d57664e9SAndroid Build Coastguard Worker   const ResXMLParser* parser_;
51*d57664e9SAndroid Build Coastguard Worker };
52*d57664e9SAndroid Build Coastguard Worker 
53*d57664e9SAndroid Build Coastguard Worker class BagAttributeFinder
54*d57664e9SAndroid Build Coastguard Worker     : public BackTrackingAttributeFinder<BagAttributeFinder, const ResolvedBag::Entry*> {
55*d57664e9SAndroid Build Coastguard Worker  public:
BagAttributeFinder(const ResolvedBag * bag)56*d57664e9SAndroid Build Coastguard Worker   explicit BagAttributeFinder(const ResolvedBag* bag)
57*d57664e9SAndroid Build Coastguard Worker       : BackTrackingAttributeFinder(bag != nullptr ? bag->entries : nullptr,
58*d57664e9SAndroid Build Coastguard Worker                                     bag != nullptr ? bag->entries + bag->entry_count : nullptr) {
59*d57664e9SAndroid Build Coastguard Worker   }
60*d57664e9SAndroid Build Coastguard Worker 
GetAttribute(const ResolvedBag::Entry * entry) const61*d57664e9SAndroid Build Coastguard Worker   inline uint32_t GetAttribute(const ResolvedBag::Entry* entry) const {
62*d57664e9SAndroid Build Coastguard Worker     return entry->key;
63*d57664e9SAndroid Build Coastguard Worker   }
64*d57664e9SAndroid Build Coastguard Worker };
65*d57664e9SAndroid Build Coastguard Worker 
GetStyleBag(Theme * theme,uint32_t theme_attribute_resid,uint32_t fallback_resid,uint32_t * out_theme_flags)66*d57664e9SAndroid Build Coastguard Worker base::expected<const ResolvedBag*, NullOrIOError> GetStyleBag(Theme* theme,
67*d57664e9SAndroid Build Coastguard Worker                                                               uint32_t theme_attribute_resid,
68*d57664e9SAndroid Build Coastguard Worker                                                               uint32_t fallback_resid,
69*d57664e9SAndroid Build Coastguard Worker                                                               uint32_t* out_theme_flags) {
70*d57664e9SAndroid Build Coastguard Worker   // Load the style from the attribute if specified.
71*d57664e9SAndroid Build Coastguard Worker   if (theme_attribute_resid != 0U) {
72*d57664e9SAndroid Build Coastguard Worker     std::optional<AssetManager2::SelectedValue> value = theme->GetAttribute(theme_attribute_resid);
73*d57664e9SAndroid Build Coastguard Worker     if (value.has_value()) {
74*d57664e9SAndroid Build Coastguard Worker       *out_theme_flags |= value->flags;
75*d57664e9SAndroid Build Coastguard Worker       auto result = theme->GetAssetManager()->ResolveBag(*value);
76*d57664e9SAndroid Build Coastguard Worker       if (result.has_value() || IsIOError(result)) {
77*d57664e9SAndroid Build Coastguard Worker         return result;
78*d57664e9SAndroid Build Coastguard Worker       }
79*d57664e9SAndroid Build Coastguard Worker     }
80*d57664e9SAndroid Build Coastguard Worker   }
81*d57664e9SAndroid Build Coastguard Worker 
82*d57664e9SAndroid Build Coastguard Worker   // Fallback to loading the style from the resource id if specified.
83*d57664e9SAndroid Build Coastguard Worker   if (fallback_resid != 0U) {
84*d57664e9SAndroid Build Coastguard Worker     return theme->GetAssetManager()->GetBag(fallback_resid);
85*d57664e9SAndroid Build Coastguard Worker   }
86*d57664e9SAndroid Build Coastguard Worker 
87*d57664e9SAndroid Build Coastguard Worker   return base::unexpected(std::nullopt);
88*d57664e9SAndroid Build Coastguard Worker }
89*d57664e9SAndroid Build Coastguard Worker 
GetXmlStyleBag(Theme * theme,ResXMLParser * xml_parser,uint32_t * out_theme_flags)90*d57664e9SAndroid Build Coastguard Worker base::expected<const ResolvedBag*, NullOrIOError> GetXmlStyleBag(Theme* theme,
91*d57664e9SAndroid Build Coastguard Worker                                                                  ResXMLParser* xml_parser,
92*d57664e9SAndroid Build Coastguard Worker                                                                  uint32_t* out_theme_flags) {
93*d57664e9SAndroid Build Coastguard Worker   if (xml_parser == nullptr) {
94*d57664e9SAndroid Build Coastguard Worker     return base::unexpected(std::nullopt);
95*d57664e9SAndroid Build Coastguard Worker   }
96*d57664e9SAndroid Build Coastguard Worker 
97*d57664e9SAndroid Build Coastguard Worker   // Retrieve the style resource ID associated with the current XML tag's style attribute.
98*d57664e9SAndroid Build Coastguard Worker   Res_value value;
99*d57664e9SAndroid Build Coastguard Worker   const ssize_t idx = xml_parser->indexOfStyle();
100*d57664e9SAndroid Build Coastguard Worker   if (idx < 0 || xml_parser->getAttributeValue(idx, &value) < 0) {
101*d57664e9SAndroid Build Coastguard Worker     return base::unexpected(std::nullopt);
102*d57664e9SAndroid Build Coastguard Worker   }
103*d57664e9SAndroid Build Coastguard Worker 
104*d57664e9SAndroid Build Coastguard Worker   if (value.dataType == Res_value::TYPE_ATTRIBUTE) {
105*d57664e9SAndroid Build Coastguard Worker     // Resolve the attribute with out theme.
106*d57664e9SAndroid Build Coastguard Worker     if (std::optional<AssetManager2::SelectedValue> result = theme->GetAttribute(value.data)) {
107*d57664e9SAndroid Build Coastguard Worker       *out_theme_flags |= result->flags;
108*d57664e9SAndroid Build Coastguard Worker       return theme->GetAssetManager()->ResolveBag(*result);
109*d57664e9SAndroid Build Coastguard Worker     }
110*d57664e9SAndroid Build Coastguard Worker   }
111*d57664e9SAndroid Build Coastguard Worker 
112*d57664e9SAndroid Build Coastguard Worker   if (value.dataType == Res_value::TYPE_REFERENCE) {
113*d57664e9SAndroid Build Coastguard Worker     return theme->GetAssetManager()->GetBag(value.data);
114*d57664e9SAndroid Build Coastguard Worker   }
115*d57664e9SAndroid Build Coastguard Worker 
116*d57664e9SAndroid Build Coastguard Worker   return base::unexpected(std::nullopt);
117*d57664e9SAndroid Build Coastguard Worker }
118*d57664e9SAndroid Build Coastguard Worker 
119*d57664e9SAndroid Build Coastguard Worker } // namespace
120*d57664e9SAndroid Build Coastguard Worker 
ResolveAttrs(Theme * theme,uint32_t def_style_attr,uint32_t def_style_res,uint32_t * src_values,size_t src_values_length,uint32_t * attrs,size_t attrs_length,uint32_t * out_values,uint32_t * out_indices)121*d57664e9SAndroid Build Coastguard Worker base::expected<std::monostate, IOError> ResolveAttrs(Theme* theme, uint32_t def_style_attr,
122*d57664e9SAndroid Build Coastguard Worker                                                      uint32_t def_style_res, uint32_t* src_values,
123*d57664e9SAndroid Build Coastguard Worker                                                      size_t src_values_length, uint32_t* attrs,
124*d57664e9SAndroid Build Coastguard Worker                                                      size_t attrs_length, uint32_t* out_values,
125*d57664e9SAndroid Build Coastguard Worker                                                      uint32_t* out_indices) {
126*d57664e9SAndroid Build Coastguard Worker   DEBUG_LOG("APPLY STYLE: theme=0x%p defStyleAttr=0x%x defStyleRes=0x%x", theme, def_style_attr,
127*d57664e9SAndroid Build Coastguard Worker             def_style_res);
128*d57664e9SAndroid Build Coastguard Worker 
129*d57664e9SAndroid Build Coastguard Worker   int indices_idx = 0;
130*d57664e9SAndroid Build Coastguard Worker   const AssetManager2* assetmanager = theme->GetAssetManager();
131*d57664e9SAndroid Build Coastguard Worker 
132*d57664e9SAndroid Build Coastguard Worker   // Load default style from attribute or resource id, if specified...
133*d57664e9SAndroid Build Coastguard Worker   uint32_t def_style_theme_flags = 0U;
134*d57664e9SAndroid Build Coastguard Worker   const auto default_style_bag = GetStyleBag(theme, def_style_attr, def_style_res,
135*d57664e9SAndroid Build Coastguard Worker                                              &def_style_theme_flags);
136*d57664e9SAndroid Build Coastguard Worker   if (UNLIKELY(IsIOError(default_style_bag))) {
137*d57664e9SAndroid Build Coastguard Worker     return base::unexpected(GetIOError(default_style_bag.error()));
138*d57664e9SAndroid Build Coastguard Worker   }
139*d57664e9SAndroid Build Coastguard Worker 
140*d57664e9SAndroid Build Coastguard Worker   BagAttributeFinder def_style_attr_finder(default_style_bag.value_or(nullptr));
141*d57664e9SAndroid Build Coastguard Worker 
142*d57664e9SAndroid Build Coastguard Worker   // Now iterate through all of the attributes that the client has requested,
143*d57664e9SAndroid Build Coastguard Worker   // filling in each with whatever data we can find.
144*d57664e9SAndroid Build Coastguard Worker   for (size_t ii = 0; ii < attrs_length; ii++) {
145*d57664e9SAndroid Build Coastguard Worker     const uint32_t cur_ident = attrs[ii];
146*d57664e9SAndroid Build Coastguard Worker     DEBUG_LOG("RETRIEVING ATTR 0x%08x...", cur_ident);
147*d57664e9SAndroid Build Coastguard Worker 
148*d57664e9SAndroid Build Coastguard Worker     // Try to find a value for this attribute...  we prioritize values
149*d57664e9SAndroid Build Coastguard Worker     // coming from, first XML attributes, then XML style, then default
150*d57664e9SAndroid Build Coastguard Worker     // style, and finally the theme.
151*d57664e9SAndroid Build Coastguard Worker 
152*d57664e9SAndroid Build Coastguard Worker     // Retrieve the current input value if available.
153*d57664e9SAndroid Build Coastguard Worker     AssetManager2::SelectedValue value{};
154*d57664e9SAndroid Build Coastguard Worker     if (src_values_length > 0 && src_values[ii] != 0) {
155*d57664e9SAndroid Build Coastguard Worker       value.type = Res_value::TYPE_ATTRIBUTE;
156*d57664e9SAndroid Build Coastguard Worker       value.data = src_values[ii];
157*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> From values: type=0x%x, data=0x%08x", value.type, value.data);
158*d57664e9SAndroid Build Coastguard Worker     } else {
159*d57664e9SAndroid Build Coastguard Worker       const ResolvedBag::Entry* const entry = def_style_attr_finder.Find(cur_ident);
160*d57664e9SAndroid Build Coastguard Worker       if (entry != def_style_attr_finder.end()) {
161*d57664e9SAndroid Build Coastguard Worker         value = AssetManager2::SelectedValue(*default_style_bag, *entry);
162*d57664e9SAndroid Build Coastguard Worker         value.flags |= def_style_theme_flags;
163*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> From def style: type=0x%x, data=0x%08x", value.type, value.data);
164*d57664e9SAndroid Build Coastguard Worker       }
165*d57664e9SAndroid Build Coastguard Worker     }
166*d57664e9SAndroid Build Coastguard Worker 
167*d57664e9SAndroid Build Coastguard Worker     if (value.type != Res_value::TYPE_NULL) {
168*d57664e9SAndroid Build Coastguard Worker       // Take care of resolving the found resource to its final value.
169*d57664e9SAndroid Build Coastguard Worker       const auto result = theme->ResolveAttributeReference(value);
170*d57664e9SAndroid Build Coastguard Worker       if (UNLIKELY(IsIOError(result))) {
171*d57664e9SAndroid Build Coastguard Worker         return base::unexpected(GetIOError(result.error()));
172*d57664e9SAndroid Build Coastguard Worker       }
173*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> Resolved attr: type=0x%x, data=0x%08x", value.type, value.data);
174*d57664e9SAndroid Build Coastguard Worker     } else if (value.data != Res_value::DATA_NULL_EMPTY) {
175*d57664e9SAndroid Build Coastguard Worker       // If we still don't have a value for this attribute, try to find it in the theme!
176*d57664e9SAndroid Build Coastguard Worker       if (auto attr_value = theme->GetAttribute(cur_ident)) {
177*d57664e9SAndroid Build Coastguard Worker         value = *attr_value;
178*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> From theme: type=0x%x, data=0x%08x", value.type, value.data);
179*d57664e9SAndroid Build Coastguard Worker 
180*d57664e9SAndroid Build Coastguard Worker         const auto result = assetmanager->ResolveReference(value, true /* cache_value */);
181*d57664e9SAndroid Build Coastguard Worker         if (UNLIKELY(IsIOError(result))) {
182*d57664e9SAndroid Build Coastguard Worker           return base::unexpected(GetIOError(result.error()));
183*d57664e9SAndroid Build Coastguard Worker         }
184*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> Resolved theme: type=0x%x, data=0x%08x", value.type, value.data);
185*d57664e9SAndroid Build Coastguard Worker       }
186*d57664e9SAndroid Build Coastguard Worker     }
187*d57664e9SAndroid Build Coastguard Worker 
188*d57664e9SAndroid Build Coastguard Worker     // Deal with the special @null value -- it turns back to TYPE_NULL.
189*d57664e9SAndroid Build Coastguard Worker     if (value.type == Res_value::TYPE_REFERENCE && value.data == 0) {
190*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> Setting to @null!");
191*d57664e9SAndroid Build Coastguard Worker       value.type = Res_value::TYPE_NULL;
192*d57664e9SAndroid Build Coastguard Worker       value.data = Res_value::DATA_NULL_UNDEFINED;
193*d57664e9SAndroid Build Coastguard Worker       value.cookie = kInvalidCookie;
194*d57664e9SAndroid Build Coastguard Worker     }
195*d57664e9SAndroid Build Coastguard Worker 
196*d57664e9SAndroid Build Coastguard Worker     DEBUG_LOG("Attribute 0x%08x: type=0x%x, data=0x%08x", cur_ident, value.type, value.data);
197*d57664e9SAndroid Build Coastguard Worker 
198*d57664e9SAndroid Build Coastguard Worker     // Write the final value back to Java.
199*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_TYPE] = value.type;
200*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DATA] = value.data;
201*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_ASSET_COOKIE] = ApkAssetsCookieToJavaCookie(value.cookie);
202*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_RESOURCE_ID] = value.resid;
203*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_CHANGING_CONFIGURATIONS] = value.flags;
204*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DENSITY] = value.config.density;
205*d57664e9SAndroid Build Coastguard Worker 
206*d57664e9SAndroid Build Coastguard Worker     if (out_indices != nullptr &&
207*d57664e9SAndroid Build Coastguard Worker         (value.type != Res_value::TYPE_NULL || value.data == Res_value::DATA_NULL_EMPTY)) {
208*d57664e9SAndroid Build Coastguard Worker       out_indices[++indices_idx] = ii;
209*d57664e9SAndroid Build Coastguard Worker     }
210*d57664e9SAndroid Build Coastguard Worker 
211*d57664e9SAndroid Build Coastguard Worker     out_values += STYLE_NUM_ENTRIES;
212*d57664e9SAndroid Build Coastguard Worker   }
213*d57664e9SAndroid Build Coastguard Worker 
214*d57664e9SAndroid Build Coastguard Worker   if (out_indices != nullptr) {
215*d57664e9SAndroid Build Coastguard Worker     out_indices[0] = indices_idx;
216*d57664e9SAndroid Build Coastguard Worker   }
217*d57664e9SAndroid Build Coastguard Worker   return {};
218*d57664e9SAndroid Build Coastguard Worker }
219*d57664e9SAndroid Build Coastguard Worker 
ApplyStyle(Theme * theme,ResXMLParser * xml_parser,uint32_t def_style_attr,uint32_t def_style_resid,const uint32_t * attrs,size_t attrs_length,uint32_t * out_values,uint32_t * out_indices)220*d57664e9SAndroid Build Coastguard Worker base::expected<std::monostate, IOError> ApplyStyle(Theme* theme, ResXMLParser* xml_parser,
221*d57664e9SAndroid Build Coastguard Worker                                                    uint32_t def_style_attr,
222*d57664e9SAndroid Build Coastguard Worker                                                    uint32_t def_style_resid,
223*d57664e9SAndroid Build Coastguard Worker                                                    const uint32_t* attrs, size_t attrs_length,
224*d57664e9SAndroid Build Coastguard Worker                                                    uint32_t* out_values, uint32_t* out_indices) {
225*d57664e9SAndroid Build Coastguard Worker   DEBUG_LOG("APPLY STYLE: theme=0x%p defStyleAttr=0x%x defStyleRes=0x%x xml=0x%p", theme,
226*d57664e9SAndroid Build Coastguard Worker             def_style_attr, def_style_resid, xml_parser);
227*d57664e9SAndroid Build Coastguard Worker 
228*d57664e9SAndroid Build Coastguard Worker   int indices_idx = 0;
229*d57664e9SAndroid Build Coastguard Worker   const AssetManager2* assetmanager = theme->GetAssetManager();
230*d57664e9SAndroid Build Coastguard Worker 
231*d57664e9SAndroid Build Coastguard Worker   // Load default style from attribute, if specified...
232*d57664e9SAndroid Build Coastguard Worker   uint32_t def_style_theme_flags = 0U;
233*d57664e9SAndroid Build Coastguard Worker   const auto default_style_bag = GetStyleBag(theme, def_style_attr, def_style_resid,
234*d57664e9SAndroid Build Coastguard Worker                                              &def_style_theme_flags);
235*d57664e9SAndroid Build Coastguard Worker   if (IsIOError(default_style_bag)) {
236*d57664e9SAndroid Build Coastguard Worker     return base::unexpected(GetIOError(default_style_bag.error()));
237*d57664e9SAndroid Build Coastguard Worker   }
238*d57664e9SAndroid Build Coastguard Worker 
239*d57664e9SAndroid Build Coastguard Worker   // Retrieve the style resource ID associated with the current XML tag's style attribute.
240*d57664e9SAndroid Build Coastguard Worker   uint32_t xml_style_theme_flags = 0U;
241*d57664e9SAndroid Build Coastguard Worker   const auto xml_style_bag = GetXmlStyleBag(theme, xml_parser, &def_style_theme_flags);
242*d57664e9SAndroid Build Coastguard Worker   if (IsIOError(xml_style_bag)) {
243*d57664e9SAndroid Build Coastguard Worker     return base::unexpected(GetIOError(xml_style_bag.error()));
244*d57664e9SAndroid Build Coastguard Worker   }
245*d57664e9SAndroid Build Coastguard Worker 
246*d57664e9SAndroid Build Coastguard Worker   BagAttributeFinder def_style_attr_finder(default_style_bag.value_or(nullptr));
247*d57664e9SAndroid Build Coastguard Worker   BagAttributeFinder xml_style_attr_finder(xml_style_bag.value_or(nullptr));
248*d57664e9SAndroid Build Coastguard Worker   XmlAttributeFinder xml_attr_finder(xml_parser);
249*d57664e9SAndroid Build Coastguard Worker 
250*d57664e9SAndroid Build Coastguard Worker   // Now iterate through all of the attributes that the client has requested,
251*d57664e9SAndroid Build Coastguard Worker   // filling in each with whatever data we can find.
252*d57664e9SAndroid Build Coastguard Worker   for (size_t ii = 0; ii < attrs_length; ii++) {
253*d57664e9SAndroid Build Coastguard Worker     const uint32_t cur_ident = attrs[ii];
254*d57664e9SAndroid Build Coastguard Worker     DEBUG_LOG("RETRIEVING ATTR 0x%08x...", cur_ident);
255*d57664e9SAndroid Build Coastguard Worker 
256*d57664e9SAndroid Build Coastguard Worker     AssetManager2::SelectedValue value{};
257*d57664e9SAndroid Build Coastguard Worker     uint32_t value_source_resid = 0;
258*d57664e9SAndroid Build Coastguard Worker 
259*d57664e9SAndroid Build Coastguard Worker     // Try to find a value for this attribute...  we prioritize values
260*d57664e9SAndroid Build Coastguard Worker     // coming from, first XML attributes, then XML style, then default
261*d57664e9SAndroid Build Coastguard Worker     // style, and finally the theme.
262*d57664e9SAndroid Build Coastguard Worker 
263*d57664e9SAndroid Build Coastguard Worker     // Walk through the xml attributes looking for the requested attribute.
264*d57664e9SAndroid Build Coastguard Worker     const size_t xml_attr_idx = xml_attr_finder.Find(cur_ident);
265*d57664e9SAndroid Build Coastguard Worker     if (xml_attr_idx != xml_attr_finder.end()) {
266*d57664e9SAndroid Build Coastguard Worker       // We found the attribute we were looking for.
267*d57664e9SAndroid Build Coastguard Worker       Res_value attribute_value{};
268*d57664e9SAndroid Build Coastguard Worker       xml_parser->getAttributeValue(xml_attr_idx, &attribute_value);
269*d57664e9SAndroid Build Coastguard Worker       value.type = attribute_value.dataType;
270*d57664e9SAndroid Build Coastguard Worker       value.data = attribute_value.data;
271*d57664e9SAndroid Build Coastguard Worker       value_source_resid = xml_parser->getSourceResourceId();
272*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> From XML: type=0x%x, data=0x%08x", value.type, value.data);
273*d57664e9SAndroid Build Coastguard Worker     }
274*d57664e9SAndroid Build Coastguard Worker 
275*d57664e9SAndroid Build Coastguard Worker     if (value.type == Res_value::TYPE_NULL && value.data != Res_value::DATA_NULL_EMPTY) {
276*d57664e9SAndroid Build Coastguard Worker       // Walk through the style class values looking for the requested attribute.
277*d57664e9SAndroid Build Coastguard Worker       const ResolvedBag::Entry* entry = xml_style_attr_finder.Find(cur_ident);
278*d57664e9SAndroid Build Coastguard Worker       if (entry != xml_style_attr_finder.end()) {
279*d57664e9SAndroid Build Coastguard Worker         value = AssetManager2::SelectedValue(*xml_style_bag, *entry);
280*d57664e9SAndroid Build Coastguard Worker         value.flags |= xml_style_theme_flags;
281*d57664e9SAndroid Build Coastguard Worker         value_source_resid = entry->style;
282*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> From style: type=0x%x, data=0x%08x, style=0x%08x", value.type, value.data,
283*d57664e9SAndroid Build Coastguard Worker                   value_source_resid);
284*d57664e9SAndroid Build Coastguard Worker       }
285*d57664e9SAndroid Build Coastguard Worker     }
286*d57664e9SAndroid Build Coastguard Worker 
287*d57664e9SAndroid Build Coastguard Worker     if (value.type == Res_value::TYPE_NULL && value.data != Res_value::DATA_NULL_EMPTY) {
288*d57664e9SAndroid Build Coastguard Worker       // Walk through the default style values looking for the requested attribute.
289*d57664e9SAndroid Build Coastguard Worker       const ResolvedBag::Entry* entry = def_style_attr_finder.Find(cur_ident);
290*d57664e9SAndroid Build Coastguard Worker       if (entry != def_style_attr_finder.end()) {
291*d57664e9SAndroid Build Coastguard Worker         value = AssetManager2::SelectedValue(*default_style_bag, *entry);
292*d57664e9SAndroid Build Coastguard Worker         value.flags |= def_style_theme_flags;
293*d57664e9SAndroid Build Coastguard Worker         value_source_resid = entry->style;
294*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> From def style: type=0x%x, data=0x%08x, style=0x%08x", value.type, value.data,
295*d57664e9SAndroid Build Coastguard Worker                   entry->style);
296*d57664e9SAndroid Build Coastguard Worker       }
297*d57664e9SAndroid Build Coastguard Worker     }
298*d57664e9SAndroid Build Coastguard Worker 
299*d57664e9SAndroid Build Coastguard Worker     if (value.type != Res_value::TYPE_NULL) {
300*d57664e9SAndroid Build Coastguard Worker       // Take care of resolving the found resource to its final value.
301*d57664e9SAndroid Build Coastguard Worker       auto result = theme->ResolveAttributeReference(value);
302*d57664e9SAndroid Build Coastguard Worker       if (UNLIKELY(IsIOError(result))) {
303*d57664e9SAndroid Build Coastguard Worker         return base::unexpected(GetIOError(result.error()));
304*d57664e9SAndroid Build Coastguard Worker       }
305*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> Resolved attr: type=0x%x, data=0x%08x", value.type, value.data);
306*d57664e9SAndroid Build Coastguard Worker     } else if (value.data != Res_value::DATA_NULL_EMPTY) {
307*d57664e9SAndroid Build Coastguard Worker       // If we still don't have a value for this attribute, try to find it in the theme!
308*d57664e9SAndroid Build Coastguard Worker       if (auto attr_value = theme->GetAttribute(cur_ident)) {
309*d57664e9SAndroid Build Coastguard Worker         value = *attr_value;
310*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> From theme: type=0x%x, data=0x%08x", value.type, value.data);
311*d57664e9SAndroid Build Coastguard Worker 
312*d57664e9SAndroid Build Coastguard Worker         auto result = assetmanager->ResolveReference(value, true /* cache_value */);
313*d57664e9SAndroid Build Coastguard Worker         if (UNLIKELY(IsIOError(result))) {
314*d57664e9SAndroid Build Coastguard Worker           return base::unexpected(GetIOError(result.error()));
315*d57664e9SAndroid Build Coastguard Worker         }
316*d57664e9SAndroid Build Coastguard Worker         DEBUG_LOG("-> Resolved theme: type=0x%x, data=0x%08x", value.type, value.data);
317*d57664e9SAndroid Build Coastguard Worker         // TODO: set value_source_resid for the style in the theme that was used.
318*d57664e9SAndroid Build Coastguard Worker       }
319*d57664e9SAndroid Build Coastguard Worker     }
320*d57664e9SAndroid Build Coastguard Worker 
321*d57664e9SAndroid Build Coastguard Worker     // Deal with the special @null value -- it turns back to TYPE_NULL.
322*d57664e9SAndroid Build Coastguard Worker     if (value.type == Res_value::TYPE_REFERENCE && value.data == 0U) {
323*d57664e9SAndroid Build Coastguard Worker       DEBUG_LOG("-> Setting to @null!");
324*d57664e9SAndroid Build Coastguard Worker       value.type = Res_value::TYPE_NULL;
325*d57664e9SAndroid Build Coastguard Worker       value.data = Res_value::DATA_NULL_UNDEFINED;
326*d57664e9SAndroid Build Coastguard Worker       value.cookie = kInvalidCookie;
327*d57664e9SAndroid Build Coastguard Worker     }
328*d57664e9SAndroid Build Coastguard Worker 
329*d57664e9SAndroid Build Coastguard Worker     DEBUG_LOG("Attribute 0x%08x: type=0x%x, data=0x%08x", cur_ident, value.type, value.data);
330*d57664e9SAndroid Build Coastguard Worker 
331*d57664e9SAndroid Build Coastguard Worker     // Write the final value back to Java.
332*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_TYPE] = value.type;
333*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DATA] = value.data;
334*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_ASSET_COOKIE] = ApkAssetsCookieToJavaCookie(value.cookie);
335*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_RESOURCE_ID] = value.resid;
336*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_CHANGING_CONFIGURATIONS] = value.flags;
337*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DENSITY] = value.config.density;
338*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_SOURCE_RESOURCE_ID] = value_source_resid;
339*d57664e9SAndroid Build Coastguard Worker 
340*d57664e9SAndroid Build Coastguard Worker     if (value.type != Res_value::TYPE_NULL || value.data == Res_value::DATA_NULL_EMPTY) {
341*d57664e9SAndroid Build Coastguard Worker       // out_indices must NOT be nullptr.
342*d57664e9SAndroid Build Coastguard Worker       out_indices[++indices_idx] = ii;
343*d57664e9SAndroid Build Coastguard Worker     }
344*d57664e9SAndroid Build Coastguard Worker     out_values += STYLE_NUM_ENTRIES;
345*d57664e9SAndroid Build Coastguard Worker   }
346*d57664e9SAndroid Build Coastguard Worker 
347*d57664e9SAndroid Build Coastguard Worker   // out_indices must NOT be nullptr.
348*d57664e9SAndroid Build Coastguard Worker   out_indices[0] = indices_idx;
349*d57664e9SAndroid Build Coastguard Worker   return {};
350*d57664e9SAndroid Build Coastguard Worker }
351*d57664e9SAndroid Build Coastguard Worker 
RetrieveAttributes(AssetManager2 * assetmanager,ResXMLParser * xml_parser,uint32_t * attrs,size_t attrs_length,uint32_t * out_values,uint32_t * out_indices)352*d57664e9SAndroid Build Coastguard Worker base::expected<std::monostate, IOError> RetrieveAttributes(AssetManager2* assetmanager,
353*d57664e9SAndroid Build Coastguard Worker                                                            ResXMLParser* xml_parser,
354*d57664e9SAndroid Build Coastguard Worker                                                            uint32_t* attrs,
355*d57664e9SAndroid Build Coastguard Worker                                                            size_t attrs_length,
356*d57664e9SAndroid Build Coastguard Worker                                                            uint32_t* out_values,
357*d57664e9SAndroid Build Coastguard Worker                                                            uint32_t* out_indices) {
358*d57664e9SAndroid Build Coastguard Worker   int indices_idx = 0;
359*d57664e9SAndroid Build Coastguard Worker 
360*d57664e9SAndroid Build Coastguard Worker   // Retrieve the XML attributes, if requested.
361*d57664e9SAndroid Build Coastguard Worker   size_t ix = 0;
362*d57664e9SAndroid Build Coastguard Worker   const size_t xml_attr_count = xml_parser->getAttributeCount();
363*d57664e9SAndroid Build Coastguard Worker   uint32_t cur_xml_attr = xml_parser->getAttributeNameResID(ix);
364*d57664e9SAndroid Build Coastguard Worker 
365*d57664e9SAndroid Build Coastguard Worker   // Now iterate through all of the attributes that the client has requested,
366*d57664e9SAndroid Build Coastguard Worker   // filling in each with whatever data we can find.
367*d57664e9SAndroid Build Coastguard Worker   for (size_t ii = 0; ii < attrs_length; ii++) {
368*d57664e9SAndroid Build Coastguard Worker     const uint32_t cur_ident = attrs[ii];
369*d57664e9SAndroid Build Coastguard Worker      AssetManager2::SelectedValue value{};
370*d57664e9SAndroid Build Coastguard Worker 
371*d57664e9SAndroid Build Coastguard Worker     // Try to find a value for this attribute...
372*d57664e9SAndroid Build Coastguard Worker     // Skip through XML attributes until the end or the next possible match.
373*d57664e9SAndroid Build Coastguard Worker     while (ix < xml_attr_count && cur_ident > cur_xml_attr) {
374*d57664e9SAndroid Build Coastguard Worker       cur_xml_attr = xml_parser->getAttributeNameResID(++ix);
375*d57664e9SAndroid Build Coastguard Worker     }
376*d57664e9SAndroid Build Coastguard Worker 
377*d57664e9SAndroid Build Coastguard Worker     // Retrieve the current XML attribute if it matches, and step to next.
378*d57664e9SAndroid Build Coastguard Worker     if (ix < xml_attr_count && cur_ident == cur_xml_attr) {
379*d57664e9SAndroid Build Coastguard Worker       Res_value attribute_value{};
380*d57664e9SAndroid Build Coastguard Worker       xml_parser->getAttributeValue(ix, &attribute_value);
381*d57664e9SAndroid Build Coastguard Worker       value.type = attribute_value.dataType;
382*d57664e9SAndroid Build Coastguard Worker       value.data = attribute_value.data;
383*d57664e9SAndroid Build Coastguard Worker       cur_xml_attr = xml_parser->getAttributeNameResID(++ix);
384*d57664e9SAndroid Build Coastguard Worker     }
385*d57664e9SAndroid Build Coastguard Worker 
386*d57664e9SAndroid Build Coastguard Worker     if (value.type != Res_value::TYPE_NULL) {
387*d57664e9SAndroid Build Coastguard Worker       // Take care of resolving the found resource to its final value.
388*d57664e9SAndroid Build Coastguard Worker       auto result = assetmanager->ResolveReference(value);
389*d57664e9SAndroid Build Coastguard Worker       if (UNLIKELY(IsIOError(result))) {
390*d57664e9SAndroid Build Coastguard Worker         return base::unexpected(GetIOError(result.error()));
391*d57664e9SAndroid Build Coastguard Worker       }
392*d57664e9SAndroid Build Coastguard Worker     }
393*d57664e9SAndroid Build Coastguard Worker 
394*d57664e9SAndroid Build Coastguard Worker     // Deal with the special @null value -- it turns back to TYPE_NULL.
395*d57664e9SAndroid Build Coastguard Worker     if (value.type == Res_value::TYPE_REFERENCE && value.data == 0U) {
396*d57664e9SAndroid Build Coastguard Worker       value.type = Res_value::TYPE_NULL;
397*d57664e9SAndroid Build Coastguard Worker       value.data = Res_value::DATA_NULL_UNDEFINED;
398*d57664e9SAndroid Build Coastguard Worker       value.cookie = kInvalidCookie;
399*d57664e9SAndroid Build Coastguard Worker     }
400*d57664e9SAndroid Build Coastguard Worker 
401*d57664e9SAndroid Build Coastguard Worker     // Write the final value back to Java.
402*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_TYPE] = value.type;
403*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DATA] = value.data;
404*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_ASSET_COOKIE] = ApkAssetsCookieToJavaCookie(value.cookie);
405*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_RESOURCE_ID] = value.resid;
406*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_CHANGING_CONFIGURATIONS] = value.flags;
407*d57664e9SAndroid Build Coastguard Worker     out_values[STYLE_DENSITY] = value.config.density;
408*d57664e9SAndroid Build Coastguard Worker 
409*d57664e9SAndroid Build Coastguard Worker     if (out_indices != nullptr &&
410*d57664e9SAndroid Build Coastguard Worker         (value.type != Res_value::TYPE_NULL ||
411*d57664e9SAndroid Build Coastguard Worker          value.data == Res_value::DATA_NULL_EMPTY)) {
412*d57664e9SAndroid Build Coastguard Worker       out_indices[++indices_idx] = ii;
413*d57664e9SAndroid Build Coastguard Worker     }
414*d57664e9SAndroid Build Coastguard Worker 
415*d57664e9SAndroid Build Coastguard Worker     out_values += STYLE_NUM_ENTRIES;
416*d57664e9SAndroid Build Coastguard Worker   }
417*d57664e9SAndroid Build Coastguard Worker 
418*d57664e9SAndroid Build Coastguard Worker   if (out_indices != nullptr) {
419*d57664e9SAndroid Build Coastguard Worker     out_indices[0] = indices_idx;
420*d57664e9SAndroid Build Coastguard Worker   }
421*d57664e9SAndroid Build Coastguard Worker   return {};
422*d57664e9SAndroid Build Coastguard Worker }
423*d57664e9SAndroid Build Coastguard Worker 
424*d57664e9SAndroid Build Coastguard Worker }  // namespace android
425