From 6f4799bb888dacd3f4a7014609b073b47add1c7f Mon Sep 17 00:00:00 2001
From: Flavio Meyer <flavio.meyer@ajila.com>
Date: Tue, 21 Jan 2020 11:18:58 +0100
Subject: [PATCH] Implementing environment variable option

---
 Dockerfile    | 14 ++++++++++++++
 entrypoint.sh | 17 +++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 1bc08bf..5ba7c5f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,4 +28,18 @@ RUN mv /go/src/clamav-rest/clamav-rest /usr/bin/ && rm -Rf /go/src/clamav-rest
 
 EXPOSE 9000
 
+ENV MAX_SCAN_SIZE=100M
+ENV MAX_FILE_SIZE=25M
+ENV MAX_RECURSION=16
+ENV MAX_FILES=10000
+ENV MAX_EMBEDDEDPE=10M
+ENV MAX_HTMLNORMALIZE=10M
+ENV MAX_HTMLNOTAGS=2M
+ENV MAX_SCRIPTNORMALIZE=5M
+ENV MAX_ZIPTYPERCG=1M
+ENV MAX_PARTITIONS=50
+ENV MAX_ICONSPE=100
+ENV PCRE_MATCHLIMIT=100000
+ENV PCRE_RECMATCHLIMIT=2000
+
 ENTRYPOINT [ "entrypoint.sh" ]
\ No newline at end of file
diff --git a/entrypoint.sh b/entrypoint.sh
index 80b1f9c..69e007e 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -1,4 +1,21 @@
 #!/bin/sh
+
+# Replace values with environment variables in clamd.conf
+sed -i 's/^#MaxScanSize .*$/MaxScanSize '"$MAX_SCAN_SIZE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#StreamMaxLength .*$/StreamMaxLength '"$MAX_FILE_SIZE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxFileSize .*$/MaxFileSize '"$MAX_FILE_SIZE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxRecursion .*$/MaxRecursion '"$MAX_RECURSION"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxFiles .*$/MaxFiles '"$MAX_FILES"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxEmbeddedPE .*$/MaxEmbeddedPE '"$MAX_EMBEDDEDPE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxHTMLNormalize .*$/MaxHTMLNormalize '"$MAX_HTMLNORMALIZE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxHTMLNoTags.*$/MaxHTMLNoTags '"$MAX_HTMLNOTAGS"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxScriptNormalize .*$/MaxScriptNormalize '"$MAX_SCRIPTNORMALIZE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxZipTypeRcg .*$/MaxZipTypeRcg '"$MAX_ZIPTYPERCG"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxPartitions .*$/MaxPartitions '"$MAX_PARTITIONS"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#MaxIconsPE .*$/MaxIconsPE '"$MAX_ICONSPE"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#PCREMatchLimit.*$/PCREMatchLimit '"$PCRE_MATCHLIMIT"'/g' /etc/clamav/clamd.conf
+sed -i 's/^#PCRERecMatchLimit .*$/PCRERecMatchLimit '"$PCRE_RECMATCHLIMIT"'/g' /etc/clamav/clamd.conf
+
 freshclam -d &
 clamd &
 /usr/bin/clamav-rest &