mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-27 02:17:54 +00:00
44d8a14629
Some checks are pending
Build / build (map[goamd64:v1 goarch:amd64 goos:darwin output:amd64-compatible]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:freebsd output:amd64-compatible]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:linux goversion:1.20 output:amd64-compatible-go120 test:test]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:linux output:amd64-compatible test:test]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:windows goversion:1.20 output:amd64-compatible-go120]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:windows output:amd64-compatible]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:darwin goversion:1.20 output:amd64-go120]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:darwin output:amd64]) (push) Waiting to run
Build / build (map[abi:1 goarch:loong64 goos:linux output:loong64-abi1]) (push) Waiting to run
Build / build (map[abi:2 goarch:loong64 goos:linux output:loong64-abi2]) (push) Waiting to run
Build / build (map[goamd64:v1 goarch:amd64 goos:darwin goversion:1.20 output:amd64-compatible-go120]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:freebsd output:amd64]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:linux goversion:1.20 output:amd64-go120]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:linux output:amd64]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:windows goversion:1.20 output:amd64-go120]) (push) Waiting to run
Build / build (map[goamd64:v3 goarch:amd64 goos:windows output:amd64]) (push) Waiting to run
Build / build (map[goarch:386 goos:android ndk:i686-linux-android34 output:386]) (push) Waiting to run
Build / build (map[goarch:386 goos:freebsd output:386]) (push) Waiting to run
Build / build (map[goarch:386 goos:linux goversion:1.20 output:386-go120]) (push) Waiting to run
Build / build (map[goarch:386 goos:linux output:386]) (push) Waiting to run
Build / build (map[goarch:386 goos:windows goversion:1.20 output:386-go120]) (push) Waiting to run
Build / build (map[goarch:386 goos:windows output:386]) (push) Waiting to run
Build / build (map[goarch:amd64 goos:android ndk:x86_64-linux-android34 output:amd64]) (push) Waiting to run
Build / build (map[goarch:arm goarm:7 goos:linux output:armv7]) (push) Waiting to run
Build / build (map[goarch:arm goarm:7 goos:windows output:armv7]) (push) Waiting to run
Build / build (map[goarch:arm goos:android ndk:armv7a-linux-androideabi34 output:armv7]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:android ndk:aarch64-linux-android34 output:arm64-v8]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:darwin goversion:1.20 output:arm64-go120]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:darwin output:arm64]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:freebsd output:arm64]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:linux output:arm64]) (push) Waiting to run
Build / build (map[goarch:arm64 goos:windows output:arm64]) (push) Waiting to run
Build / build (map[goarch:mips goos:linux mips:hardfloat output:mips-hardfloat]) (push) Waiting to run
Build / build (map[goarch:mips goos:linux mips:softfloat output:mips-softfloat]) (push) Waiting to run
Build / build (map[goarch:mips64 goos:linux output:mips64]) (push) Waiting to run
Build / build (map[goarch:mips64le goos:linux output:mips64le]) (push) Waiting to run
Build / build (map[goarch:mipsle goos:linux mips:hardfloat output:mipsle-hardfloat]) (push) Waiting to run
Build / build (map[goarch:mipsle goos:linux mips:softfloat output:mipsle-softfloat]) (push) Waiting to run
Build / build (map[goarch:riscv64 goos:linux output:riscv64]) (push) Waiting to run
Build / build (map[goarch:s390x goos:linux output:s390x]) (push) Waiting to run
Build / Upload-Prerelease (push) Blocked by required conditions
Build / Upload-Release (push) Blocked by required conditions
Build / Docker (push) Blocked by required conditions
Trigger CMFA Update / trigger-CMFA-update (push) Waiting to run
140 lines
2.8 KiB
Go
140 lines
2.8 KiB
Go
package mmdb
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"sync"
|
|
"time"
|
|
|
|
mihomoOnce "github.com/metacubex/mihomo/common/once"
|
|
mihomoHttp "github.com/metacubex/mihomo/component/http"
|
|
C "github.com/metacubex/mihomo/constant"
|
|
"github.com/metacubex/mihomo/log"
|
|
|
|
"github.com/oschwald/maxminddb-golang"
|
|
)
|
|
|
|
type databaseType = uint8
|
|
|
|
const (
|
|
typeMaxmind databaseType = iota
|
|
typeSing
|
|
typeMetaV0
|
|
)
|
|
|
|
var (
|
|
IPreader IPReader
|
|
ASNreader ASNReader
|
|
IPonce sync.Once
|
|
ASNonce sync.Once
|
|
)
|
|
|
|
func LoadFromBytes(buffer []byte) {
|
|
IPonce.Do(func() {
|
|
mmdb, err := maxminddb.FromBytes(buffer)
|
|
if err != nil {
|
|
log.Fatalln("Can't load mmdb: %s", err.Error())
|
|
}
|
|
IPreader = IPReader{Reader: mmdb}
|
|
switch mmdb.Metadata.DatabaseType {
|
|
case "sing-geoip":
|
|
IPreader.databaseType = typeSing
|
|
case "Meta-geoip0":
|
|
IPreader.databaseType = typeMetaV0
|
|
default:
|
|
IPreader.databaseType = typeMaxmind
|
|
}
|
|
})
|
|
}
|
|
|
|
func Verify(path string) bool {
|
|
instance, err := maxminddb.Open(path)
|
|
if err == nil {
|
|
instance.Close()
|
|
}
|
|
return err == nil
|
|
}
|
|
|
|
func IPInstance() IPReader {
|
|
IPonce.Do(func() {
|
|
mmdbPath := C.Path.MMDB()
|
|
log.Infoln("Load MMDB file: %s", mmdbPath)
|
|
mmdb, err := maxminddb.Open(mmdbPath)
|
|
if err != nil {
|
|
log.Fatalln("Can't load MMDB: %s", err.Error())
|
|
}
|
|
IPreader = IPReader{Reader: mmdb}
|
|
switch mmdb.Metadata.DatabaseType {
|
|
case "sing-geoip":
|
|
IPreader.databaseType = typeSing
|
|
case "Meta-geoip0":
|
|
IPreader.databaseType = typeMetaV0
|
|
default:
|
|
IPreader.databaseType = typeMaxmind
|
|
}
|
|
})
|
|
|
|
return IPreader
|
|
}
|
|
|
|
func DownloadMMDB(path string) (err error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*90)
|
|
defer cancel()
|
|
resp, err := mihomoHttp.HttpRequest(ctx, C.MmdbUrl, http.MethodGet, http.Header{"User-Agent": {C.UA}}, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0o644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
_, err = io.Copy(f, resp.Body)
|
|
|
|
return err
|
|
}
|
|
|
|
func ASNInstance() ASNReader {
|
|
ASNonce.Do(func() {
|
|
ASNPath := C.Path.ASN()
|
|
log.Infoln("Load ASN file: %s", ASNPath)
|
|
asn, err := maxminddb.Open(ASNPath)
|
|
if err != nil {
|
|
log.Fatalln("Can't load ASN: %s", err.Error())
|
|
}
|
|
ASNreader = ASNReader{Reader: asn}
|
|
})
|
|
|
|
return ASNreader
|
|
}
|
|
|
|
func DownloadASN(path string) (err error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*90)
|
|
defer cancel()
|
|
resp, err := mihomoHttp.HttpRequest(ctx, C.ASNUrl, http.MethodGet, http.Header{"User-Agent": {C.UA}}, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0o644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
_, err = io.Copy(f, resp.Body)
|
|
|
|
return err
|
|
}
|
|
|
|
func ReloadIP() {
|
|
mihomoOnce.Reset(&IPonce)
|
|
}
|
|
|
|
func ReloadASN() {
|
|
mihomoOnce.Reset(&ASNonce)
|
|
}
|