mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-10-10 19:20:04 +08:00
87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
package netLayer
|
||
|
||
import (
|
||
_ "embed"
|
||
"flag"
|
||
"log"
|
||
"net"
|
||
"os"
|
||
|
||
"github.com/hahahrfool/v2ray_simple/utils"
|
||
"github.com/oschwald/maxminddb-golang"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
var (
|
||
the_geoipdb *maxminddb.Reader
|
||
embedGeoip bool
|
||
|
||
GeoipFileName string //若运行程序指定了 geoip 参数，则该值为给定值；否则默认会被init为 GeoLite2-Country.mmdb
|
||
)
|
||
|
||
func init() {
|
||
flag.StringVar(&GeoipFileName, "geoip", "GeoLite2-Country.mmdb", "geoip maxmind file name")
|
||
|
||
}
|
||
|
||
func HasEmbedGeoip() bool {
|
||
return embedGeoip
|
||
}
|
||
|
||
func loadMaxmindGeoipBytes(bs []byte) {
|
||
db, err := maxminddb.FromBytes(bs)
|
||
if err != nil {
|
||
log.Println("loadMaxmindGeoipBytes err,", err)
|
||
return
|
||
}
|
||
the_geoipdb = db
|
||
}
|
||
|
||
//将一个外部的文件加载为我们默认的 geoip文件;若fn==""，则会自动使用 GeoipFileName 的值
|
||
func LoadMaxmindGeoipFile(fn string) {
|
||
if fn == "" {
|
||
fn = GeoipFileName
|
||
}
|
||
if fn == "" { //因为 GeoipFileName 是共有变量，所以可能会被设成"", 不排除脑残
|
||
return
|
||
}
|
||
bs, e := os.ReadFile(fn)
|
||
if e != nil {
|
||
log.Println("LoadMaxmindGeoipFile err", e)
|
||
return
|
||
}
|
||
loadMaxmindGeoipBytes(bs)
|
||
|
||
}
|
||
|
||
//使用默认的 geoip文件，会调用 GetIP_ISO_byReader
|
||
func GetIP_ISO(ip net.IP) string {
|
||
if the_geoipdb == nil {
|
||
return ""
|
||
}
|
||
return GetIP_ISO_byReader(the_geoipdb, ip)
|
||
}
|
||
|
||
//返回 iso 3166 字符串， 见 https://dev.maxmind.com/geoip/legacy/codes?lang=en ，大写，两字节
|
||
func GetIP_ISO_byReader(db *maxminddb.Reader, ip net.IP) string {
|
||
|
||
var record struct {
|
||
Country struct {
|
||
ISOCode string `maxminddb:"iso_code"`
|
||
} `maxminddb:"country"`
|
||
}
|
||
|
||
err := db.Lookup(ip, &record)
|
||
if err != nil {
|
||
|
||
if utils.ZapLogger != nil {
|
||
if ce := utils.CanLogErr("GetIP_ISO_byReader db.Lookup err"); ce != nil {
|
||
ce.Write(zap.Error(err))
|
||
}
|
||
}
|
||
|
||
return ""
|
||
}
|
||
return record.Country.ISOCode
|
||
}
|