OLD | NEW |
| (Empty) |
1 /* | |
2 * libjingle | |
3 * Copyright 2012 Google Inc. | |
4 * | |
5 * Redistribution and use in source and binary forms, with or without | |
6 * modification, are permitted provided that the following conditions are met: | |
7 * | |
8 * 1. Redistributions of source code must retain the above copyright notice, | |
9 * this list of conditions and the following disclaimer. | |
10 * 2. Redistributions in binary form must reproduce the above copyright notice, | |
11 * this list of conditions and the following disclaimer in the documentation | |
12 * and/or other materials provided with the distribution. | |
13 * 3. The name of the author may not be used to endorse or promote products | |
14 * derived from this software without specific prior written permission. | |
15 * | |
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED | |
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO | |
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
26 */ | |
27 | |
28 #include <string> | |
29 | |
30 #include "talk/media/base/constants.h" | |
31 #include "talk/media/base/fakenetworkinterface.h" | |
32 #include "talk/media/base/rtpdataengine.h" | |
33 #include "talk/media/base/rtputils.h" | |
34 #include "webrtc/base/buffer.h" | |
35 #include "webrtc/base/gunit.h" | |
36 #include "webrtc/base/helpers.h" | |
37 #include "webrtc/base/scoped_ptr.h" | |
38 #include "webrtc/base/ssladapter.h" | |
39 #include "webrtc/base/timing.h" | |
40 | |
41 class FakeTiming : public rtc::Timing { | |
42 public: | |
43 FakeTiming() : now_(0.0) {} | |
44 | |
45 virtual double TimerNow() { | |
46 return now_; | |
47 } | |
48 | |
49 void set_now(double now) { | |
50 now_ = now; | |
51 } | |
52 | |
53 private: | |
54 double now_; | |
55 }; | |
56 | |
57 class FakeDataReceiver : public sigslot::has_slots<> { | |
58 public: | |
59 FakeDataReceiver() : has_received_data_(false) {} | |
60 | |
61 void OnDataReceived( | |
62 const cricket::ReceiveDataParams& params, | |
63 const char* data, size_t len) { | |
64 has_received_data_ = true; | |
65 last_received_data_ = std::string(data, len); | |
66 last_received_data_len_ = len; | |
67 last_received_data_params_ = params; | |
68 } | |
69 | |
70 bool has_received_data() const { return has_received_data_; } | |
71 std::string last_received_data() const { return last_received_data_; } | |
72 size_t last_received_data_len() const { return last_received_data_len_; } | |
73 cricket::ReceiveDataParams last_received_data_params() const { | |
74 return last_received_data_params_; | |
75 } | |
76 | |
77 private: | |
78 bool has_received_data_; | |
79 std::string last_received_data_; | |
80 size_t last_received_data_len_; | |
81 cricket::ReceiveDataParams last_received_data_params_; | |
82 }; | |
83 | |
84 class RtpDataMediaChannelTest : public testing::Test { | |
85 protected: | |
86 virtual void SetUp() { | |
87 // Seed needed for each test to satisfy expectations. | |
88 iface_.reset(new cricket::FakeNetworkInterface()); | |
89 timing_ = new FakeTiming(); | |
90 dme_.reset(CreateEngine(timing_)); | |
91 receiver_.reset(new FakeDataReceiver()); | |
92 } | |
93 | |
94 void SetNow(double now) { | |
95 timing_->set_now(now); | |
96 } | |
97 | |
98 cricket::RtpDataEngine* CreateEngine(FakeTiming* timing) { | |
99 cricket::RtpDataEngine* dme = new cricket::RtpDataEngine(); | |
100 dme->SetTiming(timing); | |
101 return dme; | |
102 } | |
103 | |
104 cricket::RtpDataMediaChannel* CreateChannel() { | |
105 return CreateChannel(dme_.get()); | |
106 } | |
107 | |
108 cricket::RtpDataMediaChannel* CreateChannel(cricket::RtpDataEngine* dme) { | |
109 cricket::RtpDataMediaChannel* channel = | |
110 static_cast<cricket::RtpDataMediaChannel*>(dme->CreateChannel( | |
111 cricket::DCT_RTP)); | |
112 channel->SetInterface(iface_.get()); | |
113 channel->SignalDataReceived.connect( | |
114 receiver_.get(), &FakeDataReceiver::OnDataReceived); | |
115 return channel; | |
116 } | |
117 | |
118 FakeDataReceiver* receiver() { | |
119 return receiver_.get(); | |
120 } | |
121 | |
122 bool HasReceivedData() { | |
123 return receiver_->has_received_data(); | |
124 } | |
125 | |
126 std::string GetReceivedData() { | |
127 return receiver_->last_received_data(); | |
128 } | |
129 | |
130 size_t GetReceivedDataLen() { | |
131 return receiver_->last_received_data_len(); | |
132 } | |
133 | |
134 cricket::ReceiveDataParams GetReceivedDataParams() { | |
135 return receiver_->last_received_data_params(); | |
136 } | |
137 | |
138 bool HasSentData(int count) { | |
139 return (iface_->NumRtpPackets() > count); | |
140 } | |
141 | |
142 std::string GetSentData(int index) { | |
143 // Assume RTP header of length 12 | |
144 rtc::scoped_ptr<const rtc::Buffer> packet( | |
145 iface_->GetRtpPacket(index)); | |
146 if (packet->size() > 12) { | |
147 return std::string(packet->data<char>() + 12, packet->size() - 12); | |
148 } else { | |
149 return ""; | |
150 } | |
151 } | |
152 | |
153 cricket::RtpHeader GetSentDataHeader(int index) { | |
154 rtc::scoped_ptr<const rtc::Buffer> packet( | |
155 iface_->GetRtpPacket(index)); | |
156 cricket::RtpHeader header; | |
157 GetRtpHeader(packet->data(), packet->size(), &header); | |
158 return header; | |
159 } | |
160 | |
161 private: | |
162 rtc::scoped_ptr<cricket::RtpDataEngine> dme_; | |
163 // Timing passed into dme_. Owned by dme_; | |
164 FakeTiming* timing_; | |
165 rtc::scoped_ptr<cricket::FakeNetworkInterface> iface_; | |
166 rtc::scoped_ptr<FakeDataReceiver> receiver_; | |
167 }; | |
168 | |
169 TEST_F(RtpDataMediaChannelTest, SetUnknownCodecs) { | |
170 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
171 | |
172 cricket::DataCodec known_codec; | |
173 known_codec.id = 103; | |
174 known_codec.name = "google-data"; | |
175 cricket::DataCodec unknown_codec; | |
176 unknown_codec.id = 104; | |
177 unknown_codec.name = "unknown-data"; | |
178 | |
179 cricket::DataSendParameters send_parameters_known; | |
180 send_parameters_known.codecs.push_back(known_codec); | |
181 cricket::DataRecvParameters recv_parameters_known; | |
182 recv_parameters_known.codecs.push_back(known_codec); | |
183 | |
184 cricket::DataSendParameters send_parameters_unknown; | |
185 send_parameters_unknown.codecs.push_back(unknown_codec); | |
186 cricket::DataRecvParameters recv_parameters_unknown; | |
187 recv_parameters_unknown.codecs.push_back(unknown_codec); | |
188 | |
189 cricket::DataSendParameters send_parameters_mixed; | |
190 send_parameters_mixed.codecs.push_back(known_codec); | |
191 send_parameters_mixed.codecs.push_back(unknown_codec); | |
192 cricket::DataRecvParameters recv_parameters_mixed; | |
193 recv_parameters_mixed.codecs.push_back(known_codec); | |
194 recv_parameters_mixed.codecs.push_back(unknown_codec); | |
195 | |
196 EXPECT_TRUE(dmc->SetSendParameters(send_parameters_known)); | |
197 EXPECT_FALSE(dmc->SetSendParameters(send_parameters_unknown)); | |
198 EXPECT_TRUE(dmc->SetSendParameters(send_parameters_mixed)); | |
199 EXPECT_TRUE(dmc->SetRecvParameters(recv_parameters_known)); | |
200 EXPECT_FALSE(dmc->SetRecvParameters(recv_parameters_unknown)); | |
201 EXPECT_FALSE(dmc->SetRecvParameters(recv_parameters_mixed)); | |
202 } | |
203 | |
204 TEST_F(RtpDataMediaChannelTest, AddRemoveSendStream) { | |
205 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
206 | |
207 cricket::StreamParams stream1; | |
208 stream1.add_ssrc(41); | |
209 EXPECT_TRUE(dmc->AddSendStream(stream1)); | |
210 cricket::StreamParams stream2; | |
211 stream2.add_ssrc(42); | |
212 EXPECT_TRUE(dmc->AddSendStream(stream2)); | |
213 | |
214 EXPECT_TRUE(dmc->RemoveSendStream(41)); | |
215 EXPECT_TRUE(dmc->RemoveSendStream(42)); | |
216 EXPECT_FALSE(dmc->RemoveSendStream(43)); | |
217 } | |
218 | |
219 TEST_F(RtpDataMediaChannelTest, AddRemoveRecvStream) { | |
220 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
221 | |
222 cricket::StreamParams stream1; | |
223 stream1.add_ssrc(41); | |
224 EXPECT_TRUE(dmc->AddRecvStream(stream1)); | |
225 cricket::StreamParams stream2; | |
226 stream2.add_ssrc(42); | |
227 EXPECT_TRUE(dmc->AddRecvStream(stream2)); | |
228 EXPECT_FALSE(dmc->AddRecvStream(stream2)); | |
229 | |
230 EXPECT_TRUE(dmc->RemoveRecvStream(41)); | |
231 EXPECT_TRUE(dmc->RemoveRecvStream(42)); | |
232 } | |
233 | |
234 TEST_F(RtpDataMediaChannelTest, SendData) { | |
235 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
236 | |
237 cricket::SendDataParams params; | |
238 params.ssrc = 42; | |
239 unsigned char data[] = "food"; | |
240 rtc::Buffer payload(data, 4); | |
241 unsigned char padded_data[] = { | |
242 0x00, 0x00, 0x00, 0x00, | |
243 'f', 'o', 'o', 'd', | |
244 }; | |
245 cricket::SendDataResult result; | |
246 | |
247 // Not sending | |
248 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
249 EXPECT_EQ(cricket::SDR_ERROR, result); | |
250 EXPECT_FALSE(HasSentData(0)); | |
251 ASSERT_TRUE(dmc->SetSend(true)); | |
252 | |
253 // Unknown stream name. | |
254 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
255 EXPECT_EQ(cricket::SDR_ERROR, result); | |
256 EXPECT_FALSE(HasSentData(0)); | |
257 | |
258 cricket::StreamParams stream; | |
259 stream.add_ssrc(42); | |
260 ASSERT_TRUE(dmc->AddSendStream(stream)); | |
261 | |
262 // Unknown codec; | |
263 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
264 EXPECT_EQ(cricket::SDR_ERROR, result); | |
265 EXPECT_FALSE(HasSentData(0)); | |
266 | |
267 cricket::DataCodec codec; | |
268 codec.id = 103; | |
269 codec.name = cricket::kGoogleRtpDataCodecName; | |
270 cricket::DataSendParameters parameters; | |
271 parameters.codecs.push_back(codec); | |
272 ASSERT_TRUE(dmc->SetSendParameters(parameters)); | |
273 | |
274 // Length too large; | |
275 std::string x10000(10000, 'x'); | |
276 EXPECT_FALSE(dmc->SendData( | |
277 params, rtc::Buffer(x10000.data(), x10000.length()), &result)); | |
278 EXPECT_EQ(cricket::SDR_ERROR, result); | |
279 EXPECT_FALSE(HasSentData(0)); | |
280 | |
281 // Finally works! | |
282 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
283 EXPECT_EQ(cricket::SDR_SUCCESS, result); | |
284 ASSERT_TRUE(HasSentData(0)); | |
285 EXPECT_EQ(sizeof(padded_data), GetSentData(0).length()); | |
286 EXPECT_EQ(0, memcmp( | |
287 padded_data, GetSentData(0).data(), sizeof(padded_data))); | |
288 cricket::RtpHeader header0 = GetSentDataHeader(0); | |
289 EXPECT_NE(0, header0.seq_num); | |
290 EXPECT_NE(0U, header0.timestamp); | |
291 EXPECT_EQ(header0.ssrc, 42U); | |
292 EXPECT_EQ(header0.payload_type, 103); | |
293 | |
294 // Should bump timestamp by 180000 because the clock rate is 90khz. | |
295 SetNow(2); | |
296 | |
297 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
298 ASSERT_TRUE(HasSentData(1)); | |
299 EXPECT_EQ(sizeof(padded_data), GetSentData(1).length()); | |
300 EXPECT_EQ(0, memcmp( | |
301 padded_data, GetSentData(1).data(), sizeof(padded_data))); | |
302 cricket::RtpHeader header1 = GetSentDataHeader(1); | |
303 EXPECT_EQ(header1.ssrc, 42U); | |
304 EXPECT_EQ(header1.payload_type, 103); | |
305 EXPECT_EQ(static_cast<uint16_t>(header0.seq_num + 1), | |
306 static_cast<uint16_t>(header1.seq_num)); | |
307 EXPECT_EQ(header0.timestamp + 180000, header1.timestamp); | |
308 } | |
309 | |
310 TEST_F(RtpDataMediaChannelTest, SendDataMultipleClocks) { | |
311 // Timings owned by RtpDataEngines. | |
312 FakeTiming* timing1 = new FakeTiming(); | |
313 rtc::scoped_ptr<cricket::RtpDataEngine> dme1(CreateEngine(timing1)); | |
314 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc1( | |
315 CreateChannel(dme1.get())); | |
316 FakeTiming* timing2 = new FakeTiming(); | |
317 rtc::scoped_ptr<cricket::RtpDataEngine> dme2(CreateEngine(timing2)); | |
318 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc2( | |
319 CreateChannel(dme2.get())); | |
320 | |
321 ASSERT_TRUE(dmc1->SetSend(true)); | |
322 ASSERT_TRUE(dmc2->SetSend(true)); | |
323 | |
324 cricket::StreamParams stream1; | |
325 stream1.add_ssrc(41); | |
326 ASSERT_TRUE(dmc1->AddSendStream(stream1)); | |
327 cricket::StreamParams stream2; | |
328 stream2.add_ssrc(42); | |
329 ASSERT_TRUE(dmc2->AddSendStream(stream2)); | |
330 | |
331 cricket::DataCodec codec; | |
332 codec.id = 103; | |
333 codec.name = cricket::kGoogleRtpDataCodecName; | |
334 cricket::DataSendParameters parameters; | |
335 parameters.codecs.push_back(codec); | |
336 ASSERT_TRUE(dmc1->SetSendParameters(parameters)); | |
337 ASSERT_TRUE(dmc2->SetSendParameters(parameters)); | |
338 | |
339 cricket::SendDataParams params1; | |
340 params1.ssrc = 41; | |
341 cricket::SendDataParams params2; | |
342 params2.ssrc = 42; | |
343 | |
344 unsigned char data[] = "foo"; | |
345 rtc::Buffer payload(data, 3); | |
346 cricket::SendDataResult result; | |
347 | |
348 EXPECT_TRUE(dmc1->SendData(params1, payload, &result)); | |
349 EXPECT_TRUE(dmc2->SendData(params2, payload, &result)); | |
350 | |
351 // Should bump timestamp by 90000 because the clock rate is 90khz. | |
352 timing1->set_now(1); | |
353 // Should bump timestamp by 180000 because the clock rate is 90khz. | |
354 timing2->set_now(2); | |
355 | |
356 EXPECT_TRUE(dmc1->SendData(params1, payload, &result)); | |
357 EXPECT_TRUE(dmc2->SendData(params2, payload, &result)); | |
358 | |
359 ASSERT_TRUE(HasSentData(3)); | |
360 cricket::RtpHeader header1a = GetSentDataHeader(0); | |
361 cricket::RtpHeader header2a = GetSentDataHeader(1); | |
362 cricket::RtpHeader header1b = GetSentDataHeader(2); | |
363 cricket::RtpHeader header2b = GetSentDataHeader(3); | |
364 | |
365 EXPECT_EQ(static_cast<uint16_t>(header1a.seq_num + 1), | |
366 static_cast<uint16_t>(header1b.seq_num)); | |
367 EXPECT_EQ(header1a.timestamp + 90000, header1b.timestamp); | |
368 EXPECT_EQ(static_cast<uint16_t>(header2a.seq_num + 1), | |
369 static_cast<uint16_t>(header2b.seq_num)); | |
370 EXPECT_EQ(header2a.timestamp + 180000, header2b.timestamp); | |
371 } | |
372 | |
373 TEST_F(RtpDataMediaChannelTest, SendDataRate) { | |
374 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
375 | |
376 ASSERT_TRUE(dmc->SetSend(true)); | |
377 | |
378 cricket::DataCodec codec; | |
379 codec.id = 103; | |
380 codec.name = cricket::kGoogleRtpDataCodecName; | |
381 cricket::DataSendParameters parameters; | |
382 parameters.codecs.push_back(codec); | |
383 ASSERT_TRUE(dmc->SetSendParameters(parameters)); | |
384 | |
385 cricket::StreamParams stream; | |
386 stream.add_ssrc(42); | |
387 ASSERT_TRUE(dmc->AddSendStream(stream)); | |
388 | |
389 cricket::SendDataParams params; | |
390 params.ssrc = 42; | |
391 unsigned char data[] = "food"; | |
392 rtc::Buffer payload(data, 4); | |
393 cricket::SendDataResult result; | |
394 | |
395 // With rtp overhead of 32 bytes, each one of our packets is 36 | |
396 // bytes, or 288 bits. So, a limit of 872bps will allow 3 packets, | |
397 // but not four. | |
398 parameters.max_bandwidth_bps = 872; | |
399 ASSERT_TRUE(dmc->SetSendParameters(parameters)); | |
400 | |
401 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
402 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
403 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
404 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
405 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
406 | |
407 SetNow(0.9); | |
408 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
409 | |
410 SetNow(1.1); | |
411 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
412 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
413 SetNow(1.9); | |
414 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
415 | |
416 SetNow(2.2); | |
417 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
418 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
419 EXPECT_TRUE(dmc->SendData(params, payload, &result)); | |
420 EXPECT_FALSE(dmc->SendData(params, payload, &result)); | |
421 } | |
422 | |
423 TEST_F(RtpDataMediaChannelTest, ReceiveData) { | |
424 // PT= 103, SN=2, TS=3, SSRC = 4, data = "abcde" | |
425 unsigned char data[] = { | |
426 0x80, 0x67, 0x00, 0x02, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x2A, | |
427 0x00, 0x00, 0x00, 0x00, | |
428 'a', 'b', 'c', 'd', 'e' | |
429 }; | |
430 rtc::Buffer packet(data, sizeof(data)); | |
431 | |
432 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
433 | |
434 // SetReceived not called. | |
435 dmc->OnPacketReceived(&packet, rtc::PacketTime()); | |
436 EXPECT_FALSE(HasReceivedData()); | |
437 | |
438 dmc->SetReceive(true); | |
439 | |
440 // Unknown payload id | |
441 dmc->OnPacketReceived(&packet, rtc::PacketTime()); | |
442 EXPECT_FALSE(HasReceivedData()); | |
443 | |
444 cricket::DataCodec codec; | |
445 codec.id = 103; | |
446 codec.name = cricket::kGoogleRtpDataCodecName; | |
447 cricket::DataRecvParameters parameters; | |
448 parameters.codecs.push_back(codec); | |
449 ASSERT_TRUE(dmc->SetRecvParameters(parameters)); | |
450 | |
451 // Unknown stream | |
452 dmc->OnPacketReceived(&packet, rtc::PacketTime()); | |
453 EXPECT_FALSE(HasReceivedData()); | |
454 | |
455 cricket::StreamParams stream; | |
456 stream.add_ssrc(42); | |
457 ASSERT_TRUE(dmc->AddRecvStream(stream)); | |
458 | |
459 // Finally works! | |
460 dmc->OnPacketReceived(&packet, rtc::PacketTime()); | |
461 EXPECT_TRUE(HasReceivedData()); | |
462 EXPECT_EQ("abcde", GetReceivedData()); | |
463 EXPECT_EQ(5U, GetReceivedDataLen()); | |
464 } | |
465 | |
466 TEST_F(RtpDataMediaChannelTest, InvalidRtpPackets) { | |
467 unsigned char data[] = { | |
468 0x80, 0x65, 0x00, 0x02 | |
469 }; | |
470 rtc::Buffer packet(data, sizeof(data)); | |
471 | |
472 rtc::scoped_ptr<cricket::RtpDataMediaChannel> dmc(CreateChannel()); | |
473 | |
474 // Too short | |
475 dmc->OnPacketReceived(&packet, rtc::PacketTime()); | |
476 EXPECT_FALSE(HasReceivedData()); | |
477 } | |
OLD | NEW |