chore(home): mirror from kalinamall (b0abf99) with papatramp URLs
This commit is contained in:
@@ -0,0 +1,133 @@
|
||||
#!/bin/bash
|
||||
set -Eeuo pipefail
|
||||
IFS=$'\n\t'
|
||||
|
||||
CONFIG_FILE="/etc/ssh-monitor.conf"
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
if [ -f "$SCRIPT_DIR/ssh-monitor-perms.sh" ]; then
|
||||
# shellcheck source=ssh-monitor-perms.sh
|
||||
source "$SCRIPT_DIR/ssh-monitor-perms.sh"
|
||||
fi
|
||||
WATCHDOG_LOG_FILE="/var/log/ssh_monitor_watchdog.log"
|
||||
LAST_HEARTBEAT_FILE="/var/log/last_heartbeat.txt"
|
||||
WATCHDOG_MAX_HEARTBEAT_AGE=18000
|
||||
WATCHDOG_SERVICE_NAME="ssh-monitor.service"
|
||||
WATCHDOG_NOTIFY_ON_RECOVERY="0"
|
||||
|
||||
write_log() {
|
||||
local message="$1"
|
||||
local timestamp
|
||||
timestamp="$(date '+%Y-%m-%d %H:%M:%S')"
|
||||
echo "${timestamp} - ${message}" | tee -a "$WATCHDOG_LOG_FILE" >/dev/null
|
||||
}
|
||||
|
||||
send_telegram_raw() {
|
||||
local message="$1"
|
||||
if [ -z "${TELEGRAM_BOT_TOKEN:-}" ] || [ -z "${TELEGRAM_CHAT_ID:-}" ]; then
|
||||
return 1
|
||||
fi
|
||||
|
||||
local url="https://api.telegram.org/bot${TELEGRAM_BOT_TOKEN}/sendMessage"
|
||||
local body
|
||||
if ! body=$(curl -fsS -X POST "$url" \
|
||||
--data-urlencode "chat_id=${TELEGRAM_CHAT_ID}" \
|
||||
--data-urlencode "text=${message}" \
|
||||
--data-urlencode "parse_mode=HTML" \
|
||||
--max-time 10 2>/dev/null); then
|
||||
return 1
|
||||
fi
|
||||
if echo "$body" | grep -q '"ok":true'; then
|
||||
return 0
|
||||
fi
|
||||
return 1
|
||||
}
|
||||
|
||||
notify_send() {
|
||||
local message="$1"
|
||||
message="$(ssh_monitor_message_ensure_server_line "$message")"
|
||||
send_telegram_raw "$message"
|
||||
}
|
||||
|
||||
load_config() {
|
||||
if [ -f "$CONFIG_FILE" ]; then
|
||||
local _strict_peek
|
||||
_strict_peek="$(ssh_monitor_config_peek CONFIG_STRICT_PERMS "$CONFIG_FILE" "0")"
|
||||
ssh_monitor_check_config_file_perms "$CONFIG_FILE" "$_strict_peek" || true
|
||||
ssh_monitor_config_load_file "$CONFIG_FILE" || true
|
||||
fi
|
||||
}
|
||||
|
||||
is_service_active() {
|
||||
systemctl is-active --quiet "$WATCHDOG_SERVICE_NAME"
|
||||
}
|
||||
|
||||
heartbeat_age_seconds() {
|
||||
if [ ! -f "$LAST_HEARTBEAT_FILE" ]; then
|
||||
echo 999999
|
||||
return 0
|
||||
fi
|
||||
|
||||
local hb now
|
||||
hb="$(<"$LAST_HEARTBEAT_FILE")"
|
||||
now="$(date '+%s')"
|
||||
if [[ ! "$hb" =~ ^[0-9]+$ ]]; then
|
||||
echo 999999
|
||||
return 0
|
||||
fi
|
||||
echo $((now - hb))
|
||||
}
|
||||
|
||||
restart_monitor() {
|
||||
local reason="$1"
|
||||
write_log "Watchdog: перезапуск $WATCHDOG_SERVICE_NAME, причина: $reason"
|
||||
if systemctl restart "$WATCHDOG_SERVICE_NAME"; then
|
||||
local msg=" ⚠️ WATCHDOG: монитор перезапущен "$'\n'
|
||||
msg="${msg}Сервис: ${WATCHDOG_SERVICE_NAME}"$'\n'
|
||||
msg="${msg}Причина: ${reason}"$'\n'
|
||||
msg="${msg}Время: $(date '+%d.%m.%Y %H:%M:%S')"
|
||||
notify_send "$msg"
|
||||
else
|
||||
write_log "Watchdog: ошибка перезапуска $WATCHDOG_SERVICE_NAME"
|
||||
local msg=" ❌ WATCHDOG: не удалось перезапустить монитор "$'\n'
|
||||
msg="${msg}Сервис: ${WATCHDOG_SERVICE_NAME}"$'\n'
|
||||
msg="${msg}Причина: ${reason}"$'\n'
|
||||
msg="${msg}Время: $(date '+%d.%m.%Y %H:%M:%S')"
|
||||
notify_send "$msg"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
main() {
|
||||
if [ "$EUID" -ne 0 ]; then
|
||||
echo "Запустите watchdog от root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
load_config
|
||||
mkdir -p "$(dirname "$WATCHDOG_LOG_FILE")"
|
||||
touch "$WATCHDOG_LOG_FILE"
|
||||
|
||||
if ssh_monitor_update_in_progress 2>/dev/null; then
|
||||
write_log "Watchdog: пропуск — идёт agent-update (state file)"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
local reason=""
|
||||
if ! is_service_active; then
|
||||
reason="systemd reported inactive service"
|
||||
else
|
||||
local age
|
||||
age="$(heartbeat_age_seconds)"
|
||||
if [ "$age" -ge "${WATCHDOG_MAX_HEARTBEAT_AGE:-5400}" ]; then
|
||||
reason="stale heartbeat (${age}s)"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ -n "$reason" ]; then
|
||||
restart_monitor "$reason"
|
||||
elif [ "${WATCHDOG_NOTIFY_ON_RECOVERY:-1}" = "1" ]; then
|
||||
write_log "Watchdog: OK, сервис активен и heartbeat актуален"
|
||||
fi
|
||||
}
|
||||
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user