marcel/zoraxy
1
0
Fork 0
mirror of https://github.com/tobychui/zoraxy.git synced 2025-11-16 22:04:39 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add h2c support check and improve RoundTrip logic

Browse Source
This commit is contained in:
Toby Chui
2025-10-29 07:59:32 +08:00
parent 5f420b4636
commit 42d6b61770
2 changed files with 104 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/mod/dynamicproxy/modh2c/modh2c.go
View File

@@ -28,11 +28,14 @@ func (h2c *H2CRoundTripper) RoundTrip(req *http.Request) (*http.Response, error)
ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
defer cancel()
req, err := http.NewRequestWithContext(ctx, req.Method, req.RequestURI, nil)
req, err := http.NewRequestWithContext(ctx, req.Method, req.URL.String(), req.Body)
if err != nil {
return nil, err
}
// Copy headers
req.Header = req.Header.Clone()
tr := &http2.Transport{
AllowHTTP: true,
DialTLSContext: func(ctx context.Context, network, addr string, cfg *tls.Config) (net.Conn, error) {
@@ -43,3 +46,20 @@ func (h2c *H2CRoundTripper) RoundTrip(req *http.Request) (*http.Response, error)
return tr.RoundTrip(req)
}
func (h2c *H2CRoundTripper) CheckServerSupportsH2C(serverURL string) bool {
req, err := http.NewRequest("GET", serverURL, nil)
if err != nil {
return false
}
tr := &http2.Transport{
AllowHTTP: true,
DialTLSContext: func(ctx context.Context, network, addr string, cfg *tls.Config) (net.Conn, error) {
var d net.Dialer
return d.DialContext(ctx, network, addr)
},
}
_, err = tr.RoundTrip(req)
return err == nil
}