outposts/proxy: improve host header detection

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-05-11 20:02:36 +02:00
parent 84dfbcaaae
commit dd1a6a81c8
2 changed files with 12 additions and 4 deletions

View File

@ -1,12 +1,20 @@
package proxy
import "net/http"
import (
"net"
"net/http"
)
var xForwardedHost = http.CanonicalHeaderKey("X-Forwarded-Host")
func getHost(req *http.Request) string {
host := req.Host
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
}