This CL corrects a bug in which the OnHandshakeComplete callback for an ssl session was never called

net/socket/ssl_session_cache_openssl.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/socket/ssl_session_cache_openssl.h"
 
 #include <list>
 #include <map>
 
 #include <openssl/rand.h>
@@ skipping 219 matching lines @@
       return false;  // Session has not yet been marked good. Treat as a miss.
 
     // Move to front of MRU list.
     ordering_.push_front(session);
     ordering_.erase(it->second);
     it->second = ordering_.begin();
 
     return SSL_set_session(ssl, session) == 1;
   }
 
-  // Return true iff a cached session was associated with the given |cache_key|.
   bool SSLSessionIsInCache(const std::string& cache_key) const {
     base::AutoLock locked(lock_);
     KeyIndex::const_iterator it = key_index_.find(cache_key);
     if (it == key_index_.end())
       return false;
 
     SSL_SESSION* session = *it->second;
     DCHECK(session);
 
     void* session_is_good =
         SSL_SESSION_get_ex_data(session, GetSSLSessionExIndex());
 
     return session_is_good;
   }
 
-  void SetSessionAddedCallback(SSL* ssl, const base::Closure& callback) {
-    // Add |ssl| to the SSLToCallbackMap.
-    ssl_to_callback_map_.insert(SSLToCallbackMap::value_type(
-        ssl, CallbackAndCompletionCount(callback, 0)));
-  }
-
-  // Determines if the session for |ssl| is in the cache, and calls the
-  // appropriate callback if that is the case.
-  //
-  // The session must be both MarkedAsGood and Added in order for the
-  // callback to be run. These two events can occur in either order.
-  void CheckIfSessionFinished(SSL* ssl) {
-    SSLToCallbackMap::iterator it = ssl_to_callback_map_.find(ssl);
-    if (it == ssl_to_callback_map_.end())
-      return;
-    // Increment the session's completion count.
-    if (++it->second.count == 2) {
-      base::Closure callback = it->second.callback;
-      ssl_to_callback_map_.erase(it);
-      callback.Run();
-    }
-  }
-
-  void RemoveSessionAddedCallback(SSL* ssl) { ssl_to_callback_map_.erase(ssl); }
-
   void MarkSSLSessionAsGood(SSL* ssl) {
     SSL_SESSION* session = SSL_get_session(ssl);
     CHECK(session);
 
     // Mark the session as good, allowing it to be used for future connections.
     SSL_SESSION_set_ex_data(
         session, GetSSLSessionExIndex(), reinterpret_cast<void*>(1));
+  }
 
-    CheckIfSessionFinished(ssl);
+  bool SessionIsGood(SSL* ssl) {
+    SSL_SESSION* session = SSL_get_session(ssl);
+    CHECK(session);
+    return SSL_SESSION_get_ex_data(session, GetSSLSessionExIndex());
   }
 
   // Flush all entries from the cache.
   void Flush() {
     base::AutoLock lock(lock_);
     id_index_.clear();
     key_index_.clear();
     while (!ordering_.empty()) {
       SSL_SESSION* session = ordering_.front();
       ordering_.pop_front();
       SSL_SESSION_free(session);
     }
   }
 
  private:
-  // CallbackAndCompletionCounts are used to group a callback that should be
-  // run when a certain sesssion is added to the session cache with an integer
-  // indicating the status of that session.
-  struct CallbackAndCompletionCount {
-    CallbackAndCompletionCount(const base::Closure& completion_callback,
-                               int completion_count)
-        : callback(completion_callback), count(completion_count) {}
-
-    const base::Closure callback;
-    // |count| < 2 means that the ssl session associated with this object
-    // has not been added to the session cache or has not been marked as good.
-    // |count| is incremented when a session is added to the cache or marked as
-    // good, thus |count| == 2 means that the session is ready for use.
-    int count;
-  };
-
   // Type for list of SSL_SESSION handles, ordered in MRU order.
   typedef std::list<SSL_SESSION*> MRUSessionList;
   // Type for a dictionary from unique cache keys to session list nodes.
   typedef base::hash_map<std::string, MRUSessionList::iterator> KeyIndex;
   // Type for a dictionary from SessionId values to key index nodes.
   typedef base::hash_map<SessionId, KeyIndex::iterator> SessionIdIndex;
