1 files changed, 42 insertions, 0 deletions
diff --git a/media/wmf-clearkey/WMFClearKeyContentEnabler.h b/media/wmf-clearkey/WMFClearKeyContentEnabler.h
new file mode 100644
index 0000000000..ebaf5be7ad
--- /dev/null
+++ b/media/wmf-clearkey/WMFClearKeyContentEnabler.h
@@ -0,0 +1,42 @@
+/* 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_CLEARKEY_WMFCLEARKEYCONTENTENABLER_H
+#define DOM_MEDIA_PLATFORM_WMF_CLEARKEY_WMFCLEARKEYCONTENTENABLER_H
+
+#include <wrl.h>
+#include <wrl/client.h>
+
+#include "MFCDMExtra.h"
+
+namespace mozilla {
+
+// This class is used to return correct enable type.
+class WMFClearKeyContentEnabler
+    : public Microsoft::WRL::RuntimeClass<
+          Microsoft::WRL::RuntimeClassFlags<Microsoft::WRL::ClassicCom>,
+          IMFContentEnabler, Microsoft::WRL::FtmBase> {
+ public:
+  WMFClearKeyContentEnabler() = default;
+  ~WMFClearKeyContentEnabler() = default;
+  WMFClearKeyContentEnabler(const WMFClearKeyContentEnabler&) = delete;
+  WMFClearKeyContentEnabler& operator=(const WMFClearKeyContentEnabler&) =
+      delete;
+
+  HRESULT RuntimeClassInitialize();
+
+  // IMFContentEnabler
+  STDMETHODIMP AutomaticEnable() override;
+  STDMETHODIMP Cancel() override;
+  STDMETHODIMP GetEnableData(BYTE** aData, DWORD* aDataSize) override;
+  STDMETHODIMP GetEnableType(GUID* aType) override;
+  STDMETHODIMP GetEnableURL(LPWSTR* aUrl, DWORD* aUrlSize,
+                            MF_URL_TRUST_STATUS* aTrustStatus) override;
+  STDMETHODIMP IsAutomaticSupported(BOOL* automatic) override;
+  STDMETHODIMP MonitorEnable() override;
+};
+
+}  // namespace mozilla
+
+#endif  // DOM_MEDIA_PLATFORM_WMF_CLEARKEY_WMFCLEARKEYCONTENTENABLER_H