Merge pull request #96 from helloxz/dev

3.3.0

cli/Version.go

@@ -7,8 +7,8 @@ var VersionDate string
 
 // 赋值全局变量
 func init() {
-	Version = "3.2.0"
-	VersionDate = "20230109"
+	Version = "3.3.0"
+	VersionDate = "20230326"
 }
 
 // 命令行打印版本

compile.sh

@@ -5,7 +5,7 @@ OS=$1
 ARCH=$2
 
 #Zdir版本号
-VERSION=3.2.0
+VERSION=3.3.0
 
 #编译linux
 compile_linux() {

controller/Dir.go

@@ -25,6 +25,17 @@ func Mkdir(c *gin.Context) {
 	//判断上级路径是否存在
 	full_path := public_dir + path
 
+	//需要验证用户传递的path参数是否合法
+	if !V_fpath(path) {
+		c.JSON(200, gin.H{
+			"code": -1000,
+			"msg":  "参数不合法！",
+			"data": "",
+		})
+		c.Abort()
+		return
+	}
+
 	//如果路径不存在
 	if !V_dir(full_path) {
 		c.JSON(200, gin.H{

controller/FileList.go

@@ -11,12 +11,8 @@ import (
 	"zdir/config"
 
 	"github.com/gin-gonic/gin"
-	"gopkg.in/ini.v1"
 )
 
-// 声明一个全局变量用来获取.ini配置
-var cfg *ini.File
-
 // 定义一个结构体，用来存放文件或文件夹信息
 type info struct {
 	Name  string

controller/Functions.go

@@ -7,6 +7,7 @@ import (
 	"math/rand"
 	"net/url"
 	"strings"
+	"time"
 
 	"github.com/gin-gonic/gin"
 )
@@ -20,6 +21,8 @@ func md5s(str string) string {
 
 // 生成一个随机字符串
 func RandStr(n int) string {
+	//如果seed固定，那么每次程序重启后重新生成随机数会重复上一次的随机数，所以这里设置一个随机seed
+	rand.Seed(time.Now().UnixNano())
 	var bytes []byte = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz1234567890")
 	result := make([]byte, n)
 	for i := 0; i < n; i++ {

controller/Upload.go

@@ -2,12 +2,16 @@ package controller
 
 import (
 	"os"
-	"regexp"
 	"zdir/config"
 
 	"github.com/gin-gonic/gin"
 )
 
+// 声明一个结构体，用于保存文件上传成功后的信息
+type fileinfo struct {
+	Url string `json:"url"`
+}
+
 func Upload(c *gin.Context) {
 	//获取文件路径
 	path := c.PostForm("path")
@@ -23,9 +27,8 @@ func Upload(c *gin.Context) {
 	}
 	//如果上传路径不合法
 	//判断用户传递的路径是否合法
-	var validPath = regexp.MustCompile(`^(\.|\..).+`)
-	v_re := validPath.MatchString(path)
-	if v_re {
+	v_re := V_fpath(path)
+	if !v_re {
 		c.JSON(200, gin.H{
 			"code": -1000,
 			"msg":  "文件夹名称不合法！",
@@ -59,14 +62,30 @@ func Upload(c *gin.Context) {
 
 	// 单文件
 	file, _ := c.FormFile("file")
+	file_name := file.Filename
+	//验证文件名是否合法
+	if !V_fname(file_name) {
+		c.JSON(200, gin.H{
+			"code": -1000,
+			"msg":  "文件名不合法！",
+			"data": "",
+		})
+		return
+	}
 
-	dst := full_path + file.Filename
+	dst := full_path + file_name
 	// 上传文件至指定的完整文件路径
 	c.SaveUploadedFile(file, dst)
+
+	//拼接url
+	var finfo fileinfo
+	//获取公共存储的域名
+	storage_domain := config.Public_domain(c)
+	finfo.Url = storage_domain + path + file_name
 	//返回上传成功
 	c.JSON(200, gin.H{
 		"code": 200,
 		"msg":  "success",
-		"data": "",
+		"data": finfo,
 	})
 }