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 |