338d148c78
* fix external ns records * use k8s service name for ns record * update test, add func comment * expand nsAddrs() test cases * support local ipv6 ip * use less confusing pod ip in test
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/coredns/coredns/plugin/kubernetes/object"
|
|
"github.com/miekg/dns"
|
|
api "k8s.io/api/core/v1"
|
|
)
|
|
|
|
func isDefaultNS(name, zone string) bool {
|
|
return strings.Index(name, defaultNSName) == 0 && strings.Index(name, zone) == len(defaultNSName)
|
|
}
|
|
|
|
// nsAddrs returns the A or AAAA records for the CoreDNS service in the cluster. If the service cannot be found,
|
|
// it returns a record for the the local address of the machine we're running on.
|
|
func (k *Kubernetes) nsAddrs(external bool, zone string) []dns.RR {
|
|
var (
|
|
svcNames []string
|
|
svcIPs []net.IP
|
|
)
|
|
|
|
// Find the CoreDNS Endpoint
|
|
localIP := k.interfaceAddrsFunc()
|
|
endpoints := k.APIConn.EpIndexReverse(localIP.String())
|
|
|
|
// If the CoreDNS Endpoint is not found, use the locally bound IP address
|
|
if len(endpoints) == 0 {
|
|
svcNames = []string{defaultNSName + zone}
|
|
svcIPs = []net.IP{localIP}
|
|
} else {
|
|
// Collect IPs for all Services of the Endpoints
|
|
for _, endpoint := range endpoints {
|
|
svcs := k.APIConn.SvcIndex(object.ServiceKey(endpoint.Name, endpoint.Namespace))
|
|
for _, svc := range svcs {
|
|
if external {
|
|
svcName := strings.Join([]string{svc.Name, svc.Namespace, zone}, ".")
|
|
for _, exIP := range svc.ExternalIPs {
|
|
svcNames = append(svcNames, svcName)
|
|
svcIPs = append(svcIPs, net.ParseIP(exIP))
|
|
}
|
|
continue
|
|
}
|
|
svcName := strings.Join([]string{svc.Name, svc.Namespace, Svc, zone}, ".")
|
|
if svc.ClusterIP == api.ClusterIPNone {
|
|
// For a headless service, use the endpoints IPs
|
|
for _, s := range endpoint.Subsets {
|
|
for _, a := range s.Addresses {
|
|
svcNames = append(svcNames, endpointHostname(a, k.endpointNameMode)+"."+svcName)
|
|
svcIPs = append(svcIPs, net.ParseIP(a.IP))
|
|
}
|
|
}
|
|
} else {
|
|
svcNames = append(svcNames, svcName)
|
|
svcIPs = append(svcIPs, net.ParseIP(svc.ClusterIP))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Create an RR slice of collected IPs
|
|
var rrs []dns.RR
|
|
rrs = make([]dns.RR, len(svcIPs))
|
|
for i, ip := range svcIPs {
|
|
if ip.To4() == nil {
|
|
rr := new(dns.AAAA)
|
|
rr.Hdr.Class = dns.ClassINET
|
|
rr.Hdr.Rrtype = dns.TypeAAAA
|
|
rr.Hdr.Name = svcNames[i]
|
|
rr.AAAA = ip
|
|
rrs[i] = rr
|
|
continue
|
|
}
|
|
rr := new(dns.A)
|
|
rr.Hdr.Class = dns.ClassINET
|
|
rr.Hdr.Rrtype = dns.TypeA
|
|
rr.Hdr.Name = svcNames[i]
|
|
rr.A = ip
|
|
rrs[i] = rr
|
|
}
|
|
|
|
return rrs
|
|
}
|
|
|
|
const defaultNSName = "ns.dns."
|