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

extensions/renderer/user_script_set.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.h"
 
 #include "base/memory/ref_counted.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/renderer/render_thread.h"
 #include "extensions/common/extension.h"
@@ skipping 50 matching lines @@
       continue;
     DCHECK(!(*iter)->extension_id().empty());
     ids->insert((*iter)->extension_id());
   }
 }
 
 void UserScriptSet::GetInjections(
     ScopedVector<ScriptInjection>* injections,
     blink::WebFrame* web_frame,
     int tab_id,
-    UserScript::RunLocation run_location) {
+    UserScript::RunLocation run_location,
+    bool has_guest_content_script) {
   GURL document_url = GetDocumentUrlForFrame(web_frame);
   for (ScopedVector<UserScript>::const_iterator iter = scripts_.begin();
        iter != scripts_.end();
        ++iter) {
     scoped_ptr<ScriptInjection> injection = GetInjectionForScript(
         *iter,
         web_frame,
         tab_id,
         run_location,
         document_url,
-        false /* is_declarative */);
+        false /* is_declarative */,
+        has_guest_content_script);
     if (injection.get())
       injections->push_back(injection.release());
   }
 }
 
 bool UserScriptSet::UpdateUserScripts(base::SharedMemoryHandle shared_memory,
                                       const std::set<HostID>& changed_hosts) {
   bool only_inject_incognito =
       ExtensionsRendererClient::Get()->IsIncognitoProcess();
 
@@ skipping 64 matching lines @@
     const GURL& document_url) {
   for (ScopedVector<UserScript>::const_iterator it = scripts_.begin();
        it != scripts_.end();
        ++it) {
     if ((*it)->id() == script_id) {
       return GetInjectionForScript(*it,
                                    web_frame,
                                    tab_id,
                                    run_location,
                                    document_url,
-                                   true /* is_declarative */);
+                                   true /* is_declarative */,
+                                   false /*has_guest_content_script */);
     }
   }
   return scoped_ptr<ScriptInjection>();
 }
 
 // TODO(dcheng): Scripts can't be injected on a remote frame, so this function
 // signature needs to be updated.
 scoped_ptr<ScriptInjection> UserScriptSet::GetInjectionForScript(
     UserScript* script,
     blink::WebFrame* web_frame,
     int tab_id,
     UserScript::RunLocation run_location,
     const GURL& document_url,
-    bool is_declarative) {
+    bool is_declarative,
+    bool has_guest_content_script) {
   scoped_ptr<ScriptInjection> injection;
   scoped_ptr<const InjectionHost> injection_host;
 
   const HostID& host_id = script->host_id();
   if (host_id.type() == HostID::EXTENSIONS) {
     injection_host = ExtensionInjectionHost::Create(host_id.id(), extensions_);
     if (!injection_host)
       return injection.Pass();
   } else {
     DCHECK_EQ(host_id.type(), HostID::WEBUI);
     injection_host.reset(new WebUIInjectionHost(host_id));
   }
 
   if (web_frame->parent() && !script->match_all_frames())
     return injection.Pass();  // Only match subframes if the script declared it.
 
   GURL effective_document_url = ScriptContext::GetEffectiveDocumentURL(
       web_frame, document_url, script->match_about_blank());
 
   if (!script->MatchesURL(effective_document_url))
     return injection.Pass();
 
-  scoped_ptr<ScriptInjector> injector(new UserScriptInjector(script,
-                                                             this,
-                                                             is_declarative));
+  scoped_ptr<ScriptInjector> injector(new UserScriptInjector(
+      script,
+      this,
+      is_declarative,
+      has_guest_content_script));
 
   blink::WebFrame* top_frame = web_frame->top();
   // It doesn't make sense to do script injection for remote frames, since they
   // cannot host any documents or content.
   // TODO(kalman): Fix this properly by moving all security checks into the
   // browser. See http://crbug.com/466373 for ongoing work here.
   if (top_frame->isWebRemoteFrame())
     return injection.Pass();
 
   if (injector->CanExecuteOnFrame(injection_host.get(), web_frame,
@@ skipping 12 matching lines @@
         injector.Pass(),
         web_frame->toWebLocalFrame(),
         injection_host.Pass(),
         run_location,
         tab_id));
   }
   return injection.Pass();
 }
 
 }  // namespace extensions