http - 将对 GO web 服务的访问限制为来自本地子网的客户端

Question

我正在使用 GO http 包在 GO 中编写一个小型 Web 服务。我想将对 Web 服务的访问限制为本地子网(127.0.0.1、10.0.0.0/8、172.16.0.0/12、192.168.0.0/16)上的客户端。

我尝试使用子网掩码作为 ListenAndServe 的地址参数但它退出时出现“没有这样的主机”错误。

编辑:

这是我在@RickA 和@Dewy Broto 的帮助下提出的解决方案。

func JustLocal(handler http.Handler) http.Handler {
    var local_subnets []*net.IPNet
    local_subnet_s := []string{"127.0.0.1/31", "10.0.0.0/8", "172.16.0.0/12", "192.168.0.0/16"}

    for _,net_s := range local_subnet_s {
        _, n, _ := net.ParseCIDR(net_s)
        local_subnets = append(local_subnets, n)
    }

    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        fmt.Println(r.RemoteAddr)
        remote_ip := net.ParseIP(strings.Split(r.RemoteAddr, ":")[0])
        fmt.Println(remote_ip)

        local := false
        for _, local_subnet := range local_subnets {
            fmt.Println(local_subnet, remote_ip)
            if local_subnet.Contains(remote_ip) {
                local = true
                break
            }
        }
        if !local {
            http.Error(w, "go away", 403)
            return
        }
        handler.ServeHTTP(w, r)
        return
    })
}

它的边缘有点粗糙，但据我所知它是有效的。感谢所有的帮助!

Answer 1

编写一个处理程序，在委托(delegate)给另一个处理程序之前检查客户端的 IP 地址是否被允许:

type authorizationHandler struct {
    nets []*net.IPNet
    h handler
}

func (ah *authorizationHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    ok := false
    ip := net.ParseIP(r.RemoteAddr)
    for _, n := range ah.nets {
       if n.Contains(ip) {
         ok := true
         break
       }
    }
    if !ok {
       http.Error(w, "go away", 403)
       return
    }
    ah.h(w, r)
}

使用这个处理程序来包装你的根处理程序:

err := ListenAndServe(addr, 
          &authorizationHandler{nets: allowedNets, h:rootHandler))

注意:上面的代码是我在没有编译和测试的情况下输入的。可能需要围绕解析和测试远程地址进行一些调整。