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
3 // found in the LICENSE file.
4
5 #include "net/quic/test_tools/mock_crypto_client_stream.h"
6
7 #include "net/quic/crypto/quic_decrypter.h"
8 #include "net/quic/quic_client_session_base.h"
9 #include "net/quic/quic_server_id.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace net {
13
MockCryptoClientStream(const QuicServerId & server_id,QuicClientSessionBase * session,ProofVerifyContext * verify_context,QuicCryptoClientConfig * crypto_config,HandshakeMode handshake_mode,const ProofVerifyDetails * proof_verify_details)14 MockCryptoClientStream::MockCryptoClientStream(
15 const QuicServerId& server_id,
16 QuicClientSessionBase* session,
17 ProofVerifyContext* verify_context,
18 QuicCryptoClientConfig* crypto_config,
19 HandshakeMode handshake_mode,
20 const ProofVerifyDetails* proof_verify_details)
21 : QuicCryptoClientStream(server_id, session, verify_context,
22 crypto_config),
23 handshake_mode_(handshake_mode),
24 proof_verify_details_(proof_verify_details) {
25 }
26
~MockCryptoClientStream()27 MockCryptoClientStream::~MockCryptoClientStream() {
28 }
29
OnHandshakeMessage(const CryptoHandshakeMessage & message)30 void MockCryptoClientStream::OnHandshakeMessage(
31 const CryptoHandshakeMessage& message) {
32 CloseConnection(QUIC_CRYPTO_MESSAGE_AFTER_HANDSHAKE_COMPLETE);
33 }
34
CryptoConnect()35 bool MockCryptoClientStream::CryptoConnect() {
36 switch (handshake_mode_) {
37 case ZERO_RTT: {
38 encryption_established_ = true;
39 handshake_confirmed_ = false;
40 session()->connection()->SetDecrypter(QuicDecrypter::Create(kNULL),
41 ENCRYPTION_INITIAL);
42 session()->OnCryptoHandshakeEvent(
43 QuicSession::ENCRYPTION_FIRST_ESTABLISHED);
44 break;
45 }
46
47 case CONFIRM_HANDSHAKE: {
48 encryption_established_ = true;
49 handshake_confirmed_ = true;
50 crypto_negotiated_params_.key_exchange = kC255;
51 crypto_negotiated_params_.aead = kAESG;
52 if (proof_verify_details_) {
53 client_session()->OnProofVerifyDetailsAvailable(*proof_verify_details_);
54 }
55 SetConfigNegotiated();
56 session()->connection()->SetDecrypter(QuicDecrypter::Create(kNULL),
57 ENCRYPTION_FORWARD_SECURE);
58 session()->OnCryptoHandshakeEvent(QuicSession::HANDSHAKE_CONFIRMED);
59 break;
60 }
61
62 case COLD_START: {
63 handshake_confirmed_ = false;
64 encryption_established_ = false;
65 break;
66 }
67 }
68 return true;
69 }
70
SendOnCryptoHandshakeEvent(QuicSession::CryptoHandshakeEvent event)71 void MockCryptoClientStream::SendOnCryptoHandshakeEvent(
72 QuicSession::CryptoHandshakeEvent event) {
73 encryption_established_ = true;
74 if (event == QuicSession::HANDSHAKE_CONFIRMED) {
75 handshake_confirmed_ = true;
76 SetConfigNegotiated();
77 }
78 session()->OnCryptoHandshakeEvent(event);
79 }
80
SetConfigNegotiated()81 void MockCryptoClientStream::SetConfigNegotiated() {
82 ASSERT_FALSE(session()->config()->negotiated());
83 QuicTagVector cgst;
84 cgst.push_back(kINAR);
85 cgst.push_back(kQBIC);
86 session()->config()->set_congestion_feedback(cgst, kQBIC);
87 session()->config()->set_idle_connection_state_lifetime(
88 QuicTime::Delta::FromSeconds(2 * kDefaultTimeoutSecs),
89 QuicTime::Delta::FromSeconds(kDefaultTimeoutSecs));
90 session()->config()->set_max_streams_per_connection(
91 2 * kDefaultMaxStreamsPerConnection, kDefaultMaxStreamsPerConnection);
92
93 CryptoHandshakeMessage msg;
94 session()->config()->ToHandshakeMessage(&msg);
95 string error_details;
96 const QuicErrorCode error =
97 session()->config()->ProcessPeerHello(msg, CLIENT, &error_details);
98 ASSERT_EQ(QUIC_NO_ERROR, error);
99 ASSERT_TRUE(session()->config()->negotiated());
100 }
101
client_session()102 QuicClientSessionBase* MockCryptoClientStream::client_session() {
103 return reinterpret_cast<QuicClientSessionBase*>(session());
104 }
105
106 } // namespace net
107