starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-25 01:38:30 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

fix(websocket): allow localhost origin in WebSocket manager

Browse Source
This commit is contained in:
yusing
2025-08-17 17:36:07 +08:00
parent 6ec1016f29
commit 8360aa59d1
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/net/gphttp/websocket/manager.go
View File

@@ -44,10 +44,7 @@ var defaultUpgrader = websocket.Upgrader{
if err != nil { if err != nil {
return false return false
} }
if u.Scheme != "http" && u.Scheme != "https" { if u.Host == "" {
return false
}
if len(u.Host) == 0 {
return false return false
} }
originHost := strings.ToLower(u.Hostname()) originHost := strings.ToLower(u.Hostname())
@@ -55,8 +52,10 @@ var defaultUpgrader = websocket.Upgrader{
if h, _, e := net.SplitHostPort(reqHost); e == nil { if h, _, e := net.SplitHostPort(reqHost); e == nil {
reqHost = h reqHost = h
} }
if reqHost == "127.0.0.1" || reqHost == "localhost" {
return true
}
reqHost = strings.ToLower(reqHost) reqHost = strings.ToLower(reqHost)
return originHost == reqHost return originHost == reqHost
}, },
} }