starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-24 17:28:31 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

containers without port mapped will no longer be served

Browse Source
This commit is contained in:
yusing
2024-10-01 17:18:17 +08:00
parent 44cfd65f6c
commit c3b779a810
11 changed files with 50 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/net/http/middleware/x_forwarded.go
View File

@@ -4,23 +4,32 @@ import (
"net"
)
const (
xForwardedFor = "X-Forwarded-For"
xForwardedMethod = "X-Forwarded-Method"
xForwardedHost = "X-Forwarded-Host"
xForwardedProto = "X-Forwarded-Proto"
xForwardedURI = "X-Forwarded-Uri"
xForwardedPort = "X-Forwarded-Port"
)
var SetXForwarded = &Middleware{
rewrite: func(req *Request) {
req.Header.Del("Forwarded")
req.Header.Del("X-Forwarded-For")
req.Header.Del("X-Forwarded-Host")
req.Header.Del("X-Forwarded-Proto")
req.Header.Del(xForwardedFor)
req.Header.Del(xForwardedHost)
req.Header.Del(xForwardedProto)
clientIP, _, err := net.SplitHostPort(req.RemoteAddr)
if err == nil {
req.Header.Set("X-Forwarded-For", clientIP)
req.Header.Set(xForwardedFor, clientIP)
} else {
req.Header.Del("X-Forwarded-For")
req.Header.Del(xForwardedFor)
}
req.Header.Set("X-Forwarded-Host", req.Host)
req.Header.Set(xForwardedHost, req.Host)
if req.TLS == nil {
req.Header.Set("X-Forwarded-Proto", "http")
req.Header.Set(xForwardedProto, "http")
} else {
req.Header.Set("X-Forwarded-Proto", "https")
req.Header.Set(xForwardedProto, "https")
}
},
}
@@ -28,8 +37,8 @@ var SetXForwarded = &Middleware{
var HideXForwarded = &Middleware{
rewrite: func(req *Request) {
req.Header.Del("Forwarded")
req.Header.Del("X-Forwarded-For")
req.Header.Del("X-Forwarded-Host")
req.Header.Del("X-Forwarded-Proto")
req.Header.Del(xForwardedFor)
req.Header.Del(xForwardedHost)
req.Header.Del(xForwardedProto)
},
}