1 // Copyright 2013 The Chromium Authors
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/mock_crypto_client_stream_factory.h"
6
7 #include "base/lazy_instance.h"
8 #include "net/quic/quic_chromium_client_session.h"
9 #include "net/third_party/quiche/src/quiche/quic/core/quic_crypto_client_stream.h"
10
11 using std::string;
12
13 namespace net {
14
15 MockCryptoClientStreamFactory::~MockCryptoClientStreamFactory() = default;
16
MockCryptoClientStreamFactory()17 MockCryptoClientStreamFactory::MockCryptoClientStreamFactory()
18 : config_(std::make_unique<quic::QuicConfig>()) {}
19
SetConfig(const quic::QuicConfig & config)20 void MockCryptoClientStreamFactory::SetConfig(const quic::QuicConfig& config) {
21 config_ = std::make_unique<quic::QuicConfig>(config);
22 }
23
SetConfigForServerId(const quic::QuicServerId & server_id,const quic::QuicConfig & config)24 void MockCryptoClientStreamFactory::SetConfigForServerId(
25 const quic::QuicServerId& server_id,
26 const quic::QuicConfig& config) {
27 config_for_server_[server_id] = std::make_unique<quic::QuicConfig>(config);
28 }
29
30 std::unique_ptr<quic::QuicCryptoClientStream>
CreateQuicCryptoClientStream(const quic::QuicServerId & server_id,QuicChromiumClientSession * session,std::unique_ptr<quic::ProofVerifyContext>,quic::QuicCryptoClientConfig * crypto_config)31 MockCryptoClientStreamFactory::CreateQuicCryptoClientStream(
32 const quic::QuicServerId& server_id,
33 QuicChromiumClientSession* session,
34 std::unique_ptr<quic::ProofVerifyContext> /*proof_verify_context*/,
35 quic::QuicCryptoClientConfig* crypto_config) {
36 const ProofVerifyDetailsChromium* proof_verify_details = nullptr;
37 if (!proof_verify_details_queue_.empty()) {
38 proof_verify_details = proof_verify_details_queue_.front();
39 proof_verify_details_queue_.pop();
40 }
41
42 // Find a config in `config_for_server_`, falling back to `config_` if none
43 // exists.
44 auto it = config_for_server_.find(server_id);
45 quic::QuicConfig* config =
46 it == config_for_server_.end() ? config_.get() : it->second.get();
47
48 std::unique_ptr<MockCryptoClientStream> stream =
49 std::make_unique<MockCryptoClientStream>(
50 server_id, session, nullptr, *config, crypto_config, handshake_mode_,
51 proof_verify_details, use_mock_crypter_);
52 streams_.push_back(stream->GetWeakPtr());
53 return stream;
54 }
55
last_stream() const56 MockCryptoClientStream* MockCryptoClientStreamFactory::last_stream() const {
57 CHECK(!streams_.empty());
58 return streams_.back().get();
59 }
60
61 } // namespace net
62