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.
141 lines
3.7 KiB
141 lines
3.7 KiB
package core
|
|
|
|
import (
|
|
"aufs/config"
|
|
"aufs/db"
|
|
"aufs/util"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// json 结构体
|
|
type Response struct {
|
|
Status string `json:"status"` //状态
|
|
Data FilesListJson `json:"data"` //目录下的文件
|
|
Curdir string `json:"curdir"` // 扫描的目录
|
|
WorksDir string `json:"workdir"` //监听目录
|
|
Hostip string `json:"hostip"` //运行的主机ip
|
|
}
|
|
|
|
// 文件输出的结构
|
|
type FileJson struct {
|
|
Fname string `json:"fname"`
|
|
Dirflag bool `json:"dirflag"`
|
|
Isbackup int `json:"isbackup"`
|
|
Fhash string `json:"hash"` // hash
|
|
Fsize string `json:"size"` //文件大小, 输出带单位的大小
|
|
Frehash string `json:"rehash"` // 参考hash
|
|
}
|
|
|
|
type FilesListJson struct {
|
|
Flist []FileJson `json:"list"`
|
|
}
|
|
|
|
// 获取运行的路径
|
|
var Gpath string
|
|
|
|
// 遍历监视目录，发送到json中
|
|
func SerInfo(w http.ResponseWriter, r *http.Request) {
|
|
// 监听的目录通过?p=的方式传入
|
|
urlpath := r.URL.Query().Get("p")
|
|
// 防止逃逸，造成漏洞
|
|
if strings.Contains(urlpath, "../") || 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
|
|
}
|
|
// 链接数据库
|
|
db.Init()
|
|
// 获取请求的信息
|
|
rcflists := db.Fquery(util.Bas64end(dsrpath))
|
|
//
|
|
// 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
|
|
rchash := "" //参考的hash
|
|
// 如果有 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: "", Frehash: ""})
|
|
} else {
|
|
// 计算文件的hash
|
|
funame := filepath.Join(realFilePath, v.Name())
|
|
fhash := util.CalacHash(funame)
|
|
//
|
|
for _, av := range rcflists {
|
|
if av.Fname == v.Name() {
|
|
rchash = av.Fhash
|
|
}
|
|
}
|
|
// 文件大小
|
|
// 获取文件大小（以字节为单位）
|
|
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, Frehash: rchash})
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// respone file list
|
|
response := Response{
|
|
Status: "success",
|
|
Curdir: urlpath,
|
|
WorksDir: config.G.FilePath,
|
|
Hostip: config.G.LocalIP,
|
|
Data: flist,
|
|
}
|
|
|
|
// 开启跨域
|
|
uCorsHadler(w, r)
|
|
json.NewEncoder(w).Encode(response)
|
|
}
|
|
|
|
// 跨域函数
|
|
func uCorsHadler(w http.ResponseWriter, r *http.Request) {
|
|
// 设置跨域响应头
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS,PUT,DELETET")
|
|
// w.Header().Set("Access-Control-Allow-Headers", "Content-Type,Accept,Accept-Length,Accept-Encoding,X-XSRF-TOKEN,X-XSRF-TOKEN")
|
|
w.Header().Set("Access-Control-Allow-Headers", "*")
|
|
//
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
// 如果是OPTIONS请求，返回200 OK
|
|
if r.Method == "OPTIONS" {
|
|
// fmt.Printf("options is now \n")
|
|
// w.WriteHeader(http.StatusOK)
|
|
return
|
|
}
|
|
}
|
|
|