test_api.cc revision 2d566686a23fe93ada58f1c38a0d4b9a0d68556e
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 168281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgusing namespace webrtc; 178281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 189691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comnamespace webrtc { 199691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comvoid LoopBackTransport::SetSendModule(RtpRtcp* rtp_rtcp_module, 209691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RTPPayloadRegistry* payload_registry, 219691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RtpReceiver* receiver, 229691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com ReceiveStatistics* receive_statistics) { 239691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_rtcp_module_ = rtp_rtcp_module; 249691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_payload_registry_ = payload_registry; 259691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com rtp_receiver_ = receiver; 269691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com receive_statistics_ = receive_statistics; 279691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 289691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 299691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comvoid LoopBackTransport::DropEveryNthPacket(int n) { 309691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com packet_loss_ = n; 319691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 329691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 332d566686a23fe93ada58f1c38a0d4b9a0d68556epbosbool LoopBackTransport::SendRtp(const uint8_t* data, size_t len) { 349691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com count_++; 359691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (packet_loss_ > 0) { 369691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if ((count_ % packet_loss_) == 0) { 372d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 389691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 399691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 409691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RTPHeader header; 4100b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpHeaderParser> parser(RtpHeaderParser::Create()); 429691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!parser->Parse(static_cast<const uint8_t*>(data), len, &header)) { 432d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 449691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 459691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com PayloadUnion payload_specific; 469691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!rtp_payload_registry_->GetPayloadSpecifics(header.payloadType, 479691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com &payload_specific)) { 482d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 499691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 509691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com receive_statistics_->IncomingPacket(header, len, false); 519691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (!rtp_receiver_->IncomingRtpPacket(header, 529691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com static_cast<const uint8_t*>(data), len, 539691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com payload_specific, true)) { 542d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 559691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 562d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 579691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 589691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 592d566686a23fe93ada58f1c38a0d4b9a0d68556epbosbool LoopBackTransport::SendRtcp(const uint8_t* data, size_t len) { 609691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com if (rtp_rtcp_module_->IncomingRtcpPacket((const uint8_t*)data, len) < 0) { 612d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return false; 629691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com } 632d566686a23fe93ada58f1c38a0d4b9a0d68556epbos return true; 649691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 659691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 669691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.comint32_t TestRtpReceiver::OnReceivedPayloadData( 679691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const uint8_t* payload_data, 689691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const size_t payload_size, 699691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com const webrtc::WebRtcRTPHeader* rtp_header) { 709691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_LE(payload_size, sizeof(payload_data_)); 719691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com memcpy(payload_data_, payload_data, payload_size); 729691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com memcpy(&rtp_header_, rtp_header, sizeof(rtp_header_)); 739691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com payload_size_ = payload_size; 749691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com return 0; 759691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} 769691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com} // namespace webrtc 779691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com 788281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgclass RtpRtcpAPITest : public ::testing::Test { 798281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org protected: 809691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com RtpRtcpAPITest() : fake_clock_(123456) { 819691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_csrcs_.push_back(1234); 829691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_csrcs_.push_back(2345); 839691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_ssrc_ = 3456; 849691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_timestamp_ = 4567; 859691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com test_sequence_number_ = 2345; 868281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org } 878281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org ~RtpRtcpAPITest() {} 888281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 8914665ff7d4024d07e58622f498b23fd980001871kjellander@webrtc.org void SetUp() override { 902853dde5201d9ddc284ce5efb9688f6340487acepwestin@webrtc.org RtpRtcp::Configuration configuration; 912853dde5201d9ddc284ce5efb9688f6340487acepwestin@webrtc.org configuration.audio = true; 929691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com configuration.clock = &fake_clock_; 939691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_.reset(RtpRtcp::CreateRtpRtcp(configuration)); 94822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org rtp_payload_registry_.reset(new RTPPayloadRegistry( 95dc80bae2a62a1bdbe0d342b3260a7e5b2cb958dfandresp@webrtc.org RTPPayloadStrategy::CreateStrategy(true))); 96822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org rtp_receiver_.reset(RtpReceiver::CreateAudioReceiver( 97ac547a653862744d0aae560713f8418ad2852085Peter Boström &fake_clock_, NULL, NULL, NULL, rtp_payload_registry_.get())); 988281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org } 998281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 10000b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RTPPayloadRegistry> rtp_payload_registry_; 10100b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpReceiver> rtp_receiver_; 10200b8f6b3643332cce1ee711715f7fbb824d793cakwiberg@webrtc.org rtc::scoped_ptr<RtpRtcp> module_; 1039691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint32_t test_ssrc_; 1049691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint32_t test_timestamp_; 1059691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com uint16_t test_sequence_number_; 1069691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com std::vector<uint32_t> test_csrcs_; 1079691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com SimulatedClock fake_clock_; 1088281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org}; 1098281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1108281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, Basic) { 1119691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetSequenceNumber(test_sequence_number_); 1129691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_sequence_number_, module_->SequenceNumber()); 1138281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1149691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetStartTimestamp(test_timestamp_); 1159691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_timestamp_, module_->StartTimestamp()); 1168281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1179691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->Sending()); 1189691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetSendingStatus(true)); 1199691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_TRUE(module_->Sending()); 1208281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1218281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1228281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, MTU) { 1239691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetMaxTransferUnit(1234)); 1249691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 8, module_->MaxPayloadLength()); 1258281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1269691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetTransportOverhead(true, true, 12)); 1279691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 20 - 20 - 12, module_->MaxPayloadLength()); 1288281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1299691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetTransportOverhead(false, false, 0)); 1309691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(1234 - 20 - 8, module_->MaxPayloadLength()); 1318281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1328281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1338281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, SSRC) { 1349691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetSSRC(test_ssrc_); 1359691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(test_ssrc_, module_->SSRC()); 1368281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 1378281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1388281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.orgTEST_F(RtpRtcpAPITest, RTCP) { 1399691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtcpOff, module_->RTCP()); 1409691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRTCPStatus(kRtcpCompound); 1419691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtcpCompound, module_->RTCP()); 1428281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1439691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(0, module_->SetCNAME("john.doe@test.test")); 1448281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 1459691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->TMMBR()); 1469691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetTMMBRStatus(true); 1479691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_TRUE(module_->TMMBR()); 1489691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetTMMBRStatus(false); 1499691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_FALSE(module_->TMMBR()); 1508281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org 151822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org EXPECT_EQ(kNackOff, rtp_receiver_->NACK()); 1527bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtp_receiver_->SetNACKStatus(kNackRtcp); 153822fbd8b68ffdb481b9557e2950ae8d6657c8ce6wu@webrtc.org EXPECT_EQ(kNackRtcp, rtp_receiver_->NACK()); 1548281e7dd4a8ac4b00aededd8ebf52e93f1d980e8pwestin@webrtc.org} 155bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org 1567bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.orgTEST_F(RtpRtcpAPITest, RtxSender) { 1579691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxRetransmitted); 1589691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxRetransmitted, module_->RtxSendStatus()); 1590b0c24177bac6eaa27cd520595ba799e48e84a0cpbos@webrtc.org 1609691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxOff); 1619691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxOff, module_->RtxSendStatus()); 1620b0c24177bac6eaa27cd520595ba799e48e84a0cpbos@webrtc.org 1639691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com module_->SetRtxSendStatus(kRtxRetransmitted); 1649691b369951d8406b32bda2fb40667d55a3da96achangbin.shao@intel.com EXPECT_EQ(kRtxRetransmitted, module_->RtxSendStatus()); 165bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org} 166bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org 1677bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.orgTEST_F(RtpRtcpAPITest, RtxReceiver) { 1687bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org const uint32_t kRtxSsrc = 1; 1699f5ebb525130f207229dfa350ce8c2bdd22163c7mflodman@webrtc.org const int kRtxPayloadType = 119; 170e62202fedf57b74cc263246c0586ee353978caf8Shao Changbin const int kPayloadType = 100; 171ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org EXPECT_FALSE(rtp_payload_registry_->RtxEnabled()); 172ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org rtp_payload_registry_->SetRtxSsrc(kRtxSsrc); 173e62202fedf57b74cc263246c0586ee353978caf8Shao Changbin rtp_payload_registry_->SetRtxPayloadType(kRtxPayloadType, kPayloadType); 1747bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->RtxEnabled()); 1757bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org RTPHeader rtx_header; 1767bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = kRtxSsrc; 1777bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.payloadType = kRtxPayloadType; 1787bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->IsRtx(rtx_header)); 1797bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = 0; 1807bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org EXPECT_FALSE(rtp_payload_registry_->IsRtx(rtx_header)); 1817bb8f02274ecbfa1f7ef134d708369a369a78c83stefan@webrtc.org rtx_header.ssrc = kRtxSsrc; 182ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org rtx_header.payloadType = 0; 183ef92755780253c6a7940c89598a206e58e05b810stefan@webrtc.org EXPECT_TRUE(rtp_payload_registry_->IsRtx(rtx_header)); 184bda7f305c5d7d675f1c35813bd2b2a5732775bb9mikhal@webrtc.org} 185