Implement <webview>.addContentScript/removeContentScript API [1]

extensions/renderer/user_script_set_manager.cc

 // Copyright 2014 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/renderer/user_script_set_manager.h"
 
 #include "components/crx_file/id_util.h"
 #include "content/public/renderer/render_thread.h"
 #include "extensions/common/extension_messages.h"
 #include "extensions/renderer/dispatcher.h"
 #include "extensions/renderer/script_injection.h"
 #include "extensions/renderer/user_script_set.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_message_macros.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
 
 namespace extensions {
 
 UserScriptSetManager::UserScriptSetManager(const ExtensionSet* extensions)
-    : static_scripts_(extensions), extensions_(extensions) {
+    : static_scripts_(extensions),
+      is_in_guest_render_process_(false),
+      extensions_(extensions) {
   content::RenderThread::Get()->AddObserver(this);
 }
 
 UserScriptSetManager::~UserScriptSetManager() {
 }
 
 void UserScriptSetManager::AddObserver(Observer* observer) {
   observers_.AddObserver(observer);
 }
 
@@ skipping 29 matching lines @@
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void UserScriptSetManager::GetAllInjections(
     ScopedVector<ScriptInjection>* injections,
     blink::WebFrame* web_frame,
     int tab_id,
     UserScript::RunLocation run_location) {
-  static_scripts_.GetInjections(injections, web_frame, tab_id, run_location);
+  static_scripts_.GetInjections(injections, web_frame, tab_id, run_location,
+                                is_in_guest_render_process_);
   for (UserScriptSetMap::iterator it = programmatic_scripts_.begin();
        it != programmatic_scripts_.end();
        ++it) {
-    it->second->GetInjections(injections, web_frame, tab_id, run_location);
+    it->second->GetInjections(injections, web_frame, tab_id, run_location,
+                              is_in_guest_render_process_);
   }
 }
 
 void UserScriptSetManager::GetAllActiveExtensionIds(
     std::set<std::string>* ids) const {
   DCHECK(ids);
   static_scripts_.GetActiveExtensionIds(ids);
   for (UserScriptSetMap::const_iterator it = programmatic_scripts_.begin();
        it != programmatic_scripts_.end();
        ++it) {
     it->second->GetActiveExtensionIds(ids);
   }
 }
 
 UserScriptSet* UserScriptSetManager::GetProgrammaticScriptsByHostID(
     const HostID& host_id) {
   UserScriptSetMap::const_iterator it = programmatic_scripts_.find(host_id);
   return it != programmatic_scripts_.end() ? it->second.get() : NULL;
 }
 
 void UserScriptSetManager::OnUpdateUserScripts(
     base::SharedMemoryHandle shared_memory,
     const HostID& host_id,
-    const std::set<HostID>& changed_hosts) {
+    const std::set<HostID>& changed_hosts,
+    bool is_to_guest_render_process) {
+  is_in_guest_render_process_ = is_to_guest_render_process;
+
   if (!base::SharedMemory::IsHandleValid(shared_memory)) {
     NOTREACHED() << "Bad scripts handle";
     return;
   }
 
   for (const HostID& host_id : changed_hosts) {
     if (host_id.type() == HostID::EXTENSIONS &&
         !crx_file::id_util::IdIsValid(host_id.id())) {
       NOTREACHED() << "Invalid extension id: " << host_id.id();
       return;
@@ skipping 39 matching lines @@
 
   if (scripts->UpdateUserScripts(shared_memory, *effective_hosts)) {
     FOR_EACH_OBSERVER(
         Observer,
         observers_,
         OnUserScriptsUpdated(*effective_hosts, scripts->scripts()));
   }
 }
 
 }  // namespace extensions