src/net/quic/test_tools/mock_crypto_client_stream.cc

   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
  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
  27 MockCryptoClientStream::~MockCryptoClientStream() {
  28 }
  29
  30 void MockCryptoClientStream::OnHandshakeMessage(
  31     const CryptoHandshakeMessage& message) {
  32   CloseConnection(QUIC_CRYPTO_MESSAGE_AFTER_HANDSHAKE_COMPLETE);
  33 }
  34
  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
  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
  81 void MockCryptoClientStream::SetConfigNegotiated() {
  82   ASSERT_FALSE(session()->config()->negotiated());
  83   QuicTagVector cgst;
  84   cgst.push_back(kTSTP);
  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
 102 QuicClientSessionBase* MockCryptoClientStream::client_session() {
 103   return reinterpret_cast<QuicClientSessionBase*>(session());
 104 }
 105
 106 }  // namespace net