Index: net/socket/ssl_client_socket_mac.cc |
diff --git a/net/socket/ssl_client_socket_mac.cc b/net/socket/ssl_client_socket_mac.cc |
index 817b3bef25d76de67f472fd79fd126e75dead27c..f58d3409ac9fb813a1f627c9c8eb2769f1198640 100644 |
--- a/net/socket/ssl_client_socket_mac.cc |
+++ b/net/socket/ssl_client_socket_mac.cc |
@@ -532,7 +532,7 @@ SSLClientSocketMac::SSLClientSocketMac(ClientSocketHandle* transport_socket, |
host_and_port_(host_and_port), |
ssl_config_(ssl_config), |
old_user_connect_callback_(NULL), |
- user_read_callback_(NULL), |
+ old_user_read_callback_(NULL), |
user_write_callback_(NULL), |
user_read_buf_len_(0), |
user_write_buf_len_(0), |
@@ -700,7 +700,25 @@ base::TimeDelta SSLClientSocketMac::GetConnectTimeMicros() const { |
int SSLClientSocketMac::Read(IOBuffer* buf, int buf_len, |
OldCompletionCallback* callback) { |
DCHECK(completed_handshake()); |
- DCHECK(!user_read_callback_); |
+ DCHECK(!old_user_read_callback_ && user_read_callback_.is_null()); |
+ DCHECK(!user_read_buf_); |
+ |
+ user_read_buf_ = buf; |
+ user_read_buf_len_ = buf_len; |
+ |
+ int rv = DoPayloadRead(); |
+ if (rv == ERR_IO_PENDING) { |
+ old_user_read_callback_ = callback; |
+ } else { |
+ user_read_buf_ = NULL; |
+ user_read_buf_len_ = 0; |
+ } |
+ return rv; |
+} |
+int SSLClientSocketMac::Read(IOBuffer* buf, int buf_len, |
+ const CompletionCallback& callback) { |
+ DCHECK(completed_handshake()); |
+ DCHECK(!old_user_read_callback_ && user_read_callback_.is_null()); |
DCHECK(!user_read_buf_); |
user_read_buf_ = buf; |
@@ -933,15 +951,23 @@ void SSLClientSocketMac::DoConnectCallback(int rv) { |
void SSLClientSocketMac::DoReadCallback(int rv) { |
DCHECK(rv != ERR_IO_PENDING); |
- DCHECK(user_read_callback_); |
+ DCHECK(old_user_read_callback_ || !user_read_callback_.is_null()); |
// Since Run may result in Read being called, clear user_read_callback_ up |
// front. |
- OldCompletionCallback* c = user_read_callback_; |
- user_read_callback_ = NULL; |
- user_read_buf_ = NULL; |
- user_read_buf_len_ = 0; |
- c->Run(rv); |
+ if (old_user_read_callback_) { |
+ OldCompletionCallback* c = old_user_read_callback_; |
+ old_user_read_callback_ = NULL; |
+ user_read_buf_ = NULL; |
+ user_read_buf_len_ = 0; |
+ c->Run(rv); |
+ } else { |
+ CompletionCallback c = user_read_callback_; |
+ user_read_callback_.Reset(); |
+ user_read_buf_ = NULL; |
+ user_read_buf_len_ = 0; |
+ c.Run(rv); |
+ } |
} |
void SSLClientSocketMac::DoWriteCallback(int rv) { |