30f1b29257
# Conflicts: # .github/workflows/codeql-analysis.yml # .github/workflows/linter.yml # .github/workflows/release.yml # Makefile # README.md # adapter/outbound/vless.go # component/geodata/memconservative/cache.go # component/geodata/router/condition.go # component/geodata/router/condition_geoip.go # component/geodata/standard/standard.go # component/geodata/utils.go # config/config.go # config/initial.go # constant/metadata.go # constant/path.go # constant/rule.go # constant/rule_extra.go # dns/client.go # dns/filters.go # dns/resolver.go # go.mod # go.sum # hub/executor/executor.go # hub/route/configs.go # listener/listener.go # listener/tproxy/tproxy_linux_iptables.go # listener/tun/dev/dev.go # listener/tun/dev/dev_darwin.go # listener/tun/dev/dev_linux.go # listener/tun/dev/dev_windows.go # listener/tun/dev/wintun/config.go # listener/tun/dev/wintun/dll_windows.go # listener/tun/dev/wintun/session_windows.go # listener/tun/dev/wintun/wintun_windows.go # listener/tun/ipstack/commons/dns.go # listener/tun/ipstack/gvisor/tun.go # listener/tun/ipstack/gvisor/tundns.go # listener/tun/ipstack/gvisor/utils.go # listener/tun/ipstack/stack_adapter.go # listener/tun/ipstack/system/dns.go # listener/tun/ipstack/system/tcp.go # listener/tun/ipstack/system/tun.go # listener/tun/tun_adapter.go # main.go # rule/common/base.go # rule/common/domain.go # rule/common/domain_keyword.go # rule/common/domain_suffix.go # rule/common/final.go # rule/common/geoip.go # rule/common/geosite.go # rule/common/ipcidr.go # rule/common/port.go # rule/parser.go # rule/process.go # test/go.mod # test/go.sum # transport/vless/xtls.go # tunnel/tunnel.go
99 lines
2 KiB
Go
99 lines
2 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Dreamacro/clash/component/geodata"
|
|
"github.com/Dreamacro/clash/component/geodata/router"
|
|
"strings"
|
|
|
|
"github.com/Dreamacro/clash/component/mmdb"
|
|
"github.com/Dreamacro/clash/component/resolver"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/log"
|
|
)
|
|
|
|
type GEOIP struct {
|
|
*Base
|
|
country string
|
|
adapter string
|
|
noResolveIP bool
|
|
geoIPMatcher *router.GeoIPMatcher
|
|
}
|
|
|
|
func (g *GEOIP) ShouldFindProcess() bool {
|
|
return false
|
|
}
|
|
|
|
func (g *GEOIP) RuleType() C.RuleType {
|
|
return C.GEOIP
|
|
}
|
|
|
|
func (g *GEOIP) Match(metadata *C.Metadata) bool {
|
|
ip := metadata.DstIP
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
|
|
if strings.EqualFold(g.country, "LAN") {
|
|
return ip.IsPrivate() ||
|
|
ip.IsUnspecified() ||
|
|
ip.IsLoopback() ||
|
|
ip.IsMulticast() ||
|
|
ip.IsLinkLocalUnicast() ||
|
|
resolver.IsFakeBroadcastIP(ip)
|
|
}
|
|
if !C.GeodataMode {
|
|
record, _ := mmdb.Instance().Country(ip)
|
|
return strings.EqualFold(record.Country.IsoCode, g.country)
|
|
}
|
|
return g.geoIPMatcher.Match(ip)
|
|
}
|
|
|
|
func (g *GEOIP) Adapter() string {
|
|
return g.adapter
|
|
}
|
|
|
|
func (g *GEOIP) Payload() string {
|
|
return g.country
|
|
}
|
|
|
|
func (g *GEOIP) ShouldResolveIP() bool {
|
|
return !g.noResolveIP
|
|
}
|
|
|
|
func (g *GEOIP) GetCountry() string {
|
|
return g.country
|
|
}
|
|
|
|
func (g *GEOIP) GetIPMatcher() *router.GeoIPMatcher {
|
|
return g.geoIPMatcher
|
|
}
|
|
|
|
func NewGEOIP(country string, adapter string, noResolveIP bool) (*GEOIP, error) {
|
|
if !C.GeodataMode {
|
|
geoip := &GEOIP{
|
|
Base: &Base{},
|
|
country: country,
|
|
adapter: adapter,
|
|
noResolveIP: noResolveIP,
|
|
}
|
|
return geoip, nil
|
|
}
|
|
|
|
geoIPMatcher, recordsCount, err := geodata.LoadGeoIPMatcher(country)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("[GeoIP] %s", err.Error())
|
|
}
|
|
|
|
log.Infoln("Start initial GeoIP rule %s => %s, records: %d", country, adapter, recordsCount)
|
|
geoip := &GEOIP{
|
|
Base: &Base{},
|
|
country: country,
|
|
adapter: adapter,
|
|
noResolveIP: noResolveIP,
|
|
geoIPMatcher: geoIPMatcher,
|
|
}
|
|
return geoip, nil
|
|
}
|
|
|
|
var _ C.Rule = (*GEOIP)(nil)
|