| // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "net/spdy/spdy_test_util_common.h" |
| |
| #include <cstddef> |
| #include <utility> |
| |
| #include "base/base64.h" |
| #include "base/bind.h" |
| #include "base/compiler_specific.h" |
| #include "base/logging.h" |
| #include "base/optional.h" |
| #include "base/strings/string_number_conversions.h" |
| #include "base/strings/string_split.h" |
| #include "net/base/host_port_pair.h" |
| #include "net/base/http_user_agent_settings.h" |
| #include "net/cert/ct_policy_enforcer.h" |
| #include "net/cert/ct_policy_status.h" |
| #include "net/cert/do_nothing_ct_verifier.h" |
| #include "net/cert/mock_cert_verifier.h" |
| #include "net/cert/signed_certificate_timestamp_and_status.h" |
| #include "net/dns/host_resolver.h" |
| #include "net/http/http_cache.h" |
| #include "net/http/http_network_transaction.h" |
| #include "net/http/http_proxy_connect_job.h" |
| #include "net/log/net_log_with_source.h" |
| #include "net/socket/client_socket_handle.h" |
| #include "net/socket/next_proto.h" |
| #include "net/socket/socket_tag.h" |
| #include "net/socket/socks_connect_job.h" |
| #include "net/socket/ssl_client_socket.h" |
| #include "net/socket/transport_client_socket_pool.h" |
| #include "net/spdy/buffered_spdy_framer.h" |
| #include "net/spdy/spdy_http_utils.h" |
| #include "net/spdy/spdy_stream.h" |
| #include "net/test/gtest_util.h" |
| #include "net/third_party/quiche/src/spdy/core/spdy_alt_svc_wire_format.h" |
| #include "net/third_party/quiche/src/spdy/core/spdy_framer.h" |
| #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
| #include "net/url_request/url_request_job_factory_impl.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| using net::test::IsError; |
| using net::test::IsOk; |
| |
| namespace net { |
| |
| namespace { |
| |
| // Parses a URL into the scheme, host, and path components required for a |
| // SPDY request. |
| void ParseUrl(base::StringPiece url, |
| std::string* scheme, |
| std::string* host, |
| std::string* path) { |
| GURL gurl(url); |
| path->assign(gurl.PathForRequest()); |
| scheme->assign(gurl.scheme()); |
| host->assign(gurl.host()); |
| if (gurl.has_port()) { |
| host->append(":"); |
| host->append(gurl.port()); |
| } |
| } |
| |
| } // namespace |
| |
| // Chop a frame into an array of MockWrites. |
| // |frame| is the frame to chop. |
| // |num_chunks| is the number of chunks to create. |
| std::unique_ptr<MockWrite[]> ChopWriteFrame( |
| const spdy::SpdySerializedFrame& frame, |
| int num_chunks) { |
| auto chunks = std::make_unique<MockWrite[]>(num_chunks); |
| int chunk_size = frame.size() / num_chunks; |
| for (int index = 0; index < num_chunks; index++) { |
| const char* ptr = frame.data() + (index * chunk_size); |
| if (index == num_chunks - 1) |
| chunk_size += |
| frame.size() % chunk_size; // The last chunk takes the remainder. |
| chunks[index] = MockWrite(ASYNC, ptr, chunk_size); |
| } |
| return chunks; |
| } |
| |
| // Adds headers and values to a map. |
| // |extra_headers| is an array of { name, value } pairs, arranged as strings |
| // where the even entries are the header names, and the odd entries are the |
| // header values. |
| // |headers| gets filled in from |extra_headers|. |
| void AppendToHeaderBlock(const char* const extra_headers[], |
| int extra_header_count, |
| spdy::SpdyHeaderBlock* headers) { |
| if (!extra_header_count) |
| return; |
| |
| // Sanity check: Non-NULL header list. |
| DCHECK(extra_headers) << "NULL header value pair list"; |
| // Sanity check: Non-NULL header map. |
| DCHECK(headers) << "NULL header map"; |
| |
| // Copy in the headers. |
| for (int i = 0; i < extra_header_count; i++) { |
| base::StringPiece key(extra_headers[i * 2]); |
| base::StringPiece value(extra_headers[i * 2 + 1]); |
| DCHECK(!key.empty()) << "Header key must not be empty."; |
| headers->AppendValueOrAddHeader(key, value); |
| } |
| } |
| |
| // Create a MockWrite from the given spdy::SpdySerializedFrame. |
| MockWrite CreateMockWrite(const spdy::SpdySerializedFrame& req) { |
| return MockWrite(ASYNC, req.data(), req.size()); |
| } |
| |
| // Create a MockWrite from the given spdy::SpdySerializedFrame and sequence |
| // number. |
| MockWrite CreateMockWrite(const spdy::SpdySerializedFrame& req, int seq) { |
| return CreateMockWrite(req, seq, ASYNC); |
| } |
| |
| // Create a MockWrite from the given spdy::SpdySerializedFrame and sequence |
| // number. |
| MockWrite CreateMockWrite(const spdy::SpdySerializedFrame& req, |
| int seq, |
| IoMode mode) { |
| return MockWrite(mode, req.data(), req.size(), seq); |
| } |
| |
| // Create a MockRead from the given spdy::SpdySerializedFrame. |
| MockRead CreateMockRead(const spdy::SpdySerializedFrame& resp) { |
| return MockRead(ASYNC, resp.data(), resp.size()); |
| } |
| |
| // Create a MockRead from the given spdy::SpdySerializedFrame and sequence |
| // number. |
| MockRead CreateMockRead(const spdy::SpdySerializedFrame& resp, int seq) { |
| return CreateMockRead(resp, seq, ASYNC); |
| } |
| |
| // Create a MockRead from the given spdy::SpdySerializedFrame and sequence |
| // number. |
| MockRead CreateMockRead(const spdy::SpdySerializedFrame& resp, |
| int seq, |
| IoMode mode) { |
| return MockRead(mode, resp.data(), resp.size(), seq); |
| } |
| |
| // Combines the given vector of spdy::SpdySerializedFrame into a single frame. |
| spdy::SpdySerializedFrame CombineFrames( |
| std::vector<const spdy::SpdySerializedFrame*> frames) { |
| int total_size = 0; |
| for (const auto* frame : frames) { |
| total_size += frame->size(); |
| } |
| auto data = std::make_unique<char[]>(total_size); |
| char* ptr = data.get(); |
| for (const auto* frame : frames) { |
| memcpy(ptr, frame->data(), frame->size()); |
| ptr += frame->size(); |
| } |
| return spdy::SpdySerializedFrame(data.release(), total_size, |
| /* owns_buffer = */ true); |
| } |
| |
| namespace { |
| |
| class PriorityGetter : public BufferedSpdyFramerVisitorInterface { |
| public: |
| PriorityGetter() : priority_(0) {} |
| ~PriorityGetter() override = default; |
| |
| spdy::SpdyPriority priority() const { return priority_; } |
| |
| void OnError( |
| http2::Http2DecoderAdapter::SpdyFramerError spdy_framer_error) override {} |
| void OnStreamError(spdy::SpdyStreamId stream_id, |
| const std::string& description) override {} |
| void OnHeaders(spdy::SpdyStreamId stream_id, |
| bool has_priority, |
| int weight, |
| spdy::SpdyStreamId parent_stream_id, |
| bool exclusive, |
| bool fin, |
| spdy::SpdyHeaderBlock headers, |
| base::TimeTicks recv_first_byte_time) override { |
| if (has_priority) { |
| priority_ = spdy::Http2WeightToSpdy3Priority(weight); |
| } |
| } |
| void OnDataFrameHeader(spdy::SpdyStreamId stream_id, |
| size_t length, |
| bool fin) override {} |
| void OnStreamFrameData(spdy::SpdyStreamId stream_id, |
| const char* data, |
| size_t len) override {} |
| void OnStreamEnd(spdy::SpdyStreamId stream_id) override {} |
| void OnStreamPadding(spdy::SpdyStreamId stream_id, size_t len) override {} |
| void OnSettings() override {} |
| void OnSettingsAck() override {} |
| void OnSetting(spdy::SpdySettingsId id, uint32_t value) override {} |
| void OnSettingsEnd() override {} |
| void OnPing(spdy::SpdyPingId unique_id, bool is_ack) override {} |
| void OnRstStream(spdy::SpdyStreamId stream_id, |
| spdy::SpdyErrorCode error_code) override {} |
| void OnGoAway(spdy::SpdyStreamId last_accepted_stream_id, |
| spdy::SpdyErrorCode error_code, |
| base::StringPiece debug_data) override {} |
| void OnWindowUpdate(spdy::SpdyStreamId stream_id, |
| int delta_window_size) override {} |
| void OnPushPromise(spdy::SpdyStreamId stream_id, |
| spdy::SpdyStreamId promised_stream_id, |
| spdy::SpdyHeaderBlock headers) override {} |
| void OnAltSvc(spdy::SpdyStreamId stream_id, |
| base::StringPiece origin, |
| const spdy::SpdyAltSvcWireFormat::AlternativeServiceVector& |
| altsvc_vector) override {} |
| bool OnUnknownFrame(spdy::SpdyStreamId stream_id, |
| uint8_t frame_type) override { |
| return false; |
| } |
| |
| private: |
| spdy::SpdyPriority priority_; |
| }; |
| |
| } // namespace |
| |
| bool GetSpdyPriority(const spdy::SpdySerializedFrame& frame, |
| spdy::SpdyPriority* priority) { |
| NetLogWithSource net_log; |
| BufferedSpdyFramer framer(kMaxHeaderListSizeForTest, net_log); |
| PriorityGetter priority_getter; |
| framer.set_visitor(&priority_getter); |
| size_t frame_size = frame.size(); |
| if (framer.ProcessInput(frame.data(), frame_size) != frame_size) { |
| return false; |
| } |
| *priority = priority_getter.priority(); |
| return true; |
| } |
| |
| base::WeakPtr<SpdyStream> CreateStreamSynchronously( |
| SpdyStreamType type, |
| const base::WeakPtr<SpdySession>& session, |
| const GURL& url, |
| RequestPriority priority, |
| const NetLogWithSource& net_log) { |
| SpdyStreamRequest stream_request; |
| int rv = stream_request.StartRequest( |
| type, session, url, false /* no early data */, priority, SocketTag(), |
| net_log, CompletionOnceCallback(), TRAFFIC_ANNOTATION_FOR_TESTS); |
| |
| return |
| (rv == OK) ? stream_request.ReleaseStream() : base::WeakPtr<SpdyStream>(); |
| } |
| |
| StreamReleaserCallback::StreamReleaserCallback() = default; |
| |
| StreamReleaserCallback::~StreamReleaserCallback() = default; |
| |
| CompletionOnceCallback StreamReleaserCallback::MakeCallback( |
| SpdyStreamRequest* request) { |
| return base::BindOnce(&StreamReleaserCallback::OnComplete, |
| base::Unretained(this), request); |
| } |
| |
| void StreamReleaserCallback::OnComplete( |
| SpdyStreamRequest* request, int result) { |
| if (result == OK) |
| request->ReleaseStream()->Cancel(ERR_ABORTED); |
| SetResult(result); |
| } |
| |
| MockECSignatureCreator::MockECSignatureCreator(crypto::ECPrivateKey* key) |
| : key_(key) { |
| } |
| |
| bool MockECSignatureCreator::Sign(const uint8_t* data, |
| int data_len, |
| std::vector<uint8_t>* signature) { |
| std::vector<uint8_t> private_key; |
| if (!key_->ExportPrivateKey(&private_key)) |
| return false; |
| std::string head = "fakesignature"; |
| std::string tail = "/fakesignature"; |
| |
| signature->clear(); |
| signature->insert(signature->end(), head.begin(), head.end()); |
| signature->insert(signature->end(), private_key.begin(), private_key.end()); |
| signature->insert(signature->end(), '-'); |
| signature->insert(signature->end(), data, data + data_len); |
| signature->insert(signature->end(), tail.begin(), tail.end()); |
| return true; |
| } |
| |
| bool MockECSignatureCreator::DecodeSignature( |
| const std::vector<uint8_t>& signature, |
| std::vector<uint8_t>* out_raw_sig) { |
| *out_raw_sig = signature; |
| return true; |
| } |
| |
| MockECSignatureCreatorFactory::MockECSignatureCreatorFactory() { |
| crypto::ECSignatureCreator::SetFactoryForTesting(this); |
| } |
| |
| MockECSignatureCreatorFactory::~MockECSignatureCreatorFactory() { |
| crypto::ECSignatureCreator::SetFactoryForTesting(nullptr); |
| } |
| |
| std::unique_ptr<crypto::ECSignatureCreator> |
| MockECSignatureCreatorFactory::Create(crypto::ECPrivateKey* key) { |
| return std::make_unique<MockECSignatureCreator>(key); |
| } |
| |
| SpdySessionDependencies::SpdySessionDependencies() |
| : SpdySessionDependencies(ProxyResolutionService::CreateDirect()) {} |
| |
| SpdySessionDependencies::SpdySessionDependencies( |
| std::unique_ptr<ProxyResolutionService> proxy_resolution_service) |
| : host_resolver(std::make_unique<MockCachingHostResolver>()), |
| cert_verifier(std::make_unique<MockCertVerifier>()), |
| transport_security_state(std::make_unique<TransportSecurityState>()), |
| cert_transparency_verifier(std::make_unique<DoNothingCTVerifier>()), |
| ct_policy_enforcer(std::make_unique<DefaultCTPolicyEnforcer>()), |
| proxy_resolution_service(std::move(proxy_resolution_service)), |
| ssl_config_service(std::make_unique<SSLConfigServiceDefaults>()), |
| socket_factory(std::make_unique<MockClientSocketFactory>()), |
| http_auth_handler_factory(HttpAuthHandlerFactory::CreateDefault()), |
| http_server_properties(std::make_unique<HttpServerProperties>()), |
| enable_ip_pooling(true), |
| enable_ping(false), |
| enable_user_alternate_protocol_ports(false), |
| enable_quic(false), |
| enable_server_push_cancellation(false), |
| session_max_recv_window_size(kDefaultInitialWindowSize), |
| session_max_queued_capped_frames(kSpdySessionMaxQueuedCappedFrames), |
| time_func(&base::TimeTicks::Now), |
| enable_http2_alternative_service(false), |
| enable_websocket_over_http2(false), |
| net_log(nullptr), |
| disable_idle_sockets_close_on_memory_pressure(false), |
| enable_early_data(false), |
| allow_default_credentials( |
| HttpAuthPreferences::ALLOW_DEFAULT_CREDENTIALS) { |
| http2_settings[spdy::SETTINGS_INITIAL_WINDOW_SIZE] = |
| kDefaultInitialWindowSize; |
| } |
| |
| SpdySessionDependencies::~SpdySessionDependencies() = default; |
| |
| // static |
| std::unique_ptr<HttpNetworkSession> SpdySessionDependencies::SpdyCreateSession( |
| SpdySessionDependencies* session_deps) { |
| return SpdyCreateSessionWithSocketFactory(session_deps, |
| session_deps->socket_factory.get()); |
| } |
| |
| // static |
| std::unique_ptr<HttpNetworkSession> |
| SpdySessionDependencies::SpdyCreateSessionWithSocketFactory( |
| SpdySessionDependencies* session_deps, |
| ClientSocketFactory* factory) { |
| HttpNetworkSession::Params session_params = CreateSessionParams(session_deps); |
| HttpNetworkSession::Context session_context = |
| CreateSessionContext(session_deps); |
| session_context.client_socket_factory = factory; |
| auto http_session = |
| std::make_unique<HttpNetworkSession>(session_params, session_context); |
| SpdySessionPoolPeer pool_peer(http_session->spdy_session_pool()); |
| pool_peer.SetEnableSendingInitialData(false); |
| return http_session; |
| } |
| |
| // static |
| HttpNetworkSession::Params SpdySessionDependencies::CreateSessionParams( |
| SpdySessionDependencies* session_deps) { |
| HttpNetworkSession::Params params; |
| params.enable_spdy_ping_based_connection_checking = session_deps->enable_ping; |
| params.enable_user_alternate_protocol_ports = |
| session_deps->enable_user_alternate_protocol_ports; |
| params.enable_quic = session_deps->enable_quic; |
| params.enable_server_push_cancellation = |
| session_deps->enable_server_push_cancellation; |
| params.spdy_session_max_recv_window_size = |
| session_deps->session_max_recv_window_size; |
| params.spdy_session_max_queued_capped_frames = |
| session_deps->session_max_queued_capped_frames; |
| params.http2_settings = session_deps->http2_settings; |
| params.time_func = session_deps->time_func; |
| params.enable_http2_alternative_service = |
| session_deps->enable_http2_alternative_service; |
| params.enable_websocket_over_http2 = |
| session_deps->enable_websocket_over_http2; |
| params.greased_http2_frame = session_deps->greased_http2_frame; |
| params.disable_idle_sockets_close_on_memory_pressure = |
| session_deps->disable_idle_sockets_close_on_memory_pressure; |
| params.enable_early_data = session_deps->enable_early_data; |
| params.allow_default_credentials = session_deps->allow_default_credentials; |
| return params; |
| } |
| |
| HttpNetworkSession::Context SpdySessionDependencies::CreateSessionContext( |
| SpdySessionDependencies* session_deps) { |
| HttpNetworkSession::Context context; |
| context.client_socket_factory = session_deps->socket_factory.get(); |
| context.host_resolver = session_deps->GetHostResolver(); |
| context.cert_verifier = session_deps->cert_verifier.get(); |
| context.transport_security_state = |
| session_deps->transport_security_state.get(); |
| context.cert_transparency_verifier = |
| session_deps->cert_transparency_verifier.get(); |
| context.ct_policy_enforcer = session_deps->ct_policy_enforcer.get(); |
| context.proxy_resolution_service = |
| session_deps->proxy_resolution_service.get(); |
| context.http_user_agent_settings = |
| session_deps->http_user_agent_settings.get(); |
| context.ssl_config_service = session_deps->ssl_config_service.get(); |
| context.http_auth_handler_factory = |
| session_deps->http_auth_handler_factory.get(); |
| context.http_server_properties = session_deps->http_server_properties.get(); |
| context.net_log = session_deps->net_log; |
| #if BUILDFLAG(ENABLE_REPORTING) |
| context.reporting_service = session_deps->reporting_service.get(); |
| context.network_error_logging_service = |
| session_deps->network_error_logging_service.get(); |
| #endif |
| return context; |
| } |
| |
| SpdyURLRequestContext::SpdyURLRequestContext() : storage_(this) { |
| storage_.set_host_resolver(std::make_unique<MockHostResolver>()); |
| storage_.set_cert_verifier(std::make_unique<MockCertVerifier>()); |
| storage_.set_transport_security_state( |
| std::make_unique<TransportSecurityState>()); |
| storage_.set_proxy_resolution_service(ProxyResolutionService::CreateDirect()); |
| storage_.set_ct_policy_enforcer(std::make_unique<DefaultCTPolicyEnforcer>()); |
| storage_.set_cert_transparency_verifier( |
| std::make_unique<DoNothingCTVerifier>()); |
| storage_.set_ssl_config_service(std::make_unique<SSLConfigServiceDefaults>()); |
| storage_.set_http_auth_handler_factory( |
| HttpAuthHandlerFactory::CreateDefault()); |
| storage_.set_http_server_properties(std::make_unique<HttpServerProperties>()); |
| storage_.set_job_factory(std::make_unique<URLRequestJobFactoryImpl>()); |
| HttpNetworkSession::Params session_params; |
| session_params.enable_spdy_ping_based_connection_checking = false; |
| |
| HttpNetworkSession::Context session_context; |
| session_context.client_socket_factory = &socket_factory_; |
| session_context.host_resolver = host_resolver(); |
| session_context.cert_verifier = cert_verifier(); |
| session_context.transport_security_state = transport_security_state(); |
| session_context.proxy_resolution_service = proxy_resolution_service(); |
| session_context.ct_policy_enforcer = ct_policy_enforcer(); |
| session_context.cert_transparency_verifier = cert_transparency_verifier(); |
| session_context.ssl_config_service = ssl_config_service(); |
| session_context.http_auth_handler_factory = http_auth_handler_factory(); |
| session_context.http_server_properties = http_server_properties(); |
| storage_.set_http_network_session( |
| std::make_unique<HttpNetworkSession>(session_params, session_context)); |
| SpdySessionPoolPeer pool_peer( |
| storage_.http_network_session()->spdy_session_pool()); |
| pool_peer.SetEnableSendingInitialData(false); |
| storage_.set_http_transaction_factory(std::make_unique<HttpCache>( |
| storage_.http_network_session(), HttpCache::DefaultBackend::InMemory(0), |
| false)); |
| } |
| |
| SpdyURLRequestContext::~SpdyURLRequestContext() { |
| AssertNoURLRequests(); |
| } |
| |
| bool HasSpdySession(SpdySessionPool* pool, const SpdySessionKey& key) { |
| return static_cast<bool>(pool->FindAvailableSession( |
| key, /* enable_ip_based_pooling = */ true, |
| /* is_websocket = */ false, NetLogWithSource())); |
| } |
| |
| namespace { |
| |
| base::WeakPtr<SpdySession> CreateSpdySessionHelper( |
| HttpNetworkSession* http_session, |
| const SpdySessionKey& key, |
| const NetLogWithSource& net_log, |
| bool enable_ip_based_pooling, |
| bool is_trusted_proxy) { |
| EXPECT_FALSE(http_session->spdy_session_pool()->FindAvailableSession( |
| key, enable_ip_based_pooling, |
| /* is_websocket = */ false, NetLogWithSource())); |
| |
| auto connection = std::make_unique<ClientSocketHandle>(); |
| TestCompletionCallback callback; |
| |
| scoped_refptr<ClientSocketPool::SocketParams> socket_params = |
| base::MakeRefCounted<ClientSocketPool::SocketParams>( |
| std::make_unique<SSLConfig>() /* ssl_config_for_origin */, |
| nullptr /* ssl_config_for_proxy */); |
| int rv = connection->Init( |
| ClientSocketPool::GroupId(key.host_port_pair(), |
| ClientSocketPool::SocketType::kSsl, |
| key.privacy_mode()), |
| socket_params, base::nullopt /* proxy_annotation_tag */, MEDIUM, |
| key.socket_tag(), ClientSocketPool::RespectLimits::ENABLED, |
| callback.callback(), ClientSocketPool::ProxyAuthCallback(), |
| http_session->GetSocketPool(HttpNetworkSession::NORMAL_SOCKET_POOL, |
| ProxyServer::Direct()), |
| net_log); |
| rv = callback.GetResult(rv); |
| EXPECT_THAT(rv, IsOk()); |
| |
| base::WeakPtr<SpdySession> spdy_session = |
| http_session->spdy_session_pool()->CreateAvailableSessionFromSocketHandle( |
| key, is_trusted_proxy, std::move(connection), net_log); |
| // Failure is reported asynchronously. |
| EXPECT_TRUE(spdy_session); |
| EXPECT_TRUE(HasSpdySession(http_session->spdy_session_pool(), key)); |
| return spdy_session; |
| } |
| |
| } // namespace |
| |
| base::WeakPtr<SpdySession> CreateSpdySession(HttpNetworkSession* http_session, |
| const SpdySessionKey& key, |
| const NetLogWithSource& net_log) { |
| return CreateSpdySessionHelper(http_session, key, net_log, |
| /* enable_ip_based_pooling = */ true, |
| /* is_trusted_proxy = */ false); |
| } |
| |
| base::WeakPtr<SpdySession> CreateTrustedSpdySession( |
| HttpNetworkSession* http_session, |
| const SpdySessionKey& key, |
| const NetLogWithSource& net_log) { |
| return CreateSpdySessionHelper(http_session, key, net_log, |
| /* enable_ip_based_pooling = */ true, |
| /* is_trusted_proxy = */ true); |
| } |
| |
| base::WeakPtr<SpdySession> CreateSpdySessionWithIpBasedPoolingDisabled( |
| HttpNetworkSession* http_session, |
| const SpdySessionKey& key, |
| const NetLogWithSource& net_log) { |
| return CreateSpdySessionHelper(http_session, key, net_log, |
| /* enable_ip_based_pooling = */ false, |
| /* is_trusted_proxy = */ false); |
| } |
| |
| namespace { |
| |
| // A ClientSocket used for CreateFakeSpdySession() below. |
| class FakeSpdySessionClientSocket : public MockClientSocket { |
| public: |
| explicit FakeSpdySessionClientSocket(int read_result) |
| : MockClientSocket(NetLogWithSource()), read_result_(read_result) {} |
| |
| ~FakeSpdySessionClientSocket() override = default; |
| |
| int Read(IOBuffer* buf, |
| int buf_len, |
| CompletionOnceCallback callback) override { |
| return read_result_; |
| } |
| |
| int Write(IOBuffer* buf, |
| int buf_len, |
| CompletionOnceCallback callback, |
| const NetworkTrafficAnnotationTag& traffic_annotation) override { |
| return ERR_IO_PENDING; |
| } |
| |
| // Return kProtoUnknown to use the pool's default protocol. |
| NextProto GetNegotiatedProtocol() const override { return kProtoUnknown; } |
| |
| // The functions below are not expected to be called. |
| |
| int Connect(CompletionOnceCallback callback) override { |
| ADD_FAILURE(); |
| return ERR_UNEXPECTED; |
| } |
| |
| bool WasEverUsed() const override { |
| ADD_FAILURE(); |
| return false; |
| } |
| |
| bool WasAlpnNegotiated() const override { |
| ADD_FAILURE(); |
| return false; |
| } |
| |
| bool GetSSLInfo(SSLInfo* ssl_info) override { |
| ADD_FAILURE(); |
| return false; |
| } |
| |
| int64_t GetTotalReceivedBytes() const override { |
| NOTIMPLEMENTED(); |
| return 0; |
| } |
| |
| private: |
| int read_result_; |
| }; |
| |
| base::WeakPtr<SpdySession> CreateFakeSpdySessionHelper( |
| SpdySessionPool* pool, |
| const SpdySessionKey& key, |
| Error expected_status) { |
| EXPECT_NE(expected_status, ERR_IO_PENDING); |
| EXPECT_FALSE(HasSpdySession(pool, key)); |
| auto handle = std::make_unique<ClientSocketHandle>(); |
| handle->SetSocket(std::make_unique<FakeSpdySessionClientSocket>( |
| expected_status == OK ? ERR_IO_PENDING : expected_status)); |
| base::WeakPtr<SpdySession> spdy_session = |
| pool->CreateAvailableSessionFromSocketHandle( |
| key, |
| /*is_trusted_proxy=*/false, std::move(handle), NetLogWithSource()); |
| // Failure is reported asynchronously. |
| EXPECT_TRUE(spdy_session); |
| EXPECT_TRUE(HasSpdySession(pool, key)); |
| return spdy_session; |
| } |
| |
| } // namespace |
| |
| base::WeakPtr<SpdySession> CreateFakeSpdySession(SpdySessionPool* pool, |
| const SpdySessionKey& key) { |
| return CreateFakeSpdySessionHelper(pool, key, OK); |
| } |
| |
| base::WeakPtr<SpdySession> TryCreateFakeSpdySessionExpectingFailure( |
| SpdySessionPool* pool, |
| const SpdySessionKey& key, |
| Error expected_status) { |
| DCHECK_LT(expected_status, ERR_IO_PENDING); |
| return CreateFakeSpdySessionHelper(pool, key, expected_status); |
| } |
| |
| SpdySessionPoolPeer::SpdySessionPoolPeer(SpdySessionPool* pool) : pool_(pool) { |
| } |
| |
| void SpdySessionPoolPeer::RemoveAliases(const SpdySessionKey& key) { |
| pool_->RemoveAliases(key); |
| } |
| |
| void SpdySessionPoolPeer::SetEnableSendingInitialData(bool enabled) { |
| pool_->enable_sending_initial_data_ = enabled; |
| } |
| |
| SpdyTestUtil::SpdyTestUtil() |
| : headerless_spdy_framer_(spdy::SpdyFramer::ENABLE_COMPRESSION), |
| request_spdy_framer_(spdy::SpdyFramer::ENABLE_COMPRESSION), |
| response_spdy_framer_(spdy::SpdyFramer::ENABLE_COMPRESSION), |
| default_url_(GURL(kDefaultUrl)) {} |
| |
| SpdyTestUtil::~SpdyTestUtil() = default; |
| |
| void SpdyTestUtil::AddUrlToHeaderBlock(base::StringPiece url, |
| spdy::SpdyHeaderBlock* headers) const { |
| std::string scheme, host, path; |
| ParseUrl(url, &scheme, &host, &path); |
| (*headers)[spdy::kHttp2AuthorityHeader] = host; |
| (*headers)[spdy::kHttp2SchemeHeader] = scheme; |
| (*headers)[spdy::kHttp2PathHeader] = path; |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructGetHeaderBlock( |
| base::StringPiece url) { |
| return ConstructHeaderBlock("GET", url, nullptr); |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructGetHeaderBlockForProxy( |
| base::StringPiece url) { |
| return ConstructGetHeaderBlock(url); |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructHeadHeaderBlock( |
| base::StringPiece url, |
| int64_t content_length) { |
| return ConstructHeaderBlock("HEAD", url, nullptr); |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructPostHeaderBlock( |
| base::StringPiece url, |
| int64_t content_length) { |
| return ConstructHeaderBlock("POST", url, &content_length); |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructPutHeaderBlock( |
| base::StringPiece url, |
| int64_t content_length) { |
| return ConstructHeaderBlock("PUT", url, &content_length); |
| } |
| |
| std::string SpdyTestUtil::ConstructSpdyReplyString( |
| const spdy::SpdyHeaderBlock& headers) const { |
| std::string reply_string; |
| for (spdy::SpdyHeaderBlock::const_iterator it = headers.begin(); |
| it != headers.end(); ++it) { |
| std::string key = it->first.as_string(); |
| // Remove leading colon from pseudo headers. |
| if (key[0] == ':') |
| key = key.substr(1); |
| for (const std::string& value : |
| base::SplitString(it->second, base::StringPiece("\0", 1), |
| base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL)) { |
| reply_string += key + ": " + value + "\n"; |
| } |
| } |
| return reply_string; |
| } |
| |
| // TODO(jgraettinger): Eliminate uses of this method in tests (prefer |
| // spdy::SpdySettingsIR). |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdySettings( |
| const spdy::SettingsMap& settings) { |
| spdy::SpdySettingsIR settings_ir; |
| for (auto it = settings.begin(); it != settings.end(); ++it) { |
| settings_ir.AddSetting(it->first, it->second); |
| } |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(settings_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdySettingsAck() { |
| spdy::SpdySettingsIR settings_ir; |
| settings_ir.set_is_ack(true); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(settings_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPing(uint32_t ping_id, |
| bool is_ack) { |
| spdy::SpdyPingIR ping_ir(ping_id); |
| ping_ir.set_is_ack(is_ack); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(ping_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyGoAway( |
| spdy::SpdyStreamId last_good_stream_id) { |
| spdy::SpdyGoAwayIR go_ir(last_good_stream_id, spdy::ERROR_CODE_NO_ERROR, |
| "go away"); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(go_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyGoAway( |
| spdy::SpdyStreamId last_good_stream_id, |
| spdy::SpdyErrorCode error_code, |
| const std::string& desc) { |
| spdy::SpdyGoAwayIR go_ir(last_good_stream_id, error_code, desc); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(go_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyWindowUpdate( |
| const spdy::SpdyStreamId stream_id, |
| uint32_t delta_window_size) { |
| spdy::SpdyWindowUpdateIR update_ir(stream_id, delta_window_size); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeFrame(update_ir)); |
| } |
| |
| // TODO(jgraettinger): Eliminate uses of this method in tests (prefer |
| // spdy::SpdyRstStreamIR). |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyRstStream( |
| spdy::SpdyStreamId stream_id, |
| spdy::SpdyErrorCode error_code) { |
| spdy::SpdyRstStreamIR rst_ir(stream_id, error_code); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeRstStream(rst_ir)); |
| } |
| |
| // TODO(jgraettinger): Eliminate uses of this method in tests (prefer |
| // spdy::SpdyPriorityIR). |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPriority( |
| spdy::SpdyStreamId stream_id, |
| spdy::SpdyStreamId parent_stream_id, |
| RequestPriority request_priority, |
| bool exclusive) { |
| int weight = spdy::Spdy3PriorityToHttp2Weight( |
| ConvertRequestPriorityToSpdyPriority(request_priority)); |
| spdy::SpdyPriorityIR ir(stream_id, parent_stream_id, weight, exclusive); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializePriority(ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyGet( |
| const char* const url, |
| spdy::SpdyStreamId stream_id, |
| RequestPriority request_priority) { |
| spdy::SpdyHeaderBlock block(ConstructGetHeaderBlock(url)); |
| return ConstructSpdyHeaders(stream_id, std::move(block), request_priority, |
| true); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyGet( |
| const char* const extra_headers[], |
| int extra_header_count, |
| int stream_id, |
| RequestPriority request_priority) { |
| spdy::SpdyHeaderBlock block; |
| block[spdy::kHttp2MethodHeader] = "GET"; |
| AddUrlToHeaderBlock(default_url_.spec(), &block); |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| return ConstructSpdyHeaders(stream_id, std::move(block), request_priority, |
| true); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyConnect( |
| const char* const extra_headers[], |
| int extra_header_count, |
| int stream_id, |
| RequestPriority priority, |
| const HostPortPair& host_port_pair) { |
| spdy::SpdyHeaderBlock block; |
| block[spdy::kHttp2MethodHeader] = "CONNECT"; |
| block[spdy::kHttp2AuthorityHeader] = host_port_pair.ToString(); |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| return ConstructSpdyHeaders(stream_id, std::move(block), priority, false); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPush( |
| const char* const extra_headers[], |
| int extra_header_count, |
| int stream_id, |
| int associated_stream_id, |
| const char* url) { |
| spdy::SpdyHeaderBlock push_promise_header_block; |
| push_promise_header_block[spdy::kHttp2MethodHeader] = "GET"; |
| AddUrlToHeaderBlock(url, &push_promise_header_block); |
| spdy::SpdySerializedFrame push_promise_frame(ConstructSpdyPushPromise( |
| associated_stream_id, stream_id, std::move(push_promise_header_block))); |
| |
| spdy::SpdyHeaderBlock headers_header_block; |
| headers_header_block[spdy::kHttp2StatusHeader] = "200"; |
| headers_header_block["hello"] = "bye"; |
| AppendToHeaderBlock(extra_headers, extra_header_count, &headers_header_block); |
| spdy::SpdyHeadersIR headers(stream_id, std::move(headers_header_block)); |
| spdy::SpdySerializedFrame headers_frame( |
| response_spdy_framer_.SerializeFrame(headers)); |
| |
| return CombineFrames({&push_promise_frame, &headers_frame}); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPush( |
| const char* const extra_headers[], |
| int extra_header_count, |
| int stream_id, |
| int associated_stream_id, |
| const char* url, |
| const char* status, |
| const char* location) { |
| spdy::SpdyHeaderBlock push_promise_header_block; |
| push_promise_header_block[spdy::kHttp2MethodHeader] = "GET"; |
| AddUrlToHeaderBlock(url, &push_promise_header_block); |
| spdy::SpdySerializedFrame push_promise_frame(ConstructSpdyPushPromise( |
| associated_stream_id, stream_id, std::move(push_promise_header_block))); |
| |
| spdy::SpdyHeaderBlock headers_header_block; |
| headers_header_block["hello"] = "bye"; |
| headers_header_block[spdy::kHttp2StatusHeader] = status; |
| headers_header_block["location"] = location; |
| AppendToHeaderBlock(extra_headers, extra_header_count, &headers_header_block); |
| spdy::SpdyHeadersIR headers(stream_id, std::move(headers_header_block)); |
| spdy::SpdySerializedFrame headers_frame( |
| response_spdy_framer_.SerializeFrame(headers)); |
| |
| return CombineFrames({&push_promise_frame, &headers_frame}); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPushPromise( |
| spdy::SpdyStreamId associated_stream_id, |
| spdy::SpdyStreamId stream_id, |
| spdy::SpdyHeaderBlock headers) { |
| spdy::SpdyPushPromiseIR push_promise(associated_stream_id, stream_id, |
| std::move(headers)); |
| return spdy::SpdySerializedFrame( |
| response_spdy_framer_.SerializeFrame(push_promise)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPushHeaders( |
| int stream_id, |
| const char* const extra_headers[], |
| int extra_header_count) { |
| spdy::SpdyHeaderBlock header_block; |
| header_block[spdy::kHttp2StatusHeader] = "200"; |
| AppendToHeaderBlock(extra_headers, extra_header_count, &header_block); |
| spdy::SpdyHeadersIR headers(stream_id, std::move(header_block)); |
| return spdy::SpdySerializedFrame( |
| response_spdy_framer_.SerializeFrame(headers)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyResponseHeaders( |
| int stream_id, |
| spdy::SpdyHeaderBlock headers, |
| bool fin) { |
| spdy::SpdyHeadersIR spdy_headers(stream_id, std::move(headers)); |
| spdy_headers.set_fin(fin); |
| return spdy::SpdySerializedFrame( |
| response_spdy_framer_.SerializeFrame(spdy_headers)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyHeaders( |
| int stream_id, |
| spdy::SpdyHeaderBlock block, |
| RequestPriority priority, |
| bool fin) { |
| // Get the stream id of the next highest priority request |
| // (most recent request of the same priority, or last request of |
| // an earlier priority). |
| // Note that this is a duplicate of the logic in Http2PriorityDependencies |
| // (slightly transformed as this is based on RequestPriority and that logic |
| // on spdy::SpdyPriority, but only slightly transformed) and hence tests using |
| // this function do not effectively test that logic. |
| // That logic is tested by the Http2PriorityDependencies unit tests. |
| int parent_stream_id = 0; |
| for (int q = priority; q <= HIGHEST; ++q) { |
| if (!priority_to_stream_id_list_[q].empty()) { |
| parent_stream_id = priority_to_stream_id_list_[q].back(); |
| break; |
| } |
| } |
| |
| priority_to_stream_id_list_[priority].push_back(stream_id); |
| |
| spdy::SpdyHeadersIR headers(stream_id, std::move(block)); |
| headers.set_has_priority(true); |
| headers.set_weight(spdy::Spdy3PriorityToHttp2Weight( |
| ConvertRequestPriorityToSpdyPriority(priority))); |
| headers.set_parent_stream_id(parent_stream_id); |
| headers.set_exclusive(true); |
| headers.set_fin(fin); |
| return spdy::SpdySerializedFrame( |
| request_spdy_framer_.SerializeFrame(headers)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyReply( |
| int stream_id, |
| spdy::SpdyHeaderBlock headers) { |
| spdy::SpdyHeadersIR reply(stream_id, std::move(headers)); |
| return spdy::SpdySerializedFrame(response_spdy_framer_.SerializeFrame(reply)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyReplyError( |
| const char* const status, |
| const char* const* const extra_headers, |
| int extra_header_count, |
| int stream_id) { |
| spdy::SpdyHeaderBlock block; |
| block[spdy::kHttp2StatusHeader] = status; |
| block["hello"] = "bye"; |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| |
| spdy::SpdyHeadersIR reply(stream_id, std::move(block)); |
| return spdy::SpdySerializedFrame(response_spdy_framer_.SerializeFrame(reply)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyReplyError(int stream_id) { |
| return ConstructSpdyReplyError("500", nullptr, 0, 1); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyGetReply( |
| const char* const extra_headers[], |
| int extra_header_count, |
| int stream_id) { |
| spdy::SpdyHeaderBlock block; |
| block[spdy::kHttp2StatusHeader] = "200"; |
| block["hello"] = "bye"; |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| |
| return ConstructSpdyReply(stream_id, std::move(block)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPost( |
| const char* url, |
| spdy::SpdyStreamId stream_id, |
| int64_t content_length, |
| RequestPriority priority, |
| const char* const extra_headers[], |
| int extra_header_count) { |
| spdy::SpdyHeaderBlock block(ConstructPostHeaderBlock(url, content_length)); |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| return ConstructSpdyHeaders(stream_id, std::move(block), priority, false); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructChunkedSpdyPost( |
| const char* const extra_headers[], |
| int extra_header_count) { |
| spdy::SpdyHeaderBlock block; |
| block[spdy::kHttp2MethodHeader] = "POST"; |
| AddUrlToHeaderBlock(default_url_.spec(), &block); |
| AppendToHeaderBlock(extra_headers, extra_header_count, &block); |
| return ConstructSpdyHeaders(1, std::move(block), LOWEST, false); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyPostReply( |
| const char* const extra_headers[], |
| int extra_header_count) { |
| // TODO(jgraettinger): Remove this method. |
| return ConstructSpdyGetReply(extra_headers, extra_header_count, 1); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyDataFrame(int stream_id, |
| bool fin) { |
| return ConstructSpdyDataFrame(stream_id, kUploadData, fin); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyDataFrame( |
| int stream_id, |
| base::StringPiece data, |
| bool fin) { |
| spdy::SpdyDataIR data_ir(stream_id, data); |
| data_ir.set_fin(fin); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeData(data_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructSpdyDataFrame( |
| int stream_id, |
| base::StringPiece data, |
| bool fin, |
| int padding_length) { |
| spdy::SpdyDataIR data_ir(stream_id, data); |
| data_ir.set_fin(fin); |
| data_ir.set_padding_len(padding_length); |
| return spdy::SpdySerializedFrame( |
| headerless_spdy_framer_.SerializeData(data_ir)); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::ConstructWrappedSpdyFrame( |
| const spdy::SpdySerializedFrame& frame, |
| int stream_id) { |
| return ConstructSpdyDataFrame( |
| stream_id, base::StringPiece(frame.data(), frame.size()), false); |
| } |
| |
| spdy::SpdySerializedFrame SpdyTestUtil::SerializeFrame( |
| const spdy::SpdyFrameIR& frame_ir) { |
| return headerless_spdy_framer_.SerializeFrame(frame_ir); |
| } |
| |
| void SpdyTestUtil::UpdateWithStreamDestruction(int stream_id) { |
| for (auto priority_it = priority_to_stream_id_list_.begin(); |
| priority_it != priority_to_stream_id_list_.end(); ++priority_it) { |
| for (auto stream_it = priority_it->second.begin(); |
| stream_it != priority_it->second.end(); ++stream_it) { |
| if (*stream_it == stream_id) { |
| priority_it->second.erase(stream_it); |
| return; |
| } |
| } |
| } |
| NOTREACHED(); |
| } |
| |
| // static |
| spdy::SpdyHeaderBlock SpdyTestUtil::ConstructHeaderBlock( |
| base::StringPiece method, |
| base::StringPiece url, |
| int64_t* content_length) { |
| std::string scheme, host, path; |
| ParseUrl(url, &scheme, &host, &path); |
| spdy::SpdyHeaderBlock headers; |
| headers[spdy::kHttp2MethodHeader] = method.as_string(); |
| headers[spdy::kHttp2AuthorityHeader] = host.c_str(); |
| headers[spdy::kHttp2SchemeHeader] = scheme.c_str(); |
| headers[spdy::kHttp2PathHeader] = path.c_str(); |
| if (content_length) { |
| std::string length_str = base::NumberToString(*content_length); |
| headers["content-length"] = length_str; |
| } |
| return headers; |
| } |
| |
| namespace test { |
| HashValue GetTestHashValue(uint8_t label) { |
| HashValue hash_value(HASH_VALUE_SHA256); |
| memset(hash_value.data(), label, hash_value.size()); |
| return hash_value; |
| } |
| |
| TestServerPushDelegate::TestServerPushDelegate() = default; |
| |
| TestServerPushDelegate::~TestServerPushDelegate() = default; |
| |
| void TestServerPushDelegate::OnPush( |
| std::unique_ptr<ServerPushHelper> push_helper, |
| const NetLogWithSource& session_net_log) { |
| push_helpers[push_helper->GetURL()] = std::move(push_helper); |
| } |
| |
| bool TestServerPushDelegate::CancelPush(GURL url) { |
| auto itr = push_helpers.find(url); |
| DCHECK(itr != push_helpers.end()); |
| itr->second->Cancel(); |
| push_helpers.erase(itr); |
| return true; |
| } |
| |
| } // namespace test |
| } // namespace net |