140 lines
4.3 KiB
Go
140 lines
4.3 KiB
Go
package dns
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestDynamicUpdateParsing(t *testing.T) {
|
|
prefix := "example.com. IN "
|
|
for _, typ := range TypeToString {
|
|
if typ == "OPT" || typ == "AXFR" || typ == "IXFR" || typ == "ANY" || typ == "TKEY" ||
|
|
typ == "TSIG" || typ == "ISDN" || typ == "UNSPEC" || typ == "NULL" || typ == "ATMA" ||
|
|
typ == "Reserved" || typ == "None" || typ == "NXT" || typ == "MAILB" || typ == "MAILA" {
|
|
continue
|
|
}
|
|
if _, err := NewRR(prefix + typ); err != nil {
|
|
t.Errorf("failure to parse: %s %s: %v", prefix, typ, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDynamicUpdateUnpack(t *testing.T) {
|
|
// From https://github.com/miekg/dns/issues/150#issuecomment-62296803
|
|
// It should be an update message for the zone "example.",
|
|
// deleting the A RRset "example." and then adding an A record at "example.".
|
|
// class ANY, TYPE A
|
|
buf := []byte{171, 68, 40, 0, 0, 1, 0, 0, 0, 2, 0, 0, 7, 101, 120, 97, 109, 112, 108, 101, 0, 0, 6, 0, 1, 192, 12, 0, 1, 0, 255, 0, 0, 0, 0, 0, 0, 192, 12, 0, 1, 0, 1, 0, 0, 0, 0, 0, 4, 127, 0, 0, 1}
|
|
msg := new(Msg)
|
|
err := msg.Unpack(buf)
|
|
if err != nil {
|
|
t.Errorf("failed to unpack: %v\n%s", err, msg.String())
|
|
}
|
|
}
|
|
|
|
func TestDynamicUpdateZeroRdataUnpack(t *testing.T) {
|
|
m := new(Msg)
|
|
rr := &RR_Header{Name: ".", Rrtype: 0, Class: 1, Ttl: ^uint32(0), Rdlength: 0}
|
|
m.Answer = []RR{rr, rr, rr, rr, rr}
|
|
m.Ns = m.Answer
|
|
for n, s := range TypeToString {
|
|
rr.Rrtype = n
|
|
bytes, err := m.Pack()
|
|
if err != nil {
|
|
t.Errorf("failed to pack %s: %v", s, err)
|
|
continue
|
|
}
|
|
if err := new(Msg).Unpack(bytes); err != nil {
|
|
t.Errorf("failed to unpack %s: %v", s, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRemoveRRset(t *testing.T) {
|
|
// Should add a zero data RR in Class ANY with a TTL of 0
|
|
// for each set mentioned in the RRs provided to it.
|
|
rr := testRR(". 100 IN A 127.0.0.1")
|
|
m := new(Msg)
|
|
m.Ns = []RR{&RR_Header{Name: ".", Rrtype: TypeA, Class: ClassANY, Ttl: 0, Rdlength: 0}}
|
|
expectstr := m.String()
|
|
expect, err := m.Pack()
|
|
if err != nil {
|
|
t.Fatalf("error packing expected msg: %v", err)
|
|
}
|
|
|
|
m.Ns = nil
|
|
m.RemoveRRset([]RR{rr})
|
|
actual, err := m.Pack()
|
|
if err != nil {
|
|
t.Fatalf("error packing actual msg: %v", err)
|
|
}
|
|
if !bytes.Equal(actual, expect) {
|
|
tmp := new(Msg)
|
|
if err := tmp.Unpack(actual); err != nil {
|
|
t.Fatalf("error unpacking actual msg: %v\nexpected: %v\ngot: %v\n", err, expect, actual)
|
|
}
|
|
t.Errorf("expected msg:\n%s", expectstr)
|
|
t.Errorf("actual msg:\n%v", tmp)
|
|
}
|
|
}
|
|
|
|
func TestPreReqAndRemovals(t *testing.T) {
|
|
// Build a list of multiple prereqs and then somes removes followed by an insert.
|
|
// We should be able to add multiple prereqs and updates.
|
|
m := new(Msg)
|
|
m.SetUpdate("example.org.")
|
|
m.Id = 1234
|
|
|
|
// Use a full set of RRs each time, so we are sure the rdata is stripped.
|
|
rrName1 := testRR("name_used. 3600 IN A 127.0.0.1")
|
|
rrName2 := testRR("name_not_used. 3600 IN A 127.0.0.1")
|
|
rrRemove1 := testRR("remove1. 3600 IN A 127.0.0.1")
|
|
rrRemove2 := testRR("remove2. 3600 IN A 127.0.0.1")
|
|
rrRemove3 := testRR("remove3. 3600 IN A 127.0.0.1")
|
|
rrInsert := testRR("insert. 3600 IN A 127.0.0.1")
|
|
rrRrset1 := testRR("rrset_used1. 3600 IN A 127.0.0.1")
|
|
rrRrset2 := testRR("rrset_used2. 3600 IN A 127.0.0.1")
|
|
rrRrset3 := testRR("rrset_not_used. 3600 IN A 127.0.0.1")
|
|
|
|
// Handle the prereqs.
|
|
m.NameUsed([]RR{rrName1})
|
|
m.NameNotUsed([]RR{rrName2})
|
|
m.RRsetUsed([]RR{rrRrset1})
|
|
m.Used([]RR{rrRrset2})
|
|
m.RRsetNotUsed([]RR{rrRrset3})
|
|
|
|
// and now the updates.
|
|
m.RemoveName([]RR{rrRemove1})
|
|
m.RemoveRRset([]RR{rrRemove2})
|
|
m.Remove([]RR{rrRemove3})
|
|
m.Insert([]RR{rrInsert})
|
|
|
|
// This test function isn't a Example function because we print these RR with tabs at the
|
|
// end and the Example function trim these, thus they never match.
|
|
// TODO(miek): don't print these tabs and make this into an Example function.
|
|
expect := `;; opcode: UPDATE, status: NOERROR, id: 1234
|
|
;; flags:; QUERY: 1, ANSWER: 5, AUTHORITY: 4, ADDITIONAL: 0
|
|
|
|
;; QUESTION SECTION:
|
|
;example.org. IN SOA
|
|
|
|
;; ANSWER SECTION:
|
|
name_used. 0 CLASS255 ANY
|
|
name_not_used. 0 NONE ANY
|
|
rrset_used1. 0 CLASS255 A
|
|
rrset_used2. 3600 IN A 127.0.0.1
|
|
rrset_not_used. 0 NONE A
|
|
|
|
;; AUTHORITY SECTION:
|
|
remove1. 0 CLASS255 ANY
|
|
remove2. 0 CLASS255 A
|
|
remove3. 0 NONE A 127.0.0.1
|
|
insert. 3600 IN A 127.0.0.1
|
|
`
|
|
|
|
if m.String() != expect {
|
|
t.Errorf("expected msg:\n%s", expect)
|
|
t.Errorf("actual msg:\n%v", m.String())
|
|
}
|
|
}
|