fix log usage

This commit is contained in:
ginuerzh 2022-01-26 16:45:57 +08:00
parent 04dfc8c4c3
commit 713586729b
16 changed files with 183 additions and 185 deletions

View File

@ -3,7 +3,6 @@ package main
import ( import (
// Register connectors // Register connectors
_ "github.com/go-gost/gost/pkg/connector/forward" _ "github.com/go-gost/gost/pkg/connector/forward"
_ "github.com/go-gost/gost/pkg/connector/forward/ssh"
_ "github.com/go-gost/gost/pkg/connector/http" _ "github.com/go-gost/gost/pkg/connector/http"
_ "github.com/go-gost/gost/pkg/connector/http2" _ "github.com/go-gost/gost/pkg/connector/http2"
_ "github.com/go-gost/gost/pkg/connector/relay" _ "github.com/go-gost/gost/pkg/connector/relay"
@ -12,9 +11,9 @@ import (
_ "github.com/go-gost/gost/pkg/connector/socks/v5" _ "github.com/go-gost/gost/pkg/connector/socks/v5"
_ "github.com/go-gost/gost/pkg/connector/ss" _ "github.com/go-gost/gost/pkg/connector/ss"
_ "github.com/go-gost/gost/pkg/connector/ss/udp" _ "github.com/go-gost/gost/pkg/connector/ss/udp"
_ "github.com/go-gost/gost/pkg/connector/sshd"
// Register dialers // Register dialers
_ "github.com/go-gost/gost/pkg/dialer/forward/ssh"
_ "github.com/go-gost/gost/pkg/dialer/ftcp" _ "github.com/go-gost/gost/pkg/dialer/ftcp"
_ "github.com/go-gost/gost/pkg/dialer/http2" _ "github.com/go-gost/gost/pkg/dialer/http2"
_ "github.com/go-gost/gost/pkg/dialer/http2/h2" _ "github.com/go-gost/gost/pkg/dialer/http2/h2"
@ -25,6 +24,7 @@ import (
_ "github.com/go-gost/gost/pkg/dialer/pht" _ "github.com/go-gost/gost/pkg/dialer/pht"
_ "github.com/go-gost/gost/pkg/dialer/quic" _ "github.com/go-gost/gost/pkg/dialer/quic"
_ "github.com/go-gost/gost/pkg/dialer/ssh" _ "github.com/go-gost/gost/pkg/dialer/ssh"
_ "github.com/go-gost/gost/pkg/dialer/sshd"
_ "github.com/go-gost/gost/pkg/dialer/tcp" _ "github.com/go-gost/gost/pkg/dialer/tcp"
_ "github.com/go-gost/gost/pkg/dialer/tls" _ "github.com/go-gost/gost/pkg/dialer/tls"
_ "github.com/go-gost/gost/pkg/dialer/tls/mux" _ "github.com/go-gost/gost/pkg/dialer/tls/mux"

View File

@ -5,7 +5,6 @@ import (
"net" "net"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
) )
@ -15,17 +14,17 @@ func init() {
} }
type forwardConnector struct { type forwardConnector struct {
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &forwardConnector{ return &forwardConnector{
logger: options.Logger, options: options,
} }
} }
@ -34,13 +33,13 @@ func (c *forwardConnector) Init(md md.Metadata) (err error) {
} }
func (c *forwardConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *forwardConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
return conn, nil return conn, nil
} }

View File

