dnsforward: imp code

heyxkhoa · Jan 29, 2021 · e5ab957 · e5ab957
1 parent 2b84d27
commit e5ab957
Show file tree

Hide file tree

Showing 3 changed files with 239 additions and 193 deletions.
diff --git a/internal/dnsforward/dnsforward.go b/internal/dnsforward/dnsforward.go
@@ -108,6 +108,12 @@ func (s *Server) Close() {
 	s.stats = nil
 	s.queryLog = nil
 	s.dnsProxy = nil
+
+	err := s.ipset.Close()
+	if err != nil {
+		log.Error("closing ipset: %s", err)
+	}
+
 	s.Unlock()
 }
 
@@ -190,7 +196,7 @@ func (s *Server) Prepare(config *ServerConfig) error {
 
 	// Initialize IPSET configuration
 	// --
-	err := s.ipset.init(s.conf.IPSETList, nil)
+	err := s.ipset.init(s.conf.IPSETList)
 	if err != nil {
 		return err
 	}