87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
|
package upstream
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDnsUpstream(t *testing.T) {
|
||
|
|
|
||
|
|
u, err := NewDnsUpstream("8.8.8.8:53")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("cannot create a DNS upstream")
|
||
|
|
}
|
||
|
|
|
||
|
|
testUpstream(t, u)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHttpsUpstream(t *testing.T) {
|
||
|
|
|
||
|
|
testCases := []string{
|
||
|
|
"https://cloudflare-dns.com/dns-query",
|
||
|
|
"https://dns.google.com/experimental",
|
||
|
|
"https://doh.cleanbrowsing.org/doh/security-filter/",
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, url := range testCases {
|
||
|
|
u, err := NewHttpsUpstream(url)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("cannot create a DNS-over-HTTPS upstream")
|
||
|
|
}
|
||
|
|
|
||
|
|
testUpstream(t, u)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDnsOverTlsUpstream(t *testing.T) {
|
||
|
|
|
||
|
|
var tests = []struct {
|
||
|
|
endpoint string
|
||
|
|
tlsServerName string
|
||
|
|
}{
|
||
|
|
{"1.1.1.1:853", ""},
|
||
|
|
{"8.8.8.8:853", ""},
|
||
|
|
{"185.228.168.10:853", "security-filter-dns.cleanbrowsing.org"},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, test := range tests {
|
||
|
|
u, err := NewDnsOverTlsUpstream(test.endpoint, test.tlsServerName)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("cannot create a DNS-over-TLS upstream")
|
||
|
|
}
|
||
|
|
|
||
|
|
testUpstream(t, u)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func testUpstream(t *testing.T, u Upstream) {
|
||
|
|
req := dns.Msg{}
|
||
|
|
req.Id = dns.Id()
|
||
|
|
req.RecursionDesired = true
|
||
|
|
req.Question = []dns.Question{
|
||
|
|
{Name: "google-public-dns-a.google.com.", Qtype: dns.TypeA, Qclass: dns.ClassINET},
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := u.Exchange(nil, &req)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("error while making an upstream request: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Answer) != 1 {
|
||
|
|
t.Errorf("no answer section in the response")
|
||
|
|
}
|
||
|
|
if answer, ok := resp.Answer[0].(*dns.A); ok {
|
||
|
|
if !net.IPv4(8, 8, 8, 8).Equal(answer.A) {
|
||
|
|
t.Errorf("wrong IP in the response: %v", answer.A)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Printf("response: %v", resp)
|
||
|
|
}
|