feat(ssl): remove localAddresses from requests to plex.tv

Author: RoyXiang

handler/main.go

@@ -6,7 +6,6 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"os"
-	"strings"
 
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-redis/redis/v8"
@@ -47,24 +46,21 @@ func init() {
 func NewRouter() http.Handler {
 	r := mux.NewRouter()
 	r.Use(handlers.ProxyHeaders, normalizeMiddleware)
+	if !plexClient.NoRequestLogs {
+		r.Use(middleware.Logger)
+	}
 
 	plexTvUrl, _ := url.Parse("https://www." + domainPlexTv)
 	plexTvProxy := httputil.NewSingleHostReverseProxy(plexTvUrl)
 	plexTvRouter := r.Host(domainPlexTv).Subrouter()
-	plexTvRouter.Use(middleware.Logger)
-	sslRouter := plexTvRouter.MatcherFunc(func(r *http.Request, match *mux.RouteMatch) bool {
-		return strings.Index(r.URL.Path, "servers.xml") != -1
-	}).Subrouter()
+	sslRouter := plexTvRouter.Methods(http.MethodPost).Path("/servers.xml").Subrouter()
 	sslRouter.Use(sslMiddleware)
 	sslRouter.PathPrefix("/").Handler(plexTvProxy)
 	plexTvRouter.PathPrefix("/").Handler(plexTvProxy)
 
 	pmsRouter := r.MatcherFunc(func(r *http.Request, match *mux.RouteMatch) bool {
 		return r.Host != domainPlexTv
 	}).Subrouter()
-	if !plexClient.NoRequestLogs {
-		pmsRouter.Use(middleware.Logger)
-	}
 	pmsRouter.Use(wrapMiddleware, middleware.Recoverer, trafficMiddleware)
 	if redisClient != nil {
 		// bypass cache

handler/middleware.go

@@ -10,6 +10,7 @@ import (
 	"net/http/httptest"
 	"net/url"
 	"path/filepath"
+	"regexp"
 	"strconv"
 	"strings"
 	"time"
@@ -21,20 +22,23 @@ var (
 	cacheInfoCtxKey = &ctxKeyType{"cacheInfo"}
 	tokenCtxKey     = &ctxKeyType{"token"}
 	userCtxKey      = &ctxKeyType{"user"}
+
+	reLocalAddresses = regexp.MustCompile(` localAddresses="[^"]*"`)
 )
 
 func sslMiddleware(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if plexClient.sslHost != "" && r.Body != nil {
+			nr := cloneRequest(r, r.Header, r.URL.Query())
 			bodyBytes, _ := io.ReadAll(r.Body)
-			bodyStr := string(bodyBytes)
-			common.GetLogger().Print(bodyStr)
-
 			sslHost := fmt.Sprintf("address=\"%s\" scheme=\"https\"", plexClient.sslHost)
-			modifiedBody := strings.ReplaceAll(bodyStr, "host=\"\"", sslHost)
-			r.Body = io.NopCloser(strings.NewReader(modifiedBody))
+			modifiedBody := bytes.ReplaceAll(bodyBytes, []byte("host=\"\""), []byte(sslHost))
+			modifiedBody = reLocalAddresses.ReplaceAll(modifiedBody, []byte(""))
+			nr.Body = io.NopCloser(bytes.NewReader(modifiedBody))
+			next.ServeHTTP(w, nr)
+		} else {
+			next.ServeHTTP(w, r)
 		}
-		next.ServeHTTP(w, r)
 	})
 }