-  // Type for a map from SSL* to associated callbacks
-  typedef std::map<SSL*, CallbackAndCompletionCount> SSLToCallbackMap;
 
   // Return the key associated with a given session, or the empty string if
   // none exist. This shall only be used for debugging.
   std::string SessionKey(SSL_SESSION* session) {
     if (!session)
       return std::string("<null-session>");
 
     if (session->session_id_length == 0)
       return std::string("<empty-session-id>");
 
@@ skipping 59 matching lines @@
   }
 
   // Called by OpenSSL when a new |session| was created and added to a given
   // |ssl| connection. Note that the session's reference count was already
   // incremented before the function is entered. The function must return 1
   // to indicate that it took ownership of the session, i.e. that the caller
   // should not decrement its reference count after completion.
   static int NewSessionCallbackStatic(SSL* ssl, SSL_SESSION* session) {
     SSLSessionCacheOpenSSLImpl* cache = GetCache(ssl->ctx);
     cache->OnSessionAdded(ssl, session);
-    cache->CheckIfSessionFinished(ssl);
     return 1;
   }
 
   // Called by OpenSSL to indicate that a session must be removed from the
   // cache. This happens when SSL_CTX is destroyed.
   static void RemoveSessionCallbackStatic(SSL_CTX* ctx, SSL_SESSION* session) {
     GetCache(ctx)->OnSessionRemoved(session);
   }
 
   // Called by OpenSSL to generate a new session ID. This happens during a
@@ skipping 103 matching lines @@
       if (id_index_.find(SessionId(id, id_len)) == id_index_.end())
         return true;
     }
     DLOG(ERROR) << "Couldn't generate unique session ID of " << id_len
                 << "bytes after " << kMaxTries << " tries.";
     return false;
   }
 
   SSL_CTX* ctx_;
   SSLSessionCacheOpenSSL::Config config_;
-  SSLToCallbackMap ssl_to_callback_map_;
 
   // method to get the index which can later be used with SSL_CTX_get_ex_data()
   // or SSL_CTX_set_ex_data().
   mutable base::Lock lock_;  // Protects access to containers below.
 
   MRUSessionList ordering_;
   KeyIndex key_index_;
   SessionIdIndex id_index_;
 
   size_t expiration_check_;
@@ skipping 18 matching lines @@
     SSL* ssl,
     const std::string& cache_key) {
   return impl_->SetSSLSessionWithKey(ssl, cache_key);
 }
 
 bool SSLSessionCacheOpenSSL::SSLSessionIsInCache(
     const std::string& cache_key) const {
   return impl_->SSLSessionIsInCache(cache_key);
 }
 
-void SSLSessionCacheOpenSSL::RemoveSessionAddedCallback(SSL* ssl) {
-  impl_->RemoveSessionAddedCallback(ssl);
-}
-
-void SSLSessionCacheOpenSSL::SetSessionAddedCallback(SSL* ssl,
-                                                     const base::Closure& cb) {
-  impl_->SetSessionAddedCallback(ssl, cb);
-}
-
 void SSLSessionCacheOpenSSL::MarkSSLSessionAsGood(SSL* ssl) {
   return impl_->MarkSSLSessionAsGood(ssl);
 }
 
+bool SSLSessionCacheOpenSSL::SessionIsGood(SSL* ssl) {
+  return impl_->SessionIsGood(ssl);
+}
+
 void SSLSessionCacheOpenSSL::Flush() { impl_->Flush(); }
 
 }  // namespace net