| // Copyright (c) 2012 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_client_socket.h" | 
 |  | 
 | #include "base/callback_helpers.h" | 
 | #include "base/memory/ref_counted.h" | 
 | #include "base/run_loop.h" | 
 | #include "base/time/time.h" | 
 | #include "net/base/address_list.h" | 
 | #include "net/base/io_buffer.h" | 
 | #include "net/base/net_errors.h" | 
 | #include "net/base/net_log.h" | 
 | #include "net/base/net_log_unittest.h" | 
 | #include "net/base/test_completion_callback.h" | 
 | #include "net/base/test_data_directory.h" | 
 | #include "net/cert/asn1_util.h" | 
 | #include "net/cert/ct_verifier.h" | 
 | #include "net/cert/mock_cert_verifier.h" | 
 | #include "net/cert/test_root_certs.h" | 
 | #include "net/dns/host_resolver.h" | 
 | #include "net/http/transport_security_state.h" | 
 | #include "net/socket/client_socket_factory.h" | 
 | #include "net/socket/client_socket_handle.h" | 
 | #include "net/socket/socket_test_util.h" | 
 | #include "net/socket/tcp_client_socket.h" | 
 | #include "net/ssl/channel_id_service.h" | 
 | #include "net/ssl/default_channel_id_store.h" | 
 | #include "net/ssl/ssl_cert_request_info.h" | 
 | #include "net/ssl/ssl_config_service.h" | 
 | #include "net/test/cert_test_util.h" | 
 | #include "net/test/spawned_test_server/spawned_test_server.h" | 
 | #include "testing/gmock/include/gmock/gmock.h" | 
 | #include "testing/gtest/include/gtest/gtest.h" | 
 | #include "testing/platform_test.h" | 
 |  | 
 | //----------------------------------------------------------------------------- | 
 |  | 
 | using testing::_; | 
 | using testing::Return; | 
 | using testing::Truly; | 
 |  | 
 | namespace net { | 
 |  | 
 | namespace { | 
 |  | 
 | const SSLConfig kDefaultSSLConfig; | 
 |  | 
 | // WrappedStreamSocket is a base class that wraps an existing StreamSocket, | 
 | // forwarding the Socket and StreamSocket interfaces to the underlying | 
 | // transport. | 
 | // This is to provide a common base class for subclasses to override specific | 
 | // StreamSocket methods for testing, while still communicating with a 'real' | 
 | // StreamSocket. | 
 | class WrappedStreamSocket : public StreamSocket { | 
 |  public: | 
 |   explicit WrappedStreamSocket(scoped_ptr<StreamSocket> transport) | 
 |       : transport_(transport.Pass()) {} | 
 |   virtual ~WrappedStreamSocket() {} | 
 |  | 
 |   // StreamSocket implementation: | 
 |   virtual int Connect(const CompletionCallback& callback) OVERRIDE { | 
 |     return transport_->Connect(callback); | 
 |   } | 
 |   virtual void Disconnect() OVERRIDE { transport_->Disconnect(); } | 
 |   virtual bool IsConnected() const OVERRIDE { | 
 |     return transport_->IsConnected(); | 
 |   } | 
 |   virtual bool IsConnectedAndIdle() const OVERRIDE { | 
 |     return transport_->IsConnectedAndIdle(); | 
 |   } | 
 |   virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE { | 
 |     return transport_->GetPeerAddress(address); | 
 |   } | 
 |   virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE { | 
 |     return transport_->GetLocalAddress(address); | 
 |   } | 
 |   virtual const BoundNetLog& NetLog() const OVERRIDE { | 
 |     return transport_->NetLog(); | 
 |   } | 
 |   virtual void SetSubresourceSpeculation() OVERRIDE { | 
 |     transport_->SetSubresourceSpeculation(); | 
 |   } | 
 |   virtual void SetOmniboxSpeculation() OVERRIDE { | 
 |     transport_->SetOmniboxSpeculation(); | 
 |   } | 
 |   virtual bool WasEverUsed() const OVERRIDE { | 
 |     return transport_->WasEverUsed(); | 
 |   } | 
 |   virtual bool UsingTCPFastOpen() const OVERRIDE { | 
 |     return transport_->UsingTCPFastOpen(); | 
 |   } | 
 |   virtual bool WasNpnNegotiated() const OVERRIDE { | 
 |     return transport_->WasNpnNegotiated(); | 
 |   } | 
 |   virtual NextProto GetNegotiatedProtocol() const OVERRIDE { | 
 |     return transport_->GetNegotiatedProtocol(); | 
 |   } | 
 |   virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE { | 
 |     return transport_->GetSSLInfo(ssl_info); | 
 |   } | 
 |  | 
 |   // Socket implementation: | 
 |   virtual int Read(IOBuffer* buf, | 
 |                    int buf_len, | 
 |                    const CompletionCallback& callback) OVERRIDE { | 
 |     return transport_->Read(buf, buf_len, callback); | 
 |   } | 
 |   virtual int Write(IOBuffer* buf, | 
 |                     int buf_len, | 
 |                     const CompletionCallback& callback) OVERRIDE { | 
 |     return transport_->Write(buf, buf_len, callback); | 
 |   } | 
 |   virtual int SetReceiveBufferSize(int32 size) OVERRIDE { | 
 |     return transport_->SetReceiveBufferSize(size); | 
 |   } | 
 |   virtual int SetSendBufferSize(int32 size) OVERRIDE { | 
 |     return transport_->SetSendBufferSize(size); | 
 |   } | 
 |  | 
 |  protected: | 
 |   scoped_ptr<StreamSocket> transport_; | 
 | }; | 
 |  | 
 | // ReadBufferingStreamSocket is a wrapper for an existing StreamSocket that | 
 | // will ensure a certain amount of data is internally buffered before | 
 | // satisfying a Read() request. It exists to mimic OS-level internal | 
 | // buffering, but in a way to guarantee that X number of bytes will be | 
 | // returned to callers of Read(), regardless of how quickly the OS receives | 
 | // them from the TestServer. | 
 | class ReadBufferingStreamSocket : public WrappedStreamSocket { | 
 |  public: | 
 |   explicit ReadBufferingStreamSocket(scoped_ptr<StreamSocket> transport); | 
 |   virtual ~ReadBufferingStreamSocket() {} | 
 |  | 
 |   // Socket implementation: | 
 |   virtual int Read(IOBuffer* buf, | 
 |                    int buf_len, | 
 |                    const CompletionCallback& callback) OVERRIDE; | 
 |  | 
 |   // Sets the internal buffer to |size|. This must not be greater than | 
 |   // the largest value supplied to Read() - that is, it does not handle | 
 |   // having "leftovers" at the end of Read(). | 
 |   // Each call to Read() will be prevented from completion until at least | 
 |   // |size| data has been read. | 
 |   // Set to 0 to turn off buffering, causing Read() to transparently | 
 |   // read via the underlying transport. | 
 |   void SetBufferSize(int size); | 
 |  | 
 |  private: | 
 |   enum State { | 
 |     STATE_NONE, | 
 |     STATE_READ, | 
 |     STATE_READ_COMPLETE, | 
 |   }; | 
 |  | 
 |   int DoLoop(int result); | 
 |   int DoRead(); | 
 |   int DoReadComplete(int result); | 
 |   void OnReadCompleted(int result); | 
 |  | 
 |   State state_; | 
 |   scoped_refptr<GrowableIOBuffer> read_buffer_; | 
 |   int buffer_size_; | 
 |  | 
 |   scoped_refptr<IOBuffer> user_read_buf_; | 
 |   CompletionCallback user_read_callback_; | 
 | }; | 
 |  | 
 | ReadBufferingStreamSocket::ReadBufferingStreamSocket( | 
 |     scoped_ptr<StreamSocket> transport) | 
 |     : WrappedStreamSocket(transport.Pass()), | 
 |       read_buffer_(new GrowableIOBuffer()), | 
 |       buffer_size_(0) {} | 
 |  | 
 | void ReadBufferingStreamSocket::SetBufferSize(int size) { | 
 |   DCHECK(!user_read_buf_.get()); | 
 |   buffer_size_ = size; | 
 |   read_buffer_->SetCapacity(size); | 
 | } | 
 |  | 
 | int ReadBufferingStreamSocket::Read(IOBuffer* buf, | 
 |                                     int buf_len, | 
 |                                     const CompletionCallback& callback) { | 
 |   if (buffer_size_ == 0) | 
 |     return transport_->Read(buf, buf_len, callback); | 
 |  | 
 |   if (buf_len < buffer_size_) | 
 |     return ERR_UNEXPECTED; | 
 |  | 
 |   state_ = STATE_READ; | 
 |   user_read_buf_ = buf; | 
 |   int result = DoLoop(OK); | 
 |   if (result == ERR_IO_PENDING) | 
 |     user_read_callback_ = callback; | 
 |   else | 
 |     user_read_buf_ = NULL; | 
 |   return result; | 
 | } | 
 |  | 
 | int ReadBufferingStreamSocket::DoLoop(int result) { | 
 |   int rv = result; | 
 |   do { | 
 |     State current_state = state_; | 
 |     state_ = STATE_NONE; | 
 |     switch (current_state) { | 
 |       case STATE_READ: | 
 |         rv = DoRead(); | 
 |         break; | 
 |       case STATE_READ_COMPLETE: | 
 |         rv = DoReadComplete(rv); | 
 |         break; | 
 |       case STATE_NONE: | 
 |       default: | 
 |         NOTREACHED() << "Unexpected state: " << current_state; | 
 |         rv = ERR_UNEXPECTED; | 
 |         break; | 
 |     } | 
 |   } while (rv != ERR_IO_PENDING && state_ != STATE_NONE); | 
 |   return rv; | 
 | } | 
 |  | 
 | int ReadBufferingStreamSocket::DoRead() { | 
 |   state_ = STATE_READ_COMPLETE; | 
 |   int rv = | 
 |       transport_->Read(read_buffer_.get(), | 
 |                        read_buffer_->RemainingCapacity(), | 
 |                        base::Bind(&ReadBufferingStreamSocket::OnReadCompleted, | 
 |                                   base::Unretained(this))); | 
 |   return rv; | 
 | } | 
 |  | 
 | int ReadBufferingStreamSocket::DoReadComplete(int result) { | 
 |   state_ = STATE_NONE; | 
 |   if (result <= 0) | 
 |     return result; | 
 |  | 
 |   read_buffer_->set_offset(read_buffer_->offset() + result); | 
 |   if (read_buffer_->RemainingCapacity() > 0) { | 
 |     state_ = STATE_READ; | 
 |     return OK; | 
 |   } | 
 |  | 
 |   memcpy(user_read_buf_->data(), | 
 |          read_buffer_->StartOfBuffer(), | 
 |          read_buffer_->capacity()); | 
 |   read_buffer_->set_offset(0); | 
 |   return read_buffer_->capacity(); | 
 | } | 
 |  | 
 | void ReadBufferingStreamSocket::OnReadCompleted(int result) { | 
 |   result = DoLoop(result); | 
 |   if (result == ERR_IO_PENDING) | 
 |     return; | 
 |  | 
 |   user_read_buf_ = NULL; | 
 |   base::ResetAndReturn(&user_read_callback_).Run(result); | 
 | } | 
 |  | 
 | // Simulates synchronously receiving an error during Read() or Write() | 
 | class SynchronousErrorStreamSocket : public WrappedStreamSocket { | 
 |  public: | 
 |   explicit SynchronousErrorStreamSocket(scoped_ptr<StreamSocket> transport); | 
 |   virtual ~SynchronousErrorStreamSocket() {} | 
 |  | 
 |   // Socket implementation: | 
 |   virtual int Read(IOBuffer* buf, | 
 |                    int buf_len, | 
 |                    const CompletionCallback& callback) OVERRIDE; | 
 |   virtual int Write(IOBuffer* buf, | 
 |                     int buf_len, | 
 |                     const CompletionCallback& callback) OVERRIDE; | 
 |  | 
 |   // Sets the next Read() call and all future calls to return |error|. | 
 |   // If there is already a pending asynchronous read, the configured error | 
 |   // will not be returned until that asynchronous read has completed and Read() | 
 |   // is called again. | 
 |   void SetNextReadError(int error) { | 
 |     DCHECK_GE(0, error); | 
 |     have_read_error_ = true; | 
 |     pending_read_error_ = error; | 
 |   } | 
 |  | 
 |   // Sets the next Write() call and all future calls to return |error|. | 
 |   // If there is already a pending asynchronous write, the configured error | 
 |   // will not be returned until that asynchronous write has completed and | 
 |   // Write() is called again. | 
 |   void SetNextWriteError(int error) { | 
 |     DCHECK_GE(0, error); | 
 |     have_write_error_ = true; | 
 |     pending_write_error_ = error; | 
 |   } | 
 |  | 
 |  private: | 
 |   bool have_read_error_; | 
 |   int pending_read_error_; | 
 |  | 
 |   bool have_write_error_; | 
 |   int pending_write_error_; | 
 |  | 
 |   DISALLOW_COPY_AND_ASSIGN(SynchronousErrorStreamSocket); | 
 | }; | 
 |  | 
 | SynchronousErrorStreamSocket::SynchronousErrorStreamSocket( | 
 |     scoped_ptr<StreamSocket> transport) | 
 |     : WrappedStreamSocket(transport.Pass()), | 
 |       have_read_error_(false), | 
 |       pending_read_error_(OK), | 
 |       have_write_error_(false), | 
 |       pending_write_error_(OK) {} | 
 |  | 
 | int SynchronousErrorStreamSocket::Read(IOBuffer* buf, | 
 |                                        int buf_len, | 
 |                                        const CompletionCallback& callback) { | 
 |   if (have_read_error_) | 
 |     return pending_read_error_; | 
 |   return transport_->Read(buf, buf_len, callback); | 
 | } | 
 |  | 
 | int SynchronousErrorStreamSocket::Write(IOBuffer* buf, | 
 |                                         int buf_len, | 
 |                                         const CompletionCallback& callback) { | 
 |   if (have_write_error_) | 
 |     return pending_write_error_; | 
 |   return transport_->Write(buf, buf_len, callback); | 
 | } | 
 |  | 
 | // FakeBlockingStreamSocket wraps an existing StreamSocket and simulates the | 
 | // underlying transport needing to complete things asynchronously in a | 
 | // deterministic manner (e.g.: independent of the TestServer and the OS's | 
 | // semantics). | 
 | class FakeBlockingStreamSocket : public WrappedStreamSocket { | 
 |  public: | 
 |   explicit FakeBlockingStreamSocket(scoped_ptr<StreamSocket> transport); | 
 |   virtual ~FakeBlockingStreamSocket() {} | 
 |  | 
 |   // Socket implementation: | 
 |   virtual int Read(IOBuffer* buf, | 
 |                    int buf_len, | 
 |                    const CompletionCallback& callback) OVERRIDE; | 
 |   virtual int Write(IOBuffer* buf, | 
 |                     int buf_len, | 
 |                     const CompletionCallback& callback) OVERRIDE; | 
 |  | 
 |   // Blocks read results on the socket. Reads will not complete until | 
 |   // UnblockReadResult() has been called and a result is ready from the | 
 |   // underlying transport. Note: if BlockReadResult() is called while there is a | 
 |   // hanging asynchronous Read(), that Read is blocked. | 
 |   void BlockReadResult(); | 
 |   void UnblockReadResult(); | 
 |  | 
 |   // Waits for the blocked Read() call to be complete at the underlying | 
 |   // transport. | 
 |   void WaitForReadResult(); | 
 |  | 
 |   // Causes the next call to Write() to return ERR_IO_PENDING, not beginning the | 
 |   // underlying transport until UnblockWrite() has been called. Note: if there | 
 |   // is a pending asynchronous write, it is NOT blocked. For purposes of | 
 |   // blocking writes, data is considered to have reached the underlying | 
 |   // transport as soon as Write() is called. | 
 |   void BlockWrite(); | 
 |   void UnblockWrite(); | 
 |  | 
 |   // Waits for the blocked Write() call to be scheduled. | 
 |   void WaitForWrite(); | 
 |  | 
 |   // Returns the wrapped stream socket. | 
 |   StreamSocket* transport() { return transport_.get(); } | 
 |  | 
 |  private: | 
 |   // Handles completion from the underlying transport read. | 
 |   void OnReadCompleted(int result); | 
 |  | 
 |   // True if read callbacks are blocked. | 
 |   bool should_block_read_; | 
 |  | 
 |   // The user callback for the pending read call. | 
 |   CompletionCallback pending_read_callback_; | 
 |  | 
 |   // The result for the blocked read callback, or ERR_IO_PENDING if not | 
 |   // completed. | 
 |   int pending_read_result_; | 
 |  | 
 |   // WaitForReadResult() wait loop. | 
 |   scoped_ptr<base::RunLoop> read_loop_; | 
 |  | 
 |   // True if write calls are blocked. | 
 |   bool should_block_write_; | 
 |  | 
 |   // The buffer for the pending write, or NULL if not scheduled. | 
 |   scoped_refptr<IOBuffer> pending_write_buf_; | 
 |  | 
 |   // The callback for the pending write call. | 
 |   CompletionCallback pending_write_callback_; | 
 |  | 
 |   // The length for the pending write, or -1 if not scheduled. | 
 |   int pending_write_len_; | 
 |  | 
 |   // WaitForWrite() wait loop. | 
 |   scoped_ptr<base::RunLoop> write_loop_; | 
 | }; | 
 |  | 
 | FakeBlockingStreamSocket::FakeBlockingStreamSocket( | 
 |     scoped_ptr<StreamSocket> transport) | 
 |     : WrappedStreamSocket(transport.Pass()), | 
 |       should_block_read_(false), | 
 |       pending_read_result_(ERR_IO_PENDING), | 
 |       should_block_write_(false), | 
 |       pending_write_len_(-1) {} | 
 |  | 
 | int FakeBlockingStreamSocket::Read(IOBuffer* buf, | 
 |                                    int len, | 
 |                                    const CompletionCallback& callback) { | 
 |   DCHECK(pending_read_callback_.is_null()); | 
 |   DCHECK_EQ(ERR_IO_PENDING, pending_read_result_); | 
 |   DCHECK(!callback.is_null()); | 
 |  | 
 |   int rv = transport_->Read(buf, len, base::Bind( | 
 |       &FakeBlockingStreamSocket::OnReadCompleted, base::Unretained(this))); | 
 |   if (rv == ERR_IO_PENDING) { | 
 |     // Save the callback to be called later. | 
 |     pending_read_callback_ = callback; | 
 |   } else if (should_block_read_) { | 
 |     // Save the callback and read result to be called later. | 
 |     pending_read_callback_ = callback; | 
 |     OnReadCompleted(rv); | 
 |     rv = ERR_IO_PENDING; | 
 |   } | 
 |   return rv; | 
 | } | 
 |  | 
 | int FakeBlockingStreamSocket::Write(IOBuffer* buf, | 
 |                                     int len, | 
 |                                     const CompletionCallback& callback) { | 
 |   DCHECK(buf); | 
 |   DCHECK_LE(0, len); | 
 |  | 
 |   if (!should_block_write_) | 
 |     return transport_->Write(buf, len, callback); | 
 |  | 
 |   // Schedule the write, but do nothing. | 
 |   DCHECK(!pending_write_buf_.get()); | 
 |   DCHECK_EQ(-1, pending_write_len_); | 
 |   DCHECK(pending_write_callback_.is_null()); | 
 |   DCHECK(!callback.is_null()); | 
 |   pending_write_buf_ = buf; | 
 |   pending_write_len_ = len; | 
 |   pending_write_callback_ = callback; | 
 |  | 
 |   // Stop the write loop, if any. | 
 |   if (write_loop_) | 
 |     write_loop_->Quit(); | 
 |   return ERR_IO_PENDING; | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::BlockReadResult() { | 
 |   DCHECK(!should_block_read_); | 
 |   should_block_read_ = true; | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::UnblockReadResult() { | 
 |   DCHECK(should_block_read_); | 
 |   should_block_read_ = false; | 
 |  | 
 |   // If the operation is still pending in the underlying transport, immediately | 
 |   // return - OnReadCompleted() will handle invoking the callback once the | 
 |   // transport has completed. | 
 |   if (pending_read_result_ == ERR_IO_PENDING) | 
 |     return; | 
 |   int result = pending_read_result_; | 
 |   pending_read_result_ = ERR_IO_PENDING; | 
 |   base::ResetAndReturn(&pending_read_callback_).Run(result); | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::WaitForReadResult() { | 
 |   DCHECK(should_block_read_); | 
 |   DCHECK(!read_loop_); | 
 |  | 
 |   if (pending_read_result_ != ERR_IO_PENDING) | 
 |     return; | 
 |   read_loop_.reset(new base::RunLoop); | 
 |   read_loop_->Run(); | 
 |   read_loop_.reset(); | 
 |   DCHECK_NE(ERR_IO_PENDING, pending_read_result_); | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::BlockWrite() { | 
 |   DCHECK(!should_block_write_); | 
 |   should_block_write_ = true; | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::UnblockWrite() { | 
 |   DCHECK(should_block_write_); | 
 |   should_block_write_ = false; | 
 |  | 
 |   // Do nothing if UnblockWrite() was called after BlockWrite(), | 
 |   // without a Write() in between. | 
 |   if (!pending_write_buf_.get()) | 
 |     return; | 
 |  | 
 |   int rv = transport_->Write( | 
 |       pending_write_buf_.get(), pending_write_len_, pending_write_callback_); | 
 |   pending_write_buf_ = NULL; | 
 |   pending_write_len_ = -1; | 
 |   if (rv == ERR_IO_PENDING) { | 
 |     pending_write_callback_.Reset(); | 
 |   } else { | 
 |     base::ResetAndReturn(&pending_write_callback_).Run(rv); | 
 |   } | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::WaitForWrite() { | 
 |   DCHECK(should_block_write_); | 
 |   DCHECK(!write_loop_); | 
 |  | 
 |   if (pending_write_buf_.get()) | 
 |     return; | 
 |   write_loop_.reset(new base::RunLoop); | 
 |   write_loop_->Run(); | 
 |   write_loop_.reset(); | 
 |   DCHECK(pending_write_buf_.get()); | 
 | } | 
 |  | 
 | void FakeBlockingStreamSocket::OnReadCompleted(int result) { | 
 |   DCHECK_EQ(ERR_IO_PENDING, pending_read_result_); | 
 |   DCHECK(!pending_read_callback_.is_null()); | 
 |  | 
 |   if (should_block_read_) { | 
 |     // Store the result so that the callback can be invoked once Unblock() is | 
 |     // called. | 
 |     pending_read_result_ = result; | 
 |  | 
 |     // Stop the WaitForReadResult() call if any. | 
 |     if (read_loop_) | 
 |       read_loop_->Quit(); | 
 |   } else { | 
 |     // Either the Read() was never blocked or UnblockReadResult() was called | 
 |     // before the Read() completed. Either way, run the callback. | 
 |     base::ResetAndReturn(&pending_read_callback_).Run(result); | 
 |   } | 
 | } | 
 |  | 
 | // CountingStreamSocket wraps an existing StreamSocket and maintains a count of | 
 | // reads and writes on the socket. | 
 | class CountingStreamSocket : public WrappedStreamSocket { | 
 |  public: | 
 |   explicit CountingStreamSocket(scoped_ptr<StreamSocket> transport) | 
 |       : WrappedStreamSocket(transport.Pass()), | 
 |         read_count_(0), | 
 |         write_count_(0) {} | 
 |   virtual ~CountingStreamSocket() {} | 
 |  | 
 |   // Socket implementation: | 
 |   virtual int Read(IOBuffer* buf, | 
 |                    int buf_len, | 
 |                    const CompletionCallback& callback) OVERRIDE { | 
 |     read_count_++; | 
 |     return transport_->Read(buf, buf_len, callback); | 
 |   } | 
 |   virtual int Write(IOBuffer* buf, | 
 |                     int buf_len, | 
 |                     const CompletionCallback& callback) OVERRIDE { | 
 |     write_count_++; | 
 |     return transport_->Write(buf, buf_len, callback); | 
 |   } | 
 |  | 
 |   int read_count() const { return read_count_; } | 
 |   int write_count() const { return write_count_; } | 
 |  | 
 |  private: | 
 |   int read_count_; | 
 |   int write_count_; | 
 | }; | 
 |  | 
 | // CompletionCallback that will delete the associated StreamSocket when | 
 | // the callback is invoked. | 
 | class DeleteSocketCallback : public TestCompletionCallbackBase { | 
 |  public: | 
 |   explicit DeleteSocketCallback(StreamSocket* socket) | 
 |       : socket_(socket), | 
 |         callback_(base::Bind(&DeleteSocketCallback::OnComplete, | 
 |                              base::Unretained(this))) {} | 
 |   virtual ~DeleteSocketCallback() {} | 
 |  | 
 |   const CompletionCallback& callback() const { return callback_; } | 
 |  | 
 |  private: | 
 |   void OnComplete(int result) { | 
 |     if (socket_) { | 
 |       delete socket_; | 
 |       socket_ = NULL; | 
 |     } else { | 
 |       ADD_FAILURE() << "Deleting socket twice"; | 
 |     } | 
 |     SetResult(result); | 
 |   } | 
 |  | 
 |   StreamSocket* socket_; | 
 |   CompletionCallback callback_; | 
 |  | 
 |   DISALLOW_COPY_AND_ASSIGN(DeleteSocketCallback); | 
 | }; | 
 |  | 
 | // A ChannelIDStore that always returns an error when asked for a | 
 | // channel id. | 
 | class FailingChannelIDStore : public ChannelIDStore { | 
 |   virtual int GetChannelID(const std::string& server_identifier, | 
 |                            base::Time* expiration_time, | 
 |                            std::string* private_key_result, | 
 |                            std::string* cert_result, | 
 |                            const GetChannelIDCallback& callback) OVERRIDE { | 
 |     return ERR_UNEXPECTED; | 
 |   } | 
 |   virtual void SetChannelID(const std::string& server_identifier, | 
 |                             base::Time creation_time, | 
 |                             base::Time expiration_time, | 
 |                             const std::string& private_key, | 
 |                             const std::string& cert) OVERRIDE {} | 
 |   virtual void DeleteChannelID(const std::string& server_identifier, | 
 |                                const base::Closure& completion_callback) | 
 |       OVERRIDE {} | 
 |   virtual void DeleteAllCreatedBetween(base::Time delete_begin, | 
 |                                        base::Time delete_end, | 
 |                                        const base::Closure& completion_callback) | 
 |       OVERRIDE {} | 
 |   virtual void DeleteAll(const base::Closure& completion_callback) OVERRIDE {} | 
 |   virtual void GetAllChannelIDs(const GetChannelIDListCallback& callback) | 
 |       OVERRIDE {} | 
 |   virtual int GetChannelIDCount() OVERRIDE { return 0; } | 
 |   virtual void SetForceKeepSessionState() OVERRIDE {} | 
 | }; | 
 |  | 
 | // A ChannelIDStore that asynchronously returns an error when asked for a | 
 | // channel id. | 
 | class AsyncFailingChannelIDStore : public ChannelIDStore { | 
 |   virtual int GetChannelID(const std::string& server_identifier, | 
 |                            base::Time* expiration_time, | 
 |                            std::string* private_key_result, | 
 |                            std::string* cert_result, | 
 |                            const GetChannelIDCallback& callback) OVERRIDE { | 
 |     base::MessageLoop::current()->PostTask( | 
 |         FROM_HERE, base::Bind(callback, ERR_UNEXPECTED, | 
 |                               server_identifier, base::Time(), "", "")); | 
 |     return ERR_IO_PENDING; | 
 |   } | 
 |   virtual void SetChannelID(const std::string& server_identifier, | 
 |                             base::Time creation_time, | 
 |                             base::Time expiration_time, | 
 |                             const std::string& private_key, | 
 |                             const std::string& cert) OVERRIDE {} | 
 |   virtual void DeleteChannelID(const std::string& server_identifier, | 
 |                                const base::Closure& completion_callback) | 
 |       OVERRIDE {} | 
 |   virtual void DeleteAllCreatedBetween(base::Time delete_begin, | 
 |                                        base::Time delete_end, | 
 |                                        const base::Closure& completion_callback) | 
 |       OVERRIDE {} | 
 |   virtual void DeleteAll(const base::Closure& completion_callback) OVERRIDE {} | 
 |   virtual void GetAllChannelIDs(const GetChannelIDListCallback& callback) | 
 |       OVERRIDE {} | 
 |   virtual int GetChannelIDCount() OVERRIDE { return 0; } | 
 |   virtual void SetForceKeepSessionState() OVERRIDE {} | 
 | }; | 
 |  | 
 | // A mock CTVerifier that records every call to Verify but doesn't verify | 
 | // anything. | 
 | class MockCTVerifier : public CTVerifier { | 
 |  public: | 
 |   MOCK_METHOD5(Verify, int(X509Certificate*, | 
 |                            const std::string&, | 
 |                            const std::string&, | 
 |                            ct::CTVerifyResult*, | 
 |                            const BoundNetLog&)); | 
 | }; | 
 |  | 
 | class SSLClientSocketTest : public PlatformTest { | 
 |  public: | 
 |   SSLClientSocketTest() | 
 |       : socket_factory_(ClientSocketFactory::GetDefaultFactory()), | 
 |         cert_verifier_(new MockCertVerifier), | 
 |         transport_security_state_(new TransportSecurityState), | 
 |         ran_handshake_completion_callback_(false) { | 
 |     cert_verifier_->set_default_result(OK); | 
 |     context_.cert_verifier = cert_verifier_.get(); | 
 |     context_.transport_security_state = transport_security_state_.get(); | 
 |   } | 
 |  | 
 |   void RecordCompletedHandshake() { ran_handshake_completion_callback_ = true; } | 
 |  | 
 |  protected: | 
 |   // The address of the spawned test server, after calling StartTestServer(). | 
 |   const AddressList& addr() const { return addr_; } | 
 |  | 
 |   // The SpawnedTestServer object, after calling StartTestServer(). | 
 |   const SpawnedTestServer* test_server() const { return test_server_.get(); } | 
 |  | 
 |   void SetCTVerifier(CTVerifier* ct_verifier) { | 
 |     context_.cert_transparency_verifier = ct_verifier; | 
 |   } | 
 |  | 
 |   // Starts the test server with SSL configuration |ssl_options|. Returns true | 
 |   // on success. | 
 |   bool StartTestServer(const SpawnedTestServer::SSLOptions& ssl_options) { | 
 |     test_server_.reset(new SpawnedTestServer( | 
 |         SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath())); | 
 |     if (!test_server_->Start()) { | 
 |       LOG(ERROR) << "Could not start SpawnedTestServer"; | 
 |       return false; | 
 |     } | 
 |  | 
 |     if (!test_server_->GetAddressList(&addr_)) { | 
 |       LOG(ERROR) << "Could not get SpawnedTestServer address list"; | 
 |       return false; | 
 |     } | 
 |     return true; | 
 |   } | 
 |  | 
 |   // Sets up a TCP connection to a HTTPS server. To actually do the SSL | 
 |   // handshake, follow up with call to CreateAndConnectSSLClientSocket() below. | 
 |   bool ConnectToTestServer(const SpawnedTestServer::SSLOptions& ssl_options) { | 
 |     if (!StartTestServer(ssl_options)) | 
 |       return false; | 
 |  | 
 |     transport_.reset(new TCPClientSocket(addr_, &log_, NetLog::Source())); | 
 |     int rv = callback_.GetResult(transport_->Connect(callback_.callback())); | 
 |     if (rv != OK) { | 
 |       LOG(ERROR) << "Could not connect to SpawnedTestServer"; | 
 |       return false; | 
 |     } | 
 |     return true; | 
 |   } | 
 |  | 
 |   scoped_ptr<SSLClientSocket> CreateSSLClientSocket( | 
 |       scoped_ptr<StreamSocket> transport_socket, | 
 |       const HostPortPair& host_and_port, | 
 |       const SSLConfig& ssl_config) { | 
 |     scoped_ptr<ClientSocketHandle> connection(new ClientSocketHandle); | 
 |     connection->SetSocket(transport_socket.Pass()); | 
 |     return socket_factory_->CreateSSLClientSocket( | 
 |         connection.Pass(), host_and_port, ssl_config, context_); | 
 |   } | 
 |  | 
 |   // Create an SSLClientSocket object and use it to connect to a test | 
 |   // server, then wait for connection results. This must be called after | 
 |   // a successful ConnectToTestServer() call. | 
 |   // |ssl_config| the SSL configuration to use. | 
 |   // |result| will retrieve the ::Connect() result value. | 
 |   // Returns true on success, false otherwise. Success means that the socket | 
 |   // could be created and its Connect() was called, not that the connection | 
 |   // itself was a success. | 
 |   bool CreateAndConnectSSLClientSocket(SSLConfig& ssl_config, int* result) { | 
 |     sock_ = CreateSSLClientSocket( | 
 |         transport_.Pass(), test_server_->host_port_pair(), ssl_config); | 
 |  | 
 |     if (sock_->IsConnected()) { | 
 |       LOG(ERROR) << "SSL Socket prematurely connected"; | 
 |       return false; | 
 |     } | 
 |  | 
 |     *result = callback_.GetResult(sock_->Connect(callback_.callback())); | 
 |     return true; | 
 |   } | 
 |  | 
 |   ClientSocketFactory* socket_factory_; | 
 |   scoped_ptr<MockCertVerifier> cert_verifier_; | 
 |   scoped_ptr<TransportSecurityState> transport_security_state_; | 
 |   SSLClientSocketContext context_; | 
 |   scoped_ptr<SSLClientSocket> sock_; | 
 |   CapturingNetLog log_; | 
 |   bool ran_handshake_completion_callback_; | 
 |  | 
 |  private: | 
 |   scoped_ptr<StreamSocket> transport_; | 
 |   scoped_ptr<SpawnedTestServer> test_server_; | 
 |   TestCompletionCallback callback_; | 
 |   AddressList addr_; | 
 | }; | 
 |  | 
 | // Verifies the correctness of GetSSLCertRequestInfo. | 
 | class SSLClientSocketCertRequestInfoTest : public SSLClientSocketTest { | 
 |  protected: | 
 |   // Creates a test server with the given SSLOptions, connects to it and returns | 
 |   // the SSLCertRequestInfo reported by the socket. | 
 |   scoped_refptr<SSLCertRequestInfo> GetCertRequest( | 
 |       SpawnedTestServer::SSLOptions ssl_options) { | 
 |     SpawnedTestServer test_server( | 
 |         SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |     if (!test_server.Start()) | 
 |       return NULL; | 
 |  | 
 |     AddressList addr; | 
 |     if (!test_server.GetAddressList(&addr)) | 
 |       return NULL; | 
 |  | 
 |     TestCompletionCallback callback; | 
 |     CapturingNetLog log; | 
 |     scoped_ptr<StreamSocket> transport( | 
 |         new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |     int rv = transport->Connect(callback.callback()); | 
 |     if (rv == ERR_IO_PENDING) | 
 |       rv = callback.WaitForResult(); | 
 |     EXPECT_EQ(OK, rv); | 
 |  | 
 |     scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |         transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |     EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |     rv = sock->Connect(callback.callback()); | 
 |     if (rv == ERR_IO_PENDING) | 
 |       rv = callback.WaitForResult(); | 
 |     scoped_refptr<SSLCertRequestInfo> request_info = new SSLCertRequestInfo(); | 
 |     sock->GetSSLCertRequestInfo(request_info.get()); | 
 |     sock->Disconnect(); | 
 |     EXPECT_FALSE(sock->IsConnected()); | 
 |     EXPECT_TRUE( | 
 |         test_server.host_port_pair().Equals(request_info->host_and_port)); | 
 |  | 
 |     return request_info; | 
 |   } | 
 | }; | 
 |  | 
 | class SSLClientSocketFalseStartTest : public SSLClientSocketTest { | 
 |  public: | 
 |   SSLClientSocketFalseStartTest() | 
 |       : monitor_handshake_callback_(false), | 
 |         fail_handshake_after_false_start_(false) {} | 
 |  | 
 |  protected: | 
 |   // Creates an SSLClientSocket with |client_config| attached to a | 
 |   // FakeBlockingStreamSocket, returning both in |*out_raw_transport| and | 
 |   // |*out_sock|. The FakeBlockingStreamSocket is owned by the SSLClientSocket, | 
 |   // so |*out_raw_transport| is a raw pointer. | 
 |   // | 
 |   // The client socket will begin a connect using |callback| but stop before the | 
 |   // server's finished message is received. The finished message will be blocked | 
 |   // in |*out_raw_transport|. To complete the handshake and successfully read | 
 |   // data, the caller must unblock reads on |*out_raw_transport|. (Note that, if | 
 |   // the client successfully false started, |callback.WaitForResult()| will | 
 |   // return OK without unblocking transport reads. But Read() will still block.) | 
 |   // | 
 |   // Must be called after StartTestServer is called. | 
 |   void CreateAndConnectUntilServerFinishedReceived( | 
 |       const SSLConfig& client_config, | 
 |       TestCompletionCallback* callback, | 
 |       FakeBlockingStreamSocket** out_raw_transport, | 
 |       scoped_ptr<SSLClientSocket>* out_sock) { | 
 |     CHECK(test_server()); | 
 |  | 
 |     scoped_ptr<StreamSocket> real_transport(scoped_ptr<StreamSocket>( | 
 |         new TCPClientSocket(addr(), NULL, NetLog::Source()))); | 
 |     real_transport.reset( | 
 |         new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |  | 
 |     scoped_ptr<FakeBlockingStreamSocket> transport( | 
 |         new FakeBlockingStreamSocket(real_transport.Pass())); | 
 |     int rv = callback->GetResult(transport->Connect(callback->callback())); | 
 |     EXPECT_EQ(OK, rv); | 
 |  | 
 |     FakeBlockingStreamSocket* raw_transport = transport.get(); | 
 |     scoped_ptr<SSLClientSocket> sock = | 
 |         CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                               test_server()->host_port_pair(), | 
 |                               client_config); | 
 |  | 
 |     if (monitor_handshake_callback_) { | 
 |       sock->SetHandshakeCompletionCallback( | 
 |           base::Bind(&SSLClientSocketTest::RecordCompletedHandshake, | 
 |                      base::Unretained(this))); | 
 |     } | 
 |  | 
 |     // Connect. Stop before the client processes the first server leg | 
 |     // (ServerHello, etc.) | 
 |     raw_transport->BlockReadResult(); | 
 |     rv = sock->Connect(callback->callback()); | 
 |     EXPECT_EQ(ERR_IO_PENDING, rv); | 
 |     raw_transport->WaitForReadResult(); | 
 |  | 
 |     // Release the ServerHello and wait for the client to write | 
 |     // ClientKeyExchange, etc. (A proxy for waiting for the entirety of the | 
 |     // server's leg to complete, since it may span multiple reads.) | 
 |     EXPECT_FALSE(callback->have_result()); | 
 |     raw_transport->BlockWrite(); | 
 |     raw_transport->UnblockReadResult(); | 
 |     raw_transport->WaitForWrite(); | 
 |  | 
 |     if (fail_handshake_after_false_start_) { | 
 |       SynchronousErrorStreamSocket* error_socket = | 
 |           static_cast<SynchronousErrorStreamSocket*>( | 
 |               raw_transport->transport()); | 
 |       error_socket->SetNextReadError(ERR_CONNECTION_RESET); | 
 |     } | 
 |     // And, finally, release that and block the next server leg | 
 |     // (ChangeCipherSpec, Finished). | 
 |     raw_transport->BlockReadResult(); | 
 |     raw_transport->UnblockWrite(); | 
 |  | 
 |     *out_raw_transport = raw_transport; | 
 |     *out_sock = sock.Pass(); | 
 |   } | 
 |  | 
 |   void TestFalseStart(const SpawnedTestServer::SSLOptions& server_options, | 
 |                       const SSLConfig& client_config, | 
 |                       bool expect_false_start) { | 
 |     ASSERT_TRUE(StartTestServer(server_options)); | 
 |  | 
 |     TestCompletionCallback callback; | 
 |     FakeBlockingStreamSocket* raw_transport = NULL; | 
 |     scoped_ptr<SSLClientSocket> sock; | 
 |  | 
 |     ASSERT_NO_FATAL_FAILURE(CreateAndConnectUntilServerFinishedReceived( | 
 |         client_config, &callback, &raw_transport, &sock)); | 
 |  | 
 |     if (expect_false_start) { | 
 |       // When False Starting, the handshake should complete before receiving the | 
 |       // Change Cipher Spec and Finished messages. | 
 |       // | 
 |       // Note: callback.have_result() may not be true without waiting. The NSS | 
 |       // state machine sometimes lives on a separate thread, so this thread may | 
 |       // not yet have processed the signal that the handshake has completed. | 
 |       int rv = callback.WaitForResult(); | 
 |       EXPECT_EQ(OK, rv); | 
 |       EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |       const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |       static const int kRequestTextSize = | 
 |           static_cast<int>(arraysize(request_text) - 1); | 
 |       scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestTextSize)); | 
 |       memcpy(request_buffer->data(), request_text, kRequestTextSize); | 
 |  | 
 |       // Write the request. | 
 |       rv = callback.GetResult(sock->Write(request_buffer.get(), | 
 |                                           kRequestTextSize, | 
 |                                           callback.callback())); | 
 |       EXPECT_EQ(kRequestTextSize, rv); | 
 |  | 
 |       // The read will hang; it's waiting for the peer to complete the | 
 |       // handshake, and the handshake is still blocked. | 
 |       scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |       rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |  | 
 |       // After releasing reads, the connection proceeds. | 
 |       raw_transport->UnblockReadResult(); | 
 |       rv = callback.GetResult(rv); | 
 |       if (fail_handshake_after_false_start_) | 
 |         EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 |       else | 
 |         EXPECT_LT(0, rv); | 
 |     } else { | 
 |       // False Start is not enabled, so the handshake will not complete because | 
 |       // the server second leg is blocked. | 
 |       base::RunLoop().RunUntilIdle(); | 
 |       EXPECT_FALSE(callback.have_result()); | 
 |     } | 
 |   } | 
 |  | 
 |   // Indicates that the socket's handshake completion callback should | 
 |   // be monitored. | 
 |   bool monitor_handshake_callback_; | 
 |   // Indicates that this test's handshake should fail after the client | 
 |   // "finished" message is sent. | 
 |   bool fail_handshake_after_false_start_; | 
 | }; | 
 |  | 
 | class SSLClientSocketChannelIDTest : public SSLClientSocketTest { | 
 |  protected: | 
 |   void EnableChannelID() { | 
 |     channel_id_service_.reset( | 
 |         new ChannelIDService(new DefaultChannelIDStore(NULL), | 
 |                              base::MessageLoopProxy::current())); | 
 |     context_.channel_id_service = channel_id_service_.get(); | 
 |   } | 
 |  | 
 |   void EnableFailingChannelID() { | 
 |     channel_id_service_.reset(new ChannelIDService( | 
 |         new FailingChannelIDStore(), base::MessageLoopProxy::current())); | 
 |     context_.channel_id_service = channel_id_service_.get(); | 
 |   } | 
 |  | 
 |   void EnableAsyncFailingChannelID() { | 
 |     channel_id_service_.reset(new ChannelIDService( | 
 |         new AsyncFailingChannelIDStore(), | 
 |         base::MessageLoopProxy::current())); | 
 |     context_.channel_id_service = channel_id_service_.get(); | 
 |   } | 
 |  | 
 |  private: | 
 |   scoped_ptr<ChannelIDService> channel_id_service_; | 
 | }; | 
 |  | 
 | //----------------------------------------------------------------------------- | 
 |  | 
 | // LogContainsSSLConnectEndEvent returns true if the given index in the given | 
 | // log is an SSL connect end event. The NSS sockets will cork in an attempt to | 
 | // merge the first application data record with the Finished message when false | 
 | // starting. However, in order to avoid the server timing out the handshake, | 
 | // they'll give up waiting for application data and send the Finished after a | 
 | // timeout. This means that an SSL connect end event may appear as a socket | 
 | // write. | 
 | static bool LogContainsSSLConnectEndEvent( | 
 |     const CapturingNetLog::CapturedEntryList& log, | 
 |     int i) { | 
 |   return LogContainsEndEvent(log, i, NetLog::TYPE_SSL_CONNECT) || | 
 |          LogContainsEvent( | 
 |              log, i, NetLog::TYPE_SOCKET_BYTES_SENT, NetLog::PHASE_NONE); | 
 | } | 
 |  | 
 | }  // namespace | 
 |  | 
 | TEST_F(SSLClientSocketTest, Connect) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1)); | 
 |  | 
 |   sock->Disconnect(); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, ConnectExpired) { | 
 |   SpawnedTestServer::SSLOptions ssl_options( | 
 |       SpawnedTestServer::SSLOptions::CERT_EXPIRED); | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   cert_verifier_->set_default_result(ERR_CERT_DATE_INVALID); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   EXPECT_EQ(ERR_CERT_DATE_INVALID, rv); | 
 |  | 
 |   // Rather than testing whether or not the underlying socket is connected, | 
 |   // test that the handshake has finished. This is because it may be | 
 |   // desirable to disconnect the socket before showing a user prompt, since | 
 |   // the user may take indefinitely long to respond. | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1)); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, ConnectMismatched) { | 
 |   SpawnedTestServer::SSLOptions ssl_options( | 
 |       SpawnedTestServer::SSLOptions::CERT_MISMATCHED_NAME); | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   cert_verifier_->set_default_result(ERR_CERT_COMMON_NAME_INVALID); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   EXPECT_EQ(ERR_CERT_COMMON_NAME_INVALID, rv); | 
 |  | 
 |   // Rather than testing whether or not the underlying socket is connected, | 
 |   // test that the handshake has finished. This is because it may be | 
 |   // desirable to disconnect the socket before showing a user prompt, since | 
 |   // the user may take indefinitely long to respond. | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1)); | 
 | } | 
 |  | 
 | // Attempt to connect to a page which requests a client certificate. It should | 
 | // return an error code on connect. | 
 | TEST_F(SSLClientSocketTest, ConnectClientAuthCertRequested) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.request_client_certificate = true; | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   log.GetEntries(&entries); | 
 |   // Because we prematurely kill the handshake at CertificateRequest, | 
 |   // the server may still send data (notably the ServerHelloDone) | 
 |   // after the error is returned. As a result, the SSL_CONNECT may not | 
 |   // be the last entry. See http://crbug.com/54445. We use | 
 |   // ExpectLogContainsSomewhere instead of | 
 |   // LogContainsSSLConnectEndEvent to avoid assuming, e.g., only one | 
 |   // extra read instead of two. This occurs before the handshake ends, | 
 |   // so the corking logic of LogContainsSSLConnectEndEvent isn't | 
 |   // necessary. | 
 |   // | 
 |   // TODO(davidben): When SSL_RestartHandshakeAfterCertReq in NSS is | 
 |   // fixed and we can respond to the first CertificateRequest | 
 |   // without closing the socket, add a unit test for sending the | 
 |   // certificate. This test may still be useful as we'll want to close | 
 |   // the socket on a timeout if the user takes a long time to pick a | 
 |   // cert. Related bug: https://bugzilla.mozilla.org/show_bug.cgi?id=542832 | 
 |   ExpectLogContainsSomewhere( | 
 |       entries, 0, NetLog::TYPE_SSL_CONNECT, NetLog::PHASE_END); | 
 |   EXPECT_EQ(ERR_SSL_CLIENT_AUTH_CERT_NEEDED, rv); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | // Connect to a server requesting optional client authentication. Send it a | 
 | // null certificate. It should allow the connection. | 
 | // | 
 | // TODO(davidben): Also test providing an actual certificate. | 
 | TEST_F(SSLClientSocketTest, ConnectClientAuthSendNullCert) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.request_client_certificate = true; | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.send_client_cert = true; | 
 |   ssl_config.client_cert = NULL; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   // Our test server accepts certificate-less connections. | 
 |   // TODO(davidben): Add a test which requires them and verify the error. | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1)); | 
 |  | 
 |   // We responded to the server's certificate request with a Certificate | 
 |   // message with no client certificate in it.  ssl_info.client_cert_sent | 
 |   // should be false in this case. | 
 |   SSLInfo ssl_info; | 
 |   sock->GetSSLInfo(&ssl_info); | 
 |   EXPECT_FALSE(ssl_info.client_cert_sent); | 
 |  | 
 |   sock->Disconnect(); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | // TODO(wtc): Add unit tests for IsConnectedAndIdle: | 
 | //   - Server closes an SSL connection (with a close_notify alert message). | 
 | //   - Server closes the underlying TCP connection directly. | 
 | //   - Server sends data unexpectedly. | 
 |  | 
 | TEST_F(SSLClientSocketTest, Read) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   scoped_refptr<IOBuffer> request_buffer( | 
 |       new IOBuffer(arraysize(request_text) - 1)); | 
 |   memcpy(request_buffer->data(), request_text, arraysize(request_text) - 1); | 
 |  | 
 |   rv = sock->Write( | 
 |       request_buffer.get(), arraysize(request_text) - 1, callback.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(arraysize(request_text) - 1), rv); | 
 |  | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   for (;;) { | 
 |     rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |     EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |     if (rv == ERR_IO_PENDING) | 
 |       rv = callback.WaitForResult(); | 
 |  | 
 |     EXPECT_GE(rv, 0); | 
 |     if (rv <= 0) | 
 |       break; | 
 |   } | 
 | } | 
 |  | 
 | // Tests that SSLClientSocket properly handles when the underlying transport | 
 | // synchronously fails a transport read in during the handshake. The error code | 
 | // should be preserved so SSLv3 fallback logic can condition on it. | 
 | TEST_F(SSLClientSocketTest, Connect_WithSynchronousError) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> transport( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   SynchronousErrorStreamSocket* raw_transport = transport.get(); | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   raw_transport->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | // Tests that the SSLClientSocket properly handles when the underlying transport | 
 | // synchronously returns an error code - such as if an intermediary terminates | 
 | // the socket connection uncleanly. | 
 | // This is a regression test for http://crbug.com/238536 | 
 | TEST_F(SSLClientSocketTest, Read_WithSynchronousError) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> transport( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   SynchronousErrorStreamSocket* raw_transport = transport.get(); | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   static const int kRequestTextSize = | 
 |       static_cast<int>(arraysize(request_text) - 1); | 
 |   scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestTextSize)); | 
 |   memcpy(request_buffer->data(), request_text, kRequestTextSize); | 
 |  | 
 |   rv = callback.GetResult( | 
 |       sock->Write(request_buffer.get(), kRequestTextSize, callback.callback())); | 
 |   EXPECT_EQ(kRequestTextSize, rv); | 
 |  | 
 |   // Simulate an unclean/forcible shutdown. | 
 |   raw_transport->SetNextReadError(ERR_CONNECTION_RESET); | 
 |  | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |  | 
 |   // Note: This test will hang if this bug has regressed. Simply checking that | 
 |   // rv != ERR_IO_PENDING is insufficient, as ERR_IO_PENDING is a legitimate | 
 |   // result when using a dedicated task runner for NSS. | 
 |   rv = callback.GetResult(sock->Read(buf.get(), 4096, callback.callback())); | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 | } | 
 |  | 
 | // Tests that the SSLClientSocket properly handles when the underlying transport | 
 | // asynchronously returns an error code while writing data - such as if an | 
 | // intermediary terminates the socket connection uncleanly. | 
 | // This is a regression test for http://crbug.com/249848 | 
 | TEST_F(SSLClientSocketTest, Write_WithSynchronousError) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   // Note: |error_socket|'s ownership is handed to |transport|, but a pointer | 
 |   // is retained in order to configure additional errors. | 
 |   scoped_ptr<SynchronousErrorStreamSocket> error_socket( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   SynchronousErrorStreamSocket* raw_error_socket = error_socket.get(); | 
 |   scoped_ptr<FakeBlockingStreamSocket> transport( | 
 |       new FakeBlockingStreamSocket(error_socket.PassAs<StreamSocket>())); | 
 |   FakeBlockingStreamSocket* raw_transport = transport.get(); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   static const int kRequestTextSize = | 
 |       static_cast<int>(arraysize(request_text) - 1); | 
 |   scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestTextSize)); | 
 |   memcpy(request_buffer->data(), request_text, kRequestTextSize); | 
 |  | 
 |   // Simulate an unclean/forcible shutdown on the underlying socket. | 
 |   // However, simulate this error asynchronously. | 
 |   raw_error_socket->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |   raw_transport->BlockWrite(); | 
 |  | 
 |   // This write should complete synchronously, because the TLS ciphertext | 
 |   // can be created and placed into the outgoing buffers independent of the | 
 |   // underlying transport. | 
 |   rv = callback.GetResult( | 
 |       sock->Write(request_buffer.get(), kRequestTextSize, callback.callback())); | 
 |   EXPECT_EQ(kRequestTextSize, rv); | 
 |  | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |  | 
 |   rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |   EXPECT_EQ(ERR_IO_PENDING, rv); | 
 |  | 
 |   // Now unblock the outgoing request, having it fail with the connection | 
 |   // being reset. | 
 |   raw_transport->UnblockWrite(); | 
 |  | 
 |   // Note: This will cause an inifite loop if this bug has regressed. Simply | 
 |   // checking that rv != ERR_IO_PENDING is insufficient, as ERR_IO_PENDING | 
 |   // is a legitimate result when using a dedicated task runner for NSS. | 
 |   rv = callback.GetResult(rv); | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 | } | 
 |  | 
 | // If there is a Write failure at the transport with no follow-up Read, although | 
 | // the write error will not be returned to the client until a future Read or | 
 | // Write operation, SSLClientSocket should not spin attempting to re-write on | 
 | // the socket. This is a regression test for part of https://crbug.com/381160. | 
 | TEST_F(SSLClientSocketTest, Write_WithSynchronousErrorNoRead) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   // Note: intermediate sockets' ownership are handed to |sock|, but a pointer | 
 |   // is retained in order to query them. | 
 |   scoped_ptr<SynchronousErrorStreamSocket> error_socket( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   SynchronousErrorStreamSocket* raw_error_socket = error_socket.get(); | 
 |   scoped_ptr<CountingStreamSocket> counting_socket( | 
 |       new CountingStreamSocket(error_socket.PassAs<StreamSocket>())); | 
 |   CountingStreamSocket* raw_counting_socket = counting_socket.get(); | 
 |   int rv = callback.GetResult(counting_socket->Connect(callback.callback())); | 
 |   ASSERT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(counting_socket.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   ASSERT_EQ(OK, rv); | 
 |   ASSERT_TRUE(sock->IsConnected()); | 
 |  | 
 |   // Simulate an unclean/forcible shutdown on the underlying socket. | 
 |   raw_error_socket->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   static const int kRequestTextSize = | 
 |       static_cast<int>(arraysize(request_text) - 1); | 
 |   scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestTextSize)); | 
 |   memcpy(request_buffer->data(), request_text, kRequestTextSize); | 
 |  | 
 |   // This write should complete synchronously, because the TLS ciphertext | 
 |   // can be created and placed into the outgoing buffers independent of the | 
 |   // underlying transport. | 
 |   rv = callback.GetResult( | 
 |       sock->Write(request_buffer.get(), kRequestTextSize, callback.callback())); | 
 |   ASSERT_EQ(kRequestTextSize, rv); | 
 |  | 
 |   // Let the event loop spin for a little bit of time. Even on platforms where | 
 |   // pumping the state machine involve thread hops, there should be no further | 
 |   // writes on the transport socket. | 
 |   // | 
 |   // TODO(davidben): Avoid the arbitrary timeout? | 
 |   int old_write_count = raw_counting_socket->write_count(); | 
 |   base::RunLoop loop; | 
 |   base::MessageLoop::current()->PostDelayedTask( | 
 |       FROM_HERE, loop.QuitClosure(), base::TimeDelta::FromMilliseconds(100)); | 
 |   loop.Run(); | 
 |   EXPECT_EQ(old_write_count, raw_counting_socket->write_count()); | 
 | } | 
 |  | 
 | // Test the full duplex mode, with Read and Write pending at the same time. | 
 | // This test also serves as a regression test for http://crbug.com/29815. | 
 | TEST_F(SSLClientSocketTest, Read_FullDuplex) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback;  // Used for everything except Write. | 
 |  | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   // Issue a "hanging" Read first. | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |   // We haven't written the request, so there should be no response yet. | 
 |   ASSERT_EQ(ERR_IO_PENDING, rv); | 
 |  | 
 |   // Write the request. | 
 |   // The request is padded with a User-Agent header to a size that causes the | 
 |   // memio circular buffer (4k bytes) in SSLClientSocketNSS to wrap around. | 
 |   // This tests the fix for http://crbug.com/29815. | 
 |   std::string request_text = "GET / HTTP/1.1\r\nUser-Agent: long browser name "; | 
 |   for (int i = 0; i < 3770; ++i) | 
 |     request_text.push_back('*'); | 
 |   request_text.append("\r\n\r\n"); | 
 |   scoped_refptr<IOBuffer> request_buffer(new StringIOBuffer(request_text)); | 
 |  | 
 |   TestCompletionCallback callback2;  // Used for Write only. | 
 |   rv = sock->Write( | 
 |       request_buffer.get(), request_text.size(), callback2.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback2.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(request_text.size()), rv); | 
 |  | 
 |   // Now get the Read result. | 
 |   rv = callback.WaitForResult(); | 
 |   EXPECT_GT(rv, 0); | 
 | } | 
 |  | 
 | // Attempts to Read() and Write() from an SSLClientSocketNSS in full duplex | 
 | // mode when the underlying transport is blocked on sending data. When the | 
 | // underlying transport completes due to an error, it should invoke both the | 
 | // Read() and Write() callbacks. If the socket is deleted by the Read() | 
 | // callback, the Write() callback should not be invoked. | 
 | // Regression test for http://crbug.com/232633 | 
 | TEST_F(SSLClientSocketTest, Read_DeleteWhilePendingFullDuplex) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   // Note: |error_socket|'s ownership is handed to |transport|, but a pointer | 
 |   // is retained in order to configure additional errors. | 
 |   scoped_ptr<SynchronousErrorStreamSocket> error_socket( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   SynchronousErrorStreamSocket* raw_error_socket = error_socket.get(); | 
 |   scoped_ptr<FakeBlockingStreamSocket> transport( | 
 |       new FakeBlockingStreamSocket(error_socket.PassAs<StreamSocket>())); | 
 |   FakeBlockingStreamSocket* raw_transport = transport.get(); | 
 |  | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock = | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   std::string request_text = "GET / HTTP/1.1\r\nUser-Agent: long browser name "; | 
 |   request_text.append(20 * 1024, '*'); | 
 |   request_text.append("\r\n\r\n"); | 
 |   scoped_refptr<DrainableIOBuffer> request_buffer(new DrainableIOBuffer( | 
 |       new StringIOBuffer(request_text), request_text.size())); | 
 |  | 
 |   // Simulate errors being returned from the underlying Read() and Write() ... | 
 |   raw_error_socket->SetNextReadError(ERR_CONNECTION_RESET); | 
 |   raw_error_socket->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |   // ... but have those errors returned asynchronously. Because the Write() will | 
 |   // return first, this will trigger the error. | 
 |   raw_transport->BlockReadResult(); | 
 |   raw_transport->BlockWrite(); | 
 |  | 
 |   // Enqueue a Read() before calling Write(), which should "hang" due to | 
 |   // the ERR_IO_PENDING caused by SetReadShouldBlock() and thus return. | 
 |   SSLClientSocket* raw_sock = sock.get(); | 
 |   DeleteSocketCallback read_callback(sock.release()); | 
 |   scoped_refptr<IOBuffer> read_buf(new IOBuffer(4096)); | 
 |   rv = raw_sock->Read(read_buf.get(), 4096, read_callback.callback()); | 
 |  | 
 |   // Ensure things didn't complete synchronously, otherwise |sock| is invalid. | 
 |   ASSERT_EQ(ERR_IO_PENDING, rv); | 
 |   ASSERT_FALSE(read_callback.have_result()); | 
 |  | 
 | #if !defined(USE_OPENSSL) | 
 |   // NSS follows a pattern where a call to PR_Write will only consume as | 
 |   // much data as it can encode into application data records before the | 
 |   // internal memio buffer is full, which should only fill if writing a large | 
 |   // amount of data and the underlying transport is blocked. Once this happens, | 
 |   // NSS will return (total size of all application data records it wrote) - 1, | 
 |   // with the caller expected to resume with the remaining unsent data. | 
 |   // | 
 |   // This causes SSLClientSocketNSS::Write to return that it wrote some data | 
 |   // before it will return ERR_IO_PENDING, so make an extra call to Write() to | 
 |   // get the socket in the state needed for the test below. | 
 |   // | 
 |   // This is not needed for OpenSSL, because for OpenSSL, | 
 |   // SSL_MODE_ENABLE_PARTIAL_WRITE is not specified - thus | 
 |   // SSLClientSocketOpenSSL::Write() will not return until all of | 
 |   // |request_buffer| has been written to the underlying BIO (although not | 
 |   // necessarily the underlying transport). | 
 |   rv = callback.GetResult(raw_sock->Write(request_buffer.get(), | 
 |                                           request_buffer->BytesRemaining(), | 
 |                                           callback.callback())); | 
 |   ASSERT_LT(0, rv); | 
 |   request_buffer->DidConsume(rv); | 
 |  | 
 |   // Guard to ensure that |request_buffer| was larger than all of the internal | 
 |   // buffers (transport, memio, NSS) along the way - otherwise the next call | 
 |   // to Write() will crash with an invalid buffer. | 
 |   ASSERT_LT(0, request_buffer->BytesRemaining()); | 
 | #endif | 
 |  | 
 |   // Attempt to write the remaining data. NSS will not be able to consume the | 
 |   // application data because the internal buffers are full, while OpenSSL will | 
 |   // return that its blocked because the underlying transport is blocked. | 
 |   rv = raw_sock->Write(request_buffer.get(), | 
 |                        request_buffer->BytesRemaining(), | 
 |                        callback.callback()); | 
 |   ASSERT_EQ(ERR_IO_PENDING, rv); | 
 |   ASSERT_FALSE(callback.have_result()); | 
 |  | 
 |   // Now unblock Write(), which will invoke OnSendComplete and (eventually) | 
 |   // call the Read() callback, deleting the socket and thus aborting calling | 
 |   // the Write() callback. | 
 |   raw_transport->UnblockWrite(); | 
 |  | 
 |   rv = read_callback.WaitForResult(); | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 |  | 
 |   // The Write callback should not have been called. | 
 |   EXPECT_FALSE(callback.have_result()); | 
 | } | 
 |  | 
 | // Tests that the SSLClientSocket does not crash if data is received on the | 
 | // transport socket after a failing write. This can occur if we have a Write | 
 | // error in a SPDY socket. | 
 | // Regression test for http://crbug.com/335557 | 
 | TEST_F(SSLClientSocketTest, Read_WithWriteError) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   // Note: |error_socket|'s ownership is handed to |transport|, but a pointer | 
 |   // is retained in order to configure additional errors. | 
 |   scoped_ptr<SynchronousErrorStreamSocket> error_socket( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   SynchronousErrorStreamSocket* raw_error_socket = error_socket.get(); | 
 |   scoped_ptr<FakeBlockingStreamSocket> transport( | 
 |       new FakeBlockingStreamSocket(error_socket.PassAs<StreamSocket>())); | 
 |   FakeBlockingStreamSocket* raw_transport = transport.get(); | 
 |  | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   // Send a request so there is something to read from the socket. | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   static const int kRequestTextSize = | 
 |       static_cast<int>(arraysize(request_text) - 1); | 
 |   scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestTextSize)); | 
 |   memcpy(request_buffer->data(), request_text, kRequestTextSize); | 
 |  | 
 |   rv = callback.GetResult( | 
 |       sock->Write(request_buffer.get(), kRequestTextSize, callback.callback())); | 
 |   EXPECT_EQ(kRequestTextSize, rv); | 
 |  | 
 |   // Start a hanging read. | 
 |   TestCompletionCallback read_callback; | 
 |   raw_transport->BlockReadResult(); | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   rv = sock->Read(buf.get(), 4096, read_callback.callback()); | 
 |   EXPECT_EQ(ERR_IO_PENDING, rv); | 
 |  | 
 |   // Perform another write, but have it fail. Write a request larger than the | 
 |   // internal socket buffers so that the request hits the underlying transport | 
 |   // socket and detects the error. | 
 |   std::string long_request_text = | 
 |       "GET / HTTP/1.1\r\nUser-Agent: long browser name "; | 
 |   long_request_text.append(20 * 1024, '*'); | 
 |   long_request_text.append("\r\n\r\n"); | 
 |   scoped_refptr<DrainableIOBuffer> long_request_buffer(new DrainableIOBuffer( | 
 |       new StringIOBuffer(long_request_text), long_request_text.size())); | 
 |  | 
 |   raw_error_socket->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |  | 
 |   // Write as much data as possible until hitting an error. This is necessary | 
 |   // for NSS. PR_Write will only consume as much data as it can encode into | 
 |   // application data records before the internal memio buffer is full, which | 
 |   // should only fill if writing a large amount of data and the underlying | 
 |   // transport is blocked. Once this happens, NSS will return (total size of all | 
 |   // application data records it wrote) - 1, with the caller expected to resume | 
 |   // with the remaining unsent data. | 
 |   do { | 
 |     rv = callback.GetResult(sock->Write(long_request_buffer.get(), | 
 |                                         long_request_buffer->BytesRemaining(), | 
 |                                         callback.callback())); | 
 |     if (rv > 0) { | 
 |       long_request_buffer->DidConsume(rv); | 
 |       // Abort if the entire buffer is ever consumed. | 
 |       ASSERT_LT(0, long_request_buffer->BytesRemaining()); | 
 |     } | 
 |   } while (rv > 0); | 
 |  | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 |  | 
 |   // Release the read. | 
 |   raw_transport->UnblockReadResult(); | 
 |   rv = read_callback.WaitForResult(); | 
 |  | 
 | #if defined(USE_OPENSSL) | 
 |   // Should still read bytes despite the write error. | 
 |   EXPECT_LT(0, rv); | 
 | #else | 
 |   // NSS attempts to flush the write buffer in PR_Read on an SSL socket before | 
 |   // pumping the read state machine, unless configured with SSL_ENABLE_FDX, so | 
 |   // the write error stops future reads. | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 | #endif | 
 | } | 
 |  | 
 | // Tests that SSLClientSocket fails the handshake if the underlying | 
 | // transport is cleanly closed. | 
 | TEST_F(SSLClientSocketTest, Connect_WithZeroReturn) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> transport( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SynchronousErrorStreamSocket* raw_transport = transport.get(); | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             kDefaultSSLConfig)); | 
 |  | 
 |   raw_transport->SetNextReadError(0); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(ERR_CONNECTION_CLOSED, rv); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | // Tests that SSLClientSocket cleanly returns a Read of size 0 if the | 
 | // underlying socket is cleanly closed. | 
 | // This is a regression test for https://crbug.com/422246 | 
 | TEST_F(SSLClientSocketTest, Read_WithZeroReturn) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> transport( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to ensure the handshake has completed. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   SynchronousErrorStreamSocket* raw_transport = transport.get(); | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   raw_transport->SetNextReadError(0); | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   rv = callback.GetResult(sock->Read(buf.get(), 4096, callback.callback())); | 
 |   EXPECT_EQ(0, rv); | 
 | } | 
 |  | 
 | // Tests that SSLClientSocket cleanly returns a Read of size 0 if the | 
 | // underlying socket is cleanly closed asynchronously. | 
 | // This is a regression test for https://crbug.com/422246 | 
 | TEST_F(SSLClientSocketTest, Read_WithAsyncZeroReturn) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> error_socket( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   SynchronousErrorStreamSocket* raw_error_socket = error_socket.get(); | 
 |   scoped_ptr<FakeBlockingStreamSocket> transport( | 
 |       new FakeBlockingStreamSocket(error_socket.Pass())); | 
 |   FakeBlockingStreamSocket* raw_transport = transport.get(); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to ensure the handshake has completed. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.Pass(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   raw_error_socket->SetNextReadError(0); | 
 |   raw_transport->BlockReadResult(); | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |   EXPECT_EQ(ERR_IO_PENDING, rv); | 
 |  | 
 |   raw_transport->UnblockReadResult(); | 
 |   rv = callback.GetResult(rv); | 
 |   EXPECT_EQ(0, rv); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, Read_SmallChunks) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   scoped_refptr<IOBuffer> request_buffer( | 
 |       new IOBuffer(arraysize(request_text) - 1)); | 
 |   memcpy(request_buffer->data(), request_text, arraysize(request_text) - 1); | 
 |  | 
 |   rv = sock->Write( | 
 |       request_buffer.get(), arraysize(request_text) - 1, callback.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(arraysize(request_text) - 1), rv); | 
 |  | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(1)); | 
 |   for (;;) { | 
 |     rv = sock->Read(buf.get(), 1, callback.callback()); | 
 |     EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |     if (rv == ERR_IO_PENDING) | 
 |       rv = callback.WaitForResult(); | 
 |  | 
 |     EXPECT_GE(rv, 0); | 
 |     if (rv <= 0) | 
 |       break; | 
 |   } | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, Read_ManySmallRecords) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |  | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<ReadBufferingStreamSocket> transport( | 
 |       new ReadBufferingStreamSocket(real_transport.Pass())); | 
 |   ReadBufferingStreamSocket* raw_transport = transport.get(); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   ASSERT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             kDefaultSSLConfig)); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   ASSERT_EQ(OK, rv); | 
 |   ASSERT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const char request_text[] = "GET /ssl-many-small-records HTTP/1.0\r\n\r\n"; | 
 |   scoped_refptr<IOBuffer> request_buffer( | 
 |       new IOBuffer(arraysize(request_text) - 1)); | 
 |   memcpy(request_buffer->data(), request_text, arraysize(request_text) - 1); | 
 |  | 
 |   rv = callback.GetResult(sock->Write( | 
 |       request_buffer.get(), arraysize(request_text) - 1, callback.callback())); | 
 |   ASSERT_GT(rv, 0); | 
 |   ASSERT_EQ(static_cast<int>(arraysize(request_text) - 1), rv); | 
 |  | 
 |   // Note: This relies on SSLClientSocketNSS attempting to read up to 17K of | 
 |   // data (the max SSL record size) at a time. Ensure that at least 15K worth | 
 |   // of SSL data is buffered first. The 15K of buffered data is made up of | 
 |   // many smaller SSL records (the TestServer writes along 1350 byte | 
 |   // plaintext boundaries), although there may also be a few records that are | 
 |   // smaller or larger, due to timing and SSL False Start. | 
 |   // 15K was chosen because 15K is smaller than the 17K (max) read issued by | 
 |   // the SSLClientSocket implementation, and larger than the minimum amount | 
 |   // of ciphertext necessary to contain the 8K of plaintext requested below. | 
 |   raw_transport->SetBufferSize(15000); | 
 |  | 
 |   scoped_refptr<IOBuffer> buffer(new IOBuffer(8192)); | 
 |   rv = callback.GetResult(sock->Read(buffer.get(), 8192, callback.callback())); | 
 |   ASSERT_EQ(rv, 8192); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, Read_Interrupted) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   scoped_refptr<IOBuffer> request_buffer( | 
 |       new IOBuffer(arraysize(request_text) - 1)); | 
 |   memcpy(request_buffer->data(), request_text, arraysize(request_text) - 1); | 
 |  | 
 |   rv = sock->Write( | 
 |       request_buffer.get(), arraysize(request_text) - 1, callback.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(arraysize(request_text) - 1), rv); | 
 |  | 
 |   // Do a partial read and then exit.  This test should not crash! | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(512)); | 
 |   rv = sock->Read(buf.get(), 512, callback.callback()); | 
 |   EXPECT_TRUE(rv > 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   EXPECT_GT(rv, 0); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, Read_FullLogging) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   log.SetLogLevel(NetLog::LOG_ALL); | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const char request_text[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   scoped_refptr<IOBuffer> request_buffer( | 
 |       new IOBuffer(arraysize(request_text) - 1)); | 
 |   memcpy(request_buffer->data(), request_text, arraysize(request_text) - 1); | 
 |  | 
 |   rv = sock->Write( | 
 |       request_buffer.get(), arraysize(request_text) - 1, callback.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(arraysize(request_text) - 1), rv); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   size_t last_index = ExpectLogContainsSomewhereAfter( | 
 |       entries, 5, NetLog::TYPE_SSL_SOCKET_BYTES_SENT, NetLog::PHASE_NONE); | 
 |  | 
 |   scoped_refptr<IOBuffer> buf(new IOBuffer(4096)); | 
 |   for (;;) { | 
 |     rv = sock->Read(buf.get(), 4096, callback.callback()); | 
 |     EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |     if (rv == ERR_IO_PENDING) | 
 |       rv = callback.WaitForResult(); | 
 |  | 
 |     EXPECT_GE(rv, 0); | 
 |     if (rv <= 0) | 
 |       break; | 
 |  | 
 |     log.GetEntries(&entries); | 
 |     last_index = | 
 |         ExpectLogContainsSomewhereAfter(entries, | 
 |                                         last_index + 1, | 
 |                                         NetLog::TYPE_SSL_SOCKET_BYTES_RECEIVED, | 
 |                                         NetLog::PHASE_NONE); | 
 |   } | 
 | } | 
 |  | 
 | // Regression test for http://crbug.com/42538 | 
 | TEST_F(SSLClientSocketTest, PrematureApplicationData) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   TestCompletionCallback callback; | 
 |  | 
 |   static const unsigned char application_data[] = { | 
 |       0x17, 0x03, 0x01, 0x00, 0x4a, 0x02, 0x00, 0x00, 0x46, 0x03, 0x01, 0x4b, | 
 |       0xc2, 0xf8, 0xb2, 0xc1, 0x56, 0x42, 0xb9, 0x57, 0x7f, 0xde, 0x87, 0x46, | 
 |       0xf7, 0xa3, 0x52, 0x42, 0x21, 0xf0, 0x13, 0x1c, 0x9c, 0x83, 0x88, 0xd6, | 
 |       0x93, 0x0c, 0xf6, 0x36, 0x30, 0x05, 0x7e, 0x20, 0xb5, 0xb5, 0x73, 0x36, | 
 |       0x53, 0x83, 0x0a, 0xfc, 0x17, 0x63, 0xbf, 0xa0, 0xe4, 0x42, 0x90, 0x0d, | 
 |       0x2f, 0x18, 0x6d, 0x20, 0xd8, 0x36, 0x3f, 0xfc, 0xe6, 0x01, 0xfa, 0x0f, | 
 |       0xa5, 0x75, 0x7f, 0x09, 0x00, 0x04, 0x00, 0x16, 0x03, 0x01, 0x11, 0x57, | 
 |       0x0b, 0x00, 0x11, 0x53, 0x00, 0x11, 0x50, 0x00, 0x06, 0x22, 0x30, 0x82, | 
 |       0x06, 0x1e, 0x30, 0x82, 0x05, 0x06, 0xa0, 0x03, 0x02, 0x01, 0x02, 0x02, | 
 |       0x0a}; | 
 |  | 
 |   // All reads and writes complete synchronously (async=false). | 
 |   MockRead data_reads[] = { | 
 |       MockRead(SYNCHRONOUS, | 
 |                reinterpret_cast<const char*>(application_data), | 
 |                arraysize(application_data)), | 
 |       MockRead(SYNCHRONOUS, OK), }; | 
 |  | 
 |   StaticSocketDataProvider data(data_reads, arraysize(data_reads), NULL, 0); | 
 |  | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new MockTCPClientSocket(addr, NULL, &data)); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(ERR_SSL_PROTOCOL_ERROR, rv); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, CipherSuiteDisables) { | 
 |   // Rather than exhaustively disabling every RC4 ciphersuite defined at | 
 |   // http://www.iana.org/assignments/tls-parameters/tls-parameters.xml, | 
 |   // only disabling those cipher suites that the test server actually | 
 |   // implements. | 
 |   const uint16 kCiphersToDisable[] = {0x0005,  // TLS_RSA_WITH_RC4_128_SHA | 
 |   }; | 
 |  | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   // Enable only RC4 on the test server. | 
 |   ssl_options.bulk_ciphers = SpawnedTestServer::SSLOptions::BULK_CIPHER_RC4; | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config; | 
 |   for (size_t i = 0; i < arraysize(kCiphersToDisable); ++i) | 
 |     ssl_config.disabled_cipher_suites.push_back(kCiphersToDisable[i]); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |  | 
 |   // NSS has special handling that maps a handshake_failure alert received | 
 |   // immediately after a client_hello to be a mismatched cipher suite error, | 
 |   // leading to ERR_SSL_VERSION_OR_CIPHER_MISMATCH. When using OpenSSL or | 
 |   // Secure Transport (OS X), the handshake_failure is bubbled up without any | 
 |   // interpretation, leading to ERR_SSL_PROTOCOL_ERROR. Either way, a failure | 
 |   // indicates that no cipher suite was negotiated with the test server. | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_TRUE(rv == ERR_SSL_VERSION_OR_CIPHER_MISMATCH || | 
 |               rv == ERR_SSL_PROTOCOL_ERROR); | 
 |   // The exact ordering differs between SSLClientSocketNSS (which issues an | 
 |   // extra read) and SSLClientSocketMac (which does not). Just make sure the | 
 |   // error appears somewhere in the log. | 
 |   log.GetEntries(&entries); | 
 |   ExpectLogContainsSomewhere( | 
 |       entries, 0, NetLog::TYPE_SSL_HANDSHAKE_ERROR, NetLog::PHASE_NONE); | 
 |  | 
 |   // We cannot test sock->IsConnected(), as the NSS implementation disconnects | 
 |   // the socket when it encounters an error, whereas other implementations | 
 |   // leave it connected. | 
 |   // Because this an error that the test server is mutually aware of, as opposed | 
 |   // to being an error such as a certificate name mismatch, which is | 
 |   // client-only, the exact index of the SSL connect end depends on how | 
 |   // quickly the test server closes the underlying socket. If the test server | 
 |   // closes before the IO message loop pumps messages, there may be a 0-byte | 
 |   // Read event in the NetLog due to TCPClientSocket picking up the EOF. As a | 
 |   // result, the SSL connect end event will be the second-to-last entry, | 
 |   // rather than the last entry. | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1) || | 
 |               LogContainsSSLConnectEndEvent(entries, -2)); | 
 | } | 
 |  | 
 | // When creating an SSLClientSocket, it is allowed to pass in a | 
 | // ClientSocketHandle that is not obtained from a client socket pool. | 
 | // Here we verify that such a simple ClientSocketHandle, not associated with any | 
 | // client socket pool, can be destroyed safely. | 
 | TEST_F(SSLClientSocketTest, ClientSocketHandleNotFromPool) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<ClientSocketHandle> socket_handle(new ClientSocketHandle()); | 
 |   socket_handle->SetSocket(transport.Pass()); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       socket_factory_->CreateSSLClientSocket(socket_handle.Pass(), | 
 |                                              test_server.host_port_pair(), | 
 |                                              kDefaultSSLConfig, | 
 |                                              context_)); | 
 |  | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 | } | 
 |  | 
 | // Verifies that SSLClientSocket::ExportKeyingMaterial return a success | 
 | // code and different keying label results in different keying material. | 
 | TEST_F(SSLClientSocketTest, ExportKeyingMaterial) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |  | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   const int kKeyingMaterialSize = 32; | 
 |   const char* kKeyingLabel1 = "client-socket-test-1"; | 
 |   const char* kKeyingContext = ""; | 
 |   unsigned char client_out1[kKeyingMaterialSize]; | 
 |   memset(client_out1, 0, sizeof(client_out1)); | 
 |   rv = sock->ExportKeyingMaterial( | 
 |       kKeyingLabel1, false, kKeyingContext, client_out1, sizeof(client_out1)); | 
 |   EXPECT_EQ(rv, OK); | 
 |  | 
 |   const char* kKeyingLabel2 = "client-socket-test-2"; | 
 |   unsigned char client_out2[kKeyingMaterialSize]; | 
 |   memset(client_out2, 0, sizeof(client_out2)); | 
 |   rv = sock->ExportKeyingMaterial( | 
 |       kKeyingLabel2, false, kKeyingContext, client_out2, sizeof(client_out2)); | 
 |   EXPECT_EQ(rv, OK); | 
 |   EXPECT_NE(memcmp(client_out1, client_out2, kKeyingMaterialSize), 0); | 
 | } | 
 |  | 
 | // Verifies that SSLClientSocket::ClearSessionCache can be called without | 
 | // explicit NSS initialization. | 
 | TEST(SSLClientSocket, ClearSessionCache) { | 
 |   SSLClientSocket::ClearSessionCache(); | 
 | } | 
 |  | 
 | // Test that the server certificates are properly retrieved from the underlying | 
 | // SSL stack. | 
 | TEST_F(SSLClientSocketTest, VerifyServerChainProperlyOrdered) { | 
 |   // The connection does not have to be successful. | 
 |   cert_verifier_->set_default_result(ERR_CERT_INVALID); | 
 |  | 
 |   // Set up a test server with CERT_CHAIN_WRONG_ROOT. | 
 |   // This makes the server present redundant-server-chain.pem, which contains | 
 |   // intermediate certificates. | 
 |   SpawnedTestServer::SSLOptions ssl_options( | 
 |       SpawnedTestServer::SSLOptions::CERT_CHAIN_WRONG_ROOT); | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   rv = callback.GetResult(rv); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |   rv = sock->Connect(callback.callback()); | 
 |   rv = callback.GetResult(rv); | 
 |  | 
 |   EXPECT_EQ(ERR_CERT_INVALID, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |  | 
 |   // When given option CERT_CHAIN_WRONG_ROOT, SpawnedTestServer will present | 
 |   // certs from redundant-server-chain.pem. | 
 |   CertificateList server_certs = | 
 |       CreateCertificateListFromFile(GetTestCertsDirectory(), | 
 |                                     "redundant-server-chain.pem", | 
 |                                     X509Certificate::FORMAT_AUTO); | 
 |  | 
 |   // Get the server certificate as received client side. | 
 |   scoped_refptr<X509Certificate> server_certificate = | 
 |       sock->GetUnverifiedServerCertificateChain(); | 
 |  | 
 |   // Get the intermediates as received  client side. | 
 |   const X509Certificate::OSCertHandles& server_intermediates = | 
 |       server_certificate->GetIntermediateCertificates(); | 
 |  | 
 |   // Check that the unverified server certificate chain is properly retrieved | 
 |   // from the underlying ssl stack. | 
 |   ASSERT_EQ(4U, server_certs.size()); | 
 |  | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert( | 
 |       server_certificate->os_cert_handle(), server_certs[0]->os_cert_handle())); | 
 |  | 
 |   ASSERT_EQ(3U, server_intermediates.size()); | 
 |  | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(server_intermediates[0], | 
 |                                             server_certs[1]->os_cert_handle())); | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(server_intermediates[1], | 
 |                                             server_certs[2]->os_cert_handle())); | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(server_intermediates[2], | 
 |                                             server_certs[3]->os_cert_handle())); | 
 |  | 
 |   sock->Disconnect(); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | // This tests that SSLInfo contains a properly re-constructed certificate | 
 | // chain. That, in turn, verifies that GetSSLInfo is giving us the chain as | 
 | // verified, not the chain as served by the server. (They may be different.) | 
 | // | 
 | // CERT_CHAIN_WRONG_ROOT is redundant-server-chain.pem. It contains A | 
 | // (end-entity) -> B -> C, and C is signed by D. redundant-validated-chain.pem | 
 | // contains a chain of A -> B -> C2, where C2 is the same public key as C, but | 
 | // a self-signed root. Such a situation can occur when a new root (C2) is | 
 | // cross-certified by an old root (D) and has two different versions of its | 
 | // floating around. Servers may supply C2 as an intermediate, but the | 
 | // SSLClientSocket should return the chain that was verified, from | 
 | // verify_result, instead. | 
 | TEST_F(SSLClientSocketTest, VerifyReturnChainProperlyOrdered) { | 
 |   // By default, cause the CertVerifier to treat all certificates as | 
 |   // expired. | 
 |   cert_verifier_->set_default_result(ERR_CERT_DATE_INVALID); | 
 |  | 
 |   // We will expect SSLInfo to ultimately contain this chain. | 
 |   CertificateList certs = | 
 |       CreateCertificateListFromFile(GetTestCertsDirectory(), | 
 |                                     "redundant-validated-chain.pem", | 
 |                                     X509Certificate::FORMAT_AUTO); | 
 |   ASSERT_EQ(3U, certs.size()); | 
 |  | 
 |   X509Certificate::OSCertHandles temp_intermediates; | 
 |   temp_intermediates.push_back(certs[1]->os_cert_handle()); | 
 |   temp_intermediates.push_back(certs[2]->os_cert_handle()); | 
 |  | 
 |   CertVerifyResult verify_result; | 
 |   verify_result.verified_cert = X509Certificate::CreateFromHandle( | 
 |       certs[0]->os_cert_handle(), temp_intermediates); | 
 |  | 
 |   // Add a rule that maps the server cert (A) to the chain of A->B->C2 | 
 |   // rather than A->B->C. | 
 |   cert_verifier_->AddResultForCert(certs[0].get(), verify_result, OK); | 
 |  | 
 |   // Load and install the root for the validated chain. | 
 |   scoped_refptr<X509Certificate> root_cert = ImportCertFromFile( | 
 |       GetTestCertsDirectory(), "redundant-validated-chain-root.pem"); | 
 |   ASSERT_NE(static_cast<X509Certificate*>(NULL), root_cert.get()); | 
 |   ScopedTestRoot scoped_root(root_cert.get()); | 
 |  | 
 |   // Set up a test server with CERT_CHAIN_WRONG_ROOT. | 
 |   SpawnedTestServer::SSLOptions ssl_options( | 
 |       SpawnedTestServer::SSLOptions::CERT_CHAIN_WRONG_ROOT); | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, | 
 |       ssl_options, | 
 |       base::FilePath(FILE_PATH_LITERAL("net/data/ssl"))); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   CapturingNetLog log; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |   rv = sock->Connect(callback.callback()); | 
 |  | 
 |   CapturingNetLog::CapturedEntryList entries; | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsBeginEvent(entries, 5, NetLog::TYPE_SSL_CONNECT)); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |  | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   log.GetEntries(&entries); | 
 |   EXPECT_TRUE(LogContainsSSLConnectEndEvent(entries, -1)); | 
 |  | 
 |   SSLInfo ssl_info; | 
 |   sock->GetSSLInfo(&ssl_info); | 
 |  | 
 |   // Verify that SSLInfo contains the corrected re-constructed chain A -> B | 
 |   // -> C2. | 
 |   const X509Certificate::OSCertHandles& intermediates = | 
 |       ssl_info.cert->GetIntermediateCertificates(); | 
 |   ASSERT_EQ(2U, intermediates.size()); | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(ssl_info.cert->os_cert_handle(), | 
 |                                             certs[0]->os_cert_handle())); | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(intermediates[0], | 
 |                                             certs[1]->os_cert_handle())); | 
 |   EXPECT_TRUE(X509Certificate::IsSameOSCert(intermediates[1], | 
 |                                             certs[2]->os_cert_handle())); | 
 |  | 
 |   sock->Disconnect(); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketCertRequestInfoTest, NoAuthorities) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.request_client_certificate = true; | 
 |   scoped_refptr<SSLCertRequestInfo> request_info = GetCertRequest(ssl_options); | 
 |   ASSERT_TRUE(request_info.get()); | 
 |   EXPECT_EQ(0u, request_info->cert_authorities.size()); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketCertRequestInfoTest, TwoAuthorities) { | 
 |   const base::FilePath::CharType kThawteFile[] = | 
 |       FILE_PATH_LITERAL("thawte.single.pem"); | 
 |   const unsigned char kThawteDN[] = { | 
 |       0x30, 0x4c, 0x31, 0x0b, 0x30, 0x09, 0x06, 0x03, 0x55, 0x04, 0x06, 0x13, | 
 |       0x02, 0x5a, 0x41, 0x31, 0x25, 0x30, 0x23, 0x06, 0x03, 0x55, 0x04, 0x0a, | 
 |       0x13, 0x1c, 0x54, 0x68, 0x61, 0x77, 0x74, 0x65, 0x20, 0x43, 0x6f, 0x6e, | 
 |       0x73, 0x75, 0x6c, 0x74, 0x69, 0x6e, 0x67, 0x20, 0x28, 0x50, 0x74, 0x79, | 
 |       0x29, 0x20, 0x4c, 0x74, 0x64, 0x2e, 0x31, 0x16, 0x30, 0x14, 0x06, 0x03, | 
 |       0x55, 0x04, 0x03, 0x13, 0x0d, 0x54, 0x68, 0x61, 0x77, 0x74, 0x65, 0x20, | 
 |       0x53, 0x47, 0x43, 0x20, 0x43, 0x41}; | 
 |   const size_t kThawteLen = sizeof(kThawteDN); | 
 |  | 
 |   const base::FilePath::CharType kDiginotarFile[] = | 
 |       FILE_PATH_LITERAL("diginotar_root_ca.pem"); | 
 |   const unsigned char kDiginotarDN[] = { | 
 |       0x30, 0x5f, 0x31, 0x0b, 0x30, 0x09, 0x06, 0x03, 0x55, 0x04, 0x06, 0x13, | 
 |       0x02, 0x4e, 0x4c, 0x31, 0x12, 0x30, 0x10, 0x06, 0x03, 0x55, 0x04, 0x0a, | 
 |       0x13, 0x09, 0x44, 0x69, 0x67, 0x69, 0x4e, 0x6f, 0x74, 0x61, 0x72, 0x31, | 
 |       0x1a, 0x30, 0x18, 0x06, 0x03, 0x55, 0x04, 0x03, 0x13, 0x11, 0x44, 0x69, | 
 |       0x67, 0x69, 0x4e, 0x6f, 0x74, 0x61, 0x72, 0x20, 0x52, 0x6f, 0x6f, 0x74, | 
 |       0x20, 0x43, 0x41, 0x31, 0x20, 0x30, 0x1e, 0x06, 0x09, 0x2a, 0x86, 0x48, | 
 |       0x86, 0xf7, 0x0d, 0x01, 0x09, 0x01, 0x16, 0x11, 0x69, 0x6e, 0x66, 0x6f, | 
 |       0x40, 0x64, 0x69, 0x67, 0x69, 0x6e, 0x6f, 0x74, 0x61, 0x72, 0x2e, 0x6e, | 
 |       0x6c}; | 
 |   const size_t kDiginotarLen = sizeof(kDiginotarDN); | 
 |  | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.request_client_certificate = true; | 
 |   ssl_options.client_authorities.push_back( | 
 |       GetTestClientCertsDirectory().Append(kThawteFile)); | 
 |   ssl_options.client_authorities.push_back( | 
 |       GetTestClientCertsDirectory().Append(kDiginotarFile)); | 
 |   scoped_refptr<SSLCertRequestInfo> request_info = GetCertRequest(ssl_options); | 
 |   ASSERT_TRUE(request_info.get()); | 
 |   ASSERT_EQ(2u, request_info->cert_authorities.size()); | 
 |   EXPECT_EQ(std::string(reinterpret_cast<const char*>(kThawteDN), kThawteLen), | 
 |             request_info->cert_authorities[0]); | 
 |   EXPECT_EQ( | 
 |       std::string(reinterpret_cast<const char*>(kDiginotarDN), kDiginotarLen), | 
 |       request_info->cert_authorities[1]); | 
 | } | 
 |  | 
 | // cert_key_types is currently only populated on OpenSSL. | 
 | #if defined(USE_OPENSSL) | 
 | TEST_F(SSLClientSocketCertRequestInfoTest, CertKeyTypes) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.request_client_certificate = true; | 
 |   ssl_options.client_cert_types.push_back(CLIENT_CERT_RSA_SIGN); | 
 |   ssl_options.client_cert_types.push_back(CLIENT_CERT_ECDSA_SIGN); | 
 |   scoped_refptr<SSLCertRequestInfo> request_info = GetCertRequest(ssl_options); | 
 |   ASSERT_TRUE(request_info.get()); | 
 |   ASSERT_EQ(2u, request_info->cert_key_types.size()); | 
 |   EXPECT_EQ(CLIENT_CERT_RSA_SIGN, request_info->cert_key_types[0]); | 
 |   EXPECT_EQ(CLIENT_CERT_ECDSA_SIGN, request_info->cert_key_types[1]); | 
 | } | 
 | #endif  // defined(USE_OPENSSL) | 
 |  | 
 | TEST_F(SSLClientSocketTest, ConnectSignedCertTimestampsEnabledTLSExtension) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.signed_cert_timestamps_tls_ext = "test"; | 
 |  | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 ssl_options, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log_, NetLog::Source())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.signed_cert_timestamps_enabled = true; | 
 |  | 
 |   MockCTVerifier ct_verifier; | 
 |   SetCTVerifier(&ct_verifier); | 
 |  | 
 |   // Check that the SCT list is extracted as expected. | 
 |   EXPECT_CALL(ct_verifier, Verify(_, "", "test", _, _)).WillRepeatedly( | 
 |       Return(ERR_CT_NO_SCTS_VERIFIED_OK)); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   EXPECT_TRUE(sock->signed_cert_timestamps_received_); | 
 | } | 
 |  | 
 | namespace { | 
 |  | 
 | bool IsValidOCSPResponse(const base::StringPiece& input) { | 
 |   base::StringPiece ocsp_response = input; | 
 |   base::StringPiece sequence, response_status, response_bytes; | 
 |   return asn1::GetElement(&ocsp_response, asn1::kSEQUENCE, &sequence) && | 
 |       ocsp_response.empty() && | 
 |       asn1::GetElement(&sequence, asn1::kENUMERATED, &response_status) && | 
 |       asn1::GetElement(&sequence, | 
 |                        asn1::kContextSpecific | asn1::kConstructed | 0, | 
 |                        &response_status) && | 
 |       sequence.empty(); | 
 | } | 
 |  | 
 | }  // namespace | 
 |  | 
 | // Test that enabling Signed Certificate Timestamps enables OCSP stapling. | 
 | TEST_F(SSLClientSocketTest, ConnectSignedCertTimestampsEnabledOCSP) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.staple_ocsp_response = true; | 
 |   // The test server currently only knows how to generate OCSP responses | 
 |   // for a freshly minted certificate. | 
 |   ssl_options.server_certificate = SpawnedTestServer::SSLOptions::CERT_AUTO; | 
 |  | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 ssl_options, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log_, NetLog::Source())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config; | 
 |   // Enabling Signed Cert Timestamps ensures we request OCSP stapling for | 
 |   // Certificate Transparency verification regardless of whether the platform | 
 |   // is able to process the OCSP status itself. | 
 |   ssl_config.signed_cert_timestamps_enabled = true; | 
 |  | 
 |   MockCTVerifier ct_verifier; | 
 |   SetCTVerifier(&ct_verifier); | 
 |  | 
 |   // Check that the OCSP response is extracted and well-formed. It should be the | 
 |   // DER encoding of an OCSPResponse (RFC 2560), so check that it consists of a | 
 |   // SEQUENCE of an ENUMERATED type and an element tagged with [0] EXPLICIT. In | 
 |   // particular, it should not include the overall two-byte length prefix from | 
 |   // TLS. | 
 |   EXPECT_CALL(ct_verifier, | 
 |               Verify(_, Truly(IsValidOCSPResponse), "", _, _)).WillRepeatedly( | 
 |                   Return(ERR_CT_NO_SCTS_VERIFIED_OK)); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   EXPECT_TRUE(sock->stapled_ocsp_response_received_); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketTest, ConnectSignedCertTimestampsDisabled) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.signed_cert_timestamps_tls_ext = "test"; | 
 |  | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 ssl_options, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, &log_, NetLog::Source())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.signed_cert_timestamps_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   EXPECT_FALSE(sock->signed_cert_timestamps_received_); | 
 | } | 
 |  | 
 | // Tests that IsConnectedAndIdle and WasEverUsed behave as expected. | 
 | TEST_F(SSLClientSocketTest, ReuseStates) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), kDefaultSSLConfig)); | 
 |  | 
 |   rv = sock->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // The socket was just connected. It should be idle because it is speaking | 
 |   // HTTP. Although the transport has been used for the handshake, WasEverUsed() | 
 |   // returns false. | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   EXPECT_TRUE(sock->IsConnectedAndIdle()); | 
 |   EXPECT_FALSE(sock->WasEverUsed()); | 
 |  | 
 |   const char kRequestText[] = "GET / HTTP/1.0\r\n\r\n"; | 
 |   const size_t kRequestLen = arraysize(kRequestText) - 1; | 
 |   scoped_refptr<IOBuffer> request_buffer(new IOBuffer(kRequestLen)); | 
 |   memcpy(request_buffer->data(), kRequestText, kRequestLen); | 
 |  | 
 |   rv = sock->Write(request_buffer.get(), kRequestLen, callback.callback()); | 
 |   EXPECT_TRUE(rv >= 0 || rv == ERR_IO_PENDING); | 
 |  | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(static_cast<int>(kRequestLen), rv); | 
 |  | 
 |   // The socket has now been used. | 
 |   EXPECT_TRUE(sock->WasEverUsed()); | 
 |  | 
 |   // TODO(davidben): Read one byte to ensure the test server has responded and | 
 |   // then assert IsConnectedAndIdle is false. This currently doesn't work | 
 |   // because neither SSLClientSocketNSS nor SSLClientSocketOpenSSL check their | 
 |   // SSL implementation's internal buffers. Either call PR_Available and | 
 |   // SSL_pending, although the former isn't actually implemented or perhaps | 
 |   // attempt to read one byte extra. | 
 | } | 
 |  | 
 | #if defined(USE_OPENSSL) | 
 |  | 
 | TEST_F(SSLClientSocketTest, HandshakeCallbackIsRun_WithFailure) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> real_transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |   scoped_ptr<SynchronousErrorStreamSocket> transport( | 
 |       new SynchronousErrorStreamSocket(real_transport.Pass())); | 
 |   int rv = callback.GetResult(transport->Connect(callback.callback())); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   // Disable TLS False Start to avoid handshake non-determinism. | 
 |   SSLConfig ssl_config; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   SynchronousErrorStreamSocket* raw_transport = transport.get(); | 
 |   scoped_ptr<SSLClientSocket> sock( | 
 |       CreateSSLClientSocket(transport.PassAs<StreamSocket>(), | 
 |                             test_server.host_port_pair(), | 
 |                             ssl_config)); | 
 |  | 
 |   sock->SetHandshakeCompletionCallback(base::Bind( | 
 |       &SSLClientSocketTest::RecordCompletedHandshake, base::Unretained(this))); | 
 |  | 
 |   raw_transport->SetNextWriteError(ERR_CONNECTION_RESET); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |   EXPECT_EQ(ERR_CONNECTION_RESET, rv); | 
 |   EXPECT_FALSE(sock->IsConnected()); | 
 |  | 
 |   EXPECT_TRUE(ran_handshake_completion_callback_); | 
 | } | 
 |  | 
 | // Tests that the completion callback is run when an SSL connection | 
 | // completes successfully. | 
 | TEST_F(SSLClientSocketTest, HandshakeCallbackIsRun_WithSuccess) { | 
 |   SpawnedTestServer test_server(SpawnedTestServer::TYPE_HTTPS, | 
 |                                 SpawnedTestServer::kLocalhost, | 
 |                                 base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |  | 
 |   sock->SetHandshakeCompletionCallback(base::Bind( | 
 |       &SSLClientSocketTest::RecordCompletedHandshake, base::Unretained(this))); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |  | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   EXPECT_TRUE(ran_handshake_completion_callback_); | 
 | } | 
 |  | 
 | // Tests that the completion callback is run with a server that doesn't cache | 
 | // sessions. | 
 | TEST_F(SSLClientSocketTest, HandshakeCallbackIsRun_WithDisabledSessionCache) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |   ssl_options.disable_session_cache = true; | 
 |   SpawnedTestServer test_server( | 
 |       SpawnedTestServer::TYPE_HTTPS, ssl_options, base::FilePath()); | 
 |   ASSERT_TRUE(test_server.Start()); | 
 |  | 
 |   AddressList addr; | 
 |   ASSERT_TRUE(test_server.GetAddressList(&addr)); | 
 |  | 
 |   scoped_ptr<StreamSocket> transport( | 
 |       new TCPClientSocket(addr, NULL, NetLog::Source())); | 
 |  | 
 |   TestCompletionCallback callback; | 
 |   int rv = transport->Connect(callback.callback()); | 
 |   if (rv == ERR_IO_PENDING) | 
 |     rv = callback.WaitForResult(); | 
 |   EXPECT_EQ(OK, rv); | 
 |  | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.false_start_enabled = false; | 
 |  | 
 |   scoped_ptr<SSLClientSocket> sock(CreateSSLClientSocket( | 
 |       transport.Pass(), test_server.host_port_pair(), ssl_config)); | 
 |  | 
 |   sock->SetHandshakeCompletionCallback(base::Bind( | 
 |       &SSLClientSocketTest::RecordCompletedHandshake, base::Unretained(this))); | 
 |  | 
 |   rv = callback.GetResult(sock->Connect(callback.callback())); | 
 |  | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock->IsConnected()); | 
 |   EXPECT_TRUE(ran_handshake_completion_callback_); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketFalseStartTest, | 
 |        HandshakeCallbackIsRun_WithFalseStartFailure) { | 
 |   // False Start requires NPN and a forward-secret cipher suite. | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |   monitor_handshake_callback_ = true; | 
 |   fail_handshake_after_false_start_ = true; | 
 |   ASSERT_NO_FATAL_FAILURE(TestFalseStart(server_options, client_config, true)); | 
 |   ASSERT_TRUE(ran_handshake_completion_callback_); | 
 | } | 
 |  | 
 | TEST_F(SSLClientSocketFalseStartTest, | 
 |        HandshakeCallbackIsRun_WithFalseStartSuccess) { | 
 |   // False Start requires NPN and a forward-secret cipher suite. | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |   monitor_handshake_callback_ = true; | 
 |   ASSERT_NO_FATAL_FAILURE(TestFalseStart(server_options, client_config, true)); | 
 |   ASSERT_TRUE(ran_handshake_completion_callback_); | 
 | } | 
 | #endif  // defined(USE_OPENSSL) | 
 |  | 
 | TEST_F(SSLClientSocketFalseStartTest, FalseStartEnabled) { | 
 |   // False Start requires NPN and a forward-secret cipher suite. | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |   ASSERT_NO_FATAL_FAILURE( | 
 |       TestFalseStart(server_options, client_config, true)); | 
 | } | 
 |  | 
 | // Test that False Start is disabled without NPN. | 
 | TEST_F(SSLClientSocketFalseStartTest, NoNPN) { | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.clear(); | 
 |   ASSERT_NO_FATAL_FAILURE( | 
 |       TestFalseStart(server_options, client_config, false)); | 
 | } | 
 |  | 
 | // Test that False Start is disabled without a forward-secret cipher suite. | 
 | TEST_F(SSLClientSocketFalseStartTest, NoForwardSecrecy) { | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |   ASSERT_NO_FATAL_FAILURE( | 
 |       TestFalseStart(server_options, client_config, false)); | 
 | } | 
 |  | 
 | // Test that sessions are resumable after receiving the server Finished message. | 
 | TEST_F(SSLClientSocketFalseStartTest, SessionResumption) { | 
 |   // Start a server. | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |  | 
 |   // Let a full handshake complete with False Start. | 
 |   ASSERT_NO_FATAL_FAILURE( | 
 |       TestFalseStart(server_options, client_config, true)); | 
 |  | 
 |   // Make a second connection. | 
 |   TestCompletionCallback callback; | 
 |   scoped_ptr<StreamSocket> transport2( | 
 |       new TCPClientSocket(addr(), &log_, NetLog::Source())); | 
 |   EXPECT_EQ(OK, callback.GetResult(transport2->Connect(callback.callback()))); | 
 |   scoped_ptr<SSLClientSocket> sock2 = CreateSSLClientSocket( | 
 |       transport2.Pass(), test_server()->host_port_pair(), client_config); | 
 |   ASSERT_TRUE(sock2.get()); | 
 |   EXPECT_EQ(OK, callback.GetResult(sock2->Connect(callback.callback()))); | 
 |  | 
 |   // It should resume the session. | 
 |   SSLInfo ssl_info; | 
 |   EXPECT_TRUE(sock2->GetSSLInfo(&ssl_info)); | 
 |   EXPECT_EQ(SSLInfo::HANDSHAKE_RESUME, ssl_info.handshake_type); | 
 | } | 
 |  | 
 | // Test that sessions are not resumable before receiving the server Finished | 
 | // message. | 
 | TEST_F(SSLClientSocketFalseStartTest, NoSessionResumptionBeforeFinish) { | 
 |   // Start a server. | 
 |   SpawnedTestServer::SSLOptions server_options; | 
 |   server_options.key_exchanges = | 
 |       SpawnedTestServer::SSLOptions::KEY_EXCHANGE_DHE_RSA; | 
 |   server_options.enable_npn = true; | 
 |   ASSERT_TRUE(StartTestServer(server_options)); | 
 |  | 
 |   SSLConfig client_config; | 
 |   client_config.next_protos.push_back("http/1.1"); | 
 |  | 
 |   // Start a handshake up to the server Finished message. | 
 |   TestCompletionCallback callback; | 
 |   FakeBlockingStreamSocket* raw_transport1; | 
 |   scoped_ptr<SSLClientSocket> sock1; | 
 |   ASSERT_NO_FATAL_FAILURE(CreateAndConnectUntilServerFinishedReceived( | 
 |       client_config, &callback, &raw_transport1, &sock1)); | 
 |   // Although raw_transport1 has the server Finished blocked, the handshake | 
 |   // still completes. | 
 |   EXPECT_EQ(OK, callback.WaitForResult()); | 
 |  | 
 |   // Drop the old socket. This is needed because the Python test server can't | 
 |   // service two sockets in parallel. | 
 |   sock1.reset(); | 
 |  | 
 |   // Start a second connection. | 
 |   scoped_ptr<StreamSocket> transport2( | 
 |       new TCPClientSocket(addr(), &log_, NetLog::Source())); | 
 |   EXPECT_EQ(OK, callback.GetResult(transport2->Connect(callback.callback()))); | 
 |   scoped_ptr<SSLClientSocket> sock2 = CreateSSLClientSocket( | 
 |       transport2.Pass(), test_server()->host_port_pair(), client_config); | 
 |   EXPECT_EQ(OK, callback.GetResult(sock2->Connect(callback.callback()))); | 
 |  | 
 |   // No session resumption because the first connection never received a server | 
 |   // Finished message. | 
 |   SSLInfo ssl_info; | 
 |   EXPECT_TRUE(sock2->GetSSLInfo(&ssl_info)); | 
 |   EXPECT_EQ(SSLInfo::HANDSHAKE_FULL, ssl_info.handshake_type); | 
 | } | 
 |  | 
 | // Connect to a server using channel id. It should allow the connection. | 
 | TEST_F(SSLClientSocketChannelIDTest, SendChannelID) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |  | 
 |   ASSERT_TRUE(ConnectToTestServer(ssl_options)); | 
 |  | 
 |   EnableChannelID(); | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.channel_id_enabled = true; | 
 |  | 
 |   int rv; | 
 |   ASSERT_TRUE(CreateAndConnectSSLClientSocket(ssl_config, &rv)); | 
 |  | 
 |   EXPECT_EQ(OK, rv); | 
 |   EXPECT_TRUE(sock_->IsConnected()); | 
 |   EXPECT_TRUE(sock_->WasChannelIDSent()); | 
 |  | 
 |   sock_->Disconnect(); | 
 |   EXPECT_FALSE(sock_->IsConnected()); | 
 | } | 
 |  | 
 | // Connect to a server using Channel ID but failing to look up the Channel | 
 | // ID. It should fail. | 
 | TEST_F(SSLClientSocketChannelIDTest, FailingChannelID) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |  | 
 |   ASSERT_TRUE(ConnectToTestServer(ssl_options)); | 
 |  | 
 |   EnableFailingChannelID(); | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.channel_id_enabled = true; | 
 |  | 
 |   int rv; | 
 |   ASSERT_TRUE(CreateAndConnectSSLClientSocket(ssl_config, &rv)); | 
 |  | 
 |   // TODO(haavardm@opera.com): Due to differences in threading, Linux returns | 
 |   // ERR_UNEXPECTED while Mac and Windows return ERR_PROTOCOL_ERROR. Accept all | 
 |   // error codes for now. | 
 |   // http://crbug.com/373670 | 
 |   EXPECT_NE(OK, rv); | 
 |   EXPECT_FALSE(sock_->IsConnected()); | 
 | } | 
 |  | 
 | // Connect to a server using Channel ID but asynchronously failing to look up | 
 | // the Channel ID. It should fail. | 
 | TEST_F(SSLClientSocketChannelIDTest, FailingChannelIDAsync) { | 
 |   SpawnedTestServer::SSLOptions ssl_options; | 
 |  | 
 |   ASSERT_TRUE(ConnectToTestServer(ssl_options)); | 
 |  | 
 |   EnableAsyncFailingChannelID(); | 
 |   SSLConfig ssl_config = kDefaultSSLConfig; | 
 |   ssl_config.channel_id_enabled = true; | 
 |  | 
 |   int rv; | 
 |   ASSERT_TRUE(CreateAndConnectSSLClientSocket(ssl_config, &rv)); | 
 |  | 
 |   EXPECT_EQ(ERR_UNEXPECTED, rv); | 
 |   EXPECT_FALSE(sock_->IsConnected()); | 
 | } | 
 |  | 
 | }  // namespace net |