Fix closest encloser

This commit is contained in:
Miek Gieben
2016-03-30 20:13:48 +01:00
parent bf6d90600b
commit 8feef98188
4 changed files with 58 additions and 17 deletions

View File

@@ -4,13 +4,19 @@ import "github.com/miekg/dns"
// ClosestEncloser returns the closest encloser for rr.
func (z *Zone) ClosestEncloser(rr dns.RR) string {
elem := z.Tree.Prev(rr)
if elem == nil {
// SOA?
return ""
// tree/tree.go does not store a parent *Node pointer, so we can't
// just follow up the tree. TODO(miek): fix.
offset, end := dns.NextLabel(rr.Header().Name, 0)
for !end {
elem := z.Tree.Get(rr)
if elem != nil {
return elem.Name()
}
rr.Header().Name = rr.Header().Name[offset:]
offset, end = dns.NextLabel(rr.Header().Name, offset)
}
for _, r := range elem.All() {
return r.Header().Name
}
return ""
return z.SOA.Header().Name
}