121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package weavedns
|
|
|
|
import (
|
|
"github.com/miekg/dns"
|
|
"log"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
success_test_name = "test1.weave."
|
|
fail_test_name = "test2.weave."
|
|
test_addr = net.ParseIP("9.8.7.6")
|
|
)
|
|
|
|
func minimalServer(w dns.ResponseWriter, req *dns.Msg) {
|
|
//log.Println("minimalServer received:", req)
|
|
if len(req.Answer) > 0 {
|
|
return // Only interested in questions.
|
|
}
|
|
if len(req.Question) != 1 {
|
|
return // We only handle single-question messages
|
|
}
|
|
if req.Question[0].Name != success_test_name {
|
|
return // This is not the DNS record you are looking for
|
|
}
|
|
|
|
m := new(dns.Msg)
|
|
m.SetReply(req)
|
|
|
|
hdr := dns.RR_Header{Name: m.Question[0].Name, Rrtype: dns.TypeA,
|
|
Class: dns.ClassINET, Ttl: 3600}
|
|
a := &dns.A{hdr, test_addr}
|
|
m.Answer = append(m.Answer, a)
|
|
|
|
buf, err := m.Pack()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if buf == nil {
|
|
log.Fatal("Nil buffer")
|
|
}
|
|
//log.Println("minimalServer sending:", buf)
|
|
// This is a bit of a kludge - per the RFC we should send responses from 5353, but that doesn't seem to work
|
|
sendconn, err := net.DialUDP("udp", nil, ipv4Addr)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
_, err = sendconn.Write(buf)
|
|
sendconn.Close()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func RunLocalMulticastServer() (*dns.Server, error) {
|
|
multicast, err := net.ListenMulticastUDP("udp", nil, ipv4Addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server := &dns.Server{Listener: nil, PacketConn: multicast, Handler: dns.HandlerFunc(minimalServer)}
|
|
go server.ActivateAndServe()
|
|
return server, nil
|
|
}
|
|
|
|
func TestSimpleQuery(t *testing.T) {
|
|
log.Println("TestSimpleQuery starting")
|
|
mdnsClient, err := NewMDNSClient()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = mdnsClient.Start(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer mdnsClient.Shutdown()
|
|
|
|
server, err := RunLocalMulticastServer()
|
|
if err != nil {
|
|
t.Fatalf("Unable to run test server: %s", err)
|
|
}
|
|
defer server.Shutdown()
|
|
|
|
var received_addr net.IP
|
|
|
|
channel := make(chan *ResponseA, 4)
|
|
|
|
// First, a test we expect to succeed
|
|
mdnsClient.SendQuery(success_test_name, dns.TypeA, channel)
|
|
for resp := range channel {
|
|
if resp.err != nil {
|
|
t.Fatal(resp.err)
|
|
}
|
|
log.Printf("Got address response %s addr %s", resp.Name, resp.Addr)
|
|
received_addr = resp.Addr
|
|
}
|
|
|
|
if !received_addr.Equal(test_addr) {
|
|
t.Log("Unexpected result for", success_test_name, received_addr)
|
|
t.Fail()
|
|
}
|
|
|
|
received_response := false
|
|
// Now, a test we expect to time out with no responses
|
|
channel2 := make(chan *ResponseA, 4)
|
|
mdnsClient.SendQuery("test2.weave.", dns.TypeA, channel2)
|
|
for resp := range channel2 {
|
|
if resp.err != nil {
|
|
t.Fatal(resp.err)
|
|
}
|
|
log.Printf("Got address response %s addr %s", resp.Name, resp.Addr)
|
|
received_addr = resp.Addr
|
|
received_response = true
|
|
}
|
|
|
|
if received_response {
|
|
t.Log("Unexpected result for test2.weave", received_addr)
|
|
t.Fail()
|
|
}
|
|
}
|