Script to set a ZFS quota to all user of a pool
Or dataset…
This commit is contained in:
parent
c7d3012d0c
commit
3d14412124
GPG Key ID:
E759BAA22501AF32
|
|
@ -0,0 +1,375 @@
|
|||
#!/bin/sh
|
||||
#
|
||||
# Purpose {{{
|
||||
# This script will set a quota for all users of a given ZFS pool or dataset
|
||||
# 1. Verify the ZFS pool/dataset is reachable
|
||||
# 2. Get users list of pool/dataset
|
||||
# …
|
||||
#
|
||||
# 2021-11-22
|
||||
# }}}
|
||||
# Vars {{{
|
||||
readonly PROGNAME=$(basename "${0}")
|
||||
readonly PROGDIR=$(readlink -m $(dirname "${0}"))
|
||||
readonly ARGS="${*}"
|
||||
readonly NBARGS="${#}"
|
||||
[ -z "${DEBUG}" ] && DEBUG=1
|
||||
## Export DEBUG for sub-script
|
||||
export DEBUG
|
||||
|
||||
# Default values for some vars
|
||||
readonly ZFS_DATASET_DEFAULT="datastore"
|
||||
readonly ZFS_QUOTA_DEFAULT="50G"
|
||||
|
||||
## Colors
|
||||
readonly PURPLE='\033[1;35m'
|
||||
readonly RED='\033[0;31m'
|
||||
readonly RESET='\033[0m'
|
||||
readonly COLOR_DEBUG="${PURPLE}"
|
||||
# }}}
|
||||
usage() { # {{{
|
||||
|
||||
cat <<- EOF
|
||||
usage: $PROGNAME [-d|-h|-p|-q]
|
||||
|
||||
Define a quota to all users of a given ZFS pool/dataset
|
||||
|
||||
EXAMPLES :
|
||||
- Apply default quota (${ZFS_QUOTA_DEFAULT}) to default pool (${ZFS_DATASET_DEFAULT})
|
||||
${PROGNAME}
|
||||
|
||||
- Apply default quota to a given dataset
|
||||
${PROGNAME} --pool "datastore/backup"
|
||||
|
||||
OPTIONS :
|
||||
-d,--debug
|
||||
Enable debug messages.
|
||||
|
||||
-h,--help
|
||||
Print this help message.
|
||||
|
||||
-p,--pool,--dataset
|
||||
Apply quota to the given pool/dataset (default: ${ZFS_DATASET_DEFAULT}).
|
||||
|
||||
-q,--quota,--quotat
|
||||
Define the quota to apply to all users of the pool/dataset (default: ${ZFS_QUOTA_DEFAULT}).
|
||||
|
||||
EOF
|
||||
|
||||
}
|
||||
# }}}
|
||||
debug_message() { # {{{
|
||||
|
||||
local_message="${1}"
|
||||
|
||||
## Print message if DEBUG is enable (=0)
|
||||
[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6b\e[m\n' "DEBUG − ${PROGNAME} : ${local_message}"
|
||||
|
||||
return 0
|
||||
}
|
||||
# }}}
|
||||
error_message() { # {{{
|
||||
|
||||
local_error_message="${1}"
|
||||
local_error_code="${2}"
|
||||
|
||||
## Print message if DEBUG is enable (=0)
|
||||
printf '%b\n' "ERROR − ${PROGNAME} : ${RED}${local_error_message}${RESET}"
|
||||
|
||||
exit "${local_error_code:=66}"
|
||||
}
|
||||
# }}}
|
||||
define_vars() { # {{{
|
||||
|
||||
## If zfs_dataset wasn't defined (argument) {{{
|
||||
if [ -z "${zfs_dataset}" ]; then
|
||||
## Use default value
|
||||
readonly zfs_dataset="${ZFS_DATASET_DEFAULT}"
|
||||
fi
|
||||
## }}}
|
||||
## If zfs_quota wasn't defined (argument) {{{
|
||||
if [ -z "${zfs_quota}" ]; then
|
||||
## Use default value
|
||||
readonly zfs_quota="${ZFS_QUOTA_DEFAULT}"
|
||||
fi
|
||||
## }}}
|
||||
|
||||
## Temp file vars {{{
|
||||
readonly zfs_user_list_path="/tmp/${PROGNAME}.user.list"
|
||||
readonly zfs_previous_user_list_path="/tmp/${PROGNAME}.old.user.list"
|
||||
## }}}
|
||||
|
||||
}
|
||||
# }}}
|
||||
is_command_absent() { # {{{
|
||||
|
||||
local_command_absent_cmd="${1}"
|
||||
|
||||
## A command is absent by default
|
||||
return_command_absent="0"
|
||||
|
||||
if [ "$(command -v ${local_command_absent_cmd})" ]; then
|
||||
debug_message "is_command_absent − \
|
||||
${RED}${local_command_absent_cmd}${COLOR_DEBUG} seems present on this host."
|
||||
return_command_absent="1"
|
||||
else
|
||||
debug_message "is_command_absent − \
|
||||
${RED}${local_command_absent_cmd}${COLOR_DEBUG} is not available on this host."
|
||||
return_command_absent="0"
|
||||
fi
|
||||
|
||||
return "${return_command_absent}"
|
||||
|
||||
}
|
||||
# }}}
|
||||
is_zfs_dataset_exists() { # {{{
|
||||
|
||||
local_zfs_dataset="${1}"
|
||||
|
||||
## Return False by default
|
||||
return_zfs_dataset_exists="1"
|
||||
|
||||
## Use local_zfs_dataset var in zfs command and grep to avoid sub-datasets
|
||||
if zfs list -H -- "${local_zfs_dataset}" 2>/dev/null | grep --quiet "${local_zfs_dataset}"; then
|
||||
debug_message "is_zfs_dataset_exists − \
|
||||
${RED}${local_zfs_dataset}${COLOR_DEBUG} ZFS pool/dataset seems present on this host."
|
||||
return_zfs_dataset_exists="0"
|
||||
else
|
||||
debug_message "is_zfs_dataset_exists − \
|
||||
${RED}${local_zfs_dataset}${COLOR_DEBUG} ZFS pool/dataset is not available on this host."
|
||||
return_zfs_dataset_exists="1"
|
||||
fi
|
||||
|
||||
return "${return_zfs_dataset_exists}"
|
||||
|
||||
}
|
||||
# }}}
|
||||
get_dataset_user_list() { # {{{
|
||||
|
||||
local_zfs_dataset="${1}"
|
||||
|
||||
## Return False by default
|
||||
return_get_dataset_user_list="1"
|
||||
|
||||
debug_message "get_dataset_user_list − \
|
||||
Create or empty ${RED}${zfs_user_list_path}${COLOR_DEBUG} file to store user list of ${RED}${local_zfs_dataset}${COLOR_DEBUG} ZFS pool/dataset."
|
||||
true > "${zfs_user_list_path}"
|
||||
|
||||
if zfs userspace -Hp -o name "${local_zfs_dataset}" >> "${zfs_user_list_path}" 2>/dev/null; then
|
||||
if [ -s "${zfs_user_list_path}" ]; then
|
||||
debug_message "get_dataset_user_list − \
|
||||
${RED}${local_zfs_dataset}${COLOR_DEBUG} users list successfully created (see ${zfs_user_list_path} file)."
|
||||
command chmod 0400 -- "${zfs_user_list_path}"
|
||||
return_get_dataset_user_list="0"
|
||||
else
|
||||
debug_message "get_dataset_user_list − \
|
||||
Error, the user list of ${local_zfs_dataset} is empty (${zfs_user_list_path} file)."
|
||||
return_get_dataset_user_list="1"
|
||||
fi
|
||||
else
|
||||
debug_message "get_dataset_user_list − \
|
||||
Error in ${RED}zfs userspace${COLOR_DEBUG} command for ${local_zfs_dataset} ZFS pool/dataset."
|
||||
return_get_dataset_user_list="1"
|
||||
fi
|
||||
|
||||
return "${return_get_dataset_user_list}"
|
||||
|
||||
}
|
||||
# }}}
|
||||
is_file_present() { # {{{
|
||||
|
||||
local_file_present="${1}"
|
||||
|
||||
## File doesn't exist by default
|
||||
return_is_file_present="1"
|
||||
|
||||
### Check if the file exists
|
||||
# shellcheck disable=SC2086
|
||||
if find ${local_file_present} > /dev/null 2>&1; then
|
||||
return_is_file_present="0"
|
||||
debug_message "is_file_present − \
|
||||
The file ${RED}${local_file_present}${COLOR_DEBUG} exists."
|
||||
else
|
||||
return_is_file_present="1"
|
||||
debug_message "is_file_present − \
|
||||
The file ${RED}${local_file_present}${COLOR_DEBUG} doesn't exist."
|
||||
fi
|
||||
|
||||
return "${return_is_file_present}"
|
||||
|
||||
}
|
||||
# }}}
|
||||
is_file_similar() { # {{{
|
||||
|
||||
local_similar_file_one="${1}"
|
||||
local_similar_file_two="${2}"
|
||||
|
||||
## Files aren't similar by default doesn't exist by default
|
||||
return_is_file_similar="1"
|
||||
|
||||
if diff --brief -- "${local_similar_file_one}" "${local_similar_file_two}" > /dev/null; then
|
||||
debug_message "is_file_similar − \
|
||||
${local_similar_file_one} and ${local_similar_file_two} are ${RED}similar${COLOR_DEBUG}."
|
||||
return_is_file_similar="0"
|
||||
else
|
||||
debug_message "is_file_similar − \
|
||||
${local_similar_file_one} and ${local_similar_file_two} are ${RED}NOT${COLOR_DEBUG} similar."
|
||||
return_is_file_similar="1"
|
||||
fi
|
||||
|
||||
return "${return_is_file_similar}"
|
||||
}
|
||||
# }}}
|
||||
apply_zfs_userquota() { # {{{
|
||||
|
||||
local_zfs_user="${1}"
|
||||
local_zfs_quota="${2}"
|
||||
|
||||
## Return False by default
|
||||
return_apply_zfs_userquota="1"
|
||||
|
||||
## Use local_zfs_dataset var in zfs command and grep to avoid sub-datasets
|
||||
if zfs userquota@"${local_zfs_user}"="${local_zfs_quota}" "${local_zfs_dataset}" 2>/dev/null; then
|
||||
debug_message "apply_zfs_userquota − \
|
||||
Set new quota (${RED}${local_zfs_quota}${COLOR_DEBUG}) for ${RED}${local_zfs_user}${COLOR_DEBUG} user."
|
||||
return_apply_zfs_userquota="0"
|
||||
else
|
||||
debug_message "apply_zfs_userquota − \
|
||||
Can't set a new quota (${RED}${local_zfs_quota}${COLOR_DEBUG}) for ${RED}${local_zfs_user}${COLOR_DEBUG} user \
|
||||
('zfs userquota' command returns : ${?})"
|
||||
return_apply_zfs_userquota="1"
|
||||
fi
|
||||
|
||||
return "${return_apply_zfs_userquota}"
|
||||
|
||||
}
|
||||
# }}}
|
||||
|
||||
main() { # {{{
|
||||
|
||||
## This script should run as root {{{
|
||||
if ! [ $(id -u) = 0 ]; then
|
||||
error_message "Please run this script in root or with sudo." 1
|
||||
fi
|
||||
## }}}
|
||||
## If ZFS command is absent from the system {{{
|
||||
### Exit
|
||||
is_command_absent "zfs" \
|
||||
&& exit 0
|
||||
## }}}
|
||||
|
||||
## Define all vars
|
||||
define_vars
|
||||
|
||||
## If ZFS pool/dataset is not available {{{
|
||||
### Exit
|
||||
is_zfs_dataset_exists "${zfs_dataset}" \
|
||||
|| error_message "Please verify your ZFS pool (${zfs_dataset} doesn't seems available" 2
|
||||
## }}}
|
||||
|
||||
## Try to get the user list of ZFS pool/dataset {{{
|
||||
### OR Exit
|
||||
get_dataset_user_list "${zfs_dataset}" \
|
||||
|| error_message "Can't get the user list of ${zfs_dataset} ZFS pool/dataset. Please use --debug option." 3
|
||||
## }}}
|
||||
|
||||
## If a previous list of users exists {{{
|
||||
### If the two list are the same
|
||||
### Exit
|
||||
is_file_present "${zfs_previous_user_list_path}" \
|
||||
&& is_file_similar "${zfs_user_list_path}" "${zfs_previous_user_list_path}" \
|
||||
&& debug_message "main − No new user from previous run, no more actions required." \
|
||||
&& exit 0
|
||||
## }}}
|
||||
|
||||
## Information message
|
||||
debug_message "Apply quota (${RED}${zfs_quota}${COLOR_DEBUG}) to all users of ZFS pool/dataset : ${RED}${zfs_dataset}${COLOR_DEBUG}"
|
||||
|
||||
## Read user one by one
|
||||
while IFS= read -r zfs_username; do
|
||||
### Try to apply the quota to the user {{{
|
||||
### OR Exit with error
|
||||
apply_zfs_userquota "${zfs_username}" "${zfs_quota}" \
|
||||
|| error_message "Can't define the new quota (${zfs_quota}) for ${zfs_username} user in ${zfs_dataset} ZFS pool/dataset. Please use --debug option." 4
|
||||
### }}}
|
||||
done < "${zfs_user_list_path}"
|
||||
|
||||
## Rename user list for next run of the script
|
||||
## AND exit
|
||||
mv --force -- "${zfs_user_list_path}" "${zfs_previous_user_list_path}" \
|
||||
&& exit 0
|
||||
|
||||
}
|
||||
# }}}
|
||||
|
||||
# Manage arguments # {{{
|
||||
# This code can't be in a function due to argument management
|
||||
|
||||
if [ ! "${NBARGS}" -eq "0" ]; then
|
||||
|
||||
manage_arg="0"
|
||||
|
||||
## If the first argument is not an option
|
||||
if ! printf -- '%s' "${1}" | grep -q -E -- "^-+";
|
||||
then
|
||||
## Print help message and exit
|
||||
printf '%b\n' "${RED}Invalid option: ${1}${RESET}"
|
||||
printf '%b\n' "---"
|
||||
usage
|
||||
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Parse all options (start with a "-") one by one
|
||||
while printf -- '%s' "${1}" | grep -q -E -- "^-+"; do
|
||||
|
||||
case "${1}" in
|
||||
-d|--debug ) ## debug
|
||||
DEBUG=0
|
||||
;;
|
||||
-h|--help ) ## help
|
||||
usage
|
||||
## Exit after help informations
|
||||
exit 0
|
||||
;;
|
||||
-p|--pool|--dataset ) ## Define zfs_dataset
|
||||
## Move to the next argument
|
||||
shift
|
||||
## Define var
|
||||
readonly zfs_dataset="${1}"
|
||||
;;
|
||||
-q|--quota|--quotat ) ## Define zfs_quota
|
||||
## Move to the next argument
|
||||
shift
|
||||
## Define var
|
||||
readonly zfs_quota="${1}"
|
||||
;;
|
||||
* ) ## unknow option
|
||||
printf '%b\n' "${RED}Invalid option: ${1}${RESET}"
|
||||
printf '%b\n' "---"
|
||||
usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
debug_message "Arguments management − \
|
||||
${RED}${1}${COLOR_DEBUG} option managed."
|
||||
|
||||
## Move to the next argument
|
||||
shift
|
||||
manage_arg=$((manage_arg+1))
|
||||
|
||||
done
|
||||
|
||||
debug_message "Arguments management − \
|
||||
${RED}${manage_arg}${COLOR_DEBUG} argument(s) successfully managed."
|
||||
else
|
||||
debug_message "Arguments management − \
|
||||
No arguments/options to manage."
|
||||
fi
|
||||
|
||||
# }}}
|
||||
|
||||
main
|
||||
|
||||
exit 255
|
||||
Loading…
Reference in New Issue