46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
// Copyright (c) 2020 Tailscale Inc & AUTHORS All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build tailscale_go
|
|
|
|
// We want to use https://github.com/golang/go/issues/41048 but it's only in the
|
|
// Tailscale Go tree for now. Hence the build tag above.
|
|
|
|
package tshttpproxy
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
const proxyAuthHeader = "Proxy-Authorization"
|
|
|
|
func init() {
|
|
condSetTransportGetProxyConnectHeader = func(tr *http.Transport) {
|
|
tr.GetProxyConnectHeader = func(ctx context.Context, proxyURL *url.URL, target string) (http.Header, error) {
|
|
v, err := GetAuthHeader(proxyURL)
|
|
if err != nil {
|
|
log.Printf("failed to get proxy Auth header for %v; ignoring: %v", proxyURL, err)
|
|
return nil, nil
|
|
}
|
|
if v == "" {
|
|
return nil, nil
|
|
}
|
|
return http.Header{proxyAuthHeader: []string{v}}, nil
|
|
}
|
|
tr.OnProxyConnectResponse = func(ctx context.Context, proxyURL *url.URL, connectReq *http.Request, res *http.Response) error {
|
|
auth := connectReq.Header.Get(proxyAuthHeader)
|
|
const truncLen = 20
|
|
if len(auth) > truncLen {
|
|
auth = fmt.Sprintf("%s...(%d total bytes)", auth[:truncLen], len(auth))
|
|
}
|
|
log.Printf("tshttpproxy: CONNECT response from %v for target %q (auth %q): %v", proxyURL, connectReq.Host, auth, res.Status)
|
|
return nil
|
|
}
|
|
}
|
|
}
|