From dbc38848c487831fb66c6736f26c811ca162ac23 Mon Sep 17 00:00:00 2001 From: An Date: Tue, 22 Aug 2023 22:42:11 +0800 Subject: [PATCH] Fix: Cross origin websockets not allowed (#463) --- web/handlers/subscribe.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/web/handlers/subscribe.py b/web/handlers/subscribe.py index 35195a38df4..351dafe3c1b 100644 --- a/web/handlers/subscribe.py +++ b/web/handlers/subscribe.py @@ -12,13 +12,13 @@ import time import traceback from typing import Any, Dict -from urllib.parse import quote +from urllib.parse import quote, urlparse import aiohttp from tornado import httputil from tornado.web import Application -from config import proxies +from config import proxies, domain from .base import * @@ -57,6 +57,10 @@ class SubscribeUpdatingHandler(BaseWebSocketHandler): updating = False updating_start_time = 0 + def check_origin(self, origin): + parsed_origin = urlparse(origin) + return parsed_origin.netloc.endswith(domain) + async def update(self, userid): SubscribeUpdatingHandler.updating = True SubscribeUpdatingHandler.updating_start_time = int(time.time())