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_PACKET_SENDER_H_ 12 #define RTC_TOOLS_NETWORK_TESTER_PACKET_SENDER_H_ 13 14 #include <memory> 15 #include <string> 16 17 #include "api/sequence_checker.h" 18 #include "api/task_queue/pending_task_safety_flag.h" 19 #include "api/task_queue/task_queue_factory.h" 20 #include "rtc_base/ignore_wundef.h" 21 #include "rtc_base/system/no_unique_address.h" 22 23 #ifdef WEBRTC_NETWORK_TESTER_PROTO 24 RTC_PUSH_IGNORING_WUNDEF() 25 #include "rtc_tools/network_tester/network_tester_packet.pb.h" 26 RTC_POP_IGNORING_WUNDEF() 27 using webrtc::network_tester::packet::NetworkTesterPacket; 28 #else 29 class NetworkTesterPacket; 30 #endif // WEBRTC_NETWORK_TESTER_PROTO 31 32 namespace webrtc { 33 34 class TestController; 35 36 class PacketSender { 37 public: 38 PacketSender( 39 TestController* test_controller, 40 webrtc::TaskQueueBase* worker_queue, 41 rtc::scoped_refptr<webrtc::PendingTaskSafetyFlag> task_safety_flag, 42 const std::string& config_file_path); 43 ~PacketSender(); 44 45 PacketSender(const PacketSender&) = delete; 46 PacketSender& operator=(const PacketSender&) = delete; 47 48 void StartSending(); 49 void StopSending(); 50 bool IsSending() const; 51 52 void SendPacket(); 53 54 int64_t GetSendIntervalMs() const; 55 void UpdateTestSetting(size_t packet_size, int64_t send_interval_ms); 56 57 private: 58 RTC_NO_UNIQUE_ADDRESS SequenceChecker worker_queue_checker_; 59 size_t packet_size_ RTC_GUARDED_BY(worker_queue_checker_); 60 int64_t send_interval_ms_ RTC_GUARDED_BY(worker_queue_checker_); 61 int64_t sequence_number_ RTC_GUARDED_BY(worker_queue_checker_); 62 bool sending_ RTC_GUARDED_BY(worker_queue_checker_); 63 const std::string config_file_path_; 64 TestController* const test_controller_; 65 webrtc::TaskQueueBase* worker_queue_; 66 rtc::scoped_refptr<webrtc::PendingTaskSafetyFlag> task_safety_flag_; 67 }; 68 69 } // namespace webrtc 70 71 #endif // RTC_TOOLS_NETWORK_TESTER_PACKET_SENDER_H_ 72