media: MediaFoundationProtectionManager

This CL contains MediaFoundationProtectionManager to be used by
MediaFoundationRenderer. MediaFoundationProtectionManager implements
IMFContentProtectionManager and
ABI::Windows::Media::Protection::IMediaProtectionManager

Explainer:
https://github.com/MicrosoftEdge/MSEdgeExplainers/blob/master/Media/MFCdm/explainer.md
Design Doc:
https://docs.google.com/document/d/1kXMdmfsSKU3tcwTxG-wFU-MaEA6sN3by45PCGBV8zJU

Bug: 999747
Change-Id: Ib6125ac6040798b44c3b10a40851eb852880798f
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2088682
Commit-Queue: Frank Li <frankli@microsoft.com>
Reviewed-by: Xiaohan Wang <xhwang@chromium.org>
Cr-Commit-Position: refs/heads/master@{#749270}

media/renderers/BUILD.gn

@@ -107,6 +107,8 @@ if (is_win) {
       "win/media_engine_notify_impl.h",
       "win/media_foundation_audio_stream.cc",
       "win/media_foundation_audio_stream.h",
+      "win/media_foundation_protection_manager.cc",
+      "win/media_foundation_protection_manager.h",
       "win/media_foundation_source_wrapper.cc",
       "win/media_foundation_source_wrapper.h",
       "win/media_foundation_stream_wrapper.cc",

media/renderers/win/media_foundation_protection_manager.cc

+// Copyright 2020 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "media/renderers/win/media_foundation_protection_manager.h"
+
+#include <mferror.h>
+#include <windows.foundation.h>
+
+#include "base/logging.h"
+#include "base/strings/string_piece.h"
+#include "base/win/core_winrt_util.h"
+#include "base/win/scoped_hstring.h"
+#include "base/win/windows_types.h"
+#include "media/base/win/mf_helpers.h"
+
+namespace media {
+
+using Microsoft::WRL::ComPtr;
+
+MediaFoundationProtectionManager::MediaFoundationProtectionManager() = default;
+MediaFoundationProtectionManager::~MediaFoundationProtectionManager() = default;
+
+HRESULT MediaFoundationProtectionManager::RuntimeClassInitialize() {
+  DVLOG(1) << __func__ << ": this=" << this;
+
+  // Init an empty |property_set_| as MFMediaEngine could access it via
+  // |get_Properties| before we populate it within SetPMPServer.
+  base::win::ScopedHString::ResolveCoreWinRTStringDelayload();
+  base::win::ScopedHString property_set_id = base::win::ScopedHString::Create(
+      RuntimeClass_Windows_Foundation_Collections_PropertySet);
+  RETURN_IF_FAILED(
+      base::win::RoActivateInstance(property_set_id.get(), &property_set_));
+  return S_OK;
+}
+
+HRESULT MediaFoundationProtectionManager::SetCdmProxy(IMFCdmProxy* cdm_proxy) {
+  DVLOG(1) << __func__ << ": this=" << this;
+
+  DCHECK(cdm_proxy);
+  cdm_proxy_ = cdm_proxy;
+  ComPtr<ABI::Windows::Media::Protection::IMediaProtectionPMPServer> pmp_server;
+  RETURN_IF_FAILED(cdm_proxy_->GetPMPServer(IID_PPV_ARGS(&pmp_server)));
+  RETURN_IF_FAILED(SetPMPServer(pmp_server.Get()));
+  return S_OK;
+}
+
+HRESULT MediaFoundationProtectionManager::SetPMPServer(
+    ABI::Windows::Media::Protection::IMediaProtectionPMPServer* pmp_server) {
+  DVLOG(1) << __func__ << ": this=" << this;
+
+  DCHECK(pmp_server);
+  ComPtr<ABI::Windows::Foundation::Collections::IMap<HSTRING, IInspectable*>>
+      property_map;
+  RETURN_IF_FAILED(property_set_.As(&property_map));
+  boolean replaced = false;
+  base::win::ScopedHString::ResolveCoreWinRTStringDelayload();
+  // MFMediaEngine uses |pmp_server_key| to get the Protected Media Path (PMP)
+  // server used for playing protected content. This is not currently documented
+  // in MSDN.
+  base::win::ScopedHString pmp_server_key = base::win::ScopedHString::Create(
+      L"Windows.Media.Protection.MediaProtectionPMPServer");
+  RETURN_IF_FAILED(
+      property_map->Insert(pmp_server_key.get(), pmp_server, &replaced));
+  return S_OK;
+}
+
+HRESULT MediaFoundationProtectionManager::BeginEnableContent(
+    IMFActivate* enabler_activate,
+    IMFTopology* topology,
+    IMFAsyncCallback* callback,
+    IUnknown* state) {
+  DVLOG(1) << __func__ << ": this=" << this;
+
+  ComPtr<IUnknown> unknown_object;
+  ComPtr<IMFAsyncResult> async_result;
+  RETURN_IF_FAILED(
+      MFCreateAsyncResult(nullptr, callback, state, &async_result));
+  RETURN_IF_FAILED(
+      enabler_activate->ActivateObject(IID_PPV_ARGS(&unknown_object)));
+
+  // |enabler_type| can be obtained from IMFContentEnabler
+  // (https://docs.microsoft.com/en-us/windows/win32/api/mfidl/nn-mfidl-imfcontentenabler).
+  // If not, try IMediaProtectionServiceRequest
+  // (https://docs.microsoft.com/en-us/uwp/api/windows.media.protection.imediaprotectionservicerequest).
+  GUID enabler_type = GUID_NULL;
+  ComPtr<IMFContentEnabler> content_enabler;
+  if (SUCCEEDED(unknown_object.As(&content_enabler))) {
+    RETURN_IF_FAILED(content_enabler->GetEnableType(&enabler_type));
+  } else {
+    ComPtr<ABI::Windows::Media::Protection::IMediaProtectionServiceRequest>
+        service_request;
+    RETURN_IF_FAILED(unknown_object.As(&service_request));
+    RETURN_IF_FAILED(service_request->get_Type(&enabler_type));
+  }
+
+  if (enabler_type == MFENABLETYPE_MF_RebootRequired) {
+    DLOG(ERROR) << __func__ << ": MF_E_REBOOT_REQUIRED";
+    return MF_E_REBOOT_REQUIRED;
+  } else if (enabler_type == MFENABLETYPE_MF_UpdateRevocationInformation) {
+    DLOG(ERROR) << __func__ << ": MF_E_GRL_VERSION_TOO_LOW";
+    return MF_E_GRL_VERSION_TOO_LOW;
+  } else if (enabler_type == MFENABLETYPE_MF_UpdateUntrustedComponent) {
+    auto hr = HRESULT_FROM_WIN32(ERROR_INVALID_IMAGE_HASH);
+    DLOG(ERROR) << __func__ << ": hr=" << hr;
+    return hr;
+  } else {
+    RETURN_IF_FAILED(cdm_proxy_->ProcessContentEnabler(unknown_object.Get(),
+                                                       async_result.Get()));
+  }
+  return S_OK;
+}
+
+HRESULT MediaFoundationProtectionManager::EndEnableContent(
+    IMFAsyncResult* async_result) {
+  DVLOG(1) << __func__ << ": this=" << this;
+
+  // Get status from the given |async_result| for the purpose of logging.
+  // Returns S_OK as there is no additional work being done here.
+  HRESULT async_status = async_result->GetStatus();
+  if (FAILED(async_status)) {
+    DLOG(ERROR) << "Content enabling failed. hr=" << async_status;
+  } else {
+    DVLOG(2) << "Content enabling succeeded";
+  }
+  return S_OK;
+}
+
+// IMediaProtectionManager implementation
+HRESULT MediaFoundationProtectionManager::add_ServiceRequested(
+    ABI::Windows::Media::Protection::IServiceRequestedEventHandler* handler,
+    EventRegistrationToken* cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::remove_ServiceRequested(
+    EventRegistrationToken cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::add_RebootNeeded(
+    ABI::Windows::Media::Protection::IRebootNeededEventHandler* handler,
+    EventRegistrationToken* cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::remove_RebootNeeded(
+    EventRegistrationToken cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::add_ComponentLoadFailed(
+    ABI::Windows::Media::Protection::IComponentLoadFailedEventHandler* handler,
+    EventRegistrationToken* cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::remove_ComponentLoadFailed(
+    EventRegistrationToken cookie) {
+  return E_NOTIMPL;
+}
+
+HRESULT MediaFoundationProtectionManager::get_Properties(
+    ABI::Windows::Foundation::Collections::IPropertySet** properties) {
+  DVLOG(2) << __func__ << ": this=" << this;
+  if (!properties)
+    return E_POINTER;
+  return property_set_.CopyTo(properties);
+}
+
+}  // namespace media
\ No newline at end of file

media/renderers/win/media_foundation_protection_manager.h

+// Copyright 2020 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef MEDIA_RENDERERS_WIN_MEDIA_FOUNDATION_PROTECTION_MANAGER_H_
+#define MEDIA_RENDERERS_WIN_MEDIA_FOUNDATION_PROTECTION_MANAGER_H_
+
+#include <mfapi.h>
+#include <mfidl.h>
+#include <windows.media.protection.h>
+#include <wrl.h>
+
+#include "media/renderers/win/mf_cdm_proxy.h"
+
+namespace media {
+
+// Implements IMFContentProtectionManager
+// (https://docs.microsoft.com/en-us/windows/win32/api/mfidl/nn-mfidl-imfcontentprotectionmanager)
+// and ABI::Windows::Media::Protection::IMediaProtectionManager
+// (https://docs.microsoft.com/en-us/uwp/api/windows.media.protection.mediaprotectionmanager)
+// required by IMFMediaEngineProtectedContent::SetContentProtectionManager in
+// https://docs.microsoft.com/en-us/windows/win32/api/mfmediaengine/nf-mfmediaengine-imfmediaengineprotectedcontent-setcontentprotectionmanager.
+//
+class MediaFoundationProtectionManager
+    : public Microsoft::WRL::RuntimeClass<
+          Microsoft::WRL::RuntimeClassFlags<
+              Microsoft::WRL::RuntimeClassType::WinRtClassicComMix |
+              Microsoft::WRL::RuntimeClassType::InhibitRoOriginateError>,
+          IMFContentProtectionManager,
+          ABI::Windows::Media::Protection::IMediaProtectionManager> {
+ public:
+  MediaFoundationProtectionManager();
+  ~MediaFoundationProtectionManager() override;
+
+  HRESULT RuntimeClassInitialize();
+  HRESULT SetCdmProxy(IMFCdmProxy* cdm_proxy);
+
+  // IMFContentProtectionManager.
+  IFACEMETHODIMP BeginEnableContent(IMFActivate* enabler_activate,
+                                    IMFTopology* topology,
+                                    IMFAsyncCallback* callback,
+                                    IUnknown* state) override;
+  IFACEMETHODIMP EndEnableContent(IMFAsyncResult* async_result) override;
+
+  // IMediaProtectionManager.
+  // MFMediaEngine can query this interface to invoke get_Properties().
+  IFACEMETHODIMP add_ServiceRequested(
+      ABI::Windows::Media::Protection::IServiceRequestedEventHandler* handler,
+      EventRegistrationToken* cookie) override;
+  IFACEMETHODIMP remove_ServiceRequested(
+      EventRegistrationToken cookie) override;
+  IFACEMETHODIMP add_RebootNeeded(
+      ABI::Windows::Media::Protection::IRebootNeededEventHandler* handler,
+      EventRegistrationToken* cookie) override;
+  IFACEMETHODIMP remove_RebootNeeded(EventRegistrationToken cookie) override;
+  IFACEMETHODIMP add_ComponentLoadFailed(
+      ABI::Windows::Media::Protection::IComponentLoadFailedEventHandler*
+          handler,
+      EventRegistrationToken* cookie) override;
+  IFACEMETHODIMP remove_ComponentLoadFailed(
+      EventRegistrationToken cookie) override;
+  IFACEMETHODIMP get_Properties(
+      ABI::Windows::Foundation::Collections::IPropertySet** value) override;
+
+ protected:
+  Microsoft::WRL::ComPtr<ABI::Windows::Foundation::Collections::IPropertySet>
+      property_set_;
+  Microsoft::WRL::ComPtr<IMFCdmProxy> cdm_proxy_;
+
+  HRESULT SetPMPServer(
+      ABI::Windows::Media::Protection::IMediaProtectionPMPServer* pmp_server);
+};
+
+}  // namespace media
+
+#endif  // MEDIA_RENDERERS_WIN_MEDIA_FOUNDATION_PROTECTION_MANAGER_H_