Index: net/base/cert_verifier.cc |
diff --git a/net/base/cert_verifier.cc b/net/base/cert_verifier.cc |
index 4e941335653adea573ae13ae3dadcb18ebed3c01..ab939c4bf4a1ae4e81a4c2d125a161aa4215723d 100644 |
--- a/net/base/cert_verifier.cc |
+++ b/net/base/cert_verifier.cc |
@@ -17,7 +17,9 @@ |
namespace net { |
class CertVerifier::Request : |
- public base::RefCountedThreadSafe<CertVerifier::Request> { |
+ public base::RefCountedThreadSafe<CertVerifier::Request, |
+ CertVerifier::RequestTraits>, |
+ public MessageLoop::DestructionObserver { |
public: |
Request(CertVerifier* verifier, |
X509Certificate* cert, |
@@ -33,6 +35,8 @@ class CertVerifier::Request : |
callback_(callback), |
origin_loop_(MessageLoop::current()), |
error_(OK) { |
+ if (origin_loop_) |
+ origin_loop_->AddDestructionObserver(this); |
} |
void DoVerify() { |
@@ -84,15 +88,40 @@ class CertVerifier::Request : |
void Cancel() { |
verifier_ = NULL; |
+ AutoLock locked(origin_loop_lock_); |
+ if (origin_loop_) { |
+ origin_loop_->RemoveDestructionObserver(this); |
+ origin_loop_ = NULL; |
+ } |
+ } |
+ // MessageLoop::DestructionObserver override. |
+ virtual void WillDestroyCurrentMessageLoop() { |
+ LOG(ERROR) << "CertVerifier wasn't deleted before the thread was deleted."; |
AutoLock locked(origin_loop_lock_); |
origin_loop_ = NULL; |
} |
private: |
friend class base::RefCountedThreadSafe<CertVerifier::Request>; |
+ friend class DeleteTask<CertVerifier::Request>; |
+ friend struct CertVerifier::RequestTraits; |
- ~Request() {} |
+ ~Request() { |
+ Cancel(); |
+ } |
+ |
+ // If we have an origin_loop_, the observer needs to be removed on that |
+ // thread. Using OnDestruct is called by RequestTraits to ensure this |
wtc
2010/11/11 22:24:11
Nit: remove "Using".
Zachary Kuznia
2010/11/15 10:25:09
Done.
|
+ // happens. |
+ void OnDestruct() const { |
+ AutoLock locked(origin_loop_lock_); |
willchan no longer on Chromium
2010/11/12 00:32:29
Won't this crash? On line 122, you delete this.
wtc
2010/11/12 00:44:00
willchan: you're right. Good catch. By the way,
Zachary Kuznia
2010/11/15 10:25:09
Fixed.
You can actually delete 'this' in a const
|
+ if (origin_loop_ && origin_loop_ != MessageLoop::current()) { |
+ origin_loop_->DeleteSoon(FROM_HERE, this); |
+ } else { |
+ delete this; |
+ } |
+ } |
// Set on the origin thread, read on the worker thread. |
scoped_refptr<X509Certificate> cert_; |
@@ -114,6 +143,12 @@ class CertVerifier::Request : |
CertVerifyResult result_; |
}; |
+struct CertVerifier::RequestTraits { |
+ static void Destruct(const CertVerifier::Request* request) { |
+ request->OnDestruct(); |
+ } |
+}; |
+ |
//----------------------------------------------------------------------------- |
CertVerifier::CertVerifier() { |