Add reverse URL handler for shortening uber URLs

chrome/browser/chrome_content_browser_client.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/browser/chrome_content_browser_client.h"
 
 #include <set>
 #include <utility>
 #include <vector>
 
@@ skipping 142 matching lines @@
   "hhnbmknkdabfoieppbbljkhkfjcmcbjh",  // see crbug.com/134099
   "mablfbjkhmhkmefkjjacnbaikjkipphg",  // see crbug.com/134099
   "pdeelgamlgannhelgoegilelnnojegoh",  // see crbug.com/134099
   "cabapfdbkniadpollkckdnedaanlciaj",  // see crbug.com/134099
   "mapljbgnjledlpdmlchihnmeclmefbba",  // see crbug.com/134099
   "ghbfeebgmiidnnmeobbbaiamklmpbpii"   // see crbug.com/134099
 };
 
 // Handles rewriting Web UI URLs.
 bool HandleWebUI(GURL* url, content::BrowserContext* browser_context) {
+  url_canon::Replacements<char> replacements;
+  replacements.SetHost("chrome", url_parse::Component(0, 6));

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Nit: Make a const std::string in the local scope and then use its .length() for
the second arg to Component().

+  std::string host = url->host();

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Nit: Should be const.

+  replacements.SetPath(host.c_str(), url_parse::Component(0, host.length()));
+  GURL chrome_url = url->ReplaceComponents(replacements);

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Nit: Maybe extract the above block to a separate function for better
readability.

[Cem Kocagil, 2012/08/07 20:56:22]

Good idea. I went further and added two functions: AddUberHost and
RemoveUberHost.

+
+  // Handle valid "chrome://chrome/foo" URLs so the reverse handler will
+  // be called.
+  if (ChromeWebUIControllerFactory::GetInstance()->UseWebUIForURL(
+          browser_context, chrome_url))
+    return true;

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Add a comment why this doesn't need to set *url even though it returns true.

[Cem Kocagil, 2012/08/07 20:56:22]

The comment right above that block explains why it returns true (so the reverse
handler will be called).

+
   if (!ChromeWebUIControllerFactory::GetInstance()->UseWebUIForURL(
           browser_context, *url))
     return false;
 
 #if defined(OS_CHROMEOS)
   // Special case : in ChromeOS in Guest mode bookmarks and history are
   // disabled for security reasons. New tab page explains the reasons, so
   // we redirect user to new tab page.
   if (chromeos::UserManager::Get()->IsLoggedInAsGuest()) {
     if (url->SchemeIs(chrome::kChromeUIScheme) &&
@@ skipping 10 matching lines @@
   // sessions or bookmarks, so we say any URL with that scheme triggers the new
   // tab page.
   if (url->SchemeIs(chrome::kChromeInternalScheme)) {
     // Rewrite it with the proper new tab URL.
     *url = GURL(chrome::kChromeUINewTabURL);
   }
 
   return true;
 }
 
+// Reverse URL handler for Web UI. Maps "chrome://chrome/foo/" to
+// "chrome://foo/".
+bool HandleWebUIReverse(GURL* url, content::BrowserContext* browser_context) {
+  if (!url->is_valid() || !url->SchemeIs(chrome::kChromeUIScheme) ||
+      url->host() != chrome::kChromeUIUberHost)
+    return false;
+
+  std::string old_path = url->path();

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Nit: Should be const.

+
+  int separator = old_path.find('/', 1);

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Nit: Should be const.

+  std::string new_host;
+  std::string new_path;
+  if (separator == std::string::npos) {
+    new_host = old_path.substr(1);

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

What if old_path is an empty string? (Also, does find('/', 1) work in that case?

[Cem Kocagil, 2012/08/07 20:56:22]

It should work when old_path is empty. I checked the C++ spec for the behavior
of string::find where pos >= length and it says it should return npos.

+  } else {
+    new_host = old_path.substr(1, separator - 1);
+    new_path = old_path.substr(separator);
+  }
+
+  url_canon::Replacements<char> replacements;
+  replacements.ClearHost();

[Alexei Svitkine (slow), 2012/08/07 14:59:33]

Is this needed given that you set it immediately after?

+  replacements.SetHost(new_host.c_str(),
+      url_parse::Component(0, new_host.length()));
+  replacements.SetPath(new_path.c_str(),
+      url_parse::Component(0, new_path.length()));
+  *url = url->ReplaceComponents(replacements);
+  return true;
+}
+
 // Used by the GetPrivilegeRequiredByUrl() and GetProcessPrivilege() functions
 // below.  Extension, and isolated apps require different privileges to be
 // granted to their RenderProcessHosts.  This classification allows us to make
 // sure URLs are served by hosts with the right set of privileges.
 enum RenderProcessHostPrivilege {
   PRIV_NORMAL,
   PRIV_HOSTED,
   PRIV_ISOLATED,
   PRIV_EXTENSION,
 };
@@ skipping 1329 matching lines @@
   handler->AddHandlerPair(BrowserURLHandler::null_handler(),
                           &ExtensionWebUI::HandleChromeURLOverrideReverse);
 
   // about: handler. Must come before chrome: handler, since it will
   // rewrite about: urls to chrome: URLs and then expect chrome: to
   // actually handle them.
   handler->AddHandlerPair(&WillHandleBrowserAboutURL,
                           BrowserURLHandler::null_handler());
   // chrome: & friends.
   handler->AddHandlerPair(&HandleWebUI,
-                          BrowserURLHandler::null_handler());
+                          &HandleWebUIReverse);
 }
 
 void ChromeContentBrowserClient::ClearCache(RenderViewHost* rvh) {
   Profile* profile = Profile::FromBrowserContext(
       rvh->GetSiteInstance()->GetProcess()->GetBrowserContext());
   BrowsingDataRemover* remover = new BrowsingDataRemover(profile,
       BrowsingDataRemover::EVERYTHING,
       base::Time());
   remover->Remove(BrowsingDataRemover::REMOVE_CACHE,
                   BrowsingDataHelper::UNPROTECTED_WEB);
@@ skipping 113 matching lines @@
     io_thread_application_locale_ = locale;
 }
 
 void ChromeContentBrowserClient::SetApplicationLocaleOnIOThread(
     const std::string& locale) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   io_thread_application_locale_ = locale;
 }
 
 }  // namespace chrome