mirrors/dae
1
0
Fork 0
mirror of https://github.com/daeuniverse/dae.git synced 2025-07-06 08:22:38 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix: potential panic caused by dns upstream ready callback

Browse Source
This commit is contained in:
mzz2017
2023-02-27 16:21:53 +08:00
parent 1737d3f7b2
commit 95a8c0ac42
2 changed files with 34 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
component/dns/dns.go
View File

@ -50,7 +50,8 @@ func New(log *logrus.Logger, dns *config.Dns, opt *NewOption) (s *Dns, err error
if tag == "" {
return nil, fmt.Errorf("%w: '%v' has no tag", BadUpstreamFormatError, upstreamRaw)
}
u, err := url.Parse(link)
var u *url.URL
u, err = url.Parse(link)
if err != nil {
return nil, fmt.Errorf("%w: %v", BadUpstreamFormatError, err)
}
@ -73,9 +74,6 @@ func New(log *logrus.Logger, dns *config.Dns, opt *NewOption) (s *Dns, err error
}
upstreamName2Id[tag] = uint8(len(s.upstream))
s.upstream = append(s.upstream, r)
// Init immediately to avoid DNS leaking in the very beginning because param control_plane_dns_routing will
// be set in callback.
go r.GetUpstream()
}
// Optimize routings.
if dns.Routing.Request.Rules, err = routing.ApplyRulesOptimizers(dns.Routing.Request.Rules,
@ -119,6 +117,12 @@ func New(log *logrus.Logger, dns *config.Dns, opt *NewOption) (s *Dns, err error
return s, nil
}
func (s *Dns) InitUpstreams() {
for _, upstream := range s.upstream {
upstream.GetUpstream()
}
}
func (s *Dns) RequestSelect(msg *dnsmessage.Message) (upstream *Upstream, err error) {
if msg.Response {
return nil, fmt.Errorf("DNS request expected but DNS response received")

28
control/control_plane.go
View File

@ -51,6 +51,9 @@ type ControlPlane struct {
dialMode consts.DialMode
routingMatcher *RoutingMatcher
closed chan struct{}
ready chan struct{}
}
func NewControlPlane(
@ -302,7 +305,14 @@ func NewControlPlane(
outbounds: outbounds,
dialMode: dialMode,
routingMatcher: routingMatcher,
closed: make(chan struct{}),
ready: make(chan struct{}),
}
defer func() {
if err != nil {
close(c.closed)
}
}()
/// DNS upstream.
dnsUpstream, err := dns.New(log, dnsConfig, &dns.NewOption{
@ -313,7 +323,7 @@ func NewControlPlane(
}
/// Dns controller.
c.dnsController, err = NewDnsController(dnsUpstream, &DnsControllerOption{
if c.dnsController, err = NewDnsController(dnsUpstream, &DnsControllerOption{
Log: log,
CacheAccessCallback: func(cache *DnsCache) (err error) {
// Write mappings into eBPF map:
@ -331,8 +341,14 @@ func NewControlPlane(
}, nil
},
BestDialerChooser: c.chooseBestDnsDialer,
})
}); err != nil {
return nil, err
}
// Init immediately to avoid DNS leaking in the very beginning because param control_plane_dns_routing will
// be set in callback.
dnsUpstream.InitUpstreams()
close(c.ready)
return c, nil
}
@ -342,6 +358,13 @@ func (c *ControlPlane) EjectBpf() *bpfObjects {
}
func (c *ControlPlane) dnsUpstreamReadyCallback(raw *url.URL, dnsUpstream *dns.Upstream) (err error) {
// Waiting for ready.
select {
case <-c.closed:
return nil
case <-c.ready:
}
/// Notify dialers to check.
c.onceNetworkReady.Do(func() {
for _, out := range c.outbounds {
@ -736,5 +759,6 @@ func (c *ControlPlane) Close() (err error) {
}
}
}
close(c.closed)
return c.core.Close()
}