| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/cert_store_impl.h" | 5 #include "content/browser/cert_store_impl.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <functional> | 8 #include <functional> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| 11 #include "base/stl_util.h" | 11 #include "base/stl_util.h" |
| 12 #include "content/browser/renderer_host/render_process_host_impl.h" | 12 #include "content/browser/renderer_host/render_process_host_impl.h" |
| 13 #include "content/browser/renderer_host/render_view_host_impl.h" | 13 #include "content/browser/renderer_host/render_view_host_impl.h" |
| 14 #include "content/public/browser/browser_thread.h" | 14 #include "content/public/browser/browser_thread.h" |
| 15 #include "content/public/browser/notification_service.h" | 15 #include "content/public/browser/notification_service.h" |
| 16 #include "content/public/browser/notification_types.h" | 16 #include "content/public/browser/notification_types.h" |
| 17 | 17 |
| 18 namespace content { | |
| 19 | |
| 20 namespace { | |
| 21 | |
| 22 template <typename T> | 18 template <typename T> |
| 23 struct MatchSecond { | 19 struct MatchSecond { |
| 24 explicit MatchSecond(const T& t) : value(t) {} | 20 explicit MatchSecond(const T& t) : value(t) {} |
| 25 | 21 |
| 26 template<typename Pair> | 22 template<typename Pair> |
| 27 bool operator()(const Pair& p) const { | 23 bool operator()(const Pair& p) const { |
| 28 return (value == p.second); | 24 return (value == p.second); |
| 29 } | 25 } |
| 30 T value; | 26 T value; |
| 31 }; | 27 }; |
| 32 | 28 |
| 33 } // namespace | |
| 34 | |
| 35 // static | 29 // static |
| 36 CertStore* CertStore::GetInstance() { | 30 content::CertStore* content::CertStore::GetInstance() { |
| 37 return CertStoreImpl::GetInstance(); | 31 return CertStoreImpl::GetInstance(); |
| 38 } | 32 } |
| 39 | 33 |
| 40 // static | 34 // static |
| 41 CertStoreImpl* CertStoreImpl::GetInstance() { | 35 CertStoreImpl* CertStoreImpl::GetInstance() { |
| 42 return Singleton<CertStoreImpl>::get(); | 36 return Singleton<CertStoreImpl>::get(); |
| 43 } | 37 } |
| 44 | 38 |
| 45 CertStoreImpl::CertStoreImpl() : next_cert_id_(1) { | 39 CertStoreImpl::CertStoreImpl() : next_cert_id_(1) { |
| 46 if (BrowserThread::CurrentlyOn(BrowserThread::UI)) { | 40 if (content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)) { |
| 47 RegisterForNotification(); | 41 RegisterForNotification(); |
| 48 } else { | 42 } else { |
| 49 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 43 content::BrowserThread::PostTask( |
| 44 content::BrowserThread::UI, FROM_HERE, |
| 50 base::Bind(&CertStoreImpl::RegisterForNotification, | 45 base::Bind(&CertStoreImpl::RegisterForNotification, |
| 51 base::Unretained(this))); | 46 base::Unretained(this))); |
| 52 } | 47 } |
| 53 } | 48 } |
| 54 | 49 |
| 55 CertStoreImpl::~CertStoreImpl() { | 50 CertStoreImpl::~CertStoreImpl() { |
| 56 } | 51 } |
| 57 | 52 |
| 58 void CertStoreImpl::RegisterForNotification() { | 53 void CertStoreImpl::RegisterForNotification() { |
| 59 // We watch for RenderProcess termination, as this is how we clear | 54 // We watch for RenderProcess termination, as this is how we clear |
| 60 // certificates for now. | 55 // certificates for now. |
| 61 // TODO(jcampan): we should be listening to events such as resource cached/ | 56 // TODO(jcampan): we should be listening to events such as resource cached/ |
| 62 // removed from cache, and remove the cert when we know it | 57 // removed from cache, and remove the cert when we know it |
| 63 // is not used anymore. | 58 // is not used anymore. |
| 64 | 59 |
| 65 registrar_.Add(this, NOTIFICATION_RENDERER_PROCESS_TERMINATED, | 60 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, |
| 66 NotificationService::AllBrowserContextsAndSources()); | 61 content::NotificationService::AllBrowserContextsAndSources()); |
| 67 registrar_.Add(this, NOTIFICATION_RENDERER_PROCESS_CLOSED, | 62 registrar_.Add(this, content::NOTIFICATION_RENDERER_PROCESS_CLOSED, |
| 68 NotificationService::AllBrowserContextsAndSources()); | 63 content::NotificationService::AllBrowserContextsAndSources()); |
| 69 } | 64 } |
| 70 | 65 |
| 71 int CertStoreImpl::StoreCert(net::X509Certificate* cert, int process_id) { | 66 int CertStoreImpl::StoreCert(net::X509Certificate* cert, int process_id) { |
| 72 DCHECK(cert); | 67 DCHECK(cert); |
| 73 base::AutoLock auto_lock(cert_lock_); | 68 base::AutoLock auto_lock(cert_lock_); |
| 74 | 69 |
| 75 int cert_id; | 70 int cert_id; |
| 76 | 71 |
| 77 // Do we already know this cert? | 72 // Do we already know this cert? |
| 78 ReverseCertMap::iterator cert_iter = cert_to_id_.find(cert); | 73 ReverseCertMap::iterator cert_iter = cert_to_id_.find(cert); |
| (...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 // The current cert id is not referenced by any other processes, so | 161 // The current cert id is not referenced by any other processes, so |
| 167 // remove it from id_to_cert_ and cert_to_id_. | 162 // remove it from id_to_cert_ and cert_to_id_. |
| 168 RemoveCertInternal(cert_id); | 163 RemoveCertInternal(cert_id); |
| 169 } | 164 } |
| 170 } | 165 } |
| 171 if (process_ids.first != process_ids.second) | 166 if (process_ids.first != process_ids.second) |
| 172 process_id_to_cert_id_.erase(process_ids.first, process_ids.second); | 167 process_id_to_cert_id_.erase(process_ids.first, process_ids.second); |
| 173 } | 168 } |
| 174 | 169 |
| 175 void CertStoreImpl::Observe(int type, | 170 void CertStoreImpl::Observe(int type, |
| 176 const NotificationSource& source, | 171 const content::NotificationSource& source, |
| 177 const NotificationDetails& details) { | 172 const content::NotificationDetails& details) { |
| 178 DCHECK(type == NOTIFICATION_RENDERER_PROCESS_TERMINATED || | 173 DCHECK(type == content::NOTIFICATION_RENDERER_PROCESS_TERMINATED || |
| 179 type == NOTIFICATION_RENDERER_PROCESS_CLOSED); | 174 type == content::NOTIFICATION_RENDERER_PROCESS_CLOSED); |
| 180 RenderProcessHost* rph = Source<RenderProcessHost>(source).ptr(); | 175 content::RenderProcessHost* rph = |
| 176 content::Source<content::RenderProcessHost>(source).ptr(); |
| 181 DCHECK(rph); | 177 DCHECK(rph); |
| 182 RemoveCertsForRenderProcesHost(rph->GetID()); | 178 RemoveCertsForRenderProcesHost(rph->GetID()); |
| 183 } | 179 } |
| 184 | |
| 185 } // namespace content | |
| OLD | NEW |