mirror of
https://github.com/ton-blockchain/ton
synced 2025-02-12 11:12:16 +00:00
144 lines
3.6 KiB
C++
144 lines
3.6 KiB
C++
/*
|
|
This file is part of TON Blockchain Library.
|
|
|
|
TON Blockchain Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TON Blockchain Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Copyright 2017-2020 Telegram Systems LLP
|
|
*/
|
|
|
|
#include "SenderPackets.h"
|
|
|
|
#include "td/utils/bits.h"
|
|
|
|
namespace ton {
|
|
namespace rldp2 {
|
|
td::uint32 SenderPackets::next_seqno() const {
|
|
return last_seqno_ + 1;
|
|
}
|
|
|
|
SenderPackets::DropUpdate SenderPackets::drop_packets(const Limits &limits) {
|
|
while (!packets.empty()) {
|
|
auto &packet = packets.front();
|
|
if (!limits.should_drop(packet)) {
|
|
break;
|
|
}
|
|
mark_ack_or_lost(packet);
|
|
packets.pop();
|
|
}
|
|
DropUpdate update;
|
|
update.new_ack = total_ack_ - last_total_ack_;
|
|
update.new_lost = total_lost_ - last_total_lost_;
|
|
last_total_ack_ = total_ack_;
|
|
last_total_lost_ = total_lost_;
|
|
update.o_loss_at = std::move(last_loss_);
|
|
return update;
|
|
}
|
|
|
|
SenderPackets::Update SenderPackets::on_ack(Ack ack) {
|
|
ack.max_seqno = td::min(ack.max_seqno, last_seqno_);
|
|
ack.received_count = td::min(ack.received_count, ack.max_seqno);
|
|
|
|
// TODO: seqno of rldp and seqno of a packet must be completly separate seqnos
|
|
Update update;
|
|
if (received_count_ < ack.received_count) {
|
|
update.new_received = ack.received_count - received_count_;
|
|
left_ack_ += update.new_received;
|
|
left_ack_ = td::min(left_ack_, in_flight_count_);
|
|
received_count_ = ack.received_count;
|
|
}
|
|
|
|
if (max_packet_.seqno > ack.max_seqno) {
|
|
return update;
|
|
}
|
|
|
|
auto packet = get_packet(ack.max_seqno);
|
|
if (!packet) {
|
|
return update;
|
|
}
|
|
|
|
if (max_packet_.seqno < ack.max_seqno) {
|
|
update.was_max_updated = true;
|
|
max_packet_ = *packet;
|
|
}
|
|
|
|
for (td::uint32 i : td::BitsRange(ack.received_mask)) {
|
|
if (ack.max_seqno < i) {
|
|
break;
|
|
}
|
|
auto seqno = ack.max_seqno - i;
|
|
auto packet = get_packet(seqno);
|
|
if (!packet) {
|
|
break;
|
|
}
|
|
mark_ack(*packet);
|
|
}
|
|
|
|
return update;
|
|
}
|
|
void SenderPackets::mark_ack_or_lost(Packet &packet) {
|
|
if (left_ack_) {
|
|
mark_ack(packet);
|
|
} else {
|
|
mark_lost(packet);
|
|
}
|
|
}
|
|
|
|
void SenderPackets::mark_lost(Packet &packet) {
|
|
if (!packet.is_in_flight) {
|
|
return;
|
|
}
|
|
total_lost_++;
|
|
in_flight_count_--;
|
|
packet.is_in_flight = false;
|
|
last_loss_ = packet.sent_at;
|
|
}
|
|
|
|
void SenderPackets::mark_ack(Packet &packet) {
|
|
if (!packet.is_in_flight) {
|
|
return;
|
|
}
|
|
if (left_ack_ > 0) {
|
|
left_ack_--;
|
|
}
|
|
total_ack_++;
|
|
in_flight_count_--;
|
|
packet.is_in_flight = false;
|
|
}
|
|
|
|
SenderPackets::Packet *SenderPackets::get_packet(td::uint32 seqno) {
|
|
if (packets.empty()) {
|
|
return nullptr;
|
|
}
|
|
auto front_seqno = packets.front().seqno;
|
|
if (front_seqno > seqno) {
|
|
return nullptr;
|
|
}
|
|
td::uint32 index = seqno - front_seqno;
|
|
if (index >= packets.size()) {
|
|
return nullptr;
|
|
}
|
|
auto packet = packets.data() + index;
|
|
CHECK(packet->seqno == seqno);
|
|
return packet;
|
|
}
|
|
|
|
void SenderPackets::send(Packet packet) {
|
|
CHECK(next_seqno() == packet.seqno);
|
|
packets.push(packet);
|
|
last_seqno_++;
|
|
in_flight_count_ += packet.is_in_flight;
|
|
}
|
|
|
|
} // namespace rldp2
|
|
} // namespace ton
|