OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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/test_tools/mock_crypto_client_stream.h" | 5 #include "net/quic/test_tools/mock_crypto_client_stream.h" |
6 | 6 |
7 #include "net/quic/crypto/quic_decrypter.h" | 7 #include "net/quic/crypto/quic_decrypter.h" |
8 #include "net/quic/quic_client_session_base.h" | 8 #include "net/quic/quic_client_session_base.h" |
9 #include "net/quic/quic_server_id.h" | 9 #include "net/quic/quic_server_id.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 16 matching lines...) Expand all Loading... |
27 } | 27 } |
28 | 28 |
29 MockCryptoClientStream::~MockCryptoClientStream() { | 29 MockCryptoClientStream::~MockCryptoClientStream() { |
30 } | 30 } |
31 | 31 |
32 void MockCryptoClientStream::OnHandshakeMessage( | 32 void MockCryptoClientStream::OnHandshakeMessage( |
33 const CryptoHandshakeMessage& message) { | 33 const CryptoHandshakeMessage& message) { |
34 CloseConnection(QUIC_CRYPTO_MESSAGE_AFTER_HANDSHAKE_COMPLETE); | 34 CloseConnection(QUIC_CRYPTO_MESSAGE_AFTER_HANDSHAKE_COMPLETE); |
35 } | 35 } |
36 | 36 |
37 bool MockCryptoClientStream::CryptoConnect() { | 37 void MockCryptoClientStream::CryptoConnect() { |
38 switch (handshake_mode_) { | 38 switch (handshake_mode_) { |
39 case ZERO_RTT: { | 39 case ZERO_RTT: { |
40 encryption_established_ = true; | 40 encryption_established_ = true; |
41 handshake_confirmed_ = false; | 41 handshake_confirmed_ = false; |
42 session()->connection()->SetDecrypter(QuicDecrypter::Create(kNULL), | 42 session()->connection()->SetDecrypter(QuicDecrypter::Create(kNULL), |
43 ENCRYPTION_INITIAL); | 43 ENCRYPTION_INITIAL); |
44 session()->OnCryptoHandshakeEvent( | 44 session()->OnCryptoHandshakeEvent( |
45 QuicSession::ENCRYPTION_FIRST_ESTABLISHED); | 45 QuicSession::ENCRYPTION_FIRST_ESTABLISHED); |
46 break; | 46 break; |
47 } | 47 } |
(...skipping 12 matching lines...) Expand all Loading... |
60 session()->OnCryptoHandshakeEvent(QuicSession::HANDSHAKE_CONFIRMED); | 60 session()->OnCryptoHandshakeEvent(QuicSession::HANDSHAKE_CONFIRMED); |
61 break; | 61 break; |
62 } | 62 } |
63 | 63 |
64 case COLD_START: { | 64 case COLD_START: { |
65 handshake_confirmed_ = false; | 65 handshake_confirmed_ = false; |
66 encryption_established_ = false; | 66 encryption_established_ = false; |
67 break; | 67 break; |
68 } | 68 } |
69 } | 69 } |
70 return true; | |
71 } | 70 } |
72 | 71 |
73 void MockCryptoClientStream::SendOnCryptoHandshakeEvent( | 72 void MockCryptoClientStream::SendOnCryptoHandshakeEvent( |
74 QuicSession::CryptoHandshakeEvent event) { | 73 QuicSession::CryptoHandshakeEvent event) { |
75 encryption_established_ = true; | 74 encryption_established_ = true; |
76 if (event == QuicSession::HANDSHAKE_CONFIRMED) { | 75 if (event == QuicSession::HANDSHAKE_CONFIRMED) { |
77 handshake_confirmed_ = true; | 76 handshake_confirmed_ = true; |
78 SetConfigNegotiated(); | 77 SetConfigNegotiated(); |
79 } | 78 } |
80 session()->OnCryptoHandshakeEvent(event); | 79 session()->OnCryptoHandshakeEvent(event); |
(...skipping 22 matching lines...) Expand all Loading... |
103 ASSERT_EQ(QUIC_NO_ERROR, error); | 102 ASSERT_EQ(QUIC_NO_ERROR, error); |
104 ASSERT_TRUE(session()->config()->negotiated()); | 103 ASSERT_TRUE(session()->config()->negotiated()); |
105 session()->OnConfigNegotiated(); | 104 session()->OnConfigNegotiated(); |
106 } | 105 } |
107 | 106 |
108 QuicClientSessionBase* MockCryptoClientStream::client_session() { | 107 QuicClientSessionBase* MockCryptoClientStream::client_session() { |
109 return reinterpret_cast<QuicClientSessionBase*>(session()); | 108 return reinterpret_cast<QuicClientSessionBase*>(session()); |
110 } | 109 } |
111 | 110 |
112 } // namespace net | 111 } // namespace net |
OLD | NEW |