Add schema chrome-extension-resource:// for extension resources

chrome/renderer/extensions/extension_resource_request_policy.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/extension_resource_request_policy.h"
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/stringprintf.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_set.h"
 #include "googleurl/src/gurl.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebConsoleMessage.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDocument.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 
 // static
 bool ExtensionResourceRequestPolicy::CanRequestResource(

[abarth-chromium, 2012/04/13 17:11:24]

This function is getting very complicated.  Are there tests for it?  Can we add
some cases for the new functionality?

[Peng, 2012/04/17 13:52:05]

This function's input & output highly depend on environment. For example: this
function need get input from WebFrame, and send some output to WebFrame. I
didn't find a good way to create a unit test for this function. :(
Probably I can do is adding a browser_test for it. Do you think it is OK?

     const GURL& resource_url,
     WebKit::WebFrame* frame,
     const ExtensionSet* loaded_extensions) {
-  CHECK(resource_url.SchemeIs(chrome::kExtensionScheme));
+  if (resource_url.SchemeIs(chrome::kExtensionScheme)) {
+    const Extension* extension =
+        loaded_extensions->GetExtensionOrAppByURL(
+            ExtensionURLInfo(resource_url));
+    if (!extension) {
+      // Allow the load in the case of a non-existent extension. We'll just get
+      // a 404 from the browser process.
+      return true;
+    }
 
-  const Extension* extension =
-      loaded_extensions->GetExtensionOrAppByURL(ExtensionURLInfo(resource_url));
-  if (!extension) {
-    // Allow the load in the case of a non-existent extension. We'll just get a
-    // 404 from the browser process.
+    // Disallow loading of packaged resources for hosted apps. We don't allow
+    // hybrid hosted/packaged apps. The one exception is access to icons, since
+    // some extensions want to be able to do things like create their own
+    // launchers.
+    std::string resource_root_relative_path =
+        resource_url.path().empty() ? "" : resource_url.path().substr(1);
+    if (extension->is_hosted_app() &&
+        !extension->icons().ContainsPath(resource_root_relative_path)) {
+      LOG(ERROR) << "Denying load of " << resource_url.spec() << " from "
+        "hosted app.";
+      return false;
+    }
+
+    // Disallow loading of extension resources that are not explicitely listed
+    // as web accessible if the manifest version is 2 or greater.
+
+    GURL frame_url = frame->document().url();
+    GURL page_url = frame->top()->document().url();
+    // Exceptions are:
+    // - empty origin (needed for some edge cases when we have empty origins)
+    // - chrome-extension:// (for legacy reasons - some extensions interoperate)
+    // - devtools (chrome-extension:// URLs are loaded into frames of devtools
+    //     to support the devtools extension APIs)
+    if (!CommandLine::ForCurrentProcess()->HasSwitch(
+            switches::kDisableExtensionsResourceWhitelist) &&
+        !frame_url.is_empty() &&
+        !frame_url.SchemeIs(chrome::kExtensionScheme) &&
+        !(page_url.SchemeIs(chrome::kChromeDevToolsScheme) &&
+            !extension->devtools_url().is_empty()) &&
+        !extension->IsResourceWebAccessible(resource_url.path())) {
+      std::string message = base::StringPrintf(
+          "Denying load of %s. Resources must be listed in the "
+          "web_accessible_resources manifest key in order to be loaded by web "
+          "pages.",
+          resource_url.spec().c_str());
+      frame->addMessageToConsole(
+          WebKit::WebConsoleMessage(WebKit::WebConsoleMessage::LevelError,
+                                    WebKit::WebString::fromUTF8(message)));
+      return false;
+    }
+
     return true;
   }
 
-  // Disallow loading of packaged resources for hosted apps. We don't allow
-  // hybrid hosted/packaged apps. The one exception is access to icons, since
-  // some extensions want to be able to do things like create their own
-  // launchers.
-  std::string resource_root_relative_path =
-      resource_url.path().empty() ? "" : resource_url.path().substr(1);
-  if (extension->is_hosted_app() &&
-      !extension->icons().ContainsPath(resource_root_relative_path)) {
-    LOG(ERROR) << "Denying load of " << resource_url.spec() << " from "
-               << "hosted app.";
-    return false;
-  }
-
-  // Disallow loading of extension resources which are not explicitely listed
-  // as web accessible if the manifest version is 2 or greater.
-
-  GURL frame_url = frame->document().url();
-  GURL page_url = frame->top()->document().url();
-  // Exceptions are:
-  // - empty origin (needed for some edge cases when we have empty origins)
-  // - chrome-extension:// (for legacy reasons -- some extensions interop)
-  // - devtools (chrome-extension:// URLs are loaded into frames of devtools
-  //     to support the devtools extension APIs)
-  if (!CommandLine::ForCurrentProcess()->HasSwitch(
-          switches::kDisableExtensionsResourceWhitelist) &&
-      !frame_url.is_empty() &&
-      !frame_url.SchemeIs(chrome::kExtensionScheme) &&
-      !(page_url.SchemeIs(chrome::kChromeDevToolsScheme) &&
-          !extension->devtools_url().is_empty()) &&
-      !extension->IsResourceWebAccessible(resource_url.path())) {
-    std::string message = base::StringPrintf(
-        "Denying load of %s. Resources must be listed in the "
-        "web_accessible_resources manifest key in order to be loaded by web "
-        "pages.",
+  if (resource_url.SchemeIs(chrome::kExtensionResourceScheme)) {
+    GURL frame_url = frame->document().url();
+    GURL page_url = frame->top()->document().url();

[abarth-chromium, 2012/04/13 17:11:24]

It looks like the page_url variable is unused.  Perhaps remove?

[Peng, 2012/04/17 13:52:05]

Done.

+    if (!frame_url.is_empty() &&
+        !frame_url.SchemeIs(chrome::kExtensionScheme) &&
+        !frame_url.SchemeIs(chrome::kExtensionResourceScheme)) {
+      std::string message = base::StringPrintf(
+          "Denying load of %s. chrome-extension-resources:// can only be "
+          "loaded from extensions.",
         resource_url.spec().c_str());
-    frame->addMessageToConsole(
-        WebKit::WebConsoleMessage(WebKit::WebConsoleMessage::LevelError,
-                                  WebKit::WebString::fromUTF8(message)));
-    return false;
+      frame->addMessageToConsole(
+          WebKit::WebConsoleMessage(WebKit::WebConsoleMessage::LevelError,
+                                    WebKit::WebString::fromUTF8(message)));
+      return false;
+    }
+    return true;
   }
 
   return true;
 }
 
 ExtensionResourceRequestPolicy::ExtensionResourceRequestPolicy() {
 }