webrtc/tools/network_tester/test_controller.cc - Issue 2813193002: Revert of Reland "Add first part of the network_tester functionality"

Side by Side Diff: webrtc/tools/network_tester/test_controller.cc

Issue 2813193002: Revert of Reland "Add first part of the network_tester functionality" (Closed)

Patch Set: Created 3 years, 8 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

OLD	NEW
	(Empty)
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 #include "webrtc/tools/network_tester/test_controller.h"

12

13 namespace webrtc {

14

15 TestController::TestController(int min_port,

16 int max_port,

17 const std::string& config_file_path)

18 : config_file_path_(config_file_path),

19 local_test_done_(false),

20 remote_test_done_(false) {

21 RTC_DCHECK_RUN_ON(&test_controller_thread_checker_);

22 send_data_.fill(42);

23 packet_sender_checker_.Detach();

24 auto socket =

25 std::unique_ptr<rtc::AsyncPacketSocket>(socket_factory_.CreateUdpSocket(

26 rtc::SocketAddress(rtc::GetAnyIP(AF_INET), 0), min_port, max_port));

27 socket->SignalReadPacket.connect(this, &TestController::OnReadPacket);

28 udp_transport_.reset(

29 new cricket::UdpTransport("network tester transport", std::move(socket)));

30 }

31

32 void TestController::SendConnectTo(const std::string& hostname, int port) {

33 RTC_DCHECK_RUN_ON(&test_controller_thread_checker_);

34 udp_transport_->SetRemoteAddress(rtc::SocketAddress(hostname, port));

35 NetworkTesterPacket packet;

36 packet.set_type(NetworkTesterPacket::HAND_SHAKING);

37 SendData(packet, rtc::Optional<size_t>());

38 rtc::CritScope scoped_lock(&local_test_done_lock_);

39 local_test_done_ = false;

40 remote_test_done_ = false;

41 }

42

43 void TestController::Run() {

44 RTC_DCHECK_RUN_ON(&test_controller_thread_checker_);

45 rtc::Thread::Current()->ProcessMessages(0);

46 }

47

48 void TestController::SendData(const NetworkTesterPacket& packet,

49 rtc::Optional<size_t> data_size) {

50 // Can be call from packet_sender or from test_controller thread.

51 size_t packet_size = packet.ByteSize();

52 send_data_[0] = packet_size;

53 packet.SerializeToArray(&send_data_[1], std::numeric_limits<char>::max());

54 if (data_size && *data_size > packet_size)

55 packet_size = *data_size;

56 udp_transport_->SendPacket(send_data_.data(), packet_size + 1,

57 rtc::PacketOptions(), 0);

58 }

59

60 void TestController::OnTestDone() {

61 RTC_DCHECK_CALLED_SEQUENTIALLY(&packet_sender_checker_);

62 NetworkTesterPacket packet;

63 packet.set_type(NetworkTesterPacket::TEST_DONE);

64 SendData(packet, rtc::Optional<size_t>());

65 rtc::CritScope scoped_lock(&local_test_done_lock_);

66 local_test_done_ = true;

67 }

68

69 bool TestController::IsTestDone() {

70 RTC_DCHECK_RUN_ON(&test_controller_thread_checker_);

71 rtc::CritScope scoped_lock(&local_test_done_lock_);

72 return local_test_done_ && remote_test_done_;

73 }

74

75 void TestController::OnReadPacket(rtc::AsyncPacketSocket* socket,

76 const char* data,

77 size_t len,

78 const rtc::SocketAddress& remote_addr,

79 const rtc::PacketTime& packet_time) {

80 RTC_DCHECK_RUN_ON(&test_controller_thread_checker_);

81 size_t packet_size = data[0];

82 std::string receive_data(&data[1], packet_size);

83 NetworkTesterPacket packet;

84 packet.ParseFromString(receive_data);

85 RTC_CHECK(packet.has_type());

86 switch (packet.type()) {

87 case NetworkTesterPacket::HAND_SHAKING: {

88 NetworkTesterPacket packet;

89 packet.set_type(NetworkTesterPacket::TEST_START);

90 udp_transport_->SetRemoteAddress(remote_addr);

91 SendData(packet, rtc::Optional<size_t>());

92 packet_sender_.reset(new PacketSender(this, config_file_path_));

93 packet_sender_->StartSending();

94 rtc::CritScope scoped_lock(&local_test_done_lock_);

95 local_test_done_ = false;

96 remote_test_done_ = false;

97 break;

98 }

99 case NetworkTesterPacket::TEST_START: {

100 packet_sender_.reset(new PacketSender(this, config_file_path_));

101 packet_sender_->StartSending();

102 break;

103 }

104 case NetworkTesterPacket::TEST_DATA: {

105 packet.set_arrival_timestamp(packet_time.timestamp);

106 packet.set_packet_size(len);

107 // log packet

108 break;

109 }

110 case NetworkTesterPacket::TEST_DONE: {

111 remote_test_done_ = true;

112 break;

113 }

114 default: { RTC_NOTREACHED(); }

115 }

116 }

117

118 } // namespace webrtc

OLD	NEW

« no previous file with comments | « webrtc/tools/network_tester/test_controller.h ('k') | no next file » | no next file with comments »