1 /*
2 * Copyright (c) 2012 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/audio_coding/neteq/decoder_database.h"
12
13 #include <stdlib.h>
14
15 #include <string>
16
17 #include "api/audio_codecs/builtin_audio_decoder_factory.h"
18 #include "rtc_base/ref_counted_object.h"
19 #include "test/gmock.h"
20 #include "test/gtest.h"
21 #include "test/mock_audio_decoder.h"
22 #include "test/mock_audio_decoder_factory.h"
23
24 using ::testing::_;
25 using ::testing::Invoke;
26
27 namespace webrtc {
28
TEST(DecoderDatabase,CreateAndDestroy)29 TEST(DecoderDatabase, CreateAndDestroy) {
30 DecoderDatabase db(new rtc::RefCountedObject<MockAudioDecoderFactory>,
31 absl::nullopt);
32 EXPECT_EQ(0, db.Size());
33 EXPECT_TRUE(db.Empty());
34 }
35
TEST(DecoderDatabase,InsertAndRemove)36 TEST(DecoderDatabase, InsertAndRemove) {
37 rtc::scoped_refptr<MockAudioDecoderFactory> factory(
38 new rtc::RefCountedObject<MockAudioDecoderFactory>);
39 DecoderDatabase db(factory, absl::nullopt);
40 const uint8_t kPayloadType = 0;
41 const std::string kCodecName = "Robert\'); DROP TABLE Students;";
42 EXPECT_EQ(
43 DecoderDatabase::kOK,
44 db.RegisterPayload(kPayloadType, SdpAudioFormat(kCodecName, 8000, 1)));
45 EXPECT_EQ(1, db.Size());
46 EXPECT_FALSE(db.Empty());
47 EXPECT_EQ(DecoderDatabase::kOK, db.Remove(kPayloadType));
48 EXPECT_EQ(0, db.Size());
49 EXPECT_TRUE(db.Empty());
50 }
51
TEST(DecoderDatabase,InsertAndRemoveAll)52 TEST(DecoderDatabase, InsertAndRemoveAll) {
53 rtc::scoped_refptr<MockAudioDecoderFactory> factory(
54 new rtc::RefCountedObject<MockAudioDecoderFactory>);
55 DecoderDatabase db(factory, absl::nullopt);
56 const std::string kCodecName1 = "Robert\'); DROP TABLE Students;";
57 const std::string kCodecName2 = "https://xkcd.com/327/";
58 EXPECT_EQ(DecoderDatabase::kOK,
59 db.RegisterPayload(0, SdpAudioFormat(kCodecName1, 8000, 1)));
60 EXPECT_EQ(DecoderDatabase::kOK,
61 db.RegisterPayload(1, SdpAudioFormat(kCodecName2, 8000, 1)));
62 EXPECT_EQ(2, db.Size());
63 EXPECT_FALSE(db.Empty());
64 db.RemoveAll();
65 EXPECT_EQ(0, db.Size());
66 EXPECT_TRUE(db.Empty());
67 }
68
TEST(DecoderDatabase,GetDecoderInfo)69 TEST(DecoderDatabase, GetDecoderInfo) {
70 rtc::scoped_refptr<MockAudioDecoderFactory> factory(
71 new rtc::RefCountedObject<MockAudioDecoderFactory>);
72 auto* decoder = new MockAudioDecoder;
73 EXPECT_CALL(*factory, MakeAudioDecoderMock(_, _, _))
74 .WillOnce(Invoke([decoder](const SdpAudioFormat& format,
75 absl::optional<AudioCodecPairId> codec_pair_id,
76 std::unique_ptr<AudioDecoder>* dec) {
77 EXPECT_EQ("pcmu", format.name);
78 dec->reset(decoder);
79 }));
80 DecoderDatabase db(factory, absl::nullopt);
81 const uint8_t kPayloadType = 0;
82 const std::string kCodecName = "pcmu";
83 EXPECT_EQ(
84 DecoderDatabase::kOK,
85 db.RegisterPayload(kPayloadType, SdpAudioFormat(kCodecName, 8000, 1)));
86 const DecoderDatabase::DecoderInfo* info;
87 info = db.GetDecoderInfo(kPayloadType);
88 ASSERT_TRUE(info != NULL);
89 EXPECT_TRUE(info->IsType("pcmu"));
90 EXPECT_EQ(kCodecName, info->get_name());
91 EXPECT_EQ(decoder, db.GetDecoder(kPayloadType));
92 info = db.GetDecoderInfo(kPayloadType + 1); // Other payload type.
93 EXPECT_TRUE(info == NULL); // Should not be found.
94 }
95
TEST(DecoderDatabase,GetDecoder)96 TEST(DecoderDatabase, GetDecoder) {
97 DecoderDatabase db(CreateBuiltinAudioDecoderFactory(), absl::nullopt);
98 const uint8_t kPayloadType = 0;
99 EXPECT_EQ(DecoderDatabase::kOK,
100 db.RegisterPayload(kPayloadType, SdpAudioFormat("l16", 8000, 1)));
101 AudioDecoder* dec = db.GetDecoder(kPayloadType);
102 ASSERT_TRUE(dec != NULL);
103 }
104
TEST(DecoderDatabase,TypeTests)105 TEST(DecoderDatabase, TypeTests) {
106 rtc::scoped_refptr<MockAudioDecoderFactory> factory(
107 new rtc::RefCountedObject<MockAudioDecoderFactory>);
108 DecoderDatabase db(factory, absl::nullopt);
109 const uint8_t kPayloadTypePcmU = 0;
110 const uint8_t kPayloadTypeCng = 13;
111 const uint8_t kPayloadTypeDtmf = 100;
112 const uint8_t kPayloadTypeRed = 101;
113 const uint8_t kPayloadNotUsed = 102;
114 // Load into database.
115 EXPECT_EQ(
116 DecoderDatabase::kOK,
117 db.RegisterPayload(kPayloadTypePcmU, SdpAudioFormat("pcmu", 8000, 1)));
118 EXPECT_EQ(DecoderDatabase::kOK,
119 db.RegisterPayload(kPayloadTypeCng, SdpAudioFormat("cn", 8000, 1)));
120 EXPECT_EQ(DecoderDatabase::kOK,
121 db.RegisterPayload(kPayloadTypeDtmf,
122 SdpAudioFormat("telephone-event", 8000, 1)));
123 EXPECT_EQ(
124 DecoderDatabase::kOK,
125 db.RegisterPayload(kPayloadTypeRed, SdpAudioFormat("red", 8000, 1)));
126 EXPECT_EQ(4, db.Size());
127 // Test.
128 EXPECT_FALSE(db.IsComfortNoise(kPayloadNotUsed));
129 EXPECT_FALSE(db.IsDtmf(kPayloadNotUsed));
130 EXPECT_FALSE(db.IsRed(kPayloadNotUsed));
131 EXPECT_FALSE(db.IsComfortNoise(kPayloadTypePcmU));
132 EXPECT_FALSE(db.IsDtmf(kPayloadTypePcmU));
133 EXPECT_FALSE(db.IsRed(kPayloadTypePcmU));
134 EXPECT_FALSE(db.IsType(kPayloadTypePcmU, "isac"));
135 EXPECT_TRUE(db.IsType(kPayloadTypePcmU, "pcmu"));
136 EXPECT_TRUE(db.IsComfortNoise(kPayloadTypeCng));
137 EXPECT_TRUE(db.IsDtmf(kPayloadTypeDtmf));
138 EXPECT_TRUE(db.IsRed(kPayloadTypeRed));
139 }
140
TEST(DecoderDatabase,CheckPayloadTypes)141 TEST(DecoderDatabase, CheckPayloadTypes) {
142 constexpr int kNumPayloads = 10;
143 rtc::scoped_refptr<MockAudioDecoderFactory> factory(
144 new rtc::RefCountedObject<MockAudioDecoderFactory>);
145 DecoderDatabase db(factory, absl::nullopt);
146 // Load a number of payloads into the database. Payload types are 0, 1, ...,
147 // while the decoder type is the same for all payload types (this does not
148 // matter for the test).
149 for (uint8_t payload_type = 0; payload_type < kNumPayloads; ++payload_type) {
150 EXPECT_EQ(
151 DecoderDatabase::kOK,
152 db.RegisterPayload(payload_type, SdpAudioFormat("pcmu", 8000, 1)));
153 }
154 PacketList packet_list;
155 for (int i = 0; i < kNumPayloads + 1; ++i) {
156 // Create packet with payload type |i|. The last packet will have a payload
157 // type that is not registered in the decoder database.
158 Packet packet;
159 packet.payload_type = i;
160 packet_list.push_back(std::move(packet));
161 }
162
163 // Expect to return false, since the last packet is of an unknown type.
164 EXPECT_EQ(DecoderDatabase::kDecoderNotFound,
165 db.CheckPayloadTypes(packet_list));
166
167 packet_list.pop_back(); // Remove the unknown one.
168
169 EXPECT_EQ(DecoderDatabase::kOK, db.CheckPayloadTypes(packet_list));
170
171 // Delete all packets.
172 PacketList::iterator it = packet_list.begin();
173 while (it != packet_list.end()) {
174 it = packet_list.erase(it);
175 }
176 }
177
178 #if defined(WEBRTC_CODEC_ISAC) || defined(WEBRTC_CODEC_ISACFX)
179 #define IF_ISAC(x) x
180 #else
181 #define IF_ISAC(x) DISABLED_##x
182 #endif
183
184 // Test the methods for setting and getting active speech and CNG decoders.
TEST(DecoderDatabase,IF_ISAC (ActiveDecoders))185 TEST(DecoderDatabase, IF_ISAC(ActiveDecoders)) {
186 DecoderDatabase db(CreateBuiltinAudioDecoderFactory(), absl::nullopt);
187 // Load payload types.
188 ASSERT_EQ(DecoderDatabase::kOK,
189 db.RegisterPayload(0, SdpAudioFormat("pcmu", 8000, 1)));
190 ASSERT_EQ(DecoderDatabase::kOK,
191 db.RegisterPayload(103, SdpAudioFormat("isac", 16000, 1)));
192 ASSERT_EQ(DecoderDatabase::kOK,
193 db.RegisterPayload(13, SdpAudioFormat("cn", 8000, 1)));
194 // Verify that no decoders are active from the start.
195 EXPECT_EQ(NULL, db.GetActiveDecoder());
196 EXPECT_EQ(NULL, db.GetActiveCngDecoder());
197
198 // Set active speech codec.
199 bool changed; // Should be true when the active decoder changed.
200 EXPECT_EQ(DecoderDatabase::kOK, db.SetActiveDecoder(0, &changed));
201 EXPECT_TRUE(changed);
202 AudioDecoder* decoder = db.GetActiveDecoder();
203 ASSERT_FALSE(decoder == NULL); // Should get a decoder here.
204
205 // Set the same again. Expect no change.
206 EXPECT_EQ(DecoderDatabase::kOK, db.SetActiveDecoder(0, &changed));
207 EXPECT_FALSE(changed);
208 decoder = db.GetActiveDecoder();
209 ASSERT_FALSE(decoder == NULL); // Should get a decoder here.
210
211 // Change active decoder.
212 EXPECT_EQ(DecoderDatabase::kOK, db.SetActiveDecoder(103, &changed));
213 EXPECT_TRUE(changed);
214 decoder = db.GetActiveDecoder();
215 ASSERT_FALSE(decoder == NULL); // Should get a decoder here.
216
217 // Remove the active decoder, and verify that the active becomes NULL.
218 EXPECT_EQ(DecoderDatabase::kOK, db.Remove(103));
219 EXPECT_EQ(NULL, db.GetActiveDecoder());
220
221 // Set active CNG codec.
222 EXPECT_EQ(DecoderDatabase::kOK, db.SetActiveCngDecoder(13));
223 ComfortNoiseDecoder* cng = db.GetActiveCngDecoder();
224 ASSERT_FALSE(cng == NULL); // Should get a decoder here.
225
226 // Remove the active CNG decoder, and verify that the active becomes NULL.
227 EXPECT_EQ(DecoderDatabase::kOK, db.Remove(13));
228 EXPECT_EQ(NULL, db.GetActiveCngDecoder());
229
230 // Try to set non-existing codecs as active.
231 EXPECT_EQ(DecoderDatabase::kDecoderNotFound,
232 db.SetActiveDecoder(17, &changed));
233 EXPECT_EQ(DecoderDatabase::kDecoderNotFound, db.SetActiveCngDecoder(17));
234 }
235 } // namespace webrtc
236