18281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org/* 207c68b9c9de03b4110f25cb59d9d7ba8bad624bdleozwang@webrtc.org * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 38281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * 48281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * Use of this source code is governed by a BSD-style license 58281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * that can be found in the LICENSE file in the root of the source 68281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * tree. An additional intellectual property rights grant can be found 78281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * in the file PATENTS. All contributing project authors may 88281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org * be found in the AUTHORS file in the root of the source tree. 98281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org */ 108281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 11bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org#include "webrtc/modules/rtp_rtcp/test/testAPI/test_api.h" 12bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org 138281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org#include <algorithm> 148281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org#include <vector> 158281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 16b8b6fbb7a5d2f5a14f7f6f81c253747aa28e4c7fdanilchap#include "webrtc/test/null_transport.h" 17b8b6fbb7a5d2f5a14f7f6f81c253747aa28e4c7fdanilchap 189691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comnamespace webrtc { 196a6f0893dd1e653410ba4b22e7f33947d15aeb65danilchap 209691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comvoid LoopBackTransport::SetSendModule(RtpRtcp* rtp_rtcp_module, 219691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RTPPayloadRegistry* payload_registry, 229691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RtpReceiver* receiver, 239691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com ReceiveStatistics* receive_statistics) { 249691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_rtcp_module_ = rtp_rtcp_module; 259691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_payload_registry_ = payload_registry; 269691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_receiver_ = receiver; 279691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com receive_statistics_ = receive_statistics; 289691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 299691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 309691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comvoid LoopBackTransport::DropEveryNthPacket(int n) { 319691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com packet_loss_ = n; 329691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 339691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 341d8a506405734d0cef9653704b036ca4f1388960stefanbool LoopBackTransport::SendRtp(const uint8_t* data, 351d8a506405734d0cef9653704b036ca4f1388960stefan size_t len, 361d8a506405734d0cef9653704b036ca4f1388960stefan const PacketOptions& options) { 379691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com count_++; 389691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (packet_loss_ > 0) { 399691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if ((count_ % packet_loss_) == 0) { 402d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 419691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 429691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 439691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RTPHeader header; 4400b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpHeaderParser> parser(RtpHeaderParser::Create()); 459691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!parser->Parse(static_cast<const uint8_t*>(data), len, &header)) { 462d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 479691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 489691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com PayloadUnion payload_specific; 499691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!rtp_payload_registry_->GetPayloadSpecifics(header.payloadType, 509691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com &payload_specific)) { 512d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 529691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 539691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com receive_statistics_->IncomingPacket(header, len, false); 549691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!rtp_receiver_->IncomingRtpPacket(header, 559691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com static_cast<const uint8_t*>(data), len, 569691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com payload_specific, true)) { 572d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 589691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 592d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 609691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 619691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 622d566686a23fe93ada58f1c38a0d4b9a0d68556epbosbool LoopBackTransport::SendRtcp(const uint8_t* data, size_t len) { 639691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (rtp_rtcp_module_->IncomingRtcpPacket((const uint8_t*)data, len) < 0) { 642d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 659691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 662d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 679691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 689691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 699691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comint32_t TestRtpReceiver::OnReceivedPayloadData( 709691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const uint8_t* payload_data, 719691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const size_t payload_size, 729691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const webrtc::WebRtcRTPHeader* rtp_header) { 739691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_LE(payload_size, sizeof(payload_data_)); 749691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com memcpy(payload_data_, payload_data, payload_size); 759691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com memcpy(&rtp_header_, rtp_header, sizeof(rtp_header_)); 769691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com payload_size_ = payload_size; 779691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com return 0; 789691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 799691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 808281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgclass RtpRtcpAPITest : public ::testing::Test { 818281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org protected: 829691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RtpRtcpAPITest() : fake_clock_(123456) { 839691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_csrcs_.push_back(1234); 849691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_csrcs_.push_back(2345); 859691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_ssrc_ = 3456; 869691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_timestamp_ = 4567; 879691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_sequence_number_ = 2345; 888281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org } 898281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org ~RtpRtcpAPITest() {} 908281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 9114665ff7d4024d07e58622f498b23fd980001871kjellander@webrtc.org void SetUp() override { 922853dde5201d9ddc284ce5efb9688f6340487acepwestin@webrtc.org RtpRtcp::Configuration configuration; 932853dde5201d9ddc284ce5efb9688f6340487acepwestin@webrtc.org configuration.audio = true; 949691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com configuration.clock = &fake_clock_; 9586fd9ed6f9e2a38aa343db8c62764659633231fasprang configuration.outgoing_transport = &null_transport_; 969691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_.reset(RtpRtcp::CreateRtpRtcp(configuration)); 97822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org rtp_payload_registry_.reset(new RTPPayloadRegistry( 98dc80bae2a62a1bdbe0d342b3260a7e5b2cb958dfandresp@webrtc.org RTPPayloadStrategy::CreateStrategy(true))); 99822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org rtp_receiver_.reset(RtpReceiver::CreateAudioReceiver( 100ac547a653862744d0aae560713f8418ad2852085Peter Boström &fake_clock_, NULL, NULL, NULL, rtp_payload_registry_.get())); 1018281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org } 1028281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 10300b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RTPPayloadRegistry> rtp_payload_registry_; 10400b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpReceiver> rtp_receiver_; 10500b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpRtcp> module_; 1069691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint32_t test_ssrc_; 1079691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint32_t test_timestamp_; 1089691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint16_t test_sequence_number_; 1099691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com std::vector<uint32_t> test_csrcs_; 1109691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com SimulatedClock fake_clock_; 11186fd9ed6f9e2a38aa343db8c62764659633231fasprang test::NullTransport null_transport_; 1128281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org}; 1138281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1148281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, Basic) { 1159691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetSequenceNumber(test_sequence_number_); 1169691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_sequence_number_, module_->SequenceNumber()); 1178281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1189691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetStartTimestamp(test_timestamp_); 1199691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_timestamp_, module_->StartTimestamp()); 1208281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1219691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->Sending()); 1229691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetSendingStatus(true)); 1239691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_TRUE(module_->Sending()); 1248281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1258281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1268281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, MTU) { 1279691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetMaxTransferUnit(1234)); 1289691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 8, module_->MaxPayloadLength()); 1298281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1309691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetTransportOverhead(true, true, 12)); 1319691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 20 - 20 - 12, module_->MaxPayloadLength()); 1328281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1339691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetTransportOverhead(false, false, 0)); 1349691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 8, module_->MaxPayloadLength()); 1358281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1368281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1378281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, SSRC) { 1389691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetSSRC(test_ssrc_); 1399691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_ssrc_, module_->SSRC()); 1408281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1418281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1428281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, RTCP) { 143da903eaabbb6c6830efcafc3c2ade1d36f511e43pbos EXPECT_EQ(RtcpMode::kOff, module_->RTCP()); 144da903eaabbb6c6830efcafc3c2ade1d36f511e43pbos module_->SetRTCPStatus(RtcpMode::kCompound); 145da903eaabbb6c6830efcafc3c2ade1d36f511e43pbos EXPECT_EQ(RtcpMode::kCompound, module_->RTCP()); 1468281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1479691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetCNAME("john.doe@test.test")); 1488281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1499691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->TMMBR()); 1509691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetTMMBRStatus(true); 1519691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_TRUE(module_->TMMBR()); 1529691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetTMMBRStatus(false); 1539691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->TMMBR()); 1548281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 155822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org EXPECT_EQ(kNackOff, rtp_receiver_->NACK()); 1567bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtp_receiver_->SetNACKStatus(kNackRtcp); 157822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org EXPECT_EQ(kNackRtcp, rtp_receiver_->NACK()); 1588281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 159bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org 1607bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.orgTEST_F(RtpRtcpAPITest, RtxSender) { 1619691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxRetransmitted); 1629691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxRetransmitted, module_->RtxSendStatus()); 1630b0c24177bac6eaa27cd520595ba799e48e84a0cpbos@webrtc.org 1649691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxOff); 1659691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxOff, module_->RtxSendStatus()); 1660b0c24177bac6eaa27cd520595ba799e48e84a0cpbos@webrtc.org 1679691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxRetransmitted); 1689691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxRetransmitted, module_->RtxSendStatus()); 169bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org} 170bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org 1717bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.orgTEST_F(RtpRtcpAPITest, RtxReceiver) { 1727bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org const uint32_t kRtxSsrc = 1; 1739f5ebb525130f207229dfa350ce8c2bdd22163c7mflodman@webrtc.org const int kRtxPayloadType = 119; 174e62202fedf57b74cc263246c0586ee353978caf8Shao Changbin const int kPayloadType = 100; 175ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org EXPECT_FALSE(rtp_payload_registry_->RtxEnabled()); 176ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org rtp_payload_registry_->SetRtxSsrc(kRtxSsrc); 177e62202fedf57b74cc263246c0586ee353978caf8Shao Changbin rtp_payload_registry_->SetRtxPayloadType(kRtxPayloadType, kPayloadType); 1787bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->RtxEnabled()); 1797bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org RTPHeader rtx_header; 1807bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = kRtxSsrc; 1817bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.payloadType = kRtxPayloadType; 1827bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->IsRtx(rtx_header)); 1837bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = 0; 1847bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_FALSE(rtp_payload_registry_->IsRtx(rtx_header)); 1857bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = kRtxSsrc; 186ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org rtx_header.payloadType = 0; 187ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->IsRtx(rtx_header)); 188bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org} 1896a6f0893dd1e653410ba4b22e7f33947d15aeb65danilchap 1906a6f0893dd1e653410ba4b22e7f33947d15aeb65danilchap} // namespace webrtc 191