content/browser/cert_store_impl.h

   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 #ifndef CONTENT_BROWSER_CERT_STORE_IMPL_H_
   6 #define CONTENT_BROWSER_CERT_STORE_IMPL_H_
   7
   8 #include <map>
   9
  10 #include "base/memory/singleton.h"
  11 #include "base/synchronization/lock.h"
  12 #include "content/public/browser/cert_store.h"
  13 #include "content/public/browser/notification_observer.h"
  14 #include "content/public/browser/notification_registrar.h"
  15 #include "net/cert/x509_certificate.h"
  16
  17 namespace content {
  18
  19 class CertStoreImpl : public CertStore,
  20                       public NotificationObserver {
  21  public:
  22   // Returns the singleton instance of the CertStore.
  23   static CertStoreImpl* GetInstance();
  24
  25   // CertStore implementation:
  26   virtual int StoreCert(net::X509Certificate* cert,
  27                         int render_process_host_id) OVERRIDE;
  28   virtual bool RetrieveCert(int cert_id,
  29                             scoped_refptr<net::X509Certificate>* cert) OVERRIDE;
  30
  31   // NotificationObserver implementation.
  32   virtual void Observe(int type,
  33                        const NotificationSource& source,
  34                        const NotificationDetails& details) OVERRIDE;
  35  protected:
  36   CertStoreImpl();
  37   virtual ~CertStoreImpl();
  38
  39  private:
  40   friend struct DefaultSingletonTraits<CertStoreImpl>;
  41
  42   void RegisterForNotification();
  43
  44   // Remove the specified cert from id_to_cert_ and cert_to_id_.
  45   // NOTE: the caller (RemoveCertsForRenderProcesHost) must hold cert_lock_.
  46   void RemoveCertInternal(int cert_id);
  47
  48   // Removes all the certs associated with the specified process from the store.
  49   void RemoveCertsForRenderProcesHost(int render_process_host_id);
  50
  51   typedef std::multimap<int, int> IDMap;
  52   typedef std::map<int, scoped_refptr<net::X509Certificate> > CertMap;
  53   typedef std::map<net::X509Certificate*, int, net::X509Certificate::LessThan>
  54       ReverseCertMap;
  55
  56   // Is only used on the UI Thread.
  57   NotificationRegistrar registrar_;
  58
  59   IDMap process_id_to_cert_id_;
  60   IDMap cert_id_to_process_id_;
  61
  62   CertMap id_to_cert_;
  63   ReverseCertMap cert_to_id_;
  64
  65   int next_cert_id_;
  66
  67   // This lock protects: process_to_ids_, id_to_processes_, id_to_cert_ and
  68   //                     cert_to_id_.
  69   base::Lock cert_lock_;
  70
  71   DISALLOW_COPY_AND_ASSIGN(CertStoreImpl);
  72 };
  73
  74 }  // namespace content
  75
  76 #endif  // CONTENT_BROWSER_CERT_STORE_IMPL_H_