Check response len do not exceed max Buffer size

This commit is contained in:
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

View File

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