162 lines
3.6 KiB
Bash
Executable File
162 lines
3.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(realpath "$(dirname "${BASH_SOURCE[0]}")")"
|
|
SYSTEMD_UNIT_FILENAME="infection-monkey.service"
|
|
SYSTEMD_DIR="/lib/systemd/system"
|
|
MONKEY_BIN="/opt/infection-monkey/bin"
|
|
APPIMAGE_NAME="InfectionMonkey.AppImage"
|
|
|
|
echo_help() {
|
|
echo "Installs the Infection Monkey service to run on boot."
|
|
echo ""
|
|
echo "Usage:"
|
|
echo " install-infection-monkey-service.sh --user <USERNAME> --appimage <PATH>"
|
|
echo " install-infection-monkey-service.sh --uninstall"
|
|
echo " install-infection-monkey-service.sh -h|--help"
|
|
echo ""
|
|
echo "Options:"
|
|
echo " --user User to run the service as"
|
|
echo " --appimage Path to AppImage"
|
|
echo " --uninstall Uninstall the Infection Monkey service"
|
|
}
|
|
|
|
install_service() {
|
|
move_appimage "$2"
|
|
|
|
cat > "${SCRIPT_DIR}/${SYSTEMD_UNIT_FILENAME}" << EOF
|
|
[Unit]
|
|
Description=Infection Monkey Runner
|
|
After=network.target
|
|
|
|
[Service]
|
|
User=$1
|
|
Type=simple
|
|
ExecStart="${MONKEY_BIN}/${APPIMAGE_NAME}"
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
EOF
|
|
|
|
umask 077
|
|
sudo mv "${SCRIPT_DIR}/${SYSTEMD_UNIT_FILENAME}" "${SYSTEMD_DIR}/${SYSTEMD_UNIT_FILENAME}"
|
|
sudo systemctl enable "${SYSTEMD_UNIT_FILENAME}" &>/dev/null
|
|
|
|
echo -e "The Infection Monkey service has been installed and will start on boot.\n\
|
|
Run 'systemctl start infection-monkey' to start the service now."
|
|
}
|
|
|
|
uninstall_service() {
|
|
if [ -f "${MONKEY_BIN}/${APPIMAGE_NAME}" ] ; then
|
|
sudo rm -f "${MONKEY_BIN}/${APPIMAGE_NAME}"
|
|
fi
|
|
|
|
if [ -f "${SYSTEMD_DIR}/${SYSTEMD_UNIT_FILENAME}" ] ; then
|
|
sudo systemctl stop "${SYSTEMD_UNIT_FILENAME}" 2>/dev/null
|
|
sudo systemctl disable "${SYSTEMD_UNIT_FILENAME}" &>/dev/null
|
|
sudo rm "${SYSTEMD_DIR}/${SYSTEMD_UNIT_FILENAME}"
|
|
sudo systemctl daemon-reload
|
|
fi
|
|
|
|
echo "The Infection Monkey service has been uninstalled"
|
|
}
|
|
|
|
move_appimage() {
|
|
sudo mkdir --mode=0755 -p "${MONKEY_BIN}"
|
|
|
|
if [ "$1" != "${MONKEY_BIN}/${APPIMAGE_NAME}" ] ; then
|
|
umask 022
|
|
sudo cp "$appimage_path" "${MONKEY_BIN}/${APPIMAGE_NAME}"
|
|
sudo chmod 755 "${MONKEY_BIN}/${APPIMAGE_NAME}"
|
|
fi
|
|
}
|
|
|
|
user_exists() {
|
|
id -u "$1" &>/dev/null
|
|
}
|
|
|
|
assert_parameter_supplied() {
|
|
if [ -z "$2" ] ; then
|
|
echo "Error: missing required parameter '$1'"
|
|
echo_help
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
has_sudo() {
|
|
# 0 true, 1 false
|
|
sudo -nv > /dev/null 2>&1
|
|
return $?
|
|
}
|
|
|
|
exit_if_missing_argument() {
|
|
if [ -z "$2" ] || [ "${2:0:1}" == "-" ]; then
|
|
echo "Error: Argument for parameter '$1' is missing" >&2
|
|
echo_help
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
do_uninstall=false
|
|
uname=""
|
|
appimage_path=""
|
|
|
|
while (( "$#" )); do
|
|
case "$1" in
|
|
--user)
|
|
exit_if_missing_argument "$1" "$2"
|
|
|
|
uname=$2
|
|
shift 2
|
|
;;
|
|
--appimage)
|
|
exit_if_missing_argument "$1" "$2"
|
|
|
|
appimage_path=$2
|
|
shift 2
|
|
;;
|
|
--uninstall)
|
|
do_uninstall=true
|
|
shift
|
|
;;
|
|
-h|--help)
|
|
echo_help
|
|
exit 0
|
|
;;
|
|
*)
|
|
echo "Error: Unsupported parameter $1" >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if ! has_sudo; then
|
|
echo "Error: You need root permissions for some of this script operations. \
|
|
Run \`sudo -v\`, enter your password, and then re-run this script."
|
|
exit 1
|
|
fi
|
|
|
|
if $do_uninstall ; then
|
|
uninstall_service
|
|
exit 0
|
|
fi
|
|
|
|
assert_parameter_supplied "--user" "$uname"
|
|
assert_parameter_supplied "--appimage" "$appimage_path"
|
|
|
|
if ! user_exists "$uname" ; then
|
|
echo "Error: User '$uname' does not exist"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! -f "$appimage_path" ] ; then
|
|
if [ ! -f "${SCRIPT_DIR}/$appimage_path" ] ; then
|
|
echo "Error: AppImage '$appimage_path' does not exist"
|
|
exit 1
|
|
fi
|
|
appimage_path="${SCRIPT_DIR}/$appimage_path"
|
|
fi
|
|
|
|
install_service "$uname" "$appimage_path"
|