Merge pull request #4 from ahpnils/ahpnils-patch-1

added usage of nice when ionice is not available

1 files changed, 21 insertions, 1 deletions

diff --git a/phpmalwarefinder b/phpmalwarefinder
index 0a85d85..1fbe4cf 100755
--- a/phpmalwarefinder
+++ b/phpmalwarefinder
@@ -2,6 +2,8 @@
 
 YARA=$(which yara)
 CONFIG_PATH='/etc/phpmalwarefinder/malwares.yara'
+IONICE_BIN=$(which ionice)
+NICE_BIN=$(which nice)
 
 if [ ! -f "$YARA" ]
 then
@@ -13,6 +15,18 @@ then
         CONFIG_PATH='./malwares.yara'
 fi
 
+if [ -f "${IONICE_BIN}" ]
+then
+        NICE=${IONICE_BIN}
+        NICE_OPTS="-c 3"
+else
+        if [ -f "${NICE_BIN}" ]
+        then
+                NICE=${NICE_BIN}
+                NICE_OPTS="-n 20"
+        fi
+fi
+
 show_help() {
     cat << EOF
 Usage ${0##*/} [-cfhw] <file|folder> ...
@@ -65,6 +79,12 @@ then
         exit 1
 fi
 
+if [ ! -e ${NICE} ]
+then
+        echo "no nice program available. Please install ionice or nice."
+        exit 1
+fi
+
 OPTS="${OPTS} -r ${CONFIG_PATH}"
 
-ionice -c3 $YARA $OPTS $@
+${NICE} ${NICE_OPTS} $YARA $OPTS $@