123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261 |
- // Copyright (c) 2013 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "extensions/common/manifest_handler.h"
- #include <stddef.h>
- #include <map>
- #include <vector>
- #include "base/check.h"
- #include "base/containers/contains.h"
- #include "extensions/common/extension.h"
- #include "extensions/common/permissions/manifest_permission.h"
- #include "extensions/common/permissions/manifest_permission_set.h"
- namespace extensions {
- namespace {
- static base::LazyInstance<ManifestHandlerRegistry>::DestructorAtExit
- g_registry = LAZY_INSTANCE_INITIALIZER;
- static ManifestHandlerRegistry* g_registry_override = NULL;
- } // namespace
- ManifestHandler::ManifestHandler() {
- }
- ManifestHandler::~ManifestHandler() {
- }
- bool ManifestHandler::Validate(const Extension* extension,
- std::string* error,
- std::vector<InstallWarning>* warnings) const {
- return true;
- }
- bool ManifestHandler::AlwaysParseForType(Manifest::Type type) const {
- return false;
- }
- bool ManifestHandler::AlwaysValidateForType(Manifest::Type type) const {
- return false;
- }
- const std::vector<std::string> ManifestHandler::PrerequisiteKeys() const {
- return std::vector<std::string>();
- }
- ManifestPermission* ManifestHandler::CreatePermission() {
- return NULL;
- }
- ManifestPermission* ManifestHandler::CreateInitialRequiredPermission(
- const Extension* extension) {
- return NULL;
- }
- // static
- void ManifestHandler::FinalizeRegistration() {
- ManifestHandlerRegistry::Get()->Finalize();
- }
- // static
- bool ManifestHandler::IsRegistrationFinalized() {
- return ManifestHandlerRegistry::Get()->is_finalized_;
- }
- // static
- bool ManifestHandler::ParseExtension(Extension* extension,
- std::u16string* error) {
- return ManifestHandlerRegistry::Get()->ParseExtension(extension, error);
- }
- // static
- bool ManifestHandler::ValidateExtension(const Extension* extension,
- std::string* error,
- std::vector<InstallWarning>* warnings) {
- return ManifestHandlerRegistry::Get()->ValidateExtension(extension, error,
- warnings);
- }
- // static
- ManifestPermission* ManifestHandler::CreatePermission(const std::string& name) {
- return ManifestHandlerRegistry::Get()->CreatePermission(name);
- }
- // static
- void ManifestHandler::AddExtensionInitialRequiredPermissions(
- const Extension* extension, ManifestPermissionSet* permission_set) {
- return ManifestHandlerRegistry::Get()->AddExtensionInitialRequiredPermissions(
- extension, permission_set);
- }
- // static
- const std::vector<std::string> ManifestHandler::SingleKey(
- const std::string& key) {
- return std::vector<std::string>(1, key);
- }
- // static
- const size_t ManifestHandlerRegistry::kHandlerMax;
- ManifestHandlerRegistry::ManifestHandlerRegistry() : is_finalized_(false) {
- }
- ManifestHandlerRegistry::~ManifestHandlerRegistry() {
- }
- void ManifestHandlerRegistry::Finalize() {
- CHECK(!is_finalized_);
- SortManifestHandlers();
- is_finalized_ = true;
- }
- void ManifestHandlerRegistry::RegisterHandler(
- std::unique_ptr<ManifestHandler> handler) {
- CHECK(!is_finalized_);
- ManifestHandler* raw_handler = handler.get();
- owned_manifest_handlers_.push_back(std::move(handler));
- for (const char* key : raw_handler->Keys()) {
- auto insertion = handlers_.emplace(key, raw_handler);
- DCHECK(insertion.second)
- << "A ManifestHandler was already registered for key: " << key;
- }
- }
- bool ManifestHandlerRegistry::ParseExtension(Extension* extension,
- std::u16string* error) {
- std::map<int, ManifestHandler*> handlers_by_priority;
- for (ManifestHandlerMap::iterator iter = handlers_.begin();
- iter != handlers_.end(); ++iter) {
- ManifestHandler* handler = iter->second;
- if (extension->manifest()->FindPath(iter->first) ||
- handler->AlwaysParseForType(extension->GetType())) {
- handlers_by_priority[priority_map_[handler]] = handler;
- }
- }
- for (auto iter = handlers_by_priority.begin();
- iter != handlers_by_priority.end(); ++iter) {
- if (!(iter->second)->Parse(extension, error))
- return false;
- }
- return true;
- }
- bool ManifestHandlerRegistry::ValidateExtension(
- const Extension* extension,
- std::string* error,
- std::vector<InstallWarning>* warnings) {
- std::set<ManifestHandler*> handlers;
- for (ManifestHandlerMap::iterator iter = handlers_.begin();
- iter != handlers_.end(); ++iter) {
- ManifestHandler* handler = iter->second;
- if (extension->manifest()->FindPath(iter->first) ||
- handler->AlwaysValidateForType(extension->GetType())) {
- handlers.insert(handler);
- }
- }
- for (auto iter = handlers.begin(); iter != handlers.end(); ++iter) {
- if (!(*iter)->Validate(extension, error, warnings))
- return false;
- }
- return true;
- }
- ManifestPermission* ManifestHandlerRegistry::CreatePermission(
- const std::string& name) {
- ManifestHandlerMap::const_iterator it = handlers_.find(name);
- if (it == handlers_.end())
- return NULL;
- return it->second->CreatePermission();
- }
- void ManifestHandlerRegistry::AddExtensionInitialRequiredPermissions(
- const Extension* extension, ManifestPermissionSet* permission_set) {
- for (ManifestHandlerMap::const_iterator it = handlers_.begin();
- it != handlers_.end(); ++it) {
- std::unique_ptr<ManifestPermission> permission(
- it->second->CreateInitialRequiredPermission(extension));
- if (permission) {
- permission_set->insert(std::move(permission));
- }
- }
- }
- // static
- ManifestHandlerRegistry* ManifestHandlerRegistry::Get() {
- if (!g_registry_override)
- return g_registry.Pointer();
- return g_registry_override;
- }
- // static
- ManifestHandlerRegistry* ManifestHandlerRegistry::SetForTesting(
- ManifestHandlerRegistry* new_registry) {
- ManifestHandlerRegistry* old_registry = ManifestHandlerRegistry::Get();
- if (new_registry != g_registry.Pointer())
- g_registry_override = new_registry;
- else
- g_registry_override = NULL;
- return old_registry;
- }
- // static
- void ManifestHandlerRegistry::ResetForTesting() {
- ManifestHandlerRegistry* registry = Get();
- registry->priority_map_.clear();
- registry->handlers_.clear();
- registry->is_finalized_ = false;
- }
- void ManifestHandlerRegistry::SortManifestHandlers() {
- std::vector<ManifestHandler*> unsorted_handlers;
- unsorted_handlers.reserve(handlers_.size());
- for (const auto& key_value : handlers_) {
- unsorted_handlers.push_back(key_value.second);
- }
- int priority = 0;
- while (true) {
- std::vector<ManifestHandler*> next_unsorted_handlers;
- next_unsorted_handlers.reserve(unsorted_handlers.size());
- for (ManifestHandler* handler : unsorted_handlers) {
- const std::vector<std::string>& prerequisites =
- handler->PrerequisiteKeys();
- int unsatisfied = prerequisites.size();
- for (const std::string& key : prerequisites) {
- ManifestHandlerMap::const_iterator prereq_iter = handlers_.find(key);
- // If the prerequisite does not exist, crash.
- CHECK(prereq_iter != handlers_.end())
- << "Extension manifest handler depends on unrecognized key " << key;
- // Prerequisite is in our map.
- if (base::Contains(priority_map_, prereq_iter->second))
- unsatisfied--;
- }
- if (unsatisfied == 0) {
- priority_map_[handler] = priority;
- priority++;
- } else {
- // Put in the list for next time.
- next_unsorted_handlers.push_back(handler);
- }
- }
- if (next_unsorted_handlers.size() == unsorted_handlers.size())
- break;
- unsorted_handlers.swap(next_unsorted_handlers);
- }
- // If there are any leftover unsorted handlers, they must have had
- // circular dependencies.
- CHECK(unsorted_handlers.empty())
- << "Extension manifest handlers have circular dependencies!";
- }
- } // namespace extensions
|