frp/vendor/golang.org/x/net/proxy/socks5.go

43 lines
1.1 KiB
Go
Raw Normal View History

2017-11-01 16:21:57 +08:00
// 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
import (
2018-05-04 18:37:43 +08:00
"context"
2017-11-01 16:21:57 +08:00
"net"
2018-05-04 18:37:43 +08:00
"golang.org/x/net/internal/socks"
2017-11-01 16:21:57 +08:00
)
2018-05-04 18:37:43 +08:00
// SOCKS5 returns a Dialer that makes SOCKSv5 connections to the given
// address with an optional username and password.
// See RFC 1928 and RFC 1929.
func SOCKS5(network, address string, auth *Auth, forward Dialer) (Dialer, error) {
d := socks.NewDialer(network, address)
if forward != nil {
if f, ok := forward.(ContextDialer); ok {
d.ProxyDial = func(ctx context.Context, network string, address string) (net.Conn, error) {
return f.DialContext(ctx, network, address)
}
} else {
d.ProxyDial = func(ctx context.Context, network string, address string) (net.Conn, error) {
return dialContext(ctx, forward, network, address)
}
2017-11-01 16:21:57 +08:00
}
}
2018-05-04 18:37:43 +08:00
if auth != nil {
up := socks.UsernamePassword{
Username: auth.User,
Password: auth.Password,
2017-11-01 16:21:57 +08:00
}
2018-05-04 18:37:43 +08:00
d.AuthMethods = []socks.AuthMethod{
socks.AuthMethodNotRequired,
socks.AuthMethodUsernamePassword,
2017-11-01 16:21:57 +08:00
}
2018-05-04 18:37:43 +08:00
d.Authenticate = up.Authenticate
2017-11-01 16:21:57 +08:00
}
2018-05-04 18:37:43 +08:00
return d, nil
2017-11-01 16:21:57 +08:00
}