mirror of
https://gitgud.io/fatchan/haproxy-protection.git
synced 2025-05-09 02:05:37 +00:00
readme fix, improve, remove old shit cli and interaction diagram
This commit is contained in:
Thomas Lynch
281
src/cli/ddos-cli
281
src/cli/ddos-cli
@ -1,281 +0,0 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
HAPROXY_DDOS_DOMAINS_FILE="/etc/haproxy/ddos.map"
|
||||
HAPROXY_NOCAPTCHA_DOMAINS_FILE="/etc/haproxy/no_captcha.map"
|
||||
HAPROXY_GLOBAL_ACL="hdr_cnt"
|
||||
HAPROXY_SOCKET="/var/run/haproxy.sock"
|
||||
SOCAT="$(which socat)"
|
||||
|
||||
DOMAIN_REGEX='(?=^.{5,254}$)(^(?:(?!\d+\.)[a-zA-Z0-9_\-]{1,63}\.?)+(?:[a-zA-Z]{2,})$)'
|
||||
|
||||
_h_show_acl() {
|
||||
local cmd
|
||||
if [[ ${1} ]]; then
|
||||
cmd="show acl #${1}"
|
||||
else
|
||||
cmd="show acl"
|
||||
fi
|
||||
echo "${cmd}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_h_add_acl() {
|
||||
echo "add acl #${1} ${2}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_h_del_acl() {
|
||||
echo "del acl #${1} ${2}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_h_show_map() {
|
||||
local cmd
|
||||
if [[ ${1} ]]; then
|
||||
cmd="show map #${1}"
|
||||
else
|
||||
cmd="show map"
|
||||
fi
|
||||
echo "${cmd}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_h_add_map() {
|
||||
echo "add map #${1} ${2} ${2}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_h_del_map() {
|
||||
echo "del map #${1} ${2}" | ${SOCAT} ${HAPROXY_SOCKET} stdio
|
||||
}
|
||||
|
||||
_help() {
|
||||
/bin/cat <<EOF
|
||||
Usage: $0 <command> [options]
|
||||
|
||||
Command line interface to manage per-domain and global DDoS protection.
|
||||
|
||||
optional arguments:
|
||||
-h, --help Show this help message and exit.
|
||||
|
||||
Commands:
|
||||
Global management:
|
||||
$0 global status Show status of global server ddos mode.
|
||||
$0 global enable Enable global ddos mode.
|
||||
$0 global disable Disable global ddos mode.
|
||||
|
||||
Domain management:
|
||||
$0 domain list List all domains with ddos mode on.
|
||||
$0 domain nocaptcha List all domains with nocaptcha mode on.
|
||||
$0 domain status <domain> Get ddos mode status for a domain.
|
||||
$0 domain enable <domain> Enable ddos mode for a domain.
|
||||
$0 domain disable <domain> Disable ddos mode for a domain.
|
||||
$0 domain mode <domain> Toggle nocaptcha mode for a domain.
|
||||
EOF
|
||||
}
|
||||
|
||||
_ensure_domain_passed() {
|
||||
if ! [[ ${1} ]]; then
|
||||
echo "Error: 'domain' argument is required for this action"
|
||||
_help
|
||||
exit 1
|
||||
elif ! echo "${1}" | grep -qP "${DOMAIN_REGEX}"; then
|
||||
echo "Error: '${1}' is not a valid domain"
|
||||
_help
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
_domain_list() {
|
||||
local domain_acl_id
|
||||
domain_acl_id=$(_h_show_acl | grep ${HAPROXY_DDOS_DOMAINS_FILE} | cut -d' ' -f1)
|
||||
_h_show_acl "${domain_acl_id}" | cut -d' ' -f2
|
||||
}
|
||||
|
||||
_nocaptcha_list() {
|
||||
local nocaptcha_map_id
|
||||
nocaptcha_map_id=$(_h_show_map | grep ${HAPROXY_NOCAPTCHA_DOMAINS_FILE} | cut -d' ' -f1)
|
||||
_h_show_map "${nocaptcha_map_id}" | cut -d' ' -f2
|
||||
}
|
||||
|
||||
_domain_status() {
|
||||
local ddos_domains
|
||||
local nocaptcha_domains
|
||||
local global_ddos_acl_id
|
||||
local global_ddos_status
|
||||
|
||||
ddos_domains="$(_domain_list)"
|
||||
nocaptcha_domains="$(_nocaptcha_list)"
|
||||
global_ddos_acl_id=$(_h_show_acl | grep ${HAPROXY_GLOBAL_ACL} | cut -d' ' -f1)
|
||||
global_ddos_status=$(_h_show_acl "${global_ddos_acl_id}" | cut -d' ' -f2)
|
||||
|
||||
|
||||
if echo "${ddos_domains}" | grep -q "^${1}$"; then
|
||||
echo "DDoS-protection mode is enabled for ${1}"
|
||||
else
|
||||
echo "DDoS-protection mode is disabled for ${1}"
|
||||
if [[ ${global_ddos_status} -eq 0 ]]; then
|
||||
echo "ATTENTION: DDoS-protection mode is enabled globally"
|
||||
fi
|
||||
fi
|
||||
if echo "${nocaptcha_domains}" | grep -q "^${1}$"; then
|
||||
echo "Nocaptcha mode is enabled for ${1}"
|
||||
fi
|
||||
}
|
||||
|
||||
_domain_enable() {
|
||||
local ddos_domains
|
||||
local domain_acl_id
|
||||
|
||||
ddos_domains="$(_domain_list)"
|
||||
|
||||
if echo "${ddos_domains}" | grep -q "^${1}$"; then
|
||||
echo "DDoS-protection mode is already enabled for ${1}"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
domain_acl_id=$(_h_show_acl | grep ${HAPROXY_DDOS_DOMAINS_FILE} | cut -d' ' -f1)
|
||||
_h_add_acl "${domain_acl_id}" "${1}" &>/dev/null
|
||||
|
||||
if ! grep -q "^${1}$" ${HAPROXY_DDOS_DOMAINS_FILE}; then
|
||||
echo "${1}" >> ${HAPROXY_DDOS_DOMAINS_FILE}
|
||||
fi
|
||||
echo "DDoS-protection mode was enabled for ${1}"
|
||||
}
|
||||
|
||||
_domain_changemode() {
|
||||
local nocaptcha_domains
|
||||
local domain_map_id
|
||||
|
||||
nocaptcha_domains="$(_nocaptcha_list)"
|
||||
|
||||
domain_map_id=$(_h_show_acl | grep ${HAPROXY_NOCAPTCHA_DOMAINS_FILE} | cut -d' ' -f1)
|
||||
if echo "${nocaptcha_domains}" | grep -q "^${1}$"; then
|
||||
_h_del_map "${domain_acl_id}" "${1}" &>/dev/null
|
||||
if grep -q "^${1}$" ${HAPROXY_NOCAPTCHA_DOMAINS_FILE}; then
|
||||
sed -i "/^${1}$/d" ${HAPROXY_NOCAPTCHA_DOMAINS_FILE}
|
||||
fi
|
||||
echo "Nocaptcha mode was disabled for ${1}"
|
||||
exit 0
|
||||
fi
|
||||
_h_add_map "${domain_acl_id}" "${1}" &>/dev/null
|
||||
if ! grep -q "^${1}$" ${HAPROXY_NOCAPTCHA_DOMAINS_FILE}; then
|
||||
echo "${1}" >> ${HAPROXY_NOCAPTCHA_DOMAINS_FILE}
|
||||
fi
|
||||
echo "Nocaptcha mode was enabled for ${1}"
|
||||
}
|
||||
|
||||
_domain_disable() {
|
||||
local ddos_domains
|
||||
local domain_acl_id
|
||||
|
||||
ddos_domains="$(_domain_list)"
|
||||
|
||||
if ! echo "${ddos_domains}" | grep -q "^${1}$"; then
|
||||
echo "DDoS-protection mode is already disabled for ${1}"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
domain_acl_id=$(_h_show_acl | grep ${HAPROXY_DDOS_DOMAINS_FILE} | cut -d' ' -f1)
|
||||
_h_del_acl "${domain_acl_id}" "${1}" &>/dev/null
|
||||
|
||||
if grep -q "^${1}$" ${HAPROXY_DDOS_DOMAINS_FILE}; then
|
||||
sed -i "/^${1}$/d" ${HAPROXY_DDOS_DOMAINS_FILE}
|
||||
fi
|
||||
echo "DDoS-protection mode was disabled for ${1}"
|
||||
}
|
||||
|
||||
_global_status() {
|
||||
local global_ddos_acl_id
|
||||
local global_ddos_status
|
||||
|
||||
global_ddos_acl_id=$(_h_show_acl | grep ${HAPROXY_GLOBAL_ACL} | head -1 | cut -d' ' -f1)
|
||||
global_ddos_status=$(_h_show_acl "${global_ddos_acl_id}" | cut -d' ' -f2)
|
||||
|
||||
if [[ ${global_ddos_status} -eq 0 ]]; then
|
||||
echo "DDoS-protection mode is enabled globally"
|
||||
else
|
||||
echo "DDoS-protection mode is disabled globally"
|
||||
fi
|
||||
}
|
||||
|
||||
_global_enable() {
|
||||
declare -a global_ddos_acl_ids
|
||||
local global_ddos_status
|
||||
|
||||
global_ddos_acl_ids=($(_h_show_acl | grep ${HAPROXY_GLOBAL_ACL} | cut -d' ' -f1))
|
||||
global_ddos_status=$(_h_show_acl "${global_ddos_acl_ids[0]}" | cut -d' ' -f2)
|
||||
|
||||
if [[ ${global_ddos_status} -eq 0 ]]; then
|
||||
echo "DDoS-protection mode is already enabled globally"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
for id in "${global_ddos_acl_ids[@]}"; do
|
||||
_h_add_acl "${id}" 0 &>/dev/null
|
||||
_h_del_acl "${id}" 1 &>/dev/null
|
||||
done
|
||||
echo "DDoS-protection mode was enabled globally"
|
||||
}
|
||||
|
||||
_global_disable() {
|
||||
declare -a global_ddos_acl_ids
|
||||
local global_ddos_status
|
||||
|
||||
global_ddos_acl_ids=($(_h_show_acl | grep ${HAPROXY_GLOBAL_ACL} | cut -d' ' -f1))
|
||||
global_ddos_status=$(_h_show_acl "${global_ddos_acl_ids[0]}" | cut -d' ' -f2)
|
||||
|
||||
if [[ ${global_ddos_status} -eq 1 ]]; then
|
||||
echo "DDoS-protection mode is already disabled globally"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
for id in "${global_ddos_acl_ids[@]}"; do
|
||||
_h_add_acl "${id}" 1 &>/dev/null
|
||||
_h_del_acl "${id}" 0 &>/dev/null
|
||||
done
|
||||
echo "DDoS-protection mode was disabled globally"
|
||||
}
|
||||
|
||||
_handle_global_management() {
|
||||
case ${1} in
|
||||
status) _global_status;;
|
||||
enable) _global_enable;;
|
||||
disable) _global_disable;;
|
||||
*) _help; exit 1;;
|
||||
esac
|
||||
}
|
||||
|
||||
_handle_domain_management() {
|
||||
case ${1} in
|
||||
list) _domain_list;;
|
||||
nocaptcha) _nocaptcha_list;;
|
||||
status)
|
||||
_ensure_domain_passed "${2}"
|
||||
_domain_status "${2}";;
|
||||
enable)
|
||||
_ensure_domain_passed "${2}"
|
||||
_domain_enable "${2}";;
|
||||
disable)
|
||||
_ensure_domain_passed "${2}"
|
||||
_domain_disable "${2}";;
|
||||
mode)
|
||||
_ensure_domain_passed "${2}"
|
||||
_domain_changemode "${2}";;
|
||||
*) _help; exit 1;;
|
||||
esac
|
||||
}
|
||||
|
||||
if ! [[ "${*}" ]]; then
|
||||
_help
|
||||
exit 1
|
||||
fi
|
||||
|
||||
for i in "${@}"; do
|
||||
case ${i} in
|
||||
-h|--help) _help; exit 0;;
|
||||
domain) MODE=DOMAIN; shift; break;;
|
||||
global) MODE=GLOBAL; shift; break;;
|
||||
*) _help; exit 1;;
|
||||
esac
|
||||
done
|
||||
|
||||
case ${MODE} in
|
||||
DOMAIN) _handle_domain_management "${@}";;
|
||||
GLOBAL) _handle_global_management "${@}";;
|
||||
esac
|
||||
Reference in New Issue
Block a user