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

extensions/browser/user_script_loader.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/browser/user_script_loader.h"
 
 #include <set>
 #include <string>
 
 #include "base/bind.h"
@@ skipping 426 matching lines @@
     // though it removes the possibility that freeing the shared memory block
     // would open up enough FDs for long enough for a retry to succeed.
 
     // Pretend the extension change didn't happen.
     return;
   }
 
   // We've got scripts ready to go.
   shared_memory_.reset(shared_memory.release());
 
-  // If user scripts are coming from a <webview>, will only notify the
-  // RenderProcessHost of that <webview>; otherwise will notify all of the
-  // RenderProcessHosts.
-  if (user_scripts_ && !user_scripts_->empty() &&
-      (*user_scripts_)[0].consumer_instance_type() ==
-          UserScript::ConsumerInstanceType::WEBVIEW) {
-    DCHECK_EQ(1u, user_scripts_->size());
-    int render_process_id =
-        (*user_scripts_)[0].routing_info().render_process_id;
-    content::RenderProcessHost* host =
-        content::RenderProcessHost::FromID(render_process_id);
-    if (host)
-      SendUpdate(host, shared_memory_.get(), changed_hosts_);
-  } else {
-    for (content::RenderProcessHost::iterator i(
-            content::RenderProcessHost::AllHostsIterator());
-        !i.IsAtEnd(); i.Advance()) {
-      SendUpdate(i.GetCurrentValue(), shared_memory_.get(), changed_hosts_);
-    }
+  for (content::RenderProcessHost::iterator i(
+           content::RenderProcessHost::AllHostsIterator());
+       !i.IsAtEnd(); i.Advance()) {
+    SendUpdate(i.GetCurrentValue(), shared_memory_.get(), changed_hosts_);
   }
   changed_hosts_.clear();
 
   content::NotificationService::current()->Notify(
       extensions::NOTIFICATION_USER_SCRIPTS_UPDATED,
       content::Source<BrowserContext>(browser_context_),
       content::Details<base::SharedMemory>(shared_memory_.get()));
 }
 
 void UserScriptLoader::SendUpdate(content::RenderProcessHost* process,
                                   base::SharedMemory* shared_memory,
                                   const std::set<HostID>& changed_hosts) {
-  // Don't allow injection of content scripts into <webview>.
-  if (process->IsIsolatedGuest())
+  // Don't allow injection of extensions' content scripts into <webview>.
+  if (process->IsIsolatedGuest() && host_id().id().empty())
     return;
 
   // Make sure we only send user scripts to processes in our browser_context.
   if (!ExtensionsBrowserClient::Get()->IsSameContext(
           browser_context_, process->GetBrowserContext()))
     return;
 
   // If the process is being started asynchronously, early return.  We'll end up
   // calling InitUserScripts when it's created which will call this again.
   base::ProcessHandle handle = process->GetHandle();
   if (!handle)
     return;
 
   base::SharedMemoryHandle handle_for_process;
   if (!shared_memory->ShareToProcess(handle, &handle_for_process))
     return;  // This can legitimately fail if the renderer asserts at startup.
 
   if (base::SharedMemory::IsHandleValid(handle_for_process)) {
     process->Send(new ExtensionMsg_UpdateUserScripts(handle_for_process,
                                                      host_id(), changed_hosts));
   }
 }
 
 }  // namespace extensions