103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"github.com/coredns/coredns/plugin/pkg/dnsutil"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type recordRequest struct {
|
|
// The named port from the kubernetes DNS spec, this is the service part (think _https) from a well formed
|
|
// SRV record.
|
|
port string
|
|
// The protocol is usually _udp or _tcp (if set), and comes from the protocol part of a well formed
|
|
// SRV record.
|
|
protocol string
|
|
endpoint string
|
|
// The servicename used in Kubernetes.
|
|
service string
|
|
// The namespace used in Kubernetes.
|
|
namespace string
|
|
// A each name can be for a pod or a service, here we track what we've seen, either "pod" or "service".
|
|
podOrSvc string
|
|
}
|
|
|
|
// parseRequest parses the qname to find all the elements we need for querying k8s. Anything
|
|
// that is not parsed will have the wildcard "*" value (except r.endpoint).
|
|
// Potential underscores are stripped from _port and _protocol.
|
|
func parseRequest(name, zone string) (r recordRequest, err error) {
|
|
// 3 Possible cases:
|
|
// 1. _port._protocol.service.namespace.pod|svc.zone
|
|
// 2. (endpoint): endpoint.service.namespace.pod|svc.zone
|
|
// 3. (service): service.namespace.pod|svc.zone
|
|
|
|
base, _ := dnsutil.TrimZone(name, zone)
|
|
// return NODATA for apex queries
|
|
if base == "" || base == Svc || base == Pod {
|
|
return r, nil
|
|
}
|
|
segs := dns.SplitDomainName(base)
|
|
|
|
last := len(segs) - 1
|
|
if last < 0 {
|
|
return r, nil
|
|
}
|
|
r.podOrSvc = segs[last]
|
|
if r.podOrSvc != Pod && r.podOrSvc != Svc {
|
|
return r, errInvalidRequest
|
|
}
|
|
last--
|
|
if last < 0 {
|
|
return r, nil
|
|
}
|
|
|
|
r.namespace = segs[last]
|
|
last--
|
|
if last < 0 {
|
|
return r, nil
|
|
}
|
|
|
|
r.service = segs[last]
|
|
last--
|
|
if last < 0 {
|
|
return r, nil
|
|
}
|
|
|
|
// Because of ambiguity we check the labels left: 1: an endpoint. 2: port and protocol.
|
|
// Anything else is a query that is too long to answer and can safely be delegated to return an nxdomain.
|
|
switch last {
|
|
case 0: // endpoint only
|
|
r.endpoint = segs[last]
|
|
case 1: // service and port
|
|
r.protocol = stripUnderscore(segs[last])
|
|
r.port = stripUnderscore(segs[last-1])
|
|
|
|
default: // too long
|
|
return r, errInvalidRequest
|
|
}
|
|
|
|
return r, nil
|
|
}
|
|
|
|
// stripUnderscore removes a prefixed underscore from s.
|
|
func stripUnderscore(s string) string {
|
|
if len(s) == 0 {
|
|
return s
|
|
}
|
|
if s[0] != '_' {
|
|
return s
|
|
}
|
|
return s[1:]
|
|
}
|
|
|
|
// String returns a string representation of r, it just returns all fields concatenated with dots.
|
|
// This is mostly used in tests.
|
|
func (r recordRequest) String() string {
|
|
s := r.port
|
|
s += "." + r.protocol
|
|
s += "." + r.endpoint
|
|
s += "." + r.service
|
|
s += "." + r.namespace
|
|
s += "." + r.podOrSvc
|
|
return s
|
|
}
|