2018-06-10 14:50:03 +00:00
|
|
|
package rules
|
|
|
|
|
|
|
|
import (
|
2021-07-01 14:49:29 +00:00
|
|
|
"strings"
|
|
|
|
|
2021-10-27 16:06:55 +00:00
|
|
|
"github.com/Dreamacro/clash/component/mmdb"
|
2022-03-14 18:55:06 +00:00
|
|
|
"github.com/Dreamacro/clash/component/resolver"
|
2018-06-10 14:50:03 +00:00
|
|
|
C "github.com/Dreamacro/clash/constant"
|
2018-07-31 16:18:29 +00:00
|
|
|
)
|
2018-06-10 14:50:03 +00:00
|
|
|
|
|
|
|
type GEOIP struct {
|
2022-03-12 17:21:23 +00:00
|
|
|
*Base
|
2021-10-27 16:06:55 +00:00
|
|
|
country string
|
|
|
|
adapter string
|
|
|
|
noResolveIP bool
|
2018-06-10 14:50:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (g *GEOIP) RuleType() C.RuleType {
|
|
|
|
return C.GEOIP
|
|
|
|
}
|
|
|
|
|
2019-10-27 16:02:23 +00:00
|
|
|
func (g *GEOIP) Match(metadata *C.Metadata) bool {
|
|
|
|
ip := metadata.DstIP
|
|
|
|
if ip == nil {
|
2018-06-10 14:50:03 +00:00
|
|
|
return false
|
|
|
|
}
|
2021-08-18 05:26:23 +00:00
|
|
|
|
2022-02-23 06:01:53 +00:00
|
|
|
if strings.EqualFold(g.country, "LAN") {
|
|
|
|
return ip.IsPrivate() ||
|
|
|
|
ip.IsUnspecified() ||
|
|
|
|
ip.IsLoopback() ||
|
|
|
|
ip.IsMulticast() ||
|
2022-03-14 18:55:06 +00:00
|
|
|
ip.IsLinkLocalUnicast() ||
|
|
|
|
resolver.IsFakeBroadcastIP(ip)
|
2021-08-18 05:26:23 +00:00
|
|
|
}
|
2022-02-23 06:01:53 +00:00
|
|
|
|
2021-10-27 16:06:55 +00:00
|
|
|
record, _ := mmdb.Instance().Country(ip)
|
|
|
|
return strings.EqualFold(record.Country.IsoCode, g.country)
|
2018-06-10 14:50:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (g *GEOIP) Adapter() string {
|
|
|
|
return g.adapter
|
|
|
|
}
|
|
|
|
|
2018-06-20 14:41:02 +00:00
|
|
|
func (g *GEOIP) Payload() string {
|
|
|
|
return g.country
|
|
|
|
}
|
|
|
|
|
2020-07-27 03:57:55 +00:00
|
|
|
func (g *GEOIP) ShouldResolveIP() bool {
|
|
|
|
return !g.noResolveIP
|
2019-10-27 16:02:23 +00:00
|
|
|
}
|
|
|
|
|
2021-10-27 16:06:55 +00:00
|
|
|
func (g *GEOIP) GetCountry() string {
|
|
|
|
return g.country
|
|
|
|
}
|
2021-07-01 14:49:29 +00:00
|
|
|
|
2022-03-12 17:21:23 +00:00
|
|
|
func NewGEOIP(country string, adapter string, noResolveIP bool) *GEOIP {
|
2019-10-27 16:02:23 +00:00
|
|
|
geoip := &GEOIP{
|
2022-03-12 17:21:23 +00:00
|
|
|
Base: &Base{},
|
2021-10-27 16:06:55 +00:00
|
|
|
country: country,
|
|
|
|
adapter: adapter,
|
|
|
|
noResolveIP: noResolveIP,
|
2018-06-10 14:50:03 +00:00
|
|
|
}
|
2019-10-27 16:02:23 +00:00
|
|
|
|
2022-03-12 17:21:23 +00:00
|
|
|
return geoip
|
2018-06-10 14:50:03 +00:00
|
|
|
}
|
2022-03-12 17:21:23 +00:00
|
|
|
|
|
|
|
var _ C.Rule = (*GEOIP)(nil)
|