diff --git a/server/internal/ws/server.go b/server/internal/ws/server.go index 28524d9..9a8becf 100644 --- a/server/internal/ws/server.go +++ b/server/internal/ws/server.go @@ -17,7 +17,8 @@ var ( WriteBufferSize: 1024, // TODO: For now we don't really check origin CheckOrigin: func(r *http.Request) bool { - return true + origin := r.Header.Get("Origin") + return origin == "https://ptrun.develop.cloud.ptr.si" }, } )