diff options
Diffstat (limited to 'third_party/libwebrtc/rtc_tools/network_tester/test_controller.h')
-rw-r--r-- | third_party/libwebrtc/rtc_tools/network_tester/test_controller.h | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_tools/network_tester/test_controller.h b/third_party/libwebrtc/rtc_tools/network_tester/test_controller.h new file mode 100644 index 0000000000..b08fbd5dd8 --- /dev/null +++ b/third_party/libwebrtc/rtc_tools/network_tester/test_controller.h @@ -0,0 +1,95 @@ +/* + * Copyright 2017 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. + */ + +#ifndef RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ +#define RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ + +#include <stddef.h> +#include <stdint.h> + +#include <array> +#include <memory> +#include <string> + +#include "absl/types/optional.h" +#include "api/sequence_checker.h" +#include "p2p/base/basic_packet_socket_factory.h" +#include "rtc_base/async_packet_socket.h" +#include "rtc_base/ignore_wundef.h" +#include "rtc_base/socket_address.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_annotations.h" +#include "rtc_tools/network_tester/packet_logger.h" +#include "rtc_tools/network_tester/packet_sender.h" + +#ifdef WEBRTC_NETWORK_TESTER_PROTO +RTC_PUSH_IGNORING_WUNDEF() +#include "rtc_tools/network_tester/network_tester_packet.pb.h" +RTC_POP_IGNORING_WUNDEF() +using webrtc::network_tester::packet::NetworkTesterPacket; +#else +class NetworkTesterPacket; +#endif // WEBRTC_NETWORK_TESTER_PROTO + +namespace webrtc { + +constexpr size_t kEthernetMtu = 1500; + +class TestController : public sigslot::has_slots<> { + public: + TestController(int min_port, + int max_port, + const std::string& config_file_path, + const std::string& log_file_path); + ~TestController() override; + + TestController(const TestController&) = delete; + TestController& operator=(const TestController&) = delete; + + void SendConnectTo(const std::string& hostname, int port); + + void SendData(const NetworkTesterPacket& packet, + absl::optional<size_t> data_size); + + void OnTestDone(); + + bool IsTestDone(); + + private: + void OnReadPacket(rtc::AsyncPacketSocket* socket, + const char* data, + size_t len, + const rtc::SocketAddress& remote_addr, + const int64_t& packet_time_us); + RTC_NO_UNIQUE_ADDRESS SequenceChecker test_controller_thread_checker_; + std::unique_ptr<rtc::SocketServer> socket_server_; + std::unique_ptr<rtc::Thread> packet_sender_thread_; + rtc::BasicPacketSocketFactory socket_factory_ + RTC_GUARDED_BY(packet_sender_thread_); + const std::string config_file_path_; + PacketLogger packet_logger_ RTC_GUARDED_BY(packet_sender_thread_); + Mutex test_done_lock_ RTC_GUARDED_BY(test_controller_thread_checker_); + bool local_test_done_ RTC_GUARDED_BY(test_done_lock_); + bool remote_test_done_ RTC_GUARDED_BY(test_done_lock_); + std::array<char, kEthernetMtu> send_data_ + RTC_GUARDED_BY(packet_sender_thread_); + std::unique_ptr<rtc::AsyncPacketSocket> udp_socket_ + RTC_GUARDED_BY(packet_sender_thread_); + rtc::SocketAddress remote_address_; + std::unique_ptr<PacketSender> packet_sender_ + RTC_GUARDED_BY(packet_sender_thread_); + rtc::scoped_refptr<webrtc::PendingTaskSafetyFlag> task_safety_flag_; +}; + +} // namespace webrtc + +#endif // RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ |