Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1735)

Unified Diff: net/socket/ssl_client_socket_pool.cc

Issue 364943002: Makes waiting SSLConnectJobs use the message loops to resume their connection. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fixed use after free bug & other cl comments. Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« net/socket/ssl_client_socket_pool.h ('K') | « net/socket/ssl_client_socket_pool.h ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/socket/ssl_client_socket_pool.cc
diff --git a/net/socket/ssl_client_socket_pool.cc b/net/socket/ssl_client_socket_pool.cc
index e3530ef1bc4c9529434b4b15975b97c7a633b8dd..b3d10c7e1bc41961a639cd709bb32ae344f5a480 100644
--- a/net/socket/ssl_client_socket_pool.cc
+++ b/net/socket/ssl_client_socket_pool.cc
@@ -9,6 +9,7 @@
#include "base/metrics/field_trial.h"
#include "base/metrics/histogram.h"
#include "base/metrics/sparse_histogram.h"
+#include "base/thread_task_runner_handle.h"
#include "base/values.h"
#include "net/base/host_port_pair.h"
#include "net/base/net_errors.h"
@@ -115,6 +116,9 @@ void SSLConnectJobMessenger::RemoveSocket(SSLClientSocket* ssl_socket) {
}
}
+SSLConnectJobMessenger::SSLConnectJobMessenger() : weak_factory_(this) {
+}
+
bool SSLConnectJobMessenger::CanProceed(SSLClientSocket* ssl_socket) {
// If the session is in the session cache, or there are no connecting
// sockets allow the connection to proceed.
@@ -124,11 +128,11 @@ bool SSLConnectJobMessenger::CanProceed(SSLClientSocket* ssl_socket) {
void SSLConnectJobMessenger::MonitorConnectionResult(
SSLClientSocket* ssl_socket) {
connecting_sockets_.push_back(ssl_socket);
- // TODO(mshelley): Both of these callbacks will use weak_ptr in future CL.
- ssl_socket->SetHandshakeFailureCallback(
- base::Bind(&SSLConnectJobMessenger::OnJobFailed, base::Unretained(this)));
+
+ ssl_socket->SetHandshakeFailureCallback(base::Bind(
+ &SSLConnectJobMessenger::OnJobFailed, weak_factory_.GetWeakPtr()));
ssl_socket->SetHandshakeSuccessCallback(base::Bind(
- &SSLConnectJobMessenger::OnJobSucceeded, base::Unretained(this)));
+ &SSLConnectJobMessenger::OnJobSucceeded, weak_factory_.GetWeakPtr()));
}
void SSLConnectJobMessenger::AddPendingSocket(SSLClientSocket* socket,
@@ -144,23 +148,34 @@ void SSLConnectJobMessenger::OnJobSucceeded() {
}
void SSLConnectJobMessenger::OnJobFailed() {
+ connecting_sockets_.erase(connecting_sockets_.begin());
+
+ // If there were no valid pending sockets, return.
if (pending_sockets_and_callbacks_.empty())
return;
- base::Closure callback = pending_sockets_and_callbacks_[0].callback;
- connecting_sockets_.erase(connecting_sockets_.begin());
- SSLClientSocket* ssl_socket = pending_sockets_and_callbacks_[0].socket;
+
+ SocketAndCallback socket_and_callback =
+ pending_sockets_and_callbacks_.front();
pending_sockets_and_callbacks_.erase(pending_sockets_and_callbacks_.begin());
- MonitorConnectionResult(ssl_socket);
- callback.Run();
+
+ MonitorConnectionResult(socket_and_callback.socket);
+
+ // PostTask is used to avoid the recursive call sequence caused
+ // by SSLConnectJobMessengers running the callback directly.
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
+ socket_and_callback.callback);
mmenke 2014/07/22 18:06:32 Hrm... Looks like we're fine if the second socket
mshelley 2014/07/23 22:09:07 Just to confirm: by "second case" you mean the cas
}
void SSLConnectJobMessenger::RunAllCallbacks(
const SSLPendingSocketsAndCallbacks& pending_sockets_and_callbacks) {
+ scoped_refptr<base::SequencedTaskRunner> task_runner =
+ base::ThreadTaskRunnerHandle::Get();
for (std::vector<SocketAndCallback>::const_iterator it =
pending_sockets_and_callbacks.begin();
it != pending_sockets_and_callbacks.end();
- ++it)
- it->callback.Run();
+ ++it) {
+ task_runner->PostTask(FROM_HERE, it->callback);
+ }
}
// Timeout for the SSL handshake portion of the connect.
« net/socket/ssl_client_socket_pool.h ('K') | « net/socket/ssl_client_socket_pool.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698