263 lines
6.6 KiB
Bash
Executable file
263 lines
6.6 KiB
Bash
Executable file
#!/bin/sh
|
|
|
|
# Little check for quota usage on Btrfs subvolumes
|
|
# GPL v3+ (copyright chl-dev@bugness.org)
|
|
|
|
PROGPATH=$( echo $0 | sed -e 's,[\\/][^\\/][^\\/]*$,,' )
|
|
REVISION="0.1"
|
|
|
|
# Stop at the first non-catched error
|
|
set -e
|
|
|
|
# Include check_range()
|
|
# Not needed at the moment
|
|
#. $PROGPATH/utils.sh
|
|
STATE_OK=0
|
|
STATE_WARNING=1
|
|
STATE_CRITICAL=2
|
|
STATE_UNKNOWN=3
|
|
STATE_DEPENDENT=4
|
|
|
|
# Default values
|
|
RANGE_WARNING="20%"
|
|
RANGE_CRITICAL="10%"
|
|
|
|
# Output
|
|
OUTPUT_EXIT_STATUS=$STATE_OK
|
|
OUTPUT_DETAIL_OK=""
|
|
OUTPUT_DETAIL_WARNING=""
|
|
OUTPUT_DETAIL_CRITICAL=""
|
|
OUTPUT_PERFDATA=""
|
|
|
|
#
|
|
# Help function
|
|
#
|
|
usage() {
|
|
cat <<EOF
|
|
Usage :
|
|
$0 [-w warning-free-space%] [-c critical-free-space%] -V volume -V...
|
|
|
|
Example :
|
|
./check_btrfs_quota -w 20% -c 10% -V /home
|
|
|
|
We adopt the same behaviour than check_disks so thresholds are matched
|
|
against "free" space. The example above will trigger a warning if less
|
|
than 20% of maxrfer or maxecl is available.
|
|
|
|
Note: Since the volumes are checked against the latest thresholds specified, order
|
|
of the arguments is important. Ex:
|
|
./check_btrfs_quota -w 20% -V /home -w 50% -V /tmp
|
|
|
|
Default values:
|
|
warning-free-space: $RANGE_WARNING
|
|
critical-free-space: $RANGE_CRITICAL
|
|
EOF
|
|
}
|
|
|
|
check_range_syntax() {
|
|
return 0
|
|
}
|
|
# check_range 0 "$1" >/dev/null 2>&1
|
|
# if [ "$?" -eq "2" ]; then
|
|
# return 1
|
|
# fi
|
|
# return 0
|
|
#}
|
|
|
|
#
|
|
# Remove the '%' suffix (or error if no '%')
|
|
#
|
|
get_percent() {
|
|
if [ "$( echo -n "$1" | tail -c 1 )" = "%" ]; then
|
|
echo -n "$1" | head -c -1
|
|
return 0
|
|
fi
|
|
return 1
|
|
}
|
|
|
|
#
|
|
# Uniformize thresholds:
|
|
# if "20%" then calculate
|
|
# else leave as is
|
|
# args:
|
|
# 1: MAXVALUE
|
|
# 2: THRESHOLD
|
|
#
|
|
get_absolute_threshold() {
|
|
local MAXVALUE="$1"
|
|
local THRESHOLD="$2"
|
|
|
|
if get_percent "$THRESHOLD" >/dev/null; then
|
|
echo "$(( $MAXVALUE * ( 100 - $( get_percent "$THRESHOLD" ) ) / 100 ))"
|
|
else
|
|
echo "$(( $MAXVALUE - $THRESHOLD ))"
|
|
fi
|
|
}
|
|
|
|
#
|
|
# Returns 1 if value is above threshold, 0 if not
|
|
#
|
|
# args:
|
|
# - value
|
|
# - quota (aka. maximum space possible)
|
|
# - threshold (in percent with a '%' suffix, or plain unit)
|
|
#
|
|
check_above_threshold() {
|
|
local VALUE="$1"
|
|
local MAXVALUE="$2"
|
|
local THRESHOLD="$3"
|
|
|
|
if [ "$VALUE" -gt "$( get_absolute_threshold "$MAXVALUE" "$THRESHOLD" )" ]; then
|
|
return 1
|
|
fi
|
|
return 0
|
|
}
|
|
|
|
#
|
|
# Factorize output in the form of:
|
|
# /home:0/260 (refr:10M(50%) excl:5M(20%)
|
|
# args:
|
|
# 1: VOLUME
|
|
# 2: QGROUPID
|
|
# 3: REFR
|
|
# 4: EXCL
|
|
# 5: MAXRFER
|
|
# 6: MAXEXCL
|
|
# 7: SUBVOLUMEPATH
|
|
format_output_item() {
|
|
local LOCAL_OUTPUT=""
|
|
if [ "$5" != "none" ]; then
|
|
LOCAL_OUTPUT="$( printf "rfer:%s(%d%%)" \
|
|
"$( numfmt --to=iec $(( $5 - $3 )) )" \
|
|
"$(( 100 * ($5 - $3) / $5 ))" \
|
|
)"
|
|
fi
|
|
if [ "$6" != "none" ]; then
|
|
[ -n "$LOCAL_OUTPUT" ] && LOCAL_OUTPUT="$LOCAL_OUTPUT "
|
|
LOCAL_OUTPUT="$( printf "excl:%s(%d%%)" \
|
|
"$( numfmt --to=iec $(( $6 - $4 )) )" \
|
|
"$(( 100 * ($6 - $4) / $6 ))" \
|
|
)"
|
|
fi
|
|
if [ -n "$LOCAL_OUTPUT" ]; then
|
|
printf "%s:%s (%s)" "$1" "$2$( test -n "$7" && echo ":$7" )" "$LOCAL_OUTPUT"
|
|
fi
|
|
}
|
|
|
|
# Some early checks
|
|
if ! which btrfs 2>&1 >/dev/null; then
|
|
echo "UNKNOWN 'btrfs' command not found."
|
|
exit $STATE_UNKNOWN
|
|
fi
|
|
|
|
#
|
|
# Loop on parameters + tests
|
|
#
|
|
while getopts hw:c:V: f; do
|
|
case "$f" in
|
|
'h')
|
|
usage
|
|
exit
|
|
;;
|
|
|
|
'w')
|
|
if check_range_syntax "$OPTARG" >/dev/null; then
|
|
RANGE_WARNING="$OPTARG"
|
|
else
|
|
echo "UNKNOWN: invalid range."
|
|
exit 3
|
|
fi
|
|
;;
|
|
|
|
'c')
|
|
if check_range_syntax "$OPTARG" >/dev/null; then
|
|
RANGE_CRITICAL="$OPTARG"
|
|
else
|
|
echo "UNKNOWN: invalid range."
|
|
exit 3
|
|
fi
|
|
;;
|
|
|
|
'V')
|
|
VOLUME="$OPTARG"
|
|
|
|
# Quickly check if we have enough permission to launch btrfs commands
|
|
if ! btrfs qgroup show "$VOLUME" >/dev/null 2>&1 ; then
|
|
echo "UNKNOWN: unable to launch 'btrfs qgroup show $VOLUME' command."
|
|
exit $STATE_UNKNOWN
|
|
fi
|
|
|
|
# Loop on each qgroup
|
|
# (the kinda weird <<EOF at the end is to avoid entering a sub-shell, so we can keep
|
|
# our precious vars even with prosix-strict dash)
|
|
while read QGROUPID RFER EXCL MAXRFER MAXEXCL SUBVOLUMEPATH; do
|
|
LOOP_OUTPUT_STATUS=$STATE_OK
|
|
PERFDATA_RFER="$RFER""B"
|
|
PERFDATA_EXCL="$EXCL""B"
|
|
if [ "$MAXRFER" != "none" ]; then
|
|
PERFDATA_RFER="$( printf "%s;%d;%d;0;%d" "$PERFDATA_RFER" "$( get_absolute_threshold "$MAXRFER" "$RANGE_WARNING" )" "$( get_absolute_threshold "$MAXRFER" "$RANGE_CRITICAL" )" "$MAXRFER" )"
|
|
if ! check_above_threshold "$RFER" "$MAXRFER" "$RANGE_CRITICAL"; then
|
|
OUTPUT_EXIT_STATUS=$STATE_CRITICAL
|
|
LOOP_OUTPUT_STATUS=$STATE_CRITICAL
|
|
elif ! check_above_threshold "$RFER" "$MAXRFER" "$RANGE_WARNING"; then
|
|
[ "$OUTPUT_EXIT_STATUS" = "$STATE_OK" ] && OUTPUT_EXIT_STATUS=$STATE_WARNING
|
|
LOOP_OUTPUT_STATUS=$STATE_WARNING
|
|
fi
|
|
fi
|
|
if [ "$MAXEXCL" != "none" ]; then
|
|
PERFDATA_EXCL="$( printf "%s;%d;%d;0;%d" "$PERFDATA_EXCL" "$( get_absolute_threshold "$MAXEXCL" "$RANGE_WARNING" )" "$( get_absolute_threshold "$MAXEXCL" "$RANGE_CRITICAL" )" "$MAXEXCL" )"
|
|
if ! check_above_threshold "$EXCL" "$MAXEXCL" "$RANGE_CRITICAL"; then
|
|
OUTPUT_EXIT_STATUS=$STATE_CRITICAL
|
|
LOOP_OUTPUT_STATUS=$STATE_CRITICAL
|
|
elif ! check_above_threshold "$EXCL" "$MAXEXCL" "$RANGE_WARNING"; then
|
|
[ "$OUTPUT_EXIT_STATUS" = "$STATE_OK" ] && OUTPUT_EXIT_STATUS=$STATE_WARNING
|
|
[ "$LOOP_OUTPUT_STATUS" = "$STATE_OK" ] && LOOP_OUTPUT_STATUS=$STATE_WARNING
|
|
fi
|
|
fi
|
|
|
|
# Depending on the alerts raised during this loop, we store
|
|
# the output in ok/warning/critical. This way, in case of critical
|
|
# alert, we don't list all the ok data.
|
|
TMP="$( format_output_item "$VOLUME" "$QGROUPID" "$RFER" "$EXCL" "$MAXRFER" "$MAXEXCL" "$SUBVOLUMEPATH" )"
|
|
case "$LOOP_OUTPUT_STATUS" in
|
|
"$STATE_OK")
|
|
OUTPUT_DETAIL_OK="$OUTPUT_DETAIL_OK $TMP" ;;
|
|
"$STATE_WARNING")
|
|
OUTPUT_DETAIL_WARNING="$OUTPUT_DETAIL_WARNING $TMP" ;;
|
|
"$STATE_CRITICAL")
|
|
OUTPUT_DETAIL_CRITICAL="$OUTPUT_DETAIL_CRITICAL $TMP" ;;
|
|
esac
|
|
|
|
# We had some errors with pnp4nagios with labels containing '<>' like '<toplevel>' so
|
|
# we slaped a tr '<>' '_' at the end...
|
|
OUTPUT_PERFDATA="$( printf "%s\n%s\n%s" "$OUTPUT_PERFDATA" "$VOLUME:$QGROUPID:$SUBVOLUMEPATH:rfer=$PERFDATA_RFER" "$VOLUME:$QGROUPID:$SUBVOLUMEPATH:excl=$PERFDATA_EXCL" | tr '<>' '_')"
|
|
done <<EOF
|
|
$( btrfs qgroup show -re --raw "$VOLUME" | tail -n +3 )
|
|
EOF
|
|
;;
|
|
|
|
\?)
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
case "$OUTPUT_EXIT_STATUS" in
|
|
'0')
|
|
printf "OK %s" "$OUTPUT_DETAIL_OK"
|
|
;;
|
|
'1')
|
|
printf "WARNING %s" "$OUTPUT_DETAIL_WARNING"
|
|
;;
|
|
'2')
|
|
printf "CRITICAL %s" "$OUTPUT_DETAIL_CRITICAL"
|
|
;;
|
|
*)
|
|
printf "UNKNOWN"
|
|
;;
|
|
esac
|
|
|
|
# We sort the perfdata because of a weird bug in some icinga/pnp4nagios interactions
|
|
printf "|%s\n" "$( printf "%s" "$OUTPUT_PERFDATA" | sort | grep -v "^$" | tr "\n" " " )"
|
|
exit $OUTPUT_EXIT_STATUS
|