content/browser/browser_url_handler_impl.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/browser_url_handler_impl.h"
   6
   7 #include "base/string_util.h"
   8 #include "content/browser/webui/web_ui_impl.h"
   9 #include "content/public/browser/content_browser_client.h"
  10 #include "content/public/common/url_constants.h"
  11 #include "googleurl/src/gurl.h"
  12
  13 namespace content {
  14
  15 // Handles rewriting view-source URLs for what we'll actually load.
  16 static bool HandleViewSource(GURL* url,
  17                              BrowserContext* browser_context) {
  18   if (url->SchemeIs(chrome::kViewSourceScheme)) {
  19     // Load the inner URL instead.
  20     *url = GURL(url->path());
  21
  22     // Bug 26129: limit view-source to view the content and not any
  23     // other kind of 'active' url scheme like 'javascript' or 'data'.
  24     static const char* const allowed_sub_schemes[] = {
  25       chrome::kHttpScheme, chrome::kHttpsScheme, chrome::kFtpScheme,
  26       chrome::kChromeDevToolsScheme, chrome::kChromeUIScheme,
  27       chrome::kFileScheme, chrome::kFileSystemScheme
  28     };
  29
  30     bool is_sub_scheme_allowed = false;
  31     for (size_t i = 0; i < arraysize(allowed_sub_schemes); i++) {
  32       if (url->SchemeIs(allowed_sub_schemes[i])) {
  33         is_sub_scheme_allowed = true;
  34         break;
  35       }
  36     }
  37
  38     if (!is_sub_scheme_allowed) {
  39       *url = GURL(chrome::kAboutBlankURL);
  40       return false;
  41     }
  42
  43     return true;
  44   }
  45   return false;
  46 }
  47
  48 // Turns a non view-source URL into the corresponding view-source URL.
  49 static bool ReverseViewSource(GURL* url, BrowserContext* browser_context) {
  50   // No action necessary if the URL is already view-source:
  51   if (url->SchemeIs(chrome::kViewSourceScheme))
  52     return false;
  53
  54   url_canon::Replacements<char> repl;
  55   repl.SetScheme(chrome::kViewSourceScheme,
  56       url_parse::Component(0, strlen(chrome::kViewSourceScheme)));
  57   repl.SetPath(url->spec().c_str(),
  58       url_parse::Component(0, url->spec().size()));
  59   *url = url->ReplaceComponents(repl);
  60   return true;
  61 }
  62
  63 static bool HandleDebugUrl(GURL* url, BrowserContext* browser_context) {
  64   // Circumvent processing URLs that the renderer process will handle.
  65   return *url == GURL(chrome::kChromeUICrashURL) ||
  66          *url == GURL(chrome::kChromeUIHangURL) ||
  67          *url == GURL(chrome::kChromeUIKillURL) ||
  68          *url == GURL(kChromeUIShorthangURL);
  69 }
  70
  71 // static
  72 BrowserURLHandler* BrowserURLHandler::GetInstance() {
  73   return BrowserURLHandlerImpl::GetInstance();
  74 }
  75
  76 // static
  77 BrowserURLHandler::URLHandler BrowserURLHandler::null_handler() {
  78   // Required for VS2010: http://connect.microsoft.com/VisualStudio/feedback/details/520043/error-converting-from-null-to-a-pointer-type-in-std-pair
  79   return NULL;
  80 }
  81
  82 // static
  83 BrowserURLHandlerImpl* BrowserURLHandlerImpl::GetInstance() {
  84   return Singleton<BrowserURLHandlerImpl>::get();
  85 }
  86
  87 BrowserURLHandlerImpl::BrowserURLHandlerImpl() {
  88   AddHandlerPair(&HandleDebugUrl, BrowserURLHandlerImpl::null_handler());
  89
  90   GetContentClient()->browser()->BrowserURLHandlerCreated(this);
  91
  92   // view-source:
  93   AddHandlerPair(&HandleViewSource, &ReverseViewSource);
  94 }
  95
  96 BrowserURLHandlerImpl::~BrowserURLHandlerImpl() {
  97 }
  98
  99 void BrowserURLHandlerImpl::AddHandlerPair(URLHandler handler,
 100                                            URLHandler reverse_handler) {
 101   url_handlers_.push_back(HandlerPair(handler, reverse_handler));
 102 }
 103
 104 void BrowserURLHandlerImpl::RewriteURLIfNecessary(
 105     GURL* url,
 106     BrowserContext* browser_context,
 107     bool* reverse_on_redirect) {
 108   for (size_t i = 0; i < url_handlers_.size(); ++i) {
 109     URLHandler handler = *url_handlers_[i].first;
 110     if (handler && handler(url, browser_context)) {
 111       *reverse_on_redirect = (url_handlers_[i].second != NULL);
 112       return;
 113     }
 114   }
 115 }
 116
 117 bool BrowserURLHandlerImpl::ReverseURLRewrite(
 118     GURL* url, const GURL& original, BrowserContext* browser_context) {
 119   for (size_t i = 0; i < url_handlers_.size(); ++i) {
 120     URLHandler reverse_rewriter = *url_handlers_[i].second;
 121     if (reverse_rewriter) {
 122       GURL test_url(original);
 123       URLHandler handler = *url_handlers_[i].first;
 124       if (!handler) {
 125         if (reverse_rewriter(url, browser_context))
 126           return true;
 127       } else if (handler(&test_url, browser_context)) {
 128         return reverse_rewriter(url, browser_context);
 129       }
 130     }
 131   }
 132   return false;
 133 }
 134
 135 }  // namespace content