sanitise file name
This commit is contained in:
parent
b05135420c
commit
2b64d64f80
1 changed files with 19 additions and 0 deletions
|
@ -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"})
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue