Change Interceptors into URLRequestJobFactory::ProtocolHandlers.

chrome/browser/extensions/autoupdate_interceptor.cc

 // 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/extensions/autoupdate_interceptor.h"
 
 #include "base/bind.h"
 #include "base/file_util.h"
 #include "base/threading/thread_restrictions.h"
 #include "content/public/browser/browser_thread.h"
 #include "net/url_request/url_request.h"
+#include "net/url_request/url_request_filter.h"

[erikwright (departed), 2012/12/06 15:26:28]

It might be worth it to take this opportunity to de-duplicate this class and the
component_updater one.

I have a feeling you'll be back here later, so by de-duplicating now you might
save some work.

[pauljensen, 2012/12/07 18:47:42]

Done.

 #include "net/url_request/url_request_test_job.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using content::BrowserThread;
 
 namespace extensions {
 
 // This is a specialized version of net::URLRequestTestJob that lets us specify
 // response data and make sure the response code is 200, which the autoupdate
 // code relies on.
 class AutoUpdateTestRequestJob : public net::URLRequestTestJob {
  public:
   AutoUpdateTestRequestJob(net::URLRequest* request,
                            net::NetworkDelegate* network_delegate,
                            const std::string& response_data)
       : net::URLRequestTestJob(request,
                                network_delegate,
                                net::URLRequestTestJob::test_headers(),
                                response_data,
                                true) {
   }
 
   virtual int GetResponseCode() const { return 200; }
 
  private:
   ~AutoUpdateTestRequestJob() {}
 };
 
+class AutoUpdateInterceptor::Delegate
+    : public net::URLRequestJobFactory::ProtocolHandler {
+ public:
+  Delegate() {}
+  virtual ~Delegate() {
+    EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    net::URLRequestFilter::GetInstance()->RemoveHostnameHandler("http",
+                                                                "localhost");
+  }
+  void Register() {
+    EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    net::URLRequestFilter::GetInstance()->AddHostnameProtocolHandler(
+        "http", "localhost", this);
+  }
 
-AutoUpdateInterceptor::AutoUpdateInterceptor() {
-  net::URLRequest::Deprecated::RegisterRequestInterceptor(this);
+  // When computing matches, this ignores query parameters (since the autoupdate
+  // fetch code appends a bunch of them to manifest fetches).
+  virtual net::URLRequestJob* MaybeCreateJob(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate) const OVERRIDE {
+    EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    if (request->url().scheme() != "http" ||
+        request->url().host() != "localhost") {
+      return NULL;
+    }
+
+    // It's ok to do a blocking disk access on this thread; this class
+    // is just used for tests.
+    base::ThreadRestrictions::ScopedAllowIO allow_io;
+
+    // Search for this request's url, ignoring any query parameters.
+    GURL url = request->url();
+    if (url.has_query()) {
+      GURL::Replacements replacements;
+      replacements.ClearQuery();
+      url = url.ReplaceComponents(replacements);
+    }
+    std::map<GURL, FilePath>::const_iterator i = responses_.find(url);
+    if (i == responses_.end()) {
+      return NULL;
+    }
+    std::string contents;
+    EXPECT_TRUE(file_util::ReadFileToString(i->second, &contents));
+
+    return new AutoUpdateTestRequestJob(request, network_delegate, contents);
+  }
+
+  // When requests for |url| arrive, respond with the contents of |path|. The
+  // hostname of |url| must be "localhost" to avoid DNS lookups, and the scheme
+  // must be "http" so MaybeIntercept can ignore "chrome" and other schemes.
+  // Also, the match for |url| will ignore any query parameters.
+  void SetResponse(const std::string url, const FilePath& path) {
+    EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    // It's ok to do a blocking disk access on this thread; this class
+    // is just used for tests.
+    base::ThreadRestrictions::ScopedAllowIO allow_io;
+    GURL gurl(url);
+    EXPECT_EQ("http", gurl.scheme());
+    EXPECT_EQ("localhost", gurl.host());
+    EXPECT_TRUE(file_util::PathExists(path));
+    responses_[gurl] = path;
+  }
+
+ private:
+  std::map<GURL, FilePath> responses_;
+
+  DISALLOW_COPY_AND_ASSIGN(Delegate);
+};
+
+AutoUpdateInterceptor::AutoUpdateInterceptor() : delegate_(new Delegate) {
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                          base::Bind(&Delegate::Register,
+                                     base::Unretained(delegate_)));
 }
 
 AutoUpdateInterceptor::~AutoUpdateInterceptor() {
-  net::URLRequest::Deprecated::UnregisterRequestInterceptor(this);
+  BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, delegate_);
 }
 
-net::URLRequestJob* AutoUpdateInterceptor::MaybeIntercept(
-    net::URLRequest* request, net::NetworkDelegate* network_delegate) {
-  EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  if (request->url().scheme() != "http" ||
-      request->url().host() != "localhost") {
-      return NULL;
-  }
-
-  // It's ok to do a blocking disk access on this thread; this class
-  // is just used for tests.
-  base::ThreadRestrictions::ScopedAllowIO allow_io;
-
-  // Search for this request's url, ignoring any query parameters.
-  GURL url = request->url();
-  if (url.has_query()) {
-    GURL::Replacements replacements;
-    replacements.ClearQuery();
-    url = url.ReplaceComponents(replacements);
-  }
-  std::map<GURL, FilePath>::iterator i = responses_.find(url);
-  if (i == responses_.end()) {
-    return NULL;
-  }
-  std::string contents;
-  EXPECT_TRUE(file_util::ReadFileToString(i->second, &contents));
-
-  return new AutoUpdateTestRequestJob(request, network_delegate, contents);
-}
-
-
 void AutoUpdateInterceptor::SetResponse(const std::string url,
                                         const FilePath& path) {
-  EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  // It's ok to do a blocking disk access on this thread; this class
-  // is just used for tests.
-  base::ThreadRestrictions::ScopedAllowIO allow_io;
-  GURL gurl(url);
-  EXPECT_EQ("http", gurl.scheme());
-  EXPECT_EQ("localhost", gurl.host());
-  EXPECT_TRUE(file_util::PathExists(path));
-  responses_[gurl] = path;
-}
-
-
-void AutoUpdateInterceptor::SetResponseOnIOThread(const std::string url,
-                                                  const FilePath& path) {
-  BrowserThread::PostTask(
-      BrowserThread::IO, FROM_HERE,
-      base::Bind(&AutoUpdateInterceptor::SetResponse, this, url, path));
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                          base::Bind(&Delegate::SetResponse,
+                                     base::Unretained(delegate_), url, path));
 }
 
 }  // namespace extensions