mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-03 20:27:35 +03:00
Code Issues Projects Releases Packages Wiki Activity

http3: correctly use the quic.Transport (#3869)

Browse source 
* use quic.Transport in http3

* add intergrationtests to dial server with different server names

* update test
This commit is contained in:
Glonee 2023-06-01 14:31:20 +08:00 • committed by GitHub
parent 21549fcb4a
commit c96fbd2e4a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 57 additions and 90 deletions
Download patch file Download diff file Expand all files Collapse all files

18
http3/roundtrip.go
View file

@ -17,9 +17,6 @@ import (
"github.com/quic-go/quic-go"
)
// declare this as a variable, such that we can it mock it in the tests
var quicDialer = quic.DialEarly
type roundTripCloser interface {
RoundTripOpt(*http.Request, RoundTripOpt) (*http.Response, error)
HandshakeComplete() bool
@ -89,7 +86,7 @@ type RoundTripper struct {
newClient func(hostname string, tlsConf *tls.Config, opts *roundTripperOpts, conf *quic.Config, dialer dialFunc) (roundTripCloser, error) // so we can mock it in tests
clients map[string]*roundTripCloserWithCount
udpConn *net.UDPConn
transport *quic.Transport
}
// RoundTripOpt are options for the Transport.RoundTripOpt method.
@ -187,11 +184,12 @@ func (r *RoundTripper) getClient(hostname string, onlyCached bool) (rtc *roundTr
}
dial := r.Dial
if dial == nil {
if r.udpConn == nil {
r.udpConn, err = net.ListenUDP("udp", nil)
if r.transport == nil {
udpConn, err := net.ListenUDP("udp", nil)
if err != nil {
return nil, false, err
}
r.transport = &quic.Transport{Conn: udpConn}
}
dial = r.makeDialer()
}
@ -240,9 +238,9 @@ func (r *RoundTripper) Close() error {
}
}
r.clients = nil
if r.udpConn != nil {
r.udpConn.Close()
r.udpConn = nil
if r.transport != nil {
r.transport.Close()
r.transport = nil
}
return nil
}
@ -282,7 +280,7 @@ func (r *RoundTripper) makeDialer() func(ctx context.Context, addr string, tlsCf
if err != nil {
return nil, err
}
return quicDialer(ctx, r.udpConn, udpAddr, tlsCfg, cfg)
return r.transport.DialEarly(ctx, udpAddr, tlsCfg, cfg)
}
}