mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 13:20:44 +01:00
303 lines
14 KiB
C++
303 lines
14 KiB
C++
/*
|
|
* Copyright (c) 2017 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 "api/rtp_parameters.h"
|
|
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
using webrtc::RtpExtension;
|
|
|
|
static const char kExtensionUri1[] = "extension-uri1";
|
|
static const char kExtensionUri2[] = "extension-uri2";
|
|
|
|
static const RtpExtension kExtension1(kExtensionUri1, 1);
|
|
static const RtpExtension kExtension1Encrypted(kExtensionUri1, 10, true);
|
|
static const RtpExtension kExtension2(kExtensionUri2, 2);
|
|
|
|
TEST(RtpExtensionTest, DeduplicateHeaderExtensions) {
|
|
std::vector<RtpExtension> extensions;
|
|
std::vector<RtpExtension> filtered;
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kDiscardEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kPreferEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1Encrypted}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kRequireEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1Encrypted}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kDiscardEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kPreferEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1Encrypted}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kRequireEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ(std::vector<RtpExtension>{kExtension1Encrypted}, filtered);
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension2);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kDiscardEncryptedExtension);
|
|
EXPECT_EQ(2u, filtered.size());
|
|
EXPECT_EQ(extensions, filtered);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kPreferEncryptedExtension);
|
|
EXPECT_EQ(2u, filtered.size());
|
|
EXPECT_EQ(extensions, filtered);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kRequireEncryptedExtension);
|
|
EXPECT_EQ(0u, filtered.size());
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension2);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kDiscardEncryptedExtension);
|
|
EXPECT_EQ(2u, filtered.size());
|
|
EXPECT_EQ((std::vector<RtpExtension>{kExtension1, kExtension2}), filtered);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kPreferEncryptedExtension);
|
|
EXPECT_EQ(2u, filtered.size());
|
|
EXPECT_EQ((std::vector<RtpExtension>{kExtension1Encrypted, kExtension2}),
|
|
filtered);
|
|
filtered = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kRequireEncryptedExtension);
|
|
EXPECT_EQ(1u, filtered.size());
|
|
EXPECT_EQ((std::vector<RtpExtension>{kExtension1Encrypted}), filtered);
|
|
}
|
|
|
|
// Test that the filtered vector is sorted so that for a given unsorted array of
|
|
// extensions, the filtered vector will always be laied out the same (for easy
|
|
// comparison).
|
|
TEST(RtpExtensionTest, DeduplicateHeaderExtensionsSorted) {
|
|
const std::vector<RtpExtension> extensions = {
|
|
RtpExtension("cde1", 11, false), RtpExtension("cde2", 12, true),
|
|
RtpExtension("abc1", 3, false), RtpExtension("abc2", 4, true),
|
|
RtpExtension("cde3", 9, true), RtpExtension("cde4", 10, false),
|
|
RtpExtension("abc3", 1, true), RtpExtension("abc4", 2, false),
|
|
RtpExtension("bcd3", 7, false), RtpExtension("bcd1", 8, true),
|
|
RtpExtension("bcd2", 5, true), RtpExtension("bcd4", 6, false),
|
|
};
|
|
|
|
auto encrypted = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kRequireEncryptedExtension);
|
|
|
|
const std::vector<RtpExtension> expected_sorted_encrypted = {
|
|
RtpExtension("abc2", 4, true), RtpExtension("abc3", 1, true),
|
|
RtpExtension("bcd1", 8, true), RtpExtension("bcd2", 5, true),
|
|
RtpExtension("cde2", 12, true), RtpExtension("cde3", 9, true)};
|
|
EXPECT_EQ(expected_sorted_encrypted, encrypted);
|
|
|
|
auto unencypted = RtpExtension::DeduplicateHeaderExtensions(
|
|
extensions, RtpExtension::Filter::kDiscardEncryptedExtension);
|
|
|
|
const std::vector<RtpExtension> expected_sorted_unencrypted = {
|
|
RtpExtension("abc1", 3, false), RtpExtension("abc4", 2, false),
|
|
RtpExtension("bcd3", 7, false), RtpExtension("bcd4", 6, false),
|
|
RtpExtension("cde1", 11, false), RtpExtension("cde4", 10, false)};
|
|
EXPECT_EQ(expected_sorted_unencrypted, unencypted);
|
|
}
|
|
|
|
TEST(RtpExtensionTest, FindHeaderExtensionByUriAndEncryption) {
|
|
std::vector<RtpExtension> extensions;
|
|
|
|
extensions.clear();
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri1, false));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri1, false));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri1, true));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri2, false));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension2);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri1, false));
|
|
EXPECT_EQ(kExtension2, *RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri2, false));
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri1, true));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUriAndEncryption(
|
|
extensions, kExtensionUri2, true));
|
|
}
|
|
|
|
TEST(RtpExtensionTest, FindHeaderExtensionByUri) {
|
|
std::vector<RtpExtension> extensions;
|
|
|
|
extensions.clear();
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1Encrypted);
|
|
extensions.push_back(kExtension1);
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension2);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
EXPECT_EQ(kExtension2, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(kExtension2, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
|
|
extensions.clear();
|
|
extensions.push_back(kExtension1);
|
|
extensions.push_back(kExtension2);
|
|
extensions.push_back(kExtension1Encrypted);
|
|
EXPECT_EQ(kExtension1, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(kExtension1Encrypted,
|
|
*RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri1,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
EXPECT_EQ(kExtension2, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kDiscardEncryptedExtension));
|
|
EXPECT_EQ(kExtension2, *RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kPreferEncryptedExtension));
|
|
EXPECT_EQ(nullptr, RtpExtension::FindHeaderExtensionByUri(
|
|
extensions, kExtensionUri2,
|
|
RtpExtension::Filter::kRequireEncryptedExtension));
|
|
}
|
|
} // namespace webrtc
|