mirror of https://github.com/Safe3/uuWAF
131 lines
2.8 KiB
Bash
131 lines
2.8 KiB
Bash
#!/bin/bash
|
|
|
|
warning() {
|
|
echo -e "\033[33m[UUSEC WAF] $*\033[0m"
|
|
}
|
|
|
|
abort() {
|
|
echo -e "\033[31m[UUSEC WAF] $*\033[0m"
|
|
exit 1
|
|
}
|
|
|
|
if [ -z "$BASH" ]; then
|
|
abort "Please execute this script using bash and refer to the latest official technical documentation https://uuwaf.uusec.com/"
|
|
fi
|
|
|
|
if [ "$EUID" -ne "0" ]; then
|
|
abort "Please run with root privileges"
|
|
fi
|
|
|
|
SCRIPT_PATH="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
|
cd "$SCRIPT_PATH"
|
|
|
|
if [ ! $(command -v docker) ]; then
|
|
warning "Docker Engine not detected, we will automatically install it for you. The process is slow, please be patient ..."
|
|
sh install-docker.sh
|
|
if [ $? -ne "0" ]; then
|
|
abort "Automatic installation of Docker Engine failed. Please manually install it before executing this script"
|
|
fi
|
|
systemctl start docker && systemctl enable docker
|
|
fi
|
|
|
|
DC_CMD="docker compose"
|
|
$DC_CMD version > /dev/null 2>&1
|
|
if [ $? -ne "0" ]; then
|
|
abort "Your Docker version is too low and lacks the Docker compose command. Please uninstall and install the latest version"
|
|
fi
|
|
|
|
stop_uuwaf(){
|
|
$DC_CMD down
|
|
}
|
|
|
|
uninstall_uuwaf(){
|
|
stop_uuwaf
|
|
docker rm -f uuwaf wafdb > /dev/null 2>&1
|
|
docker network rm wafnet > /dev/null 2>&1
|
|
docker images|grep uuwaf|awk '{print $3}'|xargs docker rmi -f > /dev/null 2>&1
|
|
docker volume ls|grep waf|awk '{print $2}'|xargs docker volume rm -f > /dev/null 2>&1
|
|
}
|
|
|
|
start_uuwaf(){
|
|
if [ ! $(command -v netstat) ]; then
|
|
$( command -v yum || command -v apt-get ) -y install net-tools
|
|
fi
|
|
port_status=`netstat -nlt|grep -E ':(80|443|4443)\s'|wc -l`
|
|
if [ $port_status -gt 0 ]; then
|
|
echo -e "\t One or more of ports 80, 443, 4443 are occupied. Please shut down the corresponding service or modify its port"
|
|
exit 1
|
|
fi
|
|
$DC_CMD up -d
|
|
}
|
|
|
|
update_uuwaf(){
|
|
stop_uuwaf
|
|
docker images|grep uuwaf|awk '{print $3}'|xargs docker rmi -f > /dev/null 2>&1
|
|
docker volume ls|grep wafshared|awk '{print $2}'|xargs docker volume rm -f > /dev/null 2>&1
|
|
start_uuwaf
|
|
}
|
|
|
|
restart_uuwaf(){
|
|
stop_uuwaf
|
|
start_uuwaf
|
|
}
|
|
|
|
clean_uuwaf(){
|
|
docker system prune -a -f
|
|
docker volume prune -a -f
|
|
}
|
|
|
|
start_menu(){
|
|
clear
|
|
echo "========================="
|
|
echo "UUSEC WAF Docker Management"
|
|
echo "========================="
|
|
echo "1. Start"
|
|
echo "2. Stop"
|
|
echo "3. Restart"
|
|
echo "4. Update"
|
|
echo "5. Uninstall"
|
|
echo "6. Clean"
|
|
echo "7. Exit"
|
|
echo
|
|
read -p "Please enter the number: " num
|
|
case "$num" in
|
|
1)
|
|
start_uuwaf
|
|
echo "Startup completed"
|
|
;;
|
|
2)
|
|
stop_uuwaf
|
|
echo "Stop completed"
|
|
;;
|
|
3)
|
|
restart_uuwaf
|
|
echo "Restart completed"
|
|
;;
|
|
4)
|
|
update_uuwaf
|
|
echo "Update completed"
|
|
;;
|
|
5)
|
|
uninstall_uuwaf
|
|
echo "Uninstall completed"
|
|
;;
|
|
6)
|
|
clean_uuwaf
|
|
echo "Clean completed"
|
|
;;
|
|
7)
|
|
exit 1
|
|
;;
|
|
*)
|
|
clear
|
|
echo "Please enter the right number"
|
|
;;
|
|
esac
|
|
sleep 3s
|
|
start_menu
|
|
}
|
|
|
|
start_menu
|