sanitise file name

This commit is contained in:
grngxd 2025-06-09 23:11:32 +01:00
parent b05135420c
commit 2b64d64f80

View file

@ -114,12 +114,31 @@ func RegisterFileRoutes(cfg *types.StereoConfig, api *gin.RouterGroup) {
api.GET("/:name", func(c *gin.Context) {
name := c.Param("name")
name = strings.TrimSpace(name)
safe := ""
for _, r := range name {
if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') ||
(r >= '0' && r <= '9') || r == '_' || r == '.' || r == '-' {
safe += string(r)
} else {
safe += "_"
}
}
name = safe
parts := strings.SplitN(name, "_", 2)
if len(parts) != 2 {
c.JSON(400, gin.H{"error": "invalid file name"})
return
}
uid, filename := parts[0], parts[1]
if uid == "" || filename == "" {
c.JSON(400, gin.H{"error": "invalid file name"})
return
}
path := filepath.Join(cfg.ImagePath, uid, filename)
if _, err := os.Stat(path); err != nil {
c.JSON(404, gin.H{"error": "file not found"})