站点图标 嘟嘟社区

[美国VPS] 3年阿里云白**的那个脚本，在mjj的基础上完善了一下

moe

  1. #!/usr/bin/env bash
  3. # This script is called on SSH login by /etc/profile.d/sshd_telegram.sh
  4. # Modified from https://github.com/MyTheValentinus/ssh-login-alert-telegram
  6. # Your USERID or Channel ID to display alert and key, we recommend you create new bot with @BotFather on Telegram
  7. #你要修改的都在这里USERID,KEY,VPSNAME,PFTIME,LIMIT,LIMIT2
  8. #========================================================
  9. USERID=(这里也要改电报机器人id)
  10. KEY="填电报机器人key"
  11. # 设置机器名字
  12. VPSNAME="ali-hk1"
  13. # 设置流量限制（单位：GB）
  14. LIMIT=150
  15. LIMIT2=160
  16. # 设置间隔时间（单位：秒）
  17. PFTIME=1800
  18. #=========================================================
  19. for i in "${USERID[@]}"
  20. do
  21. URL="https://api.telegram.org/bot${KEY}/sendMessage"
  22. DATE="$(date "+%Y-%m-%d %H:%M:%S")"
  24. # 设置网卡名称
  25. INTERFACE="eth0"
  27. SRV_HOSTNAME=$(hostname -f)
  30. # 获取当前流量（单位：KB）====================================
  31. VNSTAT_JSON=$(vnstat -i $INTERFACE –json)
  34. # 使用 jq 解析 JSON 数据获取接收和发送的流量（单位：KB）
  35. RX=$(echo $VNSTAT_JSON | jq -r ‘.interfaces[0].traffic.total.rx’)
  36. TX=$(echo $VNSTAT_JSON | jq -r ‘.interfaces[0].traffic.total.tx’)
  39. # 检查 RX 和 TX 是否为有效的数字
  40. if ! [[ $RX =~ ^[0-9]+$ ]] || ! [[ $TX =~ ^[0-9]+$ ]]; then
  41.     exit 1
  42. fi
  44. # 计算总流量（单位：GB）
  45. TOTAL=$(echo "scale=2; ($RX + $TX) / 1024 / 1024" | bc)
  46. RX_GB=$(echo "scale=2; $RX / 1024 / 1024" | bc)
  47. TX_GB=$(echo "scale=2; $TX / 1024 / 1024" | bc)
  49. # 获取上次运行时间==========================================================
  51. # 当前时间（秒）
  52. current_time=$(date +%s)
  54. # 默认上次执行时间为很久之前（这样首次运行时会执行else分支）
  55. last_exec_time=0
  57. # 时间戳文件路径
  58. timestamp_file="/usr/unitls/else_exec_time.txt"
  60. # 检查时间戳文件是否存在
  61. if [ -f "$timestamp_file" ]; then
  62.     # 读取上次执行时间
  63.     last_exec_time=$(cat "$timestamp_file")
  64. else
  65.     touch /usr/unitls/else_exec_time.txt
  66.     echo "else_exec_time.txt 文件已经创建."  # 可选：输出提示信息
  67. fi
  69. # 计算时间差值
  70. time_diff=$((current_time – last_exec_time))
  72. #判断执行语句==============================================================
  73. if (( $(echo "$RX_GB >= $LIMIT2" | bc -l) )) || (( $(echo "$TX_GB >= $LIMIT2" | bc -l) )); then
  74.        
  75.         TEXT="${VPSNAME}(${SRV_HOSTNAME})当前流量使用情况:
  76.         入流量（接受流量）: *${RX_GB}*
  77.         出流量（发送流量）: *${TX_GB}*
  78.     时间: ${DATE}
  79.     其已超过160GB，已执行关机操作"
  81.         curl -s -d "chat_id=$i&text=${TEXT}&disable_web_page_preview=true&parse_mode=markdown" $URL > /dev/null
  82.    
  83.     sudo shutdown -h now
  85. elif (( $(echo "$RX_GB >= $LIMIT" | bc -l) )) || (( $(echo "$TX_GB >= $LIMIT" | bc -l) )); then
  86.        
  87.         TEXT="${VPSNAME}(${SRV_HOSTNAME})当前流量使用情况:
  88.         入流量（接受流量）: *${RX_GB}*
  89.         出流量（发送流量）: *${TX_GB}*
  90.     时间: ${DATE}
  91.     其已超过150GB，超过160GB将执行关机操作"
  93.         curl -s -d "chat_id=$i&text=${TEXT}&disable_web_page_preview=true&parse_mode=markdown" $URL > /dev/null
  95. else
  96.     if (( time_diff >= PFTIME )); then
  97.      # 记录操作时间
  98.     echo "$current_time" > "$timestamp_file"
  100.     TEXT="${VPSNAME}(${SRV_HOSTNAME})当前流量使用情况:
  101.         入流量（接受流量）: *${RX_GB}*
  102.         出流量（发送流量）: *${TX_GB}*
  103.     时间: ${DATE}
  104.     正常使用暂未超过150GB"
  106.         curl -s -d "chat_id=$i&text=${TEXT}&disable_web_page_preview=true&parse_mode=markdown" $URL > /dev/null
  107.     fi   
  108. fi
  109. done

复制代码

出入流量中的一个超过160就会关机并且给电报机器人发送通知。脚本每分钟执行一次。每次执行都会判定出入流量是否超标，没有超标的话，每30分钟向电报机器人发送截止当前时刻的出入流量。
前提是安装vnstat 和 jq  和bc。
vnstat修改了设置每分钟记录一次
给老哥点个赞
原贴是什么
退出移动版