wood/webp_server_go
1
0
Fork 0
mirror of https://github.com/woodchen-ink/webp_server_go.git synced 2025-07-18 05:32:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(router): improve image file type checking and error messages

Browse Source
This commit is contained in:
wood chen 2024-10-23 06:24:27 +08:00
parent 2b46043bd0
commit 11d6f95787
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
handler/router.go
View File

@ -95,7 +95,7 @@ func handleLocalImage(c *fiber.Ctx, matchedTarget, reqURI, exhaustFilename strin
rawImageAbs := path.Join(matchedTarget, reqURI)
if !helper.FileExists(rawImageAbs) {
return c.Status(fiber.StatusNotFound).SendString("本地文件不存在")
return c.Status(fiber.StatusNotFound).SendString("文件不存在")
}
if !helper.IsAllowedImageFile(path.Base(reqURI)) {
@ -115,17 +115,18 @@ func handleRemoteImage(c *fiber.Ctx, matchedTarget, matchedPrefix, reqURIwithQue
realRemoteAddr := buildRealRemoteAddr(targetUrl, matchedPrefix, reqURIwithQuery)
// 首先检查是否为允许的图片文件
if !helper.IsAllowedImageFile(path.Base(reqURIwithQuery)) {
log.Infof("不允许的文件类型或非图片文件: %s", reqURIwithQuery)
return c.Redirect(realRemoteAddr, 302)
}
rawImageAbs, isNewDownload, err := fetchRemoteImg(realRemoteAddr, targetUrl.Host)
if err != nil {
log.Errorf("获取远程图像失败: %v", err)
return c.Status(fiber.StatusInternalServerError).SendString("无法获取远程图像")
}
if !helper.IsAllowedImageFile(path.Base(reqURIwithQuery)) {
log.Infof("不允许的文件类型或非图片文件: %s", reqURIwithQuery)
return c.Redirect(realRemoteAddr, 302)
}
err = processAndSaveImage(c, rawImageAbs, exhaustFilename, extraParams)
if err != nil {
return err