Xray-core/app/router/condition_geoip.go

119 lines
2.4 KiB
Go
Raw Normal View History

2020-11-25 13:01:53 +02:00
package router
import (
"net/netip"
"strconv"
2020-11-25 13:01:53 +02:00
2020-12-04 03:36:16 +02:00
"github.com/xtls/xray-core/common/net"
"go4.org/netipx"
2020-11-25 13:01:53 +02:00
)
type GeoIPMatcher struct {
countryCode string
reverseMatch bool
ip4 *netipx.IPSet
ip6 *netipx.IPSet
2020-11-25 13:01:53 +02:00
}
func (m *GeoIPMatcher) Init(cidrs []*CIDR) error {
var builder4, builder6 netipx.IPSetBuilder
2020-11-25 13:01:53 +02:00
for _, cidr := range cidrs {
ip := net.IP(cidr.GetIp())
ipPrefixString := ip.String() + "/" + strconv.Itoa(int(cidr.GetPrefix()))
ipPrefix, err := netip.ParsePrefix(ipPrefixString)
if err != nil {
return err
}
2020-11-25 13:01:53 +02:00
switch len(ip) {
case net.IPv4len:
builder4.AddPrefix(ipPrefix)
case net.IPv6len:
builder6.AddPrefix(ipPrefix)
2020-11-25 13:01:53 +02:00
}
}
if ip4, err := builder4.IPSet(); err != nil {
return err
} else {
m.ip4 = ip4
}
2020-11-25 13:01:53 +02:00
if ip6, err := builder6.IPSet(); err != nil {
return err
} else {
m.ip6 = ip6
2020-11-25 13:01:53 +02:00
}
return nil
}
func (m *GeoIPMatcher) SetReverseMatch(isReverseMatch bool) {
m.reverseMatch = isReverseMatch
}
func (m *GeoIPMatcher) match4(ip net.IP) bool {
nip, ok := netipx.FromStdIP(ip)
if !ok {
2020-11-25 13:01:53 +02:00
return false
}
return m.ip4.Contains(nip)
2020-11-25 13:01:53 +02:00
}
func (m *GeoIPMatcher) match6(ip net.IP) bool {
nip, ok := netipx.FromStdIP(ip)
if !ok {
2020-11-25 13:01:53 +02:00
return false
}
return m.ip6.Contains(nip)
2020-11-25 13:01:53 +02:00
}
// Match returns true if the given ip is included by the GeoIP.
func (m *GeoIPMatcher) Match(ip net.IP) bool {
isMatched := false
2020-11-25 13:01:53 +02:00
switch len(ip) {
case net.IPv4len:
isMatched = m.match4(ip)
case net.IPv6len:
isMatched = m.match6(ip)
}
if m.reverseMatch {
return !isMatched
2020-11-25 13:01:53 +02:00
}
return isMatched
2020-11-25 13:01:53 +02:00
}
// GeoIPMatcherContainer is a container for GeoIPMatchers. It keeps unique copies of GeoIPMatcher by country code.
type GeoIPMatcherContainer struct {
matchers []*GeoIPMatcher
}
// Add adds a new GeoIP set into the container.
// If the country code of GeoIP is not empty, GeoIPMatcherContainer will try to find an existing one, instead of adding a new one.
func (c *GeoIPMatcherContainer) Add(geoip *GeoIP) (*GeoIPMatcher, error) {
if len(geoip.CountryCode) > 0 {
for _, m := range c.matchers {
if m.countryCode == geoip.CountryCode && m.reverseMatch == geoip.ReverseMatch {
2020-11-25 13:01:53 +02:00
return m, nil
}
}
}
m := &GeoIPMatcher{
countryCode: geoip.CountryCode,
reverseMatch: geoip.ReverseMatch,
2020-11-25 13:01:53 +02:00
}
if err := m.Init(geoip.Cidr); err != nil {
return nil, err
}
if len(geoip.CountryCode) > 0 {
c.matchers = append(c.matchers, m)
}
return m, nil
}
2021-10-19 19:57:14 +03:00
var globalGeoIPContainer GeoIPMatcherContainer