Improve listing of Checks and Groups

Change `-l` flag to print a unique list of every single check (assuming none are orphaned outside of all groups)
Allow `-g <group_id>` to be specified in combination with `-l`, to only print checks that are referenced by the specified group
When listing all checks with `-l` only, print out all groups that reference each check

Fixes: #545
This commit is contained in:
Marc Jay
2020-04-20 01:07:01 +01:00
parent 6747b208ce
commit 47a05c203a
3 changed files with 80 additions and 44 deletions

View File

@@ -108,9 +108,9 @@ This script has been written in bash using AWS-CLI and it works in Linux and OSX
./prowler ./prowler
``` ```
Use `-l` to list all available checks and group of checks (sections) Use `-l` to list all available checks and the groups (sections) that reference them
If you want to avoid installing dependences run it using Docker: If you want to avoid installing dependencies run it using Docker:
```sh ```sh
docker run -ti --rm --name prowler --env AWS_ACCESS_KEY_ID --env AWS_SECRET_ACCESS_KEY --env AWS_SESSION_TOKEN toniblyx/prowler:latest docker run -ti --rm --name prowler --env AWS_ACCESS_KEY_ID --env AWS_SECRET_ACCESS_KEY --env AWS_SESSION_TOKEN toniblyx/prowler:latest
@@ -235,7 +235,7 @@ This script has been written in bash using AWS-CLI and it works in Linux and OSX
-k keep the credential report -k keep the credential report
-n show check numbers to sort easier -n show check numbers to sort easier
(i.e.: 1.01 instead of 1.1) (i.e.: 1.01 instead of 1.1)
-l list all available checks only (does not perform any check) -l list all available checks only (does not perform any check). Add -g <group_id> to only list checks within the specified group
-L list all groups (does not perform any check) -L list all groups (does not perform any check)
-e exclude group extras -e exclude group extras
-E execute all tests except a list of specified checks separated by comma (i.e. check21,check31) -E execute all tests except a list of specified checks separated by comma (i.e. check21,check31)

View File

@@ -130,15 +130,20 @@ textTitle(){
*) ITEM_LEVEL="Unspecified or Invalid";; *) ITEM_LEVEL="Unspecified or Invalid";;
esac esac
local group_ids
if [[ -n "$5" ]]; then
group_ids="$CYAN [$5] $NORMAL"
fi
if [[ "${MODES[@]}" =~ "csv" ]]; then if [[ "${MODES[@]}" =~ "csv" ]]; then
>&2 echo "$TITLE_ID $TITLE_TEXT" | tee -a $OUTPUT_FILE_NAME.$EXTENSION_CSV >&2 echo "$TITLE_ID $TITLE_TEXT" | tee -a $OUTPUT_FILE_NAME.$EXTENSION_CSV
elif [[ "${MODES[@]}" =~ "json" || "${MODES[@]}" =~ "json-asff" ]]; then elif [[ "${MODES[@]}" =~ "json" || "${MODES[@]}" =~ "json-asff" ]]; then
: :
else else
if [[ "$ITEM_SCORED" == "Scored" ]]; then if [[ "$ITEM_SCORED" == "Scored" ]]; then
echo -e "\n$BLUE $TITLE_ID $NORMAL $TITLE_TEXT" echo -e "\n$BLUE $TITLE_ID $NORMAL $TITLE_TEXT $group_ids"
else else
echo -e "\n$PURPLE $TITLE_ID $TITLE_TEXT $NORMAL" echo -e "\n$PURPLE $TITLE_ID $TITLE_TEXT $NORMAL $group_ids"
fi fi
fi fi
} }

79
prowler
View File

