Mirrors/coredns
1
0
Fork 0
mirror of https://github.com/coredns/coredns.git synced 2025-10-27 16:24:19 -04:00
Code Issues Packages Projects Releases Wiki Activity

plugin/template: Add client IP data (#4034)

Browse Source 
Signed-off-by: Maxime Guyot <maxime@root314.com>
This commit is contained in:
Maxime Guyot
2020-08-10 10:38:18 +02:00
committed by GitHub
parent 7d5f5b87a4
commit b3d08f9b21
3 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
plugin/template/template_test.go
View File

@@ -25,6 +25,14 @@ func TestHandler(t *testing.T) {
fall: fall.Root,
zones: []string{"."},
}
exampleDomainIPATemplate := template{
regex: []*regexp.Regexp{regexp.MustCompile(".*")},
answer: []*gotmpl.Template{gotmpl.Must(gotmpl.New("answer").Parse("{{ .Name }} 60 IN A {{ .Remote }}"))},
qclass: dns.ClassINET,
qtype: dns.TypeA,
fall: fall.Root,
zones: []string{"."},
}
exampleDomainANSTemplate := template{
regex: []*regexp.Regexp{regexp.MustCompile("(^|[.])ip-10-(?P<b>[0-9]*)-(?P<c>[0-9]*)-(?P<d>[0-9]*)[.]example[.]$")},
answer: []*gotmpl.Template{gotmpl.Must(gotmpl.New("answer").Parse("{{ .Name }} 60 IN A 10.{{ .Group.b }}.{{ .Group.c }}.{{ .Group.d }}"))},
@@ -196,6 +204,25 @@ func TestHandler(t *testing.T) {
return nil
},
},
{
name: "ExampleIPMatch",
tmpl: exampleDomainIPATemplate,
qclass: dns.ClassINET,
qtype: dns.TypeA,
qname: "test.example.",
verifyResponse: func(r *dns.Msg) error {
if len(r.Answer) != 1 {
return fmt.Errorf("expected 1 answer, got %v", len(r.Answer))
}
if r.Answer[0].Header().Rrtype != dns.TypeA {
return fmt.Errorf("expected an A record answer, got %v", dns.TypeToString[r.Answer[0].Header().Rrtype])
}
if r.Answer[0].(*dns.A).A.String() != "10.240.0.1" {
return fmt.Errorf("expected an A record for 10.95.12.8, got %v", r.Answer[0].String())
}
return nil
},
},
{
name: "ExampleDomainMatch",
tmpl: exampleDomainATemplate,