// Copyright (c) 2021 Tailscale Inc & AUTHORS All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. package resolver import ( "fmt" "net" "reflect" "strings" "testing" "time" dns "golang.org/x/net/dns/dnsmessage" "tailscale.com/hostinfo" "tailscale.com/types/dnstype" ) func (rr resolverAndDelay) String() string { return fmt.Sprintf("%v+%v", rr.name, rr.startDelay) } func TestResolversWithDelays(t *testing.T) { // query q := func(ss ...string) (ipps []dnstype.Resolver) { for _, s := range ss { host, _, err := net.SplitHostPort(s) if err != nil { t.Fatal(err) } ipps = append(ipps, dnstype.Resolver{Addr: host}) } return } // output o := func(ss ...string) (rr []resolverAndDelay) { for _, s := range ss { var d time.Duration if i := strings.Index(s, "+"); i != -1 { var err error d, err = time.ParseDuration(s[i+1:]) if err != nil { panic(fmt.Sprintf("parsing duration in %q: %v", s, err)) } s = s[:i] } host, _, err := net.SplitHostPort(s) if err != nil { t.Fatal(err) } rr = append(rr, resolverAndDelay{ name: dnstype.Resolver{Addr: host}, startDelay: d, }) } return } tests := []struct { name string in []dnstype.Resolver want []resolverAndDelay }{ { name: "unknown-no-delays", in: q("1.2.3.4:53", "2.3.4.5:53"), want: o("1.2.3.4:53", "2.3.4.5:53"), }, { name: "google-all-ipv4", in: q("8.8.8.8:53", "8.8.4.4:53"), want: o("8.8.8.8:53", "8.8.4.4:53+200ms"), }, { name: "google-only-ipv6", in: q("[2001:4860:4860::8888]:53", "[2001:4860:4860::8844]:53"), want: o("[2001:4860:4860::8888]:53", "[2001:4860:4860::8844]:53+200ms"), }, { name: "google-all-four", in: q("8.8.8.8:53", "8.8.4.4:53", "[2001:4860:4860::8888]:53", "[2001:4860:4860::8844]:53"), want: o("8.8.8.8:53", "8.8.4.4:53+200ms", "[2001:4860:4860::8888]:53+2.5s", "[2001:4860:4860::8844]:53+2.7s"), }, { name: "quad9-one-v4-one-v6", in: q("9.9.9.9:53", "[2620:fe::fe]:53"), want: o("9.9.9.9:53", "[2620:fe::fe]:53+200ms"), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got := resolversWithDelays(tt.in) if !reflect.DeepEqual(got, tt.want) { t.Errorf("got %v; want %v", got, tt.want) } }) } } func TestGetRCode(t *testing.T) { tests := []struct { name string packet []byte want dns.RCode }{ { name: "empty", packet: []byte{}, want: dns.RCode(5), }, { name: "too-short", packet: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, want: dns.RCode(5), }, { name: "noerror", packet: []byte{0xC4, 0xFE, 0x81, 0xA0, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01}, want: dns.RCode(0), }, { name: "refused", packet: []byte{0xee, 0xa1, 0x81, 0x05, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01}, want: dns.RCode(5), }, { name: "nxdomain", packet: []byte{0x34, 0xf4, 0x81, 0x83, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x01}, want: dns.RCode(3), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got := getRCode(tt.packet) if got != tt.want { t.Errorf("got %d; want %d", got, tt.want) } }) } } func TestMaxDoHInFlight(t *testing.T) { tests := []struct { goos string ver string want int }{ {"ios", "", 10}, {"ios", "1532", 10}, {"ios", "9.3.2", 10}, {"ios", "14.3.2", 10}, {"ios", "15.3.2", 1000}, {"ios", "20.3.2", 1000}, {"android", "", 1000}, {"darwin", "", 1000}, {"linux", "", 1000}, } for _, tc := range tests { t.Run(fmt.Sprintf("%s-%s", tc.goos, tc.ver), func(t *testing.T) { hostinfo.SetOSVersion(tc.ver) got := maxDoHInFlight(tc.goos) if got != tc.want { t.Errorf("got %d; want %d", got, tc.want) } }) } }