implement CheckOrigin on server-side

ptr-geeks · Aug 20, 2021 · df158c1 · df158c1
1 parent f84d108
commit df158c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"
 		},
 	}
 )