chore(shell): clean up shell script formatting and structure
This commit is contained in:
@ -2,12 +2,10 @@
|
||||
|
||||
source /etc/hysteria/core/scripts/path.sh
|
||||
|
||||
# --- Configuration ---
|
||||
SERVICE_NAME="hysteria-ip-limit.service"
|
||||
DB_NAME="blitz_panel"
|
||||
CONNECTIONS_COLLECTION="active_connections"
|
||||
|
||||
# Load configurations from .configs.env
|
||||
if [ -f "$CONFIG_ENV" ]; then
|
||||
source "$CONFIG_ENV"
|
||||
BLOCK_DURATION="${BLOCK_DURATION:-60}" # Default to 60 seconds
|
||||
@ -19,17 +17,14 @@ else
|
||||
echo -e "BLOCK_DURATION=240\nMAX_IPS=5" > "$CONFIG_ENV"
|
||||
fi
|
||||
|
||||
# --- Ensure files exist ---
|
||||
[ ! -f "$BLOCK_LIST" ] && touch "$BLOCK_LIST"
|
||||
|
||||
# --- Logging function ---
|
||||
log_message() {
|
||||
local level="$1"
|
||||
local message="$2"
|
||||
echo "[$(date +"%Y-%m-%d %H:%M:%S")] [$level] $message"
|
||||
}
|
||||
|
||||
# --- Add an IP to the database for a user ---
|
||||
add_ip_to_db() {
|
||||
local username="$1"
|
||||
local ip_address="$2"
|
||||
@ -44,7 +39,6 @@ add_ip_to_db() {
|
||||
log_message "INFO" "DB Update: Added $ip_address for user $username"
|
||||
}
|
||||
|
||||
# --- Remove an IP from the database for a user ---
|
||||
remove_ip_from_db() {
|
||||
local username="$1"
|
||||
local ip_address="$2"
|
||||
@ -61,7 +55,6 @@ remove_ip_from_db() {
|
||||
log_message "INFO" "DB Update: Removed $ip_address for user $username"
|
||||
}
|
||||
|
||||
# --- Block an IP using iptables and track it ---
|
||||
block_ip() {
|
||||
local ip_address="$1"
|
||||
local username="$2"
|
||||
@ -77,7 +70,6 @@ block_ip() {
|
||||
log_message "WARN" "Blocked IP $ip_address for user $username for $BLOCK_DURATION seconds"
|
||||
}
|
||||
|
||||
# --- Explicitly unblock an IP using iptables ---
|
||||
unblock_ip() {
|
||||
local ip_address="$1"
|
||||
|
||||
@ -88,7 +80,6 @@ unblock_ip() {
|
||||
sed -i "/$ip_address,/d" "$BLOCK_LIST"
|
||||
}
|
||||
|
||||
# --- Block all IPs for a user ---
|
||||
block_all_user_ips() {
|
||||
local username="$1"
|
||||
|
||||
@ -114,7 +105,6 @@ block_all_user_ips() {
|
||||
log_message "WARN" "User $username has been completely blocked for $BLOCK_DURATION seconds"
|
||||
}
|
||||
|
||||
# --- Check for and unblock expired IPs ---
|
||||
check_expired_blocks() {
|
||||
local current_time=$(date +%s)
|
||||
local ip username expiry
|
||||
@ -129,7 +119,6 @@ check_expired_blocks() {
|
||||
done < "$BLOCK_LIST"
|
||||
}
|
||||
|
||||
# --- Check if a user has exceeded the IP limit ---
|
||||
check_ip_limit() {
|
||||
local username="$1"
|
||||
|
||||
@ -154,7 +143,6 @@ check_ip_limit() {
|
||||
fi
|
||||
}
|
||||
|
||||
# --- Parse log lines for connections and disconnections ---
|
||||
parse_log_line() {
|
||||
local log_line="$1"
|
||||
local ip_address
|
||||
@ -180,7 +168,6 @@ parse_log_line() {
|
||||
fi
|
||||
}
|
||||
|
||||
# --- Install Systemd Service ---
|
||||
install_service() {
|
||||
cat <<EOF > /etc/systemd/system/${SERVICE_NAME}
|
||||
[Unit]
|
||||
@ -204,7 +191,6 @@ EOF
|
||||
log_message "INFO" "IP Limiter service started"
|
||||
}
|
||||
|
||||
# --- Uninstall Systemd Service ---
|
||||
uninstall_service() {
|
||||
systemctl stop ${SERVICE_NAME} 2>/dev/null
|
||||
systemctl disable ${SERVICE_NAME} 2>/dev/null
|
||||
@ -213,7 +199,6 @@ uninstall_service() {
|
||||
log_message "INFO" "IP Limiter service stopped and removed"
|
||||
}
|
||||
|
||||
# --- Change Configuration ---
|
||||
change_config() {
|
||||
local new_block_duration="$1"
|
||||
local new_max_ips="$2"
|
||||
@ -244,7 +229,6 @@ change_config() {
|
||||
fi
|
||||
}
|
||||
|
||||
# --- Startup Checks ---
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo "Error: This script must be run as root."
|
||||
exit 1
|
||||
@ -257,7 +241,6 @@ if ! command -v jq &>/dev/null; then
|
||||
log_message "WARN" "'jq' is not installed. JSON parsing for blocking might fail."
|
||||
fi
|
||||
|
||||
# --- Command execution ---
|
||||
case "$1" in
|
||||
start)
|
||||
install_service
|
||||
|
||||
@ -1,6 +1,5 @@
|
||||
CLI_PATH="/etc/hysteria/core/cli.py"
|
||||
USERS_FILE="/etc/hysteria/users.json"
|
||||
TRAFFIC_FILE="/etc/hysteria/traffic_data.json"
|
||||
CONFIG_FILE="/etc/hysteria/config.json"
|
||||
CONFIG_ENV="/etc/hysteria/.configs.env"
|
||||
TELEGRAM_ENV="/etc/hysteria/core/scripts/telegrambot/.env"
|
||||
|
||||
@ -9,7 +9,6 @@ declare -a services=(
|
||||
"hysteria-telegram-bot.service"
|
||||
"hysteria-normal-sub.service"
|
||||
"hysteria-caddy-normalsub.service"
|
||||
# "hysteria-singbox.service"
|
||||
"hysteria-ip-limit.service"
|
||||
"wg-quick@wgcf.service"
|
||||
)
|
||||
|
||||
@ -1,7 +1,5 @@
|
||||
source /etc/hysteria/core/scripts/path.sh
|
||||
# source /etc/hysteria/core/scripts/services_status.sh
|
||||
|
||||
# Function to define colors
|
||||
define_colors() {
|
||||
green='\033[0;32m'
|
||||
cyan='\033[0;36m'
|
||||
@ -81,7 +79,6 @@ load_hysteria2_ips() {
|
||||
IP6=$(grep -E "^IP6=" "$CONFIG_ENV" | cut -d '=' -f 2)
|
||||
|
||||
if [[ -z "$IP4" || -z "$IP6" ]]; then
|
||||
# echo "Warning: IP4 or IP6 is not set in configs.env. Fetching from system..."
|
||||
default_interface=$(ip route | grep default | awk '{print $5}')
|
||||
|
||||
if [ -n "$default_interface" ]; then
|
||||
@ -90,7 +87,6 @@ load_hysteria2_ips() {
|
||||
if [ -n "$system_IP4" ]; then
|
||||
IP4="$system_IP4"
|
||||
else
|
||||
# echo "Attempting to fetch IPv4 from external service..."
|
||||
system_IP4=$(curl -s -4 ip.sb)
|
||||
[ -n "$system_IP4" ] && IP4="$system_IP4" || IP4="None"
|
||||
fi
|
||||
@ -101,13 +97,11 @@ load_hysteria2_ips() {
|
||||
if [ -n "$system_IP6" ]; then
|
||||
IP6="$system_IP6"
|
||||
else
|
||||
# echo "Attempting to fetch IPv6 from external service..."
|
||||
system_IP6=$(curl -s -6 ip.sb)
|
||||
[ -n "$system_IP6" ] && IP6="$system_IP6" || IP6="None"
|
||||
fi
|
||||
fi
|
||||
else
|
||||
# echo "Warning: Could not determine default interface, trying external services..."
|
||||
if [ -z "$IP4" ]; then
|
||||
system_IP4=$(curl -s -4 ip.sb)
|
||||
[ -n "$system_IP4" ] && IP4="$system_IP4" || IP4="None"
|
||||
@ -164,19 +158,3 @@ load_hysteria2_ips() {
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
|
||||
# check_services() {
|
||||
# # source /etc/hysteria/core/scripts/services_status.sh
|
||||
# for service in "${services[@]}"; do
|
||||
# service_base_name=$(basename "$service" .service)
|
||||
|
||||
# display_name=$(echo "$service_base_name" | sed -E 's/([^-]+)-?/\u\1/g')
|
||||
|
||||
# if systemctl is-active --quiet "$service"; then
|
||||
# echo -e "${NC}${display_name}:${green} Active${NC}"
|
||||
# else
|
||||
# echo -e "${NC}${display_name}:${red} Inactive${NC}"
|
||||
# fi
|
||||
# done
|
||||
# }
|
||||
|
||||
Reference in New Issue
Block a user