From 21b67e97af05e1f7af9e9bd8b284b7119b085e14 Mon Sep 17 00:00:00 2001 From: yusing Date: Sun, 6 Oct 2024 01:21:35 +0800 Subject: [PATCH] websocket fix attempt#2 --- internal/api/v1/stats.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/internal/api/v1/stats.go b/internal/api/v1/stats.go index 3b17b99..29bcddc 100644 --- a/internal/api/v1/stats.go +++ b/internal/api/v1/stats.go @@ -20,12 +20,16 @@ func Stats(cfg *config.Config, w http.ResponseWriter, r *http.Request) { } func StatsWS(cfg *config.Config, w http.ResponseWriter, r *http.Request) { - originPats := cfg.Value().MatchDomains + localAddresses := []string{"127.0.0.1", "10.0.*.*", "172.16.*.*", "192.168.*.*"} + originPats := make([]string, len(cfg.Value().MatchDomains)+len(localAddresses)) + if len(originPats) == 0 { U.Logger.Warnf("no match domains configured, accepting websocket request from all origins") originPats = []string{"*"} } else { - localAddresses := []string{"127.0.0.1", "10.0.*.*", "172.16.*.*", "192.168.*.*"} + for i, domain := range cfg.Value().MatchDomains { + originPats[i] = "*." + domain + } originPats = append(originPats, localAddresses...) } if common.IsDebug {