Ponywka/MagiTrickle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

process cnames

Browse Source
This commit is contained in:
Vladimir Avtsenov 2024-09-14 19:26:23 +03:00
parent 1cfdf406eb
commit c3aeb0cfb7
1 changed files with 48 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
kvas2.go
View File

@ -297,6 +297,11 @@ func (a *App) SyncGroup(group *Group) error {
Str("address", ip.String()). Str("address", ip.String()).
Err(err). Err(err).
Msg("failed to delete address") Msg("failed to delete address")
} else {
log.Trace().
Str("address", ip.String()).
Err(err).
Msg("add address")
} }
} }
@ -338,6 +343,7 @@ func (a *App) processARecord(aRecord dnsProxy.Address) {
names := a.Records.GetCNameRecords(aRecord.Name.String(), true) names := a.Records.GetCNameRecords(aRecord.Name.String(), true)
for _, group := range a.Groups { for _, group := range a.Groups {
Domain:
for _, domain := range group.Domains { for _, domain := range group.Domains {
if !domain.IsEnabled() { if !domain.IsEnabled() {
continue continue
@ -352,7 +358,15 @@ func (a *App) processARecord(aRecord dnsProxy.Address) {
Str("address", aRecord.Address.String()). Str("address", aRecord.Address.String()).
Err(err). Err(err).
Msg("failed to add address") Msg("failed to add address")
} else {
log.Trace().
Str("address", aRecord.Address.String()).
Str("aRecordDomain", aRecord.Name.String()).
Str("cNameDomain", name).
Err(err).
Msg("add address")
} }
break Domain
} }
} }
} }
@ -371,6 +385,40 @@ func (a *App) processCNameRecord(cNameRecord dnsProxy.CName) {
} }
a.Records.AddCNameRecord(cNameRecord.Name.String(), cNameRecord.CName.String(), ttlDuration) a.Records.AddCNameRecord(cNameRecord.Name.String(), cNameRecord.CName.String(), ttlDuration)
// TODO: Optimization
now := time.Now()
aRecords := a.Records.GetARecords(cNameRecord.Name.String())
names := a.Records.GetCNameRecords(cNameRecord.Name.String(), true)
for _, group := range a.Groups {
Domain:
for _, domain := range group.Domains {
if !domain.IsEnabled() {
continue
}
for _, name := range names {
if !domain.IsMatch(name) {
continue
}
for _, aRecord := range aRecords {
err := group.AddIPv4(aRecord.Address, now.Sub(aRecord.Deadline))
if err != nil {
log.Error().
Str("address", aRecord.Address.String()).
Err(err).
Msg("failed to add address")
} else {
log.Trace().
Str("address", aRecord.Address.String()).
Str("cNameDomain", name).
Err(err).
Msg("add address")
}
}
continue Domain
}
}
}
} }
func (a *App) handleRecord(rr dnsProxy.ResourceRecord) { func (a *App) handleRecord(rr dnsProxy.ResourceRecord) {