go-gost/gost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improve config

Browse Source
This commit is contained in:
ginuerzh
2022-01-26 18:29:48 +08:00
parent 713586729b
commit 44acdc9215
14 changed files with 160 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
cmd/gost/cmd.go
View File

@ -52,7 +52,26 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
} }
nodeConfig.Name = "node-0" nodeConfig.Name = "node-0"
var nodes []*config.NodeConfig
for _, host := range strings.Split(nodeConfig.Addr, ",") {
if host == "" {
continue
}
nodeCfg := &config.NodeConfig{}
*nodeCfg = *nodeConfig
nodeCfg.Name = fmt.Sprintf("node-%d", len(nodes))
nodeCfg.Addr = host
nodes = append(nodes, nodeCfg)
}
md := metadata.MapMetadata(nodeConfig.Connector.Metadata) md := metadata.MapMetadata(nodeConfig.Connector.Metadata)
hopConfig := &config.HopConfig{
Name: fmt.Sprintf("hop-%d", i),
Selector: parseSelector(md),
Nodes: nodes,
}
if v := metadata.GetString(md, "bypass"); v != "" { if v := metadata.GetString(md, "bypass"); v != "" {
bypassCfg := &config.BypassConfig{ bypassCfg := &config.BypassConfig{
Name: fmt.Sprintf("bypass-%d", len(cfg.Bypasses)), Name: fmt.Sprintf("bypass-%d", len(cfg.Bypasses)),
@ -67,28 +86,52 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
} }
bypassCfg.Matchers = append(bypassCfg.Matchers, s) bypassCfg.Matchers = append(bypassCfg.Matchers, s)
} }
nodeConfig.Bypass = bypassCfg.Name hopConfig.Bypass = bypassCfg.Name
cfg.Bypasses = append(cfg.Bypasses, bypassCfg) cfg.Bypasses = append(cfg.Bypasses, bypassCfg)
md.Del("bypass") md.Del("bypass")
} }
if v := metadata.GetString(md, "resolver"); v != "" {
var nodes []*config.NodeConfig resolverCfg := &config.ResolverConfig{
for _, host := range strings.Split(nodeConfig.Addr, ",") { Name: fmt.Sprintf("resolver-%d", len(cfg.Resolvers)),
if host == "" {
continue
} }
nodeCfg := &config.NodeConfig{} for _, rs := range strings.Split(v, ",") {
*nodeCfg = *nodeConfig if rs == "" {
nodeCfg.Name = fmt.Sprintf("node-%d", len(nodes)) continue
nodeCfg.Addr = host }
nodes = append(nodes, nodeCfg) resolverCfg.Nameservers = append(
resolverCfg.Nameservers,
config.NameserverConfig{
Addr: rs,
},
)
}
hopConfig.Resolver = resolverCfg.Name
cfg.Resolvers = append(cfg.Resolvers, resolverCfg)
md.Del("resolver")
}
if v := metadata.GetString(md, "hosts"); v != "" {
hostsCfg := &config.HostsConfig{
Name: fmt.Sprintf("hosts-%d", len(cfg.Hosts)),
}
for _, s := range strings.Split(v, ",") {
ss := strings.SplitN(s, ":", 2)
if len(ss) != 2 {
continue
}
hostsCfg.Mappings = append(
hostsCfg.Mappings,
config.HostMappingConfig{
Hostname: ss[0],
IP: ss[1],
},
)
}
hopConfig.Hosts = hostsCfg.Name
cfg.Hosts = append(cfg.Hosts, hostsCfg)
md.Del("hosts")
} }
chain.Hops = append(chain.Hops, &config.HopConfig{ chain.Hops = append(chain.Hops, hopConfig)
Name: fmt.Sprintf("hop-%d", i),
Selector: parseSelector(md),
Nodes: nodes,
})
} }
for i, svc := range services { for i, svc := range services {
@ -126,7 +169,7 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
} }
bypassCfg.Matchers = append(bypassCfg.Matchers, s) bypassCfg.Matchers = append(bypassCfg.Matchers, s)
} }
service.Handler.Bypass = bypassCfg.Name service.Bypass = bypassCfg.Name
cfg.Bypasses = append(cfg.Bypasses, bypassCfg) cfg.Bypasses = append(cfg.Bypasses, bypassCfg)
md.Del("bypass") md.Del("bypass")
} }
@ -145,7 +188,7 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
}, },
) )
} }
service.Handler.Resolver = resolverCfg.Name service.Resolver = resolverCfg.Name
cfg.Resolvers = append(cfg.Resolvers, resolverCfg) cfg.Resolvers = append(cfg.Resolvers, resolverCfg)
md.Del("resolver") md.Del("resolver")
} }
@ -166,11 +209,10 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
}, },
) )
} }
service.Handler.Hosts = hostsCfg.Name service.Hosts = hostsCfg.Name
cfg.Hosts = append(cfg.Hosts, hostsCfg) cfg.Hosts = append(cfg.Hosts, hostsCfg)
md.Del("hosts") md.Del("hosts")
} }
} }
return cfg, nil return cfg, nil

