1 /*
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 #include "xml/XmlActionExecutor.h"
18
19 namespace aapt {
20 namespace xml {
21
wrapper_one(XmlNodeAction::ActionFunc & f,Element * el,SourcePathDiagnostics *)22 static bool wrapper_one(XmlNodeAction::ActionFunc& f, Element* el, SourcePathDiagnostics*) {
23 return f(el);
24 }
25
wrapper_two(XmlNodeAction::ActionFuncWithDiag & f,Element * el,SourcePathDiagnostics * diag)26 static bool wrapper_two(XmlNodeAction::ActionFuncWithDiag& f, Element* el,
27 SourcePathDiagnostics* diag) {
28 return f(el, diag);
29 }
30
Action(XmlNodeAction::ActionFunc f)31 void XmlNodeAction::Action(XmlNodeAction::ActionFunc f) {
32 actions_.emplace_back(std::bind(
33 wrapper_one, std::move(f), std::placeholders::_1, std::placeholders::_2));
34 }
35
Action(XmlNodeAction::ActionFuncWithDiag f)36 void XmlNodeAction::Action(XmlNodeAction::ActionFuncWithDiag f) {
37 actions_.emplace_back(std::bind(
38 wrapper_two, std::move(f), std::placeholders::_1, std::placeholders::_2));
39 }
40
PrintElementToDiagMessage(const Element * el,DiagMessage * msg)41 static void PrintElementToDiagMessage(const Element* el, DiagMessage* msg) {
42 *msg << "<";
43 if (!el->namespace_uri.empty()) {
44 *msg << el->namespace_uri << ":";
45 }
46 *msg << el->name << ">";
47 }
48
Execute(XmlActionExecutorPolicy policy,SourcePathDiagnostics * diag,Element * el) const49 bool XmlNodeAction::Execute(XmlActionExecutorPolicy policy, SourcePathDiagnostics* diag,
50 Element* el) const {
51 bool error = false;
52 for (const ActionFuncWithDiag& action : actions_) {
53 error |= !action(el, diag);
54 }
55
56 for (Element* child_el : el->GetChildElements()) {
57 if (child_el->namespace_uri.empty()) {
58 std::map<std::string, XmlNodeAction>::const_iterator iter = map_.find(child_el->name);
59 if (iter != map_.end()) {
60 error |= !iter->second.Execute(policy, diag, child_el);
61 continue;
62 }
63
64 if (policy == XmlActionExecutorPolicy::kWhitelist) {
65 DiagMessage error_msg(child_el->line_number);
66 error_msg << "unknown element ";
67 PrintElementToDiagMessage(child_el, &error_msg);
68 error_msg << " found";
69 diag->Error(error_msg);
70 error = true;
71 }
72 }
73 }
74 return !error;
75 }
76
Execute(XmlActionExecutorPolicy policy,IDiagnostics * diag,XmlResource * doc) const77 bool XmlActionExecutor::Execute(XmlActionExecutorPolicy policy, IDiagnostics* diag,
78 XmlResource* doc) const {
79 SourcePathDiagnostics source_diag(doc->file.source, diag);
80
81 Element* el = FindRootElement(doc);
82 if (!el) {
83 if (policy == XmlActionExecutorPolicy::kWhitelist) {
84 source_diag.Error(DiagMessage() << "no root XML tag found");
85 return false;
86 }
87 return true;
88 }
89
90 if (el->namespace_uri.empty()) {
91 std::map<std::string, XmlNodeAction>::const_iterator iter = map_.find(el->name);
92 if (iter != map_.end()) {
93 return iter->second.Execute(policy, &source_diag, el);
94 }
95
96 if (policy == XmlActionExecutorPolicy::kWhitelist) {
97 DiagMessage error_msg(el->line_number);
98 error_msg << "unknown element ";
99 PrintElementToDiagMessage(el, &error_msg);
100 error_msg << " found";
101 source_diag.Error(error_msg);
102 return false;
103 }
104 }
105 return true;
106 }
107
108 } // namespace xml
109 } // namespace aapt
110