b29e652b10
Problem fixed: RTP header extensions were not properly set in tests. BUG=webrtc:5654 Review-Url: https://codereview.webrtc.org/2593963003 Cr-Commit-Position: refs/heads/master@{#15741}
113 lines
4.5 KiB
C++
113 lines
4.5 KiB
C++
/*
|
|
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "webrtc/base/array_view.h"
|
|
#include "webrtc/base/basictypes.h"
|
|
#include "webrtc/call/flexfec_receive_stream_impl.h"
|
|
#include "webrtc/modules/rtp_rtcp/include/flexfec_receiver.h"
|
|
#include "webrtc/modules/rtp_rtcp/source/byte_io.h"
|
|
#include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h"
|
|
#include "webrtc/modules/rtp_rtcp/mocks/mock_recovered_packet_receiver.h"
|
|
#include "webrtc/test/gmock.h"
|
|
#include "webrtc/test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
RtpPacketReceived ParsePacket(rtc::ArrayView<const uint8_t> packet) {
|
|
RtpPacketReceived parsed_packet(nullptr);
|
|
EXPECT_TRUE(parsed_packet.Parse(packet));
|
|
return parsed_packet;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(FlexfecReceiveStreamTest, ConstructDestruct) {
|
|
FlexfecReceiveStream::Config config;
|
|
config.payload_type = 118;
|
|
config.remote_ssrc = 424223;
|
|
config.protected_media_ssrcs = {912512};
|
|
MockRecoveredPacketReceiver recovered_packet_receiver;
|
|
|
|
FlexfecReceiveStreamImpl receive_stream(config, &recovered_packet_receiver);
|
|
}
|
|
|
|
TEST(FlexfecReceiveStreamTest, StartStop) {
|
|
FlexfecReceiveStream::Config config;
|
|
config.payload_type = 118;
|
|
config.remote_ssrc = 1652392;
|
|
config.protected_media_ssrcs = {23300443};
|
|
MockRecoveredPacketReceiver recovered_packet_receiver;
|
|
FlexfecReceiveStreamImpl receive_stream(config, &recovered_packet_receiver);
|
|
|
|
receive_stream.Start();
|
|
receive_stream.Stop();
|
|
}
|
|
|
|
// Create a FlexFEC packet that protects a single media packet and ensure
|
|
// that the callback is called. Correctness of recovery is checked in the
|
|
// FlexfecReceiver unit tests.
|
|
TEST(FlexfecReceiveStreamTest, RecoversPacketWhenStarted) {
|
|
constexpr uint8_t kFlexfecPlType = 118;
|
|
constexpr uint8_t kFlexfecSeqNum[] = {0x00, 0x01};
|
|
constexpr uint8_t kFlexfecTs[] = {0x00, 0x11, 0x22, 0x33};
|
|
constexpr uint8_t kFlexfecSsrc[] = {0x00, 0x00, 0x00, 0x01};
|
|
constexpr uint8_t kMediaPlType = 107;
|
|
constexpr uint8_t kMediaSeqNum[] = {0x00, 0x02};
|
|
constexpr uint8_t kMediaTs[] = {0xaa, 0xbb, 0xcc, 0xdd};
|
|
constexpr uint8_t kMediaSsrc[] = {0x00, 0x00, 0x00, 0x02};
|
|
|
|
// This packet mask protects a single media packet, i.e., the FlexFEC payload
|
|
// is a copy of that media packet. When inserted in the FlexFEC pipeline,
|
|
// it will thus trivially recover the lost media packet.
|
|
constexpr uint8_t kKBit0 = 1 << 7;
|
|
constexpr uint8_t kFlexfecPktMask[] = {kKBit0 | 0x00, 0x01};
|
|
constexpr uint8_t kPayloadLength[] = {0x00, 0x04};
|
|
constexpr uint8_t kSsrcCount = 1;
|
|
constexpr uint8_t kReservedBits = 0x00;
|
|
constexpr uint8_t kPayloadBits = 0x00;
|
|
// clang-format off
|
|
constexpr uint8_t kFlexfecPacket[] = {
|
|
// RTP header.
|
|
0x80, kFlexfecPlType, kFlexfecSeqNum[0], kFlexfecSeqNum[1],
|
|
kFlexfecTs[0], kFlexfecTs[1], kFlexfecTs[2], kFlexfecTs[3],
|
|
kFlexfecSsrc[0], kFlexfecSsrc[1], kFlexfecSsrc[2], kFlexfecSsrc[3],
|
|
// FlexFEC header.
|
|
0x00, kMediaPlType, kPayloadLength[0], kPayloadLength[1],
|
|
kMediaTs[0], kMediaTs[1], kMediaTs[2], kMediaTs[3],
|
|
kSsrcCount, kReservedBits, kReservedBits, kReservedBits,
|
|
kMediaSsrc[0], kMediaSsrc[1], kMediaSsrc[2], kMediaSsrc[3],
|
|
kMediaSeqNum[0], kMediaSeqNum[1], kFlexfecPktMask[0], kFlexfecPktMask[1],
|
|
// FEC payload.
|
|
kPayloadBits, kPayloadBits, kPayloadBits, kPayloadBits};
|
|
// clang-format on
|
|
|
|
FlexfecReceiveStream::Config config;
|
|
config.payload_type = kFlexfecPlType;
|
|
config.remote_ssrc = ByteReader<uint32_t>::ReadBigEndian(kFlexfecSsrc);
|
|
config.protected_media_ssrcs = {
|
|
ByteReader<uint32_t>::ReadBigEndian(kMediaSsrc)};
|
|
testing::StrictMock<MockRecoveredPacketReceiver> recovered_packet_receiver;
|
|
FlexfecReceiveStreamImpl receive_stream(config, &recovered_packet_receiver);
|
|
|
|
// Do not call back before being started.
|
|
receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
|
|
|
|
// Call back after being started.
|
|
receive_stream.Start();
|
|
EXPECT_CALL(
|
|
recovered_packet_receiver,
|
|
OnRecoveredPacket(::testing::_, kRtpHeaderSize + kPayloadLength[1]));
|
|
receive_stream.AddAndProcessReceivedPacket(ParsePacket(kFlexfecPacket));
|
|
}
|
|
|
|
} // namespace webrtc
|