Fix for GeodataMode

MetaCubeX · Feb 26, 2024 · ca36080 · ca36080
1 parent 088fdb2
commit ca36080
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/dns/filters.go b/dns/filters.go
@@ -22,6 +22,7 @@ type geoipFilter struct {
 }
 
 var geoIPMatcher *router.GeoIPMatcher
+var geoIPMatcherPrivate *router.GeoIPMatcher
 
 func (gf *geoipFilter) Match(ip netip.Addr) bool {
 	if !C.GeodataMode {
@@ -42,7 +43,15 @@ func (gf *geoipFilter) Match(ip netip.Addr) bool {
 			return false
 		}
 	}
-	return !geoIPMatcher.Match(ip)
+
+	if geoIPMatcherPrivate == nil {
+		var err error
+		geoIPMatcherPrivate, _, err = geodata.LoadGeoIPMatcher("private")
+		if err != nil {
+			return false
+		}
+	}
+	return !geoIPMatcher.Match(ip) && !geoIPMatcherPrivate.Match(ip)
 }
 
 type ipnetFilter struct {