From 1e55e61786eaedd59c067f4306fb8427dc92ef52 Mon Sep 17 00:00:00 2001 From: clsr Date: Tue, 15 Nov 2016 15:17:45 +0100 Subject: Add logging support --- main.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'main.go') diff --git a/main.go b/main.go index 40dae35..d66416f 100644 --- a/main.go +++ b/main.go @@ -76,6 +76,14 @@ func main() { grill := flag.Bool("grill", false, "enable grills") idLength := flag.Int("id-length", DefaultIdLength, "length of uploaded file IDs") idCharset := flag.String("id-charset", "", "charset for uploaded file IDs (default lowercase letters a-z)") + enableLog := flag.Bool("log", false, "enable logging") + logIP := flag.Bool("log-ip", false, "log IP addresses") + logIPHash := flag.Bool("log-ip-hash", false, "log hashed IP addresses") + logUA := flag.Bool("log-ua", false, "log User-Agent headers") + logUAHash := flag.Bool("log-ua-hash", false, "log hashed User-Agent headers") + logReferer := flag.Bool("log-referer", false, "log Referer headers") + logRefererHash := flag.Bool("log-referer-hash", false, "log hashed Referer headers") + logHashSalt := flag.String("log-hash-salt", "", "salt to use for hashed log entries") flag.Parse() @@ -96,6 +104,18 @@ func main() { storage.IdCharset = *idCharset } + if !*enableLog { + DefaultLogger = nil + } else { + DefaultLogger.LogIP = *logIP || *logIPHash + DefaultLogger.LogUserAgent = *logUA || *logUAHash + DefaultLogger.LogReferer = *logReferer || *logRefererHash + DefaultLogger.HashIP = *logIPHash + DefaultLogger.HashUserAgent = *logUAHash + DefaultLogger.HashReferer = *logRefererHash + DefaultLogger.HashSalt = *logHashSalt + } + http.HandleFunc("/upload.php", handleUpload) http.Handle("/u/", http.StripPrefix("/u/", http.HandlerFunc(handleFile))) if *grill { -- cgit