mirror of
https://github.com/ghndrx/prowler.git
synced 2026-02-10 23:05:05 +00:00
74 lines
2.5 KiB
Bash
74 lines
2.5 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Prowler - the handy cloud security tool (copyright 2021) by Toni de la Fuente
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
# use this file except in compliance with the License. You may obtain a copy
|
|
# of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software distributed
|
|
# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
# CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations under the License.
|
|
|
|
if [[ $OUTPUT_BUCKET ]]; then
|
|
# output mode has to be set to other than text
|
|
if [[ "${MODES[@]}" =~ "html" ]] || [[ "${MODES[@]}" =~ "csv" ]] || [[ "${MODES[@]}" =~ "json" ]] || [[ "${MODES[@]}" =~ "json-asff" ]]; then
|
|
OUTPUT_BUCKET_WITHOUT_FOLDERS=$(echo $OUTPUT_BUCKET | awk -F'/' '{ print $1 }')
|
|
# OUTPUT_BUCKET_STATUS=$($AWSCLI s3api head-bucket --bucket "$OUTPUT_BUCKET" 2>&1 || true)
|
|
# if [[ -z $OUTPUT_BUCKET_STATUS ]]; then
|
|
# echo "$OPTRED ERROR!$OPTNORMAL wrong bucket name or not right permissions."
|
|
# exit 1
|
|
# else
|
|
# need to make sure last / is not set to avoid // in S3
|
|
if [[ $OUTPUT_BUCKET != *"/" ]]; then
|
|
OUTPUT_BUCKET="$OUTPUT_BUCKET"
|
|
else
|
|
OUTPUT_BUCKET=${OUTPUT_BUCKET::-1}
|
|
fi
|
|
# fi
|
|
else
|
|
echo "$OPTRED ERROR!$OPTNORMAL - Mode (-M) has to be set as well. Use -h for help."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
copyToS3() {
|
|
# Prowler will copy each format to its own folder in S3, that is for better handling
|
|
# and processing by Quicksight or others.
|
|
# Also, check if -F was introduced
|
|
if [ -n "${OUTPUT_FILE_NAME+x}" ]; then
|
|
OUTPUT_PATH="${OUTPUT_FILE_NAME}"
|
|
else
|
|
OUTPUT_PATH="$OUTPUT_DIR/prowler-output-${ACCOUNT_NUM}-${OUTPUT_DATE}"
|
|
fi
|
|
|
|
for output_format in "${MODES[@]}";
|
|
do
|
|
case ${output_format} in
|
|
csv)
|
|
s3cp "${OUTPUT_PATH}" "${EXTENSION_CSV}"
|
|
;;
|
|
html)
|
|
s3cp "${OUTPUT_PATH}" "${EXTENSION_HTML}"
|
|
;;
|
|
json)
|
|
s3cp "${OUTPUT_PATH}" "${EXTENSION_JSON}"
|
|
;;
|
|
json-asff)
|
|
s3cp "${OUTPUT_PATH}" "${EXTENSION_ASFF}"
|
|
;;
|
|
*)
|
|
echo "$OPTRED ERROR!$OPTNORMAL - Invalid output format copying to S3. Use -h for help."
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
s3cp(){
|
|
OUTPUT_PATH="${1}"
|
|
EXTENSION="${2}"
|
|
"${AWSCLI}" ${PROFILE_OPT} s3 cp "${OUTPUT_PATH}.${EXTENSION}" s3://"${OUTPUT_BUCKET}"/${EXTENSION}/ --acl bucket-owner-full-control
|
|
}
|