diff options
| author | clsr <clsr@clsr.net> | 2016-06-16 02:19:44 +0200 |
|---|---|---|
| committer | clsr <clsr@clsr.net> | 2016-06-16 02:19:44 +0200 |
| commit | d1d96e35472f692ace7b08822d185f14913e0ea9 (patch) | |
| tree | 83ca1bca75a0634436328a924afe7f3a9f409e45 /main.go | |
| download | gomf-d1d96e35472f692ace7b08822d185f14913e0ea9.tar.gz gomf-d1d96e35472f692ace7b08822d185f14913e0ea9.zip | |
Initial commit
Diffstat (limited to 'main.go')
| -rw-r--r-- | main.go | 86 |
1 files changed, 86 insertions, 0 deletions
@@ -0,0 +1,86 @@ +package main + +import ( + "flag" + "fmt" + "math/rand" + "net/http" + "strings" + "time" +) + +var storage *Storage + +var ( + uploadUrl string + uploadHost string + siteName string + contactMail string + abuseMail string + hsts bool +) + +func handle(w http.ResponseWriter, r *http.Request) { + if hsts { + w.Header().Set("Strict-Transport-Security", "max-age=15552000") + } + if uploadHost != "" && r.URL.Host == uploadHost { + handleFile(w, r) + } else { + http.DefaultServeMux.ServeHTTP(w, r) + } +} + +func main() { + flag.StringVar(&uploadUrl, "upload-url", "", "URL to serve uploads from") + flag.StringVar(&uploadHost, "upload-host", "", "host to serve uploads on") + flag.StringVar(&siteName, "name", "Gomf", "website name") + flag.StringVar(&contactMail, "contact", "contact@example.com", "contact email address") + flag.StringVar(&abuseMail, "abuse", "abuse@example.com", "abuse email address") + flag.BoolVar(&hsts, "hsts", false, "enable HSTS") + listenHttp := flag.String("http", "localhost:8080", "address to listen on for HTTP") + listenHttps := flag.String("https", "", "address to listen on for HTTPS") + cert := flag.String("cert", "", "path to TLS certificate (for HTTPS)") + key := flag.String("key", "", "path to TLS key (for HTTPS)") + maxSize := flag.Int64("max-size", 50*1024*1024, "max filesize in bytes") + forbidMime := flag.String("forbid-mime", "application/x-dosexec,application/x-msdos-program", "comma-separated list of forbidden MIME types") + forbidExt := flag.String("forbid-ext", "exe,dll,msi,scr,com,pif", "comma-separated list of forbidden file extensions") + flag.Parse() + + rand.Seed(time.Now().UnixNano()) + + storage = NewStorage("upload", *maxSize) + storage.ForbiddenExt = strings.Split(*forbidExt, ",") + storage.ForbiddenMime = strings.Split(*forbidMime, ",") + + http.HandleFunc("/upload.php", handleUpload) + http.HandleFunc("/grill.php", handleGrill) + http.Handle("/u/", http.StripPrefix("/u/", http.HandlerFunc(handleFile))) + + initWebsite() + + if uploadUrl == "" { + if *listenHttps != "" { + uploadUrl = "https://" + *listenHttps + "/u/" + } else if *listenHttp != "" { + uploadUrl = "http://" + *listenHttp + "/u/" + } + } + + exit := true + if *listenHttp != "" { + exit = false + fmt.Printf("listening on http://%s/\n", *listenHttp) + go panic(http.ListenAndServe(*listenHttp, http.HandlerFunc(handle))) + } + if *listenHttps != "" { + exit = false + fmt.Printf("listening on https://%s/\n", *listenHttps) + go panic(http.ListenAndServeTLS(*listenHttps, *cert, *key, http.HandlerFunc(handle))) + } + + if !exit { + switch { + } + } +} |