tor-browser

ChromiumCDMVideoDecoder.cpp (6869B)

---

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "ChromiumCDMVideoDecoder.h"

#include "ChromiumCDMProxy.h"
#include "GMPService.h"
#include "GMPVideoDecoder.h"
#include "MP4Decoder.h"
#include "VPXDecoder.h"
#include "content_decryption_module.h"
#ifdef MOZ_AV1
#  include "AOMDecoder.h"
#endif

namespace mozilla {

ChromiumCDMVideoDecoder::ChromiumCDMVideoDecoder(
   const GMPVideoDecoderParams& aParams, CDMProxy* aCDMProxy)
   : mCDMParent(aCDMProxy->AsChromiumCDMProxy()->GetCDMParent()),
     mConfig(aParams.mConfig),
     mCrashHelper(aParams.mCrashHelper),
     mGMPThread(GetGMPThread()),
     mImageContainer(aParams.mImageContainer),
     mKnowsCompositor(aParams.mKnowsCompositor) {}

ChromiumCDMVideoDecoder::~ChromiumCDMVideoDecoder() = default;

static uint32_t ToCDMH264Profile(uint8_t aProfile) {
 switch (aProfile) {
   case 66:
     return cdm::VideoCodecProfile::kH264ProfileBaseline;
   case 77:
     return cdm::VideoCodecProfile::kH264ProfileMain;
   case 88:
     return cdm::VideoCodecProfile::kH264ProfileExtended;
   case 100:
     return cdm::VideoCodecProfile::kH264ProfileHigh;
   case 110:
     return cdm::VideoCodecProfile::kH264ProfileHigh10;
   case 122:
     return cdm::VideoCodecProfile::kH264ProfileHigh422;
   case 144:
     return cdm::VideoCodecProfile::kH264ProfileHigh444Predictive;
 }
 return cdm::VideoCodecProfile::kUnknownVideoCodecProfile;
}

#ifdef MOZ_AV1
static uint32_t ToCDMAV1Profile(uint8_t aProfile) {
 switch (aProfile) {
   case 0:
     return cdm::VideoCodecProfile::kAv1ProfileMain;
   case 1:
     return cdm::VideoCodecProfile::kAv1ProfileHigh;
   case 2:
     return cdm::VideoCodecProfile::kAv1ProfilePro;
   default:
     return cdm::VideoCodecProfile::kUnknownVideoCodecProfile;
 }
}
#endif

RefPtr<MediaDataDecoder::InitPromise> ChromiumCDMVideoDecoder::Init() {
 if (!mCDMParent) {
   // Must have failed to get the CDMParent from the ChromiumCDMProxy
   // in our constructor; the MediaKeys must have shut down the CDM
   // before we had a chance to start up the decoder.
   return InitPromise::CreateAndReject(NS_ERROR_DOM_MEDIA_FATAL_ERR, __func__);
 }

 gmp::CDMVideoDecoderConfig config;
 if (MP4Decoder::IsH264(mConfig.mMimeType)) {
   config.mCodec() = cdm::VideoCodec::kCodecH264;
   config.mProfile() =
       ToCDMH264Profile(mConfig.mExtraData->SafeElementAt(1, 0));
   config.mExtraData() = mConfig.mExtraData->Clone();
   mConvertToAnnexB = true;
#ifdef MOZ_AV1
 } else if (AOMDecoder::IsAV1(mConfig.mMimeType)) {
   AOMDecoder::AV1SequenceInfo seqInfo;
   MediaResult seqHdrResult;
   AOMDecoder::TryReadAV1CBox(mConfig.mExtraData, seqInfo, seqHdrResult);
   config.mCodec() = cdm::VideoCodec::kCodecAv1;
   config.mProfile() = NS_SUCCEEDED(seqHdrResult.Code())
                           ? ToCDMAV1Profile(seqInfo.mProfile)
                           : cdm::VideoCodecProfile::kUnknownVideoCodecProfile;
#endif
 } else if (VPXDecoder::IsVP8(mConfig.mMimeType)) {
   config.mCodec() = cdm::VideoCodec::kCodecVp8;
   config.mProfile() = cdm::VideoCodecProfile::kProfileNotNeeded;
 } else if (VPXDecoder::IsVP9(mConfig.mMimeType)) {
   config.mCodec() = cdm::VideoCodec::kCodecVp9;
   config.mProfile() = cdm::VideoCodecProfile::kProfileNotNeeded;
 } else {
   return MediaDataDecoder::InitPromise::CreateAndReject(
       NS_ERROR_DOM_MEDIA_FATAL_ERR, __func__);
 }
 config.mImageWidth() = mConfig.mImage.width;
 config.mImageHeight() = mConfig.mImage.height;
 config.mEncryptionScheme() = cdm::EncryptionScheme::kUnencrypted;
 switch (mConfig.mCrypto.mCryptoScheme) {
   case CryptoScheme::None:
     break;
   case CryptoScheme::Cenc:
     config.mEncryptionScheme() = cdm::EncryptionScheme::kCenc;
     break;
   case CryptoScheme::Cbcs:
   case CryptoScheme::Cbcs_1_9:
     config.mEncryptionScheme() = cdm::EncryptionScheme::kCbcs;
     break;
   default:
     MOZ_ASSERT_UNREACHABLE("Should not have unrecognized encryption type");
     break;
 }

 RefPtr<gmp::ChromiumCDMParent> cdm = mCDMParent;
 VideoInfo info = mConfig;
 RefPtr<layers::ImageContainer> imageContainer = mImageContainer;
 RefPtr<layers::KnowsCompositor> knowsCompositor = mKnowsCompositor;
 return InvokeAsync(mGMPThread, __func__,
                    [cdm, config, info, imageContainer, knowsCompositor]() {
                      return cdm->InitializeVideoDecoder(
                          config, info, imageContainer, knowsCompositor);
                    });
}

nsCString ChromiumCDMVideoDecoder::GetDescriptionName() const {
 return "chromium cdm video decoder"_ns;
}

nsCString ChromiumCDMVideoDecoder::GetCodecName() const {
 if (MP4Decoder::IsH264(mConfig.mMimeType)) {
   return "h264"_ns;
 }
#ifdef MOZ_AV1
 if (AOMDecoder::IsAV1(mConfig.mMimeType)) {
   return "av1"_ns;
 }
#endif
 if (VPXDecoder::IsVP8(mConfig.mMimeType)) {
   return "vp8"_ns;
 }
 if (VPXDecoder::IsVP9(mConfig.mMimeType)) {
   return "vp9"_ns;
 }
 return "unknown"_ns;
}

MediaDataDecoder::ConversionRequired ChromiumCDMVideoDecoder::NeedsConversion()
   const {
 return mConvertToAnnexB ? ConversionRequired::kNeedAnnexB
                         : ConversionRequired::kNeedNone;
}

RefPtr<MediaDataDecoder::DecodePromise> ChromiumCDMVideoDecoder::Decode(
   MediaRawData* aSample) {
 RefPtr<gmp::ChromiumCDMParent> cdm = mCDMParent;
 RefPtr<MediaRawData> sample = aSample;
 return InvokeAsync(mGMPThread, __func__, [cdm, sample]() {
   return cdm->DecryptAndDecodeFrame(sample);
 });
}

RefPtr<MediaDataDecoder::FlushPromise> ChromiumCDMVideoDecoder::Flush() {
 MOZ_ASSERT(mCDMParent);
 RefPtr<gmp::ChromiumCDMParent> cdm = mCDMParent;
 return InvokeAsync(mGMPThread, __func__,
                    [cdm]() { return cdm->FlushVideoDecoder(); });
}

RefPtr<MediaDataDecoder::DecodePromise> ChromiumCDMVideoDecoder::Drain() {
 MOZ_ASSERT(mCDMParent);
 RefPtr<gmp::ChromiumCDMParent> cdm = mCDMParent;
 return InvokeAsync(mGMPThread, __func__, [cdm]() { return cdm->Drain(); });
}

RefPtr<ShutdownPromise> ChromiumCDMVideoDecoder::Shutdown() {
 if (!mCDMParent) {
   // Must have failed to get the CDMParent from the ChromiumCDMProxy
   // in our constructor; the MediaKeys must have shut down the CDM
   // before we had a chance to start up the decoder.
   return ShutdownPromise::CreateAndResolve(true, __func__);
 }
 RefPtr<gmp::ChromiumCDMParent> cdm = mCDMParent;
 return InvokeAsync(mGMPThread, __func__,
                    [cdm]() { return cdm->ShutdownVideoDecoder(); });
}

}  // namespace mozilla