fix: 修复漏洞

Author: SilianZ

core/router.py

@@ -40,7 +40,7 @@ async def _(
             return response
 
         @self.route.get("/measure/{size}")
-        async def _(request: web.Request) -> web.StreamResponse:
+        async def _(request: web.Request) -> web.Response:
             try:
                 size = int(request.match_info.get("size", "0"))
 
@@ -53,7 +53,8 @@ async def _(request: web.Request) -> web.StreamResponse:
                 response = None
                 for storage in self.storages:
                     if isinstance(storage, AListStorage):
-                        await storage.measure(size, request, response)
+                        await storage.measure(size, response)
+                        await response.prepare(request)
                         return response
 
                 buffer = b"\x00\x66\xcc\xff" * 256 * 1024

core/storages/alist.py

@@ -89,7 +89,7 @@ async def getFileList(dir: str, pbar: tqdm) -> List[FileInfo]:
 
         return FileList(files=missing_files)
 
-    async def measure(self, size: int, request: web.Request, response):
+    async def measure(self, size: int, response):
         file_path = f"{self.path}/measure/.{size}"
         try:
             async with aiohttp.ClientSession(self.url, headers=self.headers) as session:
@@ -100,7 +100,6 @@ async def measure(self, size: int, request: web.Request, response):
                 if data["code"] == 200:
                     logger.debug(1)
                     response = web.HTTPFound(data["data"]["raw_url"])
-                    response.prepare(request)
                     return
 
                 if data["code"] != 200: