tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

remb_unittest.cc (4672B)

      1 /*
      2 *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
      3 *
      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
      6 *  tree. An additional intellectual property rights grant can be found
      7 *  in the file PATENTS.  All contributing project authors may
      8 *  be found in the AUTHORS file in the root of the source tree.
      9 */
     10 
     11 #include "modules/rtp_rtcp/source/rtcp_packet/remb.h"
     12 
     13 #include <cstddef>
     14 #include <cstdint>
     15 #include <cstring>
     16 #include <iterator>
     17 #include <vector>
     18 
     19 #include "rtc_base/buffer.h"
     20 #include "test/gmock.h"
     21 #include "test/gtest.h"
     22 #include "test/rtcp_packet_parser.h"
     23 
     24 using ::testing::ElementsAreArray;
     25 using ::testing::IsEmpty;
     26 using ::testing::make_tuple;
     27 using webrtc::rtcp::Remb;
     28 
     29 namespace webrtc {
     30 namespace {
     31 constexpr uint32_t kSenderSsrc = 0x12345678;
     32 constexpr uint32_t kRemoteSsrcs[] = {0x23456789, 0x2345678a, 0x2345678b};
     33 constexpr uint32_t kBitrateBps = 0x3fb93 * 2;  // 522022;
     34 constexpr int64_t kBitrateBps64bit = int64_t{0x3fb93} << 30;
     35 constexpr uint8_t kPacket[] = {0x8f, 206,  0x00, 0x07, 0x12, 0x34, 0x56, 0x78,
     36                               0x00, 0x00, 0x00, 0x00, 'R',  'E',  'M',  'B',
     37                               0x03, 0x07, 0xfb, 0x93, 0x23, 0x45, 0x67, 0x89,
     38                               0x23, 0x45, 0x67, 0x8a, 0x23, 0x45, 0x67, 0x8b};
     39 constexpr size_t kPacketLength = sizeof(kPacket);
     40 }  // namespace
     41 
     42 TEST(RtcpPacketRembTest, Create) {
     43  Remb remb;
     44  remb.SetSenderSsrc(kSenderSsrc);
     45  remb.SetSsrcs(
     46      std::vector<uint32_t>(std::begin(kRemoteSsrcs), std::end(kRemoteSsrcs)));
     47  remb.SetBitrateBps(kBitrateBps);
     48 
     49  Buffer packet = remb.Build();
     50 
     51  EXPECT_THAT(make_tuple(packet.data(), packet.size()),
     52              ElementsAreArray(kPacket));
     53 }
     54 
     55 TEST(RtcpPacketRembTest, Parse) {
     56  Remb remb;
     57  EXPECT_TRUE(test::ParseSinglePacket(kPacket, &remb));
     58  const Remb& parsed = remb;
     59 
     60  EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc());
     61  EXPECT_EQ(kBitrateBps, parsed.bitrate_bps());
     62  EXPECT_THAT(parsed.ssrcs(), ElementsAreArray(kRemoteSsrcs));
     63 }
     64 
     65 TEST(RtcpPacketRembTest, CreateAndParseWithoutSsrcs) {
     66  Remb remb;
     67  remb.SetSenderSsrc(kSenderSsrc);
     68  remb.SetBitrateBps(kBitrateBps);
     69  Buffer packet = remb.Build();
     70 
     71  Remb parsed;
     72  EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed));
     73  EXPECT_EQ(kSenderSsrc, parsed.sender_ssrc());
     74  EXPECT_EQ(kBitrateBps, parsed.bitrate_bps());
     75  EXPECT_THAT(parsed.ssrcs(), IsEmpty());
     76 }
     77 
     78 TEST(RtcpPacketRembTest, CreateAndParse64bitBitrate) {
     79  Remb remb;
     80  remb.SetBitrateBps(kBitrateBps64bit);
     81  Buffer packet = remb.Build();
     82 
     83  Remb parsed;
     84  EXPECT_TRUE(test::ParseSinglePacket(packet, &parsed));
     85  EXPECT_EQ(kBitrateBps64bit, parsed.bitrate_bps());
     86 }
     87 
     88 TEST(RtcpPacketRembTest, ParseFailsOnTooSmallPacketToBeRemb) {
     89  // Make it too small.
     90  constexpr size_t kTooSmallSize = (1 + 3) * 4;
     91  uint8_t packet[kTooSmallSize];
     92  memcpy(packet, kPacket, kTooSmallSize);
     93  packet[3] = 3;
     94 
     95  Remb remb;
     96  EXPECT_FALSE(test::ParseSinglePacket(packet, &remb));
     97 }
     98 
     99 TEST(RtcpPacketRembTest, ParseFailsWhenUniqueIdentifierIsNotRemb) {
    100  uint8_t packet[kPacketLength];
    101  memcpy(packet, kPacket, kPacketLength);
    102  packet[12] = 'N';  // Swap 'R' -> 'N' in the 'REMB' unique identifier.
    103 
    104  Remb remb;
    105  EXPECT_FALSE(test::ParseSinglePacket(packet, &remb));
    106 }
    107 
    108 TEST(RtcpPacketRembTest, ParseFailsWhenBitrateDoNotFitIn64bits) {
    109  uint8_t packet[kPacketLength];
    110  memcpy(packet, kPacket, kPacketLength);
    111  packet[17] |= 0xfc;  // Set exponent component to maximum of 63.
    112  packet[19] |= 0x02;  // Ensure mantissa is at least 2.
    113 
    114  Remb remb;
    115  EXPECT_FALSE(test::ParseSinglePacket(packet, &remb));
    116 }
    117 
    118 TEST(RtcpPacketRembTest, ParseFailsWhenBitrateDoNotFitIn63bits) {
    119  uint8_t packet[kPacketLength];
    120  memcpy(packet, kPacket, kPacketLength);
    121  packet[17] = 56 << 2;  // Set exponent component to 56.
    122  packet[18] = 0;        // Set mantissa to 200 > 128
    123  packet[19] = 200;
    124 
    125  // Result value 200 * 2^56 can't be represented with int64_t and thus should
    126  // be rejected.
    127  Remb remb;
    128  EXPECT_FALSE(test::ParseSinglePacket(packet, &remb));
    129 }
    130 
    131 TEST(RtcpPacketRembTest, ParseFailsWhenSsrcCountMismatchLength) {
    132  uint8_t packet[kPacketLength];
    133  memcpy(packet, kPacket, kPacketLength);
    134  packet[16]++;  // Swap 3 -> 4 in the ssrcs count.
    135 
    136  Remb remb;
    137  EXPECT_FALSE(test::ParseSinglePacket(packet, &remb));
    138 }
    139 
    140 TEST(RtcpPacketRembTest, TooManySsrcs) {
    141  Remb remb;
    142  EXPECT_FALSE(remb.SetSsrcs(
    143      std::vector<uint32_t>(Remb::kMaxNumberOfSsrcs + 1, kRemoteSsrcs[0])));
    144  EXPECT_TRUE(remb.SetSsrcs(
    145      std::vector<uint32_t>(Remb::kMaxNumberOfSsrcs, kRemoteSsrcs[0])));
    146 }
    147 
    148 }  // namespace webrtc