From b588994d23931df125f1e9486d43f64c290d7a9e Mon Sep 17 00:00:00 2001 From: xzx3344521 Date: Tue, 21 Oct 2025 22:47:48 +0800 Subject: [PATCH] =?UTF-8?q?Update=20=E5=AE=9E=E6=97=B6=20history=20?= =?UTF-8?q?=E7=9B=91=E6=8E=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 实时 history 监控 | 187 ++++++++++++++++++++++++++++++++++++++-------- 1 file changed, 156 insertions(+), 31 deletions(-) diff --git a/实时 history 监控 b/实时 history 监控 index 7b1f473..51424b2 100644 --- a/实时 history 监控 +++ b/实时 history 监控 @@ -1,20 +1,25 @@ #!/bin/bash -# 优化版实时命令监控脚本 -# 作者: 系统安全监控 -# 版本: 2.0 +# 优化版实时命令监控脚本 - 带IP地理位置查询 +# 版本: 2.1 -set -e # 遇到错误立即退出 +set -e -### 配置区域 - 可根据需要修改 ### -LOG_DIR="/root/command_monitor_logs" # 日志目录 -MAX_LOG_SIZE="100M" # 单个日志文件最大大小 -MAX_LOG_FILES=10 # 最大日志文件数量 -MEMORY_LIMIT="512M" # 内存使用限制 -CPU_LIMIT=90 # CPU使用率限制% -CHECK_INTERVAL=300 # 系统检查间隔(秒) -BACKUP_DAYS=7 # 日志保留天数 -CLEANUP_INTERVAL=3600 # 清理间隔(秒) +### 配置区域 ### +LOG_DIR="/root/command_monitor_logs" +MAX_LOG_SIZE="10M" +MAX_LOG_FILES=10 +MEMORY_LIMIT="512M" +CPU_LIMIT=90 +CHECK_INTERVAL=300 +BACKUP_DAYS=7 +CLEANUP_INTERVAL=3600 + +### IP地理位置配置 ### +IP_API_SERVICE="ipapi" # 可选: ipapi, ipapi.co, ipinfo.io, 本地 +CACHE_IP_INFO=true # 缓存IP信息避免重复查询 +IP_CACHE_FILE="/tmp/ip_geo_cache.txt" +CACHE_EXPIRE=3600 # 缓存过期时间(秒) ### 颜色定义 ### RED='\033[0;31m' @@ -29,6 +34,8 @@ NC='\033[0m' SCRIPT_PID=$$ MONITOR_PID="" LAST_CLEANUP=0 +CURRENT_LOG="" +LATEST_LOG="" # 获取时间戳 timestamp() { @@ -46,21 +53,114 @@ log_message() { "WARN") color="$YELLOW" ;; "INFO") color="$BLUE" ;; "SUCCESS") color="$GREEN" ;; + "COMMAND") color="$CYAN" ;; esac echo -e "${color}[$(timestamp)] [$level] $message${NC}" | tee -a "$CURRENT_LOG" } +# IP地理位置查询函数 +get_ip_location() { + local ip="$1" + + # 检查缓存 + if [ "$CACHE_IP_INFO" = true ] && [ -f "$IP_CACHE_FILE" ]; then + local cached_info=$(grep "^$ip|" "$IP_CACHE_FILE" | head -1) + if [ -n "$cached_info" ]; then + local cache_time=$(echo "$cached_info" | cut -d'|' -f2) + local current_time=$(date +%s) + if [ $((current_time - cache_time)) -lt $CACHE_EXPIRE ]; then + echo "$cached_info" | cut -d'|' -f3- + return 0 + fi + fi + fi + + local location_info="" + + # 使用多个API服务，优先使用免费的 + for api in $IP_API_SERVICE ipapi.co ipinfo.io; do + case $api in + ipapi) + location_info=$(curl -s -m 5 "http://ip-api.com/json/$ip" | \ + jq -r '[.country, .regionName, .city, .isp] | join(", ")' 2>/dev/null || echo "") + ;; + ipapi.co) + location_info=$(curl -s -m 5 "https://ipapi.co/$ip/json/" | \ + jq -r '[.country_name, .region, .city, .org] | join(", ")' 2>/dev/null || echo "") + ;; + ipinfo.io) + location_info=$(curl -s -m 5 "https://ipinfo.io/$ip" | \ + jq -r '[.country, .region, .city, .org] | join(", ")' 2>/dev/null || echo "") + ;; + esac + + if [ -n "$location_info" ] && [ "$location_info" != "null" ]; then + break + fi + done + + # 如果API查询失败，使用本地IP库或返回简单信息 + if [ -z "$location_info" ] || [ "$location_info" = "null" ]; then + location_info="未知位置" + fi + + # 缓存结果 + if [ "$CACHE_IP_INFO" = true ]; then + echo "$ip|$(date +%s)|$location_info" >> "$IP_CACHE_FILE" + fi + + echo "$location_info" +} + +# 简化版地理位置查询（避免依赖外部工具） +get_simple_ip_location() { + local ip="$1" + + # 检查常见IP段 + if [[ "$ip" == "192.168."* ]] || [[ "$ip" == "10."* ]] || [[ "$ip" == "172."* ]]; then + echo "内网IP" + return + fi + + if [[ "$ip" == "127.0.0.1" ]] || [[ "$ip" == "localhost" ]]; then + echo "本机" + return + fi + + # 使用whois查询（如果可用） + if command -v whois &> /dev/null; then + local whois_info=$(timeout 5 whois "$ip" 2>/dev/null | grep -iE "country:|descr:" | head -2 | \ + awk -F: '{print $2}' | sed 's/^ *//' | tr '\n' ',' | sed 's/,$//') + if [ -n "$whois_info" ]; then + echo "$whois_info" + return + fi + fi + + # 使用ping方式获取粗略位置（通过TTL判断） + local ttl=$(timeout 3 ping -c 1 "$ip" 2>/dev/null | grep "ttl=" | sed 's/.*ttl=\([0-9]*\).*/\1/' || echo "") + if [ -n "$ttl" ]; then + if [ "$ttl" -le 64 ]; then + echo "Linux系统 - 可能较近" + elif [ "$ttl" -le 128 ]; then + echo "Windows系统 - 中等距离" + else + echo "远程主机 - 可能较远" + fi + else + echo "位置未知" + fi +} + # 获取客户端IP get_client_ip() { local ip="unknown" - # 尝试多种方式获取IP if [ -n "$SSH_CLIENT" ]; then ip=$(echo "$SSH_CLIENT" | awk '{print $1}') elif [ -n "$SSH_CONNECTION" ]; then ip=$(echo "$SSH_CONNECTION" | awk '{print $1}') else - # 获取当前连接的IP ip=$(who -m | awk '{print $5}' | sed 's/[()]//g' | head -1) if [[ "$ip" == ":0" ]] || [[ "$ip" == ":1" ]]; then ip="localhost" @@ -69,6 +169,25 @@ get_client_ip() { echo "$ip" } +# 获取带地理位置的IP信息 +get_ip_with_location() { + local ip="$1" + local location="" + + # 先尝试完整查询 + if command -v curl &> /dev/null && command -v jq &> /dev/null; then + location=$(get_ip_location "$ip") + else + location=$(get_simple_ip_location "$ip") + fi + + if [ -n "$location" ] && [ "$location" != "未知位置" ]; then + echo "$ip [$location]" + else + echo "$ip" + fi +} + # 初始化日志系统 init_log_system() { mkdir -p "$LOG_DIR" @@ -77,13 +196,19 @@ init_log_system() { local log_date=$(date '+%Y%m%d_%H%M%S') CURRENT_LOG="$LOG_DIR/command_monitor_${client_ip}_${log_date}.log" - # 创建软链接指向最新日志 LATEST_LOG="$LOG_DIR/latest.log" ln -sf "$CURRENT_LOG" "$LATEST_LOG" log_message "INFO" "监控脚本启动 - PID: $$" log_message "INFO" "客户端IP: $(get_client_ip)" log_message "INFO" "日志文件: $CURRENT_LOG" + + # 显示IP地理位置 + local client_ip_original=$(get_client_ip) + if [ "$client_ip_original" != "unknown" ] && [ "$client_ip_original" != "localhost" ]; then + local ip_location=$(get_ip_with_location "$client_ip_original") + log_message "INFO" "地理位置: $ip_location" + fi } # 资源监控函数 @@ -181,12 +306,10 @@ log_rotation() { cleanup_on_exit() { log_message "INFO" "正在停止监控服务..." - # 杀死所有子进程 if [ -n "$MONITOR_PID" ]; then kill "$MONITOR_PID" 2>/dev/null || true fi - # 杀死资源监控进程 pkill -f "monitor_resources" 2>/dev/null || true log_message "SUCCESS" "监控服务已停止" @@ -212,7 +335,6 @@ export HISTTIMEFORMAT='%F %T ' export HISTSIZE=10000 export HISTFILESIZE=20000 shopt -s histappend -# 记录所有命令，包括时间戳 export HISTCONTROL=ignoredups:erasedups EOF @@ -221,7 +343,7 @@ EOF log_message "INFO" "用户 $user 已配置实时记录" fi fi - fi + done done } @@ -229,7 +351,6 @@ EOF start_main_monitor() { log_message "INFO" "启动主监控进程..." - # 声明关联数组来跟踪每个用户的文件状态 declare -A file_sizes declare -A last_commands @@ -260,11 +381,9 @@ start_main_monitor() { local last_size=${file_sizes["$user"]} if [ "$current_size" -gt "$last_size" ]; then - # 读取新内容 local new_content=$(tail -c +$((last_size + 1)) "$history_file" 2>/dev/null | tr -d '\0') if [ -n "$new_content" ]; then - # 逐行处理新命令 while IFS= read -r line; do line=$(echo "$line" | sed 's/^[ \t]*//;s/[ \t]*$//') @@ -273,7 +392,10 @@ start_main_monitor() { [ "$line" != "${last_commands["$user"]}" ] && \ [[ ! "$line" =~ ^(ls|cd|pwd|ll|la|history|exit|clear)$ ]]; then - log_message "COMMAND" "用户: $user | 命令: $line | 来源: $(get_client_ip)" + local client_ip=$(get_client_ip) + local ip_with_location=$(get_ip_with_location "$client_ip") + + log_message "COMMAND" "用户: $user | 命令: $line | 来源: $ip_with_location" last_commands["$user"]="$line" fi done <<< "$new_content" @@ -292,11 +414,9 @@ start_main_monitor() { start_background_monitor() { log_message "INFO" "启动后台监控服务..." - # 启动资源监控 monitor_resources & local resource_pid=$! - # 启动主监控 start_main_monitor & MONITOR_PID=$! @@ -305,13 +425,12 @@ start_background_monitor() { log_message "INFO" "资源监控PID: $resource_pid" log_message "INFO" "查看实时日志: tail -f $LATEST_LOG" - # 等待子进程 wait $MONITOR_PID } # 显示使用说明 show_usage() { - echo -e "${GREEN}实时命令监控系统 v2.0${NC}" + echo -e "${GREEN}实时命令监控系统 v2.1${NC}" echo "用法: $0 [选项]" echo echo "选项:" @@ -341,7 +460,6 @@ check_monitor_status() { echo "日志目录: $LOG_DIR" echo "最新日志: $LATEST_LOG" - # 显示最近的活动 if [ -f "$LATEST_LOG" ]; then echo echo -e "${YELLOW}最近10条记录:${NC}" @@ -362,7 +480,6 @@ stop_monitor() { kill $pids 2>/dev/null || true sleep 2 - # 检查是否停止成功 if pgrep -f "command_monitor" >/dev/null; then echo -e "${RED}强制停止监控进程...${NC}" kill -9 $pids 2>/dev/null || true @@ -373,7 +490,15 @@ stop_monitor() { ### 主程序 ### main() { - # 参数处理 + # 检查依赖 + if ! command -v curl &> /dev/null; then + echo -e "${YELLOW}警告: 未找到 curl，地理位置查询功能受限${NC}" + fi + + if ! command -v jq &> /dev/null; then + echo -e "${YELLOW}警告: 未找到 jq，使用简化版地理位置查询${NC}" + fi + case "${1:-}" in -d|--daemon) init_log_system