116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package ldap
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/coredns/coredns/plugin"
|
|
"github.com/coredns/coredns/plugin/file"
|
|
)
|
|
|
|
// Run updates the zone from ldap.
|
|
func (l *Ldap) Run(ctx context.Context) error {
|
|
if err := l.UpdateZones(); err != nil {
|
|
return err
|
|
}
|
|
|
|
loop := func() {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Infof("Breaking out of Ldap update loop: %v", ctx.Err())
|
|
return
|
|
case <-time.After(l.syncInterval):
|
|
if err := l.UpdateZones(); err != nil && ctx.Err() == nil {
|
|
log.Errorf("Failed to update zones: %v", err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
go loop()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *Ldap) UpdateZones() error {
|
|
zoneFileMap := make(map[string]*file.Zone, len(l.Zones.Names))
|
|
for _, zn := range l.Zones.Names {
|
|
zoneFileMap[zn] = nil
|
|
zoneFileMap[zn] = file.NewZone(zn, "")
|
|
zoneFileMap[zn].Upstream = l.Upstream
|
|
|
|
err := zoneFileMap[zn].Insert(SOA(zn))
|
|
if err != nil {
|
|
return fmt.Errorf("updating zones: %w", err)
|
|
}
|
|
}
|
|
|
|
ldapRecords, err := l.fetchLdapRecords()
|
|
if err != nil {
|
|
return fmt.Errorf("updating zones: %w", err)
|
|
}
|
|
|
|
for zn, lrpz := range l.mapLdapRecordsToZone(ldapRecords) {
|
|
if len(lrpz) == 0 {
|
|
continue
|
|
}
|
|
|
|
for _, lr := range lrpz {
|
|
err = zoneFileMap[zn].Insert(lr.AAAA())
|
|
err = zoneFileMap[zn].Insert(lr.A())
|
|
if err != nil {
|
|
return fmt.Errorf("updating zones: %w", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
l.zMu.Lock()
|
|
for zn, zf := range zoneFileMap {
|
|
l.Zones.Z[zn] = zf
|
|
}
|
|
l.zMu.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *Ldap) mapLdapRecordsToZone(ldapRecords []ldapRecord) (ldapRecordsPerZone map[string][]ldapRecord) {
|
|
lrpz := make(map[string][]ldapRecord, len(l.Zones.Names))
|
|
for _, zn := range l.Zones.Names {
|
|
lrpz[zn] = nil
|
|
}
|
|
|
|
for _, lr := range ldapRecords {
|
|
zone := plugin.Zones(l.Zones.Names).Matches(lr.fqdn)
|
|
if zone != "" {
|
|
lrpz[zone] = append(lrpz[zone], lr)
|
|
}
|
|
}
|
|
|
|
return lrpz
|
|
}
|
|
|
|
func (l *Ldap) fetchLdapRecords() (ldapRecords []ldapRecord, err error) {
|
|
searchResult, err := l.Client.SearchWithPaging(l.SearchRequest, l.pagingLimit)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("fetching data from server: %w", err)
|
|
}
|
|
|
|
ldapRecords = make([]ldapRecord, len(searchResult.Entries))
|
|
for i := 0; i < len(ldapRecords); i++ {
|
|
fqdn := searchResult.Entries[i].GetAttributeValue(l.FqdnAttr)
|
|
if !strings.HasSuffix(fqdn, ".") {
|
|
fqdn = fqdn + "."
|
|
}
|
|
ldapRecords[i] = ldapRecord{
|
|
fqdn: fqdn,
|
|
ip4: net.ParseIP(searchResult.Entries[i].GetAttributeValue(l.Ip4Attr)),
|
|
ip6: net.ParseIP(searchResult.Entries[i].GetAttributeValue(l.Ip6Attr)),
|
|
}
|
|
}
|
|
|
|
return ldapRecords, nil
|
|
}
|