package main import ( "encoding/json" "errors" "flag" "fmt" "html" "io" "io/ioutil" "log" "net/http" "net/url" "os" "path/filepath" "strconv" "strings" ) func check(e error) { if e != nil { panic(e) } } var fs http.Handler var host = flag.String("h", "127.0.0.1", "host to listen to") var port = flag.String("p", "8001", "port to listen to") var verb = flag.Bool("verb", true, "verbosity") var skipHidden = flag.Bool("k", true, "skip hidden files") var initPath = "" var css = `css_will_be_here` // js will be embedded here var jsTag = `js_will_be_here` // id. var units = [8]string{"k", "M", "G", "T", "P", "E", "Z", "Y"} type rpcCall struct { Call string `json:"call"` Args []string `json:"args"` } func logVerb(s ...interface{}) { if *verb { log.Println(s) } } func sizeToString(bytes float64) string { if bytes == 0 { return "0" } var u = -1 for { bytes = bytes / 1024 u++ if bytes < 1024 { break } } return strconv.FormatFloat(bytes, 'f', 1, 64) + units[u] } func row(name string, href string, size float64, ext string) string { if strings.HasPrefix(href, "/") { href = strings.Replace(href, "/", "", 1) } return ` ` + sizeToString(size) + ` ` + name + ` ` } func extraFolder(loc string) string { if !strings.HasSuffix(loc, "/") { loc = loc + "/" } if !strings.HasPrefix(loc, "/") { loc = "/" + loc } return `` + loc + `` } func replyList(w http.ResponseWriter, path string) { if !strings.HasSuffix(path, "/") { path += "/" } var head = ` ` + html.EscapeString(path) + `

` + extraFolder("/hols/aaa") + `

Drop here to upload

.` + html.EscapeString(path) + `

` _files, err := ioutil.ReadDir(initPath + path) check(err) if path != "/" { head += row("../", "../", 0, "folder") } var dirs = "" var files = "" for _, el := range _files { var name = el.Name() if *skipHidden && strings.HasPrefix(name, ".") { continue } if el.IsDir() { dirs += row(name+"/", name, 0, "folder") } else { var sl = strings.Split(name, ".") var ext = sl[len(sl)-1] files += row(name, name, float64(el.Size()), ext) } } var resp = head + dirs + files + `

Gossa 🎶

` w.Write([]byte(resp)) } func doContent(w http.ResponseWriter, r *http.Request) { path := html.UnescapeString(r.URL.Path) if strings.Contains(path, "/favicon.ico") { w.Write([]byte(" ")) return } fullPath, errPath := checkPath(path) stat, errStat := os.Stat(fullPath) if errStat != nil || errPath != nil { logVerb("Error", errStat, errPath) w.Write([]byte("error")) return } if stat.IsDir() { logVerb("Get list", fullPath) replyList(w, path) } else { logVerb("Get file", fullPath) fs.ServeHTTP(w, r) } } func upload(w http.ResponseWriter, r *http.Request) { unescaped, _ := url.PathUnescape(r.Header.Get("gossa-path")) fullPath, err := checkPath(unescaped) logVerb("Up", err, fullPath) if err != nil { w.Write([]byte("error")) return } reader, _ := r.MultipartReader() part, _ := reader.NextPart() dst, _ := os.Create(fullPath) io.Copy(dst, part) logVerb("Done upping", fullPath) w.Write([]byte("ok")) } func rpc(w http.ResponseWriter, r *http.Request) { var err error bodyBytes, _ := ioutil.ReadAll(r.Body) bodyString := string(bodyBytes) var payload rpcCall json.Unmarshal([]byte(bodyString), &payload) for i := range payload.Args { payload.Args[i], err = checkPath(payload.Args[i]) if err != nil { logVerb("Cant read path", err, payload) w.Write([]byte("error")) return } } if payload.Call == "mkdirp" { err = os.MkdirAll(payload.Args[0], os.ModePerm) } else if payload.Call == "mv" { err = os.Rename(payload.Args[0], payload.Args[1]) } logVerb("RPC", err, payload) w.Write([]byte("ok")) } func checkPath(p string) (string, error) { p = filepath.Join(initPath, p) fp, err := filepath.Abs(p) if err != nil || !strings.HasPrefix(fp, initPath) { return "", errors.New("error") } return fp, nil } func main() { flag.Parse() if len(flag.Args()) == 0 { initPath = "." } else { initPath = flag.Args()[0] } var err error initPath, err = filepath.Abs(initPath) check(err) var hostString = *host + ":" + *port fmt.Println("Gossa startig on directory " + initPath) fmt.Println("Listening on http://" + hostString) var root = http.Dir(initPath) fs = http.StripPrefix("/", http.FileServer(root)) http.HandleFunc("/rpc", rpc) http.HandleFunc("/post", upload) http.HandleFunc("/", doContent) err = http.ListenAndServe(hostString, nil) check(err) }