From 81d6bc357a7b0d49ba0c5034f77404dc577cf8aa Mon Sep 17 00:00:00 2001 From: clsr Date: Thu, 16 Jun 2016 19:23:18 +0200 Subject: Add --id-charset and --id-length options --- main.go | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'main.go') diff --git a/main.go b/main.go index fc51fe2..ad8711c 100644 --- a/main.go +++ b/main.go @@ -46,6 +46,9 @@ func main() { 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") grill := flag.Bool("grill", false, "enable grills") + idLength := flag.Int("id-length", 6, "length of uploaded file IDs") + idCharset := flag.String("id-charset", "", "charset for uploaded file IDs (default lowercase letters a-z)") + flag.Parse() rand.Seed(time.Now().UnixNano()) @@ -53,6 +56,10 @@ func main() { storage = NewStorage("upload", *maxSize) storage.ForbiddenExt = strings.Split(*forbidExt, ",") storage.ForbiddenMime = strings.Split(*forbidMime, ",") + storage.IdLength = *idLength + if *idCharset != "" { + storage.IdCharset = *idCharset + } http.HandleFunc("/upload.php", handleUpload) http.Handle("/u/", http.StripPrefix("/u/", http.HandlerFunc(handleFile))) -- cgit