51 lines
1.2 KiB
Bash
Executable File
51 lines
1.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -e
|
|
|
|
[[ "${DEBUG}" == 'true' ]] && set -x
|
|
|
|
# Generate password if hash not set
|
|
if [[ ! -z "${FTP_PASSWORD}" && -z "${FTP_PASSWORD_HASH}" ]]; then
|
|
FTP_PASSWORD_HASH=$(echo "${FTP_PASSWORD}" | mkpasswd -s -m sha-512)
|
|
fi
|
|
|
|
if [[ ! -z "${FTP_USER}" && ! -z "${FTP_PASSWORD_HASH}" ]]; then
|
|
echo "Adding user ${FTP_USER}"
|
|
/add-virtual-user.sh -d "${FTP_USER}" "${FTP_PASSWORD_HASH}"
|
|
fi
|
|
|
|
# Support multiple users
|
|
while read user; do
|
|
IFS=: read name pass <<< "${!user}"
|
|
echo "Adding user ${name}"
|
|
/add-virtual-user.sh "${name}" "${pass}"
|
|
done < <(env | grep "FTP_USER_" | sed 's/^\(FTP_USER_[a-zA-Z0-9]*\)=.*/\1/')
|
|
|
|
# Support user directories
|
|
if [[ ! -z "${FTP_USERS_ROOT}" ]]; then
|
|
sed -i 's/local_root=.*/local_root=\/srv\/$USER/' "/etc/vsftpd*.conf"
|
|
fi
|
|
|
|
vsftpd_stop() {
|
|
echo "Received SIGINT or SIGTERM. Shutting down vsftpd"
|
|
# Get PID
|
|
pid="$(cat /var/run/vsftpd/vsftpd.pid)"
|
|
# Set TERM
|
|
kill -SIGTERM "${pid}"
|
|
# Wait for exit
|
|
wait "${pid}"
|
|
# All done.
|
|
echo "Done"
|
|
}
|
|
|
|
if [[ "$1" == "vsftpd" ]]; then
|
|
trap vsftpd_stop SIGINT SIGTERM
|
|
echo "Running $@"
|
|
$@ &
|
|
pid="$!"
|
|
echo "${pid}" > /var/run/vsftpd/vsftpd.pid
|
|
wait "${pid}" && exit $?
|
|
else
|
|
exec "$@"
|
|
fi
|