@@ -48,6 +48,7 @@ SEND_TO_SECURITY_HUB=0
SCRIPT_START_TIME=$( date -u +"%Y-%m-%dT%H:%M:%S%z" ) SCRIPT_START_TIME=$( date -u +"%Y-%m-%dT%H:%M:%S%z" )
TITLE_ID="" TITLE_ID=""
TITLE_TEXT="CALLER ERROR - UNSET TITLE" TITLE_TEXT="CALLER ERROR - UNSET TITLE"
TOTAL_CHECKS=()
# Command usage menu # Command usage menu
usage(){ usage(){
@@ -70,7 +71,7 @@ USAGE:
-k keep the credential report -k keep the credential report
-n show check numbers to sort easier -n show check numbers to sort easier
(i.e.: 1.01 instead of 1.1) (i.e.: 1.01 instead of 1.1)
-l list all available checks only (does not perform any check) -l list all available checks only (does not perform any check). Add -g <group_id> to only list checks within the specified group
-L list all groups (does not perform any check) -L list all groups (does not perform any check)
-e exclude group extras -e exclude group extras
-E execute all tests except a list of specified checks separated by comma (i.e. check21,check31) -E execute all tests except a list of specified checks separated by comma (i.e. check21,check31)
@@ -232,14 +233,37 @@ if [[ $EXTERNAL_CHECKS_PATH ]]; then
done done
fi fi
# Function to show the title of the check # Get a list of total checks available by ID
for i in "${!GROUP_TITLE[@]}"; do
IFS=',' read -ra CHECKS <<< "${GROUP_CHECKS[$i]}"
for j in "${CHECKS[@]}"; do
TOTAL_CHECKS+=("$CHECK_ID_$j")
done
done
# Remove duplicates whilst preserving the order of checks, and store the result as an array
TOTAL_CHECKS=($(echo "${TOTAL_CHECKS[*]}" | tr ' ' '\n' | awk '!seen[$0]++'))
# Function to show the title of the check, and optionally which group(s) it belongs to
# using this way instead of arrays to keep bash3 (osx) and bash4(linux) compatibility # using this way instead of arrays to keep bash3 (osx) and bash4(linux) compatibility
show_check_title() { show_check_title() {
local check_id=CHECK_ID_$1 local check_id=CHECK_ID_$1
local check_title=CHECK_TITLE_$1 local check_title=CHECK_TITLE_$1
local check_scored=CHECK_SCORED_$1 local check_scored=CHECK_SCORED_$1
local check_type=CHECK_TYPE_$1 local check_type=CHECK_TYPE_$1
textTitle "${!check_id}" "${!check_title}" "${!check_scored}" "${!check_type}" local group_ids
# If requested ($2 is any non-null value) iterate all GROUP_CHECKS and produce a comma-separated list of all
# the GROUP_IDs that include this particular check
if [[ -n "$2" ]]; then
for i in "${!GROUP_ID[@]}"; do
if [[ "${GROUP_CHECKS[$i]}" =~ "$1" ]]; then
if [[ -n "$group_ids" ]]; then
group_ids+=", "
fi
group_ids+="${GROUP_ID[$i]}"
fi
done
fi
textTitle "${!check_id}" "${!check_title}" "${!check_scored}" "${!check_type}" "$group_ids"
} }
# Function to show the title of a group, by numeric id # Function to show the title of a group, by numeric id
@@ -343,23 +367,40 @@ execute_all() {
done done
} }
# Function to show the titles of everything # Function to show the titles of either all checks or only those in the specified group
show_all_titles() { show_all_titles() {
MAIN_GROUPS=(1 2 3 4 7) local checks
for i in "${MAIN_GROUPS[@]}"; do local check_id
show_group_title $i local group_index
# Display the title of the checks in groups 1,2,3,4 and 7 # If '-g <group_id>' has been specified, only show the titles of checks within the specified group
# Any other group has checks in these groups if [[ $GROUP_ID_READ ]];then
IFS=',' read -ra CHECKS <<< ${GROUP_CHECKS[$i]} if [[ " ${GROUP_ID[@]} " =~ " ${GROUP_ID_READ} " ]]; then
for j in ${CHECKS[@]}; do for group_index in "${!GROUP_ID[@]}"; do
show_check_title $j if [ "${GROUP_ID[$group_index]}" == "${GROUP_ID_READ}" ]; then
show_group_title "$group_index"
IFS=',' read -ra checks <<< "${GROUP_CHECKS[$i]}"
for check_id in ${checks[@]}; do
show_check_title "$check_id"
done done
fi
done done
else
textFail "Use a valid check group ID i.e.: group1, extras, forensics-ready, etc."
show_all_group_titles
exit $EXITCODE
fi
else
for check_id in "${TOTAL_CHECKS[@]}"; do
# Pass 1 so that the group IDs that this check belongs to are printed
show_check_title "$check_id" 1
done
fi
} }
show_all_group_titles() { show_all_group_titles() {
for i in "${!GROUP_TITLE[@]}"; do local group_index
show_group_title $i for group_index in "${!GROUP_TITLE[@]}"; do
show_group_title "$group_index"
done done
} }
@@ -373,16 +414,6 @@ get_all_checks_without_exclusion() {
for E_CHECK in "${E_CHECKS[@]}"; do for E_CHECK in "${E_CHECKS[@]}"; do
CHECKS_TO_EXCLUDE+=($E_CHECK) CHECKS_TO_EXCLUDE+=($E_CHECK)
done done
# Get a list of total checks available by ID
for i in "${!GROUP_TITLE[@]}"; do
# show_group_title $i
IFS=',' read -ra CHECKS <<< ${GROUP_CHECKS[$i]}
for j in ${CHECKS[@]}; do
TOTAL_CHECKS+=($CHECK_ID_$j)
done
done
# Remove duplicates whilst preserving the order of checks, and store the result as an array
TOTAL_CHECKS=($(echo "${TOTAL_CHECKS[*]}" | tr ' ' '\n' | awk '!seen[$0]++'))
# Create a list that contains all checks but excluded ones # Create a list that contains all checks but excluded ones
for i in "${TOTAL_CHECKS[@]}"; do for i in "${TOTAL_CHECKS[@]}"; do
local COINCIDENCE=false local COINCIDENCE=false