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/command_line.h" 8 #include "base/strings/string_util.h" 9 #include "cc/base/switches.h" 10 #include "content/browser/frame_host/debug_urls.h" 11 #include "content/browser/webui/web_ui_impl.h" 12 #include "content/public/browser/content_browser_client.h" 13 #include "content/public/common/url_constants.h" 14 #include "url/gurl.h" 15 16 namespace content { 17 18 // Handles rewriting view-source URLs for what we'll actually load. 19 static bool HandleViewSource(GURL* url, BrowserContext* browser_context) { 20 if (url->SchemeIs(kViewSourceScheme)) { 21 // Load the inner URL instead. 22 *url = GURL(url->GetContent()); 23 24 // Bug 26129: limit view-source to view the content and not any 25 // other kind of 'active' url scheme like 'javascript' or 'data'. 26 static const char* const default_allowed_sub_schemes[] = { 27 url::kHttpScheme, 28 url::kHttpsScheme, 29 url::kFtpScheme, 30 kChromeDevToolsScheme, 31 kChromeUIScheme, 32 url::kFileScheme, 33 url::kFileSystemScheme 34 }; 35 36 // Merge all the schemes for which view-source is allowed by default, with 37 // the WebUI schemes defined by the ContentBrowserClient. 38 std::vector<std::string> all_allowed_sub_schemes; 39 for (size_t i = 0; i < arraysize(default_allowed_sub_schemes); ++i) 40 all_allowed_sub_schemes.push_back(default_allowed_sub_schemes[i]); 41 GetContentClient()->browser()->GetAdditionalWebUISchemes( 42 &all_allowed_sub_schemes); 43 44 bool is_sub_scheme_allowed = false; 45 for (size_t i = 0; i < all_allowed_sub_schemes.size(); ++i) { 46 if (url->SchemeIs(all_allowed_sub_schemes[i].c_str())) { 47 is_sub_scheme_allowed = true; 48 break; 49 } 50 } 51 52 if (!is_sub_scheme_allowed) { 53 *url = GURL(url::kAboutBlankURL); 54 return false; 55 } 56 57 return true; 58 } 59 return false; 60 } 61 62 // Turns a non view-source URL into the corresponding view-source URL. 63 static bool ReverseViewSource(GURL* url, BrowserContext* browser_context) { 64 // No action necessary if the URL is already view-source: 65 if (url->SchemeIs(kViewSourceScheme)) 66 return false; 67 68 url::Replacements<char> repl; 69 repl.SetScheme(kViewSourceScheme, 70 url::Component(0, strlen(kViewSourceScheme))); 71 repl.SetPath(url->spec().c_str(), url::Component(0, url->spec().size())); 72 *url = url->ReplaceComponents(repl); 73 return true; 74 } 75 76 static bool DebugURLHandler(GURL* url, BrowserContext* browser_context) { 77 // If running inside the Telemetry test harness, allow automated 78 // navigations to access browser-side debug URLs. They must use the 79 // chrome:// scheme, since the about: scheme won't be rewritten in 80 // this code path. 81 if (CommandLine::ForCurrentProcess()->HasSwitch( 82 cc::switches::kEnableGpuBenchmarking)) { 83 if (HandleDebugURL(*url, PAGE_TRANSITION_FROM_ADDRESS_BAR)) { 84 return true; 85 } 86 } 87 88 // Circumvent processing URLs that the renderer process will handle. 89 return IsRendererDebugURL(*url); 90 } 91 92 // static 93 BrowserURLHandler* BrowserURLHandler::GetInstance() { 94 return BrowserURLHandlerImpl::GetInstance(); 95 } 96 97 // static 98 BrowserURLHandler::URLHandler BrowserURLHandler::null_handler() { 99 // Required for VS2010: http://connect.microsoft.com/VisualStudio/feedback/details/520043/error-converting-from-null-to-a-pointer-type-in-std-pair 100 return NULL; 101 } 102 103 // static 104 BrowserURLHandlerImpl* BrowserURLHandlerImpl::GetInstance() { 105 return Singleton<BrowserURLHandlerImpl>::get(); 106 } 107 108 BrowserURLHandlerImpl::BrowserURLHandlerImpl() { 109 AddHandlerPair(&DebugURLHandler, BrowserURLHandlerImpl::null_handler()); 110 111 GetContentClient()->browser()->BrowserURLHandlerCreated(this); 112 113 // view-source: 114 AddHandlerPair(&HandleViewSource, &ReverseViewSource); 115 } 116 117 BrowserURLHandlerImpl::~BrowserURLHandlerImpl() { 118 } 119 120 void BrowserURLHandlerImpl::AddHandlerPair(URLHandler handler, 121 URLHandler reverse_handler) { 122 url_handlers_.push_back(HandlerPair(handler, reverse_handler)); 123 } 124 125 void BrowserURLHandlerImpl::RewriteURLIfNecessary( 126 GURL* url, 127 BrowserContext* browser_context, 128 bool* reverse_on_redirect) { 129 for (size_t i = 0; i < url_handlers_.size(); ++i) { 130 URLHandler handler = *url_handlers_[i].first; 131 if (handler && handler(url, browser_context)) { 132 *reverse_on_redirect = (url_handlers_[i].second != NULL); 133 return; 134 } 135 } 136 } 137 138 bool BrowserURLHandlerImpl::ReverseURLRewrite( 139 GURL* url, const GURL& original, BrowserContext* browser_context) { 140 for (size_t i = 0; i < url_handlers_.size(); ++i) { 141 URLHandler reverse_rewriter = *url_handlers_[i].second; 142 if (reverse_rewriter) { 143 GURL test_url(original); 144 URLHandler handler = *url_handlers_[i].first; 145 if (!handler) { 146 if (reverse_rewriter(url, browser_context)) 147 return true; 148 } else if (handler(&test_url, browser_context)) { 149 return reverse_rewriter(url, browser_context); 150 } 151 } 152 } 153 return false; 154 } 155 156 } // namespace content 157