mirror of https://github.com/fatedier/frp
24 lines
530 B
Go
24 lines
530 B
Go
package socks5
|
|
|
|
import (
|
|
"net"
|
|
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
// NameResolver is used to implement custom name resolution
|
|
type NameResolver interface {
|
|
Resolve(ctx context.Context, name string) (context.Context, net.IP, error)
|
|
}
|
|
|
|
// DNSResolver uses the system DNS to resolve host names
|
|
type DNSResolver struct{}
|
|
|
|
func (d DNSResolver) Resolve(ctx context.Context, name string) (context.Context, net.IP, error) {
|
|
addr, err := net.ResolveIPAddr("ip", name)
|
|
if err != nil {
|
|
return ctx, nil, err
|
|
}
|
|
return ctx, addr.IP, err
|
|
}
|