-(dnsforward): fixed sigsegv when protection is disabled

* commit 'c82e93cfc7cf439a5e5bf9b8bc27f5b68b528b53':
  -(dnsforward): fixed sigsegv when protection is disabled
This commit is contained in:
Andrey Meshkov 2019-08-20 15:03:17 +03:00
commit 87ca9ed9c2
6 changed files with 40 additions and 8 deletions

View File

@ -13,6 +13,7 @@ run:
skip-files: skip-files:
- ".*generated.*" - ".*generated.*"
- dnsfilter/rule_to_regexp.go - dnsfilter/rule_to_regexp.go
- ".*_test.go"
# all available settings of specific linters # all available settings of specific linters

View File

@ -17,6 +17,7 @@ import (
// CheckIfOtherDHCPServersPresent sends a DHCP request to the specified network interface, // CheckIfOtherDHCPServersPresent sends a DHCP request to the specified network interface,
// and waits for a response for a period defined by defaultDiscoverTime // and waits for a response for a period defined by defaultDiscoverTime
// nolint
func CheckIfOtherDHCPServersPresent(ifaceName string) (bool, error) { func CheckIfOtherDHCPServersPresent(ifaceName string) (bool, error) {
iface, err := net.InterfaceByName(ifaceName) iface, err := net.InterfaceByName(ifaceName)
if err != nil { if err != nil {
@ -95,8 +96,6 @@ func CheckIfOtherDHCPServersPresent(ifaceName string) (bool, error) {
if c != nil { if c != nil {
defer c.Close() defer c.Close()
} }
// spew.Dump(c, err)
// spew.Printf("net.ListenUDP returned %v, %v\n", c, err)
if err != nil { if err != nil {
return false, wrapErrPrint(err, "Couldn't listen on :68") return false, wrapErrPrint(err, "Couldn't listen on :68")
} }
@ -104,7 +103,6 @@ func CheckIfOtherDHCPServersPresent(ifaceName string) (bool, error) {
// send to 255.255.255.255:67 // send to 255.255.255.255:67
cm := ipv4.ControlMessage{} cm := ipv4.ControlMessage{}
_, err = c.WriteTo(packet, &cm, dstAddr) _, err = c.WriteTo(packet, &cm, dstAddr)
// spew.Dump(n, err)
if err != nil { if err != nil {
return false, wrapErrPrint(err, "Couldn't send a packet to %s", dst) return false, wrapErrPrint(err, "Couldn't send a packet to %s", dst)
} }

View File

@ -148,7 +148,7 @@ const (
ReasonRewrite ReasonRewrite
) )
func (i Reason) String() string { func (r Reason) String() string {
names := []string{ names := []string{
"NotFilteredNotFound", "NotFilteredNotFound",
"NotFilteredWhiteList", "NotFilteredWhiteList",
@ -163,10 +163,10 @@ func (i Reason) String() string {
"Rewrite", "Rewrite",
} }
if uint(i) >= uint(len(names)) { if uint(r) >= uint(len(names)) {
return "" return ""
} }
return names[i] return names[r]
} }
type dnsFilterContext struct { type dnsFilterContext struct {

View File

@ -473,7 +473,6 @@ func (s *Server) handleDNSRequest(p *proxy.Proxy, d *proxy.DNSContext) error {
} }
if res.Reason == dnsfilter.ReasonRewrite && len(res.CanonName) != 0 { if res.Reason == dnsfilter.ReasonRewrite && len(res.CanonName) != 0 {
d.Req.Question[0] = originalQuestion d.Req.Question[0] = originalQuestion
d.Res.Question[0] = originalQuestion d.Res.Question[0] = originalQuestion
@ -520,7 +519,7 @@ func (s *Server) filterDNSRequest(d *proxy.DNSContext) (*dnsfilter.Result, error
s.RUnlock() s.RUnlock()
if !protectionEnabled { if !protectionEnabled {
return nil, nil return &dnsfilter.Result{}, nil
} }
if host != origHost { if host != origHost {

View File

@ -78,6 +78,39 @@ func TestServer(t *testing.T) {
} }
} }
func TestServerWithProtectionDisabled(t *testing.T) {
s := createTestServer(t)
s.conf.ProtectionEnabled = false
defer removeDataDir(t)
err := s.Start(nil)
if err != nil {
t.Fatalf("Failed to start server: %s", err)
}
// message over UDP
req := createGoogleATestMessage()
addr := s.dnsProxy.Addr(proxy.ProtoUDP)
client := dns.Client{Net: "udp"}
reply, _, err := client.Exchange(req, addr.String())
if err != nil {
t.Fatalf("Couldn't talk to server %s: %s", addr, err)
}
assertGoogleAResponse(t, reply)
// check query log and stats
log := s.GetQueryLog()
assert.Equal(t, 1, len(log), "Log size")
stats := s.GetStatsTop()
assert.Equal(t, 1, len(stats.Domains), "Top domains length")
assert.Equal(t, 0, len(stats.Blocked), "Top blocked length")
assert.Equal(t, 1, len(stats.Clients), "Top clients length")
err = s.Stop()
if err != nil {
t.Fatalf("DNS server failed to stop: %s", err)
}
}
func TestDotServer(t *testing.T) { func TestDotServer(t *testing.T) {
// Prepare the proxy server // Prepare the proxy server
_, certPem, keyPem := createServerTLSConfig(t) _, certPem, keyPem := createServerTLSConfig(t)

View File

@ -61,6 +61,7 @@ func Main(version string, channel string) {
// run initializes configuration and runs the AdGuard Home // run initializes configuration and runs the AdGuard Home
// run is a blocking method and it won't exit until the service is stopped! // run is a blocking method and it won't exit until the service is stopped!
// nolint
func run(args options) { func run(args options) {
// config file path can be overridden by command-line arguments: // config file path can be overridden by command-line arguments:
if args.configFilename != "" { if args.configFilename != "" {