You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.1 KiB
51 lines
1.1 KiB
package sockets |
|
|
|
import ( |
|
"net" |
|
"net/url" |
|
"os" |
|
"strings" |
|
|
|
"golang.org/x/net/proxy" |
|
) |
|
|
|
// GetProxyEnv allows access to the uppercase and the lowercase forms of |
|
// proxy-related variables. See the Go specification for details on these |
|
// variables. https://golang.org/pkg/net/http/ |
|
func GetProxyEnv(key string) string { |
|
proxyValue := os.Getenv(strings.ToUpper(key)) |
|
if proxyValue == "" { |
|
return os.Getenv(strings.ToLower(key)) |
|
} |
|
return proxyValue |
|
} |
|
|
|
// DialerFromEnvironment takes in a "direct" *net.Dialer and returns a |
|
// proxy.Dialer which will route the connections through the proxy using the |
|
// given dialer. |
|
func DialerFromEnvironment(direct *net.Dialer) (proxy.Dialer, error) { |
|
allProxy := GetProxyEnv("all_proxy") |
|
if len(allProxy) == 0 { |
|
return direct, nil |
|
} |
|
|
|
proxyURL, err := url.Parse(allProxy) |
|
if err != nil { |
|
return direct, err |
|
} |
|
|
|
proxyFromURL, err := proxy.FromURL(proxyURL, direct) |
|
if err != nil { |
|
return direct, err |
|
} |
|
|
|
noProxy := GetProxyEnv("no_proxy") |
|
if len(noProxy) == 0 { |
|
return proxyFromURL, nil |
|
} |
|
|
|
perHost := proxy.NewPerHost(proxyFromURL, direct) |
|
perHost.AddFromString(noProxy) |
|
|
|
return perHost, nil |
|
}
|
|
|