e43384b501
Add support for wildcard records, while taking care of wildcard-cnames and DNSSEC. Add enough tests to check all the corner cases.
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/miekg/coredns/middleware/proxy"
|
|
"github.com/miekg/coredns/middleware/test"
|
|
"github.com/miekg/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestLookupWildcard(t *testing.T) {
|
|
name, rm, err := test.TempFile(".", exampleOrg)
|
|
if err != nil {
|
|
t.Fatalf("failed to created zone: %s", err)
|
|
}
|
|
defer rm()
|
|
|
|
corefile := `example.org:0 {
|
|
file ` + name + `
|
|
}
|
|
`
|
|
|
|
i, err := CoreDNSServer(corefile)
|
|
if err != nil {
|
|
t.Fatalf("Could not get CoreDNS serving instance: %s", err)
|
|
}
|
|
|
|
udp, _ := CoreDNSServerPorts(i, 0)
|
|
if udp == "" {
|
|
t.Fatalf("Could not get UDP listening port")
|
|
}
|
|
defer i.Stop()
|
|
|
|
log.SetOutput(ioutil.Discard)
|
|
|
|
p := proxy.New([]string{udp})
|
|
state := request.Request{W: &test.ResponseWriter{}, Req: new(dns.Msg)}
|
|
|
|
for _, lookup := range []string{"w.example.org.", "a.w.example.org.", "a.a.w.example.org."} {
|
|
resp, err := p.Lookup(state, lookup, dns.TypeTXT)
|
|
if err != nil || resp == nil {
|
|
t.Fatal("Expected to receive reply, but didn't for %s", lookup)
|
|
}
|
|
|
|
// ;; ANSWER SECTION:
|
|
// a.w.example.org. 1800 IN TXT "Wildcard"
|
|
if resp.Rcode != dns.RcodeSuccess {
|
|
t.Errorf("Expected NOERROR RCODE, got %s for %s", dns.RcodeToString[resp.Rcode], lookup)
|
|
continue
|
|
}
|
|
if len(resp.Answer) == 0 {
|
|
t.Errorf("Expected to at least one RR in the answer section, got none for %s TXT", lookup)
|
|
t.Logf("%s", resp)
|
|
continue
|
|
}
|
|
if resp.Answer[0].Header().Name != lookup {
|
|
t.Errorf("Expected name to be %s, got: %s for TXT", lookup, resp.Answer[0].Header().Name)
|
|
continue
|
|
}
|
|
if resp.Answer[0].Header().Rrtype != dns.TypeTXT {
|
|
t.Errorf("Expected RR to be TXT, got: %d, for %s TXT", resp.Answer[0].Header().Rrtype, lookup)
|
|
continue
|
|
}
|
|
if resp.Answer[0].(*dns.TXT).Txt[0] != "Wildcard" {
|
|
t.Errorf("Expected Wildcard, got: %s, for %s TXT", resp.Answer[0].(*dns.TXT).Txt[0], lookup)
|
|
continue
|
|
}
|
|
}
|
|
|
|
for _, lookup := range []string{"w.example.org.", "a.w.example.org.", "a.a.w.example.org."} {
|
|
resp, err := p.Lookup(state, lookup, dns.TypeSRV)
|
|
if err != nil || resp == nil {
|
|
t.Fatal("Expected to receive reply, but didn't", lookup)
|
|
}
|
|
|
|
// ;; AUTHORITY SECTION:
|
|
// example.org. 1800 IN SOA linode.atoom.net. miek.miek.nl. 1454960557 14400 3600 604800 14400
|
|
if resp.Rcode != dns.RcodeSuccess {
|
|
t.Errorf("Expected NOERROR RCODE, got %s for %s", dns.RcodeToString[resp.Rcode], lookup)
|
|
continue
|
|
}
|
|
if len(resp.Answer) != 0 {
|
|
t.Errorf("Expected zero RRs in the answer section, got some, for %s SRV", lookup)
|
|
continue
|
|
}
|
|
if len(resp.Ns) == 0 {
|
|
t.Errorf("Expected to at least one RR in the authority section, got none, for %s SRV", lookup)
|
|
continue
|
|
}
|
|
if resp.Ns[0].Header().Rrtype != dns.TypeSOA {
|
|
t.Errorf("Expected RR to be SOA, got: %d, for %s SRV", resp.Ns[0].Header().Rrtype, lookup)
|
|
continue
|
|
}
|
|
}
|
|
|
|
}
|