mirror of
https://github.com/coredns/coredns.git
synced 2025-12-06 02:15:11 -05:00
plugin/transfer: only allow outgoing axfr over tcp (#4452)
* plugin/transfer: only allow outgoing axfr over tcp Return refused when the query comes in over udp. No need to add a new test case as the current crop needed to be changed to use TCP. Fixes: #4450 Signed-off-by: Miek Gieben <miek@miek.nl> * transfer tests: this needs tcp as well Signed-off-by: Miek Gieben <miek@miek.nl>
This commit is contained in:
@@ -20,7 +20,7 @@ func (w *badwriter) WriteMsg(res *dns.Msg) error { return fmt.Errorf("failed to
|
||||
func TestWriteMessageFailed(t *testing.T) {
|
||||
transfer := newTestTransfer()
|
||||
ctx := context.TODO()
|
||||
w := &badwriter{ResponseWriter: &test.ResponseWriter{}}
|
||||
w := &badwriter{ResponseWriter: &test.ResponseWriter{TCP: true}}
|
||||
m := &dns.Msg{}
|
||||
m.SetAxfr("example.org.")
|
||||
|
||||
|
||||
@@ -47,7 +47,7 @@ func TestZoneSelection(t *testing.T) {
|
||||
}
|
||||
r := new(dns.Msg)
|
||||
r.SetAxfr("sub.example.org.")
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{TCP: true})
|
||||
_, err := tr.ServeDNS(context.TODO(), w, r)
|
||||
if err == nil {
|
||||
t.Fatal("Expected error, got nil")
|
||||
|
||||
@@ -58,6 +58,10 @@ func (t *Transfer) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Ms
|
||||
return plugin.NextOrFailure(t.Name(), t.Next, ctx, w, r)
|
||||
}
|
||||
|
||||
if state.Proto() != "tcp" {
|
||||
return dns.RcodeRefused, nil
|
||||
}
|
||||
|
||||
x := longestMatch(t.xfrs, state.QName())
|
||||
if x == nil {
|
||||
return plugin.NextOrFailure(t.Name(), t.Next, ctx, w, r)
|
||||
|
||||
@@ -91,7 +91,7 @@ func TestTransferNonZone(t *testing.T) {
|
||||
ctx := context.TODO()
|
||||
|
||||
for _, tc := range []string{"sub.example.org.", "example.test."} {
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetAxfr(tc)
|
||||
|
||||
@@ -114,7 +114,7 @@ func TestTransferNotAXFRorIXFR(t *testing.T) {
|
||||
transfer := newTestTransfer()
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetQuestion("test.domain.", dns.TypeA)
|
||||
|
||||
@@ -136,7 +136,7 @@ func TestTransferAXFRExampleOrg(t *testing.T) {
|
||||
transfer := newTestTransfer()
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetAxfr(transfer.xfrs[0].Zones[0])
|
||||
|
||||
@@ -152,7 +152,7 @@ func TestTransferAXFRExampleCom(t *testing.T) {
|
||||
transfer := newTestTransfer()
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetAxfr(transfer.xfrs[1].Zones[0])
|
||||
|
||||
@@ -170,7 +170,7 @@ func TestTransferIXFRCurrent(t *testing.T) {
|
||||
testPlugin := transfer.Transferers[0].(*transfererPlugin)
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetIxfr(transfer.xfrs[0].Zones[0], testPlugin.Serial, "ns.dns."+testPlugin.Zone, "hostmaster.dns."+testPlugin.Zone)
|
||||
|
||||
@@ -200,7 +200,7 @@ func TestTransferIXFRFallback(t *testing.T) {
|
||||
testPlugin := transfer.Transferers[0].(*transfererPlugin)
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewMultiRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetIxfr(
|
||||
transfer.xfrs[0].Zones[0],
|
||||
@@ -262,7 +262,7 @@ func TestTransferNotAllowed(t *testing.T) {
|
||||
}
|
||||
|
||||
ctx := context.TODO()
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{})
|
||||
w := dnstest.NewRecorder(&test.ResponseWriter{TCP: true})
|
||||
m := &dns.Msg{}
|
||||
m.SetAxfr(transfer.xfrs[0].Zones[0])
|
||||
|
||||
|
||||
Reference in New Issue
Block a user