diff options
| -rw-r--r-- | generate.sh | 4 | ||||
| -rw-r--r-- | main.go | 148 |
2 files changed, 150 insertions, 2 deletions
diff --git a/generate.sh b/generate.sh index 1fa172ea..9631189a 100644 --- a/generate.sh +++ b/generate.sh @@ -7,8 +7,8 @@ unzip GeoLite2-Country-CSV.zip rm GeoLite2-Country-CSV.zip mv GeoLite2* geoip -CGO_ENABLED=0 go get -u v2ray.com/ext/tools/geoip/main -${GOPATH}/bin/main --country=./geoip/GeoLite2-Country-Locations-en.csv --ipv4=./geoip/GeoLite2-Country-Blocks-IPv4.csv --ipv6=./geoip/GeoLite2-Country-Blocks-IPv6.csv +go get -v -t -d github.com/v2ray/geoip/... +go run github.com/v2ray/geoip/main.go -- --country=./geoip/GeoLite2-Country-Locations-en.csv --ipv4=./geoip/GeoLite2-Country-Blocks-IPv4.csv --ipv6=./geoip/GeoLite2-Country-Blocks-IPv6.csv mkdir ./publish mv ./geoip.dat ./publish/ diff --git a/main.go b/main.go new file mode 100644 index 00000000..22a269f2 --- /dev/null +++ b/main.go @@ -0,0 +1,148 @@ +// GeoIP generator +// +// Before running this file, the GeoIP database must be downloaded and present. +// To download GeoIP database: https://dev.maxmind.com/geoip/geoip2/geolite2/ +// Inside you will find block files for IPv4 and IPv6 and country code mapping. +package main + +import ( + "encoding/csv" + "flag" + "fmt" + "io/ioutil" + "os" + "strings" + + "github.com/gogo/protobuf/proto" + "v2ray.com/core/app/router" + "v2ray.com/core/common" + "v2ray.com/ext/tools/conf" +) + +var ( + countryCodeFile = flag.String("country", "", "Path to the country code file") + ipv4File = flag.String("ipv4", "", "Path to the IPv4 block file") + ipv6File = flag.String("ipv6", "", "Path to the IPv6 block file") +) + +func getCountryCodeMap() (map[string]string, error) { + countryCodeReader, err := os.Open(*countryCodeFile) + if err != nil { + return nil, err + } + defer countryCodeReader.Close() + + m := make(map[string]string) + reader := csv.NewReader(countryCodeReader) + lines, err := reader.ReadAll() + if err != nil { + return nil, err + } + for _, line := range lines[1:] { + id := line[0] + countryCode := line[4] + if len(countryCode) == 0 { + continue + } + m[id] = strings.ToUpper(countryCode) + } + return m, nil +} + +func getCidrPerCountry(file string, m map[string]string, list map[string][]*router.CIDR) error { + fileReader, err := os.Open(file) + if err != nil { + return err + } + defer fileReader.Close() + + reader := csv.NewReader(fileReader) + lines, err := reader.ReadAll() + if err != nil { + return err + } + for _, line := range lines[1:] { + cidrStr := line[0] + countryId := line[1] + if countryCode, found := m[countryId]; found { + cidr, err := conf.ParseIP(cidrStr) + if err != nil { + return err + } + cidrs := append(list[countryCode], cidr) + list[countryCode] = cidrs + } + } + return nil +} + +func main() { + flag.Parse() + + ccMap, err := getCountryCodeMap() + if err != nil { + fmt.Println("Error reading country code map:", err) + return + } + + cidrList := make(map[string][]*router.CIDR) + if err := getCidrPerCountry(*ipv4File, ccMap, cidrList); err != nil { + fmt.Println("Error loading IPv4 file:", err) + return + } + if err := getCidrPerCountry(*ipv6File, ccMap, cidrList); err != nil { + fmt.Println("Error loading IPv6 file:", err) + return + } + + geoIPList := new(router.GeoIPList) + for cc, cidr := range cidrList { + geoIPList.Entry = append(geoIPList.Entry, &router.GeoIP{ + CountryCode: cc, + Cidr: cidr, + }) + } + geoIPList.Entry = append(geoIPList.Entry, getLocalIPs()) + + geoIPBytes, err := proto.Marshal(geoIPList) + if err != nil { + fmt.Println("Error marshalling geoip list:", err) + } + + if err := ioutil.WriteFile("geoip.dat", geoIPBytes, 0777); err != nil { + fmt.Println("Error writing geoip to file:", err) + } +} + +var ( + localIPs = []string{ + "0.0.0.0/8", + "10.0.0.0/8", + "100.64.0.0/10", + "127.0.0.0/8", + "169.254.0.0/16", + "172.16.0.0/12", + "192.0.0.0/24", + "192.0.2.0/24", + "192.168.0.0/16", + "198.18.0.0/15", + "198.51.100.0/24", + "203.0.113.0/24", + "::1/128", + "fc00::/7", + "fe80::/10", + } +) + +func getLocalIPs() *router.GeoIP { + cidr := make([]*router.CIDR, 0, 16) + for _, ip := range localIPs { + c, err := conf.ParseIP(ip) + common.Must(err) + cidr = append(cidr, c) + } + return &router.GeoIP{ + CountryCode: "PRIVATE", + Cidr: cidr, + } +} |