| Index: net/socket/ssl_client_socket_win.cc
|
| diff --git a/net/socket/ssl_client_socket_win.cc b/net/socket/ssl_client_socket_win.cc
|
| index 30f599d76d14d93eb881fd19594836c055e46149..3c855fb948296bf7e4025baf03802d9821fc022c 100644
|
| --- a/net/socket/ssl_client_socket_win.cc
|
| +++ b/net/socket/ssl_client_socket_win.cc
|
| @@ -400,7 +400,7 @@ SSLClientSocketWin::SSLClientSocketWin(ClientSocketHandle* transport_socket,
|
| old_user_connect_callback_(NULL),
|
| old_user_read_callback_(NULL),
|
| user_read_buf_len_(0),
|
| - user_write_callback_(NULL),
|
| + old_user_write_callback_(NULL),
|
| user_write_buf_len_(0),
|
| next_state_(STATE_NONE),
|
| cert_verifier_(context.cert_verifier),
|
| @@ -869,7 +869,29 @@ int SSLClientSocketWin::Read(IOBuffer* buf, int buf_len,
|
| int SSLClientSocketWin::Write(IOBuffer* buf, int buf_len,
|
| OldCompletionCallback* callback) {
|
| DCHECK(completed_handshake());
|
| - DCHECK(!user_write_callback_);
|
| + DCHECK(!old_user_write_callback_ && user_write_callback_.is_null());
|
| +
|
| + DCHECK(!user_write_buf_);
|
| + user_write_buf_ = buf;
|
| + user_write_buf_len_ = buf_len;
|
| +
|
| + int rv = DoPayloadEncrypt();
|
| + if (rv != OK)
|
| + return rv;
|
| +
|
| + rv = DoPayloadWrite();
|
| + if (rv == ERR_IO_PENDING) {
|
| + old_user_write_callback_ = callback;
|
| + } else {
|
| + user_write_buf_ = NULL;
|
| + user_write_buf_len_ = 0;
|
| + }
|
| + return rv;
|
| +}
|
| +int SSLClientSocketWin::Write(IOBuffer* buf, int buf_len,
|
| + const CompletionCallback& callback) {
|
| + DCHECK(completed_handshake());
|
| + DCHECK(!old_user_write_callback_ && user_write_callback_.is_null());
|
|
|
| DCHECK(!user_write_buf_);
|
| user_write_buf_ = buf;
|
| @@ -964,12 +986,20 @@ void SSLClientSocketWin::OnWriteComplete(int result) {
|
|
|
| int rv = DoPayloadWriteComplete(result);
|
| if (rv != ERR_IO_PENDING) {
|
| - DCHECK(user_write_callback_);
|
| - OldCompletionCallback* c = user_write_callback_;
|
| - user_write_callback_ = NULL;
|
| - user_write_buf_ = NULL;
|
| - user_write_buf_len_ = 0;
|
| - c->Run(rv);
|
| + DCHECK(old_user_write_callback_ || !user_write_callback_.is_null());
|
| + if (old_user_write_callback_) {
|
| + OldCompletionCallback* c = old_user_write_callback_;
|
| + old_user_write_callback_ = NULL;
|
| + user_write_buf_ = NULL;
|
| + user_write_buf_len_ = 0;
|
| + c->Run(rv);
|
| + } else {
|
| + CompletionCallback c = user_write_callback_;
|
| + user_write_callback_.Reset();
|
| + user_write_buf_ = NULL;
|
| + user_write_buf_len_ = 0;
|
| + c.Run(rv);
|
| + }
|
| }
|
| }
|
|
|
|
|