141 lines
4.6 KiB
C++
141 lines
4.6 KiB
C++
// Copyright (c) 2013 Google Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "base/strings/string_number_conversions.h"
|
|
#include "media/base/demuxer.h"
|
|
#include "media/base/fixed_encryptor_source.h"
|
|
#include "media/base/media_sample.h"
|
|
#include "media/base/media_stream.h"
|
|
#include "media/base/muxer.h"
|
|
#include "media/base/muxer_options.h"
|
|
#include "media/base/status_test_util.h"
|
|
#include "media/base/stream_info.h"
|
|
#include "media/base/test_data_util.h"
|
|
#include "media/mp4/mp4_muxer.h"
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
|
|
|
using ::testing::Combine;
|
|
using ::testing::Values;
|
|
using ::testing::ValuesIn;
|
|
|
|
namespace {
|
|
const char* kMediaFiles[] = {"bear-1280x720.mp4", "bear-1280x720-av_frag.mp4"};
|
|
|
|
// Encryption constants.
|
|
const char kKeyIdHex[] = "e5007e6e9dcd5ac095202ed3758382cd";
|
|
const char kKeyHex[] = "6fc96fe628a265b13aeddec0bc421f4d";
|
|
const char kPsshHex[] =
|
|
"08011210e5007e6e9dcd5ac095202ed3"
|
|
"758382cd1a0d7769646576696e655f746573742211544553545f"
|
|
"434f4e54454e545f49445f312a025344";
|
|
const uint32 kClearMilliseconds = 1500;
|
|
|
|
} // namespace
|
|
|
|
namespace media {
|
|
|
|
class TestingMuxer : public Muxer {
|
|
public:
|
|
TestingMuxer(const MuxerOptions& options, EncryptorSource* encryptor_source)
|
|
: Muxer(options, encryptor_source) {}
|
|
|
|
virtual Status Initialize() OVERRIDE {
|
|
DVLOG(1) << "Initialize is called.";
|
|
return Status::OK;
|
|
}
|
|
|
|
virtual Status AddSample(const MediaStream* stream,
|
|
scoped_refptr<MediaSample> sample) OVERRIDE {
|
|
DVLOG(1) << "Add Sample: " << sample->ToString();
|
|
DVLOG(2) << "To Stream: " << stream->ToString();
|
|
return Status::OK;
|
|
}
|
|
|
|
virtual Status Finalize() OVERRIDE {
|
|
DVLOG(1) << "Finalize is called.";
|
|
return Status::OK;
|
|
}
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(TestingMuxer);
|
|
};
|
|
|
|
typedef Muxer* CreateMuxerFunc(const std::string& input_file_name,
|
|
EncryptorSource* encryptor_source);
|
|
|
|
Muxer* CreateTestingMuxer(const std::string& input_file_name,
|
|
EncryptorSource* encryptor_source) {
|
|
MuxerOptions options;
|
|
return new TestingMuxer(options, NULL);
|
|
}
|
|
|
|
Muxer* CreateNormalMP4Muxer(const std::string& input_file_name,
|
|
EncryptorSource* encryptor_source) {
|
|
MuxerOptions options;
|
|
options.single_segment = true;
|
|
options.segment_duration = 0.005;
|
|
options.fragment_duration = 0.002;
|
|
options.segment_sap_aligned = true;
|
|
options.fragment_sap_aligned = true;
|
|
options.num_subsegments_per_sidx = 1;
|
|
options.output_file_name = "/tmp/clear_" + input_file_name;
|
|
options.segment_template = "/tmp/template$Number$.m4s";
|
|
options.temp_file_name = "/tmp/tmp.mp4";
|
|
return new mp4::MP4Muxer(options, NULL);
|
|
}
|
|
|
|
Muxer* CreateEncryptionMP4Muxer(const std::string& input_file_name,
|
|
EncryptorSource* encryptor_source) {
|
|
MuxerOptions options;
|
|
options.single_segment = true;
|
|
options.segment_duration = 0.005;
|
|
options.fragment_duration = 0.002;
|
|
options.segment_sap_aligned = true;
|
|
options.fragment_sap_aligned = true;
|
|
options.num_subsegments_per_sidx = 1;
|
|
options.output_file_name = "/tmp/enc_" + input_file_name;
|
|
options.segment_template = "/tmp/template$Number$.m4s";
|
|
options.temp_file_name = "/tmp/tmp.mp4";
|
|
return new mp4::MP4Muxer(options, encryptor_source);
|
|
}
|
|
|
|
class PackagerTest : public ::testing::TestWithParam<
|
|
::std::tr1::tuple<const char*, CreateMuxerFunc*> > {};
|
|
|
|
TEST_P(PackagerTest, Remux) {
|
|
std::string file_name = ::std::tr1::get<0>(GetParam());
|
|
CreateMuxerFunc* CreateMuxer = ::std::tr1::get<1>(GetParam());
|
|
|
|
Demuxer demuxer(GetTestDataFilePath(file_name).value(), NULL);
|
|
ASSERT_OK(demuxer.Initialize());
|
|
|
|
LOG(INFO) << "Num Streams: " << demuxer.streams().size();
|
|
for (int i = 0; i < demuxer.streams().size(); ++i) {
|
|
LOG(INFO) << "Streams " << i << " " << demuxer.streams()[i]->ToString();
|
|
}
|
|
|
|
FixedEncryptorSource encryptor_source(
|
|
kKeyIdHex, kKeyHex, kPsshHex, kClearMilliseconds);
|
|
EXPECT_OK(encryptor_source.Initialize());
|
|
|
|
scoped_ptr<Muxer> muxer(CreateMuxer(file_name, &encryptor_source));
|
|
|
|
ASSERT_OK(muxer->AddStream(demuxer.streams()[0]));
|
|
ASSERT_OK(muxer->Initialize());
|
|
|
|
// Starts remuxing process.
|
|
ASSERT_OK(demuxer.Run());
|
|
|
|
ASSERT_OK(muxer->Finalize());
|
|
}
|
|
|
|
INSTANTIATE_TEST_CASE_P(PackagerE2ETest,
|
|
PackagerTest,
|
|
Combine(ValuesIn(kMediaFiles),
|
|
Values(&CreateTestingMuxer,
|
|
&CreateNormalMP4Muxer,
|
|
&CreateEncryptionMP4Muxer)));
|
|
|
|
} // namespace media
|