mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-10 20:26:29 -05:00
140 lines
3.3 KiB
Go
140 lines
3.3 KiB
Go
package memconservative
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/v2fly/v2ray-core/v4/app/router"
|
|
"github.com/v2fly/v2ray-core/v4/common/platform"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
type GeoIPCache map[string]*router.GeoIP
|
|
|
|
func (g GeoIPCache) Has(key string) bool {
|
|
return !(g.Get(key) == nil)
|
|
}
|
|
|
|
func (g GeoIPCache) Get(key string) *router.GeoIP {
|
|
if g == nil {
|
|
return nil
|
|
}
|
|
return g[key]
|
|
}
|
|
|
|
func (g GeoIPCache) Set(key string, value *router.GeoIP) {
|
|
if g == nil {
|
|
g = make(map[string]*router.GeoIP)
|
|
}
|
|
g[key] = value
|
|
}
|
|
|
|
func (g GeoIPCache) Unmarshal(filename, code string) (*router.GeoIP, error) {
|
|
asset := platform.GetAssetLocation(filename)
|
|
idx := strings.ToUpper(asset + "|" + code)
|
|
if g.Has(idx) {
|
|
return g.Get(idx), nil
|
|
}
|
|
|
|
geoipBytes, err := Decode(asset, code)
|
|
switch err {
|
|
case nil:
|
|
var geoip router.GeoIP
|
|
if err := proto.Unmarshal(geoipBytes, &geoip); err != nil {
|
|
return nil, err
|
|
}
|
|
g.Set(idx, &geoip)
|
|
return &geoip, nil
|
|
|
|
case errCodeNotFound:
|
|
return nil, newError(code, " not found in ", filename)
|
|
|
|
case errFailedToReadBytes, errFailedToReadExpectedLenBytes,
|
|
errInvalidGeodataFile, errInvalidGeodataVarintLength:
|
|
newError("failed to decode geodata file: ", filename, ". Fallback to the original ReadFile method.").AtWarning().WriteToLog()
|
|
geoipBytes, err = ioutil.ReadFile(asset)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var geoipList router.GeoIPList
|
|
if err := proto.Unmarshal(geoipBytes, &geoipList); err != nil {
|
|
return nil, err
|
|
}
|
|
for _, geoip := range geoipList.GetEntry() {
|
|
if strings.EqualFold(code, geoip.GetCountryCode()) {
|
|
g.Set(idx, geoip)
|
|
return geoip, nil
|
|
}
|
|
}
|
|
|
|
default:
|
|
return nil, err
|
|
}
|
|
|
|
return nil, newError(code, " not found in ", filename)
|
|
}
|
|
|
|
type GeoSiteCache map[string]*router.GeoSite
|
|
|
|
func (g GeoSiteCache) Has(key string) bool {
|
|
return !(g.Get(key) == nil)
|
|
}
|
|
|
|
func (g GeoSiteCache) Get(key string) *router.GeoSite {
|
|
if g == nil {
|
|
return nil
|
|
}
|
|
return g[key]
|
|
}
|
|
|
|
func (g GeoSiteCache) Set(key string, value *router.GeoSite) {
|
|
if g == nil {
|
|
g = make(map[string]*router.GeoSite)
|
|
}
|
|
g[key] = value
|
|
}
|
|
|
|
func (g GeoSiteCache) Unmarshal(filename, code string) (*router.GeoSite, error) {
|
|
asset := platform.GetAssetLocation(filename)
|
|
idx := strings.ToUpper(asset + "|" + code)
|
|
if g.Has(idx) {
|
|
return g.Get(idx), nil
|
|
}
|
|
|
|
geositeBytes, err := Decode(asset, code)
|
|
switch err {
|
|
case nil:
|
|
var geosite router.GeoSite
|
|
if err := proto.Unmarshal(geositeBytes, &geosite); err != nil {
|
|
return nil, err
|
|
}
|
|
g.Set(idx, &geosite)
|
|
return &geosite, nil
|
|
|
|
case errCodeNotFound:
|
|
return nil, newError(code, " not found in ", filename)
|
|
|
|
case errFailedToReadBytes, errFailedToReadExpectedLenBytes,
|
|
errInvalidGeodataFile, errInvalidGeodataVarintLength:
|
|
newError("failed to decode geodata file: ", filename, ". Fallback to the original ReadFile method.").AtWarning().WriteToLog()
|
|
geositeBytes, err = ioutil.ReadFile(asset)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var geositeList router.GeoSiteList
|
|
if err := proto.Unmarshal(geositeBytes, &geositeList); err != nil {
|
|
return nil, err
|
|
}
|
|
for _, geosite := range geositeList.GetEntry() {
|
|
if strings.EqualFold(code, geosite.GetCountryCode()) {
|
|
g.Set(idx, geosite)
|
|
return geosite, nil
|
|
}
|
|
}
|
|
|
|
default:
|
|
return nil, err
|
|
}
|
|
|
|
return nil, newError(code, " not found in ", filename)
|
|
}
|