nxshock/simplefileshare
1
0
Fork 0
mirror of https://github.com/nxshock/simplefileshare.git synced 2025-07-02 00:13:36 +05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow serving unix sockets

Browse source
This commit is contained in:
NXShock 2025-06-03 12:17:51 +05:00
parent c41ff56ffa
commit 4847a2d05a
5 changed files with 35 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

13
handlers.go
View file

@ -109,6 +109,13 @@ func HandleUpload(w http.ResponseWriter, r *http.Request) {
}
func HandleDownload(w http.ResponseWriter, r *http.Request) {
root, err := os.OpenRoot(config.StoragePath)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer root.Close()
filename := filepath.Base(r.FormValue("filename"))
if filename == "" {
@ -116,7 +123,7 @@ func HandleDownload(w http.ResponseWriter, r *http.Request) {
return
}
f, err := os.Open(filepath.Join(config.StoragePath, filename))
f, err := root.Open(filename)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
@ -140,8 +147,8 @@ func HandleDownload(w http.ResponseWriter, r *http.Request) {
func HandleStream(w http.ResponseWriter, r *http.Request) {
filename := filepath.Base(r.FormValue("filename"))
if filename == "" {
http.Error(w, `"filename" field can't be empty`, http.StatusBadRequest)
if !filepath.IsLocal(filename) {
http.Error(w, `wrong "filename" specified`, http.StatusBadRequest)
return
}