mirror of
https://github.com/donaldzou/WGDashboard.git
synced 2024-11-06 16:00:28 +01:00
164 lines
6.4 KiB
Bash
164 lines
6.4 KiB
Bash
#!/bin/bash
|
|
echo "Starting the WireGuard Dashboard Docker container."
|
|
|
|
# === CLEAN UP ===
|
|
clean_up() {
|
|
echo "--------------------- CLEAN UP -----------------------"
|
|
# Cleaning out previous data such as the .pid file and starting the WireGuard Dashboard. Making sure to use the python venv.
|
|
echo "Looking for remains of previous instances..."
|
|
local pid_file="${WGDASH}/src/gunicorn.pid"
|
|
if [ -f $pid_file ]; then
|
|
echo "Found old pid file, removing."
|
|
rm $pid_file
|
|
else
|
|
echo "No pid remains found, continuing."
|
|
fi
|
|
|
|
local pycache="${WGDASH}/src/__pycache__"
|
|
if [ -d "$pycache" ]; then
|
|
local pycache_filecount=$(find "$pycache" -maxdepth 1 -type f | wc -l)
|
|
if [ "$pycache_filecount" -gt 0 ]; then
|
|
echo "Found old pycaches, removing."
|
|
rm -rf "$pycache"/*
|
|
else
|
|
echo "No pycaches found, continuing."
|
|
fi
|
|
else
|
|
echo "No pycaches found, continuing."
|
|
fi
|
|
|
|
echo "Setting permissions to not be world-accesible."
|
|
chmod 640 /etc/wireguard/*
|
|
}
|
|
|
|
# === CORE SERVICES ===
|
|
start_core() {
|
|
echo "--------------------- STARTING CORE -----------------------"
|
|
|
|
# This first step is to ensure the wg0.conf file exists, and if not, then its copied over from the ephemeral container storage.
|
|
if [ ! -f "/etc/wireguard/wg0.conf" ]; then
|
|
cp "/wg0.conf" "/etc/wireguard/wg0.conf"
|
|
echo "Standard WG0 Configuration file not found, grabbing template."
|
|
else
|
|
echo "Standard WG0 Configuration file found, using that."
|
|
fi
|
|
|
|
echo "Activating Python venv and executing the WireGuard Dashboard service."
|
|
. "${WGDASH}"/src/venv/bin/activate
|
|
cd "${WGDASH}"/src || return # If changing the directory fails (permission or presence error), then bash will exist this function, causing the WireGuard Dashboard to not be succesfully launched.
|
|
bash wgd.sh start
|
|
|
|
# Isolated peers feature:
|
|
local configurations=(/etc/wireguard/*)
|
|
IFS=',' read -r -a do_isolate <<< "${isolate}"
|
|
non_isolate=()
|
|
|
|
for config in "${configurations[@]}"; do
|
|
local config=$(echo "$config" | sed -e 's|.*/etc/wireguard/||' -e 's|\.conf$||')
|
|
found=false
|
|
for interface in "${do_isolate[@]}"; do
|
|
if [[ "$config" == "$interface" ]]; then
|
|
found=true
|
|
break
|
|
fi
|
|
done
|
|
if [ "$found" = false ]; then
|
|
non_isolate+=("$config")
|
|
fi
|
|
done
|
|
|
|
for interface in "${do_isolate[@]}"; do
|
|
if [ -f "/etc/wireguard/${interface}.conf" ]; then
|
|
echo "Isolating:" $interface
|
|
upblocking=$(grep -c "PostUp = iptables -I FORWARD -i ${interface} -o ${interface} -j DROP" /etc/wireguard/${interface}.conf)
|
|
downblocking=$(grep -c "PreDown = iptables -D FORWARD -i ${interface} -o ${interface} -j DROP" /etc/wireguard/${interface}.conf)
|
|
|
|
if [ "$upblocking" -lt 1 ] && [ "$downblocking" -lt 1 ]; then
|
|
sed -i "/PostUp =/a PostUp = iptables -I FORWARD -i ${interface} -o ${interface} -j DROP" /etc/wireguard/${interface}.conf
|
|
sed -i "/PreDown =/a PreDown = iptables -D FORWARD -i ${interface} -o ${interface} -j DROP" /etc/wireguard/${interface}.conf
|
|
fi
|
|
else
|
|
echo "Configuration for $interface does not seem to exist, continuing."
|
|
fi
|
|
done
|
|
|
|
for interface in "${non_isolate[@]}"; do
|
|
if [ -f "/etc/wireguard/${interface}.conf" ]; then
|
|
echo "Removing Isolation if present for:" $interface
|
|
sed -i "/PostUp = iptables -I FORWARD -i ${interface} -o ${interface} -j DROP/d" /etc/wireguard/${interface}.conf
|
|
sed -i "/PreDown = iptables -D FORWARD -i ${interface} -o ${interface} -j DROP/d" /etc/wireguard/${interface}.conf
|
|
else
|
|
echo "Configuration for $interface does not seem to exist, continuing."
|
|
fi
|
|
done
|
|
|
|
# The following section takes care of enabling wireguard interfaces on startup.
|
|
IFS=',' read -r -a enable_array <<< "${enable}"
|
|
|
|
for interface in "${enable_array[@]}"; do
|
|
echo "Preference for $interface to be turned on found."
|
|
if [ -f "/etc/wireguard/${interface}.conf" ]; then
|
|
echo "Found corresponding configuration file, activating..."
|
|
wg-quick up $interface
|
|
else
|
|
echo "No corresponding configuration file found for $interface doing nothing."
|
|
fi
|
|
done
|
|
}
|
|
|
|
# === SET ENV VARS ===
|
|
set_envvars() {
|
|
echo "------------- SETTING ENVIRONMENT VARIABLES ----------------"
|
|
|
|
# If the timezone is different, for example in North-America or Asia.
|
|
if [ "${tz}" != "$(cat /etc/timezone)" ]; then
|
|
echo "Changing timezone."
|
|
|
|
ln -sf /usr/share/zoneinfo/"${tz}" /etc/localtime
|
|
echo "${tz}" > /etc/timezone
|
|
fi
|
|
|
|
# Changing the DNS used for clients and the dashboard itself.
|
|
if [ "${global_dns}" != "$(grep "peer_global_dns = " /opt/wireguarddashboard/src/wg-dashboard.ini | awk '{print $NF}')" ]; then
|
|
echo "Changing default dns."
|
|
|
|
#sed -i "s/^DNS = .*/DNS = ${global_dns}/" /etc/wireguard/wg0.conf # Uncomment if you want to have DNS on server-level.
|
|
sed -i "s/^peer_global_dns = .*/peer_global_dns = ${global_dns}/" /opt/wireguarddashboard/src/wg-dashboard.ini
|
|
fi
|
|
|
|
# Setting the public IP of the WireGuard Dashboard container host. If not defined, it will trying fetching it using a curl to ifconfig.me.
|
|
if [ "${public_ip}" = "0.0.0.0" ]; then
|
|
default_ip=$(curl -s ifconfig.me)
|
|
echo "Trying to fetch the Public-IP using ifconfig.me: ${default_ip}"
|
|
|
|
sed -i "s/^remote_endpoint = .*/remote_endpoint = ${default_ip}/" /opt/wireguarddashboard/src/wg-dashboard.ini
|
|
elif [ "${public_ip}" != "$(grep "remote_endpoint = " /opt/wireguarddashboard/src/wg-dashboard.ini | awk '{print $NF}')" ]; then
|
|
echo "Setting the Public-IP using given variable: ${public_ip}"
|
|
|
|
sed -i "s/^remote_endpoint = .*/remote_endpoint = ${public_ip}/" /opt/wireguarddashboard/src/wg-dashboard.ini
|
|
fi
|
|
}
|
|
|
|
# === CLEAN UP ===
|
|
ensure_blocking() {
|
|
echo "-------------- ENSURING CONTAINER CONTINUATION -------------"
|
|
sleep 1s
|
|
echo "Ensuring container continuation."
|
|
|
|
# This function checks if the latest error log is created and tails it for docker logs uses.
|
|
if find "/opt/wireguarddashboard/src/log" -mindepth 1 -maxdepth 1 -type f | read -r; then
|
|
latestErrLog=$(find /opt/wireguarddashboard/src/log -name "error_*.log" | head -n 1)
|
|
latestAccLog=$(find /opt/wireguarddashboard/src/log -name "access_*.log" | head -n 1)
|
|
tail -f "${latestErrLog}" "${latestAccLog}"
|
|
fi
|
|
|
|
# Blocking command in case of erroring. So the container does not quit.
|
|
sleep infinity
|
|
}
|
|
|
|
# Execute functions for the WireGuard Dashboard services, then set the environment variables
|
|
clean_up
|
|
repair
|
|
start_core
|
|
set_envvars
|
|
ensure_blocking |