2014-01-02 21:12:05 +00:00
|
|
|
package agent
|
|
|
|
|
|
|
|
import (
|
2014-02-14 20:26:51 +00:00
|
|
|
"fmt"
|
2016-03-30 06:08:40 +00:00
|
|
|
"math/rand"
|
2015-04-13 20:22:45 +00:00
|
|
|
"net"
|
2016-08-12 04:46:14 +00:00
|
|
|
"reflect"
|
2014-02-14 20:26:51 +00:00
|
|
|
"strings"
|
2014-01-02 21:12:05 +00:00
|
|
|
"testing"
|
2014-06-08 23:14:21 +00:00
|
|
|
"time"
|
2014-09-02 19:47:40 +00:00
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
"github.com/hashicorp/consul/agent/config"
|
2017-07-06 10:34:00 +00:00
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
2017-04-19 23:00:11 +00:00
|
|
|
"github.com/hashicorp/consul/api"
|
2016-03-30 18:48:32 +00:00
|
|
|
"github.com/hashicorp/consul/lib"
|
2018-08-06 23:46:09 +00:00
|
|
|
"github.com/hashicorp/consul/testrpc"
|
2017-04-29 16:34:02 +00:00
|
|
|
"github.com/hashicorp/consul/testutil/retry"
|
2018-04-13 16:57:25 +00:00
|
|
|
"github.com/hashicorp/serf/coordinate"
|
2014-09-02 19:47:40 +00:00
|
|
|
"github.com/miekg/dns"
|
2017-05-29 14:08:54 +00:00
|
|
|
"github.com/pascaldekloe/goe/verify"
|
2018-03-09 17:09:21 +00:00
|
|
|
"github.com/stretchr/testify/assert"
|
2018-04-13 16:57:25 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
2014-01-02 21:12:05 +00:00
|
|
|
)
|
|
|
|
|
2016-02-12 07:58:48 +00:00
|
|
|
const (
|
2016-03-30 06:08:40 +00:00
|
|
|
configUDPAnswerLimit = 4
|
|
|
|
defaultNumUDPResponses = 3
|
2016-02-12 07:58:48 +00:00
|
|
|
testUDPTruncateLimit = 8
|
2016-03-30 06:08:40 +00:00
|
|
|
|
|
|
|
pctNodesWithIPv6 = 0.5
|
|
|
|
|
|
|
|
// generateNumNodes is the upper bounds for the number of hosts used
|
|
|
|
// in testing below. Generate an arbitrarily large number of hosts.
|
|
|
|
generateNumNodes = testUDPTruncateLimit * defaultNumUDPResponses * configUDPAnswerLimit
|
2016-02-12 07:58:48 +00:00
|
|
|
)
|
|
|
|
|
2015-04-14 02:19:22 +00:00
|
|
|
// makeRecursor creates a generic DNS server which always returns
|
|
|
|
// the provided reply. This is useful for mocking a DNS recursor with
|
|
|
|
// an expected result.
|
2017-06-25 17:42:37 +00:00
|
|
|
func makeRecursor(t *testing.T, answer dns.Msg) *dns.Server {
|
2018-08-02 14:12:52 +00:00
|
|
|
a := answer
|
2016-11-03 19:21:16 +00:00
|
|
|
mux := dns.NewServeMux()
|
|
|
|
mux.HandleFunc(".", func(resp dns.ResponseWriter, msg *dns.Msg) {
|
2018-08-02 14:12:52 +00:00
|
|
|
// The SetReply function sets the return code of the DNS
|
|
|
|
// query to SUCCESS
|
|
|
|
// We need a way to copy the variables not addressed
|
|
|
|
// in SetReply
|
2016-11-03 19:21:16 +00:00
|
|
|
answer.SetReply(msg)
|
2018-08-02 14:12:52 +00:00
|
|
|
answer.Rcode = a.Rcode
|
2016-11-03 19:21:16 +00:00
|
|
|
if err := resp.WriteMsg(&answer); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
})
|
2017-06-25 17:42:37 +00:00
|
|
|
up := make(chan struct{})
|
2016-11-03 19:21:16 +00:00
|
|
|
server := &dns.Server{
|
2017-06-25 17:42:37 +00:00
|
|
|
Addr: "127.0.0.1:0",
|
|
|
|
Net: "udp",
|
|
|
|
Handler: mux,
|
|
|
|
NotifyStartedFunc: func() { close(up) },
|
2016-11-03 19:21:16 +00:00
|
|
|
}
|
|
|
|
go server.ListenAndServe()
|
2017-06-25 17:42:37 +00:00
|
|
|
<-up
|
|
|
|
server.Addr = server.PacketConn.LocalAddr().String()
|
2016-11-03 19:21:16 +00:00
|
|
|
return server
|
|
|
|
}
|
|
|
|
|
2015-04-14 02:19:22 +00:00
|
|
|
// dnsCNAME returns a DNS CNAME record struct
|
|
|
|
func dnsCNAME(src, dest string) *dns.CNAME {
|
|
|
|
return &dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: dns.Fqdn(src),
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: dns.Fqdn(dest),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// dnsA returns a DNS A record struct
|
|
|
|
func dnsA(src, dest string) *dns.A {
|
|
|
|
return &dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: dns.Fqdn(src),
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP(dest),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-01 04:37:39 +00:00
|
|
|
// dnsTXT returns a DNS TXT record struct
|
|
|
|
func dnsTXT(src string, txt []string) *dns.TXT {
|
|
|
|
return &dns.TXT{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: dns.Fqdn(src),
|
|
|
|
Rrtype: dns.TypeTXT,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Txt: txt,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-23 01:31:11 +00:00
|
|
|
func TestRecursorAddr(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2014-02-23 01:31:11 +00:00
|
|
|
addr, err := recursorAddr("8.8.8.8")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
if addr != "8.8.8.8:53" {
|
|
|
|
t.Fatalf("bad: %v", addr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-11 01:36:50 +00:00
|
|
|
func TestEncodeKVasRFC1464(t *testing.T) {
|
|
|
|
// Test cases are from rfc1464
|
|
|
|
type rfc1464Test struct {
|
|
|
|
key, value, internalForm, externalForm string
|
|
|
|
}
|
|
|
|
tests := []rfc1464Test{
|
|
|
|
{"color", "blue", "color=blue", "color=blue"},
|
|
|
|
{"equation", "a=4", "equation=a=4", "equation=a=4"},
|
|
|
|
{"a=a", "true", "a`=a=true", "a`=a=true"},
|
|
|
|
{"a\\=a", "false", "a\\`=a=false", "a\\`=a=false"},
|
|
|
|
{"=", "\\=", "`==\\=", "`==\\="},
|
|
|
|
|
|
|
|
{"string", "\"Cat\"", "string=\"Cat\"", "string=\"Cat\""},
|
|
|
|
{"string2", "`abc`", "string2=``abc``", "string2=``abc``"},
|
|
|
|
{"novalue", "", "novalue=", "novalue="},
|
|
|
|
{"a b", "c d", "a b=c d", "a b=c d"},
|
|
|
|
{"abc ", "123 ", "abc` =123 ", "abc` =123 "},
|
|
|
|
|
|
|
|
// Additional tests
|
|
|
|
{" abc", " 321", "` abc= 321", "` abc= 321"},
|
|
|
|
{"`a", "b", "``a=b", "``a=b"},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, test := range tests {
|
|
|
|
answer := encodeKVasRFC1464(test.key, test.value)
|
|
|
|
verify.Values(t, "internalForm", answer, test.internalForm)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-09 01:57:41 +00:00
|
|
|
func TestDNS_Over_TCP(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-02-09 01:57:41 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "Foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.node.dc1.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
c.Net = "tcp"
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("empty lookup: %#v", in)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-03 01:58:58 +00:00
|
|
|
func TestDNS_NodeLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2014-01-03 01:58:58 +00:00
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
2016-02-07 18:37:34 +00:00
|
|
|
TaggedAddresses: map[string]string{
|
2016-01-26 18:30:28 +00:00
|
|
|
"wan": "127.0.0.2",
|
|
|
|
},
|
2018-07-09 15:41:58 +00:00
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
2014-01-03 01:58:58 +00:00
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2014-01-03 01:58:58 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-01-03 01:58:58 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2018-07-09 15:41:58 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, in.Answer, 2)
|
|
|
|
require.Len(t, in.Extra, 0)
|
2014-01-03 01:58:58 +00:00
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
2018-07-09 15:41:58 +00:00
|
|
|
require.True(t, ok, "First answer is not an A record")
|
|
|
|
require.Equal(t, "127.0.0.1", aRec.A.String())
|
|
|
|
require.Equal(t, uint32(0), aRec.Hdr.Ttl)
|
|
|
|
|
|
|
|
txt, ok := in.Answer[1].(*dns.TXT)
|
|
|
|
require.True(t, ok, "Second answer is not a TXT record")
|
|
|
|
require.Len(t, txt.Txt, 1)
|
|
|
|
require.Equal(t, "key=value", txt.Txt[0])
|
|
|
|
|
|
|
|
// Re-do the query, but only for an A RR
|
|
|
|
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.node.consul.", dns.TypeA)
|
|
|
|
|
|
|
|
c = new(dns.Client)
|
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, in.Answer, 1)
|
|
|
|
require.Len(t, in.Extra, 1)
|
|
|
|
|
|
|
|
aRec, ok = in.Answer[0].(*dns.A)
|
|
|
|
require.True(t, ok, "Answer is not an A record")
|
|
|
|
require.Equal(t, "127.0.0.1", aRec.A.String())
|
|
|
|
require.Equal(t, uint32(0), aRec.Hdr.Ttl)
|
|
|
|
|
|
|
|
txt, ok = in.Extra[0].(*dns.TXT)
|
|
|
|
require.True(t, ok, "Extra record is not a TXT record")
|
|
|
|
require.Len(t, txt.Txt, 1)
|
|
|
|
require.Equal(t, "key=value", txt.Txt[0])
|
2014-01-03 01:58:58 +00:00
|
|
|
|
|
|
|
// Re-do the query, but specify the DC
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.node.dc1.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c = new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2018-07-09 15:41:58 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, in.Answer, 2)
|
|
|
|
require.Len(t, in.Extra, 0)
|
2014-01-03 01:58:58 +00:00
|
|
|
|
|
|
|
aRec, ok = in.Answer[0].(*dns.A)
|
2018-07-09 15:41:58 +00:00
|
|
|
require.True(t, ok, "First answer is not an A record")
|
|
|
|
require.Equal(t, "127.0.0.1", aRec.A.String())
|
|
|
|
require.Equal(t, uint32(0), aRec.Hdr.Ttl)
|
|
|
|
|
|
|
|
txt, ok = in.Answer[1].(*dns.TXT)
|
|
|
|
require.True(t, ok, "Second answer is not a TXT record")
|
2015-06-02 21:47:18 +00:00
|
|
|
|
|
|
|
// lookup a non-existing node, we should receive a SOA
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("nofoo.node.dc1.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c = new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2018-07-09 15:41:58 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, in.Ns, 1)
|
2015-06-02 21:47:18 +00:00
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
2018-07-09 15:41:58 +00:00
|
|
|
require.True(t, ok, "NS RR is not a SOA record")
|
|
|
|
require.Equal(t, uint32(0), soaRec.Hdr.Ttl)
|
2014-01-03 01:58:58 +00:00
|
|
|
}
|
2014-01-03 21:00:03 +00:00
|
|
|
|
2014-07-17 06:38:24 +00:00
|
|
|
func TestDNS_CaseInsensitiveNodeLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-07-17 06:38:24 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "Foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-07-17 06:38:24 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
2014-07-23 08:29:28 +00:00
|
|
|
m.SetQuestion("fOO.node.dc1.consul.", dns.TypeANY)
|
2014-07-17 06:38:24 +00:00
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-07-17 06:38:24 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("empty lookup: %#v", in)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-21 22:33:01 +00:00
|
|
|
func TestDNS_NodeLookup_PeriodName(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2014-04-21 22:33:01 +00:00
|
|
|
// Register node with period in name
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo.bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2014-04-21 22:33:01 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-04-21 22:33:01 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.bar.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-04-21 22:33:01 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
func TestDNS_NodeLookup_AAAA(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "::4242:4242",
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-02-25 20:07:20 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
2015-09-08 13:54:13 +00:00
|
|
|
m.SetQuestion("bar.node.consul.", dns.TypeAAAA)
|
2014-02-25 20:07:20 +00:00
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-02-25 20:07:20 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.AAAA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.AAAA.String() != "::4242:4242" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2014-06-08 23:14:21 +00:00
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
|
|
|
|
2018-08-02 14:12:52 +00:00
|
|
|
func TestDNSCycleRecursorCheck(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
// Start a DNS recursor that returns a SERVFAIL
|
|
|
|
server1 := makeRecursor(t, dns.Msg{
|
|
|
|
MsgHdr: dns.MsgHdr{Rcode: dns.RcodeServerFailure},
|
|
|
|
})
|
|
|
|
// Start a DNS recursor that returns the result
|
|
|
|
defer server1.Shutdown()
|
|
|
|
server2 := makeRecursor(t, dns.Msg{
|
|
|
|
MsgHdr: dns.MsgHdr{Rcode: dns.RcodeSuccess},
|
|
|
|
Answer: []dns.RR{
|
|
|
|
dnsA("www.google.com", "172.21.45.67"),
|
|
|
|
},
|
|
|
|
})
|
|
|
|
defer server2.Shutdown()
|
|
|
|
//Mock the agent startup with the necessary configs
|
|
|
|
agent := NewTestAgent(t.Name(),
|
|
|
|
`recursors = ["`+server1.Addr+`", "`+server2.Addr+`"]
|
|
|
|
`)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
// DNS Message init
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("google.com.", dns.TypeA)
|
|
|
|
// Agent request
|
|
|
|
client := new(dns.Client)
|
|
|
|
in, _, _ := client.Exchange(m, agent.DNSAddr())
|
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "www.google.com.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0xAC, 0x15, 0x2D, 0x43}, // 172 , 21, 45, 67
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "Answer", in.Answer, wantAnswer)
|
|
|
|
}
|
|
|
|
func TestDNSCycleRecursorCheckAllFail(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
// Start 3 DNS recursors that returns a REFUSED status
|
|
|
|
server1 := makeRecursor(t, dns.Msg{
|
|
|
|
MsgHdr: dns.MsgHdr{Rcode: dns.RcodeRefused},
|
|
|
|
})
|
|
|
|
defer server1.Shutdown()
|
|
|
|
server2 := makeRecursor(t, dns.Msg{
|
|
|
|
MsgHdr: dns.MsgHdr{Rcode: dns.RcodeRefused},
|
|
|
|
})
|
|
|
|
defer server2.Shutdown()
|
|
|
|
server3 := makeRecursor(t, dns.Msg{
|
|
|
|
MsgHdr: dns.MsgHdr{Rcode: dns.RcodeRefused},
|
|
|
|
})
|
|
|
|
defer server3.Shutdown()
|
|
|
|
//Mock the agent startup with the necessary configs
|
|
|
|
agent := NewTestAgent(t.Name(),
|
|
|
|
`recursors = ["`+server1.Addr+`", "`+server2.Addr+`","`+server3.Addr+`"]
|
|
|
|
`)
|
|
|
|
defer agent.Shutdown()
|
|
|
|
// DNS dummy message initialization
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("google.com.", dns.TypeA)
|
|
|
|
// Agent request
|
|
|
|
client := new(dns.Client)
|
|
|
|
in, _, _ := client.Exchange(m, agent.DNSAddr())
|
|
|
|
//Verify if we hit SERVFAIL from Consul
|
|
|
|
verify.Values(t, "Answer", in.Rcode, dns.RcodeServerFailure)
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
func TestDNS_NodeLookup_CNAME(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
dnsCNAME("www.google.com", "google.com"),
|
|
|
|
dnsA("google.com", "1.2.3.4"),
|
2017-08-01 04:37:39 +00:00
|
|
|
dnsTXT("google.com", []string{"my_txt_value"}),
|
2017-06-25 17:42:37 +00:00
|
|
|
},
|
2015-04-14 02:19:22 +00:00
|
|
|
})
|
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "google",
|
|
|
|
Address: "www.google.com",
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-02-25 20:07:20 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("google.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-02-25 20:07:20 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-10 06:02:39 +00:00
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{Name: "google.node.consul.", Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Ttl: 0, Rdlength: 0x10},
|
|
|
|
Target: "www.google.com.",
|
|
|
|
},
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{Name: "www.google.com.", Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Rdlength: 0x2},
|
|
|
|
Target: "google.com.",
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "google.com.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x1, 0x2, 0x3, 0x4}, // 1.2.3.4
|
|
|
|
},
|
|
|
|
&dns.TXT{
|
|
|
|
Hdr: dns.RR_Header{Name: "google.com.", Rrtype: dns.TypeTXT, Class: dns.ClassINET, Rdlength: 0xd},
|
|
|
|
Txt: []string{"my_txt_value"},
|
|
|
|
},
|
2014-06-08 23:14:21 +00:00
|
|
|
}
|
2017-08-10 06:02:39 +00:00
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
|
|
|
|
2017-08-01 04:37:39 +00:00
|
|
|
func TestDNS_NodeLookup_TXT(t *testing.T) {
|
2017-09-26 12:39:47 +00:00
|
|
|
a := NewTestAgent(t.Name(), ``)
|
2017-08-01 04:37:39 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-08-01 04:37:39 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "google",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"rfc1035-00": "value0",
|
|
|
|
"key0": "value1",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("google.node.consul.", dns.TypeTXT)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-26 12:39:47 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-08-01 04:37:39 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should have the 1 TXT record reply
|
|
|
|
if len(in.Answer) != 2 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
txtRec, ok := in.Answer[0].(*dns.TXT)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if len(txtRec.Txt) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if txtRec.Txt[0] != "value0" && txtRec.Txt[0] != "key0=value1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-11 15:49:04 +00:00
|
|
|
func TestDNS_NodeLookup_TXT_DontSuppress(t *testing.T) {
|
2018-06-19 14:08:16 +00:00
|
|
|
a := NewTestAgent(t.Name(), `dns_config = { enable_additional_node_meta_txt = false }`)
|
2018-06-11 15:49:04 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-06-11 15:49:04 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "google",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"rfc1035-00": "value0",
|
|
|
|
"key0": "value1",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("google.node.consul.", dns.TypeTXT)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should have the 1 TXT record reply
|
|
|
|
if len(in.Answer) != 2 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
txtRec, ok := in.Answer[0].(*dns.TXT)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if len(txtRec.Txt) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if txtRec.Txt[0] != "value0" && txtRec.Txt[0] != "key0=value1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-03 06:56:24 +00:00
|
|
|
func TestDNS_NodeLookup_ANY(t *testing.T) {
|
2017-09-26 12:39:47 +00:00
|
|
|
a := NewTestAgent(t.Name(), ``)
|
2017-08-03 06:56:24 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-08-03 06:56:24 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("bar.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-26 12:39:47 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-08-03 06:56:24 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-10 06:02:39 +00:00
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
|
|
|
&dns.TXT{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.consul.", Rrtype: dns.TypeTXT, Class: dns.ClassINET, Rdlength: 0xa},
|
|
|
|
Txt: []string{"key=value"},
|
|
|
|
},
|
2017-08-03 06:56:24 +00:00
|
|
|
}
|
2017-08-10 06:02:39 +00:00
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
2017-08-03 06:56:24 +00:00
|
|
|
}
|
|
|
|
|
2018-06-19 14:08:16 +00:00
|
|
|
func TestDNS_NodeLookup_ANY_DontSuppressTXT(t *testing.T) {
|
|
|
|
a := NewTestAgent(t.Name(), `dns_config = { enable_additional_node_meta_txt = false }`)
|
2018-06-11 15:49:04 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-06-11 15:49:04 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("bar.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
2018-06-19 14:08:16 +00:00
|
|
|
&dns.TXT{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.consul.", Rrtype: dns.TypeTXT, Class: dns.ClassINET, Rdlength: 0xa},
|
|
|
|
Txt: []string{"key=value"},
|
|
|
|
},
|
2018-06-11 15:49:04 +00:00
|
|
|
}
|
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
|
|
|
}
|
|
|
|
|
2018-07-09 15:41:58 +00:00
|
|
|
func TestDNS_NodeLookup_A_SuppressTXT(t *testing.T) {
|
|
|
|
a := NewTestAgent(t.Name(), `dns_config = { enable_additional_node_meta_txt = false }`)
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-07-09 15:41:58 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
require.NoError(t, a.RPC("Catalog.Register", args, &out))
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("bar.node.consul.", dns.TypeA)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
|
|
|
|
|
|
|
// ensure TXT RR suppression
|
|
|
|
require.Len(t, in.Extra, 0)
|
|
|
|
}
|
|
|
|
|
2017-06-14 23:22:54 +00:00
|
|
|
func TestDNS_EDNS0(t *testing.T) {
|
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-06-14 23:22:54 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-06-14 23:22:54 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetEdns0(12345, true)
|
|
|
|
m.SetQuestion("foo.node.dc1.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-06-14 23:22:54 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("empty lookup: %#v", in)
|
|
|
|
}
|
|
|
|
edns := in.IsEdns0()
|
|
|
|
if edns == nil {
|
|
|
|
t.Fatalf("empty edns: %#v", in)
|
|
|
|
}
|
|
|
|
if edns.UDPSize() != 12345 {
|
|
|
|
t.Fatalf("bad edns size: %d", edns.UDPSize())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-23 08:16:37 +00:00
|
|
|
func TestDNS_ReverseLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-11-23 08:16:37 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo2",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-11-23 08:16:37 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-11-23 08:16:37 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "foo2.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-08 18:24:49 +00:00
|
|
|
func TestDNS_ReverseLookup_CustomDomain(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
domain = "custom"
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-01-08 18:24:49 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo2",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-01-08 18:24:49 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-01-08 18:24:49 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "foo2.node.dc1.custom." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-23 08:16:37 +00:00
|
|
|
func TestDNS_ReverseLookup_IPV6(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-11-23 08:16:37 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "::4242:4242",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-11-23 08:16:37 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.4.2.4.2.4.2.4.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-11-23 08:16:37 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "bar.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
2015-01-08 19:58:32 +00:00
|
|
|
|
2018-05-21 20:59:21 +00:00
|
|
|
func TestDNS_ServiceReverseLookup(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-05-21 20:59:21 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "db.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceReverseLookup_IPV6(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-05-21 20:59:21 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "2001:db8::1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
Address: "2001:db8::ff00:42:8329",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("9.2.3.8.2.4.0.0.0.0.f.f.0.0.0.0.0.0.0.0.0.0.0.0.8.b.d.0.1.0.0.2.ip6.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "db.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceReverseLookup_CustomDomain(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
domain = "custom"
|
|
|
|
`)
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-05-21 20:59:21 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "db.service.custom." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceReverseLookupNodeAddress(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-05-21 20:59:21 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("1.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ptrRec, ok := in.Answer[0].(*dns.PTR)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if ptrRec.Ptr != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", ptrRec)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-02 20:58:52 +00:00
|
|
|
func TestDNS_ServiceLookupNoMultiCNAME(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-07-02 20:58:52 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "198.18.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "foo.node.consul",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
require.NoError(t, a.RPC("Catalog.Register", args, &out))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register a second node node with the same service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "198.18.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "bar.node.consul",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
// expect a CNAME and an A RR
|
|
|
|
require.Len(t, in.Answer, 2)
|
|
|
|
require.IsType(t, &dns.CNAME{}, in.Answer[0])
|
|
|
|
require.IsType(t, &dns.A{}, in.Answer[1])
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookupPreferNoCNAME(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-07-02 20:58:52 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "198.18.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "198.18.0.1",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
require.NoError(t, a.RPC("Catalog.Register", args, &out))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register a second node node with the same service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "198.18.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "bar.node.consul",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
// expect a CNAME and an A RR
|
|
|
|
require.Len(t, in.Answer, 1)
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
require.Truef(t, ok, "Not an A RR")
|
|
|
|
|
|
|
|
require.Equal(t, "db.service.consul.", aRec.Hdr.Name)
|
|
|
|
require.Equal(t, "198.18.0.1", aRec.A.String())
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookupMultiAddrNoCNAME(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-07-02 20:58:52 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "198.18.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "198.18.0.1",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
require.NoError(t, a.RPC("Catalog.Register", args, &out))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register a second node node with the same service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "198.18.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "bar.node.consul",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register a second node node with the same service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "baz",
|
|
|
|
Address: "198.18.0.3",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
Address: "198.18.0.3",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
// expect a CNAME and an A RR
|
|
|
|
require.Len(t, in.Answer, 2)
|
|
|
|
require.IsType(t, &dns.A{}, in.Answer[0])
|
|
|
|
require.IsType(t, &dns.A{}, in.Answer[1])
|
|
|
|
}
|
|
|
|
|
2014-01-03 21:00:03 +00:00
|
|
|
func TestDNS_ServiceLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-03 21:00:03 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-03 21:00:03 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-01-03 21:00:03 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
2014-01-03 21:00:03 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Lookup a non-existing service/query, we should receive an SOA.
|
|
|
|
questions = []string{
|
|
|
|
"nodb.service.consul.",
|
|
|
|
"nope.query.consul.",
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
|
2014-06-08 23:14:21 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
|
2017-05-29 14:08:54 +00:00
|
|
|
func TestDNS_ServiceLookupWithInternalServiceAddress(t *testing.T) {
|
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
node_name = "my.test-node"
|
|
|
|
`)
|
2017-05-29 14:08:54 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-05-29 14:08:54 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
// The service is using the consul DNS name as service address
|
|
|
|
// which triggers a lookup loop and a subsequent stack overflow
|
|
|
|
// crash.
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Address: "db.service.consul",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Looking up the service should not trigger a loop
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-05-29 14:08:54 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.SRV{
|
2017-10-06 22:40:17 +00:00
|
|
|
Hdr: dns.RR_Header{Name: "db.service.consul.", Rrtype: 0x21, Class: 0x1, Rdlength: 0x1b},
|
2017-05-29 14:08:54 +00:00
|
|
|
Priority: 0x1,
|
|
|
|
Weight: 0x1,
|
|
|
|
Port: 12345,
|
|
|
|
Target: "foo.node.dc1.consul.",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
|
|
|
wantExtra := []dns.RR{
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{Name: "foo.node.dc1.consul.", Rrtype: 0x5, Class: 0x1, Rdlength: 0x2},
|
|
|
|
Target: "db.service.consul.",
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "db.service.consul.", Rrtype: 0x1, Class: 0x1, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "extra", in.Extra, wantExtra)
|
|
|
|
}
|
|
|
|
|
2018-03-09 17:09:21 +00:00
|
|
|
func TestDNS_ConnectServiceLookup(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
assert := assert.New(t)
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-03-09 17:09:21 +00:00
|
|
|
|
2018-03-12 17:13:44 +00:00
|
|
|
// Register
|
2018-03-09 17:09:21 +00:00
|
|
|
{
|
|
|
|
args := structs.TestRegisterRequestProxy(t)
|
2018-03-22 02:54:44 +00:00
|
|
|
args.Address = "127.0.0.55"
|
2018-03-09 17:09:21 +00:00
|
|
|
args.Service.ProxyDestination = "db"
|
2018-03-09 17:32:22 +00:00
|
|
|
args.Service.Address = ""
|
2018-03-09 17:09:21 +00:00
|
|
|
args.Service.Port = 12345
|
|
|
|
var out struct{}
|
|
|
|
assert.Nil(a.RPC("Catalog.Register", args, &out))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service
|
|
|
|
questions := []string{
|
|
|
|
"db.connect.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
assert.Nil(err)
|
|
|
|
assert.Len(in.Answer, 1)
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
assert.True(ok)
|
2018-03-09 17:32:22 +00:00
|
|
|
assert.Equal(uint16(12345), srvRec.Port)
|
2018-03-09 17:09:21 +00:00
|
|
|
assert.Equal("foo.node.dc1.consul.", srvRec.Target)
|
2018-03-09 17:32:22 +00:00
|
|
|
assert.Equal(uint32(0), srvRec.Hdr.Ttl)
|
2018-03-09 17:09:21 +00:00
|
|
|
|
2018-03-09 17:32:22 +00:00
|
|
|
cnameRec, ok := in.Extra[0].(*dns.A)
|
2018-03-09 17:09:21 +00:00
|
|
|
assert.True(ok)
|
|
|
|
assert.Equal("foo.node.dc1.consul.", cnameRec.Hdr.Name)
|
2018-03-09 17:32:22 +00:00
|
|
|
assert.Equal(uint32(0), srvRec.Hdr.Ttl)
|
2018-03-22 02:54:44 +00:00
|
|
|
assert.Equal("127.0.0.55", cnameRec.A.String())
|
2018-03-09 17:09:21 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-10-27 02:23:51 +00:00
|
|
|
func TestDNS_ExternalServiceLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-27 02:23:51 +00:00
|
|
|
|
|
|
|
// Register a node with an external service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "www.google.com",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-27 02:23:51 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
cnameRec, ok := in.Extra[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Target != "www.google.com." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ExternalServiceToConsulCNAMELookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
domain = "CONSUL."
|
|
|
|
node_name = "test node"
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-27 02:23:51 +00:00
|
|
|
|
|
|
|
// Register the initial node with a service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "web",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "web",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an external service pointing to the 'web' service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "alias",
|
|
|
|
Address: "web.service.consul",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "alias",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly
|
|
|
|
questions := []string{
|
|
|
|
"alias.service.consul.",
|
|
|
|
"alias.service.CoNsUl.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-27 02:23:51 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "alias.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Extra) != 2 {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
cnameRec, ok := in.Extra[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Name != "alias.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Target != "web.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[1].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "web.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
|
2016-10-27 02:23:51 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
func TestDNS_NSRecords(t *testing.T) {
|
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
domain = "CONSUL."
|
|
|
|
node_name = "server1"
|
|
|
|
`)
|
2017-08-04 17:17:04 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-08-04 17:17:04 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("something.node.consul.", dns.TypeNS)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-08-04 17:17:04 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-06 23:18:30 +00:00
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.NS{
|
|
|
|
Hdr: dns.RR_Header{Name: "consul.", Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 0, Rdlength: 0x13},
|
|
|
|
Ns: "server1.node.dc1.consul.",
|
|
|
|
},
|
2017-08-04 17:17:04 +00:00
|
|
|
}
|
2017-08-06 23:18:30 +00:00
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
|
|
|
wantExtra := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "server1.node.dc1.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4, Ttl: 0},
|
|
|
|
A: net.ParseIP("127.0.0.1").To4(),
|
|
|
|
},
|
2017-08-04 17:17:04 +00:00
|
|
|
}
|
2017-08-06 23:18:30 +00:00
|
|
|
|
|
|
|
verify.Values(t, "extra", in.Extra, wantExtra)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_NSRecords_IPV6(t *testing.T) {
|
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
domain = "CONSUL."
|
|
|
|
node_name = "server1"
|
|
|
|
advertise_addr = "::1"
|
|
|
|
`)
|
2017-08-06 23:18:30 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-08-06 23:18:30 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("server1.node.dc1.consul.", dns.TypeNS)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-08-06 23:18:30 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
2017-08-04 21:53:42 +00:00
|
|
|
}
|
2017-08-06 23:18:30 +00:00
|
|
|
|
|
|
|
wantAnswer := []dns.RR{
|
|
|
|
&dns.NS{
|
|
|
|
Hdr: dns.RR_Header{Name: "consul.", Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 0, Rdlength: 0x2},
|
|
|
|
Ns: "server1.node.dc1.consul.",
|
|
|
|
},
|
2017-08-04 21:53:42 +00:00
|
|
|
}
|
2017-08-06 23:18:30 +00:00
|
|
|
verify.Values(t, "answer", in.Answer, wantAnswer)
|
|
|
|
wantExtra := []dns.RR{
|
|
|
|
&dns.AAAA{
|
|
|
|
Hdr: dns.RR_Header{Name: "server1.node.dc1.consul.", Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Rdlength: 0x10, Ttl: 0},
|
|
|
|
AAAA: net.ParseIP("::1"),
|
|
|
|
},
|
2017-08-04 21:53:42 +00:00
|
|
|
}
|
|
|
|
|
2017-08-06 23:18:30 +00:00
|
|
|
verify.Values(t, "extra", in.Extra, wantExtra)
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
}
|
|
|
|
|
2016-10-27 02:23:51 +00:00
|
|
|
func TestDNS_ExternalServiceToConsulCNAMENestedLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
node_name = "test-node"
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-27 02:23:51 +00:00
|
|
|
|
|
|
|
// Register the initial node with a service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "web",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "web",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an external service pointing to the 'web' service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "alias",
|
|
|
|
Address: "web.service.consul",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "alias",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an external service pointing to the 'alias' service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "alias2",
|
|
|
|
Address: "alias.service.consul",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "alias2",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly
|
|
|
|
questions := []string{
|
|
|
|
"alias2.service.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-27 02:23:51 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "alias2.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Extra) != 3 {
|
2016-10-27 02:23:51 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
cnameRec, ok := in.Extra[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Name != "alias2.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Target != "alias.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
cnameRec, ok = in.Extra[1].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Name != "alias.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if cnameRec.Target != "web.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[2].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[2])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "web.service.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[1])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[2])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[2])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-02-01 03:33:41 +00:00
|
|
|
func TestDNS_ServiceLookup_ServiceAddress_A(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
2016-10-28 02:14:02 +00:00
|
|
|
if srvRec.Target != "7f000002.addr.dc1.consul." {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2016-10-28 02:14:02 +00:00
|
|
|
if aRec.Hdr.Name != "7f000002.addr.dc1.consul." {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.2" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2014-06-08 23:14:21 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
|
2017-02-01 03:33:41 +00:00
|
|
|
func TestDNS_ServiceLookup_ServiceAddress_CNAME(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-02-01 03:33:41 +00:00
|
|
|
|
|
|
|
// Register a node with a service whose address isn't an IP.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Address: "www.google.com",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2017-02-01 03:33:41 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "test",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2017-02-01 03:33:41 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-02-01 03:33:41 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
cnameRec, ok := in.Extra[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Target != "www.google.com." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if cnameRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-10-28 03:41:24 +00:00
|
|
|
func TestDNS_ServiceLookup_ServiceAddressIPV6(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-28 03:08:53 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Address: "2607:20:4005:808::200e",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-10-28 03:08:53 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2016-10-28 03:08:53 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2016-10-28 03:08:53 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-28 03:08:53 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "2607002040050808000000000000200e.addr.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.AAAA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "2607002040050808000000000000200e.addr.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.AAAA.String() != "2607:20:4005:808::200e" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-24 00:07:27 +00:00
|
|
|
func TestDNS_ServiceLookup_WanAddress(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a1 := NewTestAgent(t.Name(), `
|
|
|
|
datacenter = "dc1"
|
|
|
|
translate_wan_addrs = true
|
|
|
|
acl_datacenter = ""
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a1.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a2 := NewTestAgent(t.Name(), `
|
|
|
|
datacenter = "dc2"
|
|
|
|
translate_wan_addrs = true
|
|
|
|
acl_datacenter = ""
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a2.Shutdown()
|
2015-12-24 00:07:27 +00:00
|
|
|
|
|
|
|
// Join WAN cluster
|
2017-09-25 18:40:42 +00:00
|
|
|
addr := fmt.Sprintf("127.0.0.1:%d", a1.Config.SerfPortWAN)
|
2017-05-21 07:11:09 +00:00
|
|
|
if _, err := a2.JoinWAN([]string{addr}); err != nil {
|
2015-12-24 00:07:27 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2017-05-04 22:52:53 +00:00
|
|
|
retry.Run(t, func(r *retry.R) {
|
2017-05-21 07:11:09 +00:00
|
|
|
if got, want := len(a1.WANMembers()), 2; got < want {
|
2017-04-29 16:34:02 +00:00
|
|
|
r.Fatalf("got %d WAN members want at least %d", got, want)
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if got, want := len(a2.WANMembers()), 2; got < want {
|
2017-05-09 04:57:06 +00:00
|
|
|
r.Fatalf("got %d WAN members want at least %d", got, want)
|
|
|
|
}
|
2017-04-29 16:34:02 +00:00
|
|
|
})
|
2015-12-24 00:07:27 +00:00
|
|
|
|
2017-05-09 04:57:06 +00:00
|
|
|
// Register a remote node with a service. This is in a retry since we
|
|
|
|
// need the datacenter to have a route which takes a little more time
|
|
|
|
// beyond the join, and we don't have direct access to the router here.
|
|
|
|
retry.Run(t, func(r *retry.R) {
|
2015-12-24 00:07:27 +00:00
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc2",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
2016-02-07 18:37:34 +00:00
|
|
|
TaggedAddresses: map[string]string{
|
2016-01-26 18:30:28 +00:00
|
|
|
"wan": "127.0.0.2",
|
|
|
|
},
|
2015-12-24 00:07:27 +00:00
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a2.RPC("Catalog.Register", args, &out); err != nil {
|
2017-05-09 04:57:06 +00:00
|
|
|
r.Fatalf("err: %v", err)
|
2015-12-24 00:07:27 +00:00
|
|
|
}
|
2017-05-09 04:57:06 +00:00
|
|
|
})
|
2015-12-24 00:07:27 +00:00
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc2",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-12-24 00:07:27 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a2.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-12-24 00:07:27 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the SRV record via service and prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.dc2.consul.",
|
|
|
|
id + ".query.dc2.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
|
|
|
|
addr := a1.config.DNSAddrs[0]
|
2015-12-24 00:07:27 +00:00
|
|
|
in, _, err := c.Exchange(m, addr.String())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2016-10-28 02:14:02 +00:00
|
|
|
if aRec.Hdr.Name != "7f000002.addr.dc2.consul." {
|
2015-12-24 00:07:27 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.2" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Also check the A record directly
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeA)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
addr := a1.config.DNSAddrs[0]
|
2015-12-24 00:07:27 +00:00
|
|
|
in, _, err := c.Exchange(m, addr.String())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != question {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.2" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
2016-02-07 22:16:15 +00:00
|
|
|
|
|
|
|
// Now query from the same DC and make sure we get the local address
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
addr := a2.Config.DNSAddrs[0]
|
2016-02-07 22:16:15 +00:00
|
|
|
in, _, err := c.Exchange(m, addr.String())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc2.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Also check the A record directly from DC2
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeA)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
addr := a2.Config.DNSAddrs[0]
|
2016-02-07 22:16:15 +00:00
|
|
|
in, _, err := c.Exchange(m, addr.String())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != question {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
2015-12-24 00:07:27 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
func TestDNS_CaseInsensitiveServiceLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "Db",
|
|
|
|
Tags: []string{"Master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query, as well as a name.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "somequery",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Try some variations to make sure case doesn't matter.
|
|
|
|
questions := []string{
|
|
|
|
"master.db.service.consul.",
|
|
|
|
"mASTER.dB.service.consul.",
|
|
|
|
"MASTER.dB.service.consul.",
|
|
|
|
"db.service.consul.",
|
|
|
|
"DB.service.consul.",
|
|
|
|
"Db.service.consul.",
|
|
|
|
"somequery.query.consul.",
|
|
|
|
"SomeQuery.query.consul.",
|
|
|
|
"SOMEQUERY.query.consul.",
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("empty lookup: %#v", in)
|
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
func TestDNS_ServiceLookup_TagPeriod(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-01-08 19:58:32 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
2015-11-13 11:39:07 +00:00
|
|
|
Tags: []string{"v1.master"},
|
2015-01-08 19:58:32 +00:00
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-01-08 19:58:32 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
2015-11-13 11:39:07 +00:00
|
|
|
m.SetQuestion("v1.master.db.service.consul.", dns.TypeSRV)
|
2015-01-08 19:58:32 +00:00
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-01-08 19:58:32 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
2015-01-08 19:58:32 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
func TestDNS_PreparedQueryNearIPEDNS(t *testing.T) {
|
2018-04-11 14:33:48 +00:00
|
|
|
ipCoord := lib.GenerateCoordinate(1 * time.Millisecond)
|
2018-04-13 16:57:25 +00:00
|
|
|
serviceNodes := []struct {
|
|
|
|
name string
|
|
|
|
address string
|
|
|
|
coord *coordinate.Coordinate
|
2018-04-11 14:33:48 +00:00
|
|
|
}{
|
2018-04-13 16:57:25 +00:00
|
|
|
{"foo1", "198.18.0.1", lib.GenerateCoordinate(1 * time.Millisecond)},
|
|
|
|
{"foo2", "198.18.0.2", lib.GenerateCoordinate(10 * time.Millisecond)},
|
|
|
|
{"foo3", "198.18.0.3", lib.GenerateCoordinate(30 * time.Millisecond)},
|
2018-04-11 14:33:48 +00:00
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
added := 0
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
// Register nodes with a service
|
|
|
|
for _, cfg := range serviceNodes {
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: cfg.name,
|
|
|
|
Address: cfg.address,
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
var out struct{}
|
|
|
|
err := a.RPC("Catalog.Register", args, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
// Send coordinate updates
|
|
|
|
coordArgs := structs.CoordinateUpdateRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: cfg.name,
|
|
|
|
Coord: cfg.coord,
|
|
|
|
}
|
|
|
|
err = a.RPC("Coordinate.Update", &coordArgs, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
added += 1
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
fmt.Printf("Added %d service nodes\n", added)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
// Register a node without a service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
2018-04-12 14:40:46 +00:00
|
|
|
Address: "198.18.0.9",
|
2018-04-11 14:33:48 +00:00
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
var out struct{}
|
|
|
|
err := a.RPC("Catalog.Register", args, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
// Send coordinate updates for a few nodes.
|
|
|
|
coordArgs := structs.CoordinateUpdateRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Coord: ipCoord,
|
|
|
|
}
|
|
|
|
err = a.RPC("Coordinate.Update", &coordArgs, &out)
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 14:33:48 +00:00
|
|
|
// Register a prepared query Near = _ip
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "some.query.we.like",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
2018-04-13 16:57:25 +00:00
|
|
|
Near: "_ip",
|
2018-04-11 14:33:48 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var id string
|
|
|
|
err := a.RPC("PreparedQuery.Apply", args, &id)
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
2018-04-11 20:00:15 +00:00
|
|
|
retry.Run(t, func(r *retry.R) {
|
2018-04-12 00:32:35 +00:00
|
|
|
m := new(dns.Msg)
|
2018-04-11 20:00:15 +00:00
|
|
|
m.SetQuestion("some.query.we.like.query.consul.", dns.TypeA)
|
|
|
|
m.SetEdns0(4096, false)
|
|
|
|
o := new(dns.OPT)
|
|
|
|
o.Hdr.Name = "."
|
|
|
|
o.Hdr.Rrtype = dns.TypeOPT
|
|
|
|
e := new(dns.EDNS0_SUBNET)
|
|
|
|
e.Code = dns.EDNS0SUBNET
|
|
|
|
e.Family = 1
|
|
|
|
e.SourceNetmask = 32
|
|
|
|
e.SourceScope = 0
|
2018-04-12 14:40:46 +00:00
|
|
|
e.Address = net.ParseIP("198.18.0.9").To4()
|
2018-04-11 20:00:15 +00:00
|
|
|
o.Option = append(o.Option, e)
|
|
|
|
m.Extra = append(m.Extra, o)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 20:00:15 +00:00
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
r.Fatalf("Error with call to dns.Client.Exchange: %s", err)
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 20:00:15 +00:00
|
|
|
if len(serviceNodes) != len(in.Answer) {
|
|
|
|
r.Fatalf("Expecting %d A RRs in response, Actual found was %d", len(serviceNodes), len(in.Answer))
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:10:37 +00:00
|
|
|
for i, rr := range in.Answer {
|
|
|
|
if aRec, ok := rr.(*dns.A); ok {
|
|
|
|
if actual := aRec.A.String(); serviceNodes[i].address != actual {
|
|
|
|
r.Fatalf("Expecting A RR #%d = %s, Actual RR was %s", i, serviceNodes[i].address, actual)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
r.Fatalf("DNS Answer contained a non-A RR")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
2018-04-12 14:40:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_PreparedQueryNearIP(t *testing.T) {
|
|
|
|
ipCoord := lib.GenerateCoordinate(1 * time.Millisecond)
|
2018-04-13 16:57:25 +00:00
|
|
|
serviceNodes := []struct {
|
|
|
|
name string
|
|
|
|
address string
|
|
|
|
coord *coordinate.Coordinate
|
2018-04-12 14:40:46 +00:00
|
|
|
}{
|
2018-04-13 16:57:25 +00:00
|
|
|
{"foo1", "198.18.0.1", lib.GenerateCoordinate(1 * time.Millisecond)},
|
|
|
|
{"foo2", "198.18.0.2", lib.GenerateCoordinate(10 * time.Millisecond)},
|
|
|
|
{"foo3", "198.18.0.3", lib.GenerateCoordinate(30 * time.Millisecond)},
|
2018-04-12 14:40:46 +00:00
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), "")
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
added := 0
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
// Register nodes with a service
|
|
|
|
for _, cfg := range serviceNodes {
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: cfg.name,
|
|
|
|
Address: cfg.address,
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
var out struct{}
|
|
|
|
err := a.RPC("Catalog.Register", args, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
// Send coordinate updates
|
|
|
|
coordArgs := structs.CoordinateUpdateRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: cfg.name,
|
|
|
|
Coord: cfg.coord,
|
|
|
|
}
|
|
|
|
err = a.RPC("Coordinate.Update", &coordArgs, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
added += 1
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
fmt.Printf("Added %d service nodes\n", added)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
// Register a node without a service
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "198.18.0.9",
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
var out struct{}
|
|
|
|
err := a.RPC("Catalog.Register", args, &out)
|
|
|
|
require.NoError(t, err)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
// Send coordinate updates for a few nodes.
|
|
|
|
coordArgs := structs.CoordinateUpdateRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Coord: ipCoord,
|
|
|
|
}
|
|
|
|
err = a.RPC("Coordinate.Update", &coordArgs, &out)
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:40:46 +00:00
|
|
|
// Register a prepared query Near = _ip
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "some.query.we.like",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
2018-04-13 16:57:25 +00:00
|
|
|
Near: "_ip",
|
2018-04-12 14:40:46 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var id string
|
|
|
|
err := a.RPC("PreparedQuery.Apply", args, &id)
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:10:37 +00:00
|
|
|
retry.Run(t, func(r *retry.R) {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("some.query.we.like.query.consul.", dns.TypeA)
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:10:37 +00:00
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
r.Fatalf("Error with call to dns.Client.Exchange: %s", err)
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-12 14:10:37 +00:00
|
|
|
if len(serviceNodes) != len(in.Answer) {
|
|
|
|
r.Fatalf("Expecting %d A RRs in response, Actual found was %d", len(serviceNodes), len(in.Answer))
|
|
|
|
}
|
2018-04-13 16:57:25 +00:00
|
|
|
|
2018-04-11 20:00:15 +00:00
|
|
|
for i, rr := range in.Answer {
|
|
|
|
if aRec, ok := rr.(*dns.A); ok {
|
|
|
|
if actual := aRec.A.String(); serviceNodes[i].address != actual {
|
|
|
|
r.Fatalf("Expecting A RR #%d = %s, Actual RR was %s", i, serviceNodes[i].address, actual)
|
|
|
|
}
|
|
|
|
} else {
|
2018-04-12 00:32:35 +00:00
|
|
|
r.Fatalf("DNS Answer contained a non-A RR")
|
2018-04-11 20:00:15 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
2018-04-11 14:33:48 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
func TestDNS_ServiceLookup_PreparedQueryNamePeriod(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-07-17 06:38:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2014-07-17 06:38:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-07-17 06:38:24 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a prepared query with a period in the name.
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "some.query.we.like",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var id string
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-04-21 22:33:01 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
2015-11-13 11:39:07 +00:00
|
|
|
m.SetQuestion("some.query.we.like.query.consul.", dns.TypeSRV)
|
2014-04-21 22:33:01 +00:00
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-04-21 22:33:01 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
2014-01-03 21:00:03 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
|
2014-02-26 01:41:48 +00:00
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
2014-01-03 21:00:03 +00:00
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
2014-01-03 23:43:35 +00:00
|
|
|
|
2014-01-06 22:56:41 +00:00
|
|
|
func TestDNS_ServiceLookup_Dedup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a single node with multiple instances of a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
ID: "db2",
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"slave"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
ID: "db3",
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"slave"},
|
|
|
|
Port: 12346,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query, make sure only
|
|
|
|
// one IP is returned.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-01-06 22:56:41 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
}
|
2014-02-26 01:41:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_Dedup_SRV(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a single node with multiple instances of a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-26 01:41:48 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
ID: "db2",
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"slave"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-26 01:41:48 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
ID: "db3",
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"slave"},
|
|
|
|
Port: 12346,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-26 01:41:48 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-26 01:41:48 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query, make sure only
|
|
|
|
// one IP is returned and two unique ports are returned.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-02-26 01:41:48 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
if len(in.Answer) != 2 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 && srvRec.Port != 12346 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok = in.Answer[1].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[1])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12346 && srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Port == in.Answer[0].(*dns.SRV).Port {
|
|
|
|
t.Fatalf("should be a different port")
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2014-01-06 22:56:41 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-03 23:43:35 +00:00
|
|
|
func TestDNS_Recurse(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
|
|
|
Answer: []dns.RR{dnsA("apple.com", "1.2.3.4")},
|
|
|
|
})
|
2015-04-14 02:19:22 +00:00
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-01-03 23:43:35 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("apple.com.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-01-03 23:43:35 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) == 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
if in.Rcode != dns.RcodeSuccess {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
}
|
2014-01-15 21:30:04 +00:00
|
|
|
|
2016-11-03 19:21:16 +00:00
|
|
|
func TestDNS_Recurse_Truncation(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
|
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
2016-11-03 19:21:16 +00:00
|
|
|
MsgHdr: dns.MsgHdr{Truncated: true},
|
|
|
|
Answer: []dns.RR{dnsA("apple.com", "1.2.3.4")},
|
2017-06-25 17:42:37 +00:00
|
|
|
})
|
2016-11-03 19:21:16 +00:00
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-11-03 19:21:16 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("apple.com.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-11-03 19:26:48 +00:00
|
|
|
if err != dns.ErrTruncated {
|
2016-11-03 19:21:16 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
if in.Truncated != true {
|
|
|
|
t.Fatalf("err: message should have been truncated %v", in)
|
|
|
|
}
|
|
|
|
if len(in.Answer) == 0 {
|
|
|
|
t.Fatalf("Bad: Truncated message ignored, expected some reply %#v", in)
|
|
|
|
}
|
|
|
|
if in.Rcode != dns.RcodeSuccess {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-08-26 19:22:04 +00:00
|
|
|
func TestDNS_RecursorTimeout(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2016-08-23 20:59:38 +00:00
|
|
|
serverClientTimeout := 3 * time.Second
|
|
|
|
testClientTimeout := serverClientTimeout + 5*time.Second
|
|
|
|
|
2016-10-27 02:23:51 +00:00
|
|
|
resolverAddr, err := net.ResolveUDPAddr("udp", "127.0.0.1:0")
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
resolver, err := net.ListenUDP("udp", resolverAddr)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
defer resolver.Close()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+resolver.LocalAddr().String()+`"] // host must cause a connection|read|write timeout
|
|
|
|
dns_config {
|
|
|
|
recursor_timeout = "`+serverClientTimeout.String()+`"
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-08-23 20:59:38 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("apple.com.", dns.TypeANY)
|
|
|
|
|
|
|
|
// This client calling the server under test must have a longer timeout than the one we set internally
|
|
|
|
c := &dns.Client{Timeout: testClientTimeout}
|
|
|
|
|
|
|
|
start := time.Now()
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-08-23 20:59:38 +00:00
|
|
|
|
2017-10-17 18:38:24 +00:00
|
|
|
duration := time.Since(start)
|
2016-08-23 20:59:38 +00:00
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
if in.Rcode != dns.RcodeServerFailure {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
if duration < serverClientTimeout {
|
2016-09-02 04:52:29 +00:00
|
|
|
t.Fatalf("Expected the call to return after at least %f seconds but lasted only %f", serverClientTimeout.Seconds(), duration.Seconds())
|
2016-08-23 20:59:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2014-01-15 21:30:04 +00:00
|
|
|
func TestDNS_ServiceLookup_FilterCritical(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register nodes with health checks in various states.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "serf",
|
|
|
|
Name: "serf",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-15 21:30:04 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args2 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "serf",
|
|
|
|
Name: "serf",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args2, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-24 02:04:12 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args3 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args3, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-07-16 22:11:45 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args4 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "baz",
|
|
|
|
Address: "127.0.0.3",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args4, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-07-16 22:11:45 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args5 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "quux",
|
|
|
|
Address: "127.0.0.4",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthWarning,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args5, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-01-15 21:30:04 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2015-01-13 13:59:24 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Only 4 and 5 are not failing, so we should get 2 answers
|
|
|
|
if len(in.Answer) != 2 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
ips := make(map[string]bool)
|
|
|
|
for _, resp := range in.Answer {
|
|
|
|
aRec := resp.(*dns.A)
|
|
|
|
ips[aRec.A.String()] = true
|
|
|
|
}
|
|
|
|
|
|
|
|
if !ips["127.0.0.3"] {
|
|
|
|
t.Fatalf("Bad: %#v should contain 127.0.0.3 (state healthy)", in)
|
|
|
|
}
|
|
|
|
if !ips["127.0.0.4"] {
|
|
|
|
t.Fatalf("Bad: %#v should contain 127.0.0.4 (state warning)", in)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-07-29 22:21:16 +00:00
|
|
|
func TestDNS_ServiceLookup_OnlyFailing(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Register nodes with all health checks in a critical state.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "serf",
|
|
|
|
Name: "serf",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args2 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "serf",
|
|
|
|
Name: "serf",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args2, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args3 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args3, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2015-07-29 22:21:16 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2015-07-29 22:21:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// All 3 are failing, so we should get 0 answers and an NXDOMAIN response
|
|
|
|
if len(in.Answer) != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
if in.Rcode != dns.RcodeNameError {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2015-07-29 22:21:16 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-13 13:59:24 +00:00
|
|
|
func TestDNS_ServiceLookup_OnlyPassing(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
only_passing = true
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register nodes with health checks in various states.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthPassing,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args2 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthWarning,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args2, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args3 := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "baz",
|
|
|
|
Address: "127.0.0.3",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
Check: &structs.HealthCheck{
|
|
|
|
CheckID: "db",
|
|
|
|
Name: "db",
|
|
|
|
ServiceID: "db",
|
2017-04-19 23:00:11 +00:00
|
|
|
Status: api.HealthCritical,
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args3, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
OnlyPassing: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-01-15 21:30:04 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-07-16 22:11:45 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Only 1 is passing, so we should only get 1 answer
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2015-01-13 13:59:24 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
resp := in.Answer[0]
|
|
|
|
aRec := resp.(*dns.A)
|
|
|
|
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2014-07-16 22:11:45 +00:00
|
|
|
}
|
2014-01-15 21:30:04 +00:00
|
|
|
}
|
2014-02-14 20:26:51 +00:00
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_Randomize(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2016-03-30 06:08:40 +00:00
|
|
|
// Register a large number of nodes.
|
|
|
|
for i := 0; i < generateNumNodes; i++ {
|
2014-02-14 20:26:51 +00:00
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("foo%d", i),
|
|
|
|
Address: fmt.Sprintf("127.0.0.%d", i+1),
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "web",
|
|
|
|
Port: 8000,
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2014-02-14 20:26:51 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-02-14 20:26:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "web",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2014-02-14 20:26:51 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2014-02-14 20:26:51 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query. Ensure the
|
|
|
|
// response is randomized each time.
|
|
|
|
questions := []string{
|
|
|
|
"web.service.consul.",
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
uniques := map[string]struct{}{}
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2014-02-14 20:26:51 +00:00
|
|
|
|
2016-02-12 07:58:48 +00:00
|
|
|
c := &dns.Client{Net: "udp"}
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
2014-02-14 20:26:51 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Response length should be truncated and we should get
|
|
|
|
// an A record for each response.
|
2016-03-30 06:08:40 +00:00
|
|
|
if len(in.Answer) != defaultNumUDPResponses {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", len(in.Answer))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Collect all the names.
|
|
|
|
var names []string
|
|
|
|
for _, rec := range in.Answer {
|
|
|
|
switch v := rec.(type) {
|
|
|
|
case *dns.SRV:
|
|
|
|
names = append(names, v.Target)
|
|
|
|
case *dns.A:
|
|
|
|
names = append(names, v.A.String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
nameS := strings.Join(names, "|")
|
|
|
|
|
|
|
|
// Tally the results.
|
|
|
|
uniques[nameS] = struct{}{}
|
|
|
|
}
|
2015-01-13 00:05:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Give some wiggle room. Since the responses are randomized and
|
|
|
|
// there is a finite number of combinations, requiring 0
|
|
|
|
// duplicates every test run eventually gives us failures.
|
|
|
|
if len(uniques) < 2 {
|
|
|
|
t.Fatalf("unique response ratio too low: %d/10\n%v", len(uniques), uniques)
|
|
|
|
}
|
2015-01-13 00:05:41 +00:00
|
|
|
}
|
2014-02-14 20:26:51 +00:00
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
|
2018-04-17 07:31:30 +00:00
|
|
|
func TestBinarySearch(t *testing.T) {
|
2018-04-17 08:36:12 +00:00
|
|
|
t.Parallel()
|
2018-04-17 07:31:30 +00:00
|
|
|
msgSrc := new(dns.Msg)
|
|
|
|
msgSrc.Compress = true
|
|
|
|
msgSrc.SetQuestion("redis.service.consul.", dns.TypeSRV)
|
|
|
|
|
2018-04-21 15:18:39 +00:00
|
|
|
for i := 0; i < 5000; i++ {
|
2018-04-17 07:31:30 +00:00
|
|
|
target := fmt.Sprintf("host-redis-%d-%d.test.acme.com.node.dc1.consul.", i/256, i%256)
|
|
|
|
msgSrc.Answer = append(msgSrc.Answer, &dns.SRV{Hdr: dns.RR_Header{Name: "redis.service.consul.", Class: 1, Rrtype: dns.TypeSRV, Ttl: 0x3c}, Port: 0x4c57, Target: target})
|
|
|
|
msgSrc.Extra = append(msgSrc.Extra, &dns.CNAME{Hdr: dns.RR_Header{Name: target, Class: 1, Rrtype: dns.TypeCNAME, Ttl: 0x3c}, Target: fmt.Sprintf("fx.168.%d.%d.", i/256, i%256)})
|
|
|
|
}
|
2018-04-21 15:18:39 +00:00
|
|
|
for _, compress := range []bool{true, false} {
|
|
|
|
for idx, maxSize := range []int{12, 256, 512, 8192, 65535} {
|
|
|
|
t.Run(fmt.Sprintf("binarySearch %d", maxSize), func(t *testing.T) {
|
|
|
|
msg := new(dns.Msg)
|
|
|
|
msgSrc.Compress = compress
|
|
|
|
msgSrc.SetQuestion("redis.service.consul.", dns.TypeSRV)
|
|
|
|
msg.Answer = msgSrc.Answer
|
|
|
|
msg.Extra = msgSrc.Extra
|
|
|
|
index := make(map[string]dns.RR, len(msg.Extra))
|
|
|
|
indexRRs(msg.Extra, index)
|
|
|
|
blen := dnsBinaryTruncate(msg, maxSize, index, true)
|
|
|
|
msg.Answer = msg.Answer[:blen]
|
|
|
|
syncExtra(index, msg)
|
|
|
|
predicted := msg.Len()
|
|
|
|
buf, err := msg.Pack()
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if predicted < len(buf) {
|
|
|
|
t.Fatalf("Bug in DNS library: %d != %d", predicted, len(buf))
|
|
|
|
}
|
|
|
|
if len(buf) > maxSize || (idx != 0 && len(buf) < 16) {
|
|
|
|
t.Fatalf("bad[%d]: %d > %d", idx, len(buf), maxSize)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
2018-04-17 07:31:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-03-07 09:01:12 +00:00
|
|
|
func TestDNS_TCP_and_UDP_Truncate(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
enable_truncate = true
|
|
|
|
}
|
|
|
|
`)
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-03-07 09:01:12 +00:00
|
|
|
|
|
|
|
services := []string{"normal", "truncated"}
|
|
|
|
for index, service := range services {
|
|
|
|
numServices := (index * 5000) + 2
|
|
|
|
for i := 1; i < numServices; i++ {
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("%s-%d.acme.com", service, i),
|
2018-05-16 12:10:35 +00:00
|
|
|
Address: fmt.Sprintf("127.%d.%d.%d", 0, (i / 255), i%255),
|
2018-03-07 09:01:12 +00:00
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: service,
|
|
|
|
Port: 8000,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: service,
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: service,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly and via prepared query. Ensure the
|
|
|
|
// response is truncated each time.
|
|
|
|
questions := []string{
|
|
|
|
fmt.Sprintf("%s.service.consul.", service),
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
protocols := []string{
|
|
|
|
"tcp",
|
|
|
|
"udp",
|
|
|
|
}
|
2018-05-16 10:11:49 +00:00
|
|
|
for _, maxSize := range []uint16{8192, 65535} {
|
|
|
|
for _, qType := range []uint16{dns.TypeANY, dns.TypeA, dns.TypeSRV} {
|
|
|
|
for _, question := range questions {
|
|
|
|
for _, protocol := range protocols {
|
|
|
|
for _, compress := range []bool{true, false} {
|
|
|
|
t.Run(fmt.Sprintf("lookup %s %s (qType:=%d) compressed=%v", question, protocol, qType, compress), func(t *testing.T) {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
|
|
|
maxSz := maxSize
|
|
|
|
if protocol == "udp" {
|
|
|
|
maxSz = 8192
|
|
|
|
}
|
|
|
|
m.SetEdns0(uint16(maxSz), true)
|
|
|
|
c := new(dns.Client)
|
|
|
|
c.Net = protocol
|
|
|
|
m.Compress = compress
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil && err != dns.ErrTruncated {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for the truncate bit
|
|
|
|
buf, err := m.Pack()
|
2018-03-09 17:25:29 +00:00
|
|
|
info := fmt.Sprintf("service %s question:=%s (%s) (%d total records) sz:= %d in %v",
|
2018-05-16 10:11:49 +00:00
|
|
|
service, question, protocol, numServices, len(in.Answer), in)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("Error while packing: %v ; info:=%s", err, info)
|
|
|
|
}
|
|
|
|
if len(buf) > int(maxSz) {
|
|
|
|
t.Fatalf("len(buf) := %d > maxSz=%d for %v", len(buf), maxSz, info)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
2018-03-07 22:33:41 +00:00
|
|
|
}
|
2018-03-07 09:01:12 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-05-05 21:14:41 +00:00
|
|
|
func TestDNS_ServiceLookup_Truncate(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
enable_truncate = true
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-05-05 21:14:41 +00:00
|
|
|
|
2016-03-30 06:08:40 +00:00
|
|
|
// Register a large number of nodes.
|
|
|
|
for i := 0; i < generateNumNodes; i++ {
|
2015-05-05 21:14:41 +00:00
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("foo%d", i),
|
|
|
|
Address: fmt.Sprintf("127.0.0.%d", i+1),
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "web",
|
|
|
|
Port: 8000,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-05-05 21:14:41 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "web",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
2015-05-05 21:14:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query. Ensure the
|
|
|
|
// response is truncated each time.
|
|
|
|
questions := []string{
|
|
|
|
"web.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2015-05-05 21:14:41 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2015-05-05 21:14:41 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil && err != dns.ErrTruncated {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for the truncate bit
|
|
|
|
if !in.Truncated {
|
|
|
|
t.Fatalf("should have truncate bit")
|
|
|
|
}
|
2015-05-05 21:14:41 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-02-18 00:54:28 +00:00
|
|
|
func TestDNS_ServiceLookup_LargeResponses(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
enable_truncate = true
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-02-18 00:54:28 +00:00
|
|
|
|
|
|
|
longServiceName := "this-is-a-very-very-very-very-very-long-name-for-a-service"
|
|
|
|
|
2016-08-12 04:46:14 +00:00
|
|
|
// Register a lot of nodes.
|
|
|
|
for i := 0; i < 4; i++ {
|
2016-02-18 00:54:28 +00:00
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("foo%d", i),
|
|
|
|
Address: fmt.Sprintf("127.0.0.%d", i+1),
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: longServiceName,
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-02-18 00:54:28 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-03-07 18:37:54 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: longServiceName,
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: longServiceName,
|
|
|
|
Tags: []string{"master"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
var id string
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2016-03-07 18:37:54 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2016-02-18 00:54:28 +00:00
|
|
|
}
|
|
|
|
|
2016-03-07 18:37:54 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"_" + longServiceName + "._master.service.consul.",
|
|
|
|
longServiceName + ".query.consul.",
|
2016-02-18 00:54:28 +00:00
|
|
|
}
|
2016-03-07 18:37:54 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
2016-02-18 00:54:28 +00:00
|
|
|
|
2016-03-07 18:37:54 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-03-07 18:37:54 +00:00
|
|
|
if err != nil && err != dns.ErrTruncated {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure the response size is RFC 1035-compliant for UDP messages
|
|
|
|
if in.Len() > 512 {
|
2016-08-12 04:46:14 +00:00
|
|
|
t.Fatalf("Bad: %d", in.Len())
|
2016-03-07 18:37:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// We should only have two answers now
|
|
|
|
if len(in.Answer) != 2 {
|
2016-08-12 04:46:14 +00:00
|
|
|
t.Fatalf("Bad: %d", len(in.Answer))
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure the ADDITIONAL section matches the ANSWER section.
|
|
|
|
if len(in.Answer) != len(in.Extra) {
|
|
|
|
t.Fatalf("Bad: %d vs. %d", len(in.Answer), len(in.Extra))
|
|
|
|
}
|
|
|
|
for i := 0; i < len(in.Answer); i++ {
|
|
|
|
srv, ok := in.Answer[i].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[i])
|
|
|
|
}
|
|
|
|
|
|
|
|
a, ok := in.Extra[i].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[i])
|
|
|
|
}
|
|
|
|
|
|
|
|
if srv.Target != a.Hdr.Name {
|
|
|
|
t.Fatalf("Bad: %#v %#v", srv, a)
|
|
|
|
}
|
2016-03-07 18:37:54 +00:00
|
|
|
}
|
2016-02-18 00:54:28 +00:00
|
|
|
|
2016-03-07 18:37:54 +00:00
|
|
|
// Check for the truncate bit
|
|
|
|
if !in.Truncated {
|
|
|
|
t.Fatalf("should have truncate bit")
|
|
|
|
}
|
2016-02-18 00:54:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-01-28 18:40:13 +00:00
|
|
|
func testDNSServiceLookupResponseLimits(t *testing.T, answerLimit int, qType uint16,
|
2016-03-30 08:05:49 +00:00
|
|
|
expectedService, expectedQuery, expectedQueryID int) (bool, error) {
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
node_name = "test-node"
|
|
|
|
dns_config {
|
|
|
|
udp_answer_limit = `+fmt.Sprintf("%d", answerLimit)+`
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-08-25 19:54:11 +00:00
|
|
|
|
2016-03-30 06:08:58 +00:00
|
|
|
for i := 0; i < generateNumNodes; i++ {
|
|
|
|
nodeAddress := fmt.Sprintf("127.0.0.%d", i+1)
|
|
|
|
if rand.Float64() < pctNodesWithIPv6 {
|
|
|
|
nodeAddress = fmt.Sprintf("fe80::%d", i+1)
|
|
|
|
}
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("foo%d", i),
|
|
|
|
Address: nodeAddress,
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "api-tier",
|
|
|
|
Port: 8080,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-03-30 08:05:49 +00:00
|
|
|
return false, fmt.Errorf("err: %v", err)
|
2016-03-30 06:08:58 +00:00
|
|
|
}
|
|
|
|
}
|
2016-03-30 08:05:49 +00:00
|
|
|
var id string
|
2016-03-30 06:08:58 +00:00
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "api-tier",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "api-tier",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2016-03-30 08:05:49 +00:00
|
|
|
return false, fmt.Errorf("err: %v", err)
|
2016-03-30 06:08:58 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-02-12 07:58:48 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
2016-03-30 06:08:58 +00:00
|
|
|
"api-tier.service.consul.",
|
|
|
|
"api-tier.query.consul.",
|
2016-03-30 08:05:49 +00:00
|
|
|
id + ".query.consul.",
|
2016-02-12 07:58:48 +00:00
|
|
|
}
|
2016-03-30 08:05:49 +00:00
|
|
|
for idx, question := range questions {
|
2016-02-12 07:58:48 +00:00
|
|
|
m := new(dns.Msg)
|
2016-03-30 08:05:49 +00:00
|
|
|
m.SetQuestion(question, qType)
|
2016-02-12 07:58:48 +00:00
|
|
|
|
|
|
|
c := &dns.Client{Net: "udp"}
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-02-12 07:58:48 +00:00
|
|
|
if err != nil {
|
2016-03-30 08:05:49 +00:00
|
|
|
return false, fmt.Errorf("err: %v", err)
|
2016-02-12 07:58:48 +00:00
|
|
|
}
|
|
|
|
|
2016-03-30 08:05:49 +00:00
|
|
|
switch idx {
|
|
|
|
case 0:
|
2016-03-30 18:48:32 +00:00
|
|
|
if (expectedService > 0 && len(in.Answer) != expectedService) ||
|
|
|
|
(expectedService < -1 && len(in.Answer) < lib.AbsInt(expectedService)) {
|
2018-04-18 12:18:17 +00:00
|
|
|
return false, fmt.Errorf("%d/%d answers received for type %v for %s, sz:=%d", len(in.Answer), answerLimit, qType, question, in.Len())
|
2016-03-30 08:05:49 +00:00
|
|
|
}
|
|
|
|
case 1:
|
2016-03-30 18:48:32 +00:00
|
|
|
if (expectedQuery > 0 && len(in.Answer) != expectedQuery) ||
|
|
|
|
(expectedQuery < -1 && len(in.Answer) < lib.AbsInt(expectedQuery)) {
|
2018-04-18 12:18:17 +00:00
|
|
|
return false, fmt.Errorf("%d/%d answers received for type %v for %s, sz:=%d", len(in.Answer), answerLimit, qType, question, in.Len())
|
2016-03-30 08:05:49 +00:00
|
|
|
}
|
|
|
|
case 2:
|
2016-03-30 18:48:32 +00:00
|
|
|
if (expectedQueryID > 0 && len(in.Answer) != expectedQueryID) ||
|
|
|
|
(expectedQueryID < -1 && len(in.Answer) < lib.AbsInt(expectedQueryID)) {
|
2018-04-18 12:18:17 +00:00
|
|
|
return false, fmt.Errorf("%d/%d answers received for type %v for %s, sz:=%d", len(in.Answer), answerLimit, qType, question, in.Len())
|
2016-03-30 08:05:49 +00:00
|
|
|
}
|
|
|
|
default:
|
|
|
|
panic("abort")
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2016-03-30 08:05:49 +00:00
|
|
|
}
|
2015-08-25 19:54:11 +00:00
|
|
|
|
2016-03-30 08:05:49 +00:00
|
|
|
return true, nil
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
|
2018-03-07 17:24:41 +00:00
|
|
|
func checkDNSService(t *testing.T, generateNumNodes int, aRecordLimit int, qType uint16,
|
|
|
|
expectedResultsCount int, udpSize uint16, udpAnswerLimit int) error {
|
2018-03-06 01:07:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
node_name = "test-node"
|
|
|
|
dns_config {
|
|
|
|
a_record_limit = `+fmt.Sprintf("%d", aRecordLimit)+`
|
|
|
|
udp_answer_limit = `+fmt.Sprintf("%d", aRecordLimit)+`
|
|
|
|
}
|
|
|
|
`)
|
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-03-06 01:07:42 +00:00
|
|
|
|
|
|
|
for i := 0; i < generateNumNodes; i++ {
|
|
|
|
nodeAddress := fmt.Sprintf("127.0.0.%d", i+1)
|
|
|
|
if rand.Float64() < pctNodesWithIPv6 {
|
|
|
|
nodeAddress = fmt.Sprintf("fe80::%d", i+1)
|
|
|
|
}
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: fmt.Sprintf("foo%d", i),
|
|
|
|
Address: nodeAddress,
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "api-tier",
|
|
|
|
Port: 8080,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2018-03-07 17:24:41 +00:00
|
|
|
return fmt.Errorf("err: %v", err)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "api-tier",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "api-tier",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2018-03-07 17:24:41 +00:00
|
|
|
return fmt.Errorf("err: %v", err)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"api-tier.service.consul.",
|
|
|
|
"api-tier.query.consul.",
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
|
|
|
|
m.SetQuestion(question, qType)
|
|
|
|
protocol := "tcp"
|
|
|
|
if udpSize > 0 {
|
|
|
|
protocol = "udp"
|
|
|
|
}
|
|
|
|
if udpSize > 512 {
|
|
|
|
m.SetEdns0(udpSize, true)
|
|
|
|
}
|
|
|
|
c := &dns.Client{Net: protocol, UDPSize: 8192}
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
2018-03-07 17:24:41 +00:00
|
|
|
return fmt.Errorf("err: %v", err)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
if len(in.Answer) != expectedResultsCount {
|
2018-03-07 17:24:41 +00:00
|
|
|
return fmt.Errorf("%d/%d answers received for type %v for %s (%s)", len(in.Answer), expectedResultsCount, qType, question, protocol)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-03-07 17:24:41 +00:00
|
|
|
return nil
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_ARecordLimits(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
2018-03-27 20:31:27 +00:00
|
|
|
name string
|
|
|
|
aRecordLimit int
|
|
|
|
expectedAResults int
|
2018-03-19 16:56:00 +00:00
|
|
|
expectedAAAAResults int
|
2018-03-27 20:31:27 +00:00
|
|
|
expectedSRVResults int
|
|
|
|
numNodesTotal int
|
|
|
|
udpSize uint16
|
|
|
|
udpAnswerLimit int
|
2018-03-06 01:07:42 +00:00
|
|
|
}{
|
|
|
|
// UDP + EDNS
|
|
|
|
{"udp-edns-1", 1, 1, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-2", 2, 2, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-3", 3, 3, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-4", 4, 4, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-5", 5, 5, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-6", 6, 6, 1, 30, 30, 8192, 3},
|
|
|
|
{"udp-edns-max", 6, 3, 3, 3, 3, 8192, 3},
|
|
|
|
// All UDP without EDNS have a limit of 2 answers due to udpAnswerLimit
|
|
|
|
// Even SRV records are limit to 2 records
|
|
|
|
{"udp-limit-1", 1, 1, 1, 1, 1, 512, 2},
|
|
|
|
{"udp-limit-2", 2, 2, 2, 2, 2, 512, 2},
|
|
|
|
// AAAA results limited by size of payload
|
|
|
|
{"udp-limit-3", 3, 2, 2, 2, 2, 512, 2},
|
|
|
|
{"udp-limit-4", 4, 2, 2, 2, 2, 512, 2},
|
|
|
|
{"udp-limit-5", 5, 2, 2, 2, 2, 512, 2},
|
|
|
|
{"udp-limit-6", 6, 2, 2, 2, 2, 512, 2},
|
|
|
|
{"udp-limit-max", 6, 2, 2, 2, 2, 512, 2},
|
|
|
|
// All UDP without EDNS and no udpAnswerLimit
|
|
|
|
// Size of records is limited by UDP payload
|
|
|
|
{"udp-1", 1, 1, 1, 1, 1, 512, 0},
|
|
|
|
{"udp-2", 2, 2, 2, 2, 2, 512, 0},
|
|
|
|
{"udp-3", 3, 2, 2, 2, 2, 512, 0},
|
|
|
|
{"udp-4", 4, 2, 2, 2, 2, 512, 0},
|
|
|
|
{"udp-5", 5, 2, 2, 2, 2, 512, 0},
|
|
|
|
{"udp-6", 6, 2, 2, 2, 2, 512, 0},
|
|
|
|
// Only 3 A and 3 SRV records on 512 bytes
|
|
|
|
{"udp-max", 6, 2, 2, 2, 2, 512, 0},
|
|
|
|
|
|
|
|
{"tcp-1", 1, 1, 1, 30, 30, 0, 0},
|
|
|
|
{"tcp-2", 2, 2, 2, 30, 30, 0, 0},
|
|
|
|
{"tcp-3", 3, 3, 3, 30, 30, 0, 0},
|
|
|
|
{"tcp-4", 4, 4, 4, 30, 30, 0, 0},
|
|
|
|
{"tcp-5", 5, 5, 5, 30, 30, 0, 0},
|
|
|
|
{"tcp-6", 6, 6, 5, 30, 30, 0, 0},
|
|
|
|
{"tcp-max", 6, 2, 2, 2, 2, 0, 0},
|
|
|
|
}
|
|
|
|
for _, test := range tests {
|
|
|
|
test := test // capture loop var
|
|
|
|
|
|
|
|
queriesLimited := []uint16{
|
|
|
|
dns.TypeA,
|
|
|
|
dns.TypeAAAA,
|
|
|
|
dns.TypeANY,
|
|
|
|
}
|
|
|
|
// All those queries should have at max queriesLimited elements
|
|
|
|
for idx, qType := range queriesLimited {
|
|
|
|
t.Run(fmt.Sprintf("ARecordLimit %d qType: %d", idx, qType), func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2018-03-07 17:24:41 +00:00
|
|
|
err := checkDNSService(t, test.numNodesTotal, test.aRecordLimit, qType, test.expectedAResults, test.udpSize, test.udpAnswerLimit)
|
|
|
|
if err != nil {
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Fatalf("Expected lookup %s to pass: %v", test.name, err)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
// No limits but the size of records for SRV records, since not subject to randomization issues
|
|
|
|
t.Run("SRV lookup limitARecord", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2018-03-07 17:24:41 +00:00
|
|
|
err := checkDNSService(t, test.expectedSRVResults, test.aRecordLimit, dns.TypeSRV, test.numNodesTotal, test.udpSize, test.udpAnswerLimit)
|
|
|
|
if err != nil {
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Fatalf("Expected service SRV lookup %s to pass: %v", test.name, err)
|
2018-03-06 01:07:42 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-03-30 08:05:49 +00:00
|
|
|
func TestDNS_ServiceLookup_AnswerLimits(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2016-03-30 08:05:49 +00:00
|
|
|
// Build a matrix of config parameters (udpAnswerLimit), and the
|
2016-03-30 18:48:32 +00:00
|
|
|
// length of the response per query type and question. Negative
|
|
|
|
// values imply the test must return at least the abs(value) number
|
|
|
|
// of records in the answer section. This is required because, for
|
|
|
|
// example, on OS-X and Linux, the number of answers returned in a
|
|
|
|
// 512B response is different even though both platforms are x86_64
|
|
|
|
// and using the same version of Go.
|
|
|
|
//
|
|
|
|
// TODO(sean@): Why is it not identical everywhere when using the
|
|
|
|
// same compiler?
|
2016-03-30 08:05:49 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
udpAnswerLimit int
|
|
|
|
expectedAService int
|
|
|
|
expectedAQuery int
|
|
|
|
expectedAQueryID int
|
|
|
|
expectedAAAAService int
|
|
|
|
expectedAAAAQuery int
|
|
|
|
expectedAAAAQueryID int
|
|
|
|
expectedANYService int
|
|
|
|
expectedANYQuery int
|
|
|
|
expectedANYQueryID int
|
|
|
|
}{
|
|
|
|
{"0", 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
|
|
|
|
{"1", 1, 1, 1, 1, 1, 1, 1, 1, 1, 1},
|
|
|
|
{"2", 2, 2, 2, 2, 2, 2, 2, 2, 2, 2},
|
|
|
|
{"3", 3, 3, 3, 3, 3, 3, 3, 3, 3, 3},
|
|
|
|
{"4", 4, 4, 4, 4, 4, 4, 4, 4, 4, 4},
|
|
|
|
{"5", 5, 5, 5, 5, 5, 5, 5, 5, 5, 5},
|
2016-03-30 18:48:32 +00:00
|
|
|
{"6", 6, 6, 6, 6, 6, 6, 5, 6, 6, -5},
|
|
|
|
{"7", 7, 7, 7, 6, 7, 7, 5, 7, 7, -5},
|
|
|
|
{"8", 8, 8, 8, 6, 8, 8, 5, 8, 8, -5},
|
|
|
|
{"9", 9, 8, 8, 6, 8, 8, 5, 8, 8, -5},
|
|
|
|
{"20", 20, 8, 8, 6, 8, 8, 5, 8, -5, -5},
|
|
|
|
{"30", 30, 8, 8, 6, 8, 8, 5, 8, -5, -5},
|
2016-03-30 08:05:49 +00:00
|
|
|
}
|
|
|
|
for _, test := range tests {
|
2017-05-23 00:41:51 +00:00
|
|
|
test := test // capture loop var
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Run(fmt.Sprintf("A lookup %v", test), func(t *testing.T) {
|
2017-05-23 00:41:51 +00:00
|
|
|
t.Parallel()
|
2018-01-28 18:40:13 +00:00
|
|
|
ok, err := testDNSServiceLookupResponseLimits(t, test.udpAnswerLimit, dns.TypeA, test.expectedAService, test.expectedAQuery, test.expectedAQueryID)
|
2017-05-23 00:41:51 +00:00
|
|
|
if !ok {
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Fatalf("Expected service A lookup %s to pass: %v", test.name, err)
|
2017-05-23 00:41:51 +00:00
|
|
|
}
|
|
|
|
})
|
2015-08-25 19:54:11 +00:00
|
|
|
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Run(fmt.Sprintf("AAAA lookup %v", test), func(t *testing.T) {
|
2017-05-23 00:41:51 +00:00
|
|
|
t.Parallel()
|
2018-01-28 18:40:13 +00:00
|
|
|
ok, err := testDNSServiceLookupResponseLimits(t, test.udpAnswerLimit, dns.TypeAAAA, test.expectedAAAAService, test.expectedAAAAQuery, test.expectedAAAAQueryID)
|
2017-05-23 00:41:51 +00:00
|
|
|
if !ok {
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Fatalf("Expected service AAAA lookup %s to pass: %v", test.name, err)
|
2017-05-23 00:41:51 +00:00
|
|
|
}
|
|
|
|
})
|
2015-08-25 19:54:11 +00:00
|
|
|
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Run(fmt.Sprintf("ANY lookup %v", test), func(t *testing.T) {
|
2017-05-23 00:41:51 +00:00
|
|
|
t.Parallel()
|
2018-01-28 18:40:13 +00:00
|
|
|
ok, err := testDNSServiceLookupResponseLimits(t, test.udpAnswerLimit, dns.TypeANY, test.expectedANYService, test.expectedANYQuery, test.expectedANYQueryID)
|
2017-05-23 00:41:51 +00:00
|
|
|
if !ok {
|
2018-04-18 12:18:17 +00:00
|
|
|
t.Fatalf("Expected service ANY lookup %s to pass: %v", test.name, err)
|
2017-05-23 00:41:51 +00:00
|
|
|
}
|
|
|
|
})
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2015-08-25 19:54:11 +00:00
|
|
|
}
|
|
|
|
|
2014-02-25 20:07:20 +00:00
|
|
|
func TestDNS_ServiceLookup_CNAME(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
dnsCNAME("www.google.com", "google.com"),
|
|
|
|
dnsA("google.com", "1.2.3.4"),
|
|
|
|
},
|
2015-04-14 02:19:22 +00:00
|
|
|
})
|
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-05-07 21:47:16 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a node with a name for an address.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "google",
|
|
|
|
Address: "www.google.com",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "search",
|
|
|
|
Port: 80,
|
|
|
|
},
|
|
|
|
}
|
2014-05-07 09:43:42 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "search",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"search.service.consul.",
|
|
|
|
id + ".query.consul.",
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeANY)
|
2014-02-25 20:07:20 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Service CNAME, google CNAME, google A record
|
|
|
|
if len(in.Answer) != 3 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should have service CNAME
|
|
|
|
cnRec, ok := in.Answer[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if cnRec.Target != "www.google.com." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should have google CNAME
|
|
|
|
cnRec, ok = in.Answer[1].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[1])
|
|
|
|
}
|
|
|
|
if cnRec.Target != "google.com." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[1])
|
|
|
|
}
|
2015-04-14 02:19:22 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Check we recursively resolve
|
|
|
|
if _, ok := in.Answer[2].(*dns.A); !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[2])
|
|
|
|
}
|
2014-02-25 20:07:20 +00:00
|
|
|
}
|
|
|
|
}
|
2014-06-08 23:14:21 +00:00
|
|
|
|
|
|
|
func TestDNS_NodeLookup_TTL(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
dnsCNAME("www.google.com", "google.com"),
|
|
|
|
dnsA("google.com", "1.2.3.4"),
|
|
|
|
},
|
2015-04-14 02:19:22 +00:00
|
|
|
})
|
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
dns_config {
|
|
|
|
node_ttl = "10s"
|
|
|
|
allow_stale = true
|
|
|
|
max_stale = "1s"
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-06-08 23:14:21 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-06-08 23:14:21 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.node.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-06-08 23:14:21 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register node with IPv6
|
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "::4242:4242",
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-06-08 23:14:21 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check an IPv6 record
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("bar.node.consul.", dns.TypeANY)
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2014-06-08 23:14:21 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aaaaRec, ok := in.Answer[0].(*dns.AAAA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aaaaRec.AAAA.String() != "::4242:4242" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aaaaRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register node with CNAME
|
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "google",
|
|
|
|
Address: "www.google.com",
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-06-08 23:14:21 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("google.node.consul.", dns.TypeANY)
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2014-06-08 23:14:21 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Should have the CNAME record + a few A records
|
|
|
|
if len(in.Answer) < 2 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
cnRec, ok := in.Answer[0].(*dns.CNAME)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if cnRec.Target != "www.google.com." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if cnRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_TTL(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
service_ttl = {
|
|
|
|
"db" = "10s"
|
|
|
|
"*" = "5s"
|
|
|
|
}
|
|
|
|
allow_stale = true
|
|
|
|
max_stale = "1s"
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2014-06-08 23:14:21 +00:00
|
|
|
|
|
|
|
// Register node with 2 services
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-06-08 23:14:21 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "api",
|
|
|
|
Port: 2222,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-06-08 23:14:21 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2014-06-08 23:14:21 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("api.service.consul.", dns.TypeSRV)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2014-06-08 23:14:21 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok = in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 5 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok = in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 5 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
2014-08-18 19:45:56 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
func TestDNS_PreparedQuery_TTL(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
service_ttl = {
|
|
|
|
"db" = "10s"
|
|
|
|
"*" = "5s"
|
|
|
|
}
|
|
|
|
allow_stale = true
|
|
|
|
max_stale = "1s"
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
// Register a node and a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
|
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "api",
|
|
|
|
Port: 2222,
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 18:38:44 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 18:38:44 +00:00
|
|
|
// Register prepared queries with and without a TTL set for "db", as
|
|
|
|
// well as one for "api".
|
2015-11-13 11:39:07 +00:00
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2015-11-13 18:38:44 +00:00
|
|
|
Name: "db-ttl",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
DNS: structs.QueryDNSOptions{
|
2015-11-13 18:38:44 +00:00
|
|
|
TTL: "18s",
|
2015-11-13 11:39:07 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var id string
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
args = &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2015-11-13 18:38:44 +00:00
|
|
|
Name: "db-nottl",
|
2015-11-13 11:39:07 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
|
|
|
|
args = &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "api-nottl",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "api",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 18:38:44 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 18:38:44 +00:00
|
|
|
// Make sure the TTL is set when requested, and overrides the agent-
|
|
|
|
// specific config since the query takes precedence.
|
2015-11-13 11:39:07 +00:00
|
|
|
m := new(dns.Msg)
|
2015-11-13 18:38:44 +00:00
|
|
|
m.SetQuestion("db-ttl.query.consul.", dns.TypeSRV)
|
2015-11-13 11:39:07 +00:00
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
if srvRec.Hdr.Ttl != 18 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
if aRec.Hdr.Ttl != 18 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
// And the TTL should take the service-specific value from the agent's
|
|
|
|
// config otherwise.
|
|
|
|
m = new(dns.Msg)
|
|
|
|
m.SetQuestion("db-nottl.query.consul.", dns.TypeSRV)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2015-11-13 18:38:44 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok = in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok = in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
if aRec.Hdr.Ttl != 10 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
|
2015-11-13 18:38:44 +00:00
|
|
|
// If there's no query TTL and no service-specific value then the wild
|
|
|
|
// card value should be used.
|
2015-11-13 11:39:07 +00:00
|
|
|
m = new(dns.Msg)
|
2015-11-13 18:38:44 +00:00
|
|
|
m.SetQuestion("api-nottl.query.consul.", dns.TypeSRV)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err = c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok = in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
if srvRec.Hdr.Ttl != 5 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok = in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2015-11-13 18:38:44 +00:00
|
|
|
if aRec.Hdr.Ttl != 5 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-07-27 22:11:42 +00:00
|
|
|
func TestDNS_PreparedQuery_Failover(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a1 := NewTestAgent(t.Name(), `
|
|
|
|
datacenter = "dc1"
|
|
|
|
translate_wan_addrs = true
|
|
|
|
acl_datacenter = ""
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a1.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a2 := NewTestAgent(t.Name(), `
|
|
|
|
datacenter = "dc2"
|
|
|
|
translate_wan_addrs = true
|
|
|
|
acl_datacenter = ""
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a2.Shutdown()
|
2016-07-27 22:11:42 +00:00
|
|
|
|
2016-08-13 00:16:08 +00:00
|
|
|
// Join WAN cluster.
|
2017-09-25 18:40:42 +00:00
|
|
|
addr := fmt.Sprintf("127.0.0.1:%d", a1.Config.SerfPortWAN)
|
2017-05-21 07:11:09 +00:00
|
|
|
if _, err := a2.JoinWAN([]string{addr}); err != nil {
|
2016-07-27 22:11:42 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2017-05-04 22:52:53 +00:00
|
|
|
retry.Run(t, func(r *retry.R) {
|
2017-05-21 07:11:09 +00:00
|
|
|
if got, want := len(a1.WANMembers()), 2; got < want {
|
2017-04-29 16:34:02 +00:00
|
|
|
r.Fatalf("got %d WAN members want at least %d", got, want)
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if got, want := len(a2.WANMembers()), 2; got < want {
|
2017-05-09 04:57:06 +00:00
|
|
|
r.Fatalf("got %d WAN members want at least %d", got, want)
|
|
|
|
}
|
2017-04-29 16:34:02 +00:00
|
|
|
})
|
2016-07-27 22:11:42 +00:00
|
|
|
|
2017-05-09 04:57:06 +00:00
|
|
|
// Register a remote node with a service. This is in a retry since we
|
|
|
|
// need the datacenter to have a route which takes a little more time
|
|
|
|
// beyond the join, and we don't have direct access to the router here.
|
|
|
|
retry.Run(t, func(r *retry.R) {
|
2016-07-27 22:11:42 +00:00
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc2",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
TaggedAddresses: map[string]string{
|
|
|
|
"wan": "127.0.0.2",
|
|
|
|
},
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a2.RPC("Catalog.Register", args, &out); err != nil {
|
2017-05-09 04:57:06 +00:00
|
|
|
r.Fatalf("err: %v", err)
|
2016-07-27 22:11:42 +00:00
|
|
|
}
|
2017-05-09 04:57:06 +00:00
|
|
|
})
|
2016-07-27 22:11:42 +00:00
|
|
|
|
2016-08-13 00:16:08 +00:00
|
|
|
// Register a local prepared query.
|
2016-07-27 22:11:42 +00:00
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "my-query",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
Failover: structs.QueryDatacenterOptions{
|
|
|
|
Datacenters: []string{"dc2"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2016-08-13 00:16:08 +00:00
|
|
|
var id string
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a1.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2016-07-27 22:11:42 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-08-13 00:16:08 +00:00
|
|
|
// Look up the SRV record via the query.
|
2016-07-27 22:11:42 +00:00
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("my-query.query.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2018-01-28 18:40:13 +00:00
|
|
|
clAddr := a1.config.DNSAddrs[0]
|
|
|
|
in, _, err := c.Exchange(m, clAddr.String())
|
2016-07-27 22:11:42 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2016-08-13 00:16:08 +00:00
|
|
|
// Make sure we see the remote DC and that the address gets
|
|
|
|
// translated.
|
2016-07-27 22:11:42 +00:00
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2016-08-13 00:16:08 +00:00
|
|
|
if in.Answer[0].Header().Name != "my-query.query.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
srv, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2016-10-28 02:14:02 +00:00
|
|
|
if srv.Target != "7f000002.addr.dc2.consul." {
|
2016-08-13 00:16:08 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
2016-07-27 22:11:42 +00:00
|
|
|
|
2016-08-13 00:16:08 +00:00
|
|
|
a, ok := in.Extra[0].(*dns.A)
|
2016-07-27 22:11:42 +00:00
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2016-10-28 02:14:02 +00:00
|
|
|
if a.Hdr.Name != "7f000002.addr.dc2.consul." {
|
2016-07-27 22:11:42 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2016-08-13 00:16:08 +00:00
|
|
|
if a.A.String() != "127.0.0.2" {
|
2016-07-27 22:11:42 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-08-18 19:45:56 +00:00
|
|
|
func TestDNS_ServiceLookup_SRV_RFC(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-08-18 19:45:56 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-08-18 19:45:56 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
questions := []string{
|
|
|
|
"_db._master.service.dc1.consul.",
|
|
|
|
"_db._master.service.consul.",
|
|
|
|
"_db._master.dc1.consul.",
|
|
|
|
"_db._master.consul.",
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
2014-08-18 19:45:56 +00:00
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-01-30 18:36:48 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-08-18 19:45:56 +00:00
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
2017-01-30 18:36:48 +00:00
|
|
|
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_SRV_RFC_TCP_Default(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2014-08-18 19:45:56 +00:00
|
|
|
|
|
|
|
// Register node
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2014-08-18 19:45:56 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
questions := []string{
|
|
|
|
"_db._tcp.service.dc1.consul.",
|
|
|
|
"_db._tcp.service.consul.",
|
|
|
|
"_db._tcp.dc1.consul.",
|
|
|
|
"_db._tcp.consul.",
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
2014-08-18 19:45:56 +00:00
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-01-30 18:36:48 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2014-08-18 19:45:56 +00:00
|
|
|
|
2017-01-30 18:36:48 +00:00
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
srvRec, ok := in.Answer[0].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if srvRec.Port != 12345 {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Target != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", srvRec)
|
|
|
|
}
|
|
|
|
if srvRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Extra[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Name != "foo.node.dc1.consul." {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.A.String() != "127.0.0.1" {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Extra[0])
|
|
|
|
}
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
2017-01-30 18:36:48 +00:00
|
|
|
|
2014-08-18 19:45:56 +00:00
|
|
|
}
|
2015-06-12 22:59:18 +00:00
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_FilterACL(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-30 20:05:23 +00:00
|
|
|
tests := []struct {
|
|
|
|
token string
|
|
|
|
results int
|
|
|
|
}{
|
|
|
|
{"root", 1},
|
|
|
|
{"anonymous", 0},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run("ACLToken == "+tt.token, func(t *testing.T) {
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
acl_token = "`+tt.token+`"
|
|
|
|
acl_master_token = "root"
|
|
|
|
acl_datacenter = "dc1"
|
|
|
|
acl_down_policy = "deny"
|
|
|
|
acl_default_policy = "deny"
|
|
|
|
`)
|
2017-06-30 20:05:23 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2017-06-30 20:05:23 +00:00
|
|
|
|
|
|
|
// Register a service
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "foo",
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
WriteRequest: structs.WriteRequest{Token: "root"},
|
|
|
|
}
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-06-12 22:59:18 +00:00
|
|
|
|
2017-06-30 20:05:23 +00:00
|
|
|
// Set up the DNS query
|
|
|
|
c := new(dns.Client)
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.service.consul.", dns.TypeA)
|
2015-06-12 22:59:18 +00:00
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2017-06-30 20:05:23 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
if len(in.Answer) != tt.results {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
})
|
2015-06-12 22:59:18 +00:00
|
|
|
}
|
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
|
2018-06-11 15:49:04 +00:00
|
|
|
func TestDNS_ServiceLookup_MetaTXT(t *testing.T) {
|
2018-06-19 14:08:16 +00:00
|
|
|
a := NewTestAgent(t.Name(), `dns_config = { enable_additional_node_meta_txt = true }`)
|
2018-06-11 15:49:04 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-06-11 15:49:04 +00:00
|
|
|
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
wantAdditional := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.dc1.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
|
|
|
&dns.TXT{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.dc1.consul.", Rrtype: dns.TypeTXT, Class: dns.ClassINET, Rdlength: 0xa},
|
|
|
|
Txt: []string{"key=value"},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "additional", in.Extra, wantAdditional)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_ServiceLookup_SuppressTXT(t *testing.T) {
|
2018-06-19 14:08:16 +00:00
|
|
|
a := NewTestAgent(t.Name(), `dns_config = { enable_additional_node_meta_txt = false }`)
|
2018-06-11 15:49:04 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2018-06-11 15:49:04 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "bar",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
NodeMeta: map[string]string{
|
|
|
|
"key": "value",
|
|
|
|
},
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("db.service.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
wantAdditional := []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{Name: "bar.node.dc1.consul.", Rrtype: dns.TypeA, Class: dns.ClassINET, Rdlength: 0x4},
|
|
|
|
A: []byte{0x7f, 0x0, 0x0, 0x1}, // 127.0.0.1
|
|
|
|
},
|
|
|
|
}
|
|
|
|
verify.Values(t, "additional", in.Extra, wantAdditional)
|
|
|
|
}
|
|
|
|
|
2016-10-28 03:41:24 +00:00
|
|
|
func TestDNS_AddressLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-28 03:41:24 +00:00
|
|
|
|
|
|
|
// Look up the addresses
|
|
|
|
cases := map[string]string{
|
|
|
|
"7f000001.addr.dc1.consul.": "127.0.0.1",
|
|
|
|
}
|
|
|
|
for question, answer := range cases {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-28 03:41:24 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aRec, ok := in.Answer[0].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aRec.A.To4().String() != answer {
|
|
|
|
t.Fatalf("Bad: %#v", aRec)
|
|
|
|
}
|
|
|
|
if aRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_AddressLookupIPV6(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-10-28 03:41:24 +00:00
|
|
|
|
|
|
|
// Look up the addresses
|
|
|
|
cases := map[string]string{
|
|
|
|
"2607002040050808000000000000200e.addr.consul.": "2607:20:4005:808::200e",
|
|
|
|
"2607112040051808ffffffffffff200e.addr.consul.": "2607:1120:4005:1808:ffff:ffff:ffff:200e",
|
|
|
|
}
|
|
|
|
for question, answer := range cases {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2016-10-28 03:41:24 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(in.Answer) != 1 {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
aaaaRec, ok := in.Answer[0].(*dns.AAAA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
if aaaaRec.AAAA.To16().String() != answer {
|
|
|
|
t.Fatalf("Bad: %#v", aaaaRec)
|
|
|
|
}
|
|
|
|
if aaaaRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Answer[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-06-02 21:47:18 +00:00
|
|
|
func TestDNS_NonExistingLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-06-02 21:47:18 +00:00
|
|
|
|
|
|
|
// lookup a non-existing node, we should receive a SOA
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("nonexisting.consul.", dns.TypeANY)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-06-02 21:47:18 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-07-14 18:42:51 +00:00
|
|
|
t.Fatalf("Bad: %#v %#v", in, len(in.Answer))
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_NonExistingLookupEmptyAorAAAA(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register a v6-only service and a v4-only service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foov6",
|
|
|
|
Address: "fe80::1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "webv6",
|
|
|
|
Port: 8000,
|
|
|
|
},
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foov4",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "webv4",
|
|
|
|
Port: 8000,
|
|
|
|
},
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Register equivalent prepared queries.
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "webv4",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "webv4",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
var id string
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
args = &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
|
|
|
Name: "webv6",
|
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "webv6",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Check for ipv6 records on ipv4-only service directly and via the
|
|
|
|
// prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"webv4.service.consul.",
|
|
|
|
"webv4.query.consul.",
|
2015-09-02 14:12:22 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeAAAA)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
2015-09-02 14:12:22 +00:00
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2017-08-04 17:17:04 +00:00
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
2015-11-13 11:39:07 +00:00
|
|
|
if !ok {
|
2017-08-04 17:17:04 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
2017-08-04 17:17:04 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
2015-11-13 11:39:07 +00:00
|
|
|
}
|
2015-08-25 20:37:33 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
if in.Rcode != dns.RcodeSuccess {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
|
2015-08-25 20:37:33 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
// Check for ipv4 records on ipv6-only service directly and via the
|
|
|
|
// prepared query.
|
|
|
|
questions = []string{
|
|
|
|
"webv6.service.consul.",
|
|
|
|
"webv6.query.consul.",
|
2015-08-25 20:37:33 +00:00
|
|
|
}
|
2015-11-13 11:39:07 +00:00
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeA)
|
2015-09-02 14:12:22 +00:00
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-13 11:39:07 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-11-13 11:39:07 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
2015-11-13 11:39:07 +00:00
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
2017-08-04 17:17:04 +00:00
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
2017-08-02 21:44:40 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 11:39:07 +00:00
|
|
|
if in.Rcode != dns.RcodeSuccess {
|
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
}
|
2015-06-02 21:47:18 +00:00
|
|
|
}
|
2015-11-14 01:18:15 +00:00
|
|
|
|
|
|
|
func TestDNS_PreparedQuery_AllowStale(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
dns_config {
|
|
|
|
allow_stale = true
|
|
|
|
max_stale = "1s"
|
|
|
|
}
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-14 01:18:15 +00:00
|
|
|
|
2017-07-05 10:38:11 +00:00
|
|
|
m := MockPreparedQuery{
|
|
|
|
executeFn: func(args *structs.PreparedQueryExecuteRequest, reply *structs.PreparedQueryExecuteResponse) error {
|
|
|
|
// Return a response that's perpetually too stale.
|
|
|
|
reply.LastContact = 2 * time.Second
|
|
|
|
return nil
|
|
|
|
},
|
2015-11-14 01:18:15 +00:00
|
|
|
}
|
|
|
|
|
2017-07-05 10:38:11 +00:00
|
|
|
if err := a.registerEndpoint("PreparedQuery", &m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
2015-11-14 01:18:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure that the lookup terminates and results in an SOA since
|
|
|
|
// the query doesn't exist.
|
|
|
|
{
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("nope.query.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-14 01:18:15 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-11-14 01:18:15 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
|
2015-11-14 01:18:15 +00:00
|
|
|
}
|
|
|
|
}
|
2015-11-17 16:40:47 +00:00
|
|
|
|
|
|
|
func TestDNS_InvalidQueries(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2015-11-17 16:40:47 +00:00
|
|
|
|
|
|
|
// Try invalid forms of queries that should hit the special invalid case
|
|
|
|
// of our query parser.
|
|
|
|
questions := []string{
|
|
|
|
"consul.",
|
|
|
|
"node.consul.",
|
|
|
|
"service.consul.",
|
|
|
|
"query.consul.",
|
2017-10-20 23:49:17 +00:00
|
|
|
"foo.node.dc1.extra.consul.",
|
|
|
|
"foo.service.dc1.extra.consul.",
|
|
|
|
"foo.query.dc1.extra.consul.",
|
2015-11-17 16:40:47 +00:00
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
in, _, err := c.Exchange(m, a.DNSAddr())
|
2015-11-17 16:40:47 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
2017-08-04 17:17:04 +00:00
|
|
|
if len(in.Ns) != 1 {
|
2015-11-17 16:40:47 +00:00
|
|
|
t.Fatalf("Bad: %#v", in)
|
|
|
|
}
|
|
|
|
|
|
|
|
soaRec, ok := in.Ns[0].(*dns.SOA)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
|
|
|
if soaRec.Hdr.Ttl != 0 {
|
|
|
|
t.Fatalf("Bad: %#v", in.Ns[0])
|
|
|
|
}
|
2017-08-02 21:44:40 +00:00
|
|
|
|
2015-11-17 16:40:47 +00:00
|
|
|
}
|
|
|
|
}
|
2016-07-01 16:46:26 +00:00
|
|
|
|
|
|
|
func TestDNS_PreparedQuery_AgentSource(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-07-01 16:46:26 +00:00
|
|
|
|
2017-07-05 10:38:11 +00:00
|
|
|
m := MockPreparedQuery{
|
|
|
|
executeFn: func(args *structs.PreparedQueryExecuteRequest, reply *structs.PreparedQueryExecuteResponse) error {
|
|
|
|
// Check that the agent inserted its self-name and datacenter to
|
|
|
|
// the RPC request body.
|
|
|
|
if args.Agent.Datacenter != a.Config.Datacenter ||
|
|
|
|
args.Agent.Node != a.Config.NodeName {
|
|
|
|
t.Fatalf("bad: %#v", args.Agent)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
},
|
2016-07-01 16:46:26 +00:00
|
|
|
}
|
|
|
|
|
2017-07-05 10:38:11 +00:00
|
|
|
if err := a.registerEndpoint("PreparedQuery", &m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
2016-07-01 16:46:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
{
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("foo.query.consul.", dns.TypeSRV)
|
|
|
|
|
|
|
|
c := new(dns.Client)
|
2017-09-25 18:40:42 +00:00
|
|
|
if _, _, err := c.Exchange(m, a.DNSAddr()); err != nil {
|
2016-07-01 16:46:26 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-08-11 23:24:44 +00:00
|
|
|
|
2016-08-12 04:46:14 +00:00
|
|
|
func TestDNS_trimUDPResponse_NoTrim(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-14 23:22:54 +00:00
|
|
|
req := &dns.Msg{}
|
2016-08-12 04:46:14 +00:00
|
|
|
resp := &dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Extra: []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.185"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
cfg := config.DefaultRuntimeConfig(`data_dir = "a" bind_addr = "127.0.0.1"`)
|
|
|
|
if trimmed := trimUDPResponse(req, resp, cfg.DNSUDPAnswerLimit); trimmed {
|
2016-08-12 04:46:14 +00:00
|
|
|
t.Fatalf("Bad %#v", *resp)
|
|
|
|
}
|
|
|
|
|
|
|
|
expected := &dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Extra: []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.185"),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(resp, expected) {
|
|
|
|
t.Fatalf("Bad %#v vs. %#v", *resp, *expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_trimUDPResponse_TrimLimit(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
cfg := config.DefaultRuntimeConfig(`data_dir = "a" bind_addr = "127.0.0.1"`)
|
2016-08-12 04:46:14 +00:00
|
|
|
|
2017-06-14 23:22:54 +00:00
|
|
|
req, resp, expected := &dns.Msg{}, &dns.Msg{}, &dns.Msg{}
|
2017-09-25 18:40:42 +00:00
|
|
|
for i := 0; i < cfg.DNSUDPAnswerLimit+1; i++ {
|
2016-08-12 04:46:14 +00:00
|
|
|
target := fmt.Sprintf("ip-10-0-1-%d.node.dc1.consul.", 185+i)
|
|
|
|
srv := &dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: target,
|
|
|
|
}
|
|
|
|
a := &dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: target,
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP(fmt.Sprintf("10.0.1.%d", 185+i)),
|
|
|
|
}
|
|
|
|
|
|
|
|
resp.Answer = append(resp.Answer, srv)
|
|
|
|
resp.Extra = append(resp.Extra, a)
|
2017-09-25 18:40:42 +00:00
|
|
|
if i < cfg.DNSUDPAnswerLimit {
|
2016-08-12 04:46:14 +00:00
|
|
|
expected.Answer = append(expected.Answer, srv)
|
|
|
|
expected.Extra = append(expected.Extra, a)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
if trimmed := trimUDPResponse(req, resp, cfg.DNSUDPAnswerLimit); !trimmed {
|
2016-08-12 04:46:14 +00:00
|
|
|
t.Fatalf("Bad %#v", *resp)
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(resp, expected) {
|
|
|
|
t.Fatalf("Bad %#v vs. %#v", *resp, *expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_trimUDPResponse_TrimSize(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
cfg := config.DefaultRuntimeConfig(`data_dir = "a" bind_addr = "127.0.0.1"`)
|
2016-08-12 04:46:14 +00:00
|
|
|
|
2017-06-14 23:22:54 +00:00
|
|
|
req, resp := &dns.Msg{}, &dns.Msg{}
|
2016-08-12 04:46:14 +00:00
|
|
|
for i := 0; i < 100; i++ {
|
|
|
|
target := fmt.Sprintf("ip-10-0-1-%d.node.dc1.consul.", 185+i)
|
|
|
|
srv := &dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: target,
|
|
|
|
}
|
|
|
|
a := &dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: target,
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP(fmt.Sprintf("10.0.1.%d", 185+i)),
|
|
|
|
}
|
|
|
|
|
|
|
|
resp.Answer = append(resp.Answer, srv)
|
|
|
|
resp.Extra = append(resp.Extra, a)
|
|
|
|
}
|
|
|
|
|
|
|
|
// We don't know the exact trim, but we know the resulting answer
|
|
|
|
// data should match its extra data.
|
2017-09-25 18:40:42 +00:00
|
|
|
if trimmed := trimUDPResponse(req, resp, cfg.DNSUDPAnswerLimit); !trimmed {
|
2016-08-12 04:46:14 +00:00
|
|
|
t.Fatalf("Bad %#v", *resp)
|
|
|
|
}
|
|
|
|
if len(resp.Answer) == 0 || len(resp.Answer) != len(resp.Extra) {
|
|
|
|
t.Fatalf("Bad %#v", *resp)
|
|
|
|
}
|
2017-04-20 18:42:22 +00:00
|
|
|
for i := range resp.Answer {
|
2016-08-12 04:46:14 +00:00
|
|
|
srv, ok := resp.Answer[i].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("should be SRV")
|
|
|
|
}
|
|
|
|
|
|
|
|
a, ok := resp.Extra[i].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Fatalf("should be A")
|
|
|
|
}
|
|
|
|
|
|
|
|
if srv.Target != a.Header().Name {
|
|
|
|
t.Fatalf("Bad %#v vs. %#v", *srv, *a)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-14 23:22:54 +00:00
|
|
|
func TestDNS_trimUDPResponse_TrimSizeEDNS(t *testing.T) {
|
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
cfg := config.DefaultRuntimeConfig(`data_dir = "a" bind_addr = "127.0.0.1"`)
|
2017-06-14 23:22:54 +00:00
|
|
|
|
|
|
|
req, resp := &dns.Msg{}, &dns.Msg{}
|
|
|
|
|
|
|
|
for i := 0; i < 100; i++ {
|
|
|
|
target := fmt.Sprintf("ip-10-0-1-%d.node.dc1.consul.", 150+i)
|
|
|
|
srv := &dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: target,
|
|
|
|
}
|
|
|
|
a := &dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: target,
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP(fmt.Sprintf("10.0.1.%d", 150+i)),
|
|
|
|
}
|
|
|
|
|
|
|
|
resp.Answer = append(resp.Answer, srv)
|
|
|
|
resp.Extra = append(resp.Extra, a)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Copy over to a new slice since we are trimming both.
|
|
|
|
reqEDNS, respEDNS := &dns.Msg{}, &dns.Msg{}
|
|
|
|
reqEDNS.SetEdns0(2048, true)
|
|
|
|
respEDNS.Answer = append(respEDNS.Answer, resp.Answer...)
|
|
|
|
respEDNS.Extra = append(respEDNS.Extra, resp.Extra...)
|
|
|
|
|
|
|
|
// Trim each response
|
2017-09-25 18:40:42 +00:00
|
|
|
if trimmed := trimUDPResponse(req, resp, cfg.DNSUDPAnswerLimit); !trimmed {
|
2017-06-14 23:22:54 +00:00
|
|
|
t.Errorf("expected response to be trimmed: %#v", resp)
|
|
|
|
}
|
2017-09-25 18:40:42 +00:00
|
|
|
if trimmed := trimUDPResponse(reqEDNS, respEDNS, cfg.DNSUDPAnswerLimit); !trimmed {
|
2017-06-14 23:22:54 +00:00
|
|
|
t.Errorf("expected edns to be trimmed: %#v", resp)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check answer lengths
|
|
|
|
if len(resp.Answer) == 0 || len(resp.Answer) != len(resp.Extra) {
|
|
|
|
t.Errorf("bad response answer length: %#v", resp)
|
|
|
|
}
|
|
|
|
if len(respEDNS.Answer) == 0 || len(respEDNS.Answer) != len(respEDNS.Extra) {
|
|
|
|
t.Errorf("bad edns answer length: %#v", resp)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Due to the compression, we can't check exact equality of sizes, but we can
|
|
|
|
// make two requests and ensure that the edns one returns a larger payload
|
|
|
|
// than the non-edns0 one.
|
|
|
|
if len(resp.Answer) >= len(respEDNS.Answer) {
|
|
|
|
t.Errorf("expected edns have larger answer: %#v\n%#v", resp, respEDNS)
|
|
|
|
}
|
|
|
|
if len(resp.Extra) >= len(respEDNS.Extra) {
|
|
|
|
t.Errorf("expected edns have larger extra: %#v\n%#v", resp, respEDNS)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify that the things point where they should
|
|
|
|
for i := range resp.Answer {
|
|
|
|
srv, ok := resp.Answer[i].(*dns.SRV)
|
|
|
|
if !ok {
|
|
|
|
t.Errorf("%d should be an SRV", i)
|
|
|
|
}
|
|
|
|
|
|
|
|
a, ok := resp.Extra[i].(*dns.A)
|
|
|
|
if !ok {
|
|
|
|
t.Errorf("%d should be an A", i)
|
|
|
|
}
|
|
|
|
|
|
|
|
if srv.Target != a.Header().Name {
|
|
|
|
t.Errorf("%d: bad %#v vs. %#v", i, srv, a)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-08-12 04:46:14 +00:00
|
|
|
func TestDNS_syncExtra(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2016-08-12 04:46:14 +00:00
|
|
|
resp := &dns.Msg{
|
|
|
|
Answer: []dns.RR{
|
|
|
|
// These two are on the same host so the redundant extra
|
|
|
|
// records should get deduplicated.
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1001,
|
|
|
|
Target: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
},
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1002,
|
|
|
|
Target: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
},
|
|
|
|
// This one isn't in the Consul domain so it will get a
|
|
|
|
// CNAME and then an A record from the recursor.
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1003,
|
|
|
|
Target: "demo.consul.io.",
|
|
|
|
},
|
2016-08-12 19:16:21 +00:00
|
|
|
// This one isn't in the Consul domain and it will get
|
|
|
|
// a CNAME and A record from a recursor that alters the
|
|
|
|
// case of the name. This proves we look up in the index
|
|
|
|
// in a case-insensitive way.
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1001,
|
|
|
|
Target: "insensitive.consul.io.",
|
|
|
|
},
|
2016-08-12 04:46:14 +00:00
|
|
|
// This is also a CNAME, but it'll be set up to loop to
|
|
|
|
// make sure we don't crash.
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1001,
|
|
|
|
Target: "deadly.consul.io.",
|
|
|
|
},
|
|
|
|
// This is also a CNAME, but it won't have another record.
|
|
|
|
&dns.SRV{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "redis-cache-redis.service.consul.",
|
|
|
|
Rrtype: dns.TypeSRV,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Port: 1001,
|
|
|
|
Target: "nope.consul.io.",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Extra: []dns.RR{
|
|
|
|
// These should get deduplicated.
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.185"),
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.185"),
|
|
|
|
},
|
|
|
|
// This is a normal CNAME followed by an A record but we
|
|
|
|
// have flipped the order. The algorithm should emit them
|
|
|
|
// in the opposite order.
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "fakeserver.consul.io.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("127.0.0.1"),
|
|
|
|
},
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "demo.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "fakeserver.consul.io.",
|
|
|
|
},
|
2016-08-12 19:16:21 +00:00
|
|
|
// These differ in case to test case insensitivity.
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "INSENSITIVE.CONSUL.IO.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "Another.Server.Com.",
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "another.server.com.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("127.0.0.1"),
|
|
|
|
},
|
2016-08-12 04:46:14 +00:00
|
|
|
// This doesn't appear in the answer, so should get
|
|
|
|
// dropped.
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-186.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.186"),
|
|
|
|
},
|
|
|
|
// These two test edge cases with CNAME handling.
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "deadly.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "deadly.consul.io.",
|
|
|
|
},
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "nope.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "notthere.consul.io.",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2016-08-12 21:51:50 +00:00
|
|
|
index := make(map[string]dns.RR)
|
|
|
|
indexRRs(resp.Extra, index)
|
2016-08-12 04:46:14 +00:00
|
|
|
syncExtra(index, resp)
|
|
|
|
|
|
|
|
expected := &dns.Msg{
|
|
|
|
Answer: resp.Answer,
|
|
|
|
Extra: []dns.RR{
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "ip-10-0-1-185.node.dc1.consul.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("10.0.1.185"),
|
|
|
|
},
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "demo.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "fakeserver.consul.io.",
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "fakeserver.consul.io.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("127.0.0.1"),
|
|
|
|
},
|
2016-08-12 19:16:21 +00:00
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "INSENSITIVE.CONSUL.IO.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "Another.Server.Com.",
|
|
|
|
},
|
|
|
|
&dns.A{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "another.server.com.",
|
|
|
|
Rrtype: dns.TypeA,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
A: net.ParseIP("127.0.0.1"),
|
|
|
|
},
|
2016-08-12 04:46:14 +00:00
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "deadly.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "deadly.consul.io.",
|
|
|
|
},
|
|
|
|
&dns.CNAME{
|
|
|
|
Hdr: dns.RR_Header{
|
|
|
|
Name: "nope.consul.io.",
|
|
|
|
Rrtype: dns.TypeCNAME,
|
|
|
|
Class: dns.ClassINET,
|
|
|
|
},
|
|
|
|
Target: "notthere.consul.io.",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(resp, expected) {
|
|
|
|
t.Fatalf("Bad %#v vs. %#v", *resp, *expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_Compression_trimUDPResponse(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
cfg := config.DefaultRuntimeConfig(`data_dir = "a" bind_addr = "127.0.0.1"`)
|
2016-08-11 23:24:44 +00:00
|
|
|
|
2017-06-14 23:22:54 +00:00
|
|
|
req, m := dns.Msg{}, dns.Msg{}
|
2017-09-25 18:40:42 +00:00
|
|
|
trimUDPResponse(&req, &m, cfg.DNSUDPAnswerLimit)
|
2016-08-11 23:24:44 +00:00
|
|
|
if m.Compress {
|
|
|
|
t.Fatalf("compression should be off")
|
|
|
|
}
|
|
|
|
|
|
|
|
// The trim function temporarily turns off compression, so we need to
|
|
|
|
// make sure the setting gets restored properly.
|
|
|
|
m.Compress = true
|
2017-09-25 18:40:42 +00:00
|
|
|
trimUDPResponse(&req, &m, cfg.DNSUDPAnswerLimit)
|
2016-08-11 23:24:44 +00:00
|
|
|
if !m.Compress {
|
|
|
|
t.Fatalf("compression should be on")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_Compression_Query(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-08-11 23:24:44 +00:00
|
|
|
|
|
|
|
// Register a node with a service.
|
|
|
|
{
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo",
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Service: &structs.NodeService{
|
|
|
|
Service: "db",
|
|
|
|
Tags: []string{"master"},
|
|
|
|
Port: 12345,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-08-11 23:24:44 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register an equivalent prepared query.
|
|
|
|
var id string
|
|
|
|
{
|
|
|
|
args := &structs.PreparedQueryRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Op: structs.PreparedQueryCreate,
|
|
|
|
Query: &structs.PreparedQuery{
|
2016-12-10 19:19:08 +00:00
|
|
|
Name: "test",
|
2016-08-11 23:24:44 +00:00
|
|
|
Service: structs.ServiceQuery{
|
|
|
|
Service: "db",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("PreparedQuery.Apply", args, &id); err != nil {
|
2016-08-11 23:24:44 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Look up the service directly and via prepared query.
|
|
|
|
questions := []string{
|
|
|
|
"db.service.consul.",
|
|
|
|
id + ".query.consul.",
|
|
|
|
}
|
|
|
|
for _, question := range questions {
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion(question, dns.TypeSRV)
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
conn, err := dns.Dial("udp", a.DNSAddr())
|
2016-08-11 23:24:44 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Do a manual exchange with compression on (the default).
|
2017-06-29 14:42:17 +00:00
|
|
|
a.DNSDisableCompression(false)
|
2016-08-11 23:24:44 +00:00
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
p := make([]byte, dns.MaxMsgSize)
|
|
|
|
compressed, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Disable compression and try again.
|
2017-06-29 14:42:17 +00:00
|
|
|
a.DNSDisableCompression(true)
|
2016-08-11 23:24:44 +00:00
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
unc, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// We can't see the compressed status given the DNS API, so we
|
|
|
|
// just make sure the message is smaller to see if it's
|
|
|
|
// respecting the flag.
|
|
|
|
if compressed == 0 || unc == 0 || compressed >= unc {
|
|
|
|
t.Fatalf("'%s' doesn't look compressed: %d vs. %d", question, compressed, unc)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_Compression_ReverseLookup(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), "")
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-08-11 23:24:44 +00:00
|
|
|
|
|
|
|
// Register node.
|
|
|
|
args := &structs.RegisterRequest{
|
|
|
|
Datacenter: "dc1",
|
|
|
|
Node: "foo2",
|
|
|
|
Address: "127.0.0.2",
|
|
|
|
}
|
|
|
|
var out struct{}
|
2017-05-21 07:11:09 +00:00
|
|
|
if err := a.RPC("Catalog.Register", args, &out); err != nil {
|
2016-08-11 23:24:44 +00:00
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("2.0.0.127.in-addr.arpa.", dns.TypeANY)
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
conn, err := dns.Dial("udp", a.DNSAddr())
|
2016-08-11 23:24:44 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Do a manual exchange with compression on (the default).
|
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
p := make([]byte, dns.MaxMsgSize)
|
|
|
|
compressed, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Disable compression and try again.
|
2017-06-29 14:42:17 +00:00
|
|
|
a.DNSDisableCompression(true)
|
2016-08-11 23:24:44 +00:00
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
unc, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// We can't see the compressed status given the DNS API, so we just make
|
|
|
|
// sure the message is smaller to see if it's respecting the flag.
|
|
|
|
if compressed == 0 || unc == 0 || compressed >= unc {
|
|
|
|
t.Fatalf("doesn't look compressed: %d vs. %d", compressed, unc)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestDNS_Compression_Recurse(t *testing.T) {
|
2017-05-21 07:54:40 +00:00
|
|
|
t.Parallel()
|
2017-06-25 17:42:37 +00:00
|
|
|
recursor := makeRecursor(t, dns.Msg{
|
|
|
|
Answer: []dns.RR{dnsA("apple.com", "1.2.3.4")},
|
|
|
|
})
|
2016-08-11 23:24:44 +00:00
|
|
|
defer recursor.Shutdown()
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
a := NewTestAgent(t.Name(), `
|
|
|
|
recursors = ["`+recursor.Addr+`"]
|
|
|
|
`)
|
2017-05-21 07:11:09 +00:00
|
|
|
defer a.Shutdown()
|
2018-08-06 23:46:09 +00:00
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
2016-08-11 23:24:44 +00:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
|
|
|
m.SetQuestion("apple.com.", dns.TypeANY)
|
|
|
|
|
2017-09-25 18:40:42 +00:00
|
|
|
conn, err := dns.Dial("udp", a.DNSAddr())
|
2016-08-11 23:24:44 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Do a manual exchange with compression on (the default).
|
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
p := make([]byte, dns.MaxMsgSize)
|
|
|
|
compressed, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Disable compression and try again.
|
2017-06-29 14:42:17 +00:00
|
|
|
a.DNSDisableCompression(true)
|
2016-08-11 23:24:44 +00:00
|
|
|
if err := conn.WriteMsg(m); err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
unc, err := conn.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// We can't see the compressed status given the DNS API, so we just make
|
|
|
|
// sure the message is smaller to see if it's respecting the flag.
|
|
|
|
if compressed == 0 || unc == 0 || compressed >= unc {
|
|
|
|
t.Fatalf("doesn't look compressed: %d vs. %d", compressed, unc)
|
|
|
|
}
|
|
|
|
}
|
2017-08-04 17:17:04 +00:00
|
|
|
|
|
|
|
func TestDNSInvalidRegex(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
|
|
desc string
|
|
|
|
in string
|
|
|
|
invalid bool
|
|
|
|
}{
|
|
|
|
{"Valid Hostname", "testnode", false},
|
|
|
|
{"Valid Hostname", "test-node", false},
|
|
|
|
{"Invalid Hostname with special chars", "test#$$!node", true},
|
2017-08-06 23:18:30 +00:00
|
|
|
{"Invalid Hostname with special chars in the end", "testnode%^", true},
|
|
|
|
{"Whitespace", " ", true},
|
|
|
|
{"Only special chars", "./$", true},
|
2017-08-04 17:17:04 +00:00
|
|
|
}
|
|
|
|
for _, test := range tests {
|
|
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
|
|
if got, want := InvalidDnsRe.MatchString(test.in), test.invalid; got != want {
|
|
|
|
t.Fatalf("Expected %v to return %v", test.in, want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|