ambrose
/
alias
mirror of https://github.com/serverwentdown/alias.git
1
0
Fork 0
Code Releases Activity

Simplify loops 

Signed-off-by: Knut Ahlers <knut@ahlers.me>
pull/2/head
Knut Ahlers 2018-01-30 16:57:48 +01:00
parent 9de8988895
commit 5a79be553d
No known key found for this signature in database
GPG Key ID: DC2729FDD34BE99E
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
alias.go
View File

@ -38,8 +38,7 @@ func NewResponseModifier(w dns.ResponseWriter) *ResponseModifier {
func (r *ResponseModifier) WriteMsg(res *dns.Msg) error {
// Guess zone based on authority section.
var zone string
for i := 0; i < len(res.Ns); i++ {
rr := res.Ns[i]
for _, rr := range res.Ns {
if rr.Header().Rrtype == dns.TypeNS {
zone = rr.Header().Name
}
@ -47,8 +46,7 @@ func (r *ResponseModifier) WriteMsg(res *dns.Msg) error {
// Find and delete CNAME record on that zone, storing the canonical name.
var cname string
for i := 0; i < len(res.Answer); i++ {
rr := res.Answer[i]
for i, rr := range res.Answer {
if rr.Header().Rrtype == dns.TypeCNAME && rr.Header().Name == zone {
cname = rr.(*dns.CNAME).Target
// Remove the CNAME record
@ -58,8 +56,7 @@ func (r *ResponseModifier) WriteMsg(res *dns.Msg) error {
}
// Rename all the records with the above canonical name to the zone name
for i := 0; i < len(res.Answer); i++ {
rr := res.Answer[i]
for _, rr := range res.Answer {
if rr.Header().Name == cname {
rr.Header().Name = zone
}