chrome/browser/extensions/extension_resource_protocols.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 "chrome/browser/extensions/extension_resource_protocols.h"
   6
   7 #include "base/files/file_path.h"
   8 #include "base/memory/weak_ptr.h"
   9 #include "base/path_service.h"
  10 #include "base/threading/sequenced_worker_pool.h"
  11 #include "base/threading/thread_checker.h"
  12 #include "chrome/common/chrome_paths.h"
  13 #include "content/public/browser/browser_thread.h"
  14 #include "extensions/common/file_util.h"
  15 #include "net/url_request/url_request_file_job.h"
  16
  17 namespace {
  18
  19 base::FilePath ResolvePath(const GURL& url) {
  20   base::FilePath root_path;
  21   PathService::Get(chrome::DIR_RESOURCES_EXTENSION, &root_path);
  22   return extensions::file_util::ExtensionResourceURLToFilePath(url, root_path);
  23 }
  24
  25 class ExtensionResourcesJob : public net::URLRequestFileJob {
  26  public:
  27   ExtensionResourcesJob(net::URLRequest* request,
  28                         net::NetworkDelegate* network_delegate)
  29       : net::URLRequestFileJob(
  30             request, network_delegate, base::FilePath(),
  31             content::BrowserThread::GetBlockingPool()->
  32                 GetTaskRunnerWithShutdownBehavior(
  33                     base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)),
  34         weak_ptr_factory_(this) {}
  35
  36   virtual void Start() OVERRIDE;
  37
  38  protected:
  39   virtual ~ExtensionResourcesJob() {}
  40
  41   void ResolvePathDone(const base::FilePath& resolved_path);
  42
  43  private:
  44   base::WeakPtrFactory<ExtensionResourcesJob> weak_ptr_factory_;
  45
  46   base::ThreadChecker thread_checker_;
  47
  48   DISALLOW_COPY_AND_ASSIGN(ExtensionResourcesJob);
  49 };
  50
  51 void ExtensionResourcesJob::Start() {
  52   DCHECK(thread_checker_.CalledOnValidThread());
  53   content::BrowserThread::PostTaskAndReplyWithResult(
  54       content::BrowserThread::FILE, FROM_HERE,
  55       base::Bind(&ResolvePath, request()->url()),
  56       base::Bind(&ExtensionResourcesJob::ResolvePathDone,
  57           weak_ptr_factory_.GetWeakPtr()));
  58 }
  59
  60 void ExtensionResourcesJob::ResolvePathDone(
  61     const base::FilePath& resolved_path) {
  62   DCHECK(thread_checker_.CalledOnValidThread());
  63   file_path_ = resolved_path;
  64   net::URLRequestFileJob::Start();
  65 }
  66
  67 class ExtensionResourceProtocolHandler
  68     : public net::URLRequestJobFactory::ProtocolHandler {
  69  public:
  70   ExtensionResourceProtocolHandler() {}
  71   virtual ~ExtensionResourceProtocolHandler() {}
  72
  73   virtual net::URLRequestJob* MaybeCreateJob(
  74       net::URLRequest* request,
  75       net::NetworkDelegate* network_delegate) const OVERRIDE;
  76
  77  private:
  78   DISALLOW_COPY_AND_ASSIGN(ExtensionResourceProtocolHandler);
  79 };
  80
  81 // Creates URLRequestJobs for chrome-extension-resource:// URLs.
  82 net::URLRequestJob*
  83 ExtensionResourceProtocolHandler::MaybeCreateJob(
  84     net::URLRequest* request, net::NetworkDelegate* network_delegate) const {
  85   return new ExtensionResourcesJob(request, network_delegate);
  86 }
  87
  88 }  // namespace
  89
  90 net::URLRequestJobFactory::ProtocolHandler*
  91 CreateExtensionResourceProtocolHandler() {
  92   return new ExtensionResourceProtocolHandler();
  93 }