mirror/naiveproxy
1
0
Fork 0
mirror of https://github.com/klzgrad/naiveproxy.git synced 2024-11-21 21:06:12 +03:00
Code Issues Projects Releases Wiki Activity

quic: Add support for HTTP/3 CONNECT Fast Open

Browse Source 
SpdyProxyClientSocket uses read_callback_ for both Connect() and
Read(), and its OnIOComplete() calls read_callback_, thus its fast
connect code checks read_callback_. The code was ported to
QuicProxyClientSocket without much change.

But QuicProxyClientSocket uses a separate connect_callback_ apart from
read_callback_, and its OnIOComplete() calls connect_callback_, thus
when headers are received after Connect() it doesn't need to check
read_callback_ and should always avoid calling connect_callback_.
This commit is contained in:
klzgrad 2019-01-17 05:47:20 -05:00
parent b01404292a
commit 23b8a36e7a
2 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/net/quic/quic_proxy_client_socket.cc
View File

@ -41,6 +41,8 @@ QuicProxyClientSocket::QuicProxyClientSocket(
proxy_chain_index_(proxy_chain_index),
proxy_delegate_(proxy_delegate),
user_agent_(user_agent),
use_fastopen_(false),
read_headers_pending_(false),
net_log_(net_log) {
DCHECK(stream_->IsOpen());
@ -309,6 +311,16 @@ int QuicProxyClientSocket::DoLoop(int last_io_result) {
rv = DoReadReplyComplete(rv);
net_log_.EndEventWithNetErrorCode(
NetLogEventType::HTTP_TRANSACTION_TUNNEL_READ_HEADERS, rv);
if (use_fastopen_ && read_headers_pending_) {
read_headers_pending_ = false;
if (rv < 0) {
// read_callback_ will be called with this error and be reset.
// Further data after that will be ignored.
next_state_ = STATE_DISCONNECTED;
}
// Prevents calling connect_callback_.
rv = ERR_IO_PENDING;
}
break;
default:
NOTREACHED() << "bad state";
@ -349,6 +361,11 @@ int QuicProxyClientSocket::DoSendRequest() {
HttpRequestHeaders proxy_delegate_headers;
proxy_delegate_->OnBeforeTunnelRequestServerOnly(
proxy_chain_, proxy_chain_index_, &proxy_delegate_headers);
if (proxy_delegate_headers.HasHeader("fastopen")) {
proxy_delegate_headers.RemoveHeader("fastopen");
// TODO(klzgrad): look into why Fast Open does not work.
use_fastopen_ = true;
}
request_.extra_headers.MergeFrom(proxy_delegate_headers);
}
@ -389,6 +406,11 @@ int QuicProxyClientSocket::DoReadReply() {
&response_header_block_,
base::BindOnce(&QuicProxyClientSocket::OnReadResponseHeadersComplete,
weak_factory_.GetWeakPtr()));
if (use_fastopen_ && rv == ERR_IO_PENDING) {
read_headers_pending_ = true;
next_state_ = STATE_CONNECT_COMPLETE;
return OK;
}
if (rv == ERR_IO_PENDING)
return ERR_IO_PENDING;
if (rv < 0)
@ -437,6 +459,13 @@ int QuicProxyClientSocket::DoReadReplyComplete(int result) {
void QuicProxyClientSocket::OnReadResponseHeadersComplete(int result) {
// Convert the now-populated spdy::Http2HeaderBlock to HttpResponseInfo
if (use_fastopen_ && read_headers_pending_) {
if (next_state_ == STATE_DISCONNECTED)
return;
if (next_state_ == STATE_CONNECT_COMPLETE)
next_state_ = STATE_READ_REPLY_COMPLETE;
}
if (result > 0)
result = ProcessResponseHeaders(response_header_block_);

6
src/net/quic/quic_proxy_client_socket.h
View File

@ -146,6 +146,12 @@ class NET_EXPORT_PRIVATE QuicProxyClientSocket : public ProxyClientSocket {
std::string user_agent_;
// Session connect timing info.
LoadTimingInfo::ConnectTiming connect_timing_;
bool use_fastopen_;
bool read_headers_pending_;
const NetLogWithSource net_log_;
// The default weak pointer factory.