2022-07-02 22:55:10 +08:00
|
|
|
package route
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
|
2022-07-06 15:01:09 +08:00
|
|
|
N "github.com/sagernet/sing/common/network"
|
|
|
|
|
2022-07-02 22:55:10 +08:00
|
|
|
"github.com/sagernet/sing-box/adapter"
|
|
|
|
"github.com/sagernet/sing-box/log"
|
|
|
|
)
|
|
|
|
|
|
|
|
var _ RuleItem = (*GeoIPItem)(nil)
|
|
|
|
|
|
|
|
type GeoIPItem struct {
|
|
|
|
router adapter.Router
|
|
|
|
logger log.Logger
|
|
|
|
isSource bool
|
|
|
|
codes []string
|
|
|
|
codeMap map[string]bool
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewGeoIPItem(router adapter.Router, logger log.Logger, isSource bool, codes []string) *GeoIPItem {
|
|
|
|
codeMap := make(map[string]bool)
|
|
|
|
for _, code := range codes {
|
|
|
|
codeMap[code] = true
|
|
|
|
}
|
|
|
|
return &GeoIPItem{
|
|
|
|
router: router,
|
|
|
|
logger: logger,
|
|
|
|
codes: codes,
|
|
|
|
isSource: isSource,
|
|
|
|
codeMap: codeMap,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *GeoIPItem) Match(metadata *adapter.InboundContext) bool {
|
|
|
|
geoReader := r.router.GeoIPReader()
|
|
|
|
if geoReader == nil {
|
2022-07-03 15:57:09 +08:00
|
|
|
return r.match(metadata)
|
2022-07-02 22:55:10 +08:00
|
|
|
}
|
|
|
|
if r.isSource {
|
|
|
|
if metadata.SourceGeoIPCode == "" {
|
2022-07-05 13:23:47 +08:00
|
|
|
metadata.SourceGeoIPCode = geoReader.Lookup(metadata.Source.Addr)
|
2022-07-02 22:55:10 +08:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if metadata.Destination.IsFqdn() {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if metadata.GeoIPCode == "" {
|
2022-07-05 13:23:47 +08:00
|
|
|
metadata.GeoIPCode = geoReader.Lookup(metadata.Destination.Addr)
|
2022-07-03 15:57:09 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return r.match(metadata)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *GeoIPItem) match(metadata *adapter.InboundContext) bool {
|
|
|
|
if r.isSource {
|
|
|
|
if metadata.SourceGeoIPCode == "" {
|
|
|
|
if !N.IsPublicAddr(metadata.Source.Addr) {
|
|
|
|
metadata.SourceGeoIPCode = "private"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return r.codeMap[metadata.SourceGeoIPCode]
|
|
|
|
} else {
|
|
|
|
if metadata.Destination.IsFqdn() {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if metadata.GeoIPCode == "" {
|
|
|
|
if !N.IsPublicAddr(metadata.Destination.Addr) {
|
|
|
|
metadata.GeoIPCode = "private"
|
|
|
|
}
|
2022-07-02 22:55:10 +08:00
|
|
|
}
|
|
|
|
return r.codeMap[metadata.GeoIPCode]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *GeoIPItem) String() string {
|
|
|
|
var description string
|
|
|
|
if r.isSource {
|
|
|
|
description = "source_geoip="
|
|
|
|
} else {
|
|
|
|
description = "geoip="
|
|
|
|
}
|
|
|
|
cLen := len(r.codes)
|
|
|
|
if cLen == 1 {
|
|
|
|
description += r.codes[0]
|
|
|
|
} else if cLen > 3 {
|
|
|
|
description += "[" + strings.Join(r.codes[:3], " ") + "...]"
|
|
|
|
} else {
|
|
|
|
description += "[" + strings.Join(r.codes, " ") + "]"
|
|
|
|
}
|
|
|
|
return description
|
|
|
|
}
|