1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103


media / cdm / win / test / mock_media_protection_pmp_server.cc [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "media/cdm/win/test/mock_media_protection_pmp_server.h"

#include <mfapi.h>
#include <mferror.h>

#include "base/notreached.h"
#include "media/base/win/mf_helpers.h"

namespace media {

using Microsoft::WRL::ComPtr;
using Microsoft::WRL::MakeAndInitialize;

MockMediaProtectionPMPServer::MockMediaProtectionPMPServer() = default;

MockMediaProtectionPMPServer::~MockMediaProtectionPMPServer() {
  DVLOG_FUNC(1);
}

HRESULT MockMediaProtectionPMPServer::RuntimeClassInitialize(
    ABI::Windows::Foundation::Collections::IPropertySet* property_pmp) {
  DVLOG_FUNC(1);
  property_pmp_ = property_pmp;

  // If we create a PMP media session with `MFPMPSESSION_IN_PROCESS` flag, the
  // Protected Media Path will exist in process. This way we can use the
  // in-process PMP server and PMP host for the rest of Media Foundation
  // pipelines.
  RETURN_IF_FAILED(MFCreatePMPMediaSession(MFPMPSESSION_IN_PROCESS, nullptr,
                                           &media_session_, nullptr));
  RETURN_IF_FAILED(MFGetService(media_session_.Get(), MF_PMP_SERVER_CONTEXT,
                                IID_PPV_ARGS(&pmp_server_)));
  RETURN_IF_FAILED(MFGetService(media_session_.Get(), MF_PMP_SERVICE,
                                IID_PPV_ARGS(&pmp_host_)));

  return S_OK;
}

// IInspectable
STDMETHODIMP MockMediaProtectionPMPServer::GetIids(_COM_Outptr_ ULONG* iidCount,
                                                   _COM_Outptr_ IID** iids) {
  DVLOG_FUNC(3);

  // API not used.
  NOTIMPLEMENTED();
  return E_NOTIMPL;
}

STDMETHODIMP MockMediaProtectionPMPServer::GetRuntimeClassName(
    _COM_Outptr_ HSTRING* className) {
  DVLOG_FUNC(3);

  // API not used.
  NOTIMPLEMENTED();
  return E_NOTIMPL;
}

STDMETHODIMP MockMediaProtectionPMPServer::GetTrustLevel(
    _COM_Outptr_ TrustLevel* trustLevel) {
  DVLOG_FUNC(3);

  // API not used.
  NOTIMPLEMENTED();
  return E_NOTIMPL;
}

// IMediaProtectionPMPServer
STDMETHODIMP MockMediaProtectionPMPServer::get_Properties(
    _COM_Outptr_ ABI::Windows::Foundation::Collections::IPropertySet**
        ppProperties) {
  DVLOG_FUNC(1);
  RETURN_IF_FAILED(property_pmp_.CopyTo(ppProperties));
  return S_OK;
}

// IMFGetService
STDMETHODIMP MockMediaProtectionPMPServer::GetService(
    __RPC__in REFGUID guid_service,
    __RPC__in REFIID riid,
    __RPC__deref_out_opt LPVOID* object) {
  DVLOG_FUNC(1);
  if (!object) {
    return E_POINTER;
  }

  *object = nullptr;

  if (guid_service == MF_PMP_SERVER_CONTEXT) {
    RETURN_IF_FAILED(pmp_server_.CopyTo(riid, object));
  } else if (guid_service == MF_PMP_SERVICE && riid == IID_IMFPMPHost) {
    RETURN_IF_FAILED(pmp_host_.CopyTo(riid, object));
  } else {
    return MF_E_UNSUPPORTED_SERVICE;
  }

  return S_OK;
}

}  // namespace media