mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-04 13:07:39 +03:00
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/apernet/hysteria/extras/outbounds/acl"
|
|
"github.com/apernet/hysteria/extras/outbounds/acl/v2geo"
|
|
)
|
|
|
|
const (
|
|
geoipFilename = "geoip.dat"
|
|
geoipURL = "https://cdn.jsdelivr.net/gh/Loyalsoldier/v2ray-rules-dat@release/geoip.dat"
|
|
geositeFilename = "geosite.dat"
|
|
geositeURL = "https://cdn.jsdelivr.net/gh/Loyalsoldier/v2ray-rules-dat@release/geosite.dat"
|
|
)
|
|
|
|
var _ acl.GeoLoader = (*GeoLoader)(nil)
|
|
|
|
// GeoLoader provides the on-demand GeoIP/GeoSite database
|
|
// loading functionality required by the ACL engine.
|
|
// Empty filenames = automatic download from built-in URLs.
|
|
type GeoLoader struct {
|
|
GeoIPFilename string
|
|
GeoSiteFilename string
|
|
|
|
DownloadFunc func(filename, url string)
|
|
DownloadErrFunc func(err error)
|
|
|
|
geoipMap map[string]*v2geo.GeoIP
|
|
geositeMap map[string]*v2geo.GeoSite
|
|
}
|
|
|
|
func (l *GeoLoader) download(filename, url string) error {
|
|
l.DownloadFunc(filename, url)
|
|
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
l.DownloadErrFunc(err)
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
f, err := os.Create(filename)
|
|
if err != nil {
|
|
l.DownloadErrFunc(err)
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
_, err = io.Copy(f, resp.Body)
|
|
l.DownloadErrFunc(err)
|
|
return err
|
|
}
|
|
|
|
func (l *GeoLoader) LoadGeoIP() (map[string]*v2geo.GeoIP, error) {
|
|
if l.geoipMap != nil {
|
|
return l.geoipMap, nil
|
|
}
|
|
autoDL := false
|
|
filename := l.GeoIPFilename
|
|
if filename == "" {
|
|
autoDL = true
|
|
filename = geoipFilename
|
|
}
|
|
m, err := v2geo.LoadGeoIP(filename)
|
|
if os.IsNotExist(err) && autoDL {
|
|
// It's ok, we will download it.
|
|
err = l.download(filename, geoipURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
m, err = v2geo.LoadGeoIP(filename)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
l.geoipMap = m
|
|
return m, nil
|
|
}
|
|
|
|
func (l *GeoLoader) LoadGeoSite() (map[string]*v2geo.GeoSite, error) {
|
|
if l.geositeMap != nil {
|
|
return l.geositeMap, nil
|
|
}
|
|
autoDL := false
|
|
filename := l.GeoSiteFilename
|
|
if filename == "" {
|
|
autoDL = true
|
|
filename = geositeFilename
|
|
}
|
|
m, err := v2geo.LoadGeoSite(filename)
|
|
if os.IsNotExist(err) && autoDL {
|
|
// It's ok, we will download it.
|
|
err = l.download(filename, geositeURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
m, err = v2geo.LoadGeoSite(filename)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
l.geositeMap = m
|
|
return m, nil
|
|
}
|