OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
(...skipping 30 matching lines...) Expand all Loading... |
41 | 41 |
42 const uint16_t kSsrc1 = 1234; | 42 const uint16_t kSsrc1 = 1234; |
43 uint16_t sequence_number = 17; | 43 uint16_t sequence_number = 17; |
44 uint64_t timestamp = 7890; | 44 uint64_t timestamp = 7890; |
45 bool retransmission = false; | 45 bool retransmission = false; |
46 | 46 |
47 // Send on the first module by letting rtp_1 be sending with correct ssrc. | 47 // Send on the first module by letting rtp_1 be sending with correct ssrc. |
48 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); | 48 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); |
49 EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1)); | 49 EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1)); |
50 EXPECT_CALL(rtp_1, TimeToSendPacket(kSsrc1, sequence_number, timestamp, | 50 EXPECT_CALL(rtp_1, TimeToSendPacket(kSsrc1, sequence_number, timestamp, |
51 retransmission, 1)) | 51 retransmission)) |
52 .Times(1) | 52 .Times(1) |
53 .WillOnce(Return(true)); | 53 .WillOnce(Return(true)); |
54 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0); | 54 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0); |
55 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, | 55 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, |
56 timestamp, retransmission, 1)); | 56 timestamp, retransmission, -1)); |
57 | 57 |
58 // Send on the second module by letting rtp_2 be sending, but not rtp_1. | 58 // Send on the second module by letting rtp_2 be sending, but not rtp_1. |
59 ++sequence_number; | 59 ++sequence_number; |
60 timestamp += 30; | 60 timestamp += 30; |
61 retransmission = true; | 61 retransmission = true; |
62 const uint16_t kSsrc2 = 4567; | 62 const uint16_t kSsrc2 = 4567; |
63 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); | 63 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); |
64 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 64 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
65 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); | 65 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); |
66 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0); | 66 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0); |
67 EXPECT_CALL(rtp_2, TimeToSendPacket(kSsrc2, sequence_number, timestamp, | 67 EXPECT_CALL(rtp_2, TimeToSendPacket(kSsrc2, sequence_number, timestamp, |
68 retransmission, 2)) | 68 retransmission)) |
69 .Times(1) | 69 .Times(1) |
70 .WillOnce(Return(true)); | 70 .WillOnce(Return(true)); |
71 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc2, sequence_number, | 71 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc2, sequence_number, |
72 timestamp, retransmission, 2)); | 72 timestamp, retransmission, -1)); |
73 | 73 |
74 // No module is sending, hence no packet should be sent. | 74 // No module is sending, hence no packet should be sent. |
75 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); | 75 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); |
76 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0); | 76 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0); |
77 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false)); | 77 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false)); |
78 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0); | 78 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0); |
79 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, | 79 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, |
80 timestamp, retransmission, 1)); | 80 timestamp, retransmission, -1)); |
81 | 81 |
82 // Add a packet with incorrect ssrc and test it's dropped in the router. | 82 // Add a packet with incorrect ssrc and test it's dropped in the router. |
83 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); | 83 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); |
84 EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1)); | 84 EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1)); |
85 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 85 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
86 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); | 86 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); |
87 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0); | 87 EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0); |
88 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0); | 88 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0); |
89 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1 + kSsrc2, sequence_number, | 89 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1 + kSsrc2, sequence_number, |
90 timestamp, retransmission, 1)); | 90 timestamp, retransmission, -1)); |
91 | 91 |
92 packet_router_->RemoveRtpModule(&rtp_1); | 92 packet_router_->RemoveRtpModule(&rtp_1); |
93 | 93 |
94 // rtp_1 has been removed, try sending a packet on that ssrc and make sure | 94 // rtp_1 has been removed, try sending a packet on that ssrc and make sure |
95 // it is dropped as expected by not expecting any calls to rtp_1. | 95 // it is dropped as expected by not expecting any calls to rtp_1. |
96 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 96 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
97 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); | 97 EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2)); |
98 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0); | 98 EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0); |
99 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, | 99 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number, |
100 timestamp, retransmission, | 100 timestamp, retransmission, -1)); |
101 PacketInfo::kNotAProbe)); | |
102 | 101 |
103 packet_router_->RemoveRtpModule(&rtp_2); | 102 packet_router_->RemoveRtpModule(&rtp_2); |
104 } | 103 } |
105 | 104 |
106 TEST_F(PacketRouterTest, TimeToSendPadding) { | 105 TEST_F(PacketRouterTest, TimeToSendPadding) { |
107 const uint16_t kSsrc1 = 1234; | 106 const uint16_t kSsrc1 = 1234; |
108 const uint16_t kSsrc2 = 4567; | 107 const uint16_t kSsrc2 = 4567; |
109 | 108 |
110 MockRtpRtcp rtp_1; | 109 MockRtpRtcp rtp_1; |
111 EXPECT_CALL(rtp_1, SSRC()).WillRepeatedly(Return(kSsrc1)); | 110 EXPECT_CALL(rtp_1, SSRC()).WillRepeatedly(Return(kSsrc1)); |
112 MockRtpRtcp rtp_2; | 111 MockRtpRtcp rtp_2; |
113 EXPECT_CALL(rtp_2, SSRC()).WillRepeatedly(Return(kSsrc2)); | 112 EXPECT_CALL(rtp_2, SSRC()).WillRepeatedly(Return(kSsrc2)); |
114 packet_router_->AddRtpModule(&rtp_1); | 113 packet_router_->AddRtpModule(&rtp_1); |
115 packet_router_->AddRtpModule(&rtp_2); | 114 packet_router_->AddRtpModule(&rtp_2); |
116 | 115 |
117 // Default configuration, sending padding on all modules sending media, | 116 // Default configuration, sending padding on all modules sending media, |
118 // ordered by SSRC. | 117 // ordered by SSRC. |
119 const size_t requested_padding_bytes = 1000; | 118 const size_t requested_padding_bytes = 1000; |
120 const size_t sent_padding_bytes = 890; | 119 const size_t sent_padding_bytes = 890; |
121 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); | 120 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true)); |
122 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, 111)) | 121 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes)) |
123 .Times(1) | 122 .Times(1) |
124 .WillOnce(Return(sent_padding_bytes)); | 123 .WillOnce(Return(sent_padding_bytes)); |
125 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 124 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
126 EXPECT_CALL(rtp_2, TimeToSendPadding( | 125 EXPECT_CALL(rtp_2, |
127 requested_padding_bytes - sent_padding_bytes, 111)) | 126 TimeToSendPadding(requested_padding_bytes - sent_padding_bytes)) |
128 .Times(1) | 127 .Times(1) |
129 .WillOnce(Return(requested_padding_bytes - sent_padding_bytes)); | 128 .WillOnce(Return(requested_padding_bytes - sent_padding_bytes)); |
130 EXPECT_EQ(requested_padding_bytes, | 129 EXPECT_EQ(requested_padding_bytes, |
131 packet_router_->TimeToSendPadding(requested_padding_bytes, 111)); | 130 packet_router_->TimeToSendPadding(requested_padding_bytes)); |
132 | 131 |
133 // Let only the second module be sending and verify the padding request is | 132 // Let only the second module be sending and verify the padding request is |
134 // routed there. | 133 // routed there. |
135 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); | 134 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); |
136 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, _)).Times(0); | 135 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes)).Times(0); |
137 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 136 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
138 EXPECT_CALL(rtp_2, TimeToSendPadding(_, _)) | 137 EXPECT_CALL(rtp_2, TimeToSendPadding(_)) |
139 .Times(1) | 138 .Times(1) |
140 .WillOnce(Return(sent_padding_bytes)); | 139 .WillOnce(Return(sent_padding_bytes)); |
141 EXPECT_EQ(sent_padding_bytes, | 140 EXPECT_EQ(sent_padding_bytes, |
142 packet_router_->TimeToSendPadding(requested_padding_bytes, | 141 packet_router_->TimeToSendPadding(requested_padding_bytes)); |
143 PacketInfo::kNotAProbe)); | |
144 | 142 |
145 // No sending module at all. | 143 // No sending module at all. |
146 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); | 144 EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false)); |
147 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, _)).Times(0); | 145 EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes)).Times(0); |
148 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false)); | 146 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false)); |
149 EXPECT_CALL(rtp_2, TimeToSendPadding(_, _)).Times(0); | 147 EXPECT_CALL(rtp_2, TimeToSendPadding(_)).Times(0); |
150 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes, | 148 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes)); |
151 PacketInfo::kNotAProbe)); | |
152 | 149 |
153 packet_router_->RemoveRtpModule(&rtp_1); | 150 packet_router_->RemoveRtpModule(&rtp_1); |
154 | 151 |
155 // rtp_1 has been removed, try sending padding and make sure rtp_1 isn't asked | 152 // rtp_1 has been removed, try sending padding and make sure rtp_1 isn't asked |
156 // to send by not expecting any calls. Instead verify rtp_2 is called. | 153 // to send by not expecting any calls. Instead verify rtp_2 is called. |
157 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); | 154 EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true)); |
158 EXPECT_CALL(rtp_2, TimeToSendPadding(requested_padding_bytes, _)).Times(1); | 155 EXPECT_CALL(rtp_2, TimeToSendPadding(requested_padding_bytes)).Times(1); |
159 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes, | 156 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes)); |
160 PacketInfo::kNotAProbe)); | |
161 | 157 |
162 packet_router_->RemoveRtpModule(&rtp_2); | 158 packet_router_->RemoveRtpModule(&rtp_2); |
163 } | 159 } |
164 | 160 |
165 TEST_F(PacketRouterTest, SenderOnlyFunctionsRespectSendingMedia) { | 161 TEST_F(PacketRouterTest, SenderOnlyFunctionsRespectSendingMedia) { |
166 MockRtpRtcp rtp; | 162 MockRtpRtcp rtp; |
167 packet_router_->AddRtpModule(&rtp); | 163 packet_router_->AddRtpModule(&rtp); |
168 static const uint16_t kSsrc = 1234; | 164 static const uint16_t kSsrc = 1234; |
169 EXPECT_CALL(rtp, SSRC()).WillRepeatedly(Return(kSsrc)); | 165 EXPECT_CALL(rtp, SSRC()).WillRepeatedly(Return(kSsrc)); |
170 EXPECT_CALL(rtp, SendingMedia()).WillRepeatedly(Return(false)); | 166 EXPECT_CALL(rtp, SendingMedia()).WillRepeatedly(Return(false)); |
171 | 167 |
172 // Verify that TimeToSendPacket does not end up in a receiver. | 168 // Verify that TimeToSendPacket does not end up in a receiver. |
173 EXPECT_CALL(rtp, TimeToSendPacket(_, _, _, _, _)).Times(0); | 169 EXPECT_CALL(rtp, TimeToSendPacket(_, _, _, _)).Times(0); |
174 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc, 1, 1, false, | 170 EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc, 1, 1, false, -1)); |
175 PacketInfo::kNotAProbe)); | |
176 // Verify that TimeToSendPadding does not end up in a receiver. | 171 // Verify that TimeToSendPadding does not end up in a receiver. |
177 EXPECT_CALL(rtp, TimeToSendPadding(_, _)).Times(0); | 172 EXPECT_CALL(rtp, TimeToSendPadding(_)).Times(0); |
178 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200, PacketInfo::kNotAProbe)); | 173 EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200)); |
179 | 174 |
180 packet_router_->RemoveRtpModule(&rtp); | 175 packet_router_->RemoveRtpModule(&rtp); |
181 } | 176 } |
182 | 177 |
183 TEST_F(PacketRouterTest, AllocateSequenceNumbers) { | 178 TEST_F(PacketRouterTest, AllocateSequenceNumbers) { |
184 const uint16_t kStartSeq = 0xFFF0; | 179 const uint16_t kStartSeq = 0xFFF0; |
185 const size_t kNumPackets = 32; | 180 const size_t kNumPackets = 32; |
186 | 181 |
187 packet_router_->SetTransportWideSequenceNumber(kStartSeq - 1); | 182 packet_router_->SetTransportWideSequenceNumber(kStartSeq - 1); |
188 | 183 |
(...skipping 12 matching lines...) Expand all Loading... |
201 | 196 |
202 rtcp::TransportFeedback feedback; | 197 rtcp::TransportFeedback feedback; |
203 EXPECT_CALL(rtp_1, SendFeedbackPacket(_)).Times(1); | 198 EXPECT_CALL(rtp_1, SendFeedbackPacket(_)).Times(1); |
204 packet_router_->SendFeedback(&feedback); | 199 packet_router_->SendFeedback(&feedback); |
205 packet_router_->RemoveRtpModule(&rtp_1); | 200 packet_router_->RemoveRtpModule(&rtp_1); |
206 EXPECT_CALL(rtp_2, SendFeedbackPacket(_)).Times(1); | 201 EXPECT_CALL(rtp_2, SendFeedbackPacket(_)).Times(1); |
207 packet_router_->SendFeedback(&feedback); | 202 packet_router_->SendFeedback(&feedback); |
208 packet_router_->RemoveRtpModule(&rtp_2); | 203 packet_router_->RemoveRtpModule(&rtp_2); |
209 } | 204 } |
210 } // namespace webrtc | 205 } // namespace webrtc |
OLD | NEW |