@ -25,18 +25,18 @@ func init() {
type httpConnector struct { type httpConnector struct {
user *url.Userinfo user *url.Userinfo
md metadata md metadata
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &httpConnector{ return &httpConnector{
user: options.User, user: options.User,
logger: options.Logger, options: options,
} }
} }
@ -45,13 +45,13 @@ func (c *httpConnector) Init(md md.Metadata) (err error) {
} }
func (c *httpConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *httpConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
req := &http.Request{ req := &http.Request{
Method: http.MethodConnect, Method: http.MethodConnect,
@ -78,20 +78,20 @@ func (c *httpConnector) Connect(ctx context.Context, conn net.Conn, network, add
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
if _, ok := conn.(net.PacketConn); ok { if _, ok := conn.(net.PacketConn); ok {
err := fmt.Errorf("tcp over udp is unsupported") err := fmt.Errorf("tcp over udp is unsupported")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
req.Header.Set("X-Gost-Protocol", "udp") req.Header.Set("X-Gost-Protocol", "udp")
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
if c.logger.IsLevelEnabled(logger.DebugLevel) { if log.IsLevelEnabled(logger.DebugLevel) {
dump, _ := httputil.DumpRequest(req, false) dump, _ := httputil.DumpRequest(req, false)
c.logger.Debug(string(dump)) log.Debug(string(dump))
} }
if c.md.connectTimeout > 0 { if c.md.connectTimeout > 0 {
@ -110,9 +110,9 @@ func (c *httpConnector) Connect(ctx context.Context, conn net.Conn, network, add
} }
defer resp.Body.Close() defer resp.Body.Close()
if c.logger.IsLevelEnabled(logger.DebugLevel) { if log.IsLevelEnabled(logger.DebugLevel) {
dump, _ := httputil.DumpResponse(resp, false) dump, _ := httputil.DumpResponse(resp, false)
c.logger.Debug(string(dump)) log.Debug(string(dump))
} }
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {

View File

@ -26,18 +26,18 @@ func init() {
type http2Connector struct { type http2Connector struct {
user *url.Userinfo user *url.Userinfo
md metadata md metadata
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &http2Connector{ return &http2Connector{
user: options.User, user: options.User,
logger: options.Logger, options: options,
} }
} }
@ -46,18 +46,18 @@ func (c *http2Connector) Init(md md.Metadata) (err error) {
} }
func (c *http2Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *http2Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
cc, ok := conn.(*http2_util.ClientConn) cc, ok := conn.(*http2_util.ClientConn)
if !ok { if !ok {
err := errors.New("wrong connection type") err := errors.New("wrong connection type")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
@ -83,9 +83,9 @@ func (c *http2Connector) Connect(ctx context.Context, conn net.Conn, network, ad
"Basic "+base64.StdEncoding.EncodeToString([]byte(u+":"+p))) "Basic "+base64.StdEncoding.EncodeToString([]byte(u+":"+p)))
} }
if c.logger.IsLevelEnabled(logger.DebugLevel) { if log.IsLevelEnabled(logger.DebugLevel) {
dump, _ := httputil.DumpRequest(req, false) dump, _ := httputil.DumpRequest(req, false)
c.logger.Debug(string(dump)) log.Debug(string(dump))
} }
if c.md.connectTimeout > 0 { if c.md.connectTimeout > 0 {
@ -95,19 +95,19 @@ func (c *http2Connector) Connect(ctx context.Context, conn net.Conn, network, ad
resp, err := cc.Client().Do(req.WithContext(ctx)) resp, err := cc.Client().Do(req.WithContext(ctx))
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
cc.Close() cc.Close()
return nil, err return nil, err
} }
if c.logger.IsLevelEnabled(logger.DebugLevel) { if log.IsLevelEnabled(logger.DebugLevel) {
dump, _ := httputil.DumpResponse(resp, false) dump, _ := httputil.DumpResponse(resp, false)
c.logger.Debug(string(dump)) log.Debug(string(dump))
} }
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
resp.Body.Close() resp.Body.Close()
err = fmt.Errorf("%s", resp.Status) err = fmt.Errorf("%s", resp.Status)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }

View File

@ -10,16 +10,17 @@ import (
"github.com/go-gost/gost/pkg/common/util/socks" "github.com/go-gost/gost/pkg/common/util/socks"
"github.com/go-gost/gost/pkg/common/util/udp" "github.com/go-gost/gost/pkg/common/util/udp"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
"github.com/go-gost/relay" "github.com/go-gost/relay"
) )
// Bind implements connector.Binder. // Bind implements connector.Binder.
func (c *relayConnector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) { func (c *relayConnector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("bind on %s/%s", address, network) log.Infof("bind on %s/%s", address, network)
options := connector.BindOptions{} options := connector.BindOptions{}
for _, opt := range opts { for _, opt := range opts {
@ -28,21 +29,22 @@ func (c *relayConnector) Bind(ctx context.Context, conn net.Conn, network, addre
switch network { switch network {
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
return c.bindTCP(ctx, conn, network, address) return c.bindTCP(ctx, conn, network, address, log)
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
return c.bindUDP(ctx, conn, network, address, &options) return c.bindUDP(ctx, conn, network, address, &options, log)
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
} }
func (c *relayConnector) bindTCP(ctx context.Context, conn net.Conn, network, address string) (net.Listener, error) { func (c *relayConnector) bindTCP(ctx context.Context, conn net.Conn, network, address string, log logger.Logger) (net.Listener, error) {
laddr, err := c.bind(conn, relay.BIND, network, address) laddr, err := c.bind(conn, relay.BIND, network, address)
if err != nil { if err != nil {
return nil, err return nil, err
} }
log.Debugf("bind on %s/%s OK", laddr, laddr.Network())
session, err := mux.ServerSession(conn) session, err := mux.ServerSession(conn)
if err != nil { if err != nil {
@ -52,15 +54,16 @@ func (c *relayConnector) bindTCP(ctx context.Context, conn net.Conn, network, ad
return &tcpListener{ return &tcpListener{
addr: laddr, addr: laddr,
session: session, session: session,
logger: c.logger, logger: log,
}, nil }, nil
} }
func (c *relayConnector) bindUDP(ctx context.Context, conn net.Conn, network, address string, opts *connector.BindOptions) (net.Listener, error) { func (c *relayConnector) bindUDP(ctx context.Context, conn net.Conn, network, address string, opts *connector.BindOptions, log logger.Logger) (net.Listener, error) {
laddr, err := c.bind(conn, relay.FUDP|relay.BIND, network, address) laddr, err := c.bind(conn, relay.FUDP|relay.BIND, network, address)
if err != nil { if err != nil {
return nil, err return nil, err
} }
log.Debugf("bind on %s/%s OK", laddr, laddr.Network())
ln := udp.NewListener( ln := udp.NewListener(
socks.UDPTunClientPacketConn(conn), socks.UDPTunClientPacketConn(conn),
@ -68,7 +71,7 @@ func (c *relayConnector) bindUDP(ctx context.Context, conn net.Conn, network, ad
opts.Backlog, opts.Backlog,
opts.UDPDataQueueSize, opts.UDPDataBufferSize, opts.UDPDataQueueSize, opts.UDPDataBufferSize,
opts.UDPConnTTL, opts.UDPConnTTL,
c.logger) log)
return ln, nil return ln, nil
} }
@ -125,7 +128,6 @@ func (c *relayConnector) bind(conn net.Conn, cmd uint8, network, address string)
if err != nil { if err != nil {
return nil, err return nil, err
} }
c.logger.Debugf("bind on %s/%s OK", baddr, baddr.Network())
return baddr, nil return baddr, nil
} }

View File

@ -9,7 +9,6 @@ import (
"github.com/go-gost/gost/pkg/common/util/socks" "github.com/go-gost/gost/pkg/common/util/socks"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
"github.com/go-gost/relay" "github.com/go-gost/relay"
@ -21,19 +20,19 @@ func init() {
type relayConnector struct { type relayConnector struct {
user *url.Userinfo user *url.Userinfo
logger logger.Logger
md metadata md metadata
options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &relayConnector{ return &relayConnector{
user: options.User, user: options.User,
logger: options.Logger, options: options,
} }
} }
@ -42,13 +41,13 @@ func (c *relayConnector) Init(md md.Metadata) (err error) {
} }
func (c *relayConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *relayConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
if c.md.connectTimeout > 0 { if c.md.connectTimeout > 0 {
conn.SetDeadline(time.Now().Add(c.md.connectTimeout)) conn.SetDeadline(time.Now().Add(c.md.connectTimeout))
@ -68,7 +67,7 @@ func (c *relayConnector) Connect(ctx context.Context, conn net.Conn, network, ad
if err != nil { if err != nil {
return nil, err return nil, err
} }
c.logger.Debugf("associate on %s OK", baddr) log.Debugf("associate on %s OK", baddr)
return socks.UDPTunClientConn(conn, nil), nil return socks.UDPTunClientConn(conn, nil), nil
} }
@ -123,7 +122,7 @@ func (c *relayConnector) Connect(ctx context.Context, conn net.Conn, network, ad
conn = cc conn = cc
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }

View File

@ -5,7 +5,6 @@ import (
"net" "net"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
) )
@ -16,17 +15,17 @@ func init() {
type sniConnector struct { type sniConnector struct {
md metadata md metadata
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &sniConnector{ return &sniConnector{
logger: options.Logger, options: options,
} }
} }
@ -35,13 +34,13 @@ func (c *sniConnector) Init(md md.Metadata) (err error) {
} }
func (c *sniConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *sniConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
return &sniClientConn{Conn: conn, host: c.md.host}, nil return &sniClientConn{Conn: conn, host: c.md.host}, nil
} }

View File

@ -11,7 +11,6 @@ import (
"github.com/go-gost/gosocks4" "github.com/go-gost/gosocks4"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
) )
@ -24,18 +23,18 @@ func init() {
type socks4Connector struct { type socks4Connector struct {
user *url.Userinfo user *url.Userinfo
md metadata md metadata
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &socks4Connector{ return &socks4Connector{
user: options.User, user: options.User,
logger: options.Logger, options: options,
} }
} }
@ -44,24 +43,24 @@ func (c *socks4Connector) Init(md md.Metadata) (err error) {
} }
func (c *socks4Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *socks4Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
switch network { switch network {
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
if _, ok := conn.(net.PacketConn); ok { if _, ok := conn.(net.PacketConn); ok {
err := fmt.Errorf("tcp over udp is unsupported") err := fmt.Errorf("tcp over udp is unsupported")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
@ -70,7 +69,7 @@ func (c *socks4Connector) Connect(ctx context.Context, conn net.Conn, network, a
if c.md.disable4a { if c.md.disable4a {
taddr, err := net.ResolveTCPAddr("tcp4", address) taddr, err := net.ResolveTCPAddr("tcp4", address)
if err != nil { if err != nil {
c.logger.Error("resolve: ", err) log.Error("resolve: ", err)
return nil, err return nil, err
} }
if len(taddr.IP) == 0 { if len(taddr.IP) == 0 {
@ -105,21 +104,21 @@ func (c *socks4Connector) Connect(ctx context.Context, conn net.Conn, network, a
} }
req := gosocks4.NewRequest(gosocks4.CmdConnect, addr, userid) req := gosocks4.NewRequest(gosocks4.CmdConnect, addr, userid)
if err := req.Write(conn); err != nil { if err := req.Write(conn); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(req) log.Debug(req)
reply, err := gosocks4.ReadReply(conn) reply, err := gosocks4.ReadReply(conn)
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(reply) log.Debug(reply)
if reply.Code != gosocks4.Granted { if reply.Code != gosocks4.Granted {
err = errors.New("host unreachable") err = errors.New("host unreachable")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }

View File

@ -10,15 +10,18 @@ import (
"github.com/go-gost/gost/pkg/common/util/socks" "github.com/go-gost/gost/pkg/common/util/socks"
"github.com/go-gost/gost/pkg/common/util/udp" "github.com/go-gost/gost/pkg/common/util/udp"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
) )
// Bind implements connector.Binder. // Bind implements connector.Binder.
func (c *socks5Connector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) { func (c *socks5Connector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("bind: %s/%s", address, network) log.Infof("bind: %s/%s", address, network)
options := connector.BindOptions{} options := connector.BindOptions{}
for _, opt := range opts { for _, opt := range opts {
@ -28,20 +31,20 @@ func (c *socks5Connector) Bind(ctx context.Context, conn net.Conn, network, addr
switch network { switch network {
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
if options.Mux { if options.Mux {
return c.muxBindTCP(ctx, conn, network, address) return c.muxBindTCP(ctx, conn, network, address, log)
} }
return c.bindTCP(ctx, conn, network, address) return c.bindTCP(ctx, conn, network, address, log)
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
return c.bindUDP(ctx, conn, network, address, &options) return c.bindUDP(ctx, conn, network, address, &options, log)
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
} }
func (c *socks5Connector) bindTCP(ctx context.Context, conn net.Conn, network, address string) (net.Listener, error) { func (c *socks5Connector) bindTCP(ctx context.Context, conn net.Conn, network, address string, log logger.Logger) (net.Listener, error) {
laddr, err := c.bind(conn, gosocks5.CmdBind, network, address) laddr, err := c.bind(conn, gosocks5.CmdBind, network, address, log)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -49,12 +52,12 @@ func (c *socks5Connector) bindTCP(ctx context.Context, conn net.Conn, network, a
return &tcpListener{ return &tcpListener{
addr: laddr, addr: laddr,
conn: conn, conn: conn,
logger: c.logger, logger: log,
}, nil }, nil
} }
func (c *socks5Connector) muxBindTCP(ctx context.Context, conn net.Conn, network, address string) (net.Listener, error) { func (c *socks5Connector) muxBindTCP(ctx context.Context, conn net.Conn, network, address string, log logger.Logger) (net.Listener, error) {
laddr, err := c.bind(conn, socks.CmdMuxBind, network, address) laddr, err := c.bind(conn, socks.CmdMuxBind, network, address, log)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -67,12 +70,12 @@ func (c *socks5Connector) muxBindTCP(ctx context.Context, conn net.Conn, network
return &tcpMuxListener{ return &tcpMuxListener{
addr: laddr, addr: laddr,
session: session, session: session,
logger: c.logger, logger: log,
}, nil }, nil
} }
func (c *socks5Connector) bindUDP(ctx context.Context, conn net.Conn, network, address string, opts *connector.BindOptions) (net.Listener, error) { func (c *socks5Connector) bindUDP(ctx context.Context, conn net.Conn, network, address string, opts *connector.BindOptions, log logger.Logger) (net.Listener, error) {
laddr, err := c.bind(conn, socks.CmdUDPTun, network, address) laddr, err := c.bind(conn, socks.CmdUDPTun, network, address, log)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -83,19 +86,19 @@ func (c *socks5Connector) bindUDP(ctx context.Context, conn net.Conn, network, a
opts.Backlog, opts.Backlog,
opts.UDPDataQueueSize, opts.UDPDataBufferSize, opts.UDPDataQueueSize, opts.UDPDataBufferSize,
opts.UDPConnTTL, opts.UDPConnTTL,
c.logger) log)
return ln, nil return ln, nil
} }
func (l *socks5Connector) bind(conn net.Conn, cmd uint8, network, address string) (net.Addr, error) { func (l *socks5Connector) bind(conn net.Conn, cmd uint8, network, address string, log logger.Logger) (net.Addr, error) {
addr := gosocks5.Addr{} addr := gosocks5.Addr{}
addr.ParseFrom(address) addr.ParseFrom(address)
req := gosocks5.NewRequest(cmd, &addr) req := gosocks5.NewRequest(cmd, &addr)
if err := req.Write(conn); err != nil { if err := req.Write(conn); err != nil {
return nil, err return nil, err
} }
l.logger.Debug(req) log.Debug(req)
// first reply, bind status // first reply, bind status
reply, err := gosocks5.ReadReply(conn) reply, err := gosocks5.ReadReply(conn)
@ -103,7 +106,7 @@ func (l *socks5Connector) bind(conn net.Conn, cmd uint8, network, address string
return nil, err return nil, err
} }
l.logger.Debug(reply) log.Debug(reply)
if reply.Rep != gosocks5.Succeeded { if reply.Rep != gosocks5.Succeeded {
return nil, fmt.Errorf("bind on %s/%s failed", address, network) return nil, fmt.Errorf("bind on %s/%s failed", address, network)
@ -121,7 +124,7 @@ func (l *socks5Connector) bind(conn net.Conn, cmd uint8, network, address string
if err != nil { if err != nil {
return nil, err return nil, err
} }
l.logger.Debugf("bind on %s/%s OK", baddr, baddr.Network()) log.Debugf("bind on %s/%s OK", baddr, baddr.Network())
return baddr, nil return baddr, nil
} }

View File

@ -23,7 +23,6 @@ func init() {
type socks5Connector struct { type socks5Connector struct {
selector gosocks5.Selector selector gosocks5.Selector
logger logger.Logger
md metadata md metadata
options connector.Options options connector.Options
} }
@ -44,16 +43,14 @@ func (c *socks5Connector) Init(md md.Metadata) (err error) {
return return
} }
c.logger = c.options.Logger
selector := &clientSelector{ selector := &clientSelector{
methods: []uint8{ methods: []uint8{
gosocks5.MethodNoAuth, gosocks5.MethodNoAuth,
gosocks5.MethodUserPass, gosocks5.MethodUserPass,
}, },
logger: c.logger,
User: c.options.User, User: c.options.User,
TLSConfig: c.options.TLSConfig, TLSConfig: c.options.TLSConfig,
logger: c.options.Logger,
} }
if !c.md.noTLS { if !c.md.noTLS {
selector.methods = append(selector.methods, socks.MethodTLS) selector.methods = append(selector.methods, socks.MethodTLS)
@ -70,7 +67,7 @@ func (c *socks5Connector) Init(md md.Metadata) (err error) {
// Handshake implements connector.Handshaker. // Handshake implements connector.Handshaker.
func (c *socks5Connector) Handshake(ctx context.Context, conn net.Conn) (net.Conn, error) { func (c *socks5Connector) Handshake(ctx context.Context, conn net.Conn) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
}) })
@ -82,7 +79,7 @@ func (c *socks5Connector) Handshake(ctx context.Context, conn net.Conn) (net.Con
cc := gosocks5.ClientConn(conn, c.selector) cc := gosocks5.ClientConn(conn, c.selector)
if err := cc.Handleshake(); err != nil { if err := cc.Handleshake(); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
@ -90,11 +87,13 @@ func (c *socks5Connector) Handshake(ctx context.Context, conn net.Conn) (net.Con
} }
func (c *socks5Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *socks5Connector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
if c.md.connectTimeout > 0 { if c.md.connectTimeout > 0 {
conn.SetDeadline(time.Now().Add(c.md.connectTimeout)) conn.SetDeadline(time.Now().Add(c.md.connectTimeout))
@ -103,68 +102,68 @@ func (c *socks5Connector) Connect(ctx context.Context, conn net.Conn, network, a
switch network { switch network {
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
return c.connectUDP(ctx, conn, network, address) return c.connectUDP(ctx, conn, network, address, log)
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
if _, ok := conn.(net.PacketConn); ok { if _, ok := conn.(net.PacketConn); ok {
err := fmt.Errorf("tcp over udp is unsupported") err := fmt.Errorf("tcp over udp is unsupported")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
addr := gosocks5.Addr{} addr := gosocks5.Addr{}
if err := addr.ParseFrom(address); err != nil { if err := addr.ParseFrom(address); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
req := gosocks5.NewRequest(gosocks5.CmdConnect, &addr) req := gosocks5.NewRequest(gosocks5.CmdConnect, &addr)
if err := req.Write(conn); err != nil { if err := req.Write(conn); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(req) log.Debug(req)
reply, err := gosocks5.ReadReply(conn) reply, err := gosocks5.ReadReply(conn)
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(reply) log.Debug(reply)
if reply.Rep != gosocks5.Succeeded { if reply.Rep != gosocks5.Succeeded {
err = errors.New("host unreachable") err = errors.New("host unreachable")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
return conn, nil return conn, nil
} }
func (c *socks5Connector) connectUDP(ctx context.Context, conn net.Conn, network, address string) (net.Conn, error) { func (c *socks5Connector) connectUDP(ctx context.Context, conn net.Conn, network, address string, log logger.Logger) (net.Conn, error) {
addr, err := net.ResolveUDPAddr(network, address) addr, err := net.ResolveUDPAddr(network, address)
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
req := gosocks5.NewRequest(socks.CmdUDPTun, nil) req := gosocks5.NewRequest(socks.CmdUDPTun, nil)
if err := req.Write(conn); err != nil { if err := req.Write(conn); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(req) log.Debug(req)
reply, err := gosocks5.ReadReply(conn) reply, err := gosocks5.ReadReply(conn)
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
c.logger.Debug(reply) log.Debug(reply)
if reply.Rep != gosocks5.Succeeded { if reply.Rep != gosocks5.Succeeded {
return nil, errors.New("get socks5 UDP tunnel failure") return nil, errors.New("get socks5 UDP tunnel failure")

View File

@ -10,7 +10,6 @@ import (
"github.com/go-gost/gost/pkg/common/bufpool" "github.com/go-gost/gost/pkg/common/bufpool"
"github.com/go-gost/gost/pkg/common/util/ss" "github.com/go-gost/gost/pkg/common/util/ss"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
"github.com/shadowsocks/go-shadowsocks2/core" "github.com/shadowsocks/go-shadowsocks2/core"
@ -23,7 +22,6 @@ func init() {
type ssConnector struct { type ssConnector struct {
cipher core.Cipher cipher core.Cipher
md metadata md metadata
logger logger.Logger
options connector.Options options connector.Options
} }
@ -35,7 +33,6 @@ func NewConnector(opts ...connector.Option) connector.Connector {
return &ssConnector{ return &ssConnector{
options: options, options: options,
logger: options.Logger,
} }
} }
@ -54,30 +51,30 @@ func (c *ssConnector) Init(md md.Metadata) (err error) {
} }
func (c *ssConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *ssConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
switch network { switch network {
case "tcp", "tcp4", "tcp6": case "tcp", "tcp4", "tcp6":
if _, ok := conn.(net.PacketConn); ok { if _, ok := conn.(net.PacketConn); ok {
err := fmt.Errorf("tcp over udp is unsupported") err := fmt.Errorf("tcp over udp is unsupported")
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
addr := gosocks5.Addr{} addr := gosocks5.Addr{}
if err := addr.ParseFrom(address); err != nil { if err := addr.ParseFrom(address); err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
rawaddr := bufpool.Get(512) rawaddr := bufpool.Get(512)
@ -85,7 +82,7 @@ func (c *ssConnector) Connect(ctx context.Context, conn net.Conn, network, addre
n, err := addr.Encode(*rawaddr) n, err := addr.Encode(*rawaddr)
if err != nil { if err != nil {
c.logger.Error("encoding addr: ", err) log.Error("encoding addr: ", err)
return nil, err return nil, err
} }

View File

@ -9,7 +9,6 @@ import (
"github.com/go-gost/gost/pkg/common/util/socks" "github.com/go-gost/gost/pkg/common/util/socks"
"github.com/go-gost/gost/pkg/common/util/ss" "github.com/go-gost/gost/pkg/common/util/ss"
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
"github.com/shadowsocks/go-shadowsocks2/core" "github.com/shadowsocks/go-shadowsocks2/core"
@ -22,7 +21,6 @@ func init() {
type ssuConnector struct { type ssuConnector struct {
cipher core.Cipher cipher core.Cipher
md metadata md metadata
logger logger.Logger
options connector.Options options connector.Options
} }
@ -34,7 +32,6 @@ func NewConnector(opts ...connector.Option) connector.Connector {
return &ssuConnector{ return &ssuConnector{
options: options, options: options,
logger: options.Logger,
} }
} }
@ -53,19 +50,19 @@ func (c *ssuConnector) Init(md md.Metadata) (err error) {
} }
func (c *ssuConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *ssuConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
switch network { switch network {
case "udp", "udp4", "udp6": case "udp", "udp4", "udp6":
default: default:
err := fmt.Errorf("network %s is unsupported", network) err := fmt.Errorf("network %s is unsupported", network)
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }

View File

@ -1,4 +1,4 @@
package ssh package sshd
import ( import (
"context" "context"
@ -7,7 +7,6 @@ import (
"github.com/go-gost/gost/pkg/connector" "github.com/go-gost/gost/pkg/connector"
ssh_util "github.com/go-gost/gost/pkg/internal/util/ssh" ssh_util "github.com/go-gost/gost/pkg/internal/util/ssh"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
) )
@ -16,33 +15,33 @@ func init() {
registry.RegiserConnector("sshd", NewConnector) registry.RegiserConnector("sshd", NewConnector)
} }
type forwardConnector struct { type sshdConnector struct {
logger logger.Logger options connector.Options
} }
func NewConnector(opts ...connector.Option) connector.Connector { func NewConnector(opts ...connector.Option) connector.Connector {
options := &connector.Options{} options := connector.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &forwardConnector{ return &sshdConnector{
logger: options.Logger, options: options,
} }
} }
func (c *forwardConnector) Init(md md.Metadata) (err error) { func (c *sshdConnector) Init(md md.Metadata) (err error) {
return nil return nil
} }
func (c *forwardConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) { func (c *sshdConnector) Connect(ctx context.Context, conn net.Conn, network, address string, opts ...connector.ConnectOption) (net.Conn, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(), "remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(), "local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("connect %s/%s", address, network) log.Infof("connect %s/%s", address, network)
cc, ok := conn.(*ssh_util.ClientConn) cc, ok := conn.(*ssh_util.ClientConn)
if !ok { if !ok {
@ -51,7 +50,7 @@ func (c *forwardConnector) Connect(ctx context.Context, conn net.Conn, network,
conn, err := cc.Client().Dial(network, address) conn, err := cc.Client().Dial(network, address)
if err != nil { if err != nil {
c.logger.Error(err) log.Error(err)
return nil, err return nil, err
} }
@ -59,12 +58,14 @@ func (c *forwardConnector) Connect(ctx context.Context, conn net.Conn, network,
} }
// Bind implements connector.Binder. // Bind implements connector.Binder.
func (c *forwardConnector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) { func (c *sshdConnector) Bind(ctx context.Context, conn net.Conn, network, address string, opts ...connector.BindOption) (net.Listener, error) {
c.logger = c.logger.WithFields(map[string]interface{}{ log := c.options.Logger.WithFields(map[string]interface{}{
"remote": conn.RemoteAddr().String(),
"local": conn.LocalAddr().String(),
"network": network, "network": network,
"address": address, "address": address,
}) })
c.logger.Infof("bind on %s/%s", address, network) log.Infof("bind on %s/%s", address, network)
cc, ok := conn.(*ssh_util.ClientConn) cc, ok := conn.(*ssh_util.ClientConn)
if !ok { if !ok {

View File

@ -1,4 +1,4 @@
package ssh package sshd
import ( import (
"net" "net"

View File

@ -1,4 +1,4 @@
package ssh package sshd
import ( import (
"context" "context"
@ -10,7 +10,6 @@ import (
"github.com/go-gost/gost/pkg/dialer" "github.com/go-gost/gost/pkg/dialer"
ssh_util "github.com/go-gost/gost/pkg/internal/util/ssh" ssh_util "github.com/go-gost/gost/pkg/internal/util/ssh"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata" md "github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry" "github.com/go-gost/gost/pkg/registry"
"golang.org/x/crypto/ssh" "golang.org/x/crypto/ssh"
@ -20,28 +19,28 @@ func init() {
registry.RegisterDialer("sshd", NewDialer) registry.RegisterDialer("sshd", NewDialer)
} }
type forwardDialer struct { type sshdDialer struct {
user *url.Userinfo user *url.Userinfo
sessions map[string]*sshSession sessions map[string]*sshSession
sessionMutex sync.Mutex sessionMutex sync.Mutex
logger logger.Logger
md metadata md metadata
options dialer.Options
} }
func NewDialer(opts ...dialer.Option) dialer.Dialer { func NewDialer(opts ...dialer.Option) dialer.Dialer {
options := &dialer.Options{} options := dialer.Options{}
for _, opt := range opts { for _, opt := range opts {
opt(options) opt(&options)
} }
return &forwardDialer{ return &sshdDialer{
user: options.User, user: options.User,
sessions: make(map[string]*sshSession), sessions: make(map[string]*sshSession),
logger: options.Logger, options: options,
} }
} }
func (d *forwardDialer) Init(md md.Metadata) (err error) { func (d *sshdDialer) Init(md md.Metadata) (err error) {
if err = d.parseMetadata(md); err != nil { if err = d.parseMetadata(md); err != nil {
return return
} }
@ -50,11 +49,11 @@ func (d *forwardDialer) Init(md md.Metadata) (err error) {
} }
// Multiplex implements dialer.Multiplexer interface. // Multiplex implements dialer.Multiplexer interface.
func (d *forwardDialer) Multiplex() bool { func (d *sshdDialer) Multiplex() bool {
return true return true
} }
func (d *forwardDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialOption) (conn net.Conn, err error) { func (d *sshdDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialOption) (conn net.Conn, err error) {
var options dialer.DialOptions var options dialer.DialOptions
for _, opt := range opts { for _, opt := range opts {
opt(&options) opt(&options)
@ -85,7 +84,7 @@ func (d *forwardDialer) Dial(ctx context.Context, addr string, opts ...dialer.Di
} }
// Handshake implements dialer.Handshaker // Handshake implements dialer.Handshaker
func (d *forwardDialer) Handshake(ctx context.Context, conn net.Conn, options ...dialer.HandshakeOption) (net.Conn, error) { func (d *sshdDialer) Handshake(ctx context.Context, conn net.Conn, options ...dialer.HandshakeOption) (net.Conn, error) {
opts := &dialer.HandshakeOptions{} opts := &dialer.HandshakeOptions{}
for _, option := range options { for _, option := range options {
option(opts) option(opts)
@ -99,10 +98,12 @@ func (d *forwardDialer) Handshake(ctx context.Context, conn net.Conn, options ..
defer conn.SetDeadline(time.Time{}) defer conn.SetDeadline(time.Time{})
} }
log := d.options.Logger
session, ok := d.sessions[opts.Addr] session, ok := d.sessions[opts.Addr]
if session != nil && session.conn != conn { if session != nil && session.conn != conn {
err := errors.New("ssh: unrecognized connection") err := errors.New("ssh: unrecognized connection")
d.logger.Error(err) log.Error(err)
conn.Close() conn.Close()
delete(d.sessions, opts.Addr) delete(d.sessions, opts.Addr)
return nil, err return nil, err
@ -111,7 +112,7 @@ func (d *forwardDialer) Handshake(ctx context.Context, conn net.Conn, options ..
if !ok || session.client == nil { if !ok || session.client == nil {
s, err := d.initSession(ctx, opts.Addr, conn) s, err := d.initSession(ctx, opts.Addr, conn)
if err != nil { if err != nil {
d.logger.Error(err) log.Error(err)
conn.Close() conn.Close()
delete(d.sessions, opts.Addr) delete(d.sessions, opts.Addr)
return nil, err return nil, err
@ -119,7 +120,7 @@ func (d *forwardDialer) Handshake(ctx context.Context, conn net.Conn, options ..
session = s session = s
go func() { go func() {
s.wait() s.wait()
d.logger.Debug("session closed") log.Debug("session closed")
}() }()
d.sessions[opts.Addr] = session d.sessions[opts.Addr] = session
} }
@ -131,14 +132,16 @@ func (d *forwardDialer) Handshake(ctx context.Context, conn net.Conn, options ..
return ssh_util.NewClientConn(session.conn, session.client), nil return ssh_util.NewClientConn(session.conn, session.client), nil
} }
func (d *forwardDialer) dial(ctx context.Context, network, addr string, opts *dialer.DialOptions) (net.Conn, error) { func (d *sshdDialer) dial(ctx context.Context, network, addr string, opts *dialer.DialOptions) (net.Conn, error) {
log := d.options.Logger
dial := opts.DialFunc dial := opts.DialFunc
if dial != nil { if dial != nil {
conn, err := dial(ctx, addr) conn, err := dial(ctx, addr)
if err != nil { if err != nil {
d.logger.Error(err) log.Error(err)
} else { } else {
d.logger.WithFields(map[string]interface{}{ log.WithFields(map[string]interface{}{
"src": conn.LocalAddr().String(), "src": conn.LocalAddr().String(),
"dst": addr, "dst": addr,
}).Debug("dial with dial func") }).Debug("dial with dial func")
@ -149,9 +152,9 @@ func (d *forwardDialer) dial(ctx context.Context, network, addr string, opts *di
var netd net.Dialer var netd net.Dialer
conn, err := netd.DialContext(ctx, network, addr) conn, err := netd.DialContext(ctx, network, addr)
if err != nil { if err != nil {
d.logger.Error(err) log.Error(err)
} else { } else {
d.logger.WithFields(map[string]interface{}{ log.WithFields(map[string]interface{}{
"src": conn.LocalAddr().String(), "src": conn.LocalAddr().String(),
"dst": addr, "dst": addr,
}).Debugf("dial direct %s/%s", addr, network) }).Debugf("dial direct %s/%s", addr, network)
@ -159,7 +162,7 @@ func (d *forwardDialer) dial(ctx context.Context, network, addr string, opts *di
return conn, err return conn, err
} }
func (d *forwardDialer) initSession(ctx context.Context, addr string, conn net.Conn) (*sshSession, error) { func (d *sshdDialer) initSession(ctx context.Context, addr string, conn net.Conn) (*sshSession, error) {
config := ssh.ClientConfig{ config := ssh.ClientConfig{
// Timeout: timeout, // Timeout: timeout,
HostKeyCallback: ssh.InsecureIgnoreHostKey(), HostKeyCallback: ssh.InsecureIgnoreHostKey(),

View File

@ -1,4 +1,4 @@
package ssh package sshd
import ( import (
"io/ioutil" "io/ioutil"
@ -13,7 +13,7 @@ type metadata struct {
signer ssh.Signer signer ssh.Signer
} }
func (d *forwardDialer) parseMetadata(md mdata.Metadata) (err error) { func (d *sshdDialer) parseMetadata(md mdata.Metadata) (err error) {
const ( const (
handshakeTimeout = "handshakeTimeout" handshakeTimeout = "handshakeTimeout"
privateKeyFile = "privateKeyFile" privateKeyFile = "privateKeyFile"