2022-12-19 14:19:26 +00:00
|
|
|
# shellcheck shell=bash
|
2022-10-16 19:46:42 +00:00
|
|
|
# shellcheck disable=SC2034
|
|
|
|
# SC2034: "Variable appears unused. Verify it or export it."
|
|
|
|
# Those are intentional here, as the file is meant to be included elsewhere.
|
|
|
|
|
2023-10-02 14:26:32 +00:00
|
|
|
# NOTE: If using another privilege escalation binary make sure it is configured or has the appropriate flag
|
2022-11-23 20:01:11 +00:00
|
|
|
# to keep the current environment variables in the launched process (in sudo's case this is achieved
|
|
|
|
# through the -E flag described in sudo(8).
|
2022-10-16 19:46:42 +00:00
|
|
|
die() {
|
|
|
|
echo "die: $*"
|
|
|
|
exit 1
|
|
|
|
}
|
2022-12-06 17:17:41 +00:00
|
|
|
|
2023-04-24 19:09:23 +00:00
|
|
|
exit_if_running_as_root() {
|
|
|
|
if [ "$(id -u)" -eq 0 ]; then
|
|
|
|
die "$*"
|
|
|
|
fi
|
|
|
|
}
|
|
|
|
|
2022-12-06 17:17:41 +00:00
|
|
|
find_executable() {
|
|
|
|
paths=("/usr/sbin" "/sbin")
|
|
|
|
|
2022-12-18 23:56:47 +00:00
|
|
|
if [ "$(uname -s)" = "Darwin" ]; then
|
|
|
|
if [ -n "${HOMEBREW_PREFIX}" ]; then
|
|
|
|
paths+=("${HOMEBREW_PREFIX}/opt/e2fsprogs/bin" "${HOMEBREW_PREFIX}/opt/e2fsprogs/sbin")
|
|
|
|
elif command -v brew > /dev/null 2>&1; then
|
|
|
|
if prefix=$(brew --prefix e2fsprogs 2>/dev/null); then
|
|
|
|
paths+=("${prefix}/bin" "${prefix}/sbin")
|
|
|
|
fi
|
|
|
|
fi
|
2022-12-06 17:17:41 +00:00
|
|
|
fi
|
|
|
|
|
|
|
|
executable="${1}"
|
|
|
|
|
|
|
|
# Prefer tools from PATH over fallback paths
|
|
|
|
if command -v "${executable}"; then
|
|
|
|
return 0
|
|
|
|
fi
|
|
|
|
|
|
|
|
for path in "${paths[@]}"; do
|
|
|
|
if command -v "${path}/${executable}"; then
|
|
|
|
return 0
|
|
|
|
fi
|
|
|
|
done
|
|
|
|
|
|
|
|
# We return the executable's name back to provide meaningful messages on future failure
|
|
|
|
echo "${executable}"
|
|
|
|
}
|
|
|
|
|
|
|
|
FUSE2FS_PATH="$(find_executable fuse2fs)"
|
|
|
|
RESIZE2FS_PATH="$(find_executable resize2fs)"
|
2022-12-10 18:11:59 +00:00
|
|
|
E2FSCK_PATH="$(find_executable e2fsck)"
|
2022-12-15 15:25:44 +00:00
|
|
|
MKE2FS_PATH="$(find_executable mke2fs)"
|
2023-05-02 19:04:35 +00:00
|
|
|
|
|
|
|
get_number_of_processing_units() {
|
|
|
|
number_of_processing_units="nproc"
|
|
|
|
SYSTEM_NAME="$(uname -s)"
|
|
|
|
|
|
|
|
if [ "$SYSTEM_NAME" = "OpenBSD" ]; then
|
|
|
|
number_of_processing_units="sysctl -n hw.ncpuonline"
|
|
|
|
elif [ "$SYSTEM_NAME" = "FreeBSD" ]; then
|
|
|
|
number_of_processing_units="sysctl -n hw.ncpu"
|
|
|
|
elif [ "$SYSTEM_NAME" = "Darwin" ]; then
|
|
|
|
number_of_processing_units="sysctl -n hw.ncpu"
|
|
|
|
fi
|
|
|
|
|
|
|
|
($number_of_processing_units)
|
|
|
|
}
|
2023-07-08 05:46:49 +00:00
|
|
|
|
|
|
|
# We depend on GNU coreutils du for the --apparent-size extension.
|
|
|
|
# GNU coreutils is a build dependency.
|
|
|
|
if command -v gdu > /dev/null 2>&1 && gdu --version | grep -q "GNU coreutils"; then
|
|
|
|
GNUDU="gdu"
|
|
|
|
else
|
|
|
|
GNUDU="du"
|
|
|
|
fi
|
|
|
|
|
|
|
|
disk_usage() {
|
|
|
|
# shellcheck disable=SC2003,SC2307
|
|
|
|
expr "$(${GNUDU} -sbm "$1" | cut -f1)"
|
|
|
|
}
|
|
|
|
|
|
|
|
inode_usage() {
|
|
|
|
find "$1" | wc -l
|
|
|
|
}
|