| 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_server_stream.h" | 5 #include "net/quic/quic_crypto_server_stream.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| (...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 88 reinterpret_cast<const uint8*>(orbit.data()), | 88 reinterpret_cast<const uint8*>(orbit.data()), |
| 89 StrikeRegister::NO_STARTUP_PERIOD_NEEDED); | 89 StrikeRegister::NO_STARTUP_PERIOD_NEEDED); |
| 90 strike_register_client_->StartDelayingVerification(); | 90 strike_register_client_->StartDelayingVerification(); |
| 91 server_crypto_config_.SetStrikeRegisterClient(strike_register_client_); | 91 server_crypto_config_.SetStrikeRegisterClient(strike_register_client_); |
| 92 } | 92 } |
| 93 } | 93 } |
| 94 | 94 |
| 95 // Initializes the crypto server stream state for testing. May be | 95 // Initializes the crypto server stream state for testing. May be |
| 96 // called multiple times. | 96 // called multiple times. |
| 97 void InitializeServer() { | 97 void InitializeServer() { |
| 98 TestServerSession* server_session = nullptr; | 98 TestQuicSpdyServerSession* server_session = nullptr; |
| 99 CreateServerSessionForTest(server_id_, QuicTime::Delta::FromSeconds(100000), | 99 CreateServerSessionForTest(server_id_, QuicTime::Delta::FromSeconds(100000), |
| 100 &server_crypto_config_, &server_connection_, | 100 &server_crypto_config_, &server_connection_, |
| 101 &server_session); | 101 &server_session); |
| 102 CHECK(server_session); | 102 CHECK(server_session); |
| 103 server_session_.reset(server_session); | 103 server_session_.reset(server_session); |
| 104 CryptoTestUtils::SetupCryptoServerConfigForTest( | 104 CryptoTestUtils::SetupCryptoServerConfigForTest( |
| 105 server_connection_->clock(), server_connection_->random_generator(), | 105 server_connection_->clock(), server_connection_->random_generator(), |
| 106 server_session_->config(), &server_crypto_config_); | 106 server_session_->config(), &server_crypto_config_); |
| 107 } | 107 } |
| 108 | 108 |
| 109 QuicCryptoServerStream* server_stream() { | 109 QuicCryptoServerStream* server_stream() { |
| 110 return server_session_->GetCryptoStream(); | 110 return server_session_->GetCryptoStream(); |
| 111 } | 111 } |
| 112 | 112 |
| 113 QuicCryptoClientStream* client_stream() { | 113 QuicCryptoClientStream* client_stream() { |
| 114 return client_session_->GetCryptoStream(); | 114 return client_session_->GetCryptoStream(); |
| 115 } | 115 } |
| 116 | 116 |
| 117 // Initializes a fake client, and all its associated state, for | 117 // Initializes a fake client, and all its associated state, for |
| 118 // testing. May be called multiple times. | 118 // testing. May be called multiple times. |
| 119 void InitializeFakeClient(bool supports_stateless_rejects) { | 119 void InitializeFakeClient(bool supports_stateless_rejects) { |
| 120 TestClientSession* client_session = nullptr; | 120 TestQuicSpdyClientSession* client_session = nullptr; |
| 121 CreateClientSessionForTest(server_id_, supports_stateless_rejects, | 121 CreateClientSessionForTest(server_id_, supports_stateless_rejects, |
| 122 QuicTime::Delta::FromSeconds(100000), | 122 QuicTime::Delta::FromSeconds(100000), |
| 123 &client_crypto_config_, &client_connection_, | 123 &client_crypto_config_, &client_connection_, |
| 124 &client_session); | 124 &client_session); |
| 125 CHECK(client_session); | 125 CHECK(client_session); |
| 126 client_session_.reset(client_session); | 126 client_session_.reset(client_session); |
| 127 } | 127 } |
| 128 | 128 |
| 129 bool AsyncStrikeRegisterVerification() { | 129 bool AsyncStrikeRegisterVerification() { |
| 130 return GetParam(); | 130 return GetParam(); |
| (...skipping 19 matching lines...) Expand all Loading... |
| 150 | 150 |
| 151 EXPECT_CALL(*client_session_, OnProofValid(_)).Times(testing::AnyNumber()); | 151 EXPECT_CALL(*client_session_, OnProofValid(_)).Times(testing::AnyNumber()); |
| 152 client_stream()->CryptoConnect(); | 152 client_stream()->CryptoConnect(); |
| 153 CryptoTestUtils::AdvanceHandshake(client_connection_, client_stream(), 0, | 153 CryptoTestUtils::AdvanceHandshake(client_connection_, client_stream(), 0, |
| 154 server_connection_, server_stream(), 0); | 154 server_connection_, server_stream(), 0); |
| 155 } | 155 } |
| 156 | 156 |
| 157 protected: | 157 protected: |
| 158 // Server state | 158 // Server state |
| 159 PacketSavingConnection* server_connection_; | 159 PacketSavingConnection* server_connection_; |
| 160 scoped_ptr<TestServerSession> server_session_; | 160 scoped_ptr<TestQuicSpdyServerSession> server_session_; |
| 161 QuicCryptoServerConfig server_crypto_config_; | 161 QuicCryptoServerConfig server_crypto_config_; |
| 162 QuicServerId server_id_; | 162 QuicServerId server_id_; |
| 163 | 163 |
| 164 // Client state | 164 // Client state |
| 165 PacketSavingConnection* client_connection_; | 165 PacketSavingConnection* client_connection_; |
| 166 QuicCryptoClientConfig client_crypto_config_; | 166 QuicCryptoClientConfig client_crypto_config_; |
| 167 scoped_ptr<TestClientSession> client_session_; | 167 scoped_ptr<TestQuicSpdyClientSession> client_session_; |
| 168 | 168 |
| 169 CryptoHandshakeMessage message_; | 169 CryptoHandshakeMessage message_; |
| 170 scoped_ptr<QuicData> message_data_; | 170 scoped_ptr<QuicData> message_data_; |
| 171 CryptoTestUtils::FakeClientOptions client_options_; | 171 CryptoTestUtils::FakeClientOptions client_options_; |
| 172 DelayedVerifyStrikeRegisterClient* strike_register_client_; | 172 DelayedVerifyStrikeRegisterClient* strike_register_client_; |
| 173 }; | 173 }; |
| 174 | 174 |
| 175 INSTANTIATE_TEST_CASE_P(Tests, QuicCryptoServerStreamTest, testing::Bool()); | 175 INSTANTIATE_TEST_CASE_P(Tests, QuicCryptoServerStreamTest, testing::Bool()); |
| 176 | 176 |
| 177 TEST_P(QuicCryptoServerStreamTest, NotInitiallyConected) { | 177 TEST_P(QuicCryptoServerStreamTest, NotInitiallyConected) { |
| (...skipping 237 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 415 message_.Clear(); | 415 message_.Clear(); |
| 416 QuicConfig stateful_reject_config = DefaultQuicConfig(); | 416 QuicConfig stateful_reject_config = DefaultQuicConfig(); |
| 417 stateful_reject_config.ToHandshakeMessage(&message_); | 417 stateful_reject_config.ToHandshakeMessage(&message_); |
| 418 EXPECT_FALSE( | 418 EXPECT_FALSE( |
| 419 QuicCryptoServerStreamPeer::DoesPeerSupportStatelessRejects(message_)); | 419 QuicCryptoServerStreamPeer::DoesPeerSupportStatelessRejects(message_)); |
| 420 } | 420 } |
| 421 | 421 |
| 422 } // namespace | 422 } // namespace |
| 423 } // namespace test | 423 } // namespace test |
| 424 } // namespace net | 424 } // namespace net |
| OLD | NEW |