在golang上的大型ip /子网列表中搜索ip的最快方法?

时间:2018-11-20 16:28:09

标签: go

请帮助我以最快的方式解决下一个任务

我有大量的IP /子网,例如...

35.132.199.128/27
8.44.144.248/32
87.117.185.193
45.23.45.45

等 而且我需要在该列表中找到一些禁食的IP。

当我尝试使用字符串切片和范围时,在大型列表上速度非常慢。

我可以使用map,例如map [string] string,它看起来可用,但仅用于ip检查,而不用于子网检查。

有人可以帮助我解决此任务吗?谢谢。

我的代码

func (app *application) validateIP(ip string) bool {

for _, item := range app.IPList {

    itemIsIP := net.ParseIP(item)

    if itemIsIP != nil {
        if ip == itemIsIP.String() {
            return true
        }
        continue
    }

    _, itemNet, err := net.ParseCIDR(item)
    if err != nil {
        log.Printf("[ERROR] %+v", err)
    }

    checkedIP := net.ParseIP(ip)

    if itemNet.Contains(checkedIP) {
        return true
    }
}
return false

}

1 个答案:

答案 0 :(得分:-1)

我们在项目中解决了这个问题:

isIPV4inCIDRList("35.132.199.128", []string{"35.132.199.128/27"})

func isIPV4inCIDRList(ip []byte, list []string) bool {
    for i := 0; i < 32; i++ {
        sm := strconv.Itoa(i)
        m, _ := mask(sm, 4)
        inv := andIP(ip, []byte(m))
        if len(inv) == 0 {
            continue
        }

        for _, cidr := range list {
            if cidr == inv.String() + "/" + sm {
                return true
            }
        }
    }

    return false
}

func andIP(ip, mask []byte) net.IP {
    inv := net.IP{}
    for i, v := range ip {
        inv = append(inv, mask[i]&v)
    }
    return inv
}

// Bigger than we need, not too big to worry about overflow
const big = 0xFFFFFF

// Decimal to integer.
// Returns number, characters consumed, success.
func dtoi(s string) (n int, i int, ok bool) {
    n = 0
    for i = 0; i < len(s) && '0' <= s[i] && s[i] <= '9'; i++ {
        n = n*10 + int(s[i]-'0')
        if n >= big {
            return big, i, false
        }
    }
    if i == 0 {
        return 0, 0, false
    }
    return n, i, true
}

func mask(m string, iplen int) (net.IPMask, error) {
    n, i, ok := dtoi(m)
    if !ok || i != len(m) || n < 0 || n > 8*iplen {
        return nil, &net.ParseError{Type: "CIDR address"}
    }
    return net.CIDRMask(n, 8*iplen), nil
}