Make it possible to proxy through a Unix domain socket.

2manymws · Nov 24, 2023 · 88e3ac0 · 88e3ac0
1 parent f804e59
commit 88e3ac0
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/rp.go b/rp.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"net/http/httputil"
 	"net/url"
+	"strings"
 )
 
 const errorKey = "X-Proxy-Error"
@@ -73,7 +74,11 @@ func NewRouter(r Relayer) http.Handler {
 					return
 				}
 				if u != nil {
-					pr.Out.Host = u.Host
+					if strings.HasPrefix(u.Host, "/") {
+						pr.Out.Host = pr.In.Host
+					} else {
+						pr.Out.Host = u.Host
+					}
 					pr.Out.URL = u
 					pr.SetXForwarded()
 				}
@@ -89,9 +94,13 @@ func NewRouter(r Relayer) http.Handler {
 				return
 			}
 			if u != nil {
-				pr.Out.Host = u.Host
-				pr.Out.URL = u
+				if strings.HasPrefix(u.Host, "/") {
+					pr.Out.Host = pr.In.Host
+				} else {
+					pr.Out.Host = u.Host
+				}
 
+				pr.Out.URL = u
 			}
 			if err := rr.Rewrite(pr); err != nil {
 				pr.Out.Header.Set(errorKey, err.Error())