49 lines
1.7 KiB
C++
49 lines
1.7 KiB
C++
#include "congestion_control.hpp"
|
|
#include "VpnPeer.hpp"
|
|
|
|
CongestionController::CongestionController(const VpnPeer& peer):
|
|
_peer(peer)
|
|
{
|
|
_last_seqno = _peer.get_loss_logger().get_cur_seqno();
|
|
_loss_based.bandwidth = 3e5; // 300kBps seems a good value to start with
|
|
_loss_based.prev_tot_sent = _peer.get_tot_bytes_sent();
|
|
_last_bucket_update = _loss_based.prev_time =
|
|
std::chrono::steady_clock::now();
|
|
update_params();
|
|
}
|
|
|
|
void CongestionController::update_lossbased() {
|
|
const VpnPeer::LossReports& loss_rep = _peer.get_loss_reports();
|
|
|
|
uint32_t delta_seqno = loss_rep.last_seqno - loss_rep.prev_seqno;
|
|
unsigned int delta_losses = loss_rep.last_losses - loss_rep.prev_losses;
|
|
|
|
double loss_rate = (double)delta_losses / (double)delta_seqno;
|
|
|
|
auto cur_time = std::chrono::steady_clock::now();
|
|
uint64_t cur_sent = _peer.get_tot_bytes_sent();
|
|
double instant_bandwidth = /* byte per second */
|
|
(double)(cur_sent - _loss_based.prev_tot_sent)
|
|
/ (double)(std::chrono::duration_cast<std::chrono::microseconds>(
|
|
cur_time - _loss_based.prev_time).count())
|
|
* 1e6;
|
|
|
|
_loss_based.prev_time = cur_time;
|
|
_loss_based.prev_tot_sent = cur_sent;
|
|
|
|
if(loss_rate < 0.02) {
|
|
if(instant_bandwidth > 0.75 * _loss_based.bandwidth)
|
|
_loss_based.bandwidth *= 1.05;
|
|
}
|
|
else if(loss_rate >= 0.1)
|
|
_loss_based.bandwidth *= (1 - 0.5 * loss_rate);
|
|
|
|
update_params();
|
|
}
|
|
|
|
void CongestionController::update_params() {
|
|
_bandwidth = _loss_based.bandwidth; // meant to integrate other controllers
|
|
_bucket_max_level = _bandwidth * _peer.get_rtt().avg_rtt();
|
|
_bucket_level = std::min(_bucket_level, _bucket_max_level);
|
|
}
|