[go: nahoru, domu]
Google Git
Sign in
chromium / chromium / src / 7c7e28ccfd2c3ae4b19378db40f323a1c13cc3cf / . / media / filters / decrypting_audio_decoder_unittest.cc
blob: 926d869ecf2070a7e65ca9d1dab922f747fde5a3 [file] [log] [blame]
Avi Drissmand387f0922022-09-14 20:51:31[diff] [blame]1// Copyright 2012 The Chromium Authors
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
Daniel Cheng9e37dfb2022-02-28 07:30:12[diff] [blame]5#include "media/filters/decrypting_audio_decoder.h"
6
avi1323b9c22015-12-23 06:22:36[diff] [blame]7#include <stdint.h>
8
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]9#include <string>
10#include <vector>
11
Avi Drissmand70f89a2023-01-11 23:52:55[diff] [blame]12#include "base/functional/bind.h"
13#include "base/functional/callback_helpers.h"
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]14#include "base/run_loop.h"
Marijn Kruisselbrink52a35242019-06-05 22:58:29[diff] [blame]15#include "base/test/gmock_callback_support.h"
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]16#include "base/test/gmock_move_support.h"
Gabriel Charettec7108742019-08-23 03:31:40[diff] [blame]17#include "base/test/task_environment.h"
jrummell@chromium.org47b37a62013-07-10 03:56:10[diff] [blame]18#include "media/base/audio_buffer.h"
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]19#include "media/base/decoder_buffer.h"
20#include "media/base/decrypt_config.h"
chcunningham9812dd82015-10-20 01:42:09[diff] [blame]21#include "media/base/media_util.h"
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]22#include "media/base/mock_filters.h"
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]23#include "media/base/mock_media_log.h"
scherkus@chromium.org1848da02012-12-06 06:58:38[diff] [blame]24#include "media/base/test_helpers.h"
watk9f9dfdc92015-09-04 21:33:29[diff] [blame]25#include "media/base/timestamp_constants.h"
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]26#include "testing/gmock/include/gmock/gmock.h"
27
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]28using ::base::test::RunOnceCallback;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]29using ::testing::_;
30using ::testing::AtMost;
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]31using ::testing::HasSubstr;
xhwang00f69db72015-11-12 07:32:37[diff] [blame]32using ::testing::Return;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]33using ::testing::StrictMock;
34
35namespace media {
36
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]37const int kSampleRate = 44100;
rileya@chromium.org16ecd5f2014-03-20 05:41:07[diff] [blame]38
xhwang@chromium.orgecf16912013-01-12 16:55:22[diff] [blame]39// Make sure the kFakeAudioFrameSize is a valid frame size for all audio decoder
40// configs used in this test.
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]41const int kFakeAudioFrameSize = 48;
Avi Drissman97785ea2015-12-19 01:11:31[diff] [blame]42const uint8_t kFakeKeyId[] = {0x4b, 0x65, 0x79, 0x20, 0x49, 0x44};
43const uint8_t kFakeIv[DecryptConfig::kDecryptionKeySize] = {0};
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]44const int kDecodingDelay = 3;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]45
46// Create a fake non-empty encrypted buffer.
47static scoped_refptr<DecoderBuffer> CreateFakeEncryptedBuffer() {
48 const int buffer_size = 16; // Need a non-empty buffer;
Vikram Pasupathy7c7e28c2023-04-14 23:30:21[diff] [blame^]49 scoped_refptr<DecoderBuffer> buffer(
50 base::MakeRefCounted<DecoderBuffer>(buffer_size));
John Rummell9044a762018-04-19 01:24:07[diff] [blame]51 buffer->set_decrypt_config(DecryptConfig::CreateCencConfig(
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]52 std::string(reinterpret_cast<const char*>(kFakeKeyId),
Daniel Cheng9e37dfb2022-02-28 07:30:12[diff] [blame]53 std::size(kFakeKeyId)),
54 std::string(reinterpret_cast<const char*>(kFakeIv), std::size(kFakeIv)),
John Rummell9044a762018-04-19 01:24:07[diff] [blame]55 std::vector<SubsampleEntry>()));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]56 return buffer;
57}
58
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]59class DecryptingAudioDecoderTest : public testing::Test {
60 public:
61 DecryptingAudioDecoderTest()
Vikram Pasupathy7c7e28c2023-04-14 23:30:21[diff] [blame^]62 : decoder_(std::make_unique<DecryptingAudioDecoder>(
Gabriel Charettedfa36042019-08-19 17:30:11[diff] [blame]63 task_environment_.GetMainThreadTaskRunner(),
Carlos Caballero9de7c722019-05-31 21:25:07[diff] [blame]64 &media_log_)),
Vikram Pasupathy7c7e28c2023-04-14 23:30:21[diff] [blame^]65 cdm_context_(std::make_unique<StrictMock<MockCdmContext>>()),
66 decryptor_(std::make_unique<StrictMock<MockDecryptor>>()),
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]67 num_decrypt_and_decode_calls_(0),
68 num_frames_in_decryptor_(0),
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]69 encrypted_buffer_(CreateFakeEncryptedBuffer()),
kylechar938dc442019-10-29 14:40:24[diff] [blame]70 decoded_frame_(nullptr),
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]71 decoded_frame_list_() {}
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]72
Peter Boström4c42559f2021-09-22 20:24:34[diff] [blame]73 DecryptingAudioDecoderTest(const DecryptingAudioDecoderTest&) = delete;
74 DecryptingAudioDecoderTest& operator=(const DecryptingAudioDecoderTest&) =
75 delete;
76
Daniel Cheng9f0b7012018-04-26 21:09:05[diff] [blame]77 ~DecryptingAudioDecoderTest() override { Destroy(); }
rileya@chromium.orgb960fd02014-01-23 04:48:54[diff] [blame]78
xhwang59c321f2015-06-05 23:46:55[diff] [blame]79 void InitializeAndExpectResult(const AudioDecoderConfig& config,
80 bool success) {
dalecurtis@google.come81f5f9d2014-03-29 01:32:16[diff] [blame]81 // Initialize data now that the config is known. Since the code uses
82 // invalid values (that CreateEmptyBuffer() doesn't support), tweak them
83 // just for CreateEmptyBuffer().
84 int channels = ChannelLayoutToChannelCount(config.channel_layout());
85 if (channels < 0)
86 channels = 0;
dalecurtis39a7f932016-07-19 18:34:59[diff] [blame]87 decoded_frame_ = AudioBuffer::CreateEmptyBuffer(
88 config.channel_layout(), channels, kSampleRate, kFakeAudioFrameSize,
89 kNoTimestamp);
jrummell@chromium.org47b37a62013-07-10 03:56:10[diff] [blame]90 decoded_frame_list_.push_back(decoded_frame_);
91
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]92 decoder_->Initialize(
93 config, cdm_context_.get(),
94 base::BindOnce(
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]95 [](bool success, DecoderStatus status) {
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]96 EXPECT_EQ(status.is_ok(), success);
97 },
98 success),
99 base::BindRepeating(&DecryptingAudioDecoderTest::FrameReady,
100 base::Unretained(this)),
101 base::BindRepeating(&DecryptingAudioDecoderTest::OnWaiting,
102 base::Unretained(this)));
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]103 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]104 }
105
xhwanga935e442016-02-11 02:22:45[diff] [blame]106 enum CdmType { CDM_WITHOUT_DECRYPTOR, CDM_WITH_DECRYPTOR };
xhwang00f69db72015-11-12 07:32:37[diff] [blame]107
108 void SetCdmType(CdmType cdm_type) {
xhwang00f69db72015-11-12 07:32:37[diff] [blame]109 const bool has_decryptor = cdm_type == CDM_WITH_DECRYPTOR;
xhwanga935e442016-02-11 02:22:45[diff] [blame]110 EXPECT_CALL(*cdm_context_, GetDecryptor())
111 .WillRepeatedly(Return(has_decryptor ? decryptor_.get() : nullptr));
jrummell@chromium.org9ebc3b032014-08-13 04:01:23[diff] [blame]112 }
113
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]114 void Initialize() {
xhwang00f69db72015-11-12 07:32:37[diff] [blame]115 SetCdmType(CDM_WITH_DECRYPTOR);
xhwang@chromium.orga3e28672013-03-14 14:54:59[diff] [blame]116 EXPECT_CALL(*decryptor_, InitializeAudioDecoder(_, _))
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]117 .Times(AtMost(1))
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]118 .WillOnce(RunOnceCallback<1>(true));
Xiaohan Wangd72ce5d2020-07-14 01:34:43[diff] [blame]119 EXPECT_CALL(*cdm_context_, RegisterEventCB(_)).WillOnce([&](auto cb) {
120 event_cb_ = cb;
121 return std::make_unique<CallbackRegistration>();
122 });
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]123
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]124 config_.Initialize(AudioCodec::kVorbis, kSampleFormatPlanarF32,
chcunningham9812dd82015-10-20 01:42:09[diff] [blame]125 CHANNEL_LAYOUT_STEREO, kSampleRate, EmptyExtraData(),
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]126 EncryptionScheme::kCenc, base::TimeDelta(), 0);
xhwang59c321f2015-06-05 23:46:55[diff] [blame]127 InitializeAndExpectResult(config_, true);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]128 }
129
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]130 void Reinitialize() { ReinitializeConfigChange(config_); }
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]131
dalecurtis@google.com6fb86542014-04-18 19:58:13[diff] [blame]132 void ReinitializeConfigChange(const AudioDecoderConfig& new_config) {
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]133 EXPECT_CALL(*decryptor_, DeinitializeDecoder(Decryptor::kAudio));
134 EXPECT_CALL(*decryptor_, InitializeAudioDecoder(_, _))
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]135 .WillOnce(RunOnceCallback<1>(true));
Ted Meyer2a41e242020-03-20 08:01:27[diff] [blame]136 decoder_->Initialize(
137 new_config, cdm_context_.get(),
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]138 base::BindOnce(
139 [](DecoderStatus status) { EXPECT_TRUE(status.is_ok()); }),
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]140 base::BindRepeating(&DecryptingAudioDecoderTest::FrameReady,
141 base::Unretained(this)),
142 base::BindRepeating(&DecryptingAudioDecoderTest::OnWaiting,
143 base::Unretained(this)));
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]144 }
145
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]146 // Decode |buffer| and expect DecodeDone to get called with |status|.
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]147 void DecodeAndExpect(scoped_refptr<DecoderBuffer> buffer,
148 DecoderStatus status) {
Frank Liberato3d157ba2020-09-24 18:19:30[diff] [blame]149 EXPECT_CALL(*this, DecodeDone(HasStatusCode(status)));
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]150 decoder_->Decode(buffer,
151 base::BindOnce(&DecryptingAudioDecoderTest::DecodeDone,
152 base::Unretained(this)));
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]153 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]154 }
155
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]156 // Helper function to simulate the decrypting and decoding process in the
157 // |decryptor_| with a decoding delay of kDecodingDelay buffers.
Dale Curtisca8ad982018-04-09 20:23:14[diff] [blame]158 void DecryptAndDecodeAudio(scoped_refptr<DecoderBuffer> encrypted,
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]159 Decryptor::AudioDecodeCB audio_decode_cb) {
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]160 num_decrypt_and_decode_calls_++;
161 if (!encrypted->end_of_stream())
162 num_frames_in_decryptor_++;
163
164 if (num_decrypt_and_decode_calls_ <= kDecodingDelay ||
165 num_frames_in_decryptor_ == 0) {
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]166 std::move(audio_decode_cb)
167 .Run(Decryptor::kNeedMoreData, Decryptor::AudioFrames());
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]168 return;
169 }
170
171 num_frames_in_decryptor_--;
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]172 std::move(audio_decode_cb)
173 .Run(Decryptor::kSuccess, Decryptor::AudioFrames(1, decoded_frame_));
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]174 }
175
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]176 // Sets up expectations and actions to put DecryptingAudioDecoder in an
177 // active normal decoding state.
178 void EnterNormalDecodingState() {
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]179 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
180 .WillRepeatedly(
181 Invoke(this, &DecryptingAudioDecoderTest::DecryptAndDecodeAudio));
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]182 EXPECT_CALL(*this, FrameReady(decoded_frame_));
183 for (int i = 0; i < kDecodingDelay + 1; ++i)
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]184 DecodeAndExpect(encrypted_buffer_, DecoderStatus::Codes::kOk);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]185 }
186
187 // Sets up expectations and actions to put DecryptingAudioDecoder in an end
188 // of stream state. This function must be called after
189 // EnterNormalDecodingState() to work.
190 void EnterEndOfStreamState() {
xhwang@chromium.orgff757032014-06-19 01:42:24[diff] [blame]191 // The codec in the |decryptor_| will be flushed.
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]192 EXPECT_CALL(*this, FrameReady(decoded_frame_)).Times(kDecodingDelay);
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]193 DecodeAndExpect(DecoderBuffer::CreateEOSBuffer(),
194 DecoderStatus::Codes::kOk);
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]195 EXPECT_EQ(0, num_frames_in_decryptor_);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]196 }
197
198 // Make the audio decode callback pending by saving and not firing it.
199 void EnterPendingDecodeState() {
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]200 EXPECT_TRUE(!pending_audio_decode_cb_);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]201 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(encrypted_buffer_, _))
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]202 .WillOnce(MoveArg<1>(&pending_audio_decode_cb_));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]203
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]204 decoder_->Decode(encrypted_buffer_,
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]205 base::BindOnce(&DecryptingAudioDecoderTest::DecodeDone,
206 base::Unretained(this)));
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]207 base::RunLoop().RunUntilIdle();
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]208 // Make sure the Decode() on the decoder triggers a DecryptAndDecode() on
209 // the decryptor.
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]210 EXPECT_FALSE(!pending_audio_decode_cb_);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]211 }
212
213 void EnterWaitingForKeyState() {
xhwang@chromium.org2b454ff82012-11-15 03:57:58[diff] [blame]214 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(encrypted_buffer_, _))
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]215 .WillRepeatedly(
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]216 RunOnceCallback<1>(Decryptor::kNoKey, Decryptor::AudioFrames()));
Xiaohan Wang640b41d2018-12-18 19:00:46[diff] [blame]217 EXPECT_CALL(*this, OnWaiting(WaitingReason::kNoDecryptionKey));
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]218 decoder_->Decode(encrypted_buffer_,
David Bienvenub8c87522020-11-11 22:59:50[diff] [blame]219 base::BindOnce(&DecryptingAudioDecoderTest::DecodeDone,
220 base::Unretained(this)));
Xiaohan Wang689cd4282017-08-04 07:54:55[diff] [blame]221
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]222 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]223 }
224
225 void AbortPendingAudioDecodeCB() {
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]226 if (pending_audio_decode_cb_) {
227 std::move(pending_audio_decode_cb_)
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]228 .Run(Decryptor::kSuccess, Decryptor::AudioFrames());
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]229 }
230 }
231
rileya@chromium.orgb960fd02014-01-23 04:48:54[diff] [blame]232 void AbortAllPendingCBs() {
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]233 if (pending_init_cb_) {
234 ASSERT_TRUE(!pending_audio_decode_cb_);
235 std::move(pending_init_cb_).Run(false);
rileya@chromium.orgb960fd02014-01-23 04:48:54[diff] [blame]236 return;
237 }
238
239 AbortPendingAudioDecodeCB();
240 }
241
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]242 void Reset() {
243 EXPECT_CALL(*decryptor_, ResetDecoder(Decryptor::kAudio))
244 .WillRepeatedly(InvokeWithoutArgs(
245 this, &DecryptingAudioDecoderTest::AbortPendingAudioDecodeCB));
246
247 decoder_->Reset(NewExpectedClosure());
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]248 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]249 }
250
xhwang@chromium.org83fb2a9e2014-07-16 23:36:41[diff] [blame]251 void Destroy() {
rileya@chromium.orgb960fd02014-01-23 04:48:54[diff] [blame]252 EXPECT_CALL(*decryptor_, DeinitializeDecoder(Decryptor::kAudio))
253 .WillRepeatedly(InvokeWithoutArgs(
254 this, &DecryptingAudioDecoderTest::AbortAllPendingCBs));
255
xhwang@chromium.org83fb2a9e2014-07-16 23:36:41[diff] [blame]256 decoder_.reset();
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]257 base::RunLoop().RunUntilIdle();
rileya@chromium.orgb960fd02014-01-23 04:48:54[diff] [blame]258 }
259
John Rummell1b2b63d2019-04-26 20:45:57[diff] [blame]260 MOCK_METHOD1(FrameReady, void(scoped_refptr<AudioBuffer>));
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]261 MOCK_METHOD1(DecodeDone, void(DecoderStatus));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]262
Xiaohan Wang640b41d2018-12-18 19:00:46[diff] [blame]263 MOCK_METHOD1(OnWaiting, void(WaitingReason));
jrummell74fc4f942015-03-02 22:48:27[diff] [blame]264
Gabriel Charette311b0e92019-09-09 20:01:25[diff] [blame]265 base::test::SingleThreadTaskEnvironment task_environment_;
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]266 StrictMock<MockMediaLog> media_log_;
dcheng254c5362016-04-22 01:12:48[diff] [blame]267 std::unique_ptr<DecryptingAudioDecoder> decoder_;
268 std::unique_ptr<StrictMock<MockCdmContext>> cdm_context_;
269 std::unique_ptr<StrictMock<MockDecryptor>> decryptor_;
xhwang@chromium.org7c39486c2013-01-06 15:36:09[diff] [blame]270 AudioDecoderConfig config_;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]271
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]272 // Variables to help the |decryptor_| to simulate decoding delay and flushing.
273 int num_decrypt_and_decode_calls_;
274 int num_frames_in_decryptor_;
275
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]276 Decryptor::DecoderInitCB pending_init_cb_;
Xiaohan Wangd72ce5d2020-07-14 01:34:43[diff] [blame]277 CdmContext::EventCB event_cb_;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]278 Decryptor::AudioDecodeCB pending_audio_decode_cb_;
279
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]280 // Constant buffer/frames, to be used/returned by |decoder_| and |decryptor_|.
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]281 scoped_refptr<DecoderBuffer> encrypted_buffer_;
jrummell@chromium.org47b37a62013-07-10 03:56:10[diff] [blame]282 scoped_refptr<AudioBuffer> decoded_frame_;
anujk.sharmaf1cb600a2015-01-07 19:11:13[diff] [blame]283 Decryptor::AudioFrames decoded_frame_list_;
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]284};
285
286TEST_F(DecryptingAudioDecoderTest, Initialize_Normal) {
287 Initialize();
288}
289
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]290// Ensure decoder handles invalid audio configs without crashing.
291TEST_F(DecryptingAudioDecoderTest, Initialize_InvalidAudioConfig) {
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]292 AudioDecoderConfig config(AudioCodec::kUnknown, kUnknownSampleFormat,
dougsteed8d5275f2016-03-12 00:04:30[diff] [blame]293 CHANNEL_LAYOUT_STEREO, 0, EmptyExtraData(),
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]294 EncryptionScheme::kCenc);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]295
xhwang59c321f2015-06-05 23:46:55[diff] [blame]296 InitializeAndExpectResult(config, false);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]297}
298
299// Ensure decoder handles unsupported audio configs without crashing.
300TEST_F(DecryptingAudioDecoderTest, Initialize_UnsupportedAudioConfig) {
xhwang00f69db72015-11-12 07:32:37[diff] [blame]301 SetCdmType(CDM_WITH_DECRYPTOR);
Xiaohan Wangd72ce5d2020-07-14 01:34:43[diff] [blame]302 EXPECT_CALL(*cdm_context_, RegisterEventCB(_)).WillOnce([&](auto cb) {
303 event_cb_ = cb;
304 return std::make_unique<CallbackRegistration>();
305 });
xhwang@chromium.orga3e28672013-03-14 14:54:59[diff] [blame]306 EXPECT_CALL(*decryptor_, InitializeAudioDecoder(_, _))
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]307 .WillOnce(RunOnceCallback<1>(false));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]308
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]309 AudioDecoderConfig config(AudioCodec::kVorbis, kSampleFormatPlanarF32,
chcunningham9812dd82015-10-20 01:42:09[diff] [blame]310 CHANNEL_LAYOUT_STEREO, kSampleRate,
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]311 EmptyExtraData(), EncryptionScheme::kCenc);
xhwang59c321f2015-06-05 23:46:55[diff] [blame]312 InitializeAndExpectResult(config, false);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]313}
314
xhwang00f69db72015-11-12 07:32:37[diff] [blame]315TEST_F(DecryptingAudioDecoderTest, Initialize_CdmWithoutDecryptor) {
316 SetCdmType(CDM_WITHOUT_DECRYPTOR);
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]317 AudioDecoderConfig config(AudioCodec::kVorbis, kSampleFormatPlanarF32,
chcunningham9812dd82015-10-20 01:42:09[diff] [blame]318 CHANNEL_LAYOUT_STEREO, kSampleRate,
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]319 EmptyExtraData(), EncryptionScheme::kCenc);
xhwang59c321f2015-06-05 23:46:55[diff] [blame]320 InitializeAndExpectResult(config, false);
xhwang@chromium.org699b64a2013-06-19 07:36:05[diff] [blame]321}
322
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]323// Test normal decrypt and decode case.
324TEST_F(DecryptingAudioDecoderTest, DecryptAndDecode_Normal) {
325 Initialize();
326 EnterNormalDecodingState();
327}
328
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]329// Test the case where the decryptor errors for mismatched subsamples
330TEST_F(DecryptingAudioDecoderTest, DecryptAndDecode_SubsampleError) {
331 Initialize();
332
333 scoped_refptr<media::DecoderBuffer> mismatched_encrypted_buffer =
334 CreateMismatchedBufferForTest();
335
336 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
337 .WillRepeatedly(
338 RunOnceCallback<1>(Decryptor::kError, Decryptor::AudioFrames()));
339 EXPECT_MEDIA_LOG(
340 HasSubstr("DecryptingAudioDecoder: Subsamples for Buffer do not match"));
341
342 DecodeAndExpect(mismatched_encrypted_buffer, DecoderStatus::Codes::kFailed);
343}
344
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]345// Test the case where the decryptor returns error when doing decrypt and
346// decode.
347TEST_F(DecryptingAudioDecoderTest, DecryptAndDecode_DecodeError) {
348 Initialize();
349
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]350 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]351 .WillRepeatedly(
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]352 RunOnceCallback<1>(Decryptor::kError, Decryptor::AudioFrames()));
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]353 EXPECT_MEDIA_LOG(HasSubstr("DecryptingAudioDecoder: decode error"));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]354
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]355 DecodeAndExpect(encrypted_buffer_, DecoderStatus::Codes::kFailed);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]356}
357
358// Test the case where the decryptor returns multiple decoded frames.
359TEST_F(DecryptingAudioDecoderTest, DecryptAndDecode_MultipleFrames) {
360 Initialize();
361
jrummell@chromium.org47b37a62013-07-10 03:56:10[diff] [blame]362 scoped_refptr<AudioBuffer> frame_a = AudioBuffer::CreateEmptyBuffer(
rileya@chromium.org16ecd5f2014-03-20 05:41:07[diff] [blame]363 config_.channel_layout(),
dalecurtis39a7f932016-07-19 18:34:59[diff] [blame]364 ChannelLayoutToChannelCount(config_.channel_layout()), kSampleRate,
365 kFakeAudioFrameSize, kNoTimestamp);
jrummell@chromium.org47b37a62013-07-10 03:56:10[diff] [blame]366 scoped_refptr<AudioBuffer> frame_b = AudioBuffer::CreateEmptyBuffer(
rileya@chromium.org16ecd5f2014-03-20 05:41:07[diff] [blame]367 config_.channel_layout(),
dalecurtis39a7f932016-07-19 18:34:59[diff] [blame]368 ChannelLayoutToChannelCount(config_.channel_layout()), kSampleRate,
369 kFakeAudioFrameSize, kNoTimestamp);
xhwang@chromium.orgecbb9762012-10-24 22:33:54[diff] [blame]370 decoded_frame_list_.push_back(frame_a);
371 decoded_frame_list_.push_back(frame_b);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]372
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]373 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]374 .WillOnce(RunOnceCallback<1>(Decryptor::kSuccess, decoded_frame_list_));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]375
xhwang@chromium.org5c5e9b72014-06-17 19:21:12[diff] [blame]376 EXPECT_CALL(*this, FrameReady(decoded_frame_));
377 EXPECT_CALL(*this, FrameReady(frame_a));
378 EXPECT_CALL(*this, FrameReady(frame_b));
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]379 DecodeAndExpect(encrypted_buffer_, DecoderStatus::Codes::kOk);
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]380}
381
382// Test the case where the decryptor receives end-of-stream buffer.
383TEST_F(DecryptingAudioDecoderTest, DecryptAndDecode_EndOfStream) {
384 Initialize();
385 EnterNormalDecodingState();
386 EnterEndOfStreamState();
387}
388
xhwang65c23032017-02-17 04:37:42[diff] [blame]389// Test reinitializing decode with a new encrypted config.
390TEST_F(DecryptingAudioDecoderTest, Reinitialize_EncryptedToEncrypted) {
xhwang@chromium.org7c39486c2013-01-06 15:36:09[diff] [blame]391 Initialize();
392
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]393 EXPECT_CALL(*decryptor_, InitializeAudioDecoder(_, _))
394 .Times(AtMost(1))
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]395 .WillOnce(RunOnceCallback<1>(true));
xhwang@chromium.org7c39486c2013-01-06 15:36:09[diff] [blame]396
xhwang@chromium.orgecf16912013-01-12 16:55:22[diff] [blame]397 // The new config is different from the initial config in bits-per-channel,
398 // channel layout and samples_per_second.
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]399 AudioDecoderConfig new_config(AudioCodec::kVorbis, kSampleFormatPlanarS16,
chcunningham9812dd82015-10-20 01:42:09[diff] [blame]400 CHANNEL_LAYOUT_5_1, 88200, EmptyExtraData(),
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]401 EncryptionScheme::kCenc);
xhwang@chromium.orgecf16912013-01-12 16:55:22[diff] [blame]402 EXPECT_NE(new_config.bits_per_channel(), config_.bits_per_channel());
403 EXPECT_NE(new_config.channel_layout(), config_.channel_layout());
404 EXPECT_NE(new_config.samples_per_second(), config_.samples_per_second());
xhwang65c23032017-02-17 04:37:42[diff] [blame]405 ASSERT_TRUE(new_config.is_encrypted());
406
407 ReinitializeConfigChange(new_config);
408 base::RunLoop().RunUntilIdle();
409}
410
411// Test reinitializing decode with a new clear config.
412TEST_F(DecryptingAudioDecoderTest, Reinitialize_EncryptedToClear) {
413 Initialize();
414
415 EXPECT_CALL(*decryptor_, InitializeAudioDecoder(_, _))
416 .Times(AtMost(1))
Jose Lopes109230ec2020-02-24 10:28:12[diff] [blame]417 .WillOnce(RunOnceCallback<1>(true));
xhwang65c23032017-02-17 04:37:42[diff] [blame]418
419 // The new config is different from the initial config in bits-per-channel,
420 // channel layout and samples_per_second.
Dale Curtisbafd563172021-08-13 02:07:47[diff] [blame]421 AudioDecoderConfig new_config(AudioCodec::kVorbis, kSampleFormatPlanarS16,
xhwang65c23032017-02-17 04:37:42[diff] [blame]422 CHANNEL_LAYOUT_5_1, 88200, EmptyExtraData(),
Yuchen Liub33bfc12019-11-08 20:16:12[diff] [blame]423 EncryptionScheme::kUnencrypted);
xhwang65c23032017-02-17 04:37:42[diff] [blame]424 EXPECT_NE(new_config.bits_per_channel(), config_.bits_per_channel());
425 EXPECT_NE(new_config.channel_layout(), config_.channel_layout());
426 EXPECT_NE(new_config.samples_per_second(), config_.samples_per_second());
427 ASSERT_FALSE(new_config.is_encrypted());
xhwang@chromium.orgecf16912013-01-12 16:55:22[diff] [blame]428
rileya@chromium.orgd70157792014-03-17 21:45:03[diff] [blame]429 ReinitializeConfigChange(new_config);
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]430 base::RunLoop().RunUntilIdle();
xhwang@chromium.org7c39486c2013-01-06 15:36:09[diff] [blame]431}
432
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]433// Test the case where the a key is added when the decryptor is in
434// kWaitingForKey state.
435TEST_F(DecryptingAudioDecoderTest, KeyAdded_DuringWaitingForKey) {
436 Initialize();
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]437 EXPECT_MEDIA_LOG(HasSubstr("DecryptingAudioDecoder: no key for key"));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]438 EnterWaitingForKeyState();
439
440 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]441 .WillRepeatedly(
442 RunOnceCallback<1>(Decryptor::kSuccess, decoded_frame_list_));
sergeyu@chromium.org49cea862014-06-11 11:11:50[diff] [blame]443 EXPECT_CALL(*this, FrameReady(decoded_frame_));
Frank Liberato3d157ba2020-09-24 18:19:30[diff] [blame]444 EXPECT_CALL(*this, DecodeDone(IsOkStatus()));
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]445 EXPECT_MEDIA_LOG(
446 HasSubstr("DecryptingAudioDecoder: key added, resuming decode"));
Xiaohan Wangd72ce5d2020-07-14 01:34:43[diff] [blame]447 event_cb_.Run(CdmContext::Event::kHasAdditionalUsableKey);
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]448 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]449}
450
451// Test the case where the a key is added when the decryptor is in
452// kPendingDecode state.
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]453TEST_F(DecryptingAudioDecoderTest, KeyAdded_DuringPendingDecode) {
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]454 Initialize();
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]455 EXPECT_MEDIA_LOG(HasSubstr("DecryptingAudioDecoder: no key for key"));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]456 EnterPendingDecodeState();
457
458 EXPECT_CALL(*decryptor_, DecryptAndDecodeAudio(_, _))
Daniel Cheng868ab272021-02-24 17:07:15[diff] [blame]459 .WillRepeatedly(
460 RunOnceCallback<1>(Decryptor::kSuccess, decoded_frame_list_));
sergeyu@chromium.org49cea862014-06-11 11:11:50[diff] [blame]461 EXPECT_CALL(*this, FrameReady(decoded_frame_));
Frank Liberato3d157ba2020-09-24 18:19:30[diff] [blame]462 EXPECT_CALL(*this, DecodeDone(IsOkStatus()));
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]463 EXPECT_MEDIA_LOG(
464 HasSubstr("DecryptingAudioDecoder: key was added, resuming decode"));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]465 // The audio decode callback is returned after the correct decryption key is
466 // added.
Xiaohan Wangd72ce5d2020-07-14 01:34:43[diff] [blame]467 event_cb_.Run(CdmContext::Event::kHasAdditionalUsableKey);
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]468 std::move(pending_audio_decode_cb_)
Xiaohan Wangc03ce232018-05-30 21:07:58[diff] [blame]469 .Run(Decryptor::kNoKey, Decryptor::AudioFrames());
fdorayf920dcf2016-06-27 17:16:59[diff] [blame]470 base::RunLoop().RunUntilIdle();
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]471}
472
473// Test resetting when the decoder is in kIdle state but has not decoded any
474// frame.
475TEST_F(DecryptingAudioDecoderTest, Reset_DuringIdleAfterInitialization) {
476 Initialize();
477 Reset();
478}
479
480// Test resetting when the decoder is in kIdle state after it has decoded one
481// frame.
482TEST_F(DecryptingAudioDecoderTest, Reset_DuringIdleAfterDecodedOneFrame) {
483 Initialize();
484 EnterNormalDecodingState();
485 Reset();
486}
487
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]488// Test resetting when the decoder is in kPendingDecode state.
489TEST_F(DecryptingAudioDecoderTest, Reset_DuringPendingDecode) {
490 Initialize();
491 EnterPendingDecodeState();
492
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]493 EXPECT_CALL(*this, DecodeDone(HasStatusCode(DecoderStatus::Codes::kAborted)));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]494
495 Reset();
496}
497
498// Test resetting when the decoder is in kWaitingForKey state.
499TEST_F(DecryptingAudioDecoderTest, Reset_DuringWaitingForKey) {
500 Initialize();
Vikram Pasupathyabfd18a2023-01-26 20:56:20[diff] [blame]501 EXPECT_MEDIA_LOG(HasSubstr("DecryptingAudioDecoder: no key for key"));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]502 EnterWaitingForKeyState();
503
Ted Meyera8afec222022-01-12 02:37:27[diff] [blame]504 EXPECT_CALL(*this, DecodeDone(HasStatusCode(DecoderStatus::Codes::kAborted)));
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]505
506 Reset();
507}
508
509// Test resetting when the decoder has hit end of stream and is in
510// kDecodeFinished state.
511TEST_F(DecryptingAudioDecoderTest, Reset_AfterDecodeFinished) {
512 Initialize();
513 EnterNormalDecodingState();
514 EnterEndOfStreamState();
515 Reset();
516}
517
518// Test resetting after the decoder has been reset.
519TEST_F(DecryptingAudioDecoderTest, Reset_AfterReset) {
520 Initialize();
521 EnterNormalDecodingState();
522 Reset();
523 Reset();
524}
525
xhwang@chromium.org97a9ce42012-10-19 10:06:43[diff] [blame]526} // namespace media