1 files changed, 71 insertions, 0 deletions
diff --git a/dom/media/platforms/wmf/MFCDMProxy.h b/dom/media/platforms/wmf/MFCDMProxy.h
new file mode 100644
index 0000000000..605755705c
--- /dev/null
+++ b/dom/media/platforms/wmf/MFCDMProxy.h
@@ -0,0 +1,71 @@
+/* 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/. */
+
+#ifndef DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H
+#define DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H
+
+#include <map>
+#include <mfobjects.h>
+#include <unknwn.h>
+#include <windef.h>
+#include <wrl.h>
+
+#include "MFCDMExtra.h"
+#include "nsISupportsImpl.h"
+
+namespace mozilla {
+
+/**
+ * MFCDMProxy wraps a IMFContentDecryptionModule and provides some high level
+ * helper methods in order to allow caller to interact with the wrapped CDM.
+ */
+class MFCDMProxy {
+  NS_INLINE_DECL_REFCOUNTING(MFCDMProxy);
+
+  explicit MFCDMProxy(IMFContentDecryptionModule* aCDM) : mCDM(aCDM) {}
+
+ public:
+  // Return a IMediaProtectionPMPServer from the existing CDM.
+  HRESULT GetPMPServer(REFIID aRiid, LPVOID* aPMPServerOut);
+
+  // Return a IMFInputTrustAuthority for given stream id, the same stream ID
+  // always maps to the same IMFInputTrustAuthority. In addition,
+  // `aContentInitData` is optional initialization data as in
+  // https://www.w3.org/TR/encrypted-media/#initialization-data
+  HRESULT GetInputTrustAuthority(uint32_t aStreamId,
+                                 const uint8_t* aContentInitData,
+                                 uint32_t aContentInitDataSize, REFIID aRiid,
+                                 IUnknown** aInputTrustAuthorityOut);
+
+  // Set IMFContentEnabler to the existing CDM, `aRequest` should be a inherited
+  // class of `IMFContentEnabler`.
+  HRESULT SetContentEnabler(IUnknown* aRequest, IMFAsyncResult* aResult);
+
+  // Notify the CDM on DRM_E_TEE_INVALID_HWDRM_STATE (0x8004cd12), which happens
+  // in cases like OS Sleep. In this case, the CDM should close all sessions
+  // because they are in bad state.
+  void OnHardwareContextReset();
+
+  // TODO : set last key id in order to let CDM use the key IDs information to
+  // perform some optimization.
+
+ private:
+  ~MFCDMProxy() = default;
+
+  Microsoft::WRL::ComPtr<IMFContentDecryptionModule> mCDM;
+
+  // The same ITA is always mapping to the same stream Id.
+  std::map<uint32_t /* stream Id */,
+           Microsoft::WRL::ComPtr<IMFInputTrustAuthority>>
+      mInputTrustAuthorities;
+
+  Microsoft::WRL::ComPtr<IMFTrustedInput> mTrustedInput;
+
+  // TODO : need some events? (Eg. significant playback, error, hardware context
+  // reset)
+};
+
+}  // namespace mozilla
+
+#endif  // DOM_MEDIA_PLATFORM_WMF_MFCDMPROXY_H