Use full paths when local directory is used for includedir or dbdir

1 files changed, 18 insertions, 9 deletions

diff --git a/lynis b/lynis
index da12f6d8..8fc594b9 100755
--- a/lynis
+++ b/lynis
@@ -56,11 +56,20 @@
 #
 #################################################################################
 #
+    # Work directory
+    WORKDIR=`pwd`
+
     # Test from which directories we can use all functions and tests
 
         INCLUDEDIR=""  # Set default include directory to none
         tINCLUDE_TARGETS="/usr/local/include/lynis /usr/local/lynis/include /usr/share/lynis/include ./include"  # Default paths to check (CWD as last option, in case we run from standalone)
-        for I in ${tINCLUDE_TARGETS}; do if [ -d ${I} ]; then INCLUDEDIR=${I}; fi; done
+        for I in ${tINCLUDE_TARGETS}; do
+            if [ "${I}" = "./include" ]; then
+                if [ -d ${WORKDIR}/include ]; then INCLUDEDIR="${WORKDIR}/include"; fi
+            elif [ -d ${I} ]; then
+                INCLUDEDIR=${I}
+            fi
+        done
 
     # Drop out if our include directory can't be found
         if [ "${INCLUDEDIR}" = "" ]; then
@@ -73,7 +82,13 @@ Make sure to execute ${PROGRAM_NAME} from untarred directory or check your insta
     # Test for database directory
 
         DBDIR=""; tDB_TARGETS="/usr/local/share/lynis/db /usr/local/lynis/db /usr/share/lynis/db ./db"
-        for I in ${tDB_TARGETS}; do if [ -d ${I} ]; then DBDIR=${I}; fi; done
+        for I in ${tDB_TARGETS}; do
+            if [ "${I}" = "./db" ]; then
+                if [ -d ${WORKDIR}/db ]; then DBDIR="${WORKDIR}/db"; fi
+            elif [ -d ${I} ]; then
+                DBDIR="${I}"
+            fi
+        done
 #
 #################################################################################
 #
@@ -88,13 +103,7 @@ Make sure to execute ${PROGRAM_NAME} from untarred directory or check your insta
 #
 #################################################################################
 #
-    WORKDIR=`pwd`
-#
-#################################################################################
-#
-
-# Consts
-# (bin paths, text strings, colors)
+# Consts (bin paths, text strings, colors)
 #
 #################################################################################
 #