7e56cc74e5
* middleware/kubernetes: pull TXT out of parseRequest Put the TXT handling one layer higher and remove it from parseRequest. Also rename the podsvc field in there to podOrSvc. Now that it isn't used anymore for TXT record (dns-version) that was put in there. We can make this a boolean (in a future PR). Make parseRequest get an optional Zone that is from state.Zone and use that instead of its own code. Removed some tests and other smaller cleanups. Fixes #836 * add this reverse * another check * readd * Rename to kPod and kService for some clarity
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/coredns/coredns/middleware"
|
|
"github.com/coredns/coredns/middleware/pkg/dnsutil"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
// ServeDNS implements the middleware.Handler interface.
|
|
func (k Kubernetes) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
|
|
state := request.Request{W: w, Req: r}
|
|
if state.QClass() != dns.ClassINET {
|
|
return dns.RcodeServerFailure, middleware.Error(k.Name(), errors.New("can only deal with ClassINET"))
|
|
}
|
|
|
|
m := new(dns.Msg)
|
|
m.SetReply(r)
|
|
m.Authoritative, m.RecursionAvailable, m.Compress = true, true, true
|
|
// Check that query matches one of the zones served by this middleware,
|
|
// otherwise delegate to the next in the pipeline.
|
|
zone := middleware.Zones(k.Zones).Matches(state.Name())
|
|
if zone == "" {
|
|
if state.Type() != "PTR" {
|
|
return middleware.NextOrFailure(k.Name(), k.Next, ctx, w, r)
|
|
}
|
|
// If this is a PTR request, and the request is in a defined
|
|
// pod/service cidr range, process the request in this middleware,
|
|
// otherwise pass to next middleware.
|
|
if !k.isRequestInReverseRange(state.Name()) {
|
|
return middleware.NextOrFailure(k.Name(), k.Next, ctx, w, r)
|
|
}
|
|
|
|
// Set the zone to this specific request, as we want to handle this reverse request.
|
|
zone = state.Name()
|
|
}
|
|
|
|
state.Zone = zone
|
|
|
|
var (
|
|
records []dns.RR
|
|
extra []dns.RR
|
|
err error
|
|
)
|
|
|
|
switch state.Type() {
|
|
case "A":
|
|
records, _, err = middleware.A(&k, zone, state, nil, middleware.Options{})
|
|
case "AAAA":
|
|
records, _, err = middleware.AAAA(&k, zone, state, nil, middleware.Options{})
|
|
case "TXT":
|
|
records, _, err = middleware.TXT(&k, zone, state, middleware.Options{})
|
|
case "CNAME":
|
|
records, _, err = middleware.CNAME(&k, zone, state, middleware.Options{})
|
|
case "PTR":
|
|
records, _, err = middleware.PTR(&k, zone, state, middleware.Options{})
|
|
case "MX":
|
|
records, extra, _, err = middleware.MX(&k, zone, state, middleware.Options{})
|
|
case "SRV":
|
|
records, extra, _, err = middleware.SRV(&k, zone, state, middleware.Options{})
|
|
case "SOA":
|
|
records, _, err = middleware.SOA(&k, zone, state, middleware.Options{})
|
|
case "NS":
|
|
if state.Name() == zone {
|
|
records, extra, _, err = middleware.NS(&k, zone, state, middleware.Options{})
|
|
break
|
|
}
|
|
fallthrough
|
|
default:
|
|
// Do a fake A lookup, so we can distinguish between NODATA and NXDOMAIN
|
|
_, _, err = middleware.A(&k, zone, state, nil, middleware.Options{})
|
|
}
|
|
|
|
if k.IsNameError(err) {
|
|
if k.Fallthrough {
|
|
return middleware.NextOrFailure(k.Name(), k.Next, ctx, w, r)
|
|
}
|
|
return middleware.BackendError(&k, zone, dns.RcodeNameError, state, nil /*debug*/, nil /* err */, middleware.Options{})
|
|
}
|
|
if err != nil {
|
|
return dns.RcodeServerFailure, err
|
|
}
|
|
|
|
if len(records) == 0 {
|
|
return middleware.BackendError(&k, zone, dns.RcodeSuccess, state, nil /*debug*/, nil, middleware.Options{})
|
|
}
|
|
|
|
m.Answer = append(m.Answer, records...)
|
|
m.Extra = append(m.Extra, extra...)
|
|
|
|
m = dnsutil.Dedup(m)
|
|
state.SizeAndDo(m)
|
|
m, _ = state.Scrub(m)
|
|
w.WriteMsg(m)
|
|
return dns.RcodeSuccess, nil
|
|
}
|
|
|
|
// Name implements the Handler interface.
|
|
func (k Kubernetes) Name() string { return "kubernetes" }
|