mirror of
https://github.com/soypat/lneto.git
synced 2026-09-11 00:59:30 +00:00
refactor(dns): eagerly decode CNAME target into r.data, drop target field
Address review feedback on #189: - Remove Resource.target: Resource.Decode expands CNAME RDATA in place into r.data (uncompressed wire format) while the full message is still available and updates header.Length, storing the name bytes exactly once. - Add Resource.CNAMEView returning a length-bounded view of the expanded target; Message.WriteAnswers uses it. - Rename matchesHost to ResourceHeader.pertainsTo. - Make ResolveConfig.MaxCNAMEs explicit: drop the silent default in Client.StartResolve and let callers opt in (x/xnet). - Reduce test suite to regression coverage only. refs #189
This commit is contained in:
+69
-33
@@ -48,7 +48,6 @@ type Question struct {
|
||||
type Resource struct {
|
||||
header ResourceHeader
|
||||
data []byte
|
||||
target Name
|
||||
}
|
||||
|
||||
// A ResourceHeader is the header of a DNS resource record. There are
|
||||
@@ -300,50 +299,66 @@ func (m *Message) AppendTo(buf []byte, txid uint16, flags HeaderFlags) (_ []byte
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
// WriteAnswers follows CNAMEs from host among the answers and writes the
|
||||
// resulting addresses into dst.
|
||||
func (m *Message) WriteAnswers(dst []netip.Addr, host string) (n uint16, err error) {
|
||||
var alias Name
|
||||
// Each pass follows at most one alias hop
|
||||
// The pass limit also bounds CNAME cycles in malformed responses.
|
||||
for range m.Answers {
|
||||
var next Name
|
||||
n = 0
|
||||
var (
|
||||
next Name
|
||||
hasAddrs bool
|
||||
)
|
||||
for i := range m.Answers {
|
||||
if int(n) >= len(dst) {
|
||||
break
|
||||
}
|
||||
ans := &m.Answers[i]
|
||||
hdr := ans.Header()
|
||||
matched := alias.Len() == 0 && hdr.Name.EqualString(host) ||
|
||||
alias.Len() != 0 && NamesEqual(hdr.Name, alias)
|
||||
if !matched {
|
||||
if !hdr.pertainsTo(alias, host) {
|
||||
continue
|
||||
}
|
||||
switch hdr.Type {
|
||||
case TypeA, TypeAAAA:
|
||||
var ok bool
|
||||
dst[n], ok = netip.AddrFromSlice(ans.RawData())
|
||||
if !ok {
|
||||
err = lneto.ErrInvalidAddr
|
||||
continue
|
||||
} else {
|
||||
n++
|
||||
}
|
||||
hasAddrs = true
|
||||
case TypeCNAME:
|
||||
if ans.target.Len() != 0 {
|
||||
next = ans.target
|
||||
if cname, ok := ans.CNAMEView(); ok && cname.Len() != 0 {
|
||||
next = cname
|
||||
}
|
||||
}
|
||||
}
|
||||
if next.Len() == 0 || n > 0 {
|
||||
// Either no new alias was found or addresses were resolved for
|
||||
// the current name; a name cannot alias and hold records at once.
|
||||
if hasAddrs || next.Len() == 0 {
|
||||
break
|
||||
}
|
||||
alias = next
|
||||
}
|
||||
|
||||
for i := range m.Answers {
|
||||
if int(n) >= len(dst) {
|
||||
return n, lneto.ErrExhausted
|
||||
}
|
||||
ans := &m.Answers[i]
|
||||
hdr := ans.Header()
|
||||
isAddr := hdr.Type == TypeA || hdr.Type == TypeAAAA
|
||||
if !isAddr || !hdr.pertainsTo(alias, host) {
|
||||
continue
|
||||
}
|
||||
var ok bool
|
||||
dst[n], ok = netip.AddrFromSlice(ans.RawData())
|
||||
if !ok {
|
||||
err = lneto.ErrInvalidAddr
|
||||
} else {
|
||||
n++
|
||||
}
|
||||
}
|
||||
return n, err
|
||||
}
|
||||
|
||||
// pertainsTo reports whether the record's owner name is the given name: host
|
||||
// or one of the aliases resolved so far.
|
||||
func (h *ResourceHeader) pertainsTo(alias Name, host string) bool {
|
||||
if alias.Len() == 0 {
|
||||
return h.Name.EqualString(host)
|
||||
}
|
||||
return NamesEqual(h.Name, alias)
|
||||
}
|
||||
|
||||
func (m *Message) Len() uint16 {
|
||||
return SizeHeader + m.lenResources()
|
||||
}
|
||||
@@ -423,7 +438,6 @@ func (h *ResourceHeader) String() string {
|
||||
func (r *Resource) Reset() {
|
||||
r.header.Reset()
|
||||
r.data = r.data[:0]
|
||||
r.target.Reset()
|
||||
}
|
||||
|
||||
func (r *Resource) Header() ResourceHeader { return r.header }
|
||||
@@ -436,6 +450,13 @@ func (r *Resource) RawData() []byte {
|
||||
return r.data[:length]
|
||||
}
|
||||
|
||||
func (r *Resource) CNAMEView() (Name, bool) {
|
||||
if r.header.Type == TypeCNAME {
|
||||
return Name{data: r.RawData()}, true
|
||||
}
|
||||
return Name{}, false
|
||||
}
|
||||
|
||||
func (q *Question) Reset() {
|
||||
q.Name.Reset()
|
||||
*q = Question{Name: q.Name} // Reuse Name's buffer.
|
||||
@@ -485,16 +506,32 @@ func (r *Resource) Decode(b []byte, off uint16) (uint16, error) {
|
||||
if r.header.Length > uint16(len(b[off:])) {
|
||||
return off, errResourceLen
|
||||
}
|
||||
end := off + r.header.Length
|
||||
r.data = append(r.data[:0], b[off:end]...)
|
||||
if r.header.Type == TypeCNAME {
|
||||
_, err = r.target.Decode(b, off)
|
||||
if err != nil {
|
||||
r.target.Reset() // Tolerate undecodable target; CNAME chain lookup skips it.
|
||||
raw := b[off:end]
|
||||
data, derr := expandName(r.data[:0], b, off)
|
||||
if derr == nil {
|
||||
r.data = data
|
||||
r.header.Length = uint16(len(r.data))
|
||||
} else {
|
||||
r.data = append(r.data[:0], raw...)
|
||||
}
|
||||
} else {
|
||||
r.target.Reset()
|
||||
}
|
||||
r.data = append(r.data[:0], b[off:off+r.header.Length]...)
|
||||
return off + r.header.Length, nil
|
||||
return end, nil
|
||||
}
|
||||
|
||||
func expandName(buf []byte, msg []byte, off uint16) ([]byte, error) {
|
||||
appended := 0
|
||||
_, err := visitAllLabels(msg, off, func(label []byte) {
|
||||
buf = append(buf, byte(len(label)))
|
||||
buf = append(buf, label...)
|
||||
appended += 1 + len(label)
|
||||
}, allowCompression)
|
||||
if err != nil {
|
||||
return buf[:len(buf)-appended], err
|
||||
}
|
||||
return append(buf, 0), nil
|
||||
}
|
||||
|
||||
func (r *Resource) appendTo(buf []byte) (_ []byte, err error) {
|
||||
@@ -808,7 +845,6 @@ func (dst *Question) CopyFrom(q Question) {
|
||||
func (dst *Resource) CopyFrom(r Resource) {
|
||||
dst.header.CopyFrom(r.header)
|
||||
dst.data = append(dst.data[:0], r.data...)
|
||||
dst.target.CopyFrom(r.target)
|
||||
}
|
||||
|
||||
// SetA sets an A (IPv4 address) resource record, reusing internal buffers.
|
||||
|
||||
Reference in New Issue
Block a user