blob: e44e67f7e932a74fbfd81dc2b77f6e8da97996e1 [file] [log] [blame]
// Copyright (c) 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "quic/test_tools/simulator/traffic_policer.h"
#include <algorithm>
namespace quic {
namespace simulator {
TrafficPolicer::TrafficPolicer(Simulator* simulator,
std::string name,
QuicByteCount initial_bucket_size,
QuicByteCount max_bucket_size,
QuicBandwidth target_bandwidth,
Endpoint* input)
: PacketFilter(simulator, name, input),
initial_bucket_size_(initial_bucket_size),
max_bucket_size_(max_bucket_size),
target_bandwidth_(target_bandwidth),
last_refill_time_(clock_->Now()) {}
TrafficPolicer::~TrafficPolicer() {}
void TrafficPolicer::Refill() {
QuicTime::Delta time_passed = clock_->Now() - last_refill_time_;
QuicByteCount refill_size = time_passed * target_bandwidth_;
for (auto& bucket : token_buckets_) {
bucket.second = std::min(bucket.second + refill_size, max_bucket_size_);
}
last_refill_time_ = clock_->Now();
}
bool TrafficPolicer::FilterPacket(const Packet& packet) {
// Refill existing buckets.
Refill();
// Create a new bucket if one does not exist.
if (token_buckets_.count(packet.destination) == 0) {
token_buckets_.insert(
std::make_pair(packet.destination, initial_bucket_size_));
}
auto bucket = token_buckets_.find(packet.destination);
DCHECK(bucket != token_buckets_.end());
// Silently drop the packet on the floor if out of tokens
if (bucket->second < packet.size) {
return false;
}
bucket->second -= packet.size;
return true;
}
} // namespace simulator
} // namespace quic