fix redir stuff i hope

master v0.1
Bel LaPointe 2019-04-06 17:02:42 -06:00
parent ae56c1744e
commit 4c4791ee0f
2 changed files with 26 additions and 5 deletions

View File

@ -1,11 +1,11 @@
p: 54243
r:
- echo:http://localhost:49982
- echo2:http://localhost:49983
crt: ./testdata/rproxy3server.crt
key: ./testdata/rproxy3server.key
user: bel
pass: bel
- echo2:http://192.168.0.86:38090
#crt: ./testdata/rproxy3server.crt
#key: ./testdata/rproxy3server.key
#user: bel
#pass: bel
rate: 1
burst: 2
timeout: 10

View File

@ -9,8 +9,17 @@ import (
"strings"
)
type redirPurge struct {
proxyHost string
targetHost string
}
func (s *Server) Proxy(w http.ResponseWriter, r *http.Request) {
newURL, err := s.lookup(r.Host)
transport := &redirPurge{
proxyHost: r.Host,
targetHost: newURL.Host,
}
if err != nil {
http.NotFound(w, r)
log.Printf("unknown host lookup %q", r.Host)
@ -18,6 +27,7 @@ func (s *Server) Proxy(w http.ResponseWriter, r *http.Request) {
}
r.Host = newURL.Host
proxy := httputil.NewSingleHostReverseProxy(newURL)
proxy.Transport = transport
proxy.ServeHTTP(w, r)
}
@ -28,3 +38,14 @@ func (s *Server) lookup(host string) (*url.URL, error) {
err := s.db.Get(nsRouting, host, v)
return v.URL(), err
}
func (rp *redirPurge) RoundTrip(r *http.Request) (*http.Response, error) {
resp, err := http.DefaultTransport.RoundTrip(r)
if err != nil {
return resp, err
}
if loc := resp.Header.Get("Location"); loc != "" {
resp.Header.Set("Location", strings.Replace(loc, rp.targetHost, rp.proxyHost, 1))
}
return resp, err
}