2015-09-24 19:58:50 -04:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"net/http"
|
2015-10-13 22:52:55 -04:00
|
|
|
"net/url"
|
2015-09-24 19:58:50 -04:00
|
|
|
"os"
|
|
|
|
"path"
|
2015-09-29 19:28:10 -04:00
|
|
|
"strings"
|
2015-09-24 19:58:50 -04:00
|
|
|
|
|
|
|
"github.com/zenazn/goji/web"
|
|
|
|
)
|
|
|
|
|
|
|
|
func fileServeHandler(c web.C, w http.ResponseWriter, r *http.Request) {
|
2015-09-24 22:20:44 -04:00
|
|
|
fileName := c.URLParams["name"]
|
|
|
|
filePath := path.Join(Config.filesDir, fileName)
|
2015-09-24 19:58:50 -04:00
|
|
|
|
2015-10-07 22:45:34 -04:00
|
|
|
err := checkFile(fileName)
|
|
|
|
if err == NotFoundErr {
|
2015-09-25 12:00:14 -04:00
|
|
|
notFoundHandler(c, w, r)
|
2015-09-24 19:58:50 -04:00
|
|
|
return
|
2015-10-07 22:45:34 -04:00
|
|
|
} else if err == BadMetadata {
|
|
|
|
oopsHandler(c, w, r, RespAUTO, "Corrupt metadata.")
|
|
|
|
return
|
2015-09-24 19:58:50 -04:00
|
|
|
}
|
|
|
|
|
2015-09-29 19:28:10 -04:00
|
|
|
if !Config.allowHotlink {
|
|
|
|
referer := r.Header.Get("Referer")
|
2015-10-13 22:52:55 -04:00
|
|
|
u, _ := url.Parse(referer)
|
|
|
|
p, _ := url.Parse(Config.siteURL)
|
|
|
|
if referer != "" && !sameOrigin(u, p) {
|
2015-09-29 19:28:10 -04:00
|
|
|
w.WriteHeader(403)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-04 17:58:00 -04:00
|
|
|
w.Header().Set("Content-Security-Policy", Config.fileContentSecurityPolicy)
|
|
|
|
|
2015-09-28 00:25:57 -04:00
|
|
|
http.ServeFile(w, r, filePath)
|
|
|
|
}
|
2015-09-27 22:17:12 -04:00
|
|
|
|
2015-09-30 15:54:30 -04:00
|
|
|
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 {
|
2015-10-04 12:58:30 -04:00
|
|
|
if path == "/favicon.ico" {
|
|
|
|
path = "/static/images/favicon.gif"
|
|
|
|
}
|
|
|
|
|
2015-09-30 15:54:30 -04:00
|
|
|
filePath := strings.TrimPrefix(path, "/static/")
|
2015-10-14 14:58:27 -04:00
|
|
|
file, err := staticBox.Open(filePath)
|
|
|
|
if err != nil {
|
|
|
|
notFoundHandler(c, w, r)
|
|
|
|
return
|
2015-09-30 15:54:30 -04:00
|
|
|
}
|
|
|
|
|
2015-10-01 10:32:59 -04:00
|
|
|
w.Header().Set("Etag", timeStartedStr)
|
|
|
|
w.Header().Set("Cache-Control", "max-age=86400")
|
2015-10-14 14:58:27 -04:00
|
|
|
http.ServeContent(w, r, filePath, timeStarted, file)
|
2015-09-30 15:54:30 -04:00
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-07 22:45:34 -04:00
|
|
|
func checkFile(filename string) error {
|
2015-09-28 00:25:57 -04:00
|
|
|
filePath := path.Join(Config.filesDir, filename)
|
|
|
|
|
|
|
|
_, err := os.Stat(filePath)
|
|
|
|
if err != nil {
|
2015-10-07 22:45:34 -04:00
|
|
|
return NotFoundErr
|
|
|
|
}
|
|
|
|
|
|
|
|
expired, err := isFileExpired(filename)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2015-09-27 22:17:12 -04:00
|
|
|
}
|
2015-09-24 19:58:50 -04:00
|
|
|
|
2015-10-07 22:45:34 -04:00
|
|
|
if expired {
|
2015-09-28 00:25:57 -04:00
|
|
|
os.Remove(path.Join(Config.filesDir, filename))
|
|
|
|
os.Remove(path.Join(Config.metaDir, filename))
|
2015-10-07 22:45:34 -04:00
|
|
|
return NotFoundErr
|
2015-09-28 00:25:57 -04:00
|
|
|
}
|
|
|
|
|
2015-10-07 22:45:34 -04:00
|
|
|
return nil
|
2015-09-24 19:58:50 -04:00
|
|
|
}
|