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
|
/* 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 "MFCDMProxy.h"
#include "MFCDMParent.h"
#include "MFMediaEngineUtils.h"
namespace mozilla {
using Microsoft::WRL::ComPtr;
#define LOG(msg, ...) \
MOZ_LOG(gMFMediaEngineLog, LogLevel::Debug, \
("MFCDMProxy=%p, " msg, this, ##__VA_ARGS__))
MFCDMProxy::MFCDMProxy(IMFContentDecryptionModule* aCDM, uint64_t aCDMParentId)
: mCDM(aCDM), mCDMParentId(aCDMParentId) {
LOG("MFCDMProxy created, created by %" PRId64 " MFCDMParent", mCDMParentId);
}
MFCDMProxy::~MFCDMProxy() { LOG("MFCDMProxy destroyed"); }
void MFCDMProxy::Shutdown() {
if (mTrustedInput) {
mTrustedInput = nullptr;
}
for (auto& inputAuthorities : mInputTrustAuthorities) {
SHUTDOWN_IF_POSSIBLE(inputAuthorities.second);
}
mInputTrustAuthorities.clear();
if (auto* parent = MFCDMParent::GetCDMById(mCDMParentId)) {
parent->ShutdownCDM();
}
mCDM = nullptr;
LOG("MFCDMProxy Shutdowned");
}
HRESULT MFCDMProxy::GetPMPServer(REFIID aRiid, LPVOID* aPMPServerOut) {
ComPtr<IMFGetService> cdmServices;
RETURN_IF_FAILED(mCDM.As(&cdmServices));
RETURN_IF_FAILED(cdmServices->GetService(MF_CONTENTDECRYPTIONMODULE_SERVICE,
aRiid, aPMPServerOut));
return S_OK;
}
HRESULT MFCDMProxy::GetInputTrustAuthority(uint32_t aStreamId,
const uint8_t* aContentInitData,
uint32_t aContentInitDataSize,
REFIID aRiid,
IUnknown** aInputTrustAuthorityOut) {
if (mInputTrustAuthorities.count(aStreamId)) {
RETURN_IF_FAILED(
mInputTrustAuthorities[aStreamId].CopyTo(aInputTrustAuthorityOut));
return S_OK;
}
if (!mTrustedInput) {
RETURN_IF_FAILED(mCDM->CreateTrustedInput(
aContentInitData, aContentInitDataSize, &mTrustedInput));
LOG("Created a trust input for stream %u", aStreamId);
}
// GetInputTrustAuthority takes IUnknown* as the output. Using other COM
// interface will have a v-table mismatch issue.
ComPtr<IUnknown> unknown;
RETURN_IF_FAILED(
mTrustedInput->GetInputTrustAuthority(aStreamId, aRiid, &unknown));
ComPtr<IMFInputTrustAuthority> inputTrustAuthority;
RETURN_IF_FAILED(unknown.As(&inputTrustAuthority));
RETURN_IF_FAILED(unknown.CopyTo(aInputTrustAuthorityOut));
mInputTrustAuthorities[aStreamId] = inputTrustAuthority;
return S_OK;
}
HRESULT MFCDMProxy::SetContentEnabler(IUnknown* aRequest,
IMFAsyncResult* aResult) {
LOG("SetContentEnabler");
ComPtr<IMFContentEnabler> contentEnabler;
RETURN_IF_FAILED(aRequest->QueryInterface(IID_PPV_ARGS(&contentEnabler)));
return mCDM->SetContentEnabler(contentEnabler.Get(), aResult);
}
void MFCDMProxy::OnHardwareContextReset() {
LOG("OnHardwareContextReset");
// Hardware context reset happens, all the crypto sessions are in invalid
// states. So drop everything here.
mTrustedInput.Reset();
mInputTrustAuthorities.clear();
}
#undef LOG
} // namespace mozilla
|
