mirror of https://github.com/v2fly/v2ray-core.git
68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package standard
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/app/router"
|
|
"github.com/v2fly/v2ray-core/v4/common/platform/filesystem"
|
|
"github.com/v2fly/v2ray-core/v4/infra/conf/geodata"
|
|
)
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v4/common/errors/errorgen
|
|
|
|
func loadIP(filename, country string) ([]*router.CIDR, error) {
|
|
geoipBytes, err := filesystem.ReadAsset(filename)
|
|
if err != nil {
|
|
return nil, newError("failed to open file: ", filename).Base(err)
|
|
}
|
|
var geoipList router.GeoIPList
|
|
if err := proto.Unmarshal(geoipBytes, &geoipList); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, geoip := range geoipList.Entry {
|
|
if strings.EqualFold(geoip.CountryCode, country) {
|
|
return geoip.Cidr, nil
|
|
}
|
|
}
|
|
|
|
return nil, newError("country not found in ", filename, ": ", country)
|
|
}
|
|
|
|
func loadSite(filename, list string) ([]*router.Domain, error) {
|
|
geositeBytes, err := filesystem.ReadAsset(filename)
|
|
if err != nil {
|
|
return nil, newError("failed to open file: ", filename).Base(err)
|
|
}
|
|
var geositeList router.GeoSiteList
|
|
if err := proto.Unmarshal(geositeBytes, &geositeList); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, site := range geositeList.Entry {
|
|
if strings.EqualFold(site.CountryCode, list) {
|
|
return site.Domain, nil
|
|
}
|
|
}
|
|
|
|
return nil, newError("list not found in ", filename, ": ", list)
|
|
}
|
|
|
|
type standardLoader struct{}
|
|
|
|
func (d standardLoader) LoadSite(filename, list string) ([]*router.Domain, error) {
|
|
return loadSite(filename, list)
|
|
}
|
|
|
|
func (d standardLoader) LoadIP(filename, country string) ([]*router.CIDR, error) {
|
|
return loadIP(filename, country)
|
|
}
|
|
|
|
func init() {
|
|
geodata.RegisterGeoDataLoaderImplementationCreator("standard", func() geodata.LoaderImplementation {
|
|
return standardLoader{}
|
|
})
|
|
}
|