Mirrors/juanfont.headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2024-12-30 00:09:42 +01:00
Code Issues Projects Releases Wiki Activity

Generate MagicDNS search domains for any tailnet range

Browse Source
This commit is contained in:
Juan Font Alonso 2021-10-09 12:22:13 +02:00
parent b02a9f9769
commit fc5153af3e
2 changed files with 23 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app.go
View File

@ -12,7 +12,7 @@ import (
"github.com/rs/zerolog/log" "github.com/rs/zerolog/log"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/zsais/go-gin-prometheus" ginprometheus "github.com/zsais/go-gin-prometheus"
"golang.org/x/crypto/acme" "golang.org/x/crypto/acme"
"golang.org/x/crypto/acme/autocert" "golang.org/x/crypto/acme/autocert"
"gorm.io/gorm" "gorm.io/gorm"
@ -111,7 +111,7 @@ func NewHeadscale(cfg Config) (*Headscale, error) {
} }
if h.cfg.DNSConfig != nil && h.cfg.DNSConfig.Proxied { // if MagicDNS if h.cfg.DNSConfig != nil && h.cfg.DNSConfig.Proxied { // if MagicDNS
magicDNSDomains, err := h.generateMagicDNSRootDomains() magicDNSDomains, err := generateMagicDNSRootDomains(h.cfg.IPPrefix, h.cfg.BaseDomain)
if err != nil { if err != nil {
return nil, err return nil, err
} }

27
dns.go
View File

@ -2,12 +2,14 @@ package headscale
import ( import (
"fmt" "fmt"
"strings"
"inet.af/netaddr"
"tailscale.com/util/dnsname" "tailscale.com/util/dnsname"
) )
func (h *Headscale) generateMagicDNSRootDomains() (*[]dnsname.FQDN, error) { func generateMagicDNSRootDomains(ipPrefix netaddr.IPPrefix, baseDomain string) (*[]dnsname.FQDN, error) {
base, err := dnsname.ToFQDN(h.cfg.BaseDomain) base, err := dnsname.ToFQDN(baseDomain)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -17,14 +19,27 @@ func (h *Headscale) generateMagicDNSRootDomains() (*[]dnsname.FQDN, error) {
ipv6base := dnsname.FQDN("0.e.1.a.c.5.1.1.a.7.d.f.ip6.arpa.") ipv6base := dnsname.FQDN("0.e.1.a.c.5.1.1.a.7.d.f.ip6.arpa.")
fqdns := []dnsname.FQDN{base, ipv6base} fqdns := []dnsname.FQDN{base, ipv6base}
for i := 64; i <= 127; i++ { netRange := ipPrefix.IPNet()
fqdn, err := dnsname.ToFQDN(fmt.Sprintf("%d.100.in-addr.arpa.", i)) maskBits, _ := netRange.Mask.Size()
lastByte := maskBits / 8
unmaskedBits := 8 - maskBits%8
min := uint(netRange.IP[lastByte])
max := uint((min + 1<<uint(unmaskedBits)) - 1)
rdnsSlice := []string{}
for i := lastByte - 1; i >= 0; i-- {
rdnsSlice = append(rdnsSlice, fmt.Sprintf("%d", netRange.IP[i]))
}
rdnsSlice = append(rdnsSlice, "in-addr.arpa.")
rdnsBase := strings.Join(rdnsSlice, ".")
for i := min; i <= max; i++ {
fqdn, err := dnsname.ToFQDN(fmt.Sprintf("%d.%s", i, rdnsBase))
if err != nil { if err != nil {
// TODO: propagate error
continue continue
} }
fqdns = append(fqdns, fqdn) fqdns = append(fqdns, fqdn)
} }
return &fqdns, nil return &fqdns, nil
} }