36
cmd/gost/config.go
View File

@ -119,12 +119,12 @@ func buildService(cfg *config.Config) (services []*service.Service) {
} }
h := registry.GetHandler(svc.Handler.Type)( h := registry.GetHandler(svc.Handler.Type)(
handler.AuthsOption(authsFromConfig(svc.Handler.Auths...)...),
handler.RetriesOption(svc.Handler.Retries), handler.RetriesOption(svc.Handler.Retries),
handler.ChainOption(chains[svc.Handler.Chain]), handler.ChainOption(chains[svc.Handler.Chain]),
handler.ResolverOption(resolvers[svc.Handler.Resolver]), handler.BypassOption(bypasses[svc.Bypass]),
handler.HostsOption(hosts[svc.Handler.Hosts]), handler.ResolverOption(resolvers[svc.Resolver]),
handler.BypassOption(bypasses[svc.Handler.Bypass]), handler.HostsOption(hosts[svc.Hosts]),
handler.AuthsOption(authsFromConfig(svc.Handler.Auths...)...),
handler.TLSConfigOption(tlsConfig), handler.TLSConfigOption(tlsConfig),
handler.LoggerOption(handlerLogger), handler.LoggerOption(handlerLogger),
) )
@ -252,9 +252,25 @@ func chainFromConfig(cfg *config.ChainConfig) *chain.Chain {
WithDialer(d). WithDialer(d).
WithAddr(v.Addr) WithAddr(v.Addr)
node := chain.NewNode(v.Name, v.Addr). if v.Bypass == "" {
WithTransport(tr). v.Bypass = hop.Bypass
WithBypass(bypasses[v.Bypass]) }
if v.Resolver == "" {
v.Resolver = hop.Resolver
}
if v.Hosts == "" {
v.Hosts = hop.Hosts
}
node := &chain.Node{
Name: v.Name,
Addr: v.Addr,
Transport: tr,
Bypass: bypasses[v.Bypass],
Resolver: resolvers[v.Resolver],
Hosts: hosts[v.Hosts],
Marker: &chain.FailMarker{},
}
group.AddNode(node) group.AddNode(node)
} }
@ -277,7 +293,11 @@ func forwarderFromConfig(cfg *config.ForwarderConfig) *chain.NodeGroup {
group := &chain.NodeGroup{} group := &chain.NodeGroup{}
for _, target := range cfg.Targets { for _, target := range cfg.Targets {
if v := strings.TrimSpace(target); v != "" { if v := strings.TrimSpace(target); v != "" {
group.AddNode(chain.NewNode(target, target)) group.AddNode(&chain.Node{
Name: target,
Addr: target,
Marker: &chain.FailMarker{},
})
} }
} }
return group.WithSelector(selectorFromConfig(cfg.Selector)) return group.WithSelector(selectorFromConfig(cfg.Selector))

6
gost.yml
View File

@ -6,10 +6,10 @@ log:
services: services:
- name: http+tcp - name: http+tcp
addr: ":28000" addr: ":28000"
# bypass: bypass01
handler: handler:
type: http type: http
chain: chain01 chain: chain01
# bypass: bypass01
metadata: metadata:
proxyAgent: "gost/3.0" proxyAgent: "gost/3.0"
auths: auths:
@ -23,10 +23,10 @@ services:
keepAlive: 15s keepAlive: 15s
- name: ss - name: ss
addr: ":28338" addr: ":28338"
# bypass: bypass01
handler: handler:
type: ss type: ss
# chain: chain01 # chain: chain01
# bypass: bypass01
metadata: metadata:
method: chacha20-ietf method: chacha20-ietf
password: gost password: gost
@ -39,10 +39,10 @@ services:
keepAlive: 15s keepAlive: 15s
- name: socks5 - name: socks5
addr: ":21080" addr: ":21080"
# bypass: bypass01
handler: handler:
type: socks5 type: socks5
# chain: chain-ss # chain: chain-ss
# bypass: bypass01
metadata: metadata:
auths: auths:
- gost:gost - gost:gost

10
pkg/chain/chain.go
View File

@ -23,15 +23,15 @@ func (c *Chain) GetRouteFor(network, address string) (r *route) {
if node == nil { if node == nil {
return return
} }
if node.bypass != nil && node.bypass.Contains(address) { if node.Bypass != nil && node.Bypass.Contains(address) {
break break
} }
if node.transport.Multiplex() { if node.Transport.Multiplex() {
tr := node.transport.Copy(). tr := node.Transport.Copy().
WithRoute(r) WithRoute(r)
node = node.Copy(). node = node.Copy()
WithTransport(tr) node.Transport = tr
r = &route{} r = &route{}
} }

44
pkg/chain/node.go
View File

@ -5,44 +5,18 @@ import (
"time" "time"
"github.com/go-gost/gost/pkg/bypass" "github.com/go-gost/gost/pkg/bypass"
"github.com/go-gost/gost/pkg/hosts"
"github.com/go-gost/gost/pkg/resolver"
) )
type Node struct { type Node struct {
name string Name string
addr string Addr string
transport *Transport Transport *Transport
bypass bypass.Bypass Bypass bypass.Bypass
marker *FailMarker Resolver resolver.Resolver
} Hosts hosts.HostMapper
Marker *FailMarker
func NewNode(name, addr string) *Node {
return &Node{
name: name,
addr: addr,
marker: &FailMarker{},
}
}
func (node *Node) Name() string {
return node.name
}
func (node *Node) Addr() string {
return node.addr
}
func (node *Node) Marker() *FailMarker {
return node.marker
}
func (node *Node) WithTransport(tr *Transport) *Node {
node.transport = tr
return node
}
func (node *Node) WithBypass(bp bypass.Bypass) *Node {
node.bypass = bp
return node
} }
func (node *Node) Copy() *Node { func (node *Node) Copy() *Node {

28
pkg/chain/route.go
View File

@ -29,35 +29,35 @@ func (r *route) connect(ctx context.Context) (conn net.Conn, err error) {
} }
node := r.nodes[0] node := r.nodes[0]
cc, err := node.transport.Dial(ctx, r.nodes[0].Addr()) cc, err := node.Transport.Dial(ctx, r.nodes[0].Addr)
if err != nil { if err != nil {
node.Marker().Mark() node.Marker.Mark()
return return
} }
cn, err := node.transport.Handshake(ctx, cc) cn, err := node.Transport.Handshake(ctx, cc)
if err != nil { if err != nil {
cc.Close() cc.Close()
node.Marker().Mark() node.Marker.Mark()
return return
} }
node.Marker().Reset() node.Marker.Reset()
preNode := node preNode := node
for _, node := range r.nodes[1:] { for _, node := range r.nodes[1:] {
cc, err = preNode.transport.Connect(ctx, cn, "tcp", node.Addr()) cc, err = preNode.Transport.Connect(ctx, cn, "tcp", node.Addr)
if err != nil { if err != nil {
cn.Close() cn.Close()
node.Marker().Mark() node.Marker.Mark()
return return
} }
cc, err = node.transport.Handshake(ctx, cc) cc, err = node.Transport.Handshake(ctx, cc)
if err != nil { if err != nil {
cn.Close() cn.Close()
node.Marker().Mark() node.Marker.Mark()
return return
} }
node.Marker().Reset() node.Marker.Reset()
cn = cc cn = cc
preNode = node preNode = node
@ -77,7 +77,7 @@ func (r *route) Dial(ctx context.Context, network, address string) (net.Conn, er
return nil, err return nil, err
} }
cc, err := r.Last().transport.Connect(ctx, conn, network, address) cc, err := r.Last().Transport.Connect(ctx, conn, network, address)
if err != nil { if err != nil {
conn.Close() conn.Close()
return nil, err return nil, err
@ -108,7 +108,7 @@ func (r *route) Bind(ctx context.Context, network, address string, opts ...conne
return nil, err return nil, err
} }
ln, err := r.Last().transport.Bind(ctx, conn, network, address, opts...) ln, err := r.Last().Transport.Bind(ctx, conn, network, address, opts...)
if err != nil { if err != nil {
conn.Close() conn.Close()
return nil, err return nil, err
@ -134,8 +134,8 @@ func (r *route) Path() (path []*Node) {
} }
for _, node := range r.nodes { for _, node := range r.nodes {
if node.transport != nil && node.transport.route != nil { if node.Transport != nil && node.Transport.route != nil {
path = append(path, node.transport.route.Path()...) path = append(path, node.Transport.route.Path()...)
} }
path = append(path, node) path = append(path, node)
} }

4
pkg/chain/router.go
View File

@ -46,7 +46,7 @@ func (r *Router) dial(ctx context.Context, network, address string) (conn net.Co
if r.Logger.IsLevelEnabled(logger.DebugLevel) { if r.Logger.IsLevelEnabled(logger.DebugLevel) {
buf := bytes.Buffer{} buf := bytes.Buffer{}
for _, node := range route.Path() { for _, node := range route.Path() {
fmt.Fprintf(&buf, "%s@%s > ", node.Name(), node.Addr()) fmt.Fprintf(&buf, "%s@%s > ", node.Name, node.Addr)
} }
fmt.Fprintf(&buf, "%s", address) fmt.Fprintf(&buf, "%s", address)
r.Logger.Debugf("route(retry=%d) %s", i, buf.String()) r.Logger.Debugf("route(retry=%d) %s", i, buf.String())
@ -114,7 +114,7 @@ func (r *Router) Bind(ctx context.Context, network, address string, opts ...conn
if r.Logger.IsLevelEnabled(logger.DebugLevel) { if r.Logger.IsLevelEnabled(logger.DebugLevel) {
buf := bytes.Buffer{} buf := bytes.Buffer{}
for _, node := range route.Path() { for _, node := range route.Path() {
fmt.Fprintf(&buf, "%s@%s > ", node.Name(), node.Addr()) fmt.Fprintf(&buf, "%s@%s > ", node.Name, node.Addr)
} }
fmt.Fprintf(&buf, "%s", address) fmt.Fprintf(&buf, "%s", address)
r.Logger.Debugf("route(retry=%d) %s", i, buf.String()) r.Logger.Debugf("route(retry=%d) %s", i, buf.String())

6
pkg/chain/selector.go
View File

@ -141,8 +141,8 @@ func (f *failFilter) Filter(nodes ...*Node) []*Node {
} }
var nl []*Node var nl []*Node
for _, node := range nodes { for _, node := range nodes {
if node.Marker().FailCount() < int64(maxFails) || if node.Marker.FailCount() < int64(maxFails) ||
time.Since(time.Unix(node.Marker().FailTime(), 0)) >= failTimeout { time.Since(time.Unix(node.Marker.FailTime(), 0)) >= failTimeout {
nl = append(nl, node) nl = append(nl, node)
} }
} }
@ -161,7 +161,7 @@ func InvalidFilter() Filter {
func (f *invalidFilter) Filter(nodes ...*Node) []*Node { func (f *invalidFilter) Filter(nodes ...*Node) []*Node {
var nl []*Node var nl []*Node
for _, node := range nodes { for _, node := range nodes {
_, sport, _ := net.SplitHostPort(node.Addr()) _, sport, _ := net.SplitHostPort(node.Addr)
if port, _ := strconv.Atoi(sport); port > 0 { if port, _ := strconv.Atoi(sport); port > 0 {
nl = append(nl, node) nl = append(nl, node)
} }

11
pkg/config/config.go
View File

@ -93,9 +93,6 @@ type HandlerConfig struct {
Type string Type string
Retries int `yaml:",omitempty"` Retries int `yaml:",omitempty"`
Chain string `yaml:",omitempty"` Chain string `yaml:",omitempty"`
Bypass string `yaml:",omitempty"`
Resolver string `yaml:",omitempty"`
Hosts string `yaml:",omitempty"`
Auths []*AuthConfig `yaml:",omitempty"` Auths []*AuthConfig `yaml:",omitempty"`
TLS *TLSConfig `yaml:",omitempty"` TLS *TLSConfig `yaml:",omitempty"`
Metadata map[string]interface{} `yaml:",omitempty"` Metadata map[string]interface{} `yaml:",omitempty"`
@ -123,6 +120,9 @@ type ConnectorConfig struct {
type ServiceConfig struct { type ServiceConfig struct {
Name string Name string
Addr string `yaml:",omitempty"` Addr string `yaml:",omitempty"`
Bypass string `yaml:",omitempty"`
Resolver string `yaml:",omitempty"`
Hosts string `yaml:",omitempty"`
Handler *HandlerConfig `yaml:",omitempty"` Handler *HandlerConfig `yaml:",omitempty"`
Listener *ListenerConfig `yaml:",omitempty"` Listener *ListenerConfig `yaml:",omitempty"`
Forwarder *ForwarderConfig `yaml:",omitempty"` Forwarder *ForwarderConfig `yaml:",omitempty"`
@ -137,6 +137,9 @@ type ChainConfig struct {
type HopConfig struct { type HopConfig struct {
Name string Name string
Selector *SelectorConfig `yaml:",omitempty"` Selector *SelectorConfig `yaml:",omitempty"`
Bypass string `yaml:",omitempty"`
Resolver string `yaml:",omitempty"`
Hosts string `yaml:",omitempty"`
Nodes []*NodeConfig Nodes []*NodeConfig
} }
@ -144,6 +147,8 @@ type NodeConfig struct {
Name string Name string
Addr string `yaml:",omitempty"` Addr string `yaml:",omitempty"`
Bypass string `yaml:",omitempty"` Bypass string `yaml:",omitempty"`
Resolver string `yaml:",omitempty"`
Hosts string `yaml:",omitempty"`
Connector *ConnectorConfig `yaml:",omitempty"` Connector *ConnectorConfig `yaml:",omitempty"`
Dialer *DialerConfig `yaml:",omitempty"` Dialer *DialerConfig `yaml:",omitempty"`
} }

16
pkg/handler/forward/local/handler.go
View File

@ -43,7 +43,7 @@ func (h *forwardHandler) Init(md md.Metadata) (err error) {
if h.group == nil { if h.group == nil {
// dummy node used by relay connector. // dummy node used by relay connector.
h.group = chain.NewNodeGroup(chain.NewNode("dummy", ":0")) h.group = chain.NewNodeGroup(&chain.Node{Name: "dummy", Addr: ":0"})
} }
h.router = &chain.Router{ h.router = &chain.Router{
@ -90,26 +90,26 @@ func (h *forwardHandler) Handle(ctx context.Context, conn net.Conn) {
} }
log = log.WithFields(map[string]interface{}{ log = log.WithFields(map[string]interface{}{
"dst": fmt.Sprintf("%s/%s", target.Addr(), network), "dst": fmt.Sprintf("%s/%s", target.Addr, network),
}) })
log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr)
cc, err := h.router.Dial(ctx, network, target.Addr()) cc, err := h.router.Dial(ctx, network, target.Addr)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
// TODO: the router itself may be failed due to the failed node in the router, // TODO: the router itself may be failed due to the failed node in the router,
// the dead marker may be a wrong operation. // the dead marker may be a wrong operation.
target.Marker().Mark() target.Marker.Mark()
return return
} }
defer cc.Close() defer cc.Close()
target.Marker().Reset() target.Marker.Reset()
t := time.Now() t := time.Now()
log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr)
handler.Transport(conn, cc) handler.Transport(conn, cc)
log.WithFields(map[string]interface{}{ log.WithFields(map[string]interface{}{
"duration": time.Since(t), "duration": time.Since(t),
}).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr()) }).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr)
} }

14
pkg/handler/forward/remote/handler.go
View File

@ -84,26 +84,26 @@ func (h *forwardHandler) Handle(ctx context.Context, conn net.Conn) {
} }
log = log.WithFields(map[string]interface{}{ log = log.WithFields(map[string]interface{}{
"dst": fmt.Sprintf("%s/%s", target.Addr(), network), "dst": fmt.Sprintf("%s/%s", target.Addr, network),
}) })
log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr)
cc, err := h.router.Dial(ctx, network, target.Addr()) cc, err := h.router.Dial(ctx, network, target.Addr)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
// TODO: the router itself may be failed due to the failed node in the router, // TODO: the router itself may be failed due to the failed node in the router,
// the dead marker may be a wrong operation. // the dead marker may be a wrong operation.
target.Marker().Mark() target.Marker.Mark()
return return
} }
defer cc.Close() defer cc.Close()
target.Marker().Reset() target.Marker.Reset()
t := time.Now() t := time.Now()
log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr)
handler.Transport(conn, cc) handler.Transport(conn, cc)
log.WithFields(map[string]interface{}{ log.WithFields(map[string]interface{}{
"duration": time.Since(t), "duration": time.Since(t),
}).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr()) }).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr)
} }

14
pkg/handler/relay/forward.go
View File

@ -25,17 +25,17 @@ func (h *relayHandler) handleForward(ctx context.Context, conn net.Conn, network
} }
log = log.WithFields(map[string]interface{}{ log = log.WithFields(map[string]interface{}{
"dst": fmt.Sprintf("%s/%s", target.Addr(), network), "dst": fmt.Sprintf("%s/%s", target.Addr, network),
"cmd": "forward", "cmd": "forward",
}) })
log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr)
cc, err := h.router.Dial(ctx, network, target.Addr()) cc, err := h.router.Dial(ctx, network, target.Addr)
if err != nil { if err != nil {
// TODO: the router itself may be failed due to the failed node in the router, // TODO: the router itself may be failed due to the failed node in the router,
// the dead marker may be a wrong operation. // the dead marker may be a wrong operation.
target.Marker().Mark() target.Marker.Mark()
resp.Status = relay.StatusHostUnreachable resp.Status = relay.StatusHostUnreachable
resp.WriteTo(conn) resp.WriteTo(conn)
@ -44,7 +44,7 @@ func (h *relayHandler) handleForward(ctx context.Context, conn net.Conn, network
return return
} }
defer cc.Close() defer cc.Close()
target.Marker().Reset() target.Marker.Reset()
if h.md.noDelay { if h.md.noDelay {
if _, err := resp.WriteTo(conn); err != nil { if _, err := resp.WriteTo(conn); err != nil {
@ -79,9 +79,9 @@ func (h *relayHandler) handleForward(ctx context.Context, conn net.Conn, network
} }
t := time.Now() t := time.Now()
log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s <-> %s", conn.RemoteAddr(), target.Addr)
handler.Transport(conn, cc) handler.Transport(conn, cc)
log.WithFields(map[string]interface{}{ log.WithFields(map[string]interface{}{
"duration": time.Since(t), "duration": time.Since(t),
}).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr()) }).Infof("%s >-< %s", conn.RemoteAddr(), target.Addr)
} }

4
pkg/handler/tap/handler.go
View File

@ -109,7 +109,7 @@ func (h *tapHandler) Handle(ctx context.Context, conn net.Conn) {
target := h.group.Next() target := h.group.Next()
if target != nil { if target != nil {
raddr, err = net.ResolveUDPAddr(network, target.Addr()) raddr, err = net.ResolveUDPAddr(network, target.Addr)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
return return
@ -117,7 +117,7 @@ func (h *tapHandler) Handle(ctx context.Context, conn net.Conn) {
log = log.WithFields(map[string]interface{}{ log = log.WithFields(map[string]interface{}{
"dst": fmt.Sprintf("%s/%s", raddr.String(), raddr.Network()), "dst": fmt.Sprintf("%s/%s", raddr.String(), raddr.Network()),
}) })
log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr)
} }
h.handleLoop(ctx, conn, raddr, cc.Config(), log) h.handleLoop(ctx, conn, raddr, cc.Config(), log)

4
pkg/handler/tun/handler.go
View File

@ -112,7 +112,7 @@ func (h *tunHandler) Handle(ctx context.Context, conn net.Conn) {
target := h.group.Next() target := h.group.Next()
if target != nil { if target != nil {
raddr, err = net.ResolveUDPAddr(network, target.Addr()) raddr, err = net.ResolveUDPAddr(network, target.Addr)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
return return
@ -120,7 +120,7 @@ func (h *tunHandler) Handle(ctx context.Context, conn net.Conn) {
log = log.WithFields(map[string]interface{}{ log = log.WithFields(map[string]interface{}{
"dst": fmt.Sprintf("%s/%s", raddr.String(), raddr.Network()), "dst": fmt.Sprintf("%s/%s", raddr.String(), raddr.Network()),
}) })
log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr()) log.Infof("%s >> %s", conn.RemoteAddr(), target.Addr)
} }
h.handleLoop(ctx, conn, raddr, cc.Config(), log) h.handleLoop(ctx, conn, raddr, cc.Config(), log)