如何在gin-gonic/gin中间件中实现高效的IP白名单

Question

我有一个应用程序需要限制为少数 IP。如果请求 IP 不在允许列表中，我可以编写一个中间件并返回，但是我希望这个过程尽可能高效。 IE。我想尽早断开连接。我可以断开连接的最早阶段是什么，最好是使用 HTTP 响应。 我无法控制主机防火墙或边界防火墙来过滤流量，而且即使我控制了防火墙，我也无法提供 HTTP 响应。

如果我能在 gin 中获得关于 HTTP 请求生命周期的描述，我更愿意。

Answer 1

添加一个如 Lansana 所述的中间件。

尽早在链中声明它很重要。

r := gin.New()

whitelist := make(map[string]bool)
whitelist["127.0.0.1"] = true

r.Use(middleware.IPWhiteList(whitelist))

我会这样写中间件，如果你不在白名单中，返回一个合适的错误，在下面的sn-p中我返回一个JSON错误。

package middleware

import (
    "log"
    "net/http"

    "github.com/gin-gonic/gin"
)

func IPWhiteList(whitelist map[string]bool) gin.HandlerFunc {
    return func(c *gin.Context) {
        if !whitelist[c.ClientIP()] {
            c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
                "status":  http.StatusForbidden,
                "message": "Permission denied",
            })
            return
        }
    }
}

Answer 2

你读过Gin documentation吗？

您可以根据this section 中的示例添加中间件。使用router.Use() 链接中间件。如果您希望它尽可能高效，请将其设置为链中的第一个中间件，并使用允许 O(1) 对允许的 IP 进行读取访问的数据结构。

例如：

var myWhiteList map[string]bool = map[string]bool{
    "1.2.3.4": true,
    "4.3.2.1": true,
}

func main() {
    // Creates a router without any middleware by default
    r := gin.New()

    // Add whitelist middleware
    r.Use(middleware.IPWhitelist(myWhiteList))

    // Listen and serve on 0.0.0.0:8080
    r.Run(":8080")
}

middleware.IPWhitelist 的实现可能如下所示：

func IPWhiteList(whitelist map[string]bool) func(http.Handler) http.Handler {
    f := func(h http.Handler) http.Handler {
        fn := func(w http.ResponseWriter, r *http.Request) {
            // Get IP of this request
            ip := doSomething()

            // If the IP isn't in the whitelist, forbid the request.
            if !whitelist[ip] {
                w.Header().Set("Content-Type", "text/plain")
                w.WriteHeader(http.StatusForbidden)
                w.Write([]byte("."))
                return
            }

            h.ServeHTTP(w, r)
        }

        return http.HandlerFunc(fn)
    }

    return f
}