1 // Copyright 2024 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 #ifndef NET_QUIC_QUIC_SESSION_POOL_PROXY_JOB_H_ 6 #define NET_QUIC_QUIC_SESSION_POOL_PROXY_JOB_H_ 7 8 #include <memory> 9 10 #include "net/base/http_user_agent_settings.h" 11 #include "net/base/request_priority.h" 12 #include "net/log/net_log_with_source.h" 13 #include "net/quic/quic_session_attempt.h" 14 #include "net/quic/quic_session_pool.h" 15 #include "net/quic/quic_session_pool_job.h" 16 #include "net/spdy/multiplexed_session_creation_initiator.h" 17 #include "net/third_party/quiche/src/quiche/quic/core/quic_versions.h" 18 #include "net/traffic_annotation/network_traffic_annotation.h" 19 20 namespace net { 21 22 // A ProxyJob is a QuicSessionPool::Job that handles connections to the 23 // destination over a (QUIC) proxy. 24 class QuicSessionPool::ProxyJob : public QuicSessionPool::Job { 25 public: 26 ProxyJob(QuicSessionPool* pool, 27 quic::ParsedQuicVersion target_quic_version, 28 QuicSessionAliasKey key, 29 NetworkTrafficAnnotationTag proxy_annotation_tag, 30 MultiplexedSessionCreationInitiator session_creation_initiator, 31 const HttpUserAgentSettings* http_user_agent_settings, 32 std::unique_ptr<CryptoClientConfigHandle> client_config_handle, 33 RequestPriority priority, 34 int cert_verify_flags, 35 const NetLogWithSource& net_log); 36 37 ~ProxyJob() override; 38 39 // QuicSessionPool::Job implementation. 40 int Run(CompletionOnceCallback callback) override; 41 void SetRequestExpectations(QuicSessionRequest* request) override; 42 void PopulateNetErrorDetails(NetErrorDetails* details) const override; 43 44 private: 45 int DoLoop(int rv); 46 int DoCreateProxySession(); 47 int DoCreateProxySessionComplete(int rv); 48 int DoCreateProxyStream(); 49 int DoCreateProxyStreamComplete(int rv); 50 int DoAttemptSession(); 51 52 void OnIOComplete(int rv); 53 void OnSessionAttemptComplete(int rv); 54 GetWeakPtr()55 base::WeakPtr<ProxyJob> GetWeakPtr() { return weak_factory_.GetWeakPtr(); } 56 57 enum IoState { 58 STATE_NONE, 59 STATE_CREATE_PROXY_SESSION, 60 STATE_CREATE_PROXY_SESSION_COMPLETE, 61 STATE_CREATE_PROXY_STREAM, 62 STATE_CREATE_PROXY_STREAM_COMPLETE, 63 STATE_ATTEMPT_SESSION, 64 }; 65 66 CompletionRepeatingCallback io_callback_; 67 IoState io_state_ = STATE_CREATE_PROXY_SESSION; 68 69 std::unique_ptr<QuicSessionRequest> proxy_session_request_; 70 std::unique_ptr<QuicChromiumClientSession::Handle> proxy_session_; 71 std::unique_ptr<QuicChromiumClientStream::Handle> proxy_stream_; 72 NetErrorDetails net_error_details_; 73 74 // The QUIC version for the tunneled session created by this job. 75 quic::ParsedQuicVersion target_quic_version_; 76 77 NetworkTrafficAnnotationTag proxy_annotation_tag_; 78 MultiplexedSessionCreationInitiator session_creation_initiator_; 79 const int cert_verify_flags_; 80 raw_ptr<const HttpUserAgentSettings> http_user_agent_settings_; 81 CompletionOnceCallback callback_; 82 std::unique_ptr<QuicSessionAttempt> session_attempt_; 83 base::WeakPtrFactory<ProxyJob> weak_factory_{this}; 84 }; 85 86 } // namespace net 87 88 #endif // NET_QUIC_QUIC_SESSION_POOL_PROXY_JOB_H_ 89