From d1d96e35472f692ace7b08822d185f14913e0ea9 Mon Sep 17 00:00:00 2001
From: clsr <clsr@clsr.net>
Date: Thu, 16 Jun 2016 02:19:44 +0200
Subject: Initial commit

---
 main.go | 86 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 main.go

(limited to 'main.go')

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..fb8810f
--- /dev/null
+++ b/main.go
@@ -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 {
+		}
+	}
+}
-- 
cgit