feat(cos): add self domain

MorvanZhou · May 24, 2024 · 9a293ed · 9a293ed
1 parent 8fa25c1
commit 9a293ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/retk/utils.py b/src/retk/utils.py
@@ -223,7 +223,7 @@ def ssrf_check(url: str) -> bool:
         return True
     # cos url
     settings = config.get_settings()
-    if host == f"{settings.COS_BUCKET_NAME}.cos.{settings.COS_REGION}.myqcloud.com":
+    if host == f"{settings.COS_BUCKET_NAME}.cos.{settings.COS_REGION}.myqcloud.com" or host == settings.COS_DOMAIN:
         return True
     try:
         ip = ipaddress.IPv4Address(socket.gethostbyname(host))