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

@ -57,7 +57,7 @@ func (s *Server) handler(w http.ResponseWriter, r *http.Request) {
for k := range s.Handlers {
hostKeys = append(hostKeys, k)
}
s.logger.WithField("host", host).WithField("known-hosts", strings.Join(hostKeys, ", ")).Debug("Host header does not match any we know of")
s.logger.WithField("host", host).WithField("known-hosts", strings.Join(hostKeys, ",")).Debug("Host header does not match any we know of")
w.WriteHeader(404)
return
}

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
}