mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00
26a082ce36
Bug: b/169531206 Change-Id: I02c19385ff7078944f7509ecc07358b4315f7b08 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/350181 Commit-Queue: Sergey Sukhanov <sergeysu@webrtc.org> Reviewed-by: Victor Boivie <boivie@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Per Kjellander <perkj@webrtc.org> Cr-Commit-Position: refs/heads/main@{#42261}
112 lines
3.8 KiB
C++
112 lines
3.8 KiB
C++
/*
|
|
* Copyright 2019 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
#include "test/network/feedback_generator.h"
|
|
|
|
#include "absl/memory/memory.h"
|
|
#include "api/transport/network_types.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "test/network/network_emulation_manager.h"
|
|
|
|
namespace webrtc {
|
|
|
|
FeedbackGeneratorImpl::FeedbackGeneratorImpl(
|
|
FeedbackGeneratorImpl::Config config)
|
|
: conf_(config),
|
|
net_({.time_mode = TimeMode::kSimulated}),
|
|
send_link_{new SimulatedNetwork(conf_.send_link)},
|
|
ret_link_{new SimulatedNetwork(conf_.return_link)},
|
|
route_(this,
|
|
net_.CreateRoute(
|
|
{net_.CreateEmulatedNode(absl::WrapUnique(send_link_))}),
|
|
net_.CreateRoute(
|
|
{net_.CreateEmulatedNode(absl::WrapUnique(ret_link_))})) {}
|
|
|
|
Timestamp FeedbackGeneratorImpl::Now() {
|
|
return net_.Now();
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::Sleep(TimeDelta duration) {
|
|
net_.time_controller()->AdvanceTime(duration);
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::SendPacket(size_t size) {
|
|
SentPacket sent;
|
|
sent.send_time = Now();
|
|
sent.size = DataSize::Bytes(size);
|
|
sent.sequence_number = sequence_number_++;
|
|
sent_packets_.push(sent);
|
|
route_.SendRequest(size, sent);
|
|
}
|
|
|
|
std::vector<TransportPacketsFeedback> FeedbackGeneratorImpl::PopFeedback() {
|
|
std::vector<TransportPacketsFeedback> ret;
|
|
ret.swap(feedback_);
|
|
return ret;
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::SetSendConfig(BuiltInNetworkBehaviorConfig config) {
|
|
conf_.send_link = config;
|
|
send_link_->SetConfig(conf_.send_link);
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::SetReturnConfig(
|
|
BuiltInNetworkBehaviorConfig config) {
|
|
conf_.return_link = config;
|
|
ret_link_->SetConfig(conf_.return_link);
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::SetSendLinkCapacity(DataRate capacity) {
|
|
conf_.send_link.link_capacity_kbps = capacity.kbps<int>();
|
|
send_link_->SetConfig(conf_.send_link);
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::OnRequest(SentPacket packet,
|
|
Timestamp arrival_time) {
|
|
PacketResult result;
|
|
result.sent_packet = packet;
|
|
result.receive_time = arrival_time;
|
|
received_packets_.push_back(result);
|
|
Timestamp first_recv = received_packets_.front().receive_time;
|
|
if (Now() - first_recv > conf_.feedback_interval) {
|
|
route_.SendResponse(conf_.feedback_packet_size.bytes<size_t>(),
|
|
std::move(received_packets_));
|
|
received_packets_ = {};
|
|
}
|
|
}
|
|
|
|
void FeedbackGeneratorImpl::OnResponse(std::vector<PacketResult> packet_results,
|
|
Timestamp arrival_time) {
|
|
TransportPacketsFeedback feedback;
|
|
feedback.feedback_time = arrival_time;
|
|
std::vector<PacketResult>::const_iterator received_packet_iterator =
|
|
packet_results.begin();
|
|
while (received_packet_iterator != packet_results.end()) {
|
|
RTC_DCHECK(!sent_packets_.empty() &&
|
|
sent_packets_.front().sequence_number <=
|
|
received_packet_iterator->sent_packet.sequence_number)
|
|
<< "reordering not implemented";
|
|
if (sent_packets_.front().sequence_number <
|
|
received_packet_iterator->sent_packet.sequence_number) {
|
|
// Packet lost.
|
|
PacketResult lost;
|
|
lost.sent_packet = sent_packets_.front();
|
|
feedback.packet_feedbacks.push_back(lost);
|
|
}
|
|
if (sent_packets_.front().sequence_number ==
|
|
received_packet_iterator->sent_packet.sequence_number) {
|
|
feedback.packet_feedbacks.push_back(*received_packet_iterator);
|
|
++received_packet_iterator;
|
|
}
|
|
sent_packets_.pop();
|
|
}
|
|
feedback_.push_back(feedback);
|
|
}
|
|
|
|
} // namespace webrtc
|