黑帽联盟

标题: keepalived通知脚本 [打印本页]
作者: yun    时间: 2017-6-21 20:50
标题: keepalived通知脚本
下面的脚本可以接受选项，其中：

-s, --service SERVICE,...：指定服务脚本名称，当状态切换时可自动启动、重启或关闭此服务；

-a, --address VIP: 指定相关虚拟路由器的VIP地址；

-m, --mode {mm|mb}：指定虚拟路由的模型，mm表示主主，mb表示主备；它们表示相对于同一种服务而方，其VIP的工作类型；

-n, --notify {master|backup|fault}：指定通知的类型，即vrrp角色切换的目标角色；

-h, --help：获取脚本的使用帮助；

  1. #!/bin/bash
  2. # description: An example of notify script
  3. # Usage: notify.sh -m|--mode {mm|mb} -s|--service SERVICE1,... -a|--address VIP  -n|--notify {master|backup|falut} -h|--help

  5. helpflag=0
  6. serviceflag=0
  7. modeflag=0
  8. addressflag=0
  9. notifyflag=0

  11. contact='root@localhost'

  13. Usage() {
  14.   echo "Usage: notify.sh [-m|--mode {mm|mb}] [-s|--service SERVICE1,...] <-a|--address VIP>  <-n|--notify {master|backup|falut}>"
  15.   echo "Usage: notify.sh -h|--help"
  16. }

  18. ParseOptions() {
  19.   local I=1;
  20.   if [ $# -gt 0 ]; then
  21.     while [ $I -le $# ]; do
  22.       case $1 in
  23.   -s|--service)
  24. [ $# -lt 2 ] && return 3
  25.          serviceflag=1
  26.          services=(`echo $2|awk -F"," '{for(i=1;i<=NF;i++) print $i}'`)
  27. shift 2 ;;
  28.   -h|--help)
  29.          helpflag=1
  30. return 0
  31.         shift
  32. ;;
  33.   -a|--address)
  34. [ $# -lt 2 ] && return 3
  35.     addressflag=1
  36. vip=$2
  37. shift 2
  38. ;;
  39.   -m|--mode)
  40. [ $# -lt 2 ] && return 3
  41. mode=$2
  42. shift 2
  43. ;;
  44.   -n|--notify)
  45. [ $# -lt 2 ] && return 3
  46. notifyflag=1
  47. notify=$2
  48. shift 2
  49. ;;
  50.   *)
  51. echo "Wrong options..."
  52. Usage
  53. return 7
  54. ;;
  55.        esac
  56.     done
  57.     return 0
  58.   fi
  59. }

  61. #workspace=$(dirname $0)

  63. RestartService() {
  64.   if [ ${#@} -gt 0 ]; then
  65.     for I in $@; do
  66.       if [ -x /etc/rc.d/init.d/$I ]; then
  67.         /etc/rc.d/init.d/$I restart
  68.       else
  69.         echo "$I is not a valid service..."
  70.       fi
  71.     done
  72.   fi
  73. }

  75. StopService() {
  76.   if [ ${#@} -gt 0 ]; then
  77.     for I in $@; do
  78.       if [ -x /etc/rc.d/init.d/$I ]; then
  79.         /etc/rc.d/init.d/$I stop
  80.       else
  81.         echo "$I is not a valid service..."
  82.       fi
  83.     done
  84.   fi
  85. }


  88. Notify() {
  89.     mailsubject="`hostname` to be $1: $vip floating"
  90.     mailbody="`date '+%F %H:%M:%S'`, vrrp transition, `hostname` changed to be $1."
  91.     echo $mailbody | mail -s "$mailsubject" $contact
  92. }


  95. # Main Function
  96. ParseOptions $@
  97. [ $? -ne 0 ] && Usage && exit 5

  99. [ $helpflag -eq 1 ] && Usage && exit 0

  101. if [ $addressflag -ne 1 -o $notifyflag -ne 1 ]; then
  102.   Usage
  103.   exit 2
  104. fi

  106. mode=${mode:-mb}

  108. case $notify in
  109. 'master')
  110.   if [ $serviceflag -eq 1 ]; then
  111.       RestartService ${services[*]}
  112.   fi
  113.   Notify master
  114.   ;;
  115. 'backup')
  116.   if [ $serviceflag -eq 1 ]; then
  117.     if [ "$mode" == 'mb' ]; then
  118.       StopService ${services[*]}
  119.     else
  120.       RestartService ${services[*]}
  121.     fi
  122.   fi
  123.   Notify backup
  124.   ;;
  125. 'fault')
  126.   Notify fault
  127.   ;;
  128. *)
  129.   Usage
  130.   exit 4
  131.   ;;
  132. esac
复制代码

在keepalived.conf配置文件中，其调用方法如下所示：
  1. notify_master "/etc/keepalived/notify.sh -n master -a 172.16.100.1"  
  2. notify_backup "/etc/keepalived/notify.sh -n backup -a 172.16.100.1"  
  3. notify_fault "/etc/keepalived/notify.sh -n fault -a 172.16.100.1"  
复制代码






欢迎光临 黑帽联盟 (https://bbs.cnblackhat.com/) Powered by Discuz! X2.5