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 19namespace aapt { 20namespace xml { 21 22static bool wrapper_one(XmlNodeAction::ActionFunc& f, Element* el, SourcePathDiagnostics*) { 23 return f(el); 24} 25 26static bool wrapper_two(XmlNodeAction::ActionFuncWithDiag& f, Element* el, 27 SourcePathDiagnostics* diag) { 28 return f(el, diag); 29} 30 31void 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 36void 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 41static 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 49bool 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 77bool 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