src/chrome/browser/browsing_data/browsing_data_server_bound_cert_helper.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/browsing_data/browsing_data_server_bound_cert_helper.h"
   6
   7 #include "base/bind.h"
   8 #include "base/logging.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "chrome/browser/profiles/profile.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "net/ssl/server_bound_cert_service.h"
  14 #include "net/url_request/url_request_context.h"
  15 #include "net/url_request/url_request_context_getter.h"
  16
  17 namespace {
  18
  19 class BrowsingDataServerBoundCertHelperImpl
  20     : public BrowsingDataServerBoundCertHelper {
  21  public:
  22   explicit BrowsingDataServerBoundCertHelperImpl(Profile* profile);
  23
  24   // BrowsingDataServerBoundCertHelper methods.
  25   virtual void StartFetching(const FetchResultCallback& callback) OVERRIDE;
  26   virtual void DeleteServerBoundCert(const std::string& server_id) OVERRIDE;
  27
  28  private:
  29   virtual ~BrowsingDataServerBoundCertHelperImpl();
  30
  31   // Fetch the certs. This must be called in the IO thread.
  32   void FetchOnIOThread();
  33
  34   void OnFetchComplete(
  35       const net::ServerBoundCertStore::ServerBoundCertList& cert_list);
  36
  37   // Notifies the completion callback. This must be called in the UI thread.
  38   void NotifyInUIThread(
  39       const net::ServerBoundCertStore::ServerBoundCertList& cert_list);
  40
  41   // Delete a single cert. This must be called in IO thread.
  42   void DeleteOnIOThread(const std::string& server_id);
  43
  44   // Called when deletion is done.
  45   void DeleteCallback();
  46
  47   // Indicates whether or not we're currently fetching information:
  48   // it's true when StartFetching() is called in the UI thread, and it's reset
  49   // after we notify the callback in the UI thread.
  50   // This only mutates on the UI thread.
  51   bool is_fetching_;
  52
  53   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
  54
  55   // This only mutates on the UI thread.
  56   FetchResultCallback completion_callback_;
  57
  58   DISALLOW_COPY_AND_ASSIGN(BrowsingDataServerBoundCertHelperImpl);
  59 };
  60
  61 BrowsingDataServerBoundCertHelperImpl::
  62 BrowsingDataServerBoundCertHelperImpl(Profile* profile)
  63     : is_fetching_(false),
  64       request_context_getter_(profile->GetRequestContext()) {
  65   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
  66 }
  67
  68 BrowsingDataServerBoundCertHelperImpl::
  69 ~BrowsingDataServerBoundCertHelperImpl() {
  70 }
  71
  72 void BrowsingDataServerBoundCertHelperImpl::StartFetching(
  73     const FetchResultCallback& callback) {
  74   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
  75   DCHECK(!is_fetching_);
  76   DCHECK(!callback.is_null());
  77   DCHECK(completion_callback_.is_null());
  78   is_fetching_ = true;
  79   completion_callback_ = callback;
  80   content::BrowserThread::PostTask(
  81       content::BrowserThread::IO, FROM_HERE,
  82       base::Bind(&BrowsingDataServerBoundCertHelperImpl::FetchOnIOThread,
  83                  this));
  84 }
  85
  86 void BrowsingDataServerBoundCertHelperImpl::DeleteServerBoundCert(
  87     const std::string& server_id) {
  88   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
  89   content::BrowserThread::PostTask(
  90       content::BrowserThread::IO, FROM_HERE,
  91       base::Bind(&BrowsingDataServerBoundCertHelperImpl::DeleteOnIOThread,
  92                  this, server_id));
  93 }
  94
  95 void BrowsingDataServerBoundCertHelperImpl::FetchOnIOThread() {
  96   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
  97   net::ServerBoundCertStore* cert_store =
  98       request_context_getter_->GetURLRequestContext()->
  99       server_bound_cert_service()->GetCertStore();
 100   if (cert_store) {
 101     cert_store->GetAllServerBoundCerts(base::Bind(
 102         &BrowsingDataServerBoundCertHelperImpl::OnFetchComplete, this));
 103   } else {
 104     OnFetchComplete(net::ServerBoundCertStore::ServerBoundCertList());
 105   }
 106 }
 107
 108 void BrowsingDataServerBoundCertHelperImpl::OnFetchComplete(
 109     const net::ServerBoundCertStore::ServerBoundCertList& cert_list) {
 110   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
 111   content::BrowserThread::PostTask(
 112       content::BrowserThread::UI, FROM_HERE,
 113       base::Bind(&BrowsingDataServerBoundCertHelperImpl::NotifyInUIThread,
 114                  this, cert_list));
 115 }
 116
 117 void BrowsingDataServerBoundCertHelperImpl::NotifyInUIThread(
 118     const net::ServerBoundCertStore::ServerBoundCertList& cert_list) {
 119   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 120   DCHECK(is_fetching_);
 121   is_fetching_ = false;
 122   completion_callback_.Run(cert_list);
 123   completion_callback_.Reset();
 124 }
 125
 126 void BrowsingDataServerBoundCertHelperImpl::DeleteOnIOThread(
 127     const std::string& server_id) {
 128   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
 129   net::ServerBoundCertStore* cert_store =
 130       request_context_getter_->GetURLRequestContext()->
 131       server_bound_cert_service()->GetCertStore();
 132   if (cert_store) {
 133     cert_store->DeleteServerBoundCert(
 134         server_id,
 135         base::Bind(&BrowsingDataServerBoundCertHelperImpl::DeleteCallback,
 136                    this));
 137   }
 138 }
 139
 140 void BrowsingDataServerBoundCertHelperImpl::DeleteCallback() {
 141   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO));
 142   // Need to close open SSL connections which may be using the channel ids we
 143   // are deleting.
 144   // TODO(mattm): http://crbug.com/166069 Make the server bound cert
 145   // service/store have observers that can notify relevant things directly.
 146   request_context_getter_->GetURLRequestContext()->ssl_config_service()->
 147       NotifySSLConfigChange();
 148 }
 149
 150 }  // namespace
 151
 152 // static
 153 BrowsingDataServerBoundCertHelper*
 154 BrowsingDataServerBoundCertHelper::Create(Profile* profile) {
 155   return new BrowsingDataServerBoundCertHelperImpl(profile);
 156 }
 157
 158 CannedBrowsingDataServerBoundCertHelper::
 159 CannedBrowsingDataServerBoundCertHelper() {}
 160
 161 CannedBrowsingDataServerBoundCertHelper::
 162 ~CannedBrowsingDataServerBoundCertHelper() {}
 163
 164 CannedBrowsingDataServerBoundCertHelper*
 165 CannedBrowsingDataServerBoundCertHelper::Clone() {
 166   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 167   CannedBrowsingDataServerBoundCertHelper* clone =
 168       new CannedBrowsingDataServerBoundCertHelper();
 169
 170   clone->server_bound_cert_map_ = server_bound_cert_map_;
 171   return clone;
 172 }
 173
 174 void CannedBrowsingDataServerBoundCertHelper::AddServerBoundCert(
 175     const net::ServerBoundCertStore::ServerBoundCert& server_bound_cert) {
 176   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 177   server_bound_cert_map_[server_bound_cert.server_identifier()] =
 178       server_bound_cert;
 179 }
 180
 181 void CannedBrowsingDataServerBoundCertHelper::Reset() {
 182   server_bound_cert_map_.clear();
 183 }
 184
 185 bool CannedBrowsingDataServerBoundCertHelper::empty() const {
 186   return server_bound_cert_map_.empty();
 187 }
 188
 189 size_t CannedBrowsingDataServerBoundCertHelper::GetCertCount() const {
 190   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 191   return server_bound_cert_map_.size();
 192 }
 193
 194 void CannedBrowsingDataServerBoundCertHelper::StartFetching(
 195     const FetchResultCallback& callback) {
 196   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 197   if (callback.is_null())
 198     return;
 199   // We post a task to emulate async fetching behavior.
 200   completion_callback_ = callback;
 201   base::MessageLoop::current()->PostTask(
 202       FROM_HERE,
 203       base::Bind(&CannedBrowsingDataServerBoundCertHelper::FinishFetching,
 204                  this));
 205 }
 206
 207 void CannedBrowsingDataServerBoundCertHelper::FinishFetching() {
 208   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 209   net::ServerBoundCertStore::ServerBoundCertList cert_list;
 210   for (ServerBoundCertMap::iterator i = server_bound_cert_map_.begin();
 211        i != server_bound_cert_map_.end(); ++i)
 212     cert_list.push_back(i->second);
 213   completion_callback_.Run(cert_list);
 214 }
 215
 216 void CannedBrowsingDataServerBoundCertHelper::DeleteServerBoundCert(
 217     const std::string& server_id) {
 218   NOTREACHED();
 219 }