From 2e3ef77731f3ffffa24eae057acce3435f2babcb Mon Sep 17 00:00:00 2001 From: Michael Kashin <mmkashin@gmail.com> Date: Fri, 29 May 2020 10:04:23 -0700 Subject: [PATCH] k8s_external can now resolve CNAME returned by AWS ELB/NLB (#3916) Automatically submitted. --- plugin/k8s_external/external.go | 7 ++-- plugin/k8s_external/external_test.go | 51 ++++++++++++++++++++++------ plugin/k8s_external/msg_to_dns.go | 21 +++++++++--- plugin/k8s_external/setup.go | 3 ++ plugin/kubernetes/object/service.go | 7 +++- 5 files changed, 72 insertions(+), 17 deletions(-) diff --git a/plugin/k8s_external/external.go b/plugin/k8s_external/external.go index 3ca188ed8..1dffe1daa 100644 --- a/plugin/k8s_external/external.go +++ b/plugin/k8s_external/external.go @@ -16,6 +16,7 @@ import ( "github.com/coredns/coredns/plugin" "github.com/coredns/coredns/plugin/etcd/msg" + "github.com/coredns/coredns/plugin/pkg/upstream" "github.com/coredns/coredns/request" "github.com/miekg/dns" @@ -39,6 +40,8 @@ type External struct { apex string ttl uint32 + upstream *upstream.Upstream + externalFunc func(request.Request) ([]msg.Service, int) externalAddrFunc func(request.Request) []dns.RR } @@ -90,9 +93,9 @@ func (e *External) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Ms switch state.QType() { case dns.TypeA: - m.Answer = e.a(svc, state) + m.Answer = e.a(ctx, svc, state) case dns.TypeAAAA: - m.Answer = e.aaaa(svc, state) + m.Answer = e.aaaa(ctx, svc, state) case dns.TypeSRV: m.Answer, m.Extra = e.srv(svc, state) default: diff --git a/plugin/k8s_external/external_test.go b/plugin/k8s_external/external_test.go index 1123c6f6f..45584b6b1 100644 --- a/plugin/k8s_external/external_test.go +++ b/plugin/k8s_external/external_test.go @@ -41,6 +41,7 @@ func TestExternal(t *testing.T) { } resp := w.Msg + if resp == nil { t.Fatalf("Test %d, got nil message and no error for %q", i, r.Question[0].Name) } @@ -147,21 +148,33 @@ var tests = []test.Case{ test.SOA("example.com. 5 IN SOA ns1.dns.example.com. hostmaster.example.com. 1499347823 7200 1800 86400 5"), }, }, + { + Qname: "svc11.testns.example.com.", Qtype: dns.TypeA, Rcode: dns.RcodeSuccess, + Answer: []dns.RR{ + test.A("svc11.testns.example.com. 5 IN A 1.2.3.4"), + }, + }, + { + Qname: "svc12.testns.example.com.", Qtype: dns.TypeA, Rcode: dns.RcodeSuccess, + Answer: []dns.RR{ + test.CNAME("svc12.testns.example.com. 5 IN CNAME dummy.hostname"), + }, + }, } type external struct{} -func (external) HasSynced() bool { return true } -func (external) Run() {} -func (external) Stop() error { return nil } -func (external) EpIndexReverse(string) []*object.Endpoints { return nil } -func (external) SvcIndexReverse(string) []*object.Service { return nil } -func (external) Modified() int64 { return 0 } -func (external) EpIndex(s string) []*object.Endpoints { return nil } -func (external) EndpointsList() []*object.Endpoints { return nil } +func (external) HasSynced() bool { return true } +func (external) Run() {} +func (external) Stop() error { return nil } +func (external) EpIndexReverse(string) []*object.Endpoints { return nil } +func (external) SvcIndexReverse(string) []*object.Service { return nil } +func (external) Modified() int64 { return 0 } +func (external) EpIndex(s string) []*object.Endpoints { return nil } +func (external) EndpointsList() []*object.Endpoints { return nil } func (external) GetNodeByName(ctx context.Context, name string) (*api.Node, error) { return nil, nil } -func (external) SvcIndex(s string) []*object.Service { return svcIndexExternal[s] } -func (external) PodIndex(string) []*object.Pod { return nil } +func (external) SvcIndex(s string) []*object.Service { return svcIndexExternal[s] } +func (external) PodIndex(string) []*object.Pod { return nil } func (external) GetNamespaceByName(name string) (*api.Namespace, error) { return &api.Namespace{ @@ -192,6 +205,24 @@ var svcIndexExternal = map[string][]*object.Service{ Ports: []api.ServicePort{{Name: "http", Protocol: "tcp", Port: 80}}, }, }, + "svc11.testns": { + { + Name: "svc11", + Namespace: "testns", + Type: api.ServiceTypeLoadBalancer, + ExternalIPs: []string{"1.2.3.4"}, + Ports: []api.ServicePort{{Name: "http", Protocol: "tcp", Port: 80}}, + }, + }, + "svc12.testns": { + { + Name: "svc12", + Namespace: "testns", + Type: api.ServiceTypeLoadBalancer, + ExternalIPs: []string{"dummy.hostname"}, + Ports: []api.ServicePort{{Name: "http", Protocol: "tcp", Port: 80}}, + }, + }, } func (external) ServiceList() []*object.Service { diff --git a/plugin/k8s_external/msg_to_dns.go b/plugin/k8s_external/msg_to_dns.go index d09229d48..14a198063 100644 --- a/plugin/k8s_external/msg_to_dns.go +++ b/plugin/k8s_external/msg_to_dns.go @@ -1,6 +1,7 @@ package external import ( + "context" "math" "github.com/coredns/coredns/plugin/etcd/msg" @@ -9,7 +10,7 @@ import ( "github.com/miekg/dns" ) -func (e *External) a(services []msg.Service, state request.Request) (records []dns.RR) { +func (e *External) a(ctx context.Context, services []msg.Service, state request.Request) (records []dns.RR) { dup := make(map[string]struct{}) for _, s := range services { @@ -18,7 +19,13 @@ func (e *External) a(services []msg.Service, state request.Request) (records []d switch what { case dns.TypeCNAME: - // can't happen + rr := s.NewCNAME(state.QName(), s.Host) + records = append(records, rr) + if resp, err := e.upstream.Lookup(ctx, state, dns.Fqdn(s.Host), dns.TypeA); err == nil { + for _, rr := range resp.Answer { + records = append(records, rr) + } + } case dns.TypeA: if _, ok := dup[s.Host]; !ok { @@ -35,7 +42,7 @@ func (e *External) a(services []msg.Service, state request.Request) (records []d return records } -func (e *External) aaaa(services []msg.Service, state request.Request) (records []dns.RR) { +func (e *External) aaaa(ctx context.Context, services []msg.Service, state request.Request) (records []dns.RR) { dup := make(map[string]struct{}) for _, s := range services { @@ -44,7 +51,13 @@ func (e *External) aaaa(services []msg.Service, state request.Request) (records switch what { case dns.TypeCNAME: - // can't happen + rr := s.NewCNAME(state.QName(), s.Host) + records = append(records, rr) + if resp, err := e.upstream.Lookup(ctx, state, dns.Fqdn(s.Host), dns.TypeAAAA); err == nil { + for _, rr := range resp.Answer { + records = append(records, rr) + } + } case dns.TypeA: // nada diff --git a/plugin/k8s_external/setup.go b/plugin/k8s_external/setup.go index 5c2dce0b2..651ed1ba8 100644 --- a/plugin/k8s_external/setup.go +++ b/plugin/k8s_external/setup.go @@ -5,6 +5,7 @@ import ( "github.com/coredns/coredns/core/dnsserver" "github.com/coredns/coredns/plugin" + "github.com/coredns/coredns/plugin/pkg/upstream" "github.com/caddyserver/caddy" ) @@ -30,6 +31,8 @@ func setup(c *caddy.Controller) error { return nil }) + e.upstream = upstream.New() + dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler { e.Next = next return e diff --git a/plugin/kubernetes/object/service.go b/plugin/kubernetes/object/service.go index 3dc061528..4c0ea6901 100644 --- a/plugin/kubernetes/object/service.go +++ b/plugin/kubernetes/object/service.go @@ -62,7 +62,12 @@ func toService(skipCleanup bool, svc *api.Service) *Service { li := copy(s.ExternalIPs, svc.Spec.ExternalIPs) for i, lb := range svc.Status.LoadBalancer.Ingress { - s.ExternalIPs[li+i] = lb.IP + if lb.IP != "" { + s.ExternalIPs[li+i] = lb.IP + continue + } + s.ExternalIPs[li+i] = lb.Hostname + } if !skipCleanup {