From 838a708e8f0ed659cbd141ac4b71777c0d0e9d3d Mon Sep 17 00:00:00 2001
From: Dan Kelleher <dkelleher@g.harvard.edu>
Date: Tue, 27 Mar 2018 16:44:33 -0400
Subject: [PATCH] Add functionality for keepalived check script

This will add a hook to allow keepalived to run a check script if the
script variable is defined.
---
 src/etc/one-context.d/loc-15-keepalived##apk | 46 +++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/src/etc/one-context.d/loc-15-keepalived##apk b/src/etc/one-context.d/loc-15-keepalived##apk
index e63b950..d75b92e 100755
--- a/src/etc/one-context.d/loc-15-keepalived##apk
+++ b/src/etc/one-context.d/loc-15-keepalived##apk
@@ -53,6 +53,12 @@ get_vrouter_password() {
     echo $password
 }
 
+get_check_script() {
+    script="$VROUTER_KEEPALIVED_CHECK_SCRIPT"
+
+    echo $script
+}
+
 get_iface_var() {
     var_name="$1_$2"
     var=$(eval "echo \"\${$var_name}\"")
@@ -81,6 +87,22 @@ EOT
 EOT
 }
 
+gen_check_script() {
+    script="$(get_check_script)"
+
+    if [ -n "$script" ]; then
+        cat <<EOT
+vrrp_script chk_script {
+  script "$script"
+  interval 2
+  fall 2
+  rise 2
+}
+
+EOT
+    fi
+}
+
 gen_auth() {
     password=$(get_vrouter_password)
 
@@ -94,6 +116,26 @@ EOT
     fi
 }
 
+gen_track() {
+    script="$(get_check_script)"
+
+    if [ -n "$script" ]; then
+        cat <<EOT
+track_script {
+    chk_script
+  }
+EOT
+    fi
+}
+
+gen_preempt() {
+    script="$(get_check_script)"
+
+    if [ -z "$script" ]; then
+        echo nopreempt
+    fi
+}
+
 gen_instances() {
     for interface in $(get_context_interfaces); do
         vrouter_ip=$(get_iface_var $interface VROUTER_IP)
@@ -111,7 +153,8 @@ vrrp_instance $interface {
     $vrouter_ip
   }
   $(gen_auth)
-  nopreempt
+  $(gen_track)
+  $(gen_preempt)
 }
 
 EOT
@@ -133,6 +176,7 @@ fi
 (
 
 gen_group
+gen_check_script
 gen_instances
 
 ) > /etc/keepalived/keepalived.conf