1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
package config
import (
"net/url"
"golang.org/x/net/proxy"
ournet "github.com/twstrike/coyim/net"
)
func socks5UnixProxy(u *url.URL, d proxy.Dialer) (proxy.Dialer, error) {
var auth *proxy.Auth
if u.User != nil {
auth = &proxy.Auth{
User: u.User.Username(),
}
if p, ok := u.User.Password(); ok {
auth.Password = p
}
}
return proxy.SOCKS5("unix", u.Path, auth, d)
}
func genTorAutoString() string {
s := [10]byte{}
randomString(s[:])
return "randomTorAuto-" + string(s[:])
}
func genTorAutoUsername() string {
return genTorAutoString()
}
func genTorAutoPassword() string {
return genTorAutoString()
}
func genTorAutoAuth(u *url.URL) *proxy.Auth {
auth := &proxy.Auth{}
if u.User != nil {
auth.User = u.User.Username()
if p, ok := u.User.Password(); ok {
auth.Password = p
}
} else {
auth.User = genTorAutoUsername()
auth.Password = genTorAutoPassword()
}
return auth
}
func genTorAutoAddr(u *url.URL) string {
if u.Host == "" {
return ournet.Tor.Address()
}
return u.Host
}
func torAutoProxy(u *url.URL, d proxy.Dialer) (proxy.Dialer, error) {
auth := genTorAutoAuth(u)
addr := genTorAutoAddr(u)
if addr == "" {
return nil, ErrTorNotRunning
}
return proxy.SOCKS5("tcp", addr, auth, d)
}
func init() {
proxy.RegisterDialerType("socks5+unix", socks5UnixProxy)
proxy.RegisterDialerType("tor-auto", torAutoProxy)
}
|