2 Commits

Author SHA1 Message Date
Vladislav Yarmak
6ed529d8eb fix panic 2021-06-30 00:22:49 +03:00
Vladislav Yarmak
6232a3dea9 drop old resolver 2021-06-26 14:52:43 +03:00
2 changed files with 3 additions and 67 deletions

View File

@@ -21,73 +21,6 @@ const (
DNS_CACHE_SIZE_LIMIT = 1024
)
func NewResolver(address string, timeout time.Duration) (*Resolver, error) {
opts := upstream.Options{Timeout: timeout}
u, err := upstream.AddressToUpstream(address, opts)
if err != nil {
return nil, err
}
return &Resolver{upstream: u}, nil
}
func (r *Resolver) ResolveA(domain string) []string {
res := make([]string, 0)
if len(domain) == 0 {
return res
}
domain = absDomain(domain)
req := dns.Msg{}
req.Id = dns.Id()
req.RecursionDesired = true
req.Question = []dns.Question{
{Name: domain, Qtype: dns.TypeA, Qclass: dns.ClassINET},
}
reply, err := r.upstream.Exchange(&req)
if err != nil {
return res
}
for _, rr := range reply.Answer {
if a, ok := rr.(*dns.A); ok {
res = append(res, a.A.String())
}
}
return res
}
func (r *Resolver) ResolveAAAA(domain string) []string {
res := make([]string, 0)
if len(domain) == 0 {
return res
}
domain = absDomain(domain)
req := dns.Msg{}
req.Id = dns.Id()
req.RecursionDesired = true
req.Question = []dns.Question{
{Name: domain, Qtype: dns.TypeAAAA, Qclass: dns.ClassINET},
}
reply, err := r.upstream.Exchange(&req)
if err != nil {
return res
}
for _, rr := range reply.Answer {
if a, ok := rr.(*dns.AAAA); ok {
res = append(res, a.AAAA.String())
}
}
return res
}
func (r *Resolver) Resolve(domain string) []string {
res := r.ResolveA(domain)
if len(res) == 0 {
res = r.ResolveAAAA(domain)
}
return res
}
type ResolvingDialer struct {
next ContextDialer
upstream upstream.Upstream

View File

@@ -199,6 +199,9 @@ func NewNoSNIDialer(caPool *x509.CertPool, nextDialer ContextDialer) *NoSNIDiale
func (d *NoSNIDialer) DialTLSContext(ctx context.Context, network, addr string) (net.Conn, error) {
conn, err := d.next.DialContext(ctx, network, addr)
if err != nil {
return nil, err
}
name, _, err := net.SplitHostPort(addr)
if err != nil {