mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-11-27 11:01:27 +00:00
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package dns
|
|
|
|
import (
|
|
"github.com/xtls/xray-core/common/errors"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/common/serial"
|
|
"github.com/xtls/xray-core/features"
|
|
)
|
|
|
|
// IPOption is an object for IP query options.
|
|
type IPOption struct {
|
|
IPv4Enable bool
|
|
IPv6Enable bool
|
|
FakeEnable bool
|
|
}
|
|
|
|
func (p *IPOption) Copy() *IPOption {
|
|
return &IPOption{p.IPv4Enable, p.IPv6Enable, p.FakeEnable}
|
|
}
|
|
|
|
type Option func(dopt *IPOption) *IPOption
|
|
|
|
// Client is a Xray feature for querying DNS information.
|
|
//
|
|
// xray:api:stable
|
|
type Client interface {
|
|
features.Feature
|
|
|
|
// LookupIP returns IP address for the given domain. IPs may contain IPv4 and/or IPv6 addresses.
|
|
LookupIP(domain string) ([]net.IP, error)
|
|
|
|
// LookupOptions query IP address for domain with *IPOption.
|
|
LookupOptions(domain string, opt ...Option) ([]net.IP, error)
|
|
}
|
|
|
|
// IPv4Lookup is an optional feature for querying IPv4 addresses only.
|
|
//
|
|
// xray:api:beta
|
|
type IPv4Lookup interface {
|
|
LookupIPv4(domain string) ([]net.IP, error)
|
|
}
|
|
|
|
// IPv6Lookup is an optional feature for querying IPv6 addresses only.
|
|
//
|
|
// xray:api:beta
|
|
type IPv6Lookup interface {
|
|
LookupIPv6(domain string) ([]net.IP, error)
|
|
}
|
|
|
|
// ClientType returns the type of Client interface. Can be used for implementing common.HasType.
|
|
//
|
|
// xray:api:beta
|
|
func ClientType() interface{} {
|
|
return (*Client)(nil)
|
|
}
|
|
|
|
// ErrEmptyResponse indicates that DNS query succeeded but no answer was returned.
|
|
var ErrEmptyResponse = errors.New("empty response")
|
|
|
|
type RCodeError uint16
|
|
|
|
func (e RCodeError) Error() string {
|
|
return serial.Concat("rcode: ", uint16(e))
|
|
}
|
|
|
|
func RCodeFromError(err error) uint16 {
|
|
if err == nil {
|
|
return 0
|
|
}
|
|
cause := errors.Cause(err)
|
|
if r, ok := cause.(RCodeError); ok {
|
|
return uint16(r)
|
|
}
|
|
return 0
|
|
}
|
|
|
|
var (
|
|
LookupIPv4Only = func(d *IPOption) *IPOption {
|
|
d.IPv4Enable = true
|
|
d.IPv6Enable = false
|
|
return d
|
|
}
|
|
LookupIPv6Only = func(d *IPOption) *IPOption {
|
|
d.IPv4Enable = false
|
|
d.IPv6Enable = true
|
|
return d
|
|
}
|
|
LookupIP = func(d *IPOption) *IPOption {
|
|
d.IPv4Enable = true
|
|
d.IPv6Enable = true
|
|
return d
|
|
}
|
|
LookupFake = func(d *IPOption) *IPOption {
|
|
d.FakeEnable = true
|
|
return d
|
|
}
|
|
LookupNoFake = func(d *IPOption) *IPOption {
|
|
d.FakeEnable = false
|
|
return d
|
|
}
|
|
|
|
LookupAll = func(d *IPOption) *IPOption {
|
|
LookupIP(d)
|
|
LookupFake(d)
|
|
return d
|
|
}
|
|
)
|