github
/
Cloudreve
mirror of https://github.com/cloudreve/Cloudreve
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix: letter case should be ignored while validating file extension

pull/247/head
HFO4 2019-11-17 14:35:28 +08:00
parent 3741da692b
commit 5aa61301af
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/filesystem/validator.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"github.com/HFO4/cloudreve/pkg/util" "github.com/HFO4/cloudreve/pkg/util"
"path/filepath" "path/filepath"
"strings"
) )
// ValidateFileSize 验证上传的文件大小是否超出限制 // ValidateFileSize 验证上传的文件大小是否超出限制
@ -26,7 +27,7 @@ func (fs *FileSystem) ValidateExtension(ctx context.Context, fileName string) bo
return true return true
} }
ext := filepath.Ext(fileName) ext := strings.ToLower(filepath.Ext(fileName))
// 无扩展名时 // 无扩展名时
if len(ext) == 0 { if len(ext) == 0 {
return false return false

1
pkg/filesystem/validator_test.go
View File

@ -86,5 +86,6 @@ func TestFileSystem_ValidateExtension(t *testing.T) {
asserts.False(fs.ValidateExtension(ctx, "1.jpg.png")) asserts.False(fs.ValidateExtension(ctx, "1.jpg.png"))
asserts.True(fs.ValidateExtension(ctx, "1.txt")) asserts.True(fs.ValidateExtension(ctx, "1.txt"))
asserts.True(fs.ValidateExtension(ctx, "1.png.jpg")) asserts.True(fs.ValidateExtension(ctx, "1.png.jpg"))
asserts.True(fs.ValidateExtension(ctx, "1.png.jpG"))
asserts.False(fs.ValidateExtension(ctx, "1.png")) asserts.False(fs.ValidateExtension(ctx, "1.png"))
} }