mirror of https://github.com/v2ray/v2ray-core
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
// +build generate
|
|
|
|
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"math"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
)
|
|
|
|
const (
|
|
apnicFile = "http://ftp.apnic.net/apnic/stats/apnic/delegated-apnic-latest"
|
|
)
|
|
|
|
func main() {
|
|
resp, err := http.Get(apnicFile)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if resp.StatusCode != 200 {
|
|
panic(fmt.Errorf("Unexpected status %d", resp.StatusCode))
|
|
}
|
|
defer resp.Body.Close()
|
|
scanner := bufio.NewScanner(resp.Body)
|
|
|
|
ipNet := v2net.NewIPNet()
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
line = strings.TrimSpace(line)
|
|
parts := strings.Split(line, "|")
|
|
if len(parts) < 5 {
|
|
continue
|
|
}
|
|
if strings.ToLower(parts[1]) != "cn" || strings.ToLower(parts[2]) != "ipv4" {
|
|
continue
|
|
}
|
|
ip := parts[3]
|
|
count, err := strconv.Atoi(parts[4])
|
|
if err != nil {
|
|
continue
|
|
}
|
|
mask := 32 - int(math.Floor(math.Log2(float64(count))+0.5))
|
|
cidr := fmt.Sprintf("%s/%d", ip, mask)
|
|
_, t, err := net.ParseCIDR(cidr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
ipNet.Add(t)
|
|
}
|
|
dump := ipNet.Serialize()
|
|
|
|
file, err := os.OpenFile("chinaip_init.go", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
|
|
if err != nil {
|
|
log.Fatalf("Failed to generate chinaip_init.go: %v", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
fmt.Fprintln(file, "package rules")
|
|
fmt.Fprintln(file, "import (")
|
|
fmt.Fprintln(file, "v2net \"github.com/v2ray/v2ray-core/common/net\"")
|
|
fmt.Fprintln(file, ")")
|
|
|
|
fmt.Fprintln(file, "var (")
|
|
fmt.Fprintln(file, "chinaIPNet *v2net.IPNet")
|
|
fmt.Fprintln(file, ")")
|
|
|
|
fmt.Fprintln(file, "func init() {")
|
|
|
|
fmt.Fprintln(file, "chinaIPNet = v2net.NewIPNetInitialValue(map[uint32]byte {")
|
|
for i := 0; i < len(dump); i += 2 {
|
|
fmt.Fprintln(file, dump[i], ": ", dump[i+1], ",")
|
|
}
|
|
fmt.Fprintln(file, "})")
|
|
fmt.Fprintln(file, "}")
|
|
}
|