97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/andreimarcu/linx-server/backends"
|
|
"github.com/andreimarcu/linx-server/expiry"
|
|
"github.com/zenazn/goji/web"
|
|
)
|
|
|
|
func fileServeHandler(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
fileName := c.URLParams["name"]
|
|
|
|
metadata, err := checkFile(fileName)
|
|
if err == backends.NotFoundErr {
|
|
notFoundHandler(c, w, r)
|
|
return
|
|
} else if err != nil {
|
|
oopsHandler(c, w, r, RespAUTO, "Corrupt metadata.")
|
|
return
|
|
}
|
|
|
|
if !Config.allowHotlink {
|
|
referer := r.Header.Get("Referer")
|
|
u, _ := url.Parse(referer)
|
|
p, _ := url.Parse(getSiteURL(r))
|
|
if referer != "" && !sameOrigin(u, p) {
|
|
http.Redirect(w, r, Config.sitePath+fileName, 303)
|
|
return
|
|
}
|
|
}
|
|
|
|
w.Header().Set("Content-Security-Policy", Config.fileContentSecurityPolicy)
|
|
w.Header().Set("Referrer-Policy", Config.fileReferrerPolicy)
|
|
|
|
_, reader, err := storageBackend.Get(fileName)
|
|
if err != nil {
|
|
oopsHandler(c, w, r, RespAUTO, err.Error())
|
|
}
|
|
|
|
w.Header().Set("Content-Type", metadata.Mimetype)
|
|
w.Header().Set("Content-Length", strconv.FormatInt(metadata.Size, 10))
|
|
w.Header().Set("Etag", metadata.Sha256sum)
|
|
w.Header().Set("Cache-Control", "max-age=0")
|
|
|
|
if r.Method != "HEAD" {
|
|
defer reader.Close()
|
|
|
|
if _, err = io.CopyN(w, reader, metadata.Size); err != nil {
|
|
oopsHandler(c, w, r, RespAUTO, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func staticHandler(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
path := r.URL.Path
|
|
if path[len(path)-1:] == "/" {
|
|
notFoundHandler(c, w, r)
|
|
return
|
|
} else {
|
|
if path == "/favicon.ico" {
|
|
path = Config.sitePath + "/static/images/favicon.gif"
|
|
}
|
|
|
|
filePath := strings.TrimPrefix(path, Config.sitePath+"static/")
|
|
file, err := staticBox.Open(filePath)
|
|
if err != nil {
|
|
notFoundHandler(c, w, r)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Etag", timeStartedStr)
|
|
w.Header().Set("Cache-Control", "max-age=86400")
|
|
http.ServeContent(w, r, filePath, timeStarted, file)
|
|
return
|
|
}
|
|
}
|
|
|
|
func checkFile(filename string) (metadata backends.Metadata, err error) {
|
|
metadata, err = storageBackend.Head(filename)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if expiry.IsTsExpired(metadata.Expiry) {
|
|
storageBackend.Delete(filename)
|
|
err = backends.NotFoundErr
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|