diff options
Diffstat (limited to 'src/lbdata.cc')
| -rw-r--r-- | src/lbdata.cc | 212 |
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; + } +} |