github/sower
4
0
Fork 0
mirror of https://github.com/wweir/sower.git synced 2026-10-21 03:51:28 +00:00
Code Issues Releases Wiki Activity

Add timeout for dns exchange

Browse Source
This commit is contained in:
wweir
2019-03-19 13:30:36 +08:00
parent 6add7afd19
commit 45ca063a7e
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dns/dns.go
+7 -2
View File
@@ -1,6 +1,7 @@
package dns
import (
"context"
"net"
"strings"
"sync/atomic"
@@ -73,7 +74,10 @@ func matchAndServe(w dns.ResponseWriter, r *dns.Msg, domain, listenIP, dnsServer
return
}
msg, err := dns.Exchange(r, dnsServer)
ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond)
defer cancel()
msg, err := dns.ExchangeContext(ctx, r, dnsServer)
if err != nil {
if dhcpCh != nil {
select {
@@ -81,9 +85,10 @@ func matchAndServe(w dns.ResponseWriter, r *dns.Msg, domain, listenIP, dnsServer
default:
}
}
glog.V(1).Infof("get dns of %s fail: %s", domain, err)
return
} else if msg == nil { // expose any response except nil
glog.V(1).Infof("get dns of %s fail: %s", domain, err)
glog.V(1).Infof("get dns of %s return empty", domain)
return
}