53 lines
851 B
Go
53 lines
851 B
Go
|
package rules
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
|
||
|
|
C "github.com/Dreamacro/clash/constant"
|
||
|
|
|
||
|
|
"github.com/oschwald/geoip2-golang"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
var mmdb *geoip2.Reader
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
var err error
|
||
|
|
mmdb, err = geoip2.Open(C.MMDBPath)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Can't load mmdb: %s", err.Error())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type GEOIP struct {
|
||
|
|
country string
|
||
|
|
adapter string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *GEOIP) RuleType() C.RuleType {
|
||
|
|
return C.GEOIP
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *GEOIP) IsMatch(addr *C.Addr) bool {
|
||
|
|
if addr.AddrType == C.AtypDomainName {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
dstIP := net.ParseIP(addr.Host)
|
||
|
|
if dstIP == nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
record, _ := mmdb.Country(dstIP)
|
||
|
|
return record.Country.IsoCode == g.country
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *GEOIP) Adapter() string {
|
||
|
|
return g.adapter
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGEOIP(country string, adapter string) *GEOIP {
|
||
|
|
return &GEOIP{
|
||
|
|
country: country,
|
||
|
|
adapter: adapter,
|
||
|
|
}
|
||
|
|
}
|