1 /* 2 * Copyright 2017 The WebRTC project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ 12 #define RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ 13 14 #include <stddef.h> 15 #include <stdint.h> 16 17 #include <array> 18 #include <memory> 19 #include <string> 20 21 #include "absl/types/optional.h" 22 #include "p2p/base/basic_packet_socket_factory.h" 23 #include "rtc_base/async_packet_socket.h" 24 #include "rtc_base/constructor_magic.h" 25 #include "rtc_base/critical_section.h" 26 #include "rtc_base/ignore_wundef.h" 27 #include "rtc_base/socket_address.h" 28 #include "rtc_base/synchronization/sequence_checker.h" 29 #include "rtc_base/third_party/sigslot/sigslot.h" 30 #include "rtc_base/thread_annotations.h" 31 #include "rtc_base/thread_checker.h" 32 #include "rtc_tools/network_tester/packet_logger.h" 33 #include "rtc_tools/network_tester/packet_sender.h" 34 35 #ifdef WEBRTC_NETWORK_TESTER_PROTO 36 RTC_PUSH_IGNORING_WUNDEF() 37 #include "rtc_tools/network_tester/network_tester_packet.pb.h" 38 39 RTC_POP_IGNORING_WUNDEF() 40 using webrtc::network_tester::packet::NetworkTesterPacket; 41 #else 42 class NetworkTesterPacket; 43 #endif // WEBRTC_NETWORK_TESTER_PROTO 44 45 namespace webrtc { 46 47 constexpr size_t kEthernetMtu = 1500; 48 49 class TestController : public sigslot::has_slots<> { 50 public: 51 TestController(int min_port, 52 int max_port, 53 const std::string& config_file_path, 54 const std::string& log_file_path); 55 56 void Run(); 57 58 void SendConnectTo(const std::string& hostname, int port); 59 60 void SendData(const NetworkTesterPacket& packet, 61 absl::optional<size_t> data_size); 62 63 void OnTestDone(); 64 65 bool IsTestDone(); 66 67 private: 68 void OnReadPacket(rtc::AsyncPacketSocket* socket, 69 const char* data, 70 size_t len, 71 const rtc::SocketAddress& remote_addr, 72 const int64_t& packet_time_us); 73 rtc::ThreadChecker test_controller_thread_checker_; 74 SequenceChecker packet_sender_checker_; 75 rtc::BasicPacketSocketFactory socket_factory_; 76 const std::string config_file_path_; 77 PacketLogger packet_logger_; 78 rtc::CriticalSection local_test_done_lock_; 79 bool local_test_done_ RTC_GUARDED_BY(local_test_done_lock_); 80 bool remote_test_done_; 81 std::array<char, kEthernetMtu> send_data_; 82 std::unique_ptr<rtc::AsyncPacketSocket> udp_socket_; 83 rtc::SocketAddress remote_address_; 84 std::unique_ptr<PacketSender> packet_sender_; 85 86 RTC_DISALLOW_COPY_AND_ASSIGN(TestController); 87 }; 88 89 } // namespace webrtc 90 91 #endif // RTC_TOOLS_NETWORK_TESTER_TEST_CONTROLLER_H_ 92