Index: net/socket/ssl_client_socket_nss.cc |
diff --git a/net/socket/ssl_client_socket_nss.cc b/net/socket/ssl_client_socket_nss.cc |
index 18c0cf7ae9f57b3fd09f5f3a4e66f20e3543e3f0..262193cb784bb8abf20480fac4dd6c282421728e 100644 |
--- a/net/socket/ssl_client_socket_nss.cc |
+++ b/net/socket/ssl_client_socket_nss.cc |
@@ -445,7 +445,7 @@ SSLClientSocketNSS::SSLClientSocketNSS(ClientSocketHandle* transport_socket, |
transport_(transport_socket), |
host_and_port_(host_and_port), |
ssl_config_(ssl_config), |
- user_connect_callback_(NULL), |
+ old_user_connect_callback_(NULL), |
user_read_callback_(NULL), |
user_write_callback_(NULL), |
user_read_buf_len_(0), |
@@ -569,7 +569,55 @@ int SSLClientSocketNSS::Connect(OldCompletionCallback* callback) { |
DCHECK(next_handshake_state_ == STATE_NONE); |
DCHECK(!user_read_callback_); |
DCHECK(!user_write_callback_); |
- DCHECK(!user_connect_callback_); |
+ DCHECK(!old_user_connect_callback_); |
csilv
2011/12/06 21:03:18
make this check same as line 620
James Hawkins
2011/12/06 22:19:30
Done.
|
+ DCHECK(!user_read_buf_); |
+ DCHECK(!user_write_buf_); |
+ |
+ EnsureThreadIdAssigned(); |
+ |
+ net_log_.BeginEvent(NetLog::TYPE_SSL_CONNECT, NULL); |
+ |
+ int rv = Init(); |
+ if (rv != OK) { |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SSL_CONNECT, rv); |
+ return rv; |
+ } |
+ |
+ rv = InitializeSSLOptions(); |
+ if (rv != OK) { |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SSL_CONNECT, rv); |
+ return rv; |
+ } |
+ |
+ rv = InitializeSSLPeerName(); |
+ if (rv != OK) { |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SSL_CONNECT, rv); |
+ return rv; |
+ } |
+ |
+ if (ssl_config_.cached_info_enabled && ssl_host_info_.get()) { |
+ GotoState(STATE_LOAD_SSL_HOST_INFO); |
+ } else { |
+ GotoState(STATE_HANDSHAKE); |
+ } |
+ |
+ rv = DoHandshakeLoop(OK); |
+ if (rv == ERR_IO_PENDING) { |
+ old_user_connect_callback_ = callback; |
+ } else { |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SSL_CONNECT, rv); |
+ } |
+ |
+ LeaveFunction(""); |
+ return rv > OK ? OK : rv; |
+} |
+int SSLClientSocketNSS::Connect(const CompletionCallback& callback) { |
+ EnterFunction(""); |
+ DCHECK(transport_.get()); |
+ DCHECK(next_handshake_state_ == STATE_NONE); |
+ DCHECK(!user_read_callback_); |
+ DCHECK(!user_write_callback_); |
+ DCHECK(!old_user_connect_callback_ && user_connect_callback_.is_null()); |
DCHECK(!user_read_buf_); |
DCHECK(!user_write_buf_); |
@@ -636,7 +684,8 @@ void SSLClientSocketNSS::Disconnect() { |
// Reset object state |
transport_send_busy_ = false; |
transport_recv_busy_ = false; |
- user_connect_callback_ = NULL; |
+ old_user_connect_callback_ = NULL; |
+ user_connect_callback_.Reset(); |
user_read_callback_ = NULL; |
user_write_callback_ = NULL; |
user_read_buf_ = NULL; |
@@ -758,7 +807,7 @@ int SSLClientSocketNSS::Read(IOBuffer* buf, int buf_len, |
DCHECK(completed_handshake_); |
DCHECK(next_handshake_state_ == STATE_NONE); |
DCHECK(!user_read_callback_); |
- DCHECK(!user_connect_callback_); |
+ DCHECK(!old_user_connect_callback_); |
DCHECK(!user_read_buf_); |
DCHECK(nss_bufs_); |
@@ -783,7 +832,7 @@ int SSLClientSocketNSS::Write(IOBuffer* buf, int buf_len, |
DCHECK(completed_handshake_); |
DCHECK(next_handshake_state_ == STATE_NONE); |
DCHECK(!user_write_callback_); |
- DCHECK(!user_connect_callback_); |
+ DCHECK(!old_user_connect_callback_); |
DCHECK(!user_write_buf_); |
DCHECK(nss_bufs_); |
@@ -1179,11 +1228,17 @@ void SSLClientSocketNSS::DoWriteCallback(int rv) { |
void SSLClientSocketNSS::DoConnectCallback(int rv) { |
EnterFunction(rv); |
DCHECK_NE(rv, ERR_IO_PENDING); |
- DCHECK(user_connect_callback_); |
+ DCHECK(old_user_connect_callback_ || !user_connect_callback_.is_null()); |
- OldCompletionCallback* c = user_connect_callback_; |
- user_connect_callback_ = NULL; |
- c->Run(rv > OK ? OK : rv); |
+ if (old_user_connect_callback_) { |
+ OldCompletionCallback* c = old_user_connect_callback_; |
+ old_user_connect_callback_ = NULL; |
+ c->Run(rv > OK ? OK : rv); |
+ } else { |
+ CompletionCallback c = user_connect_callback_; |
+ user_connect_callback_.Reset(); |
+ c.Run(rv > OK ? OK : rv); |
+ } |
LeaveFunction(""); |
} |