Adds navigator.registerProtocolHandler.

chrome/browser/custom_handlers/protocol_handler_registry.cc

Index: chrome/browser/custom_handlers/protocol_handler_registry.cc
diff --git a/chrome/browser/custom_handlers/protocol_handler_registry.cc b/chrome/browser/custom_handlers/protocol_handler_registry.cc
new file mode 100644
index 0000000000000000000000000000000000000000..1c20ffd690af3dba8699bfa77bcbdd030c61ef22
--- /dev/null
+++ b/chrome/browser/custom_handlers/protocol_handler_registry.cc
@@ -0,0 +1,167 @@
+// Copyright (c) 2011 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/custom_handlers/protocol_handler_registry.h"
+
+#include "base/scoped_ptr.h"
+#include "base/string_util.h"
+#include "chrome/browser/child_process_security_policy.h"
+#include "chrome/browser/custom_handlers/register_protocol_handler_infobar_delegate.h"
+#include "chrome/browser/net/chrome_url_request_context.h"
+#include "chrome/common/pref_names.h"
+#include "net/base/escape.h"
+#include "net/url_request/url_request_redirect_job.h"
+
+// ProtocolHandler -------------------------------------------------------------
+
+ProtocolHandler::ProtocolHandler(const std::string& protocol,
+                                 const GURL& url,
+                                 const string16& title)
+  :protocol_(protocol),
+   url_(url),
+   title_(title) {
+}
+
+ProtocolHandler* ProtocolHandler::CreateProtocolHandler(
+    const std::string& protocol,
+    const GURL& url,
+    const string16& title) {
+  // TODO(koz) Check for bad protocol and url here.

[tony, 2011/02/14 23:40:07]

We should probably make sure we have this check before .registerProtocolHandler
is available on the dev channel.  I'm worried about users who will get junk in
their prefs and once checks are added, it'll be hard to fix.

Alternately, this feature might want to start out behind a command line flag
until it's a bit more mature.  If you put it behind a command line flag, try and
make sure window.registerProtocolHandler is undefined without the flag.

[koz (OOO until 15th September), 2011/02/15 03:37:27]

I talked with Ojan earlier about validating input and he said that it is
desirable to do all our sanitising in WebKit itself so that all ports benefit
from it. There are some checks in WebKit currently, but there are a few more
checks I would like to add, such as ensuring that only valid characters appear
in the scheme (which I'll have to research). I can't think of any checks we
would want to do for the handler URL templates (though I'd appreciate any
thoughts you have :-) ).

Currently this feature is turned on and off by flipping WebKit flags in
features.gypi and features_override.gypi, which controls whether or not the
navigator.registerProtocolHandler stub is defined. Is it possible to put
something like that behind a command line flag?

+  std::string lower_protocol(protocol);
+  lower_protocol = StringToLowerASCII(protocol);
+  return new ProtocolHandler(lower_protocol, url, title);
+}
+
+ProtocolHandler* ProtocolHandler::CreateProtocolHandler(const DictionaryValue* value) {

[tony, 2011/02/14 23:40:07]

80 cols here and in lots of other places.  Please run the cpplinter.

[koz (OOO until 15th September), 2011/02/15 03:37:27]

Done.

+  std::string protocol, url;
+  string16 title;
+  value->GetString("protocol", &protocol);
+  value->GetString("url", &url);
+  value->GetString("title", &title);
+  return ProtocolHandler::CreateProtocolHandler(protocol, GURL(url), title);
+}
+
+GURL ProtocolHandler::TranslateUrl(const GURL& url) {
+  std::string translatedUrlSpec(url_.spec());
+  ReplaceSubstringsAfterOffset(&translatedUrlSpec, 0, "%s",
+      EscapeQueryParamValue(url.spec(), true));
+  return GURL(translatedUrlSpec);
+}
+
+Value* ProtocolHandler::Encode() {
+  DictionaryValue* d = new DictionaryValue();
+  d->Set("protocol", Value::CreateStringValue(protocol_));
+  d->Set("url", Value::CreateStringValue(url_.spec()));
+  d->Set("title", Value::CreateStringValue(title_));
+  return d;
+}
+
+bool ProtocolHandler::operator==(const ProtocolHandler &other) const {
+  return protocol_ == other.protocol_ &&
+    url_ == other.url_ &&
+    title_ == other.title_;
+}
+
+// ProtocolHandlerRegistry -------------------------------------------------------------
+
+ProtocolHandlerRegistry::ProtocolHandlerRegistry(Profile* profile)
+  :profile_(profile) {
+}
+
+void ProtocolHandlerRegistry::RegisterProtocolHandler(ProtocolHandler* handler) {
+  if (protocolHandlers_.find(handler->protocol()) == protocolHandlers_.end()) {
+    ChildProcessSecurityPolicy* policy = ChildProcessSecurityPolicy::GetInstance();
+    if (!policy->IsWebSafeScheme(handler->protocol())) {
+      policy->RegisterWebSafeScheme(handler->protocol());
+    }
+    net::URLRequest::RegisterProtocolFactory(handler->protocol(),
+        &ProtocolHandlerRegistry::Factory);
+  }
+  protocolHandlers_[handler->protocol()] = handler;
+}
+
+void ProtocolHandlerRegistry::Load() {
+  PrefService* prefs = profile_->GetPrefs();
+  if (!prefs->HasPrefPath(prefs::kRegisteredProtocolHandlers)) {
+    return;
+  }
+  const ListValue* protocolHandlers = prefs->GetList(prefs::kRegisteredProtocolHandlers);
+
+  for (size_t i = 0; i < protocolHandlers->GetSize(); i++) {
+    DictionaryValue* dict;
+    protocolHandlers->GetDictionary(i, &dict);
+    RegisterHandlerFromValue(dict);
+  }
+}
+
+void ProtocolHandlerRegistry::RegisterHandlerFromValue(const DictionaryValue* value) {
+  ProtocolHandler* handler = ProtocolHandler::CreateProtocolHandler(value);
+  if (handler) {
+    RegisterProtocolHandler(handler);
+  } else {
+  }

[tony, 2011/02/14 23:40:07]

Remove empty else?

[koz (OOO until 15th September), 2011/02/15 03:37:27]

Done.

+}
+
+void ProtocolHandlerRegistry::Save() {
+  scoped_ptr<Value> value(Encode());
+  profile_->GetPrefs()->Set(prefs::kRegisteredProtocolHandlers, *value);
+  profile_->GetPrefs()->ScheduleSavePersistentPrefs();
+}
+
+ProtocolHandler* ProtocolHandlerRegistry::GetHandlerFor(const std::string& scheme) const {
+  ProtocolHandlerMap::const_iterator i = protocolHandlers_.find(scheme);
+  return i == protocolHandlers_.end() ? NULL : i->second;
+}
+
+bool ProtocolHandlerRegistry::IsAlreadyRegistered(const ProtocolHandler* handler) const {
+  ProtocolHandler* currentHandler = GetHandlerFor(handler->protocol());
+  return currentHandler && *currentHandler == *handler;
+}
+
+net::URLRequestJob* ProtocolHandlerRegistry::CreateJob(net::URLRequest* request,
+                                                       const std::string& scheme) const {
+  ProtocolHandler* handler = GetHandlerFor(scheme);
+
+  if (!handler) {
+    return NULL;
+  }
+
+  GURL translatedUrl(handler->TranslateUrl(request->url()));
+
+  if (!translatedUrl.is_valid()) {
+    return NULL;
+  }
+
+  return new net::URLRequestRedirectJob(request, translatedUrl);
+}
+
+net::URLRequestJob* ProtocolHandlerRegistry::Factory(net::URLRequest* request,
+                                                const std::string& scheme) {
+  ChromeURLRequestContext* context =
+    reinterpret_cast<ChromeURLRequestContext*>(request->context());

[tony, 2011/02/14 23:40:07]

I think this should be a static_cast.

[koz (OOO until 15th September), 2011/02/15 03:37:27]

Done.

+  return context->protocol_handler_registry()->CreateJob(request, scheme);
+}
+
+Value* ProtocolHandlerRegistry::Encode() {
+  ListValue* protocolHandlers = new ListValue();
+
+  for (ProtocolHandlerMap::iterator i = protocolHandlers_.begin();
+      i != protocolHandlers_.end(); ++i) {
+    protocolHandlers->Append(i->second->Encode());
+  }
+  return protocolHandlers;
+}
+
+void ProtocolHandlerRegistry::OnAcceptRegisterProtocolHandler(ProtocolHandler* handler) {
+  RegisterProtocolHandler(handler);
+  Save();
+}
+
+void ProtocolHandlerRegistry::OnDenyRegisterProtocolHandler(ProtocolHandler* handler) {
+
+}
+
+void ProtocolHandlerRegistry::RegisterPrefs(PrefService* prefService) {
+  prefService->RegisterListPref(prefs::kRegisteredProtocolHandlers);
+}