Merge pull request #314 from cuteLittleDevil/patch-1

[fix] 修复火狐浏览器播放ws-flv失败的问题
q191201771 · Aug 31, 2023 · 6915b75 · 6915b75
2 parents 6af11b2 + 2263c7f
commit 6915b75
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/logic/http_server_handler.go b/pkg/logic/http_server_handler.go
@@ -61,7 +61,8 @@ func (h *HttpServerHandler) ServeSubSession(writer http.ResponseWriter, req *htt
 		isWebSocket  bool
 		webSocketKey string
 	)
-	if req.Header.Get("Connection") == "Upgrade" && req.Header.Get("Upgrade") == "websocket" {
+	// 火狐浏览器 Connection = [keep-alive, Upgrade]
+	if strings.Contains(req.Header.Get("Connection"), "Upgrade") && req.Header.Get("Upgrade") == "websocket" {
 		isWebSocket = true
 		webSocketKey = req.Header.Get("Sec-WebSocket-Key")
 	}