| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_crypto_client_stream.h" | 5 #include "net/quic/quic_crypto_client_stream.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "net/quic/crypto/aes_128_gcm_12_encrypter.h" | 8 #include "net/quic/crypto/aes_128_gcm_12_encrypter.h" |
| 9 #include "net/quic/crypto/quic_decrypter.h" | 9 #include "net/quic/crypto/quic_decrypter.h" |
| 10 #include "net/quic/crypto/quic_encrypter.h" | 10 #include "net/quic/crypto/quic_encrypter.h" |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 namespace net { | 23 namespace net { |
| 24 namespace test { | 24 namespace test { |
| 25 namespace { | 25 namespace { |
| 26 | 26 |
| 27 const char kServerHostname[] = "example.com"; | 27 const char kServerHostname[] = "example.com"; |
| 28 const uint16 kServerPort = 80; | 28 const uint16 kServerPort = 80; |
| 29 | 29 |
| 30 class QuicCryptoClientStreamTest : public ::testing::Test { | 30 class QuicCryptoClientStreamTest : public ::testing::Test { |
| 31 public: | 31 public: |
| 32 QuicCryptoClientStreamTest() | 32 QuicCryptoClientStreamTest() |
| 33 : connection_(new PacketSavingConnection(false)), | 33 : connection_(new PacketSavingConnection(/*is_server=*/false)), |
| 34 session_(new TestClientSession(connection_, DefaultQuicConfig())), | 34 session_(new TestClientSession(connection_, DefaultQuicConfig())), |
| 35 server_id_(kServerHostname, kServerPort, false, PRIVACY_MODE_DISABLED), | 35 server_id_(kServerHostname, kServerPort, false, PRIVACY_MODE_DISABLED), |
| 36 stream_(new QuicCryptoClientStream(server_id_, session_.get(), nullptr, | 36 stream_(new QuicCryptoClientStream(server_id_, |
| 37 session_.get(), |
| 38 nullptr, |
| 37 &crypto_config_)) { | 39 &crypto_config_)) { |
| 38 session_->SetCryptoStream(stream_.get()); | 40 session_->SetCryptoStream(stream_.get()); |
| 39 // Advance the time, because timers do not like uninitialized times. | 41 // Advance the time, because timers do not like uninitialized times. |
| 40 connection_->AdvanceTime(QuicTime::Delta::FromSeconds(1)); | 42 connection_->AdvanceTime(QuicTime::Delta::FromSeconds(1)); |
| 41 } | 43 } |
| 42 | 44 |
| 43 void CompleteCryptoHandshake() { | 45 void CompleteCryptoHandshake() { |
| 44 stream_->CryptoConnect(); | 46 stream_->CryptoConnect(); |
| 45 CryptoTestUtils::HandshakeWithFakeServer(connection_, stream_.get()); | 47 CryptoTestUtils::HandshakeWithFakeServer(connection_, stream_.get()); |
| 46 } | 48 } |
| (...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 114 | 116 |
| 115 CompleteCryptoHandshake(); | 117 CompleteCryptoHandshake(); |
| 116 EXPECT_TRUE(stream_->encryption_established()); | 118 EXPECT_TRUE(stream_->encryption_established()); |
| 117 EXPECT_TRUE(stream_->handshake_confirmed()); | 119 EXPECT_TRUE(stream_->handshake_confirmed()); |
| 118 } | 120 } |
| 119 | 121 |
| 120 TEST_F(QuicCryptoClientStreamTest, ExpiredServerConfig) { | 122 TEST_F(QuicCryptoClientStreamTest, ExpiredServerConfig) { |
| 121 // Seed the config with a cached server config. | 123 // Seed the config with a cached server config. |
| 122 CompleteCryptoHandshake(); | 124 CompleteCryptoHandshake(); |
| 123 | 125 |
| 124 connection_ = new PacketSavingConnection(true); | 126 connection_ = new PacketSavingConnection(/*is_server=*/false); |
| 125 session_.reset(new TestClientSession(connection_, DefaultQuicConfig())); | 127 session_.reset(new TestClientSession(connection_, DefaultQuicConfig())); |
| 126 stream_.reset(new QuicCryptoClientStream(server_id_, session_.get(), nullptr, | 128 stream_.reset(new QuicCryptoClientStream(server_id_, session_.get(), nullptr, |
| 127 &crypto_config_)); | 129 &crypto_config_)); |
| 128 | 130 |
| 129 session_->SetCryptoStream(stream_.get()); | 131 session_->SetCryptoStream(stream_.get()); |
| 130 | 132 |
| 131 // Advance time 5 years to ensure that we pass the expiry time of the cached | 133 // Advance time 5 years to ensure that we pass the expiry time of the cached |
| 132 // server config. | 134 // server config. |
| 133 connection_->AdvanceTime( | 135 connection_->AdvanceTime( |
| 134 QuicTime::Delta::FromSeconds(60 * 60 * 24 * 365 * 5)); | 136 QuicTime::Delta::FromSeconds(60 * 60 * 24 * 365 * 5)); |
| 135 | 137 |
| 136 stream_->CryptoConnect(); | 138 stream_->CryptoConnect(); |
| 137 // Check that a client hello was sent. | 139 // Check that a client hello was sent. |
| 138 ASSERT_EQ(1u, connection_->packets_.size()); | 140 ASSERT_EQ(1u, connection_->encrypted_packets_.size()); |
| 139 } | 141 } |
| 140 | 142 |
| 141 TEST_F(QuicCryptoClientStreamTest, ServerConfigUpdate) { | 143 TEST_F(QuicCryptoClientStreamTest, ServerConfigUpdate) { |
| 142 // Test that the crypto client stream can receive server config updates after | 144 // Test that the crypto client stream can receive server config updates after |
| 143 // the connection has been established. | 145 // the connection has been established. |
| 144 CompleteCryptoHandshake(); | 146 CompleteCryptoHandshake(); |
| 145 | 147 |
| 146 QuicCryptoClientConfig::CachedState* state = | 148 QuicCryptoClientConfig::CachedState* state = |
| 147 crypto_config_.LookupOrCreate(server_id_); | 149 crypto_config_.LookupOrCreate(server_id_); |
| 148 | 150 |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 194 CryptoHandshakeMessage server_config_update; | 196 CryptoHandshakeMessage server_config_update; |
| 195 server_config_update.set_tag(kSCUP); | 197 server_config_update.set_tag(kSCUP); |
| 196 scoped_ptr<QuicData> data( | 198 scoped_ptr<QuicData> data( |
| 197 CryptoFramer::ConstructHandshakeMessage(server_config_update)); | 199 CryptoFramer::ConstructHandshakeMessage(server_config_update)); |
| 198 stream_->ProcessRawData(data->data(), data->length()); | 200 stream_->ProcessRawData(data->data(), data->length()); |
| 199 } | 201 } |
| 200 | 202 |
| 201 } // namespace | 203 } // namespace |
| 202 } // namespace test | 204 } // namespace test |
| 203 } // namespace net | 205 } // namespace net |
| OLD | NEW |