Golang IP地址片中IP检查的有效方法

使用“集合”

打造我们的系列

没有内置 Set 输入Go,但您可以优雅地使用 map[Type]bool 作为一套,例如:

// Create a set with 2 values in it: [1, 2]
m := map[int]bool{1: true, 2: true}

// Test an element:
fmt.Println(m[1]) // true
fmt.Println(m[3]) // false

// Set an element:
m[3] = true
fmt.Println(m[3]) // true

// Delete an element:
delete(m, 1)
fmt.Println(m[1]) // false

zero value 值类型,即 false 万一 bool

在 Go Playground .

^{注释#2 :有一些技巧可以使处理映射集的代码更短,您可以在下面的答案中进行检查: Check if a value is in a list .}

使用 net.IP 在集合中

现在我们只需要一个表示 net.IP How can I prevent a type being used as a map key? ).

不幸地 网络IP 它本身不合格,因为它是一个切片:

type IP []byte

切片是不可比的。有关详细信息,请参阅此问题: Hash with key as an array type 还有这个: Why have arrays in Go?

一种简单的方法是将其转换为规范 string 价值,我们就完成了。为此,我们可以简单地将IP的字节转换为十六进制 。但IPv4地址可能表示为IPv6,因此我们应首先将其转换为IPv6:

func Key(ip net.IP) string {
    return hex.EncodeToString(ip.To16())
}

注意:IP地址的字节可能不是有效的UTF-8编码 一串 (这就是Go商店的方式 一串 内存中的),但 一串 Go中的值表示任意字节序列,因此以下内容也适用,更简单,效率更高:

func Key(ip net.IP) string {
    return string(ip.To16())  // Simple []byte => string conversion
}

// Populate forbidden IPs:
forbIPs := map[string]bool{
    Key(ip1): true,
    Key(ip2): true,
}

// Now check a single IP:
ipToCheck := ...
if forbIPs[Key(ipToCheck)] {
    fmt.Println("Forbidden!")
} else {
    fmt.Println("Allowed.")
}

net.LookupIP() ),这是一个单人间 for 循环:

ips, err := net.LookupIP(host)
// Check err
for _, ip := range ips {
    if forbIPs[Key(ip)] {
        // FORBIDDEN!
    }
}

备用密钥类型

注意,如上所述,片是不可比较的,但数组是可比较的。所以我们也可以使用数组作为键。这就是它的样子:

func Key(ip net.IP) (a [16]byte) {
    copy(a[:], ip)
    return
}

// And the IP set:
forbIPs := map[[16]byte]bool{
    // ...
}

选择

已排序切片

或者,我们可以简单地将禁止的IP存储在一个切片中 []net.IP 和 保持排序 。如果已排序,我们可以使用二进制搜索在其中查找IP(标准库 sort.Search() ).

是的,二进制搜索有 O(log2(n)) 复杂性与 O(1) 上述(散列)映射解决方案的复杂性。但这种替代方案有另一个优点:

枚举单个IP并不总是可行的。有时(通常)更容易列出IP范围。第一种解决方案不适用于处理IP范围,但此解决方案可能是:您可以在中找到覆盖IP地址的范围 O(log2(n)) 时间

您可以使用 func (ip IP) Equal(x IP) bool 从…起 net 包裹:

Equal报告ip和x是否是相同的ip地址。IPv4 地址和IPv6形式的相同地址被视为相等。

就像这个工作示例:

package main

import (
    "fmt"
    "net"
)

func main() {
    ip := net.ParseIP("127.0.0.1")
    ips, err := net.LookupIP("localhost")
    if err != nil {
        panic(err)
    }
    for _, v := range ips {
        if v.Equal(ip) {
            fmt.Println(v)
        }
    }
}