github
/
consul
mirror of https://github.com/hashicorp/consul
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Check response len do not exceed max Buffer size

pull/10009/head
Dhia Ayachi 2021-06-18 13:23:13 -04:00 committed by Daniel Nephin
parent f8f2756967
commit ad2065f2aa
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
agent/dns_test.go
View File

@ -6751,8 +6751,13 @@ func TestDNS_EDNS_Truncate_AgentSource(t *testing.T) {
} }
t.Parallel() t.Parallel()
a := NewTestAgent(t, "") a := NewTestAgent(t, `
dns_config {
enable_truncate = true
}
`)
defer a.Shutdown() defer a.Shutdown()
a.DNSDisableCompression(true)
testrpc.WaitForLeader(t, a.RPC, "dc1") testrpc.WaitForLeader(t, a.RPC, "dc1")
m := MockPreparedQuery{ m := MockPreparedQuery{
@ -6778,11 +6783,12 @@ func TestDNS_EDNS_Truncate_AgentSource(t *testing.T) {
m := new(dns.Msg) m := new(dns.Msg)
m.SetQuestion("foo.query.consul.", dns.TypeSRV) m.SetQuestion("foo.query.consul.", dns.TypeSRV)
m.SetEdns0(2048, true) m.SetEdns0(2048, true)
m.Compress = false
c := new(dns.Client) c := new(dns.Client)
r, _, err := c.Exchange(m, a.DNSAddr()) r, _, err := c.Exchange(m, a.DNSAddr())
require.NoError(t, err) require.NoError(t, err)
require.Len(t, r.Answer, 53) require.True(t, r.Len() < 2048)
} }
} }