1 files changed, 20 insertions, 0 deletions

diff --git a/echo/middleware/xff.go b/echo/middleware/xff.go
new file mode 100644
index 0000000..d110e12
--- /dev/null
+++ b/echo/middleware/xff.go
@@ -0,0 +1,20 @@
+package middleware
+
+import (
+        "net"
+
+        "github.com/labstack/echo/v4"
+)
+
+func ExtractIPFromXFFHeaders(trustLinkLocal bool, ips []string) (echo.IPExtractor, error) {
+        xffOpts := []echo.TrustOption{echo.TrustLinkLocal(trustLinkLocal)}
+        for _, r := range ips {
+                _, pn, err := net.ParseCIDR(r)
+                if err != nil {
+                        return nil, err
+                }
+                xffOpts = append(xffOpts, echo.TrustIPRange(pn))
+        }
+
+        return echo.ExtractIPFromXFFHeader(xffOpts...), nil
+}