outposts/proxy: improve host header detection
Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
parent
84dfbcaaae
commit
dd1a6a81c8
|
@ -1,12 +1,20 @@
|
||||||
package proxy
|
package proxy
|
||||||
|
|
||||||
import "net/http"
|
import (
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
var xForwardedHost = http.CanonicalHeaderKey("X-Forwarded-Host")
|
var xForwardedHost = http.CanonicalHeaderKey("X-Forwarded-Host")
|
||||||
|
|
||||||
func getHost(req *http.Request) string {
|
func getHost(req *http.Request) string {
|
||||||
|
host := req.Host
|
||||||
if req.Header.Get(xForwardedHost) != "" {
|
if req.Header.Get(xForwardedHost) != "" {
|
||||||
return req.Header.Get(xForwardedHost)
|
host = req.Header.Get(xForwardedHost)
|
||||||
}
|
}
|
||||||
return req.Host
|
hostOnly, _, err := net.SplitHostPort(host)
|
||||||
|
if err != nil {
|
||||||
|
return host
|
||||||
|
}
|
||||||
|
return hostOnly
|
||||||
}
|
}
|
||||||
|
|
Reference in New Issue