95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
|
// Copyright 2011 The Go Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
// Package proxy provides support for a variety of protocols to proxy network
|
||
|
|
// data.
|
||
|
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net"
|
||
|
|
"net/url"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
// A Dialer is a means to establish a connection.
|
||
|
|
type Dialer interface {
|
||
|
|
// Dial connects to the given address via the proxy.
|
||
|
|
Dial(network, addr string) (c net.Conn, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Auth contains authentication parameters that specific Dialers may require.
|
||
|
|
type Auth struct {
|
||
|
|
User, Password string
|
||
|
|
}
|
||
|
|
|
||
|
|
// FromEnvironment returns the dialer specified by the proxy related variables in
|
||
|
|
// the environment.
|
||
|
|
func FromEnvironment() Dialer {
|
||
|
|
allProxy := os.Getenv("all_proxy")
|
||
|
|
if len(allProxy) == 0 {
|
||
|
|
return Direct
|
||
|
|
}
|
||
|
|
|
||
|
|
proxyURL, err := url.Parse(allProxy)
|
||
|
|
if err != nil {
|
||
|
|
return Direct
|
||
|
|
}
|
||
|
|
proxy, err := FromURL(proxyURL, Direct)
|
||
|
|
if err != nil {
|
||
|
|
return Direct
|
||
|
|
}
|
||
|
|
|
||
|
|
noProxy := os.Getenv("no_proxy")
|
||
|
|
if len(noProxy) == 0 {
|
||
|
|
return proxy
|
||
|
|
}
|
||
|
|
|
||
|
|
perHost := NewPerHost(proxy, Direct)
|
||
|
|
perHost.AddFromString(noProxy)
|
||
|
|
return perHost
|
||
|
|
}
|
||
|
|
|
||
|
|
// proxySchemes is a map from URL schemes to a function that creates a Dialer
|
||
|
|
// from a URL with such a scheme.
|
||
|
|
var proxySchemes map[string]func(*url.URL, Dialer) (Dialer, error)
|
||
|
|
|
||
|
|
// RegisterDialerType takes a URL scheme and a function to generate Dialers from
|
||
|
|
// a URL with that scheme and a forwarding Dialer. Registered schemes are used
|
||
|
|
// by FromURL.
|
||
|
|
func RegisterDialerType(scheme string, f func(*url.URL, Dialer) (Dialer, error)) {
|
||
|
|
if proxySchemes == nil {
|
||
|
|
proxySchemes = make(map[string]func(*url.URL, Dialer) (Dialer, error))
|
||
|
|
}
|
||
|
|
proxySchemes[scheme] = f
|
||
|
|
}
|
||
|
|
|
||
|
|
// FromURL returns a Dialer given a URL specification and an underlying
|
||
|
|
// Dialer for it to make network requests.
|
||
|
|
func FromURL(u *url.URL, forward Dialer) (Dialer, error) {
|
||
|
|
var auth *Auth
|
||
|
|
if u.User != nil {
|
||
|
|
auth = new(Auth)
|
||
|
|
auth.User = u.User.Username()
|
||
|
|
if p, ok := u.User.Password(); ok {
|
||
|
|
auth.Password = p
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
switch u.Scheme {
|
||
|
|
case "socks5":
|
||
|
|
return SOCKS5("tcp", u.Host, auth, forward)
|
||
|
|
}
|
||
|
|
|
||
|
|
// If the scheme doesn't match any of the built-in schemes, see if it
|
||
|
|
// was registered by another package.
|
||
|
|
if proxySchemes != nil {
|
||
|
|
if f, ok := proxySchemes[u.Scheme]; ok {
|
||
|
|
return f(u, forward)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, errors.New("proxy: unknown scheme: " + u.Scheme)
|
||
|
|
}
|