package tools import ( "fmt" "io" "io/ioutil" "net/http" "os" "path/filepath" "strconv" "strings" "b612.me/starainrt" "b612.me/starlog" "github.com/spf13/cobra" ) var port, ip, path string var up bool type TraceHandler struct { h http.Handler } // httpCmd represents the http command var httpcmd = &cobra.Command{ Use: "http", Short: "HTTP文件服务器", Long: `HTTP文件服务器`, Run: func(cmd *cobra.Command, args []string) { http.HandleFunc("/", httplisten) path, _ = filepath.Abs(path) fmt.Println("Listening On Port:" + port) if up { fmt.Println("upload is openned,path is /vtqeupload1127") http.HandleFunc("/vtqeupload1127", uploadfile) } err := http.ListenAndServe(ip+":"+port, nil) if err != nil { starlog.Println("Error:"+err.Error(), "red", "") } }, } func uploadfile(w http.ResponseWriter, r *http.Request) { if r.Method != "POST" { w.Write([]byte("USE POST METHOD!")) return } r.ParseMultipartForm(10485760) file, handler, err := r.FormFile("victorique") if err != nil { fmt.Println(err) w.WriteHeader(502) w.Write([]byte(err.Error())) return } defer file.Close() fmt.Printf("Upload %s From %s\n", handler.Filename, r.RemoteAddr) fmt.Fprintf(w, `

Return To Web Page

`, handler.Header) os.Mkdir("./vtqeupload1127", 0755) f, err := os.OpenFile("./vtqeupload1127/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0755) if err != nil { fmt.Println(err) return } defer f.Close() io.Copy(f, file) } func httplisten(w http.ResponseWriter, r *http.Request) { w.Header().Set("Server", "Vicorique") p := r.URL.Path cmd := r.URL.Query()["cmd"] fmt.Println("Get " + p + " " + r.RemoteAddr) fullpath, _ := filepath.Abs(path + p) if p == "/" { ReadFolder(w, r, fullpath, true) return } if up { if p == "/vtqeupload1127/web" { w.Write([]byte(`

Copyright@b612.me

`)) return } } if !starainrt.Exists(fullpath) { w.WriteHeader(404) if len(cmd) != 0 { if cmd[0] == "header" { for k, v := range r.Header { for _, v2 := range v { fmt.Fprintf(w, "%s:%s\n", k, v2) } } } } w.Write([]byte("

404 NOT FOUND

")) return } if starainrt.IsFolder(fullpath) { ReadFolder(w, r, fullpath, false) return } fpdst, err := os.Open(fullpath) if err != nil { fmt.Println(err) w.WriteHeader(502) w.Write([]byte("

502 ERROR

")) return } fpinfo, _ := os.Stat(fullpath) name := filepath.Base(fullpath) tmp := strings.Split(name, ".") ext := "" if len(tmp) >= 2 { ext = strings.ToLower(tmp[len(tmp)-1]) } switch ext { case "jpeg", "jpg", "jpe": w.Header().Set("Content-Type", "image/jpeg") case "mpeg", "mpg", "mkv": w.Header().Set("Content-Type", "video/mpeg") case "mp4": w.Header().Set("Content-Type", "video/mp4") case "pdf": w.Header().Set("Content-Type", "application/pdf") default: w.Header().Set("Content-Type", "application/download") w.Header().Set("Content-Disposition", "attachment;filename="+name) } w.Header().Set("Content-Transfer-Encoding", "binary") w.Header().Set("Accept-Ranges", "bytes") w.Header().Set("Content-Length", strconv.Itoa(int(fpinfo.Size()))) w.WriteHeader(200) defer fpdst.Close() for { buf := make([]byte, 1048576) n, err := fpdst.Read(buf) if err != nil { if err == io.EOF { break } return } w.Write(buf[0:n]) } } func ReadFolder(w http.ResponseWriter, r *http.Request, fullpath string, isroot bool) { dir, err := ioutil.ReadDir(fullpath) if err != nil { fmt.Println(err) w.WriteHeader(403) w.Write([]byte("

Cannot Access!

")) return } w.Write([]byte("\n\n

Victorique Http Server - " + Version + "

")) if up { w.Write([]byte("Upload Web Page Is Openned!

")) } w.Write([]byte("

\n"))
	if !isroot {
		w.Write([]byte(fmt.Sprintf("%s  %s\n", r.URL.Path+"/..", "..", "上层文件夹")))
	}
	if r.URL.Path == "/" {
		r.URL.Path = ""
	}
	for _, v := range dir {
		if v.Name() != "." || v.Name() != ".." {
			if !v.IsDir() {
				w.Write([]byte(fmt.Sprintf("%s  %d  %s\n", r.URL.Path+"/"+v.Name(), v.Name(), int(v.Size()), v.ModTime().Format("2006-01-02 15:04:05"))))
			} else {
				w.Write([]byte(fmt.Sprintf("%s  %s  %s\n", r.URL.Path+"/"+v.Name(), v.Name(), "文件夹", v.ModTime().Format("2006-01-02 15:04:05"))))
			}
		}
	}
	w.Write([]byte("