mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-10-22 16:29:36 +08:00
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package netLayer
|
||
|
||
import (
|
||
_ "embed"
|
||
"log"
|
||
"net"
|
||
"os"
|
||
|
||
"github.com/e1732a364fed/v2ray_simple/utils"
|
||
"github.com/oschwald/maxminddb-golang"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
const MMDB_DownloadLink = "https://cdn.jsdelivr.net/gh/Loyalsoldier/geoip@release/Country.mmdb"
|
||
|
||
var (
|
||
the_geoipdb *maxminddb.Reader
|
||
embedGeoip bool
|
||
|
||
GeoipFileName string
|
||
)
|
||
|
||
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(utils.GetFilePath(fn))
|
||
if e != nil {
|
||
log.Printf("LoadMaxmindGeoipFile err, %s\n", e.Error())
|
||
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
|
||
}
|