2014-01-23 19:14:06 +00:00
|
|
|
package socks5
|
|
|
|
|
|
|
|
import (
|
2020-04-19 09:08:22 +00:00
|
|
|
"context"
|
2014-01-23 19:14:06 +00:00
|
|
|
"net"
|
|
|
|
)
|
|
|
|
|
|
|
|
// NameResolver is used to implement custom name resolution
|
|
|
|
type NameResolver interface {
|
2016-03-08 14:29:11 +00:00
|
|
|
Resolve(ctx context.Context, name string) (context.Context, net.IP, error)
|
2014-01-23 19:14:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// DNSResolver uses the system DNS to resolve host names
|
|
|
|
type DNSResolver struct{}
|
|
|
|
|
2016-03-08 14:29:11 +00:00
|
|
|
func (d DNSResolver) Resolve(ctx context.Context, name string) (context.Context, net.IP, error) {
|
2014-01-23 21:07:13 +00:00
|
|
|
addr, err := net.ResolveIPAddr("ip", name)
|
|
|
|
if err != nil {
|
2016-03-08 14:29:11 +00:00
|
|
|
return ctx, nil, err
|
2014-01-23 21:07:13 +00:00
|
|
|
}
|
2016-03-08 14:29:11 +00:00
|
|
|
return ctx, addr.IP, err
|
2014-01-23 19:14:06 +00:00
|
|
|
}
|