Enhance WebSocket upgrade detection in Mihomo

- Updated the `isWebSocketUpgrade` function to improve handling of WebSocket upgrade requests by considering the presence of the "Sec-WebSocket-Key" header as a strong indicator.
- Added a new test case in `TestIsWebSocketUpgrade` to validate the detection logic for requests with the "Sec-WebSocket-Key" header, ensuring comprehensive test coverage.

internal/proxy/mihomo.go

@@ -36,10 +36,14 @@ func isWebSocketUpgrade(r *http.Request) bool {
 	if r == nil {
 		return false
 	}
-	// Some clients/proxies pass comma-separated tokens or extra spaces.
+	// Be tolerant to proxy/header quirks:
-	// Treat request as WS only when both headers contain required upgrade tokens.
+	// - RFC path: Connection: upgrade + Upgrade: websocket
-	return headerHasToken(r.Header, "Connection", "upgrade") &&
+	// - Fallback: Sec-WebSocket-Key presence strongly indicates WS handshake.
-		headerHasToken(r.Header, "Upgrade", "websocket")
+	if headerHasToken(r.Header, "Upgrade", "websocket") &&
+		headerHasToken(r.Header, "Connection", "upgrade") {
+		return true
+	}
+	return strings.TrimSpace(r.Header.Get("Sec-WebSocket-Key")) != ""
 }
 
 func headerHasToken(h http.Header, key, token string) bool {

internal/proxy/mihomo_test.go

@@ -53,4 +53,10 @@ func TestIsWebSocketUpgrade(t *testing.T) {
 	if isWebSocketUpgrade(r2) {
 		t.Fatal("expected non-websocket when Connection lacks upgrade")
 	}
+
+	r3 := httptest.NewRequest(http.MethodGet, "http://gw/api/mtg/mihomo/traffic", nil)
+	r3.Header.Set("Sec-WebSocket-Key", "dGhlIHNhbXBsZSBub25jZQ==")
+	if !isWebSocketUpgrade(r3) {
+		t.Fatal("expected websocket upgrade when Sec-WebSocket-Key is present")
+	}
 }