remove Referer header

fix #73
LiterMC · Oct 9, 2024 · 1865ea3 · 1865ea3
1 parent dd8238d
commit 1865ea3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cluster.go b/cluster.go
@@ -159,9 +159,11 @@ func NewCluster(
 
 		client: &http.Client{
 			Transport: transport,
+			CheckRedirect: redirectChecker,
 		},
 		cachedCli: &http.Client{
 			Transport: cachedTransport,
+			CheckRedirect: redirectChecker,
 		},
 
 		wsUpgrader: &websocket.Upgrader{
@@ -193,6 +195,14 @@ func NewCluster(
 	return
 }
 
+func redirectChecker(req *http.Request, via []*http.Request) error {
+	req.Header.Del("Referer")
+	if len(via) > 10 {
+		return errors.New("More than 10 redirects detected")
+	}
+	return nil
+}
+
 func (cr *Cluster) Init(ctx context.Context) (err error) {
 	// create data folder
 	os.MkdirAll(cr.dataDir, 0755)