Enable check_origin on gateway websocket communication (#5471)

This is a port of the same change in NB2KG (PR 22). Fixes #5460 Co-authored-by: Dylan <dylan.yi@icloud.com> Co-authored-by: Dylan <dylan.yi@icloud.com>
jupyter · May 25, 2020 · 66ad3f4 · 66ad3f4
1 parent 01e298f
commit 66ad3f4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/notebook/gateway/handlers.py b/notebook/gateway/handlers.py
@@ -32,6 +32,9 @@ class WebSocketChannelsHandler(WebSocketHandler, IPythonHandler):
     kernel_id = None
     ping_callback = None
 
+    def check_origin(self, origin=None):
+        return IPythonHandler.check_origin(self, origin)
+
     def set_default_headers(self):
         """Undo the set_default_headers in IPythonHandler which doesn't make sense for websockets"""
         pass