Allow content script insertion on about:-URLs.

chrome/renderer/extensions/user_script_slave.cc

 // Copyright (c) 2012 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 "chrome/renderer/extensions/user_script_slave.h"
 
 #include <map>
 
 #include "base/command_line.h"
 #include "base/logging.h"
@@ skipping 20 matching lines @@
 #include "third_party/WebKit/public/web/WebDataSource.h"
 #include "third_party/WebKit/public/web/WebDocument.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
 #include "third_party/WebKit/public/web/WebSecurityOrigin.h"
 #include "third_party/WebKit/public/web/WebSecurityPolicy.h"
 #include "third_party/WebKit/public/web/WebView.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "url/gurl.h"
 
 using blink::WebFrame;
+using blink::WebDocument;
 using blink::WebSecurityOrigin;
 using blink::WebSecurityPolicy;
 using blink::WebString;
 using blink::WebVector;
 using blink::WebView;
 using content::RenderThread;
 
 namespace extensions {
 
 // These two strings are injected before and after the Greasemonkey API and
@@ skipping 136 matching lines @@
   // the compose iframe's dataSource URL is about:blank, but the document URL
   // changes to match the parent document after Gmail document.writes into
   // it to create the editor.
   // http://code.google.com/p/chromium/issues/detail?id=86742
   blink::WebDataSource* data_source = frame->provisionalDataSource() ?
       frame->provisionalDataSource() : frame->dataSource();
   CHECK(data_source);
   return GURL(data_source->request().url());
 }
 
+GURL UserScriptSlave::GetEffectiveDocumentURL(const WebFrame* frame,
+                                              const GURL& document_url,
+                                              bool match_about_blank) {
+  if (!match_about_blank || !document_url.SchemeIs(content::kAboutScheme))
+    return document_url;
+
+  // Non-sandboxed about:blank and about:srcdoc pages inherit their security
+  // origin from their parent frame/window. So, traverse the frame/window
+  // hierarchy to find the closest non-about:-page and return its URL.
+  const WebFrame* parent = frame;
+  do {
+    parent = parent->parent() ? parent->parent() : parent->opener();
+  } while (parent != NULL &&
+           GURL(parent->document().url()).SchemeIs(content::kAboutScheme));
+

[not at google - send to devlin, 2014/05/02 16:01:48]

not quite sure about this. if we're in an non-about-blank iframe on the same
domain then |parent| will be the non-NULL parent at this point.

e.g. "google.com/foo" has an iframe to "google.com/bar".

then the "if (parent)" check will pass, and the "securityOrigin.canAccess" check
will pass, so we'll end up returning "google.com/bar" here when really it should
be "google.com/foo".

this might matter in practice, if an extension requests a content script to be
executed on "google.com/foo/*" with allFrames=true, it will end up being
injected into the google.com/bar iframe too. unless there are checks elsewhere,
which is very possible. not a big deal for security reasons, but, it's still
wrong.

[robwu, 2014/05/02 17:03:40]

If we're on a non-about:blank frame, then the code doesn't even reach this line.
This loop only runs for frames whose URL is about:blank. The caller of the
function explicitly passes the URL of the frame via |document_url|. If the
scheme is not "about:", then the function returns the same value again. I've
added a comment to emphasize this. 

+  if (parent) {
+    // Only return the parent URL if the frame can access it.
+    const WebDocument& parentDocument = parent->document();

[not at google - send to devlin, 2014/05/02 16:01:48]

as you pointed out, parent_document not parentDocument.

[robwu, 2014/05/02 17:03:40]

Done.

+    if (frame->document().securityOrigin().canAccess(
+            parentDocument.securityOrigin()))
+      return GURL(parentDocument.url());

[not at google - send to devlin, 2014/05/02 16:01:48]

WebURL has a GURL operator so you don't need the explicit GURL(..) constructor:

https://code.google.com/p/chromium/codesearch#chromium/src/third_party/WebKit...

[robwu, 2014/05/02 17:03:40]

Done.

+  }
+  return document_url;
+}
+
 void UserScriptSlave::InjectScripts(WebFrame* frame,
                                     UserScript::RunLocation location) {
   GURL data_source_url = GetDataSourceURLForFrame(frame);
   if (data_source_url.is_empty())
     return;
 
   if (frame->isViewSourceModeEnabled())
     data_source_url = GURL(content::kViewSourceScheme + std::string(":") +
                            data_source_url.spec());
 
@@ skipping 10 matching lines @@
     if (frame->parent() && !script->match_all_frames())
       continue;  // Only match subframes if the script declared it wanted to.
 
     const Extension* extension = extensions_->GetByID(script->extension_id());
 
     // Since extension info is sent separately from user script info, they can
     // be out of sync. We just ignore this situation.
     if (!extension)
       continue;
 
+    const GURL& document_url = GetEffectiveDocumentURL(
+        frame, data_source_url, script->match_about_blank());
+
     // Content scripts are not tab-specific.
     const int kNoTabId = -1;
     // We don't have a process id in this context.
     const int kNoProcessId = -1;
     if (!PermissionsData::CanExecuteScriptOnPage(extension,
-                                                 data_source_url,
+                                                 document_url,
                                                  frame->top()->document().url(),
                                                  kNoTabId,
                                                  script,
                                                  kNoProcessId,
                                                  NULL)) {
       continue;
     }
 
     if (location == UserScript::DOCUMENT_START) {
       num_css += script->css_scripts().size();
@@ skipping 75 matching lines @@
   } else if (location == UserScript::DOCUMENT_IDLE) {
     UMA_HISTOGRAM_COUNTS_100("Extensions.InjectIdle_ScriptCount", num_scripts);
     if (num_scripts)
       UMA_HISTOGRAM_TIMES("Extensions.InjectIdle_Time", timer.Elapsed());
   } else {
     NOTREACHED();
   }
 }
 
 }  // namespace extensions