|
|
@ -2,7 +2,7 @@
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
|
|
|
|
package dns
|
|
|
|
package resolver
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"bytes"
|
|
|
@ -194,7 +194,7 @@ func TestRDNSNameToIPv6(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func TestResolve(t *testing.T) {
|
|
|
|
func TestResolve(t *testing.T) {
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
@ -240,7 +240,7 @@ func TestResolve(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func TestResolveReverse(t *testing.T) {
|
|
|
|
func TestResolveReverse(t *testing.T) {
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
@ -318,7 +318,7 @@ func TestDelegate(t *testing.T) {
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetUpstreams([]net.Addr{
|
|
|
|
r.SetUpstreams([]net.Addr{
|
|
|
|
v4server.PacketConn.LocalAddr(),
|
|
|
|
v4server.PacketConn.LocalAddr(),
|
|
|
@ -397,7 +397,7 @@ func TestDelegateCollision(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
defer server.Shutdown()
|
|
|
|
defer server.Shutdown()
|
|
|
|
|
|
|
|
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetUpstreams([]net.Addr{server.PacketConn.LocalAddr()})
|
|
|
|
r.SetUpstreams([]net.Addr{server.PacketConn.LocalAddr()})
|
|
|
|
|
|
|
|
|
|
|
@ -463,7 +463,7 @@ func TestDelegateCollision(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func TestConcurrentSetMap(t *testing.T) {
|
|
|
|
func TestConcurrentSetMap(t *testing.T) {
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
t.Fatalf("start: %v", err)
|
|
|
|
t.Fatalf("start: %v", err)
|
|
|
@ -499,7 +499,7 @@ func TestConcurrentSetUpstreams(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
defer server.Shutdown()
|
|
|
|
defer server.Shutdown()
|
|
|
|
|
|
|
|
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: true})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
@ -670,7 +670,7 @@ var emptyResponse = []byte{
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func TestFull(t *testing.T) {
|
|
|
|
func TestFull(t *testing.T) {
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
@ -709,7 +709,7 @@ func TestFull(t *testing.T) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func TestAllocs(t *testing.T) {
|
|
|
|
func TestAllocs(t *testing.T) {
|
|
|
|
r := NewResolver(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r := New(ResolverConfig{Logf: t.Logf, Forward: false})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
|
|
|
|
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
|
if err := r.Start(); err != nil {
|
|
|
@ -778,7 +778,7 @@ func BenchmarkFull(b *testing.B) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
defer server.Shutdown()
|
|
|
|
defer server.Shutdown()
|
|
|
|
|
|
|
|
|
|
|
|
r := NewResolver(ResolverConfig{Logf: b.Logf, Forward: true})
|
|
|
|
r := New(ResolverConfig{Logf: b.Logf, Forward: true})
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetMap(dnsMap)
|
|
|
|
r.SetUpstreams([]net.Addr{server.PacketConn.LocalAddr()})
|
|
|
|
r.SetUpstreams([]net.Addr{server.PacketConn.LocalAddr()})
|
|
|
|
|
|
|
|
|