aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPer K <perkj@webrtc.org>2023-12-13 08:18:38 +0100
committerWebRTC LUCI CQ <webrtc-scoped@luci-project-accounts.iam.gserviceaccount.com>2023-12-13 07:52:23 +0000
commit0b11bd4c2e885f108d852221af0cd624eff3ff01 (patch)
tree9429c024fea0dcf8e6062fc52a4cb642d3c33c21
parent14a7e8b300ca5d714a17bc5d09db285f0114aafa (diff)
downloadwebrtc-0b11bd4c2e885f108d852221af0cd624eff3ff01.tar.gz
Refactor NetworkTester to use rtc::ReceivedPackets
Instead of using raw pointers. Bug: webrtc:11943 Change-Id: I296fb4f234938126f834b4a6ac5e64d31b191d34 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/331280 Commit-Queue: Per Kjellander <perkj@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/main@{#41370}
-rw-r--r--rtc_tools/network_tester/BUILD.gn3
-rw-r--r--rtc_tools/network_tester/test_controller.cc25
-rw-r--r--rtc_tools/network_tester/test_controller.h11
3 files changed, 21 insertions, 18 deletions
diff --git a/rtc_tools/network_tester/BUILD.gn b/rtc_tools/network_tester/BUILD.gn
index 5930431ab9..e44681441a 100644
--- a/rtc_tools/network_tester/BUILD.gn
+++ b/rtc_tools/network_tester/BUILD.gn
@@ -43,6 +43,7 @@ if (rtc_enable_protobuf) {
"../../api/task_queue",
"../../api/task_queue:default_task_queue_factory",
"../../api/task_queue:pending_task_safety_flag",
+ "../../api/units:timestamp",
"../../p2p:rtc_p2p",
"../../rtc_base:async_packet_socket",
"../../rtc_base:checks",
@@ -55,9 +56,9 @@ if (rtc_enable_protobuf) {
"../../rtc_base:socket_server",
"../../rtc_base:threading",
"../../rtc_base:timeutils",
+ "../../rtc_base/network:received_packet",
"../../rtc_base/synchronization:mutex",
"../../rtc_base/system:no_unique_address",
- "../../rtc_base/third_party/sigslot",
]
absl_deps = [
"//third_party/abseil-cpp/absl/functional:any_invocable",
diff --git a/rtc_tools/network_tester/test_controller.cc b/rtc_tools/network_tester/test_controller.cc
index 3d9af380f1..f8641aacb6 100644
--- a/rtc_tools/network_tester/test_controller.cc
+++ b/rtc_tools/network_tester/test_controller.cc
@@ -13,10 +13,12 @@
#include <limits>
#include "absl/types/optional.h"
+#include "api/units/timestamp.h"
#include "rtc_base/checks.h"
#include "rtc_base/internal/default_socket_server.h"
#include "rtc_base/ip_address.h"
#include "rtc_base/logging.h"
+#include "rtc_base/network/received_packet.h"
#include "rtc_base/thread.h"
namespace webrtc {
@@ -44,7 +46,10 @@ TestController::TestController(int min_port,
std::unique_ptr<rtc::AsyncPacketSocket>(socket_factory_.CreateUdpSocket(
rtc::SocketAddress(rtc::GetAnyIP(AF_INET), 0), min_port, max_port));
RTC_CHECK(udp_socket_ != nullptr);
- udp_socket_->SignalReadPacket.connect(this, &TestController::OnReadPacket);
+ udp_socket_->RegisterReceivedPacketCallback(
+ [&](rtc::AsyncPacketSocket* socket, const rtc::ReceivedPacket& packet) {
+ OnReadPacket(socket, packet);
+ });
});
}
@@ -103,14 +108,13 @@ bool TestController::IsTestDone() {
}
void TestController::OnReadPacket(rtc::AsyncPacketSocket* socket,
- const char* data,
- size_t len,
- const rtc::SocketAddress& remote_addr,
- const int64_t& packet_time_us) {
+ const rtc::ReceivedPacket& received_packet) {
RTC_DCHECK_RUN_ON(packet_sender_thread_.get());
RTC_LOG(LS_VERBOSE) << "OnReadPacket";
- size_t packet_size = data[0];
- std::string receive_data(&data[1], packet_size);
+ size_t packet_size = received_packet.payload()[0];
+ std::string receive_data(
+ reinterpret_cast<const char*>(&received_packet.payload()[1]),
+ packet_size);
NetworkTesterPacket packet;
packet.ParseFromString(receive_data);
RTC_CHECK(packet.has_type());
@@ -118,7 +122,7 @@ void TestController::OnReadPacket(rtc::AsyncPacketSocket* socket,
case NetworkTesterPacket::HAND_SHAKING: {
NetworkTesterPacket packet;
packet.set_type(NetworkTesterPacket::TEST_START);
- remote_address_ = remote_addr;
+ remote_address_ = received_packet.source_address();
SendData(packet, absl::nullopt);
packet_sender_.reset(new PacketSender(this, packet_sender_thread_.get(),
task_safety_flag_,
@@ -140,8 +144,9 @@ void TestController::OnReadPacket(rtc::AsyncPacketSocket* socket,
break;
}
case NetworkTesterPacket::TEST_DATA: {
- packet.set_arrival_timestamp(packet_time_us);
- packet.set_packet_size(len);
+ packet.set_arrival_timestamp(
+ received_packet.arrival_time().value_or(Timestamp::Zero()).us());
+ packet.set_packet_size(received_packet.payload().size());
packet_logger_.LogPacket(packet);
break;
}
diff --git a/rtc_tools/network_tester/test_controller.h b/rtc_tools/network_tester/test_controller.h
index 3638c75af1..423eb08d0c 100644
--- a/rtc_tools/network_tester/test_controller.h
+++ b/rtc_tools/network_tester/test_controller.h
@@ -22,11 +22,11 @@
#include "api/sequence_checker.h"
#include "p2p/base/basic_packet_socket_factory.h"
#include "rtc_base/async_packet_socket.h"
+#include "rtc_base/network/received_packet.h"
#include "rtc_base/socket_address.h"
#include "rtc_base/socket_server.h"
#include "rtc_base/synchronization/mutex.h"
#include "rtc_base/system/no_unique_address.h"
-#include "rtc_base/third_party/sigslot/sigslot.h"
#include "rtc_base/thread.h"
#include "rtc_base/thread_annotations.h"
#include "rtc_tools/network_tester/packet_logger.h"
@@ -43,13 +43,13 @@ namespace webrtc {
constexpr size_t kEthernetMtu = 1500;
-class TestController : public sigslot::has_slots<> {
+class TestController {
public:
TestController(int min_port,
int max_port,
const std::string& config_file_path,
const std::string& log_file_path);
- ~TestController() override;
+ ~TestController();
TestController(const TestController&) = delete;
TestController& operator=(const TestController&) = delete;
@@ -65,10 +65,7 @@ class TestController : public sigslot::has_slots<> {
private:
void OnReadPacket(rtc::AsyncPacketSocket* socket,
- const char* data,
- size_t len,
- const rtc::SocketAddress& remote_addr,
- const int64_t& packet_time_us);
+ const rtc::ReceivedPacket& received_packet);
RTC_NO_UNIQUE_ADDRESS SequenceChecker test_controller_thread_checker_;
std::unique_ptr<rtc::SocketServer> socket_server_;
std::unique_ptr<rtc::Thread> packet_sender_thread_;