added configuration hook to execute script on interface state change. configurable via conf mode. environment variables identify interface and new state.

1 files changed, 23 insertions, 1 deletions

diff --git a/src/lbdecision.cc b/src/lbdecision.cc
index a26476b..c6107d5 100644
--- a/src/lbdecision.cc
+++ b/src/lbdecision.cc
@@ -229,9 +229,31 @@ LBDecision::run(LBData &lb_data)
   update_paths(lb_data);
 
   //first determine if we need to alter the rule set
-  if (!lb_data.state_changed()) {
+  map<string,string> state_changed_coll;
+  state_changed_coll = lb_data.state_changed();
+  if (state_changed_coll.empty() == true) {
     return;
   }
+  else { 
+    //state has changed execute script now
+    
+    map<string,string>::iterator iter = state_changed_coll.begin();
+    while (iter != state_changed_coll.end()) {
+      //set state
+      //set interface
+      setenv("WLB_INTERFACE_NAME",iter->first.c_str(),1);
+      setenv("WLB_INTERFACE_STATE",iter->second.c_str(),1);
+
+      syslog(LOG_WARNING, "executing script: %s",lb_data._hook.c_str());
+
+      execute(lb_data._hook, stdout);
+      //unset state
+      //unset interface
+      unsetenv("WLB_INTERFACE_NAME");
+      unsetenv("WLB_INTERFACE_STATE");
+      ++iter;
+    }
+  }
 
   //then if we do, flush all
   execute("iptables -t mangle -F PREROUTING", stdout);