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

add resolver & hosts for chain

Browse Source
This commit is contained in:
ginuerzh
2022-01-30 00:43:36 +08:00
parent 0e9a975a26
commit c4e9b35484
3 changed files with 107 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
pkg/chain/router.go
View File

@ -52,12 +52,16 @@ func (r *Router) dial(ctx context.Context, network, address string) (conn net.Co
r.Logger.Debugf("route(retry=%d) %s", i, buf.String())
}
address, err = r.resolve(ctx, address)
address, err = resolve(ctx, address, r.Resolver, r.Hosts, r.Logger)
if err != nil {
r.Logger.Error(err)
break
}
if route != nil {
route.logger = r.Logger
}
conn, err = route.Dial(ctx, network, address)
if err == nil {
break
@ -68,39 +72,6 @@ func (r *Router) dial(ctx context.Context, network, address string) (conn net.Co
return
}
func (r *Router) resolve(ctx context.Context, addr string) (string, error) {
if addr == "" {
return addr, nil
}
host, port, err := net.SplitHostPort(addr)
if err != nil {
return "", err
}
if host == "" {
return addr, nil
}
if r.Hosts != nil {
if ips, _ := r.Hosts.Lookup("ip", host); len(ips) > 0 {
r.Logger.Debugf("hit host mapper: %s -> %s", host, ips)
return net.JoinHostPort(ips[0].String(), port), nil
}
}
if r.Resolver != nil {
ips, err := r.Resolver.Resolve(ctx, host)
if err != nil {
r.Logger.Error(err)
}
if len(ips) == 0 {
return "", fmt.Errorf("resolver: domain %s does not exists", host)
}
return net.JoinHostPort(ips[0].String(), port), nil
}
return addr, nil
}
func (r *Router) Bind(ctx context.Context, network, address string, opts ...connector.BindOption) (ln net.Listener, err error) {
count := r.Retries + 1
if count <= 0 {