extensions/renderer/user_script_set_manager.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "extensions/renderer/user_script_set_manager.h"
   6
   7 #include "components/crx_file/id_util.h"
   8 #include "content/public/renderer/render_thread.h"
   9 #include "extensions/common/extension_messages.h"
  10 #include "extensions/renderer/dispatcher.h"
  11 #include "extensions/renderer/user_script_set.h"
  12 #include "ipc/ipc_message.h"
  13 #include "ipc/ipc_message_macros.h"
  14 #include "third_party/WebKit/public/web/WebFrame.h"
  15
  16 namespace extensions {
  17
  18 UserScriptSetManager::UserScriptSetManager(const ExtensionSet* extensions)
  19     : static_scripts_(extensions), extensions_(extensions) {
  20   content::RenderThread::Get()->AddObserver(this);
  21 }
  22
  23 UserScriptSetManager::~UserScriptSetManager() {
  24 }
  25
  26 void UserScriptSetManager::AddObserver(Observer* observer) {
  27   observers_.AddObserver(observer);
  28 }
  29
  30 void UserScriptSetManager::RemoveObserver(Observer* observer) {
  31   observers_.RemoveObserver(observer);
  32 }
  33
  34 bool UserScriptSetManager::OnControlMessageReceived(
  35     const IPC::Message& message) {
  36   bool handled = true;
  37   IPC_BEGIN_MESSAGE_MAP(UserScriptSetManager, message)
  38     IPC_MESSAGE_HANDLER(ExtensionMsg_UpdateUserScripts, OnUpdateUserScripts)
  39     IPC_MESSAGE_UNHANDLED(handled = false)
  40   IPC_END_MESSAGE_MAP()
  41   return handled;
  42 }
  43
  44 const UserScriptSet* UserScriptSetManager::GetProgrammaticScriptsByExtension(
  45     const ExtensionId& extension_id) {
  46   UserScriptSetMap::const_iterator it =
  47       programmatic_scripts_.find(extension_id);
  48   return it != programmatic_scripts_.end() ? it->second.get() : NULL;
  49 }
  50
  51 void UserScriptSetManager::GetAllInjections(
  52     ScopedVector<ScriptInjection>* injections,
  53     blink::WebFrame* web_frame,
  54     int tab_id,
  55     UserScript::RunLocation run_location) {
  56   static_scripts_.GetInjections(injections, web_frame, tab_id, run_location);
  57   for (UserScriptSetMap::iterator it = programmatic_scripts_.begin();
  58        it != programmatic_scripts_.end();
  59        ++it) {
  60     it->second->GetInjections(injections, web_frame, tab_id, run_location);
  61   }
  62 }
  63
  64 void UserScriptSetManager::GetAllActiveExtensionIds(
  65     std::set<std::string>* ids) const {
  66   DCHECK(ids);
  67   static_scripts_.GetActiveExtensionIds(ids);
  68   for (UserScriptSetMap::const_iterator it = programmatic_scripts_.begin();
  69        it != programmatic_scripts_.end();
  70        ++it) {
  71     it->second->GetActiveExtensionIds(ids);
  72   }
  73 }
  74
  75 void UserScriptSetManager::OnUpdateUserScripts(
  76     base::SharedMemoryHandle shared_memory,
  77     const ExtensionId& extension_id,
  78     const std::set<std::string>& changed_extensions) {
  79   if (!base::SharedMemory::IsHandleValid(shared_memory)) {
  80     NOTREACHED() << "Bad scripts handle";
  81     return;
  82   }
  83
  84   for (std::set<std::string>::const_iterator iter = changed_extensions.begin();
  85        iter != changed_extensions.end();
  86        ++iter) {
  87     if (!crx_file::id_util::IdIsValid(*iter)) {
  88       NOTREACHED() << "Invalid extension id: " << *iter;
  89       return;
  90     }
  91   }
  92
  93   UserScriptSet* scripts = NULL;
  94   if (!extension_id.empty()) {
  95     // The expectation when there is an extensions that "owns" this shared
  96     // memory region is that it will list itself as the only changed extension.
  97     CHECK(changed_extensions.size() == 1 &&
  98           changed_extensions.find(extension_id) != changed_extensions.end());
  99     if (programmatic_scripts_.find(extension_id) ==
 100         programmatic_scripts_.end()) {
 101       scripts = new UserScriptSet(extensions_);
 102       programmatic_scripts_[extension_id] = make_linked_ptr(scripts);
 103     } else {
 104       scripts = programmatic_scripts_[extension_id].get();
 105     }
 106   } else {
 107     scripts = &static_scripts_;
 108   }
 109   DCHECK(scripts);
 110
 111   // If no extensions are included in the set, that indicates that all
 112   // extensions were updated. Add them all to the set so that observers and
 113   // individual UserScriptSets don't need to know this detail.
 114   const std::set<std::string>* effective_extensions = &changed_extensions;
 115   std::set<std::string> all_extensions;
 116   if (changed_extensions.empty()) {
 117     all_extensions = extensions_->GetIDs();
 118     effective_extensions = &all_extensions;
 119   }
 120
 121   if (scripts->UpdateUserScripts(shared_memory, *effective_extensions)) {
 122     FOR_EACH_OBSERVER(
 123         Observer,
 124         observers_,
 125         OnUserScriptsUpdated(*effective_extensions, scripts->scripts()));
 126   }
 127 }
 128
 129 }  // namespace extensions