stig tests option to lockdown permissions

src/freedombone-tests

 
 export TEXTDOMAIN=${PROJECT_NAME}-tests
 export TEXTDOMAINDIR="/usr/share/locale"
+source /usr/share/${PROJECT_NAME}/utils/${PROJECT_NAME}-utils-setup
 
 # Whether to run STIG tests
 RUN_STIG=
     echo ''
     echo $'Runs tests on the system'
     echo ''
-    echo $'     --stig [yes|no]      Run STIG tests'
+    echo $'     --stig [yes|no|fix]  Run STIG tests'
     echo $'     --help               Show help'
     echo ''
     exit 0
     wait $1
 }
 
+function fix_stig {
+    if [[ $RUN_STIG != 'fix' ]]; then
+        return
+    fi
+    lockdown_permissions
+}
+
 function test_stig {
     if [ ! $RUN_STIG ]; then
         return
             if [[ "$1" == 'showall' ]]; then
                 SHOW_ALL_TESTS=1
             fi
-            RUN_STIG=1
+            RUN_STIG="$1"
             ;;
         *)
             # unknown option
 
 test_app_functions
 test_unique_onion_ports
+fix_stig
 test_stig
 
 if [ ! $RUN_STIG ]; then