Skip to content

Commit

Permalink
Fix: DNS NCACHE TTL and OPT RRs (#2900)
Browse files Browse the repository at this point in the history
* Fix: DNS NCACHE TTL and OPT RRs

1. DNS NCACHE was not correctly implemented.
2. OPT RRs must not be cached or forwarded.

Closes #2889.
  • Loading branch information
lujjjh authored Aug 24, 2023
1 parent 8a4c46a commit 45c10a1
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 6 deletions.
6 changes: 5 additions & 1 deletion dns/resolver.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ import (
C "github.com/Dreamacro/clash/constant"

D "github.com/miekg/dns"
"github.com/samber/lo"
"golang.org/x/sync/singleflight"
)

Expand Down Expand Up @@ -166,7 +167,10 @@ func (r *Resolver) exchangeWithoutCache(ctx context.Context, m *D.Msg) (msg *D.M
}

msg := result.(*D.Msg)

// OPT RRs MUST NOT be cached, forwarded, or stored in or loaded from master files.
msg.Extra = lo.Filter(msg.Extra, func(rr D.RR, index int) bool {
return rr.Header().Rrtype != D.TypeOPT
})
putMsgToCache(r.lruCache, q.String(), q, msg)
}()

Expand Down
28 changes: 23 additions & 5 deletions dns/util.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,22 +6,27 @@ import (
"errors"
"fmt"
"net"
"strings"
"time"

"github.com/Dreamacro/clash/common/cache"
"github.com/Dreamacro/clash/common/picker"
"github.com/Dreamacro/clash/log"

D "github.com/miekg/dns"
"github.com/samber/lo"
)

const serverFailureCacheTTL uint32 = 5

func minimalTTL(records []D.RR) uint32 {
if len(records) == 0 {
rr := lo.MinBy(records, func(r1 D.RR, r2 D.RR) bool {
return r1.Header().Ttl < r2.Header().Ttl
})
if rr == nil {
return 0
}
return lo.MinBy(records, func(r1 D.RR, r2 D.RR) bool {
return r1.Header().Ttl < r2.Header().Ttl
}).Header().Ttl
return rr.Header().Ttl
}

func updateTTL(records []D.RR, ttl uint32) {
Expand All @@ -35,7 +40,20 @@ func updateTTL(records []D.RR, ttl uint32) {
}

func putMsgToCache(c *cache.LruCache, key string, q D.Question, msg *D.Msg) {
ttl := minimalTTL(msg.Answer)
// skip dns cache for acme challenge
if q.Qtype == D.TypeTXT && strings.HasPrefix(q.Name, "_acme-challenge.") {
log.Debugln("[DNS] dns cache ignored because of acme challenge for: %s", q.Name)
return
}

var ttl uint32
if msg.Rcode == D.RcodeServerFailure {
// [...] a resolver MAY cache a server failure response.
// If it does so it MUST NOT cache it for longer than five (5) minutes [...]
ttl = serverFailureCacheTTL
} else {
ttl = minimalTTL(append(append(msg.Answer, msg.Ns...), msg.Extra...))
}
if ttl == 0 {
return
}
Expand Down

0 comments on commit 45c10a1

Please sign in to comment.