You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.6 KiB
105 lines
2.6 KiB
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"xtcfs/config"
|
|
"xtcfs/util"
|
|
)
|
|
|
|
// json 结构体
|
|
type Response struct {
|
|
Status string `json:"status"` //状态
|
|
Data FilesListJson `json:"data"` //目录下的文件
|
|
Curdir string `json:"curdir"` // 扫描的目录
|
|
WorksDir string `json:"workdir"` //监听目录
|
|
}
|
|
|
|
// 文件输出的结构
|
|
type FileJson struct {
|
|
Fname string `json:"fname"`
|
|
Dirflag bool `json:"dirflag"`
|
|
Isbackup int `json:"isbackup"`
|
|
Fhash string `json:"hash"` // hash
|
|
Fsize string `json:"size"` //文件大小, 输出带单位的大小
|
|
}
|
|
|
|
type FilesListJson struct {
|
|
Flist []FileJson `json:"list"`
|
|
}
|
|
|
|
// 遍历监视目录，发送到json中
|
|
func SerInfo(w http.ResponseWriter, r *http.Request) {
|
|
// 监听的目录通过?p=的方式传入
|
|
urlpath := r.URL.Query().Get("p")
|
|
// 防止逃逸，造成漏洞
|
|
if strings.Contains(urlpath, "../") {
|
|
urlpath = "."
|
|
}
|
|
|
|
// urlpath 进行base64 解码
|
|
dsrpath := util.Base64dec(urlpath)
|
|
// 监听的根目录
|
|
realFilePath := filepath.Join(config.G.FilePath, dsrpath)
|
|
// 时间目录的情况
|
|
fileInfo, err := os.Stat(realFilePath)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
// list json
|
|
var flist FilesListJson
|
|
//针对目录的情况才输出
|
|
// todo 如果是文件的话 暂时不处理
|
|
if fileInfo.IsDir() {
|
|
// 遍历目录
|
|
files, err := os.ReadDir(realFilePath)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
// 遍历
|
|
for _, v := range files {
|
|
// fmt.Printf("rang v:%v\n", v)
|
|
isbak := 0
|
|
// 如果有 backup
|
|
if strings.Contains(v.Name(), "backup") {
|
|
isbak = 1
|
|
}
|
|
// 如果是文件的话，就计算hash和大小
|
|
if v.IsDir() {
|
|
//While entry is A Directory
|
|
flist.Flist = append(flist.Flist, FileJson{Fname: v.Name(), Dirflag: v.IsDir(), Isbackup: isbak, Fhash: "", Fsize: ""})
|
|
} else {
|
|
// 计算文件的hash
|
|
funame := filepath.Join(realFilePath, v.Name())
|
|
fhash := util.CalacHash(funame)
|
|
// 文件大小
|
|
// 获取文件大小（以字节为单位）
|
|
sizeInBytes := fileInfo.Size()
|
|
sizeStr := fmt.Sprintf("%.2f KB", float64(sizeInBytes)/1024)
|
|
// output
|
|
flist.Flist = append(flist.Flist, FileJson{Fname: v.Name(), Dirflag: v.IsDir(), Isbackup: isbak, Fhash: fhash, Fsize: sizeStr})
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// respone file list
|
|
response := Response{
|
|
Status: "success",
|
|
Curdir: urlpath,
|
|
WorksDir: config.G.FilePath,
|
|
Data: flist,
|
|
}
|
|
|
|
// 开启跨域
|
|
util.CorsHadler(w, r)
|
|
json.NewEncoder(w).Encode(response)
|
|
}
|
|
|