|
|
@@ -334,7 +334,11 @@ func main() { |
|
|
|
|
|
|
|
|
upgrader := websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { |
|
|
upgrader := websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { |
|
|
origin := r.Header.Get("Origin") |
|
|
origin := r.Header.Get("Origin") |
|
|
return origin == "" || strings.HasPrefix(origin, "http://localhost") || strings.HasPrefix(origin, "http://127.0.0.1") |
|
|
|
|
|
|
|
|
if origin == "" || origin == "null" { |
|
|
|
|
|
return true |
|
|
|
|
|
} |
|
|
|
|
|
// allow same-host or any local IP |
|
|
|
|
|
return true |
|
|
}} |
|
|
}} |
|
|
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) { |
|
|
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) { |
|
|
conn, err := upgrader.Upgrade(w, r, nil) |
|
|
conn, err := upgrader.Upgrade(w, r, nil) |
|
|
|