93 lines
1.6 KiB
Go
93 lines
1.6 KiB
Go
package dns
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func getIP(s string) string {
|
|
a, err := net.LookupAddr(s)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return a[0]
|
|
}
|
|
|
|
// flaky, need to setup local server and test from
|
|
// that.
|
|
func testClientAXFR(t *testing.T) {
|
|
if testing.Short() {
|
|
return
|
|
}
|
|
m := new(Msg)
|
|
m.SetAxfr("miek.nl.")
|
|
|
|
server := getIP("linode.atoom.net")
|
|
|
|
tr := new(Transfer)
|
|
|
|
if a, err := tr.In(m, net.JoinHostPort(server, "53")); err != nil {
|
|
t.Fatal("failed to setup axfr: ", err)
|
|
} else {
|
|
for ex := range a {
|
|
if ex.Error != nil {
|
|
t.Errorf("error %v", ex.Error)
|
|
break
|
|
}
|
|
for _, rr := range ex.RR {
|
|
t.Logf("%s\n", rr.String())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// fails.
|
|
func testClientAXFRMultipleEnvelopes(t *testing.T) {
|
|
if testing.Short() {
|
|
return
|
|
}
|
|
m := new(Msg)
|
|
m.SetAxfr("nlnetlabs.nl.")
|
|
|
|
server := getIP("open.nlnetlabs.nl.")
|
|
|
|
tr := new(Transfer)
|
|
if a, err := tr.In(m, net.JoinHostPort(server, "53")); err != nil {
|
|
t.Fatalf("Failed to setup axfr %v for server: %v", err, server)
|
|
} else {
|
|
for ex := range a {
|
|
if ex.Error != nil {
|
|
t.Errorf("Error %v", ex.Error)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func testClientTsigAXFR(t *testing.T) {
|
|
if testing.Short() {
|
|
return
|
|
}
|
|
m := new(Msg)
|
|
m.SetAxfr("example.nl.")
|
|
m.SetTsig("axfr.", HmacMD5, 300, time.Now().Unix())
|
|
|
|
tr := new(Transfer)
|
|
tr.TsigSecret = map[string]string{"axfr.": "so6ZGir4GPAqINNh9U5c3A=="}
|
|
|
|
if a, err := tr.In(m, "176.58.119.54:53"); err != nil {
|
|
t.Fatal("failed to setup axfr: ", err)
|
|
} else {
|
|
for ex := range a {
|
|
if ex.Error != nil {
|
|
t.Errorf("error %v", ex.Error)
|
|
break
|
|
}
|
|
for _, rr := range ex.RR {
|
|
t.Logf("%s\n", rr.String())
|
|
}
|
|
}
|
|
}
|
|
}
|