diff --git a/transfersh-server/handlers.go b/transfersh-server/handlers.go index 651ee96..a6c04ef 100644 --- a/transfersh-server/handlers.go +++ b/transfersh-server/handlers.go @@ -195,7 +195,7 @@ func scanHandler(w http.ResponseWriter, r *http.Request) { reader = r.Body - c := clamd.NewClamd("/tmp/clamd.socket") + c := clamd.NewClamd(config.CLAMAV_DAEMON_HOST) response, err := c.ScanStream(reader) if err != nil { diff --git a/transfersh-server/main.go b/transfersh-server/main.go index c41f5dd..1f4a997 100644 --- a/transfersh-server/main.go +++ b/transfersh-server/main.go @@ -44,21 +44,28 @@ const SERVER_INFO = "transfer.sh" const _24K = (1 << 20) * 24 var config struct { - AWS_ACCESS_KEY string - AWS_SECRET_KEY string - BUCKET string - VIRUSTOTAL_KEY string - Temp string + AWS_ACCESS_KEY string + AWS_SECRET_KEY string + BUCKET string + VIRUSTOTAL_KEY string + CLAMAV_DAEMON_HOST string "/tmp/clamd.socket" + Temp string } var storage Storage func init() { - config.AWS_ACCESS_KEY = os.Getenv("AWS_ACCESS_KEY") + config.AWS_ACCESS_KEY = os.Getenv("AWS_ACCESS_KEY_ID") config.AWS_SECRET_KEY = os.Getenv("AWS_SECRET_KEY") config.BUCKET = os.Getenv("BUCKET") + config.VIRUSTOTAL_KEY = os.Getenv("VIRUSTOTAL_KEY") - config.Temp = "" + + if os.Getenv("CLAMAV_DAEMON_HOST") != "" { + config.CLAMAV_DAEMON_HOST = os.Getenv("CLAMAV_DAEMON_HOST") + } + + config.Temp = os.TempDir() } func main() { @@ -112,7 +119,7 @@ func main() { r.NotFoundHandler = http.HandlerFunc(notFoundHandler) port := flag.String("port", "8080", "port number, default: 8080") - temp := flag.String("temp", "", "") + temp := flag.String("temp", config.Temp, "") basedir := flag.String("basedir", "", "") logpath := flag.String("log", "", "") provider := flag.String("provider", "s3", "")