chrome/browser/ssl/ssl_client_auth_observer.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/ssl/ssl_client_auth_observer.h"
   6
   7 #include <utility>
   8
   9 #include "base/bind.h"
  10 #include "base/logging.h"
  11 #include "chrome/browser/chrome_notification_types.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "content/public/browser/client_certificate_delegate.h"
  14 #include "content/public/browser/notification_service.h"
  15 #include "net/cert/x509_certificate.h"
  16 #include "net/ssl/ssl_cert_request_info.h"
  17
  18 using content::BrowserThread;
  19
  20 typedef std::pair<net::SSLCertRequestInfo*, net::X509Certificate*> CertDetails;
  21
  22 SSLClientAuthObserver::SSLClientAuthObserver(
  23     const content::BrowserContext* browser_context,
  24     const scoped_refptr<net::SSLCertRequestInfo>& cert_request_info,
  25     scoped_ptr<content::ClientCertificateDelegate> delegate)
  26     : browser_context_(browser_context),
  27       cert_request_info_(cert_request_info),
  28       delegate_(delegate.Pass()) {
  29 }
  30
  31 SSLClientAuthObserver::~SSLClientAuthObserver() {
  32 }
  33
  34 void SSLClientAuthObserver::CertificateSelected(
  35     net::X509Certificate* certificate) {
  36   if (!delegate_)
  37     return;
  38
  39   // Stop listening now that the delegate has been resolved. This is also to
  40   // avoid getting a self-notification.
  41   StopObserving();
  42
  43   CertDetails details;
  44   details.first = cert_request_info_.get();
  45   details.second = certificate;
  46   content::NotificationService* service =
  47       content::NotificationService::current();
  48   service->Notify(chrome::NOTIFICATION_SSL_CLIENT_AUTH_CERT_SELECTED,
  49                   content::Source<content::BrowserContext>(browser_context_),
  50                   content::Details<CertDetails>(&details));
  51
  52   delegate_->ContinueWithCertificate(certificate);
  53   delegate_.reset();
  54 }
  55
  56 void SSLClientAuthObserver::CancelCertificateSelection() {
  57   if (!delegate_)
  58     return;
  59
  60   // Stop observing now that the delegate has been resolved.
  61   StopObserving();
  62   delegate_.reset();
  63 }
  64
  65 void SSLClientAuthObserver::Observe(
  66     int type,
  67     const content::NotificationSource& source,
  68     const content::NotificationDetails& details) {
  69   DVLOG(1) << "SSLClientAuthObserver::Observe " << this;
  70   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  71   DCHECK(type == chrome::NOTIFICATION_SSL_CLIENT_AUTH_CERT_SELECTED);
  72
  73   CertDetails* cert_details = content::Details<CertDetails>(details).ptr();
  74   if (!cert_details->first->host_and_port.Equals(
  75            cert_request_info_->host_and_port))
  76     return;
  77
  78   DVLOG(1) << this << " got matching notification and selecting cert "
  79            << cert_details->second;
  80   StopObserving();
  81   delegate_->ContinueWithCertificate(cert_details->second);
  82   delegate_.reset();
  83   OnCertSelectedByNotification();
  84 }
  85
  86 void SSLClientAuthObserver::StartObserving() {
  87   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  88   notification_registrar_.Add(
  89       this, chrome::NOTIFICATION_SSL_CLIENT_AUTH_CERT_SELECTED,
  90       content::Source<content::BrowserContext>(browser_context_));
  91 }
  92
  93 void SSLClientAuthObserver::StopObserving() {
  94   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  95   notification_registrar_.RemoveAll();
  96 }