black-internet-days-ahead/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2024-11-22 08:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add disableCache/disableExpire option for dns client

Browse source
This commit is contained in:
世界 2022-07-06 23:39:17 +08:00
parent 8a761d7e3b
commit ecac383477
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
5 changed files with 138 additions and 105 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
dns/client.go
View file

@ -13,6 +13,7 @@ import (
"github.com/sagernet/sing-box/adapter"
C "github.com/sagernet/sing-box/constant"
"github.com/sagernet/sing-box/option"
"golang.org/x/net/dns/dnsmessage"
)
@ -27,12 +28,21 @@ var (
var _ adapter.DNSClient = (*Client)(nil)
type Client struct {
cache *cache.LruCache[dnsmessage.Question, dnsmessage.Message]
cache *cache.LruCache[dnsmessage.Question, *dnsmessage.Message]
disableCache bool
disableExpire bool
}
func NewClient() *Client {
func NewClient(options option.DNSClientOptions) *Client {
if options.DisableCache {
return &Client{
cache: cache.New[dnsmessage.Question, dnsmessage.Message](),
disableCache: true,
}
} else {
return &Client{
cache: cache.New[dnsmessage.Question, *dnsmessage.Message](),
disableExpire: options.DisableExpire,
}
}
}
@ -41,10 +51,12 @@ func (c *Client) Exchange(ctx context.Context, transport adapter.DNSTransport, m
return nil, E.New("empty query")
}
question := message.Questions[0]
if !c.disableCache {
cachedAnswer, cached := c.cache.Load(question)
if cached {
cachedAnswer.ID = message.ID
return &cachedAnswer, nil
return cachedAnswer, nil
}
}
if !transport.Raw() {
if question.Type == dnsmessage.TypeA || question.Type == dnsmessage.TypeAAAA {
@ -56,7 +68,9 @@ func (c *Client) Exchange(ctx context.Context, transport adapter.DNSTransport, m
if err != nil {
return nil, err
}
c.cache.StoreWithExpire(question, *response, calculateExpire(message))
if !c.disableCache {
c.storeCache(question, response)
}
return message, err
}
@ -93,6 +107,7 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
}
return sortAddresses(response4, response6, strategy), nil
}
if !c.disableCache {
if strategy == C.DomainStrategyUseIPv4 {
response, err := c.questionCache(dnsmessage.Question{
Name: dnsName,
@ -126,6 +141,7 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
return sortAddresses(response4, response6, strategy), nil
}
}
}
var rCode dnsmessage.RCode
response, err := transport.Lookup(ctx, domain, strategy)
if err != nil {
@ -135,13 +151,16 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
} else {
rCode = dnsmessage.RCode(rCodeError)
}
if c.disableCache {
return nil, err
}
}
header := dnsmessage.Header{
Response: true,
Authoritative: true,
RCode: rCode,
}
expire := time.Now().Add(time.Second * time.Duration(DefaultTTL))
if !c.disableCache {
if strategy != C.DomainStrategyUseIPv6 {
question4 := dnsmessage.Question{
Name: dnsName,
@ -151,7 +170,7 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
response4 := common.Filter(response, func(addr netip.Addr) bool {
return addr.Is4() || addr.Is4In6()
})
message4 := dnsmessage.Message{
message4 := &dnsmessage.Message{
Header: header,
Questions: []dnsmessage.Question{question4},
}
@ -169,7 +188,7 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
})
}
}
c.cache.StoreWithExpire(question4, message4, expire)
c.storeCache(question4, message4)
}
if strategy != C.DomainStrategyUseIPv4 {
question6 := dnsmessage.Question{
@ -180,7 +199,7 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
response6 := common.Filter(response, func(addr netip.Addr) bool {
return addr.Is6() && !addr.Is4In6()
})
message6 := dnsmessage.Message{
message6 := &dnsmessage.Message{
Header: header,
Questions: []dnsmessage.Question{question6},
}
@ -198,7 +217,8 @@ func (c *Client) Lookup(ctx context.Context, transport adapter.DNSTransport, dom
})
}
}
c.cache.StoreWithExpire(question6, message6, expire)
c.storeCache(question6, message6)
}
}
return response, err
}
@ -211,14 +231,19 @@ func sortAddresses(response4 []netip.Addr, response6 []netip.Addr, strategy C.Do
}
}
func calculateExpire(message *dnsmessage.Message) time.Time {
func (c *Client) storeCache(question dnsmessage.Question, message *dnsmessage.Message) {
if c.disableExpire {
c.cache.Store(question, message)
return
}
timeToLive := DefaultTTL
for _, answer := range message.Answers {
if int(answer.Header.TTL) < timeToLive {
timeToLive = int(answer.Header.TTL)
}
}
return time.Now().Add(time.Second * time.Duration(timeToLive))
expire := time.Now().Add(time.Second * time.Duration(timeToLive))
c.cache.StoreWithExpire(question, message, expire)
}
func (c *Client) exchangeToLookup(ctx context.Context, transport adapter.DNSTransport, message *dnsmessage.Message, question dnsmessage.Question) (*dnsmessage.Message, error) {
@ -275,10 +300,12 @@ func (c *Client) lookupToExchange(ctx context.Context, transport adapter.DNSTran
Type: qType,
Class: dnsmessage.ClassINET,
}
if !c.disableCache {
cachedAddresses, err := c.questionCache(question)
if err != ErrNotCached {
return cachedAddresses, err
}
}
message := dnsmessage.Message{
Header: dnsmessage.Header{
ID: 0,
@ -298,7 +325,7 @@ func (c *Client) questionCache(question dnsmessage.Question) ([]netip.Addr, erro
if !cached {
return nil, ErrNotCached
}
return messageToAddresses(&response)
return messageToAddresses(response)
}
func messageToAddresses(response *dnsmessage.Message) ([]netip.Addr, error) {

2
dns/transport_tcp.go
View file

@ -91,7 +91,7 @@ func (t *TCPTransport) newConnection(conn *dnsConnection) {
cancel()
conn.err = err
if err != nil {
t.logger.Warn("connection closed: ", err)
t.logger.Debug("connection closed: ", err)
}
}

2
dns/transport_tls.go
View file

@ -99,7 +99,7 @@ func (t *TLSTransport) newConnection(conn *dnsConnection) {
cancel()
conn.err = err
if err != nil {
t.logger.Warn("connection closed: ", err)
t.logger.Debug("connection closed: ", err)
}
}

2
dns/transport_udp.go
View file

@ -87,7 +87,7 @@ func (t *UDPTransport) newConnection(conn *dnsConnection) {
cancel()
conn.err = err
if err != nil {
t.logger.Warn("connection closed: ", err)
t.logger.Debug("connection closed: ", err)
}
}

8
option/dns.go
View file

@ -2,11 +2,17 @@ package option
type DNSOptions struct {
Servers []DNSServerOptions `json:"servers,omitempty"`
DNSClientOptions
}
type DNSClientOptions struct {
DisableCache bool `json:"disable_cache,omitempty"`
DisableExpire bool `json:"disable_expire,omitempty"`
}
type DNSServerOptions struct {
Tag string `json:"tag,omitempty"`
Address string `json:"address"`
Detour string `json:"detour,omitempty"`
AddressResolver string `json:"address_resolver,omitempty"`
DialerOptions
}