143 lines
3.6 KiB
Go
143 lines
3.6 KiB
Go
package dns
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestPackUnpack(t *testing.T) {
|
|
out := new(Msg)
|
|
out.Answer = make([]RR, 1)
|
|
key := new(RR_DNSKEY)
|
|
key = &RR_DNSKEY{Flags: 257, Protocol: 3, Algorithm: RSASHA1}
|
|
key.Hdr = RR_Header{Name: "miek.nl.", Rrtype: TypeDNSKEY, Class: ClassINET, Ttl: 3600}
|
|
key.PublicKey = "AwEAAaHIwpx3w4VHKi6i1LHnTaWeHCL154Jug0Rtc9ji5qwPXpBo6A5sRv7cSsPQKPIwxLpyCrbJ4mr2L0EPOdvP6z6YfljK2ZmTbogU9aSU2fiq/4wjxbdkLyoDVgtO+JsxNN4bjr4WcWhsmk1Hg93FV9ZpkWb0Tbad8DFqNDzr//kZ"
|
|
|
|
out.Answer[0] = key
|
|
msg, ok := out.Pack()
|
|
if !ok {
|
|
t.Log("Failed to pack msg with DNSKEY")
|
|
t.Fail()
|
|
}
|
|
in := new(Msg)
|
|
if !in.Unpack(msg) {
|
|
t.Log("Failed to unpack msg with DNSKEY")
|
|
t.Fail()
|
|
}
|
|
|
|
sig := new(RR_RRSIG)
|
|
sig = &RR_RRSIG{TypeCovered: TypeDNSKEY, Algorithm: RSASHA1, Labels: 2,
|
|
OrigTtl: 3600, Expiration: 4000, Inception: 4000, KeyTag: 34641, SignerName: "miek.nl.",
|
|
Signature: "AwEAAaHIwpx3w4VHKi6i1LHnTaWeHCL154Jug0Rtc9ji5qwPXpBo6A5sRv7cSsPQKPIwxLpyCrbJ4mr2L0EPOdvP6z6YfljK2ZmTbogU9aSU2fiq/4wjxbdkLyoDVgtO+JsxNN4bjr4WcWhsmk1Hg93FV9ZpkWb0Tbad8DFqNDzr//kZ"}
|
|
sig.Hdr = RR_Header{Name: "miek.nl.", Rrtype: TypeRRSIG, Class: ClassINET, Ttl: 3600}
|
|
|
|
out.Answer[0] = sig
|
|
msg, ok = out.Pack()
|
|
if !ok {
|
|
t.Log("Failed to pack msg with RRSIG")
|
|
t.Fail()
|
|
}
|
|
|
|
if !in.Unpack(msg) {
|
|
t.Log("Failed to unpack msg with RRSIG")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestPackUnpack2(t *testing.T) {
|
|
m := new(Msg)
|
|
m.Extra = make([]RR, 1)
|
|
m.Answer = make([]RR, 1)
|
|
dom := "miek.nl."
|
|
rr := new(RR_A)
|
|
rr.Hdr = RR_Header{Name: dom, Rrtype: TypeA, Class: ClassINET, Ttl: 0}
|
|
rr.A = net.IPv4(127, 0, 0, 1)
|
|
|
|
x := new(RR_TXT)
|
|
x.Hdr = RR_Header{Name: dom, Rrtype: TypeTXT, Class: ClassINET, Ttl: 0}
|
|
x.Txt = []string{"heelalaollo"}
|
|
|
|
m.Extra[0] = x
|
|
m.Answer[0] = rr
|
|
_, ok := m.Pack()
|
|
if !ok {
|
|
t.Log("Packing failed")
|
|
t.Fail()
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestBailiwick(t *testing.T) {
|
|
yes := map[string]string{
|
|
"miek.nl": "ns.miek.nl",
|
|
".": "miek.nl",
|
|
}
|
|
for parent, child := range yes {
|
|
if !IsSubDomain(parent, child) {
|
|
t.Logf("%s should be child of %s\n", child, parent)
|
|
t.Logf("comparelabels %d", CompareLabels(parent, child))
|
|
t.Logf("lenlabels %d %d", LenLabels(parent), LenLabels(child))
|
|
t.Fail()
|
|
}
|
|
}
|
|
no := map[string]string{
|
|
"www.miek.nl": "ns.miek.nl",
|
|
"miek.nl": ".",
|
|
}
|
|
for parent, child := range no {
|
|
if IsSubDomain(parent, child) {
|
|
t.Logf("%s should not be child of %s\n", child, parent)
|
|
t.Logf("comparelabels %d", CompareLabels(parent, child))
|
|
t.Logf("lenlabels %d %d", LenLabels(parent), LenLabels(child))
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPack(t *testing.T) {
|
|
rr := []string{"US. 86400 IN NSEC 0-.us. NS SOA RRSIG NSEC DNSKEY TYPE65534"}
|
|
m := new(Msg)
|
|
var err error
|
|
m.Answer = make([]RR, 1)
|
|
for _, r := range rr {
|
|
m.Answer[0], err = NewRR(r)
|
|
if err != nil {
|
|
t.Logf("Failed to create RR: %s\n", err.Error())
|
|
t.Fail()
|
|
continue
|
|
}
|
|
if _, ok := m.Pack(); !ok {
|
|
t.Log("Packing failed")
|
|
t.Fail()
|
|
}
|
|
}
|
|
x := new(Msg)
|
|
ns, _ := NewRR("pool.ntp.org. 390 IN NS a.ntpns.org")
|
|
ns.(*RR_NS).Ns = "a.ntpns.org."
|
|
x.Ns = append(m.Ns, ns)
|
|
x.Ns = append(m.Ns, ns)
|
|
x.Ns = append(m.Ns, ns)
|
|
// This crashes due to the fact the a.ntpns.org isn't a FQDN
|
|
// How to recover() from a remove panic()?
|
|
if _, ok := x.Pack(); !ok {
|
|
t.Log("Packing failed")
|
|
t.Fail()
|
|
}
|
|
x.Answer = make([]RR, 1)
|
|
x.Answer[0], err = NewRR(rr[0])
|
|
if _, ok := x.Pack(); !ok {
|
|
t.Log("Packing failed")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestCompressLenght(t *testing.T) {
|
|
m := new(Msg)
|
|
m.SetQuestion("miek.nl", TypeMX)
|
|
ul := m.Len()
|
|
m.Compress = true
|
|
if ul != m.Len(){
|
|
t.Fatalf("Should be equal")
|
|
}
|
|
}
|