initial check wan lbdebian/0.1

Signed-off-by: Michael Larson <slioch@eng-140.vyatta.com>

1 files changed, 212 insertions, 0 deletions

diff --git a/src/lbdata.cc b/src/lbdata.cc
new file mode 100644
index 0000000..14a04c0
--- /dev/null
+++ b/src/lbdata.cc
@@ -0,0 +1,212 @@
+/*
+ * Module: lbdata.cc
+ *
+ * **** License ****
+ * Version: VPL 1.0
+ *
+ * The contents of this file are subject to the Vyatta Public License
+ * Version 1.0 ("License"); you may not use this file except in
+ * compliance with the License. You may obtain a copy of the License at
+ * http://www.vyatta.com/vpl
+ *
+ * Software distributed under the License is distributed on an "AS IS"
+ * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
+ * the License for the specific language governing rights and limitations
+ * under the License.
+ *
+ * This code was originally developed by Vyatta, Inc.
+ * Portions created by Vyatta are Copyright (C) 2007 Vyatta, Inc.
+ * All Rights Reserved.
+ *
+ * Author: Michael Larson
+ * Date: 2007
+ * Description:
+ *
+ * **** End License ****
+ *
+ */
+#include <sys/time.h>
+#include <time.h>
+#include <iostream>
+
+#include "lbdata.hh"
+
+int LBHealthHistory::_buffer_size = 10;
+
+/**
+ *
+ *
+ **/
+void
+LBHealth::put(int rtt)
+{
+  int activity_ct = _hresults.push(rtt);
+
+  if (rtt == -1) {
+    if (activity_ct >= _failure_ct) {
+      if (_is_active == true) {
+	_state_changed = true;
+      }
+      _is_active = false;
+    }
+  }
+  else {
+    if (activity_ct >= _success_ct) {
+      if (_is_active == false) {
+	_state_changed = true;
+      }
+      _is_active = true;
+    }
+  }
+}
+
+/**
+ *
+ *
+ **/
+LBHealthHistory::LBHealthHistory(int buffer_size) :
+  _last_success(0),
+  _last_failure(0),
+  _index(0)
+{
+  _resp_data.resize(10);
+
+  for (int i = 0; i < _buffer_size; ++i) {
+    _resp_data[i] = 0;
+  }
+}
+
+
+
+/**
+ *
+ *
+ **/
+int
+LBHealthHistory::push(int rtt)
+{
+  struct timeval tv;
+  gettimeofday(&tv,NULL);
+
+  if (rtt == -1) {
+    _last_failure = tv.tv_sec;
+  }
+  else {
+    _last_success = tv.tv_sec;
+  }
+
+  _resp_data[_index % _buffer_size] = rtt;
+  ++_index;
+
+  //compute count of sequence of same responses
+  int ct = 0;
+  int start_index = (_index - 1) % _buffer_size;
+  for (int i = 0; i < _buffer_size; ++i) {
+    int index = start_index - i;
+    if (index < 0) {
+      //handle wrap around of index here
+      index = _buffer_size - index;
+    }
+
+    if (_resp_data[index] == -1 && rtt == -1) {
+      ++ct;
+    }
+    else if (_resp_data[index] != -1 && rtt != -1) {
+      ++ct;
+    }
+    else {
+      return ct;
+    }
+  }
+  return ct;
+}
+
+/**
+ *
+ *
+ **/
+bool
+LBData::is_active(const string &iface) 
+{
+  InterfaceHealthIter iter = _iface_health_coll.find(iface);
+  if (iter != _iface_health_coll.end()) {
+    return iter->second._is_active;
+  }
+  return false;
+}
+
+/**
+ *
+ *
+ **/
+void
+LBData::dump()
+{
+
+  cout << "health" << endl;
+  LBData::InterfaceHealthIter h_iter = _iface_health_coll.begin();
+  while (h_iter != _iface_health_coll.end()) {
+    cout << "  " << h_iter->first << endl;
+    cout << "    " << h_iter->second._success_ct << endl;
+    cout << "    " << h_iter->second._failure_ct << endl;
+    cout << "    " << h_iter->second._ping_target << endl;
+    cout << "    " << h_iter->second._ping_resp_time << endl;
+    ++h_iter;
+  }
+
+  cout << endl << "wan" << endl;
+  LBData::LBRuleIter r_iter = _lb_rule_coll.begin();
+  while (r_iter != _lb_rule_coll.end()) {
+    cout << "  rule: " << r_iter->first << endl;
+    cout << "    " << r_iter->second._proto << endl;
+    cout << "    " << r_iter->second._s_addr << endl;
+    cout << "    " << r_iter->second._s_net << endl;
+    cout << "    " << r_iter->second._s_port_num << endl;
+    cout << "    " << r_iter->second._s_port_name << endl;
+
+    cout << "    " << r_iter->second._d_addr << endl;
+    cout << "    " << r_iter->second._d_net << endl;
+    cout << "    " << r_iter->second._d_port_num << endl;
+    cout << "    " << r_iter->second._d_port_name << endl;
+    
+    LBRule::InterfaceDistIter ri_iter = r_iter->second._iface_dist_coll.begin();
+    while (ri_iter != r_iter->second._iface_dist_coll.end()) {
+      cout << "      interface: " << ri_iter->first << endl;
+      cout << "      weight: " << ri_iter->second << endl;
+      ++ri_iter;
+    }
+    ++r_iter;
+  }
+  cout << "end dump" << endl;
+}
+
+/**
+ *
+ *
+ **/
+bool
+LBData::state_changed()
+{
+  LBData::InterfaceHealthIter h_iter = _iface_health_coll.begin();
+  while (h_iter != _iface_health_coll.end()) {
+    if (h_iter->second.state_changed()) {
+      return true;
+    }
+    ++h_iter;
+  }
+  return false;
+}
+
+/**
+ *
+ *
+ **/
+void
+LBData::reset_state_changed()
+{
+  LBData::InterfaceHealthIter h_iter = _iface_health_coll.begin();
+  while (h_iter != _iface_health_coll.end()) {
+    h_iter->second._state_changed = false;
+    ++h_iter;
+  }
+}