mirror of
https://github.com/versity/versitygw.git
synced 2026-01-08 04:35:15 +00:00
97 lines
2.4 KiB
Bash
97 lines
2.4 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright 2024 Versity Software
|
|
# This file is 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.
|
|
|
|
# levels: 1 - crit, 2 - err, 3 - warn, 4 - info, 5 - debug, 6 - trace
|
|
|
|
export LOG_LEVEL_INT=4
|
|
|
|
log() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "log function requires level, message"
|
|
return 1
|
|
fi
|
|
# shellcheck disable=SC2153
|
|
if [[ $1 -gt $LOG_LEVEL_INT ]]; then
|
|
return 0
|
|
fi
|
|
log_level=""
|
|
case "$1" in
|
|
1) log_level="CRIT";;
|
|
2) log_level="ERROR";;
|
|
3) log_level="WARN";;
|
|
4) log_level="INFO";;
|
|
5) log_level="DEBUG";;
|
|
6) log_level="TRACE";;
|
|
*) echo "invalid log level $1"; return 1
|
|
esac
|
|
if [[ "$2" == *"secret"* ]]; then
|
|
log_mask "$log_level" "$2"
|
|
return 0
|
|
fi
|
|
log_message "$log_level" "$2"
|
|
}
|
|
|
|
log_mask() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "mask and log requires level, string"
|
|
return 1
|
|
fi
|
|
masked_args=() # Initialize an array to hold the masked arguments
|
|
|
|
IFS=' ' read -r -a array <<< "$2"
|
|
|
|
mask_next=false
|
|
for arg in "${array[@]}"; do
|
|
if ! check_arg_for_mask "$arg"; then
|
|
echo "error checking arg for mask"
|
|
return 1
|
|
fi
|
|
done
|
|
log_message "$log_level" "${masked_args[*]}"
|
|
}
|
|
|
|
check_arg_for_mask() {
|
|
if [ $# -ne 1 ]; then
|
|
echo "'check_arg_for_mask' requires arg"
|
|
return 1
|
|
fi
|
|
if [[ $mask_next == true ]]; then
|
|
masked_args+=("********")
|
|
mask_next=false
|
|
elif [[ "$arg" == --secret_key=* ]]; then
|
|
masked_args+=("--secret_key=********")
|
|
elif [[ "$arg" == --secret=* ]]; then
|
|
masked_args+=("--secret=********")
|
|
else
|
|
if [[ "$arg" == "--secret_key" ]] || [[ "$arg" == "--secret" ]] || [[ "$arg" == "--s3-iam-secret" ]]; then
|
|
mask_next=true
|
|
fi
|
|
masked_args+=("$arg")
|
|
fi
|
|
}
|
|
|
|
log_message() {
|
|
if [[ $# -ne 2 ]]; then
|
|
echo "log message requires level, message"
|
|
return 1
|
|
fi
|
|
now="$(date "+%Y-%m-%d %H:%M:%S")"
|
|
echo "$now $1 $2"
|
|
if [[ -n "$TEST_LOG_FILE" ]]; then
|
|
echo "$now $1 $2" >> "$TEST_LOG_FILE"
|
|
fi
|
|
}
|