history-museum/Clash.Meta
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: reuse code

Browse Source
This commit is contained in:
Skyxim 2023-03-12 14:37:46 +08:00
parent a510909228
commit c555cec490
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
dns/middleware.go
View File

@ -31,21 +31,27 @@ func withHosts(hosts R.Hosts, mapping *cache.LruCache[netip.Addr, string]) middl
} }
host := strings.TrimRight(q.Name, ".") host := strings.TrimRight(q.Name, ".")
handleCName := func(resp *D.Msg, domain string) {
rr := &D.CNAME{}
rr.Hdr = D.RR_Header{Name: q.Name, Rrtype: D.TypeCNAME, Class: D.ClassINET, Ttl: 10}
rr.Target = domain + "."
resp.Answer = append([]D.RR{rr}, resp.Answer...)
}
record, ok := hosts.Search(host, q.Qtype != D.TypeA && q.Qtype != D.TypeAAAA) record, ok := hosts.Search(host, q.Qtype != D.TypeA && q.Qtype != D.TypeAAAA)
if !ok { if !ok {
if record != nil && record.IsDomain { if record != nil && record.IsDomain {
// replace request domain // replace request domain
newR := r.Copy() newR := r.Copy()
newR.Question[0].Name = record.Domain+"." newR.Question[0].Name = record.Domain + "."
resp, err := next(ctx, newR) resp, err := next(ctx, newR)
if err==nil{ if err == nil {
resp.Id=r.Id resp.Id = r.Id
resp.Question=r.Question resp.Question = r.Question
handleCName(resp,record.Domain)
} }
return resp,err return resp, err
} }
return next(ctx,r) return next(ctx, r)
} }
msg := r.Copy() msg := r.Copy()
@ -78,10 +84,7 @@ func withHosts(hosts R.Hosts, mapping *cache.LruCache[netip.Addr, string]) middl
case D.TypeAAAA: case D.TypeAAAA:
handleIPs() handleIPs()
case D.TypeCNAME: case D.TypeCNAME:
rr := &D.CNAME{} handleCName(r,record.Domain)
rr.Hdr = D.RR_Header{Name: q.Name, Rrtype: D.TypeCNAME, Class: D.ClassINET, Ttl: 10}
rr.Target = record.Domain + "."
msg.Answer = append(msg.Answer, rr)
default: default:
return next(ctx, r) return next(ctx, r)
} }