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